small fixes from previous

[platal.git] / classes / XDB.php

<?php
/***************************************************************************
 *  Copyright (C) 2003-2006 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 XDB
{
    // {{{ function _prepare

    function _prepare($args) {
        $query    = array_map(Array('XDB', '_db_escape'), $args);
        $query[0] = str_replace('{?}', '%s', str_replace('%',   '%%', $args[0]));
        return call_user_func_array('sprintf', $query);
    }

    // }}}
    // {{{ function query

    function &query()
    {
        return new XOrgDBResult(XDB::_prepare(func_get_args()));
    }

    // }}}
    // {{{ function execute()

    function execute() {
        global $globals;
        return $globals->db->query(XDB::_prepare(func_get_args()));
    }

    // }}}
    // {{{ function iterator()

    function &iterator()
    {
        return new XOrgDBIterator(XDB::_prepare(func_get_args()));
    }

    // }}}
    // {{{ function iterRow()

    function &iterRow()
    {
        return new XOrgDBIterator(XDB::_prepare(func_get_args()), MYSQL_NUM);
    }

    // }}}
    // {{{ function insertId()

    function insertId()
    {
        return mysql_insert_id();
    }

    // }}}
    // {{{ function _db_escape

    function _db_escape($var)
    {
        switch (gettype($var)) {
          case 'boolean':
            return $var ? 1 : 0;

          case 'integer':
          case 'double':
          case 'float':
            return $var;

          case 'string':
            return "'".addslashes($var)."'";

          case 'NULL':
            return 'NULL';

          case 'object':
          case 'array':
            return "'".addslashes(serialize($var))."'";

          default:
            die(var_export($var, true).' is not a valid for a database entry');
        }
    }

    // }}}
}

class XOrgDBResult
{
    // {{{ properties

    var $_res;

    // }}}
    // {{{ constructor

    function XOrgDBResult($query)
    {
        global $globals;
        if (strpos($query, 'SQL_CALC_FOUND_ROWS') === false) {
            $this->_res = $globals->db->query($query);
        } else {
            $this->_res = mysql_query($query);
        }
    }

    // }}}
    // {{{ destructor

    function free()
    {
        mysql_free_result($this->_res);
        unset($this);
    }

    // }}}
    // {{{ function fetchRow

    function _fetchRow()
    {
        return mysql_fetch_row($this->_res);
    }

    // }}}
    // {{{ function fetchAssoc

    function _fetchAssoc()
    {
        return mysql_fetch_assoc($this->_res);
    }

    // }}}
    // {{{ function fetchAllRow

    function fetchAllRow()
    {
        $result = Array();
        while ($result[] = mysql_fetch_row($this->_res)) { }
        array_pop($result);
        $this->free();
        return $result;
    }

    // }}}
    // {{{ function fetchAllAssoc

    function fetchAllAssoc()
    {
        $result = Array();
        while ($result[] = mysql_fetch_assoc($this->_res)) { }
        array_pop($result);
        $this->free();
        return $result;
    }

    // }}}
    // {{{ function fetchOneAssoc()

    function fetchOneAssoc()
    {
        $tmp = $this->_fetchAssoc();
        $this->free();
        return $tmp;
    }

    // }}}
    // {{{ function fetchOneRow()

    function fetchOneRow()
    {
        $tmp = $this->_fetchRow();
        $this->free();
        return $tmp;
    }

    // }}}
    // {{{ function fetchOneCell()

    function fetchOneCell()
    {
        $tmp = $this->_fetchRow();
        $this->free();
        return $tmp[0];
    }

    // }}}
    // {{{ function fetchColumn()

    function fetchColumn($key = 0)
    {
        $res = Array();
        if (is_numeric($key)) {
            while($tmp = $this->_fetchRow()) {
                $res[] = $tmp[$key];
            }
        } else {
            while($tmp = $this->_fetchAssoc()) {
                $res[] = $tmp[$key];
            }
        }
        $this->free();
        return $res;
    }

    // }}}
    // {{{ function numRows

    function numRows()
    {
        return mysql_num_rows($this->_res);
    }

    // }}}
}

class XOrgDBIterator
{
    // {{{ properties

    var $_result;
    var $_pos;
    var $_total;
    var $_mode = MYSQL_ASSOC;

    // }}}
    // {{{ constructor

    function XOrgDBIterator($query, $mode = MYSQL_ASSOC)
    {
        $this->_result =& new XOrgDBResult($query);
        $this->_pos    = 0;
        $this->_total  = $this->_result->numRows();
        $this->_mode   = $mode;
    }

    // }}}
    // {{{ function next ()

    function next()
    {
        $this->_pos ++;
        if ($this->_pos > $this->_total) {
            $this->_result->free();
            unset($this);
            return null;
        }
        return $this->_mode != MYSQL_ASSOC ? $this->_result->_fetchRow() : $this->_result->_fetchAssoc();
    }

    // }}}
    // {{{ function first

    function first()
    {
        return $this->_pos == 1;
    }

    // }}}
    // {{{ function last

    function last()
    {
        return $this->_last == $this->_total;
    }

    // }}}
    // {{{ function total()

    function total()
    {
        return $this->_total;
    }

    // }}}
}

// vim:set et sw=4 sts=4 sws=4 foldmethod=marker:
?>