OpenSQLManager/sys/db/query_builder.php

604 lines
13 KiB
PHP
Raw Normal View History

<?php
/**
* OpenSQLManager
*
* Free Database manager for Open Source Databases
*
* @author Timothy J. Warren
* @copyright Copyright (c) 2012
* @link https://github.com/aviat4ion/OpenSQLManager
* @license http://philsturgeon.co.uk/code/dbad-license
*/
// --------------------------------------------------------------------------
/**
* Convienience class for creating sql queries - also the class that
* instantiates the specific db driver
*/
class Query_Builder {
2012-03-09 14:27:13 -05:00
private $table,
$sql,
$select_string,
$from_string,
$where_array,
2012-03-09 15:55:56 -05:00
$where_string,
2012-03-12 16:43:02 -04:00
$insert_string,
$update_string,
2012-03-12 15:28:52 -04:00
$set_array,
2012-03-13 09:59:34 -04:00
$set_array_keys,
2012-03-12 15:28:52 -04:00
$set_string,
2012-03-09 15:55:56 -05:00
$limit,
$offset;
/**
* Constructor
*
2012-03-07 18:57:52 -05:00
* @param object $conn_name - the name of the connection/parameters
*/
2012-03-07 18:57:52 -05:00
public function __construct($params)
{
$params->type = strtolower($params->type);
$dbtype = ($params->type !== 'postgresql') ? $params->type : 'pgsql';
2012-03-07 18:57:52 -05:00
// Initiate the constructor for the selected database
switch($dbtype)
{
default:
$this->db = new $dbtype("host={$params->host};port={$params->port};", $params->user, $params->pass);
break;
case "sqlite":
2012-03-01 12:05:32 -05:00
if ( ! empty($params->user) && ! empty($params->pass))
{
$this->db = new $dbtype($params->file, $params->user, $params->pass);
}
else
{
$this->db = new $dbtype($params->file);
}
break;
case "firebird":
$this->db = new $dbtype("{$params->host}:{$params->file}", $params->user, $params->pass);
break;
}
2012-03-08 12:37:33 -05:00
// Make things just slightly shorter
$this->sql =& $this->db->sql;
2012-03-01 12:05:32 -05:00
}
2012-03-12 16:17:30 -04:00
// --------------------------------------------------------------------------
// ! Select Queries
// --------------------------------------------------------------------------
/**
* Specifies rows to select in a query
*
* @param string $fields
* @return $this
*/
public function select($fields)
{
2012-03-09 08:04:47 -05:00
// Split fields by comma
$fields_array = explode(",", $fields);
$fields_array = array_map('trim', $fields_array);
// Split on 'As'
foreach ($fields_array as $key => $field)
{
if (stripos($field, 'as') !== FALSE)
{
$fields_array[$key] = preg_split('`as`i', $field);
2012-03-09 08:26:16 -05:00
$fields_array[$key] = array_map('trim', $fields_array[$key]);
2012-03-09 08:04:47 -05:00
}
}
// Quote the identifiers
$safe_array = array_map(array($this->db, 'quote_ident'), $fields_array);
2012-03-09 12:52:35 -05:00
unset($fields_array);
2012-03-09 08:04:47 -05:00
// Join the strings back together
2012-03-09 08:26:16 -05:00
for($i = 0, $c = count($safe_array); $i < $c; $i++)
2012-03-09 08:04:47 -05:00
{
2012-03-09 08:10:37 -05:00
if (is_array($safe_array[$i]))
2012-03-09 08:04:47 -05:00
{
$safe_array[$i] = implode(' AS ', $safe_array[$i]);
}
}
$this->select_string = implode(', ', $safe_array);
2012-03-09 08:26:16 -05:00
2012-03-09 12:52:35 -05:00
unset($safe_array);
2012-03-09 08:04:47 -05:00
return $this;
}
// --------------------------------------------------------------------------
2012-03-12 16:17:30 -04:00
/**
* Specify the database table to select from
*
* @param string $dbname
* @return $this
*/
public function from($dbname)
{
// Split identifiers on spaces
$ident_array = explode(' ', trim($dbname));
$ident_array = array_map('trim', $ident_array);
// Quote the identifiers
$ident_array = array_map(array($this->db, 'quote_ident'), $ident_array);
// Paste it back together
$this->from_string = implode(' ', $ident_array);
return $this;
}
// --------------------------------------------------------------------------
/**
* Creates a Like clause in the sql statement
*
* @param string $field
* @param mixed $val
* @param string $pos
* @return $this
*/
public function like($field, $val, $pos='both')
{
// @todo implement like method
return $this;
}
// --------------------------------------------------------------------------
/**
* Specify condition(s) in the where clause of a query
* Note: this function works with key / value, or a
* passed array with key / value pairs
*
* @param mixed $key
* @param mixed $val
* @return $this
*/
public function where($key, $val=array())
{
2012-03-09 11:56:14 -05:00
// Key and value passed? Add them to the where array
if (is_scalar($key) && is_scalar($val))
{
$this->where_array[$key] = $val;
}
// Array or object, loop through and add to the where array
elseif ( ! is_scalar($key))
{
foreach($key as $k => $v)
{
$this->where_array[$k] = $v;
}
}
// The values are irrelevant until the query is actually run
$fields = array_keys($this->where_array);
// Array of conditions
$kv_array = array();
// Create key/value placeholders
foreach($fields as $f)
{
// Split each key by spaces, incase there
// is an operator such as >, <, !=, etc.
$f_array = explode(' ', trim($f));
// Simple key = val
if (count($f_array) === 1)
{
$kv_array[] = $this->db->quote_ident($f_array[0]) . '= ?';
}
else // Other operators
{
$kv_array[] = $this->db->quote_ident($f_array[0]) . " {$f_array[1]} ?";
}
}
// Create the where portion of the string
2012-03-09 14:27:13 -05:00
$this->where_string = ' WHERE '.implode(' AND ', $kv_array);
2012-03-09 12:52:35 -05:00
unset($kv_array);
unset($fields);
2012-03-09 11:56:14 -05:00
return $this;
}
// --------------------------------------------------------------------------
/**
* Where clause prefixed with "OR"
*
* @param string $field
* @param mixed $value
* @return $this
*/
public function or_where($field, $value)
{
// @todo Implement or_where method
return $this;
}
// --------------------------------------------------------------------------
/**
* Where clause with 'IN' statement
*
* @param mixed $field
* @param mixed $val
* @return $this
*/
public function where_in($field, $val)
{
// @todo Implement Where_in method
return $this;
}
// --------------------------------------------------------------------------
/**
* Where in statement prefixed with "or"
*
* @param string $field
* @param mixed $val
* @return $this
*/
public function or_where_in($field, $val)
{
// @todo Implement or_where_in method
return $this;
}
// --------------------------------------------------------------------------
/**
* WHERE NOT IN (FOO) clause
*
* @param string $field
* @param mixed $val
* @return $this
*/
public function where_not_in($field, $val)
{
// @todo Implement where_not_in method
return $this;
}
// --------------------------------------------------------------------------
/**
* OR WHERE NOT IN (FOO) clause
*
* @param string $field
* @param mixed $val
* @return $this
*/
public function or_where_not_in($field, $val)
{
2012-03-12 16:17:30 -04:00
// @todo Implement or_where_not_in method
return $this;
}
2012-03-12 16:17:30 -04:00
// --------------------------------------------------------------------------
/**
* Creates a join phrase in a compiled query
*
* @param string $table
* @param string $condition
* @param string $type
* @return $this
*/
public function join($table, $condition, $type='inner')
{
// @todo Implement join method
return $this;
}
// --------------------------------------------------------------------------
2012-03-12 16:43:02 -04:00
/**
* Group the results by the selected field(s)
*
* @param mixed $field
* @return $this
*/
public function group_by($field)
{
// @todo Implement group_by method
return $this;
}
// --------------------------------------------------------------------------
/**
* Order the results by the selected field(s)
*
* @param string $field
* @param string $type
* @return $this
*/
public function order_by($field, $type="")
{
// @todo implement order_by method
return $this;
}
// --------------------------------------------------------------------------
/**
2012-03-12 16:17:30 -04:00
* Set a limit on the current sql statement
*
2012-03-12 16:17:30 -04:00
* @param int $limit
* @param int $offset
* @return string
*/
2012-03-12 16:17:30 -04:00
public function limit($limit, $offset=FALSE)
{
2012-03-12 16:17:30 -04:00
$this->limit = $limit;
$this->offset = $offset;
2012-03-09 14:27:13 -05:00
return $this;
}
2012-03-08 13:29:30 -05:00
// --------------------------------------------------------------------------
2012-03-12 16:17:30 -04:00
2012-03-09 15:55:56 -05:00
/**
2012-03-12 16:17:30 -04:00
* Select and retrieve all records from the current table, and/or
* execute current compiled query
2012-03-09 15:55:56 -05:00
*
2012-03-12 16:17:30 -04:00
* @param $table
2012-03-09 15:55:56 -05:00
* @param int $limit
* @param int $offset
2012-03-12 16:17:30 -04:00
* @return object
2012-03-09 15:55:56 -05:00
*/
2012-03-12 16:17:30 -04:00
public function get($table='', $limit=FALSE, $offset=FALSE)
2012-03-09 15:55:56 -05:00
{
2012-03-12 16:17:30 -04:00
// Set the table
if ( ! empty($table))
{
$this->from_string = $this->db->quote_ident($table);
}
// Set the limit, if it exists
if ($limit !== FALSE)
{
$this->limit($limit, $offset);
}
2012-03-09 15:55:56 -05:00
2012-03-12 16:17:30 -04:00
$sql = $this->_compile('select');
// Do prepared statements for anything involving a "where" clause
if ( ! empty($this->where_string))
{
$result = $this->db->prepare_execute($sql, array_values($this->where_array));
}
else
{
// Otherwise, a simple query will do.
$result = $this->db->query($sql);
}
// Reset for next query
$this->_reset();
return $result;
2012-03-09 15:55:56 -05:00
}
2012-03-12 16:17:30 -04:00
// --------------------------------------------------------------------------
// ! Insert/Update/Delete Queries
// --------------------------------------------------------------------------
2012-03-12 16:17:30 -04:00
/**
* Sets values for inserts / updates / deletes
*
* @param mixed $key
* @param mixed $val
* @return $this
*/
public function set($key, $val)
{
// Plain key, value pair
if (is_scalar($key) && is_scalar($val))
{
$this->set_array[$key] = $val;
}
// Object or array
elseif ( ! is_scalar($key))
{
foreach($key as $k => $v)
{
$this->set_array[$k] = $v;
}
}
// Use the keys of the array to make the insert/update string
2012-03-13 09:59:34 -04:00
$this->set_array_keys = array_keys($this->set_array);
2012-03-12 16:17:30 -04:00
// Escape the field names
2012-03-13 09:59:34 -04:00
$this->set_array_keys = array_map(array($this, 'quote_ident'), $this->set_array_keys);
2012-03-12 16:17:30 -04:00
// Generate the "set" string
2012-03-13 09:59:34 -04:00
$this->set_string = implode('=?, ', $this->set_array_keys);
2012-03-12 16:17:30 -04:00
$this->set_string .= '=?';
return $this;
}
// --------------------------------------------------------------------------
/**
* Creates an insert clause, and executes it
*
* @param string $table
* @param mixed $data
* @return mixed
*/
public function insert($table, $data=array())
{
// No use duplicating logic!
if ( ! empty($data))
{
$this->set($data);
}
$params = array_values($this->set_array);
$sql = $this->_compile("insert", $table);
return $this->db->prepare_execute($sql, $params);
}
// --------------------------------------------------------------------------
/**
* Creates an update clause, and executes it
*
* @param string $table
* @param mixed $data
2012-03-13 09:59:34 -04:00
* @return mixed
*/
public function update($table, $data=array())
{
// No use duplicating logic!
if ( ! empty($data))
{
$this->set($data);
}
2012-03-13 09:59:34 -04:00
$sql = 'UPDATE '.$this->quote_ident($table). ' SET '. $this->set_string;
$params = array_values($this->set_array);
// Do a linear array merge if there is a where string.
// We need all the parameters to line up, even when
// there are placeholders in the where string and
// the set string
if ( ! empty($this->where_string))
{
$sql .= $this->where_string;
$where_params = array_values($this->where_array);
foreach($where_params as $w)
{
$params[] = $w;
}
}
// Run the query
return $this->db->prepare_execute($sql, $params);
}
2012-03-09 15:55:56 -05:00
// --------------------------------------------------------------------------
/**
* Deletes data from a table
*
* @param string $table
* @param mixed $where
* @return
*/
public function delete($table, $where='')
{
// @todo implement delete method
}
2012-03-09 15:55:56 -05:00
// --------------------------------------------------------------------------
/**
* Clear out the class variables, so the next query can be run
*/
private function _reset()
{
// Only unset class variables that
// are not callable. Otherwise, we'll
// delete class methods!
foreach($this as $name => $var)
{
// Skip properties that are needed for every query
$save_properties = array(
'db',
'sql'
);
if (in_array($name, $save_properties))
{
continue;
}
// Nothing query-generation related is safe!
if ( ! is_callable($this->$name))
{
unset($this->$name);
}
}
}
// --------------------------------------------------------------------------
2012-03-08 13:29:30 -05:00
/**
* String together the sql statements for sending to the db
*
* @param string $type
* @param string $table
2012-03-08 13:29:30 -05:00
* @return $string
*/
private function _compile($type="select", $table="")
2012-03-08 13:29:30 -05:00
{
2012-03-09 15:55:56 -05:00
$sql = '';
switch($type)
{
default:
case "select":
$sql = 'SELECT * FROM '.$this->from_string;
// Set the select string
if ( ! empty($this->select_string))
{
// Replace the star with the selected fields
$sql = str_replace('*', $this->select_string, $sql);
}
// Set the where string
if ( ! empty($this->where_string))
{
$sql .= $this->where_string;
}
// Set the limit via the class variables
if (is_numeric($this->limit))
{
$sql = $this->sql->limit($sql, $this->limit, $this->offset);
}
break;
case "insert":
$param_count = count($this->set_array);
$params = array_file(0, $param_count, '?');
$sql = 'INSERT (' . implode(', ', $this->set_array_keys) .
') INTO '. $this->quote_ident($table) .
'VALUES ('.implode(', ', $params).')';
2012-03-09 15:55:56 -05:00
break;
case "delete":
// @todo Implement delete statements
break;
}
return $sql;
2012-03-08 13:29:30 -05:00
}
2012-03-09 15:55:56 -05:00
}
// End of query_builder.php