Kleeja/includes/mysqli.php

<?php
/**
*
* @package Kleeja
* @copyright (c) 2007 Kleeja.com
* @license ./docs/license.txt
*
*/


//no for directly open
if (! defined('IN_COMMON'))
{
    exit();
}  

if (! defined('SQL_LAYER')):

define('SQL_LAYER', 'mysqli');

class KleejaDatabase
{
    public $connect_id		= null;		
    public $result;
    public $query_num		    = 0;
    public $in_transaction = 0;
    public $debugr			      = false;
    public $show_errors 	  = true;


    /*
     * initiate the class
     * with basic data
    */
    public function __construct($host, $db_username, $db_password, $db_name, $new_link = false)
    {
        global $script_encoding;

        $host 	.= strpos($host, ':') !== false ? '' : ':';
        $this->host 		      = substr($host, 0, strpos($host, ':'));
        $this->port 		      = (int) substr($host, strpos($host, ':')+1);
        $this->db_username 	= $db_username;
        $this->db_name     	= $db_name;
        $this->db_password 	= 'hidden';

        $this->connect_id = @mysqli_connect($this->host, $this->db_username, $db_password, $this->db_name, (! $this->port ? 3306 : $this->port ));

        //no error
        if (defined('MYSQL_NO_ERRORS'))
        {
            $this->show_errors = false;
        }


        if (! $this->connect_id)
        {
            //loggin -> no database -> close connection
            $this->close();
            $this->error_msg('we can not connect to the server ...');
            return false;
        }

        //loggin -> connecting 
        kleeja_log('[Connected] : ' . kleeja_get_page());


        if ((! preg_match('/utf/i', strtolower($script_encoding)) && ! defined('IN_LOGINPAGE') && ! defined('IN_ADMIN_LOGIN') && ! defined('DISABLE_INTR')) || (empty($script_encoding) || preg_match('/utf/i', strtolower($script_encoding)) || defined('DISABLE_INTR')))
        {
            if (mysqli_set_charset($this->connect_id, 'utf8'))
            {
                //loggin -> set utf8 
                kleeja_log('[Set to UTF8] : --> ');
            }
        }

        return $this->connect_id;
    }

    public function is_connected()
    {
        return ! (is_null($this->connect_id) || empty($this->connect_id));
    }

    // close the connection
    public function close()
    {
        if ($this->connect_id)
        {
            // Commit any remaining transactions
            if ($this->in_transaction)
            {
                mysqli_commit($this->connect_id);
            }

            //loggin -> close connection
            kleeja_log('[Closing connection] : ' . kleeja_get_page());

            return @mysqli_close($this->connect_id);
        }
        else
        {
            return false;
        }
    }

    // encoding functions
    public function set_utf8()
    {
        $this->set_names('utf8');
    }

    public function set_names($charset)
    {
        @mysqli_set_charset($this->connect_id, $charset);
    }

    public function client_encoding()
    {
        return mysqli_character_set_name($this->connect_id);
    }

    public function mysql_version()
    {
        $vr = $this->query('SELECT VERSION() AS v');
        $vs = $this->fetch_array($vr);
        $vs = $vs['v'];
        return preg_replace('/^([^-]+).*$/', '\\1', $vs);
    }

    /*
     * the query func . its so important to do 
     * the quries and give results
     */
    public function query($query, $transaction = false)
    {
        //no connection
        if (! $this->connect_id)
        {
            return false;
        }

        //
        // Remove any pre-existing queries
        //
        unset($this->result);

        if (! empty($query))
        {
            //debug .. //////////////
            $srartum_sql = get_microtime();
            ////////////////

            if ($transaction == 1 && ! $this->in_transaction)
            {
                if (! mysqli_autocommit($this->connect_id, false))
                {
                    return false;
                }

                $this->in_transaction = true;
            }

            $this->result = mysqli_query($this->connect_id, $query);

            //debug .. //////////////
            $this->debugr[$this->query_num+1] = [$query, sprintf('%.5f', get_microtime() - $srartum_sql)];
            ////////////////

            if (! $this->result)
            {
                $this->error_msg('Error In query');
            }
            else
            {
                //let's debug it
                kleeja_log('[Query] : --> ' . $query);
            }
        }
        else
        {
            if ( $transaction == 2 && $this->in_transaction )
            {
                $this->result = mysqli_commit($this->connect_id);
            }
        }

        //is there any result
        if ($this->result)
        {
            if ($transaction == 2 && $this->in_transaction)
            {
                $this->in_transaction = false;

                if (! mysqli_commit($this->connect_id))
                {
                    mysqli_rollback($this->connect_id);
                    return false;
                }
            }

            $this->query_num++;
            return $this->result;
        }
        else
        {
            if ($this->in_transaction)
            {
                mysqli_rollback($this->connect_id);
                $this->in_transaction = false;
            }
            return false;
        }
    }

