X-Git-Url: https://vcs.fsf.org/?a=blobdiff_plain;f=functions%2Fdb_prefs.php;h=40c0b7c5c3a27f63c3ff9f30b0e9268ae67a04ab;hb=e976319c39def9b07d04b41b8cf6f1918472bf9e;hp=442e9d26d4cb293f7b0cbff3fff3eb6638382dcc;hpb=355861846a51d35efe50848869d680fd3b4a5316;p=squirrelmail.git

diff --git a/functions/db_prefs.php b/functions/db_prefs.php
index 442e9d26..40c0b7c5 100644
--- a/functions/db_prefs.php
+++ b/functions/db_prefs.php
@@ -1,8 +1,9 @@
 <?php
-/**
+
+/*
  * db_prefs.php
  *
- * Copyright (c) 1999-2001 The Squirrelmail Development Team
+ * 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
@@ -15,57 +16,101 @@
  * Database:
  * ---------
  *
- * The preferences table should have tree columns:
- *    username   char  \  primary
+ * The preferences table should have three columns:
+ *    user       char  \  primary
  *    prefkey    char  /  key
  *    prefval    blob
  *
- *   CREATE TABLE userprefs (user CHAR(32) NOT NULL DEFAULT '',
+ *   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.
+ * by using conf.pl or the administrator plugin
  *
  * $Id$
  */
 
-/*****************************************************************/
-/*** THIS FILE NEEDS TO HAVE ITS FORMATTING FIXED!!!           ***/
-/*** PLEASE DO SO AND REMOVE THIS COMMENT SECTION.             ***/
-/***    + Base level indent should begin at left margin, as    ***/
-/***      the require_once below.                              ***/
-/***    + All identation should consist of four space blocks   ***/
-/***    + Tab characters are evil.                             ***/
-/***    + all comments should use "slash-star ... star-slash"  ***/
-/***      style -- no pound characters, no slash-slash style   ***/
-/***    + FLOW CONTROL STATEMENTS (if, while, etc) SHOULD      ***/
-/***      ALWAYS USE { AND } CHARACTERS!!!                     ***/
-/***    + Please use ' instead of ", when possible. Note "     ***/
-/***      should always be used in _( ) function calls.        ***/
-/*** Thank you for your help making the SM code more readable. ***/
-/*****************************************************************/
+define('SMDB_UNKNOWN', 0);
+define('SMDB_MYSQL', 1);
+define('SMDB_PGSQL', 2);
 
 require_once('DB.php');
+require_once('../config/config.php');
+
+global $prefs_are_cached, $prefs_cache;
+
+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@host/database';
-      var $table = 'userprefs';
+    var $table = 'userprefs';
+    var $user_field = 'user';
+    var $key_field = 'prefkey';
+    var $val_field = 'prefval';
+
+    var $dbh   = NULL;
+    var $error = NULL;
+    var $db_type = SMDB_UNKNOWN;
+
+    var $default = Array('chosen_theme' => '../themes/default_theme.php',
+                         'show_html_default' => '0');
 
-      var $dbh   = NULL;
-      var $error = NULL;
+    function open() {
+        global $prefs_dsn, $prefs_table;
+        global $prefs_user_field, $prefs_key_field, $prefs_val_field;
 
-      var $default = Array('chosen_theme'      => '../themes/default_theme.php',
-                           'show_html_default' => '0');
+        if(isset($this->dbh)) {
+            return true;
+        }
 
-      function dbPrefs() {
-         $this->open();
-      }
+        if (preg_match('/^mysql/', $prefs_dsn)) {
+            $this->db_type = SMDB_MYSQL;
+        } elseif (preg_match('/^pgsql/', $prefs_dsn)) {
+            $this->db_type = SMDB_PGSQL;
+        }
 
-      function open() {
-        if(isset($this->dbh)) return true;
-        $dbh = DB::connect($this->DSN, true);
+        if (!empty($prefs_table)) {
+            $this->table = $prefs_table;
+        }
+        if (!empty($prefs_user_field)) {
+            $this->user_field = $prefs_user_field;
+        }
+        if (!empty($prefs_key_field)) {
+            $this->key_field = $prefs_key_field;
+        }
+        if (!empty($prefs_val_field)) {
+            $this->val_field = $prefs_val_field;
+        }
+        $dbh = DB::connect($prefs_dsn, true);
 
         if(DB::isError($dbh) || DB::isWarning($dbh)) {
             $this->error = DB::errorMessage($dbh);
@@ -74,103 +119,194 @@ class dbPrefs {
 
         $this->dbh = $dbh;
         return true;
-      }
+    }
 
-
-      function failQuery($res = NULL) {
+    function failQuery($res = NULL) {
         if($res == NULL) {
             printf(_("Preference database error (%s). Exiting abnormally"),
-            $this->error);
+                  $this->error);
         } else {
             printf(_("Preference database error (%s). Exiting abnormally"),
-            DB::errorMessage($res));
+                  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));
+    function getKey($user, $key, $default = '') {
+        global $prefs_cache;
 
-        $res = $this->dbh->query($query);
-        if(DB::isError($res))
-            $this->failQuery($res);
+        cachePrefValues($user);
 
-        if($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
-            return $row['prefval'];
+        if (isset($prefs_cache[$key])) {
+            return $prefs_cache[$key];
         } else {
-            if(isset($this->default[$key])) {
-            return $this->default[$key];
+            if (isset($this->default[$key])) {
+                return $this->default[$key];
             } else {
-            return '';
+                return $default;
             }
         }
+    }
 
-        return '';
-      }
+    function deleteKey($user, $key) {
+        global $prefs_cache;
 
-      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));
+        if (!$this->open()) {
+            return false;
+        }
+        $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
+                         $this->table,
+                         $this->user_field,
+                         $this->dbh->quoteString($user),
+                         $this->key_field,
+                         $this->dbh->quoteString($key));
 
         $res = $this->dbh->simpleQuery($query);
-            if(DB::isError($res))
+        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));
+    function setKey($user, $key, $value) {
+        if (!$this->open()) {
+            return false;
+        }
+        if ($this->db_type == SMDB_MYSQL) {
+            $query = sprintf("REPLACE INTO %s (%s, %s, %s) ".
+                             "VALUES('%s','%s','%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $this->dbh->quoteString($user),
+                             $this->dbh->quoteString($key),
+                             $this->dbh->quoteString($value));
 
             $res = $this->dbh->simpleQuery($query);
-            if(DB::isError($res))
-            $this->failQuery($res);
+            if(DB::isError($res)) {
+                $this->failQuery($res);
+            }
+        } elseif ($this->db_type == SMDB_PGSQL) {
+            $this->dbh->simpleQuery("BEGIN TRANSACTION");
+            $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
+                             $this->table,
+                             $this->user_field,
+                             $this->dbh->quoteString($user),
+                             $this->key_field,
+                             $this->dbh->quoteString($key));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
+                $this->failQuery($res);
+            }
+            $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $this->dbh->quoteString($user),
+                             $this->dbh->quoteString($key),
+                             $this->dbh->quoteString($value));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
+                $this->failQuery($res);
+            }
+            $this->dbh->simpleQuery("COMMIT TRANSACTION");
+        } else {
+            $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
+                             $this->table,
+                             $this->user_field,
+                             $this->dbh->quoteString($user),
+                             $this->key_field,
+                             $this->dbh->quoteString($key));
+            $res = $this->dbh->simpleQuery($query);
+            if (DB::isError($res)) {
+                $this->failQuery($res);
+            }
+            $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
+                             $this->table,
+                             $this->user_field,
+                             $this->key_field,
+                             $this->val_field,
+                             $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;
 
