alecpl
2010-04-01 70318e5463986edff014e881e7e121483679726b
program/include/rcube_mdb2.php
@@ -46,6 +46,8 @@
  var $a_query_results = array('dummy');
  var $last_res_id = 0;
  private $tables;
  /**
@@ -106,7 +108,7 @@
      if (!filesize($dsn_array['database']) && !empty($this->sqlite_initials))
        $this->_sqlite_create_database($dbh, $this->sqlite_initials);
      }
    else
    else if ($this->db_provider!='mssql' && $this->db_provider!='sqlsrv')
      $dbh->setCharset('utf8');
    return $dbh;
@@ -178,6 +180,17 @@
    
  /**
   * Connection state checker
   *
   * @param  boolean  True if in connected state
   */
  function is_connected()
    {
    return PEAR::isError($this->db_handle) ? false : true;
    }
  /**
   * Execute a SQL query
   *
   * @param  string  SQL query to execute
@@ -187,6 +200,9 @@
   */
  function query()
    {
    if (!$this->is_connected())
      return NULL;
    $params = func_get_args();
    $query = array_shift($params);
@@ -228,7 +244,7 @@
  function _query($query, $offset, $numrows, $params)
    {
    // Read or write ?
    if (strtolower(trim(substr($query,0,6)))=='select')
    if (strtolower(substr(trim($query),0,6))=='select')
      $mode='r';
    else
      $mode='w';
@@ -242,18 +258,19 @@
      $result = $this->db_handle->setLimit($numrows,$offset);
    if (empty($params))
        $result = $this->db_handle->query($query);
      $result = $mode=='r' ? $this->db_handle->query($query) : $this->db_handle->exec($query);
    else
      {
      $params = (array)$params;
      $q = $this->db_handle->prepare($query);
      $q = $this->db_handle->prepare($query, null, $mode=='w' ? MDB2_PREPARE_MANIP : null);
      if ($this->db_handle->isError($q))
        {
        $this->db_error = TRUE;
        $this->db_error_msg = $q->userinfo;
        raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
                          'message' => $this->db_error_msg), TRUE, TRUE);
        raise_error(array('code' => 500, 'type' => 'db',
          'line' => __LINE__, 'file' => __FILE__,
          'message' => $this->db_error_msg), TRUE, TRUE);
        }
      else
        {
@@ -307,16 +324,27 @@
   * Get last inserted record ID
   * For Postgres databases, a sequence name is required
   *
   * @param  string  Sequence name for increment
   * @param  string  Table name (to find the incremented sequence)
   * @return mixed   ID or FALSE on failure
   * @access public
   */
  function insert_id($sequence = '')
  function insert_id($table = '')
    {
    if (!$this->db_handle || $this->db_mode=='r')
      return FALSE;
    return $this->db_handle->lastInsertID($sequence);
    if ($table) {
      if ($this->db_provider == 'pgsql')
        // find sequence name
        $table = get_sequence_name($table);
      else
        // resolve table name
        $table = get_table_name($table);
    }
    $id = $this->db_handle->lastInsertID($table);
    return $this->db_handle->isError($id) ? null : $id;
    }
@@ -360,11 +388,31 @@
   */
  function _fetch_row($result, $mode)
    {
    if ($result === FALSE || PEAR::isError($result))
    if ($result === FALSE || PEAR::isError($result) || !$this->is_connected())
      return FALSE;
    return $result->fetchRow($mode);
    }
  /**
   * Wrapper for the SHOW TABLES command
   *
   * @return array List of all tables of the current database
   */
  function list_tables()
  {
    // get tables if not cached
    if (!$this->tables) {
      $this->db_handle->loadModule('Manager');
      if (!PEAR::isError($result = $this->db_handle->listTables()))
        $this->tables = $result;
      else
        $this->tables = array();
    }
    return $this->tables;
  }
  /**
@@ -447,11 +495,32 @@
    switch($this->db_provider)
      {
      case 'mssql':
      case 'sqlsrv':
        return "getdate()";
      default:
        return "now()";
      }
    }
  /**
   * Return list of elements for use with SQL's IN clause
   *
   * @param  string Input array
   * @return string Elements list string
   * @access public
   */
  function array2list($arr, $type=null)
    {
    if (!is_array($arr))
      return $this->quote($arr, $type);
    $res = array();
    foreach ($arr as $item)
      $res[] = $this->quote($item, $type);
    return implode(',', $res);
    }
@@ -471,7 +540,8 @@
        break;
      case 'mssql':
        return "datediff(s, '1970-01-01 00:00:00', $field)";
      case 'sqlsrv':
   return "DATEDIFF(second, '19700101', $field) + DATEDIFF(second, GETDATE(), GETUTCDATE())";
      default:
        return "UNIX_TIMESTAMP($field)";
@@ -523,6 +593,54 @@
  /**
   * Encodes non-UTF-8 characters in string/array/object (recursive)
   *
   * @param  mixed  Data to fix
   * @return mixed  Properly UTF-8 encoded data
   * @access public
   */
  function encode($input)
    {
    if (is_object($input)) {
      foreach (get_object_vars($input) as $idx => $value)
        $input->$idx = $this->encode($value);
      return $input;
      }
    else if (is_array($input)) {
      foreach ($input as $idx => $value)
        $input[$idx] = $this->encode($value);
      return $input;
      }
    return utf8_encode($input);
    }
  /**
   * Decodes encoded UTF-8 string/object/array (recursive)
   *
   * @param  mixed  Input data
   * @return mixed  Decoded data
   * @access public
   */
  function decode($input)
    {
    if (is_object($input)) {
      foreach (get_object_vars($input) as $idx => $value)
        $input->$idx = $this->decode($value);
      return $input;
      }
    else if (is_array($input)) {
      foreach ($input as $idx => $value)
        $input[$idx] = $this->decode($value);
      return $input;
      }
    return utf8_decode($input);
    }
  /**
   * Adds a query result and returns a handle ID
   *
   * @param  object  Query handle
@@ -536,7 +654,8 @@
      {
      $this->db_error = TRUE;
      $this->db_error_msg = $res->getMessage();
      raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
      raise_error(array('code' => 500, 'type' => 'db',
           'line' => __LINE__, 'file' => __FILE__,
           'message' => $res->getMessage() . " Query: " 
       . substr(preg_replace('/[\r\n]+\s*/', ' ', $res->userinfo), 0, 512)),
       TRUE, FALSE);
@@ -585,7 +704,9 @@
    $data = file_get_contents($file_name);
    if (strlen($data))
      sqlite_exec($dbh->connection, $data);
      if (!sqlite_exec($dbh->connection, $data, $error) || MDB2::isError($dbh))
        raise_error(array('code' => 500, 'type' => 'db',
       'line' => __LINE__, 'file' => __FILE__, 'message' => $error), TRUE, FALSE);
    }
@@ -617,8 +738,6 @@
  {
    $debug_output = $scope . '('.$db->db_index.'): ';
    $debug_output .= $message . $db->getOption('log_line_break');
    write_log('sqllog', $debug_output);
    write_log('sql', $debug_output);
  }
}