KjwSql.php

<?php // vim:ts=4:sw=4:noet
/*====================================================================== 
Copyright (C) 2006-2009, Walter Doekes
This file is part of KjwLib-php45.
 
KjwLib-php45 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 3 of the License, or 
(at your option) any later version. 

KjwLib-php45 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 KjwLib-php45.  If not, see <http://www.gnu.org/licenses/>. 
======================================================================*/

require_once('KjwObject.php');

class KjwSql extends KjwObject {
        var $_sqltype;          
        var $_hostname;         
        var $_portnum;          
        var $_username;         
        var $_password;         
        var $_database;         
        var $_debug_query_count; 
        var $_last_query;       
        function KjwSql($sqltype, $hostname, $portnum, $username, $password, $database) {
                parent::KjwObject();
                $this->_sqltype = $sqltype;
                $this->_hostname = $hostname;
                $this->_portnum = $portnum;
                $this->_username = $username;
                $this->_password = $password;
                $this->_database = $database;
                $this->_debug_query_count = 0;
                if (KJW_DEBUG)
                        $this->_last_query = array();
                else
                        $this->_last_query = '';
        }

        /* [INHERITED DOCS] */
        function destroy() {
                $this->disconnect(); // Hope for succes.
                parent::destroy();
        }

        function getType() {
                return $this->_sqltype;
        }

        function connect() {
                return $this->notImplemented();
        }

        function disconnect() {
                return $this->notImplemented();
        }

        function beginWork() {
                return $this->execute('BEGIN WORK');
        }

        function commitWork() {
                return $this->execute('COMMIT');
        }

        function rollbackWork() {
                return $this->execute('ROLLBACK');
        }
        
        function execute($query) {
                $this->_last_query = $query;
                return $this->notImplemented();
        }

        function selectAll($query, $max = -1, $skip = 0) {
                $result = $this->execute("SELECT $query");
                if (!is_bool($result)) {
                        $ret = array();
                        $result->seekAbs($skip);
                        while (($row = $result->getNext()) && $max--)
                                array_push($ret, $row);
                        $result->destroy();
                        return $ret;
                } elseif ($result === false) {
                        return false;
                }
                $this->croak('Database Data Transfer Failure', 'Execute returned \'true\'. Query was: ' . $this->_getLastQuery(), 2);
        }

        function selectAtom($query) {
                $result = $this->execute("SELECT $query");
                if (!is_bool($result) && $result->size() === 1) {
                        $arr = $result->getNext();
                        $result->destroy();
                        if (sizeof($arr) === 1)
                                return array_shift($arr);
                } elseif ($result === false) {
                        return false;
                }
                $this->croak('Database Data Transfer Failure', 'Atom: result count != 1. Query was: ' . $this->_getLastQuery(), 2);
        }

        function selectAtomOrNull($query) {
                $result = $this->execute("SELECT $query");
                if (!is_bool($result)) {
                        if ($result->size() === 0) {
                                $result->destroy();
                                return null;
                        } elseif ($result->size() === 1) {
                                $arr = $result->getNext();
                                $result->destroy();
                                if (sizeof($arr) === 1)
                                        return array_shift($arr);
                        }
                } elseif ($result === false) {
                        return false;
                }
                $this->croak('Database Data Transfer Failure', 'AtomOrNull: result count > 1. Query was: ' . $this->_getLastQuery(), 2);
        }

        function selectAtomRow($query) {
                $result = $this->execute("SELECT $query");
                if (!is_bool($result) && $result->size() === 1) {
                        $arr = $result->getNext();
                        $result->destroy();
                        return $arr;
                } elseif ($result === false) {
                        return false;
                }
                $this->croak('Database Data Transfer Failure', 'AtomRow, result count != 1. Query was: ' . $this->_getLastQuery(), 2);
        }

        function selectAtomRowOrNull($query) {
                $result = $this->execute("SELECT $query");
                if (!is_bool($result)) {
                        if ($result->size() === 0) {
                                $result->destroy();
                                return null;
                        } elseif ($result->size() === 1) {
                                $arr = $result->getNext();
                                $result->destroy();
                                return $arr;
                        }
                } elseif ($result === false) {
                        return false;
                }
                $this->croak('Database Data Transfer Failure', 'AtomRowOrNull, result count > 1. Query was: ' . $this->_getLastQuery(), 2);
        }