    // query build
    public function build($query)
    {
        $sql = '';

        if (isset($query['SELECT']))
        {
            $sql = 'SELECT ' . $query['SELECT'] . ' FROM ' . $query['FROM'];

            if (isset($query['JOINS']))
            {
                foreach ($query['JOINS'] as $cur_join)
                {
                    $sql .= ' ' . key($cur_join) . ' ' . @current($cur_join) . ' ON ' . $cur_join['ON'];
                }
            }

            if (! empty($query['WHERE']))
            {
                $sql .= ' WHERE ' . $query['WHERE'];
            }

            if (! empty($query['GROUP BY']))
            {
                $sql .= ' GROUP BY ' . $query['GROUP BY'];
            }

            if (! empty($query['HAVING']))
            {
                $sql .= ' HAVING ' . $query['HAVING'];
            }

            if (! empty($query['ORDER BY']))
            {
                $sql .= ' ORDER BY ' . $query['ORDER BY'];
            }

            if (! empty($query['LIMIT']))
            {
                $sql .= ' LIMIT ' . $query['LIMIT'];
            }
        }
        elseif (isset($query['INSERT']))
        {
            $sql = 'INSERT INTO ' . $query['INTO'];

            if (! empty($query['INSERT']))
            {
                $sql .= ' (' . $query['INSERT'] . ')';
            }

            $sql .= ' VALUES(' . $query['VALUES'] . ')';
        }
        elseif (isset($query['UPDATE']))
        {
            if (isset($query['PARAMS']['LOW_PRIORITY']))
            {
                $query['UPDATE'] = 'LOW_PRIORITY ' . $query['UPDATE'];
            }

            $sql = 'UPDATE ' . $query['UPDATE'] . ' SET ' . $query['SET'];

            if (! empty($query['WHERE']))
            {
                $sql .= ' WHERE ' . $query['WHERE'];
            }
        }
        elseif (isset($query['DELETE']))
        {
            $sql = 'DELETE FROM ' . $query['DELETE'];

            if (! empty($query['WHERE']))
            {
                $sql .= ' WHERE ' . $query['WHERE'];
            }
        }
        elseif (isset($query['REPLACE']))
        {
            $sql = 'REPLACE INTO ' . $query['INTO'];

            if (! empty($query['REPLACE']))
            {
                $sql .= ' (' . $query['REPLACE'] . ')';
            }

            $sql .= ' VALUES(' . $query['VALUES'] . ')';
        }

        return $this->query($sql);
    }

    // free the memmory from the last results
    public function free($query_id = 0)
    {
        return $this->freeresult($query_id);
    }

    public function freeresult($query_id = 0)
    {
        if (! $query_id)
        {
            $query_id = $this->result;
        }

        if ($query_id)
        {
            mysqli_free_result($query_id);
            return true;
        }
        else
        {
            return false;
        }
    }

    /*
     * if the result is an arry ,
     * this func is so important to order them as a array
     */
    public function fetch($query_id = 0)
    {
        return $this->fetch_array($query_id);
    }

    public function fetch_array($query_id = 0)
    {
        if (! $query_id)
        {
            $query_id = $this->result;
        }

        return $query_id  ? mysqli_fetch_array($query_id, MYSQLI_ASSOC) : false;
    }

    /*
     * if we have a result and we have to know 
     * the number of it , this is a func ..
     */
    public function num_rows($query_id = 0)
    {
        if (! $query_id)
        {
            $query_id = $this->result;
        }

        return $query_id ? mysqli_num_rows($query_id) : false;
    }


    // last id inserted in sql
    public function insert_id()
    {
        return $this->connect_id ? mysqli_insert_id($this->connect_id) : false;
    }

    // clean the qurery before insert it
    public function escape($msg)
    {
        $msg = htmlspecialchars($msg, ENT_QUOTES);
        //$msg = (!get_magic_quotes_gpc()) ? addslashes ($msg) : $msg;
        $msg = $this->real_escape($msg);
        return $msg;
    }

