classes/xdb.php

   1 <?php
   2 /***************************************************************************
   3  *  Copyright (C) 2003-2008 Polytechnique.org                              *
   4  *  http://opensource.polytechnique.org/                                   *
   5  *                                                                         *
   6  *  This program is free software; you can redistribute it and/or modify   *
   7  *  it under the terms of the GNU General Public License as published by   *
   8  *  the Free Software Foundation; either version 2 of the License, or      *
   9  *  (at your option) any later version.                                    *
  10  *                                                                         *
  11  *  This program is distributed in the hope that it will be useful,        *
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of         *
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the          *
  14  *  GNU General Public License for more details.                           *
  15  *                                                                         *
  16  *  You should have received a copy of the GNU General Public License      *
  17  *  along with this program; if not, write to the Free Software            *
  18  *  Foundation, Inc.,                                                      *
  19  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA                *
  20  **************************************************************************/
  21
  22 class XDB
  23 {
  24     private static $mysqli = null;
  25
  26     public static function connect()
  27     {
  28         global $globals;
  29         XDB::$mysqli = new mysqli($globals->dbhost, $globals->dbuser, $globals->dbpwd, $globals->dbdb);
  30         if ($globals->debug & DEBUG_BT) {
  31             $bt = new PlBacktrace('MySQL');
  32             if (mysqli_connect_errno()) {
  33                 $bt->newEvent("MySQLI connection", 0, mysqli_connect_error());
  34                 return false;
  35             }
  36         }
  37         XDB::$mysqli->autocommit(true);
  38         XDB::$mysqli->set_charset($globals->dbcharset);
  39         return true;
  40     }
  41
  42     public static function _prepare($args)
  43     {
  44         $query    = array_map(Array('XDB', 'escape'), $args);
  45         $query[0] = str_replace('{?}', '%s', str_replace('%',   '%%', $args[0]));
  46         return call_user_func_array('sprintf', $query);
  47     }
  48
  49     public static function _reformatQuery($query)
  50     {
  51         $query  = preg_split("/\n\\s*/", trim($query));
  52         $length = 0;
  53         foreach ($query as $key=>$line) {
  54             $local = -2;
  55             if (preg_match('/^([A-Z]+(?:\s+(?:JOIN|BY|FROM|INTO))?)\s+(.*)/u', $line, $matches)
  56             && $matches[1] != 'AND' && $matches[1] != 'OR')
  57             {
  58                 $local  = strlen($matches[1]);
  59                 $line   = $matches[1] . '  ' . $matches[2];
  60                 $length = max($length, $local);
  61             }
  62             $query[$key] = array($line, $local);
  63         }
  64         $res = '';
  65         foreach ($query as $array) {
  66             list($line, $local) = $array;
  67             $local   = max(0, $length - $local);
  68             $res    .= str_repeat(' ', $local) . $line . "\n";
  69             $length += 2 * (substr_count($line, '(') - substr_count($line, ')'));
  70         }
  71         return $res;
  72     }
  73
  74     public static function _query($query)
  75     {
  76         global $globals;
  77
  78         if (!XDB::$mysqli && !XDB::connect()) {
  79             Platal::page()->kill('Impossible de se connecter à la base de données.');
  80             exit;
  81         }
  82
  83         if ($globals->debug & DEBUG_BT) {
  84             $explain = array();
  85             if (strpos($query, 'FOUND_ROWS()') === false) {
  86                 $res = XDB::$mysqli->query("EXPLAIN $query");
  87                 if ($res) {
  88                     while ($row = $res->fetch_assoc()) {
  89                         $explain[] = $row;
  90                     }
  91                     $res->free();
  92                 }
  93             }
  94             PlBacktrace::$bt['MySQL']->start(XDB::_reformatQuery($query));
  95         }
  96
  97         $res = XDB::$mysqli->query($query);
  98
  99         if ($globals->debug & DEBUG_BT) {
 100             PlBacktrace::$bt['MySQL']->stop(@$res->num_rows ? $res->num_rows : XDB::$mysqli->affected_rows,
 101                                             XDB::$mysqli->error,
 102                                             $explain);
 103         }
 104
 105         if ($res === false) {
 106             if (strpos($query, 'INSERT') === false && strpos($query, 'UPDATE') === false
 107                 && strpos($query, 'REPLACE') === false && strpos($query, 'DELETE') === false) {
 108                 Platal::page()->kill('Erreur lors de l\'interrogation de la base de données');
 109             } else {
 110                 Platal::page()->kill('Erreur lors de l\'écriture dans la base de données');
 111             }
 112             exit;
 113         }
 114         return $res;
 115     }
 116
 117     public static function query()
 118     {
 119         return new XOrgDBResult(XDB::_prepare(func_get_args()));
 120     }
 121
 122     public static function execute()
 123     {
 124         global $globals;
 125         $args = func_get_args();
 126         if ($globals->mode != 'rw' && !strpos($args[0], 'logger')) {
 127             return;
 128         }
 129         return XDB::_query(XDB::_prepare($args));
 130     }
 131
 132     public static function iterator()
 133     {
 134         return new XOrgDBIterator(XDB::_prepare(func_get_args()));
 135     }
 136
 137     public static function iterRow()
 138     {
 139         return new XOrgDBIterator(XDB::_prepare(func_get_args()), MYSQL_NUM);
 140     }
 141
 142     public static function insertId()
 143     {
 144         return XDB::$mysqli->insert_id;
 145     }
 146
 147     public static function errno()
 148     {
 149         return XDB::$mysqli->errno;
 150     }
 151
 152     public static function error()
 153     {
 154         return XDB::$mysqli->error;
 155     }
 156
 157     public static function affectedRows()
 158     {
 159         return XDB::$mysqli->affected_rows;
 160     }
 161
 162     public static function escape($var)
 163     {
 164         switch (gettype($var)) {
 165           case 'boolean':
 166             return $var ? 1 : 0;
 167
 168           case 'integer':
 169           case 'double':
 170           case 'float':
 171             return $var;
 172
 173           case 'string':
 174             return "'".addslashes($var)."'";
 175
 176           case 'NULL':
 177             return 'NULL';
 178
 179           case 'object':
 180             if ($var instanceof PlFlagSet) {
 181                 return "'" . addslashes($var->flags()) . "'";
 182             }
 183           case 'array':
 184             return "'".addslashes(serialize($var))."'";
 185
 186           default:
 187             die(var_export($var, true).' is not a valid for a database entry');
 188         }
 189     }
 190 }
 191
 192 class XOrgDBResult
 193 {
 194
 195     private $_res;
 196
 197     public function __construct($query)
 198     {
 199         $this->_res = XDB::_query($query);
 200     }
 201
 202     public function free()
 203     {
 204         if ($this->_res) {
 205             $this->_res->free();
 206         }
 207         unset($this);
 208     }
 209
 210     protected function _fetchRow()
 211     {
 212         return $this->_res ? $this->_res->fetch_row() : null;
 213     }
 214
 215     protected function _fetchAssoc()
 216     {
 217         return $this->_res ? $this->_res->fetch_assoc() : null;
 218     }
 219
 220     public function fetchAllRow()
 221     {
 222         $result = Array();
 223         if (!$this->_res) {
 224             return $result;
 225         }
 226         while ($result[] = $this->_res->fetch_row());
 227         array_pop($result);
 228         $this->free();
 229         return $result;
 230     }
 231
 232     public function fetchAllAssoc()
 233     {
 234         $result = Array();
 235         if (!$this->_res) {
 236             return $result;
 237         }
 238         while ($result[] = $this->_res->fetch_assoc());
 239         array_pop($result);
 240         $this->free();
 241         return $result;
 242     }
 243
 244     public function fetchOneAssoc()
 245     {
 246         $tmp = $this->_fetchAssoc();
 247         $this->free();
 248         return $tmp;
 249     }
 250
 251     public function fetchOneRow()
 252     {
 253         $tmp = $this->_fetchRow();
 254         $this->free();
 255         return $tmp;
 256     }
 257
 258     public function fetchOneCell()
 259     {
 260         $tmp = $this->_fetchRow();
 261         $this->free();
 262         return $tmp[0];
 263     }
 264
 265     public function fetchColumn($key = 0)
 266     {
 267         $res = Array();
 268         if (is_numeric($key)) {
 269             while($tmp = $this->_fetchRow()) {
 270                 $res[] = $tmp[$key];
 271             }
 272         } else {
 273             while($tmp = $this->_fetchAssoc()) {
 274                 $res[] = $tmp[$key];
 275             }
 276         }
 277         $this->free();
 278         return $res;
 279     }
 280
 281     public function fetchOneField()
 282     {
 283         return $this->_res ? $this->_res->fetch_field() : null;
 284     }
 285
 286     public function fetchFields()
 287     {
 288         $res = array();
 289         while ($res[] = $this->fetchOneField());
 290         return $res;
 291     }
 292
 293     public function numRows()
 294     {
 295         return $this->_res ? $this->_res->num_rows : 0;
 296     }
 297
 298     public function fieldCount()
 299     {
 300         return $this->_res ? $this->_res->field_count : 0;
 301     }
 302 }
 303
 304 require_once dirname(__FILE__) . '/pliterator.php';
 305
 306 class XOrgDBIterator extends XOrgDBResult implements PlIterator
 307 {
 308     private $_result;
 309     private $_pos;
 310     private $_total;
 311     private $_fpos;
 312     private $_fields;
 313     private $_mode = MYSQL_ASSOC;
 314
 315     public function __construct($query, $mode = MYSQL_ASSOC)
 316     {
 317         parent::__construct($query);
 318         $this->_pos    = 0;
 319         $this->_total  = $this->numRows();
 320         $this->_fpost  = 0;
 321         $this->_fields = $this->fieldCount();
 322         $this->_mode   = $mode;
 323     }
 324
 325     public function next()
 326     {
 327         $this->_pos ++;
 328         if ($this->_pos > $this->_total) {
 329             $this->free();
 330             unset($this);
 331             return null;
 332         }
 333         return $this->_mode != MYSQL_ASSOC ? $this->_fetchRow() : $this->_fetchAssoc();
 334     }
 335
 336     public function first()
 337     {
 338         return $this->_pos == 1;
 339     }
 340
 341     public function last()
 342     {
 343         return $this->_pos == $this->_total;
 344     }
 345
 346     public function total()
 347     {
 348         return $this->_total;
 349     }
 350
 351     public function nextField()
 352     {
 353         $this->_fpos++;
 354         if ($this->_fpos > $this->_fields) {
 355             return null;
 356         }
 357         return $this->fetchOneField();
 358     }
 359
 360     public function firstField()
 361     {
 362         return $this->_fpos == 1;
 363     }
 364
 365     public function lastField()
 366     {
 367         return $this->_fpos == $this->_fields;
 368     }
 369
 370     public function totalFields()
 371     {
 372         return $this->_fields;
 373     }
 374 }
 375
 376 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 377 ?>