526 lines
12 KiB
PHP
526 lines
12 KiB
PHP
<?php
|
|
/**
|
|
* Query
|
|
*
|
|
* Free Query Builder / Database Abstraction Layer
|
|
*
|
|
* @package Query
|
|
* @author Timothy J. Warren
|
|
* @copyright Copyright (c) 2012 - 2013
|
|
* @link https://github.com/aviat4ion/Query
|
|
* @license http://philsturgeon.co.uk/code/dbad-license
|
|
*/
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Base Database class
|
|
*
|
|
* Extends PDO to simplify cross-database issues
|
|
*
|
|
* @package Query
|
|
* @subpackage Query
|
|
*/
|
|
abstract class DB_PDO extends PDO {
|
|
|
|
// Reference to the last executed query
|
|
protected $statement;
|
|
|
|
// Character to escape identifiers
|
|
protected $escape_char = '"';
|
|
|
|
// Reference to sql sub class
|
|
public $sql;
|
|
|
|
// Reference to util sub class
|
|
public $util;
|
|
|
|
// Last query executed
|
|
public $last_query;
|
|
|
|
// Prefix to apply to table namesa
|
|
public $table_prefix = '';
|
|
|
|
/**
|
|
* PDO constructor wrapper
|
|
*
|
|
* @param string $dsn
|
|
* @param string $username
|
|
* @param string $password
|
|
* @param array $driver_options
|
|
*/
|
|
public function __construct($dsn, $username=NULL, $password=NULL, $driver_options=array())
|
|
{
|
|
parent::__construct($dsn, $username, $password, $driver_options);
|
|
|
|
// Load the sql class for the driver
|
|
$class = get_class($this)."_sql";
|
|
$this->sql = new $class();
|
|
|
|
// Load the util class for the driver
|
|
$class = get_class($this)."_util";
|
|
$this->util = new $class($this);
|
|
|
|
$this->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
|
|
|
// Set additional driver options, if they exist
|
|
if ( ! empty($driver_options) && is_array($driver_options))
|
|
{
|
|
foreach($driver_options as $key => $val)
|
|
{
|
|
$this->setAttribute($key, $val);
|
|
}
|
|
}
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
// ! Concrete functions that can be overridden in child classes
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Simplifies prepared statements for database queries
|
|
*
|
|
* @param string $sql
|
|
* @param array $data
|
|
* @return mixed PDOStatement / FALSE
|
|
*/
|
|
public function prepare_query($sql, $data)
|
|
{
|
|
// Prepare the sql
|
|
$query = $this->prepare($sql);
|
|
|
|
if( ! (is_object($query) || is_resource($query)))
|
|
{
|
|
$this->get_last_error();
|
|
return FALSE;
|
|
}
|
|
|
|
// Set the statement in the class variable for easy later access
|
|
$this->statement = $query;
|
|
|
|
|
|
if( ! (is_array($data) || is_object($data)))
|
|
{
|
|
trigger_error("Invalid data argument");
|
|
return FALSE;
|
|
}
|
|
|
|
// Bind the parameters
|
|
foreach($data as $k => $value)
|
|
{
|
|
if(is_numeric($k))
|
|
{
|
|
$k++;
|
|
}
|
|
|
|
$res = $query->bindValue($k, $value);
|
|
|
|
if( ! $res)
|
|
{
|
|
trigger_error("Parameter not successfully bound");
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
return $query;
|
|
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Create and execute a prepared statement with the provided parameters
|
|
*
|
|
* @param string $sql
|
|
* @param array $params
|
|
* @return PDOStatement
|
|
*/
|
|
public function prepare_execute($sql, $params)
|
|
{
|
|
$this->statement = $this->prepare_query($sql, $params);
|
|
$this->statement->execute();
|
|
|
|
return $this->statement;
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Retreives the data from a select query
|
|
*
|
|
* @param PDOStatement $statement
|
|
* @return array
|
|
*/
|
|
public function get_query_data($statement)
|
|
{
|
|
$this->statement =& $statement;
|
|
|
|
// Execute the query
|
|
$this->statement->execute();
|
|
|
|
// Return the data array fetched
|
|
return $this->statement->fetchAll(PDO::FETCH_ASSOC);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Returns number of rows affected by an INSERT, UPDATE, DELETE type query
|
|
*
|
|
* @param PDOStatement $statement
|
|
* @return int
|
|
*/
|
|
public function affected_rows($statement='')
|
|
{
|
|
if ( ! empty($statement))
|
|
{
|
|
$this->statement = $statement;
|
|
}
|
|
|
|
// Return number of rows affected
|
|
return $this->statement->rowCount();
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return the last error for the current database connection
|
|
*
|
|
* @return string
|
|
*/
|
|
public function get_last_error()
|
|
{
|
|
list($i1, $i2, $i3) = $this->errorInfo();
|
|
|
|
echo "Error: <pre>{$i1}:{$i2}\n{$i3}</pre>";
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Quote database table name, and set prefix
|
|
*
|
|
* @param string $table
|
|
* @return string
|
|
*/
|
|
public function quote_table($table)
|
|
{
|
|
// If there isn't a prefix set, just quote the table name
|
|
if (empty($this->table_prefix))
|
|
{
|
|
return $this->quote_ident($table);
|
|
}
|
|
|
|
// Split indentifier by period, will split into:
|
|
// database.schema.table OR
|
|
// schema.table OR
|
|
// database.table OR
|
|
// table
|
|
$idents = (array) explode('.', $table);
|
|
$segments = count($idents);
|
|
|
|
// Reference the last item in the split string
|
|
$last =& $idents[$segments - 1];
|
|
|
|
// Quote the last item
|
|
$last = $this->_prefix($last);
|
|
|
|
// Rejoin
|
|
$table = implode('.', $idents);
|
|
|
|
// Finally, quote the table
|
|
return $this->quote_ident($table);
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Sets the table prefix on the passed string
|
|
*
|
|
* @param string $str
|
|
* @return string
|
|
*/
|
|
protected function _prefix($str)
|
|
{
|
|
// Don't prefix an already prefixed table
|
|
if (strpos($str, $this->table_prefix) !== FALSE)
|
|
{
|
|
return $str;
|
|
}
|
|
|
|
return $this->table_prefix.$str;
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Surrounds the string with the databases identifier escape characters
|
|
*
|
|
* @param mixed $ident
|
|
* @return string
|
|
*/
|
|
public function quote_ident($ident)
|
|
{
|
|
if (is_array($ident))
|
|
{
|
|
return array_map(array($this, __METHOD__), $ident);
|
|
}
|
|
|
|
// Handle comma-separated identifiers
|
|
if (strpos($ident, ',') !== FALSE)
|
|
{
|
|
$parts = explode(',', $ident);
|
|
$parts = array_map('mb_trim', $parts);
|
|
$parts = array_map(array($this, __METHOD__), $parts);
|
|
$ident = implode(',', $parts);
|
|
}
|
|
|
|
// Split each identifier by the period
|
|
$hiers = explode('.', $ident);
|
|
$hiers = array_map('mb_trim', $hiers);
|
|
|
|
// Return the re-compiled string
|
|
return implode('.', array_map(array($this, '_quote'), $hiers));
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Helper method for quote_ident
|
|
*
|
|
* @param mixed $str
|
|
* @return mixed
|
|
*/
|
|
protected function _quote($str)
|
|
{
|
|
// Don't quote numbers
|
|
if ( ! is_string($str) && is_numeric($str))
|
|
{
|
|
return $str;
|
|
}
|
|
|
|
// Don't add additional quotes
|
|
if (strpos($str, $this->escape_char) === 0 || strrpos($str, $this->escape_char) === 0)
|
|
{
|
|
return $str;
|
|
}
|
|
|
|
return "{$this->escape_char}{$str}{$this->escape_char}";
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Deletes all the rows from a table. Does the same as the truncate
|
|
* method if the database does not support 'TRUNCATE';
|
|
*
|
|
* @param string $table
|
|
* @return mixed
|
|
*/
|
|
public function empty_table($table)
|
|
{
|
|
$sql = 'DELETE FROM '.$this->quote_ident($table);
|
|
|
|
return $this->query($sql);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return schemas for databases that list them
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_schemas()
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of tables for the current database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_tables()
|
|
{
|
|
return $this->driver_query($this->sql->table_list());
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of dbs for the current connection, if possible
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_dbs()
|
|
{
|
|
return $this->driver_query($this->sql->db_list());
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of views for the current database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_views()
|
|
{
|
|
return $this->driver_query($this->sql->view_list());
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of sequences for the current database, if they exist
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_sequences()
|
|
{
|
|
return $this->driver_query($this->sql->sequence_list());
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of function for the current database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_functions()
|
|
{
|
|
return $this->driver_query($this->sql->function_list(), FALSE);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of stored procedures for the current database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_procedures()
|
|
{
|
|
return $this->driver_query($this->sql->procedure_list(), FALSE);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return list of triggers for the current database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_triggers()
|
|
{
|
|
return $this->driver_query($this->sql->trigger_list(), FALSE);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Retreives an array of non-user-created tables for
|
|
* the connection/database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_system_tables()
|
|
{
|
|
return $this->driver_query($this->sql->system_table_list());
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Retrieve column information for the current database table
|
|
*
|
|
* @param string $table
|
|
* @return array
|
|
*/
|
|
public function get_columns($table)
|
|
{
|
|
return $this->driver_query($this->sql->column_list($table), FALSE);
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Retrieve list of data types for the database
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_types()
|
|
{
|
|
return $this->driver_query($this->sql->type_list(), FALSE);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Method to simplify retreiving db results for meta-data queries
|
|
*
|
|
* @param string $sql
|
|
* @param bool $filtered_index
|
|
* @return mixed
|
|
*/
|
|
public function driver_query($sql, $filtered_index=TRUE)
|
|
{
|
|
// Return if the query doesn't apply to the driver
|
|
if ($sql === FALSE)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
// Return predefined data
|
|
if (is_array($sql))
|
|
{
|
|
return $sql;
|
|
}
|
|
|
|
$res = $this->query($sql);
|
|
|
|
$flag = ($filtered_index) ? PDO::FETCH_NUM : PDO::FETCH_ASSOC;
|
|
$all = $res->fetchAll($flag);
|
|
|
|
return ($filtered_index) ? db_filter($all, 0) : $all;
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Return the number of rows returned for a SELECT query
|
|
*
|
|
* @see http://us3.php.net/manual/en/pdostatement.rowcount.php#87110
|
|
* @return int
|
|
*/
|
|
public function num_rows()
|
|
{
|
|
$regex = '/^SELECT\s+(?:ALL\s+|DISTINCT\s+)?(?:.*?)\s+FROM\s+(.*)$/i';
|
|
$output = array();
|
|
|
|
if (preg_match($regex, $this->last_query, $output) > 0)
|
|
{
|
|
$stmt = $this->query("SELECT COUNT(*) FROM {$output[1]}", PDO::FETCH_NUM);
|
|
return $stmt->fetchColumn();
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
// ! Abstract public functions to implement in child classes
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* Empty the passed table
|
|
*
|
|
* @param string $table
|
|
* @return void
|
|
*/
|
|
abstract public function truncate($table);
|
|
}
|
|
// End of db_pdo.php
|