    // real escape ..
    /**
     * @param  string|array $msg
     * @return int|string
     */
    public function real_escape($msg)
    {
        if (is_array($msg) && ! $this->connect_id)
        {
            return $msg;
        }

        if (! $this->connect_id)
        {
            return 0;
        }

        //escaping _ made alot of problems
        //return addcslashes(mysqli_real_escape_string($this->connect_id, $msg), '%_');
        return mysqli_real_escape_string($this->connect_id, $msg);
    }

    // get affected records
    public function affected()
    {
        return $this->connect_id ? mysqli_affected_rows($this->connect_id) : false;
    }

    // get the information of mysql server
    public function server_info()
    {
        return 'MySQLi ' . $this->mysql_version;
    }

    // error message func
    public function error_msg($msg)
    {
        global $dbprefix;

        if (! $this->show_errors)
        {
            return false;
        }

        $error_no  = $this->connect_id ? @mysqli_errno($this->connect_id) : @mysqli_connect_errno();
        $error_msg = $this->connect_id ? @mysqli_error($this->connect_id) : @mysqli_connect_error();
        $error_sql = @current($this->debugr[$this->query_num+1]);

        //some ppl want hide their table names
        if (! defined('DEV_STAGE'))
        {
            $error_sql = preg_replace("#\s{1,3}`*{$dbprefix}([a-z0-9]+)`*\s{1,3}#e", "' <span style=\"color:blue\">' . substr('$1', 0, 1) . '</span> '", $error_sql);
            $error_msg = preg_replace("#{$this->db_name}.{$dbprefix}([a-z0-9]+)#e", "' <span style=\"color:blue\">' . substr('$1', 0, 1) . '</span> '", $error_msg);
            $error_sql = preg_replace("#\s{1,3}(from|update|into)\s{1,3}([a-z0-9]+)\s{1,3}#ie", "' $1 <span style=\"color:blue\">' . substr('$2', 0, 1) . '</span> '", $error_sql);
            $error_msg = preg_replace("#\s{1,3}(from|update|into)\s{1,3}([a-z0-9]+)\s{1,3}#ie", "' $1 <span style=\"color:blue\">' . substr('$2', 0, 1) . '</span> '", $error_msg);
            $error_msg = preg_replace("#\s'([^']+)'@'([^']+)'#ie", "' <span style=\"color:blue\">hidden</span>@$2 '", $error_msg);
            $error_sql = preg_replace("#password\s*=\s*'[^']+'#i", "password='<span style=\"color:blue\">hidden</span>'", $error_sql);
        }

        //is this error related to updating?
        $updating_related = false;

        if (strpos($error_msg, 'Unknown column') !== false)
        {
            $updating_related = true;
        }

        header('HTTP/1.1 500 Internal Server Error');
        $error_message = '<html><head><title>ERROR IM MYSQL</title>';
        $error_message .= "<style>BODY{font-family:'Tahoma',serif;font-size:12px;}.error {}</style></head><body>";
        $error_message .= '<br />';
        $error_message .= '<div class="error">';
        $error_message .= " <a href='#' onclick='window.location.reload( false );'>click to Refresh this page ...</a><br />";
        $error_message .= '<h2>Sorry , There is an error in mysql ' . ($msg !='' ? ", error : $msg" : '') . '</h2>';

        if ($error_sql != '')
        {
            $error_message .= "<br />--[query]-------------------------- <br />$error_sql<br />---------------------------------<br /><br />";
        }
        $error_message .= "[$error_no : $error_msg] <br />";

        if ($updating_related)
        {
            global $config;
            $error_message .= '<br /><strong>Your Kleeja database might be old, try to update it now from: ' . rtrim($config['siteurl'], '/') . '/install</strong>';
            $error_message .= "<br /><br><strong>If this error happened after installing a plugin, add <span style=\"background-color:#ccc; padding:2px\">define('STOP_PLUGINS', true);</span> to end of config.php file.</strong>";
        }
        $error_message .= "<br /><br /><strong>Script: Kleeja <br /><a href='http://www.kleeja.com'>Kleeja Website</a></strong>";
        $error_message .= '</b></div>';
        $error_message .= '</body></html>';


        print $error_message;


        //loggin -> error
        kleeja_log('[SQL ERROR] : "' . $error_no . ' : ' . $error_msg . '" -->');

        @$this->close();

        exit();
    }

    // return last error
    public function get_error()
    {
        if ($this->connect_id)
        {
            return [@mysqli_errno($this->connect_id), @mysqli_error($this->connect_id)];
        }
        else
        {
            return [@mysqli_connect_errno(), @mysqli_connect_error()];
        }
    }
}//end of class

endif;