thomascube
2008-04-05 e70d6ea64e711096af36b1234f8545b870ea5f45
program/include/rcube_db.inc
old mode 100755 new mode 100644
@@ -5,7 +5,7 @@
 | program/include/rcube_db.inc                                          |
 |                                                                       |
 | This file is part of the RoundCube Webmail client                     |
 | Copyright (C) 2005, RoundCube Dev. - Switzerland                      |
 | Copyright (C) 2005-2007, RoundCube Dev. - Switzerland                 |
 | Licensed under the GNU GPL                                            |
 |                                                                       |
 | PURPOSE:                                                              |
@@ -14,6 +14,7 @@
 |                                                                       |
 +-----------------------------------------------------------------------+
 | Author: David Saez Padros <david@ols.es>                              |
 |         Thomas Bruederli <roundcube@gmail.com>                        |
 +-----------------------------------------------------------------------+
 $Id$
@@ -24,18 +25,17 @@
/**
 * Obtain the PEAR::DB class that is used for abstraction
 */
require_once('DB.php');
require_once 'DB.php';
/**
 * Database independent query interface
 *
 * This is a wrapper for the PEAR::DB class
 *
 * @package    RoundCube Webmail
 * @package    Database
 * @author     David Saez Padros <david@ols.es>
 * @author     Thomas Bruederli <roundcube@gmail.com>
 * @version    1.14
 * @version    1.17
 * @link       http://pear.php.net/package/DB
 */
class rcube_db
@@ -45,6 +45,9 @@
  var $db_connected = false;  // Already connected ?
  var $db_mode = '';          // Connection mode
  var $db_handle = 0;         // Connection handle
  var $db_pconn = false;      // Use persistent connections
  var $db_error = false;
  var $db_error_msg = '';
  var $a_query_results = array('dummy');
  var $last_res_id = 0;
@@ -56,13 +59,14 @@
   * @param  string  DSN for read/write operations
   * @param  string  Optional DSN for read only operations
   */
  function __construct($db_dsnw, $db_dsnr='')
  function __construct($db_dsnw, $db_dsnr='', $pconn=false)
    {
    if ($db_dsnr=='')
      $db_dsnr=$db_dsnw;
        
    $this->db_dsnw = $db_dsnw;
    $this->db_dsnr = $db_dsnr;
    $this->db_pconn = $pconn;
        
    $dsn_array = DB::parseDSN($db_dsnw);
    $this->db_provider = $dsn_array['phptype'];        
@@ -74,9 +78,9 @@
   *
   * @see  rcube_db::__construct
   */
  function rcube_db($db_dsnw,$db_dsnr='')
  function rcube_db($db_dsnw, $db_dsnr='', $pconn=false)
    {
    $this->__construct($db_dsnw,$db_dsnr);
    $this->__construct($db_dsnw, $db_dsnr, $pconn);
    }
@@ -90,12 +94,17 @@
  function dsn_connect($dsn)
    {
    // Use persistent connections if available
    $dbh = DB::connect($dsn, array('persistent' => TRUE));
    $dbh = DB::connect($dsn, array('persistent' => $this->db_pconn));
        
    if (DB::isError($dbh))
      {
      raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
                        'message' => $dbh->getMessage()), TRUE, FALSE);
      $this->db_error = TRUE;
      $this->db_error_msg = $dbh->getMessage();
      raise_error(array('code' => 603, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
                        'message' => $this->db_error_msg), TRUE, FALSE);
      return FALSE;
      }
    else if ($this->db_provider=='sqlite')
@@ -142,7 +151,28 @@
      $dsn = $this->db_dsnw;
    $this->db_handle = $this->dsn_connect($dsn);
    $this->db_connected = true;
    $this->db_connected = $this->db_handle ? TRUE : FALSE;
    }
  /**
   * Activate/deactivate debug mode
   * (not implemented)
   */
  function set_debug($dbg = true)
  {
  }
  /**
   * Getter for error state
   *
   * @param  boolean  True on error
   */
  function is_error()
    {
    return $this->db_error ? $this->db_error_msg : FALSE;
    }
