source: trunk/web/punbb/include/dblayer/sqlite.php @ 2

<?php
/***********************************************************************

  Copyright (C) 2002-2005  Rickard Andersson (rickard@punbb.org)

  This file is part of PunBB.

  PunBB 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.

  PunBB 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

************************************************************************/


// Make sure we have built in support for SQLite
if (!function_exists('sqlite_open'))
        exit('This PHP environment doesn\'t have SQLite support built in. SQLite support is required if you want to use a SQLite database to run this forum. Consult the PHP documentation for further assistance.');


class DBLayer
{
        var $prefix;
        var $link_id;
        var $query_result;
        var $in_transaction = 0;

        var $saved_queries = array();
        var $num_queries = 0;

        var $error_no = false;
        var $error_msg = 'Unknown';


        function DBLayer($db_host, $db_username, $db_password, $db_name, $db_prefix, $p_connect)
        {
                // Prepend $db_name with the path to the forum root directory
                $db_name = PUN_ROOT.$db_name;

                $this->prefix = $db_prefix;

                if (!file_exists($db_name))
                {
                        @touch($db_name);
                        @chmod($db_name, 0666);
                        if (!file_exists($db_name))
                                error('Unable to create new database \''.$db_name.'\'. Permission denied', __FILE__, __LINE__);
                }

                if (!is_readable($db_name))
                        error('Unable to open database \''.$db_name.'\' for reading. Permission denied', __FILE__, __LINE__);

                if (!is_writable($db_name))
                        error('Unable to open database \''.$db_name.'\' for writing. Permission denied', __FILE__, __LINE__);

                if ($p_connect)
                        $this->link_id = @sqlite_popen($db_name, 0666, $sqlite_error);
                else
                        $this->link_id = @sqlite_open($db_name, 0666, $sqlite_error);

                if (!$this->link_id)
                        error('Unable to open database \''.$db_name.'\'. SQLite reported: '.$sqlite_error, __FILE__, __LINE__);
                else
                        return $this->link_id;
        }


        function start_transaction()
        {
                ++$this->in_transaction;

                return (@sqlite_query($this->link_id, 'BEGIN')) ? true : false;
        }


        function end_transaction()
        {
                --$this->in_transaction;

                if (@sqlite_query($this->link_id, 'COMMIT'))
                        return true;
                else
                {
                        @sqlite_query($this->link_id, 'ROLLBACK');
                        return false;
                }
        }


        function query($sql, $unbuffered = false)
        {
                if (defined('PUN_SHOW_QUERIES'))
                        $q_start = get_microtime();

                if ($unbuffered)
                        $this->query_result = @sqlite_unbuffered_query($this->link_id, $sql);
                else
                        $this->query_result = @sqlite_query($this->link_id, $sql);

                if ($this->query_result)
                {
                        if (defined('PUN_SHOW_QUERIES'))
                                $this->saved_queries[] = array($sql, sprintf('%.5f', get_microtime() - $q_start));

                        ++$this->num_queries;

                        return $this->query_result;
                }
                else
                {
                        if (defined('PUN_SHOW_QUERIES'))
                                $this->saved_queries[] = array($sql, 0);

                        $this->error_no = @sqlite_last_error($this->link_id);
                        $this->error_msg = @sqlite_error_string($this->error_no);

                        if ($this->in_transaction)
                                @sqlite_query($this->link_id, 'ROLLBACK');

                        --$this->in_transaction;

                        return false;
                }
        }


        function result($query_id = 0, $row = 0)
        {
                if ($query_id)
                {
                        if ($row != 0)
                                @sqlite_seek($query_id, $row);

                        return @current(@sqlite_current($query_id));
                }
                else
                        return false;
        }


        function fetch_assoc($query_id = 0)
        {
                if ($query_id)
                {
                        $cur_row = @sqlite_fetch_array($query_id, SQLITE_ASSOC);
                        if ($cur_row)
                        {
                                // Horrible hack to get rid of table names and table aliases from the array keys
                                while (list($key, $value) = @each($cur_row))
                                {
                                    $dot_spot = strpos($key, '.');
                                    if ($dot_spot !== false)
                                    {
                                        unset($cur_row[$key]);
                                        $key = substr($key, $dot_spot+1);
                                        $cur_row[$key] = $value;
                                    }
                                }
                        }

                        return $cur_row;
                }
                else
                        return false;
        }


        function fetch_row($query_id = 0)
        {
                return ($query_id) ? @sqlite_fetch_array($query_id, SQLITE_NUM) : false;
        }


        function num_rows($query_id = 0)
        {
                return ($query_id) ? @sqlite_num_rows($query_id) : false;
        }


        function affected_rows()
        {
                return ($this->query_result) ? @sqlite_changes($this->query_result) : false;
        }


        function insert_id()
        {
                return ($this->link_id) ? @sqlite_last_insert_rowid($this->link_id) : false;
        }


        function get_num_queries()
        {
                return $this->num_queries;
        }


        function get_saved_queries()
        {
                return $this->saved_queries;
        }


        function free_result($query_id = false)
        {
                return true;
        }


        function escape($str)
        {
                return sqlite_escape_string($str);
        }


        function error()
        {
                $result['error_sql'] = @current(@end($this->saved_queries));
                $result['error_no'] = $this->error_no;
                $result['error_msg'] = $this->error_msg;

                return $result;
        }


        function close()
        {
                if ($this->link_id)
                {
                        if ($this->in_transaction)
                        {
                                if (defined('PUN_SHOW_QUERIES'))
                                        $this->saved_queries[] = array('COMMIT', 0);

                                @sqlite_query($this->link_id, 'COMMIT');
                        }

                        return @sqlite_close($this->link_id);
                }
                else
                        return false;
        }
}