        function safeQuote($mixed) {
                if (is_string($mixed))
                        return "'" . preg_replace("/[\x01-\x1f]/", '', str_replace("\x00", '', $mixed)) . "'";
                if (is_int($mixed) or is_float($mixed))
                        return (double)$mixed;
                if (is_bool($mixed))
                        return $mixed ? 1 : 0;
                if (is_null($mixed))
                        return 'NULL';
                if (is_array($mixed) || is_object($mixed)) {
                        $ret = array();
                        foreach($mixed as $k => $v)
                                $ret[$this->nameQuote($k)] = $this->safeQuote($v);
                        return $ret;
                }
                return $this->croak('Library Component Error', 'Sql safeQuote fell through switch. Mixed parameter is "'
                                . $mixed . '" as string, type is ' . gettype($mixed) . '.', 1);
        }

        function nameQuote($string) {
                return $this->notImplemented();
        }

        function insertArray($table, $args) {
                assert(is_array($args));
                if (is_string($table)) {
                        $table = $this->nameQuote($table);
                } elseif (is_array($table)) {
                        foreach ($table as $k => $v)
                                $table[$k] = $this->nameQuote($v);
                        $table = implode('.', $table);
                } else {
                        assert(0);
                }
                $keys = array();
                $values = array();
                foreach ($args as $k => $v) {
                        $keys[] = $this->nameQuote($k);
                        if (is_array($v)) {
                                $values[] = $v[0]; // Special.. no quoting.
                        } else {
                                $values[] = $this->safeQuote($v);
                        }
                }
                        
                $query = "INSERT INTO $table ("
                                . implode(', ', $keys) . ") VALUES ("
                                . implode(', ', $values) . ")";
                return $this->execute($query);
        }
        
        function updateArray($table, $args, $where = null) {
                assert(is_array($args) && is_string($where));
                if (is_string($table)) {
                        $table = $this->nameQuote($table);
                } elseif (is_array($table)) {
                        foreach ($table as $k => $v)
                                $table[$k] = $this->nameQuote($v);
                        $table = implode('.', $table);
                } else {
                        assert(0);
                }
                $setArgs = array();
                foreach($args AS $k => $v) {
                        if (is_array($v)) {
                                $setArgs[] = $this->nameQuote($k) . ' = ' . $v[0]; // Special.. no quoting.
                        } else {
                                $setArgs[] = $this->nameQuote($k) . ' = ' . $this->safeQuote($v);
                        }
                }               
                return $this->execute(
                        "UPDATE $table SET " . implode(', ', $setArgs) . ($where !== null ? " WHERE $where" : '')
                );
        }
        
        function affectedRows() {
                return $this->notImplemented();
        }
        
        function insertId() {
                return $this->notImplemented();
        }

        function debugStat() {
                return array('query_count' => $this->_debug_query_count);
        }

        function _getLastQuery() {
                if (KJW_DEBUG)
                        return end($this->_last_query);
                return $this->_last_query;
        }

        function _setLastQuery($query) {
                if (KJW_DEBUG) {
                        array_push($this->_last_query, $query);
                        $this->trace($query, 'DEBUG');
                } else {
                        $this->_last_query = $query;
                }
                ++$this->_debug_query_count;
        }
}


function kjwsql_from_url($url, $create_new = false) {
        static $sqls = array();
        if ($create_new || !isset($sqls[$url])) {
                $ua = parse_url($url);
                $port = (int)@$ua['port'];
                switch (@$ua['scheme']) {
//              case 'csvsql': require_once(dirname(__FILE__) . '/KjwCsvSql.php');
//                      
                case 'mysql': require_once(dirname(__FILE__) . '/KjwMySql.php');
                        $sqls[$url] = new KjwMySql(@$ua['host'], $port, @$ua['user'], @$ua['pass'], substr(@$ua['path'], 1)); break;
                case 'mssql': require_once(dirname(__FILE__) . '/KjwMsSql.php');
                        $sqls[$url] = new KjwMsSql(@$ua['host'], $port, @$ua['user'], @$ua['pass'], substr(@$ua['path'], 1)); break;
                case 'pgsql': require_once(dirname(__FILE__) . '/KjwPgSql.php');
                        $sqls[$url] = new KjwPgSql(@$ua['host'], $port, @$ua['user'], @$ua['pass'], substr(@$ua['path'], 1)); break;
                case 'xmlsql': require_once(dirname(__FILE__) . '/KjwXmlSql.php');
                        $sqls[$url] = new KjwXmlSql(@$ua['host'], $port, @$ua['user'], @$ua['pass'], substr(@$ua['path'], 1)); break;
                default: trigger_error('Unknown SQL type: ' . @$ua['scheme'], E_USER_ERROR);
                        return null;
                }
        }
        return $sqls[$url];
}

?>

---