@@ -203,6 +233,9 @@
      $mode='w';
        
    $this->db_connect($mode);
    if (!$this->db_connected)
      return FALSE;
    if ($this->db_provider == 'sqlite')
      $this->_sqlite_prepare();
@@ -268,14 +301,17 @@
    switch($this->db_provider)
      {
      case 'pgsql':
        // PostgreSQL uses sequences
        $result = &$this->db_handle->getOne("SELECT CURRVAL('$sequence')");
        if (DB::isError($result))
          {
          raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__, 
                            'message' => $result->getMessage()), TRUE, FALSE);
          }
        return $result;
      case 'mssql':
        $result = &$this->db_handle->getOne("SELECT @@IDENTITY");
        if (DB::isError($result))
          raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
                            'message' => $result->getMessage()), TRUE, FALSE);
        return $result;
                
      case 'mysql': // This is unfortuneate
@@ -283,7 +319,7 @@
      case 'mysqli':
        return mysqli_insert_id($this->db_handle->connection);
      case 'sqlite':
        return sqlite_last_insert_rowid($this->db_handle->connection);
@@ -304,17 +340,47 @@
  function fetch_assoc($res_id=NULL)
    {
    $result = $this->_get_result($res_id);
    return $this->_fetch_row($result, DB_FETCHMODE_ASSOC);
    }
    if (DB::isError($result))
  /**
   * Get an index array for one row
   * If no query handle is specified, the last query will be taken as reference
   *
   * @param  number  Optional query handle identifier
   * @return mixed   Array with col values or FALSE on failure
   * @access public
   */
  function fetch_array($res_id=NULL)
    {
    $result = $this->_get_result($res_id);
    return $this->_fetch_row($result, DB_FETCHMODE_ORDERED);
    }
  /**
   * Get co values for a result row
   *
   * @param  object  Query result handle
   * @param  number  Fetch mode identifier
   * @return mixed   Array with col values or FALSE on failure
   * @access private
   */
  function _fetch_row($result, $mode)
    {
    if (!$result || DB::isError($result))
      {
      raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
                        'message' => $this->db_link->getMessage()), TRUE, FALSE);
      return FALSE;
      }
      elseif (!is_object($result))
      return FALSE;
                         
    return $result->fetchRow(DB_FETCHMODE_ASSOC);
    return $result->fetchRow($mode);
    }
  /**
   * Formats input so it can be safely used in a query
@@ -370,6 +436,41 @@
  /**
   * Escapes a string
   *
   * @param  string  The string to be escaped
   * @return string  The escaped string
   * @access public
   */
  function escapeSimple($str)
    {
    if (!$this->db_handle)
      $this->db_connect('r');
    return $this->db_handle->escapeSimple($str);
    }
  /*
   * Return SQL function for current time and date
   *
   * @return string SQL function to use in query
   * @access public
   */
  function now()
    {
    switch($this->db_provider)
      {
      case 'mssql':
        return "getdate()";
      default:
        return "now()";
      }
    }
  /**
   * Return SQL statement to convert a field value into a unix timestamp
   *
   * @param  string  Field name
@@ -382,7 +483,9 @@
      {
      case 'pgsql':
        return "EXTRACT (EPOCH FROM $field)";
        break;
      case 'mssql':
        return "datediff(s, '1970-01-01 00:00:00', $field)";
      default:
        return "UNIX_TIMESTAMP($field)";
@@ -404,7 +507,7 @@
      case 'mysqli':
      case 'mysql':
      case 'sqlite':
        return "FROM_UNIXTIME($timestamp)";
        return sprintf("FROM_UNIXTIME(%d)", $timestamp);
      default:
        return date("'Y-m-d H:i:s'", $timestamp);
@@ -502,4 +605,4 @@
  }  // end class rcube_db
?>
?>