yet another db_prefs patch
[squirrelmail.git] / functions / db_prefs.php
index 6a606db74eebf50b652d392ec6a0a32a76c55d1b..7bc0800a6e956525d21fbc879c3a7e9c2e78e24c 100644 (file)
 <?php
-   /**
-    **  db_prefs.php
-    **
-    **  This contains functions for manipulating user preferences
-    **  stored in a database, accessed though the Pear DB layer.
-    **
-    **  To use this instead of the regular prefs.php, create a 
-    **  database as described below, and replace prefs.php
-    **  with this file.
-    **
-    **  Database:
-    **  ---------
-    **
-    **  The preferences table should have tree columns:
-    **     username   char  \  primary
-    **     prefkey    char  /  key
-    **     prefval    blob
-    **
-    **    CREATE TABLE userprefs (user CHAR(32) NOT NULL DEFAULT '', 
-    **                            prefkey CHAR(64) NOT NULL DEFAULT '', 
-    **                            prefval BLOB NOT NULL DEFAULT '', 
-    **                            primary key (user,prefkey));
-    **
-    **  Configuration of databasename, username and password is done
-    **  by changing $DSN below.
-    **
-    **  $Id$
-    **/
-
-   if (defined('prefs_php'))
-       return;
-   define('prefs_php', true);
-
-   require_once('DB.php');
-
-   class dbPrefs {
-      var $DSN   = 'mysql://user@host/database';
-      var $table = 'userprefs';
-      
-      var $dbh   = NULL;
-      var $error = NULL;
-
-      var $default = Array('chosen_theme'      => '../themes/default_theme.php',
-                          'show_html_default' => '0');
-      
-      function dbPrefs() {
-        $this->open();
-      }
-      
-      function open() {
-        if(isset($this->dbh)) return true;
-        $dbh = DB::connect($this->DSN, true);
-        
-        if(DB::isError($dbh) || DB::isWarning($dbh)) {
-           $this->error = DB::errorMessage($dbh);
-           return false;
-        }
-        
-        $this->dbh = $dbh;
-        return true;
-      }
-      
-
-      function failQuery($res = NULL) {
-        if($res == NULL) {
-           printf(_("Preference database error (%s). Exiting abnormally"),
-                  $this->error);
-        } else {
-           printf(_("Preference database error (%s). Exiting abnormally"),
-                  DB::errorMessage($res));
-        }
-        exit;
-      }
-
-
-      function getKey($user, $key) {
-        $this->open();
-        $query = sprintf("SELECT prefval FROM %s ".
-                         "WHERE user='%s' AND prefkey='%s'",
-                         $this->table, 
-                         $this->dbh->quoteString($user),
-                         $this->dbh->quoteString($key));
-
-        $res = $this->dbh->query($query);
-        if(DB::isError($res))
-           $this->failQuery($res);
-
-        if($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
-           return $row['prefval'];
-        } else {
-           if(isset($this->default[$key])) {
-              return $this->default[$key];
-           } else {
-              return '';
-           }
-        }
-
-        return '';
-      }
-
-      function deleteKey($user, $key) {
-        $this->open();
-        $query = sprintf("DELETE FROM %s WHERE user='%s' AND prefkey='%s'",
-                         $this->table, 
-                         $this->dbh->quoteString($user),
-                         $this->dbh->quoteString($key));
-
-        $res = $this->dbh->simpleQuery($query);
-         if(DB::isError($res))
-           $this->failQuery($res);
-
-        if(substr($key, 0, 9) == 'highlight') {
-           $this->renumberHighlightList($user);
-        }
-
-        return true;
-      }
-
-      function setKey($user, $key, $value) {
-        $this->open();
-        $query = sprintf("REPLACE INTO %s (user,prefkey,prefval) ".
-                         "VALUES('%s','%s','%s')",
-                         $this->table, 
-                         $this->dbh->quoteString($user),
-                         $this->dbh->quoteString($key),
-                         $this->dbh->quoteString($value));
-
-         $res = $this->dbh->simpleQuery($query);
-         if(DB::isError($res)) 
-           $this->failQuery($res);
-
-        return true;
-      }
-
-      
-      /**
-       ** When a highlight option is deleted the preferences module
-       ** must renumber the list.  This should be done somewhere else,
-       ** but it is not, so....  
-       **/
-      function renumberHighlightList($user) {
-        $this->open();
-        $query = sprintf("SELECT * FROM %s WHERE user='%s' ".
-                         "AND prefkey LIKE 'highlight%%' ORDER BY prefkey",
-                         $this->table, 
-                         $this->dbh->quoteString($user));
-
-        $res = $this->dbh->query($query);
-         if(DB::isError($res))
-           $this->failQuery($res);
-
-        // Store old data in array
-        $rows = Array();
-        while($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) 
-           $rows[] = $row;
-
-        // Renumber keys of old data
-        $hilinum = 0;
-        for($i = 0; $i < count($rows) ; $i++) {
-           $oldkey = $rows[$i]['prefkey'];
-           $newkey = substr($oldkey, 0, 9) . $hilinum;
-           $hilinum++;
-
-           if($oldkey != $newkey) {
-              $query = sprintf("UPDATE %s SET prefkey='%s' WHERE user='%s' ".
-                               "AND prefkey='%s'", 
-                               $this->table,
-                               $this->dbh->quoteString($newkey),
-                               $this->dbh->quoteString($user),
-                               $this->dbh->quoteString($oldkey));
-              
-              $res = $this->dbh->simpleQuery($query);
-              if(DB::isError($res))
-                 $this->failQuery($res);
-           }
-        }
-
-        return;
-      }
-
-   } // end class dbPrefs
-
-
-   /** returns the value for the pref $string **/
-   function getPref($data_dir, $username, $string, $default ) {
-      $db = new dbPrefs;
-      if(isset($db->error)) {
-        printf(_("Preference database error (%s). Exiting abnormally"),
-               $db->error);
-        exit;
-      }
-
-      return $db->getKey($username, $string);
-   }
-
-   /** Remove the pref $string **/
-   function removePref($data_dir, $username, $string) {
-      $db = new dbPrefs;
-      if(isset($db->error)) $db->failQuery();
-
-      $db->deleteKey($username, $string);
-      return;
-   }
+
+/*
+ * db_prefs.php
+ *
+ * Copyright (c) 1999-2002 The SquirrelMail Project Team
+ * Licensed under the GNU GPL. For full terms see the file COPYING.
+ *
+ * This contains functions for manipulating user preferences
+ * stored in a database, accessed though the Pear DB layer.
+ *
+ * To use this instead of the regular prefs.php, create a
+ * database as described below, and replace prefs.php
+ * with this file.
+ *
+ * Database:
+ * ---------
+ *
+ * The preferences table should have tree columns:
+ *    username   char  \  primary
+ *    prefkey    char  /  key
+ *    prefval    blob
+ *
+ *   CREATE TABLE userprefs (user CHAR(128) NOT NULL DEFAULT '',
+ *                           prefkey CHAR(64) NOT NULL DEFAULT '',
+ *                           prefval BLOB NOT NULL DEFAULT '',
+ *                           primary key (user,prefkey));
+ *
+ * Configuration of databasename, username and password is done
+ * by changing $DSN below.
+ *
+ * $Id$
+ */
+
+require_once('DB.php');
+
+global $prefs_are_cached, $prefs_cache;
+
+if ( !session_is_registered('prefs_are_cached') ||
+     !isset( $prefs_cache) ||
+     !is_array( $prefs_cache) ||
+     substr( phpversion(), 0, 3 ) == '4.1' ) {
+    $prefs_are_cached = false;
+    $prefs_cache = array();
+}
+
+function cachePrefValues($username) {
+    global $prefs_are_cached, $prefs_cache;
+
+    if ($prefs_are_cached) {
+        return;
+    }
+
+    session_unregister('prefs_cache');
+    session_unregister('prefs_are_cached');
+
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        printf( _("Preference database error (%s). Exiting abnormally"),
+              $db->error);
+        exit;
+    }
+
+    $db->fillPrefsCache($username);
+    if (isset($db->error)) {
+        printf( _("Preference database error (%s). Exiting abnormally"),
+              $db->error);
+        exit;
+    }
+
+    $prefs_are_cached = true;
+
+    session_register('prefs_cache');
+    session_register('prefs_are_cached');
+}
+
+class dbPrefs {
+    var $DSN   = 'mysql://user:passwd@host/database';
+    var $table = 'userprefs';
+
+    var $dbh   = NULL;
+    var $error = NULL;
+
+    var $default = Array('chosen_theme' => '../themes/default_theme.php',
+                         'show_html_default' => '0');
+
+    function open() {
+        if(isset($this->dbh)) {
+            return true;
+        }
+        $dbh = DB::connect($this->DSN, true);
+
+        if(DB::isError($dbh) || DB::isWarning($dbh)) {
+            $this->error = DB::errorMessage($dbh);
+            return false;
+        }
+
+        $this->dbh = $dbh;
+        return true;
+    }
+
+    function failQuery($res = NULL) {
+        if($res == NULL) {
+            printf(_("Preference database error (%s). Exiting abnormally"),
+                  $this->error);
+        } else {
+            printf(_("Preference database error (%s). Exiting abnormally"),
+                  DB::errorMessage($res));
+        }
+        exit;
+    }
+
+
+    function getKey($user, $key, $default = '') {
+        global $prefs_cache;
+
+        cachePrefValues($user);
+
+        if (isset($prefs_cache[$key])) {
+            return $prefs_cache[$key];
+        } else {
+            if (isset($this->default[$key])) {
+                return $this->default[$key];
+            } else {
+                return $default;
+            }
+        }
+    }
+
+    function deleteKey($user, $key) {
+        global $prefs_cache;
+
+        $this->open();
+        $query = sprintf("DELETE FROM %s WHERE user='%s' AND prefkey='%s'",
+                         $this->table,
+                         $this->dbh->quoteString($user),
+                         $this->dbh->quoteString($key));
+
+        $res = $this->dbh->simpleQuery($query);
+        if(DB::isError($res)) {
+            $this->failQuery($res);
+        }
+
+        unset($prefs_cache[$key]);
+
+        if(substr($key, 0, 9) == 'highlight') {
+            $this->renumberHighlightList($user);
+        }
+
+        return true;
+    }
+
+    function setKey($user, $key, $value) {
+        $this->open();
+        $query = sprintf("REPLACE INTO %s (user,prefkey,prefval) ".
+                         "VALUES('%s','%s','%s')",
+                         $this->table,
+                         $this->dbh->quoteString($user),
+                         $this->dbh->quoteString($key),
+                         $this->dbh->quoteString($value));
+
+        $res = $this->dbh->simpleQuery($query);
+        if(DB::isError($res)) {
+            $this->failQuery($res);
+        }
+
+        return true;
+    }
+
+    function fillPrefsCache($user) {
+        global $prefs_cache;
+
+        $this->open();
+
+        $prefs_cache = array();
+        $query = sprintf("SELECT prefkey, prefval FROM %s ".
+                         "WHERE user = '%s'",
+                         $this->table,
+                         $this->dbh->quoteString($user));
+        $res = $this->dbh->query($query);
+        if (DB::isError($res)) {
+            $this->failQuery($res);
+        }
+
+        while ($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
+            $prefs_cache[$row['prefkey']] = $row['prefval'];
+        }
+    }
+
+    /*
+     * When a highlight option is deleted the preferences module
+     * must renumber the list.  This should be done somewhere else,
+     * but it is not, so....
+     */
+    function renumberHighlightList($user) {
+        $this->open();
+        $query = sprintf("SELECT * FROM %s WHERE user='%s' ".
+                         "AND prefkey LIKE 'highlight%%' ORDER BY prefkey",
+                         $this->table,
+                         $this->dbh->quoteString($user));
+
+        $res = $this->dbh->query($query);
+        if(DB::isError($res)) {
+            $this->failQuery($res);
+        }
+
+        /* Store old data in array */
+        $rows = Array();
+        while($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
+            $rows[] = $row;
+        }
+
+        /* Renumber keys of old data */
+        $hilinum = 0;
+        for($i = 0; $i < count($rows) ; $i++) {
+            $oldkey = $rows[$i]['prefkey'];
+            $newkey = substr($oldkey, 0, 9) . $hilinum;
+            $hilinum++;
+
+            if($oldkey != $newkey) {
+                $query = sprintf("UPDATE %s SET prefkey='%s' ".
+                                 "WHERE user ='%s' AND prefkey='%s'",
+                                 $this->table,
+                                 $this->dbh->quoteString($newkey),
+                                 $this->dbh->quoteString($user),
+                                 $this->dbh->quoteString($oldkey));
+
+                $res = $this->dbh->simpleQuery($query);
+                if(DB::isError($res)) {
+                    $this->failQuery($res);
+                }
+            }
+        }
+
+        return;
+    }
+
+} /* end class dbPrefs */
+
+
+/* returns the value for the pref $string */
+function getPref($data_dir, $username, $string, $default = '') {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        printf( _("Preference database error (%s). Exiting abnormally"),
+              $db->error);
+        exit;
+    }
+
+    return $db->getKey($username, $string, $default);
+}
+
+/* Remove the pref $string */
+function removePref($data_dir, $username, $string) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    $db->deleteKey($username, $string);
+    return;
+}
+
+/* sets the pref, $string, to $set_to */
+function setPref($data_dir, $username, $string, $set_to) {
+    global $prefs_cache;
+
+    if (isset($prefs_cache[$string]) && ($prefs_cache[$string] == $set_to)) {
+        return;
+    }
+
+    if ($set_to == '') {
+        removePref($data_dir, $username, $string);
+        return;
+    }
+
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    $db->setKey($username, $string, $set_to);
+    $prefs_cache[$string] = $set_to;
+    assert_options(ASSERT_ACTIVE, 1);
+    assert_options(ASSERT_BAIL, 1);
+    assert ('$set_to == $prefs_cache[$string]');
+
+    return;
+}
+
+/* This checks if the prefs are available */
+function checkForPrefs($data_dir, $username) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+}
+
+/* Writes the Signature */
+function setSig($data_dir, $username, $string) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    $db->setKey($username, '___signature___', $string);
+    return;
+}
+
+/* Gets the signature */
+function getSig($data_dir, $username) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    return $db->getKey($username, '___signature___');
+}
+
+/* Hashing functions */
+
+function getHashedFile($username, $dir, $datafile, $hash_search = true) {
+    global $dir_hash_level;
+
+    /* Remove trailing slash from $dir if found */
+    if (substr($dir, -1) == '/') {
+        $dir = substr($dir, 0, strlen($dir) - 1);
+    }
+
+    /* Compute the hash for this user and extract the hash directories. */
+    $hash_dirs = computeHashDirs($username);
+
+    /* First, get and make sure the full hash directory exists. */
+    $real_hash_dir = getHashedDir($username, $dir, $hash_dirs);
+     
+    /* Set the value of our real data file. */
+    $result = "$real_hash_dir/$datafile";
+
+    /* Check for this file in the real hash directory. */
+    if ($hash_search && !@file_exists($result)) {
+        /* First check the base directory, the most common location. */
+        if (@file_exists("$dir/$datafile")) {
+            rename("$dir/$datafile", $result);
+
+        /* Then check the full range of possible hash directories. */
+        } else {
+            $check_hash_dir = $dir;
+            for ($h = 0; $h < 4; ++$h) {
+                $check_hash_dir .= '/' . $hash_dirs[$h];
+                if (@is_readable("$check_hash_dir/$datafile")) {
+                    rename("$check_hash_dir/$datafile", $result);
+                    break;
+                }
+            }
+        }
+    }
+
+    /* Return the full hashed datafile path. */
+    return ($result);
+}
+
+function getHashedDir($username, $dir, $hash_dirs = '') {
+    global $dir_hash_level;
+
+    /* Remove trailing slash from $dir if found */
+    if (substr($dir, -1) == '/') {
+        $dir = substr($dir, 0, strlen($dir) - 1);
+    }
    
-   /** sets the pref, $string, to $set_to **/
-   function setPref($data_dir, $username, $string, $set_to) {
-      $db = new dbPrefs;
-      if(isset($db->error))
-        $db->failQuery();
-
-      $db->setKey($username, $string, $set_to);
-      return;
-   }
-
-   /** This checks if the prefs are available **/
-   function checkForPrefs($data_dir, $username) {
-      $db = new dbPrefs;
-      if(isset($db->error))
-        $db->failQuery();
-   }
-
-   /** Writes the Signature **/
-   function setSig($data_dir, $username, $string) {
-      $db = new dbPrefs;
-      if(isset($db->error)) 
-        $db->failQuery();
-
-      $db->setKey($username, "___signature___", $string);
-      return;
-   }
-
-   /** Gets the signature **/
-   function getSig($data_dir, $username) {
-      $db = new dbPrefs;
-      if(isset($db->error))
-        $db->failQuery();
-
-      return $db->getKey($username, "___signature___");
-   }
+    /* If necessary, populate the hash dir variable. */
+    if ($hash_dirs == '') {
+        $hash_dirs = computeHashDirs($username);
+    }
+
+    /* Make sure the full hash directory exists. */
+    $real_hash_dir = $dir;
+    for ($h = 0; $h < $dir_hash_level; ++$h) {
+        $real_hash_dir .= '/' . $hash_dirs[$h];
+        if (!@is_dir($real_hash_dir)) {
+            if (!@mkdir($real_hash_dir, 0770)) {
+                echo sprintf(_("Error creating directory %s."), $real_hash_dir) . '<br>';
+                echo _("Could not create hashed directory structure!") . "<br>\n";
+                echo _("Please contact your system administrator and report this error.") . "<br>\n";
+                exit;
+            }
+        }
+    }
+
+    /* And return that directory. */
+    return ($real_hash_dir);
+}
+
+function computeHashDirs($username) {
+    /* Compute the hash for this user and extract the hash directories. */
+    $hash = base_convert(crc32($username), 10, 16);
+    $hash_dirs = array();
+    for ($h = 0; $h < 4; ++ $h) {
+        $hash_dirs[] = substr($hash, $h, 1);
+    }
+        
+    /* Return our array of hash directories. */
+    return ($hash_dirs);
+}
 
 ?>