+        if (!$this->open()) {
+            return;
+        }
+
+        $prefs_cache = array();
+        $query = sprintf("SELECT %s as prefkey, %s as prefval FROM %s ".
+                         "WHERE %s = '%s'",
+                         $this->key_field,
+                         $this->val_field,
+                         $this->table,
+                         $this->user_field,
+                         $this->dbh->quoteString($user));
+        $res = $this->dbh->query($query);
+        if (DB::isError($res)) {
+            $this->failQuery($res);
+        }
 
-      /**
-       ** 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));
+        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) {
+        if (!$this->open()) {
+            return;
+        }
+        $query = sprintf("SELECT %s, %s as prefkey, %s as prefval FROM %s WHERE %s='%s' ".
+                         "AND %s LIKE 'highlight%%' ORDER BY %s",
+                         $this->user_field,
+                         $this->key_field,
+                         $this->val_field,
+                         $this->table,
+                         $this->user_field,
+                         $this->dbh->quoteString($user),
+                         $this->key_field,
+                         $this->key_field);
 
         $res = $this->dbh->query($query);
-            if(DB::isError($res))
+        if(DB::isError($res)) {
             $this->failQuery($res);
+        }
 
-        // Store old data in array
+        /* Store old data in array */
         $rows = Array();
-        while($row = $res->fetchRow(DB_FETCHMODE_ASSOC))
+        while($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
             $rows[] = $row;
+        }
 
-        // Renumber keys of old data
+        /* Renumber keys of old data */
         $hilinum = 0;
         for($i = 0; $i < count($rows) ; $i++) {
             $oldkey = $rows[$i]['prefkey'];
@@ -178,80 +314,116 @@ class dbPrefs {
             $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);
+                $query = sprintf("UPDATE %s SET %s='%s' ".
+                                 "WHERE %s ='%s' AND %s='%s'",
+                                 $this->table,
+                                 $this->key_field,
+                                 $this->dbh->quoteString($newkey),
+                                 $this->user_field,
+                                 $this->dbh->quoteString($user),
+                                 $this->key_field,
+                                 $this->dbh->quoteString($oldkey));
+
+                $res = $this->dbh->simpleQuery($query);
+                if(DB::isError($res)) {
+                    $this->failQuery($res);
+                }
             }
         }
 
         return;
-      }
+    }
 
-   } // end class dbPrefs
+} /* end class dbPrefs */
 
 
-   /** returns the value for the pref $string **/
-   function getPref($data_dir, $username, $string, $default ) {
-      $db = new dbPrefs;
-      if(isset($db->error)) {
+/* 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);
+              $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;
-   }
-
-   /** 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___");
-   }
+    }
+
+    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, $number, $string) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    if ($number == "g") {
+        $key = '___signature___';
+    } else {
+        $key = sprintf('___sig%s___', $number);
+    }
+    $db->setKey($username, $key, $string);
+    return;
+}
+
+/* Gets the signature */
+function getSig($data_dir, $username, $number) {
+    $db = new dbPrefs;
+    if(isset($db->error)) {
+        $db->failQuery();
+    }
+
+    if ($number == "g") {
+        $key = '___signature___';
+    } else {
+        $key = sprintf('___sig%d___', $number);
+    }
+    return $db->getKey($username, $key);
+}
 
 ?>