diogenes.database.inc.php

<?php
/*
 * Copyright (C) 2003-2004 Polytechnique.org
 * http://opensource.polytechnique.org/
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */


class DiogenesDatabase {
  var $connect_id;
  
  var $_trace = false;
  
  var $_trace_data = Array();

  var $_fatal = false;

  var $_errno = 0;
  
  var $_errstr = '';

  var $_errinfo = '';

  function DiogenesDatabase($database, $host, $user, $password) {
    global $globals;
    
    // make sure that we have MySQL support, try loading it
    if (!extension_loaded('mysql') && !dl('mysql.so'))
    {
      echo "MySQL support needs to be activated in your PHP configuration!<br>\n";
      echo "Add a line with 'extension=mysql.so' in your php.ini file.\n";
      exit(1);
    }

    $this->_fatal = @$this->database_error_fatal;
        
    if(empty($user)){
      $this->connect_id=@mysql_connect();
    } else {
      $this->connect_id=@mysql_connect($host, $user, $password);
    }

    if (!$this->connect_id) {
      $this->_handleError("");
      return;
    }
        
    if (!@mysql_select_db($database,$this->connect_id))
    {
      $this->_handleError("");
      return;
    }

    // when the script exits, we close the connection to the DB
    register_shutdown_function(array(&$this, 'close'));
  }


  function close() {
      mysql_close($this->connect_id);
      $this->connect_id = FALSE;
  }


  function trace_off() {
      $this->_trace = false;
  }

  
  function trace_on() {
      $this->_trace = true;
  }


  function trace_format(&$page,$template='') {
      global $globals;
      if(empty($template))
          $template = $globals->libroot . '/templates/database-debug.tpl';
      $page->assign_by_ref('trace_data', $this->_trace_data);
      return $page->fetch($template);
  }


  function query($query) {
    if (!empty($query)) {
      
      if ($this->_trace) {
        $_res = mysql_query("EXPLAIN $query", $this->connect_id);
        $explain = Array();
        while($row = @mysql_fetch_assoc($_res)) $explain[] = $row;
        $trace_data = Array('query' => $query, 'explain' => $explain );
        @mysql_free_result($_res);
      }

      $res = mysql_query($query, $this->connect_id);
      
      if ($this->_trace) {
        $trace_data['error'] = $this->error();
        $this->_trace_data[] = $trace_data;
      }

      if (!$res)
      {
        $this->_handleError($query);
      }

      return $res;
    }
  }


  function insert_id()
  {
    return @mysql_insert_id($this->connect_id);
  }

  
  function err()
  {
    return ($this->_errno != 0);
  }

  
  function error() {
    return $this->_errstr;
  }

  
  function errno() {
    return $this->_errno;
  }

  
  function errinfo()
  {
    return $this->_errinfo;
  }

  
  function ResetError()
  {
    $this->_errno = 0;
    $this->_errstr = '';
    $this->_errinfo = '';
  }

  
  function affected_rows() {
    return @mysql_affected_rows($this->connect_id);
  }


  function get_set($table,$column) {
    $res = $this->query("show columns from $table like '$column'");
    $line = mysql_fetch_assoc($res);
    $set = $line['Type'];
    $set = substr($set,5,strlen($set)-7);
    return preg_split("/','/",$set);
  }

  
  function _handleError($extras = '')
  {
    $this->_errinfo = $extras;
    
    if ($this->connect_id)
    {
      $this->_errno = mysql_errno($this->connect_id);
      $this->_errstr = mysql_error($this->connect_id);
    }
    else
    {
      $this->_errno = mysql_errno();
      $this->_errstr = mysql_error();
    }

    if ($this->_fatal)
    {
      die(sprintf("Database error: (%i) %s\n%s\n", $this->_errno, $this->_errstr, $this->_errinfo));
    }
  }
}

?>

---