diff --git a/db.class.php b/db.class.php
new file mode 100644
index 0000000..34f175e
--- /dev/null
+++ b/db.class.php
@@ -0,0 +1,1580 @@
+.
+*/
+
+class DB
+{
+ // initial connection
+ public static $dbName = '';
+ public static $user = '';
+ public static $password = '';
+ public static $host = 'localhost';
+ public static $port = 3306; //hhvm complains if this is null
+ public static $socket = null;
+ public static $encoding = 'latin1';
+
+ // configure workings
+ public static $param_char = '%';
+ public static $named_param_seperator = '_';
+ public static $success_handler = false;
+ public static $error_handler = true;
+ public static $throw_exception_on_error = false;
+ public static $nonsql_error_handler = null;
+ public static $pre_sql_handler = false;
+ public static $throw_exception_on_nonsql_error = false;
+ public static $nested_transactions = false;
+ public static $usenull = true;
+ public static $ssl = array(
+ 'key' => '',
+ 'cert' => '',
+ 'ca_cert' => '',
+ 'ca_path' => '',
+ 'cipher' => ''
+ );
+ public static $connect_options = array(MYSQLI_OPT_CONNECT_TIMEOUT => 30);
+
+ // internal
+ protected static $mdb = null;
+ public static $variables_to_sync = array(
+ 'param_char',
+ 'named_param_seperator',
+ 'success_handler',
+ 'error_handler',
+ 'throw_exception_on_error',
+ 'nonsql_error_handler',
+ 'pre_sql_handler',
+ 'throw_exception_on_nonsql_error',
+ 'nested_transactions',
+ 'usenull',
+ 'ssl',
+ 'connect_options'
+ );
+
+ public static function getMDB()
+ {
+ $mdb = DB::$mdb;
+
+ if ($mdb === null) {
+ $mdb = DB::$mdb = new MeekroDB();
+ }
+
+ // Sync everytime because settings might have changed. It's fast.
+ $mdb->sync_config();
+
+ return $mdb;
+ }
+
+ // yes, this is ugly. __callStatic() only works in 5.3+
+ public static function get()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'get'), $args);
+ }
+ public static function disconnect()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'disconnect'), $args);
+ }
+ public static function query()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'query'), $args);
+ }
+ public static function queryFirstRow()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryFirstRow'), $args);
+ }
+ public static function queryOneRow()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryOneRow'), $args);
+ }
+ public static function queryAllLists()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryAllLists'), $args);
+ }
+ public static function queryFullColumns()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryFullColumns'), $args);
+ }
+ public static function queryFirstList()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryFirstList'), $args);
+ }
+ public static function queryOneList()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryOneList'), $args);
+ }
+ public static function queryFirstColumn()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryFirstColumn'), $args);
+ }
+ public static function queryOneColumn()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryOneColumn'), $args);
+ }
+ public static function queryFirstField()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryFirstField'), $args);
+ }
+ public static function queryOneField()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryOneField'), $args);
+ }
+ public static function queryRaw()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryRaw'), $args);
+ }
+ public static function queryRawUnbuf()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'queryRawUnbuf'), $args);
+ }
+
+ public static function insert()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'insert'), $args);
+ }
+ public static function insertIgnore()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'insertIgnore'), $args);
+ }
+ public static function insertUpdate()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'insertUpdate'), $args);
+ }
+ public static function replace()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'replace'), $args);
+ }
+ public static function update()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'update'), $args);
+ }
+ public static function delete()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'delete'), $args);
+ }
+
+ public static function insertId()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'insertId'), $args);
+ }
+ public static function count()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'count'), $args);
+ }
+ public static function affectedRows()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'affectedRows'), $args);
+ }
+
+ public static function useDB()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'useDB'), $args);
+ }
+ public static function startTransaction()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'startTransaction'), $args);
+ }
+ public static function commit()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'commit'), $args);
+ }
+ public static function rollback()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'rollback'), $args);
+ }
+ public static function tableList()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'tableList'), $args);
+ }
+ public static function columnList()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'columnList'), $args);
+ }
+
+ public static function sqlEval()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'sqlEval'), $args);
+ }
+ public static function nonSQLError()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'nonSQLError'), $args);
+ }
+
+ public static function serverVersion()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'serverVersion'), $args);
+ }
+ public static function transactionDepth()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array(DB::getMDB(), 'transactionDepth'), $args);
+ }
+
+ public static function debugMode($handler = true)
+ {
+ DB::$success_handler = $handler;
+ }
+}
+
+class MeekroDB
+{
+ // initial connection
+ public $dbName = '';
+ public $user = '';
+ public $password = '';
+ public $host = 'localhost';
+ public $port = 3306;
+ public $socket = null;
+ public $encoding = 'latin1';
+
+ // configure workings
+ public $param_char = '%';
+ public $named_param_seperator = '_';
+ public $success_handler = false;
+ public $error_handler = true;
+ public $throw_exception_on_error = false;
+ public $nonsql_error_handler = null;
+ public $pre_sql_handler = false;
+ public $throw_exception_on_nonsql_error = false;
+ public $nested_transactions = false;
+ public $usenull = true;
+ public $ssl = array(
+ 'key' => '',
+ 'cert' => '',
+ 'ca_cert' => '',
+ 'ca_path' => '',
+ 'cipher' => ''
+ );
+ public $connect_options = array(MYSQLI_OPT_CONNECT_TIMEOUT => 30);
+
+ // internal
+ public $internal_mysql = null;
+ public $server_info = null;
+ public $insert_id = 0;
+ public $num_rows = 0;
+ public $affected_rows = 0;
+ public $current_db = null;
+ public $nested_transactions_count = 0;
+
+ public function __construct(
+ $host = null,
+ $user = null,
+ $password = null,
+ $dbName = null,
+ $port = null,
+ $encoding = null,
+ $socket = null
+ ) {
+ if ($host === null) {
+ $host = DB::$host;
+ }
+ if ($user === null) {
+ $user = DB::$user;
+ }
+ if ($password === null) {
+ $password = DB::$password;
+ }
+ if ($dbName === null) {
+ $dbName = DB::$dbName;
+ }
+ if ($port === null) {
+ $port = DB::$port;
+ }
+ if ($socket === null) {
+ $socket = DB::$socket;
+ }
+ if ($encoding === null) {
+ $encoding = DB::$encoding;
+ }
+
+ $this->host = $host;
+ $this->user = $user;
+ $this->password = $password;
+ $this->dbName = $dbName;
+ $this->port = $port;
+ $this->socket = $socket;
+ $this->encoding = $encoding;
+
+ $this->sync_config();
+ }
+
+ // suck in config settings from static class
+ public function sync_config()
+ {
+ foreach (DB::$variables_to_sync as $variable) {
+ if ($this->$variable !== DB::$$variable) {
+ $this->$variable = DB::$$variable;
+ }
+ }
+ }
+
+ public function get()
+ {
+ $mysql = $this->internal_mysql;
+
+ if (!($mysql instanceof MySQLi)) {
+ if (!$this->port) {
+ $this->port = ini_get('mysqli.default_port');
+ }
+ $this->current_db = $this->dbName;
+ $mysql = new mysqli();
+
+ $connect_flags = 0;
+ if ($this->ssl['key']) {
+ $mysql->ssl_set(
+ $this->ssl['key'],
+ $this->ssl['cert'],
+ $this->ssl['ca_cert'],
+ $this->ssl['ca_path'],
+ $this->ssl['cipher']
+ );
+ $connect_flags |= MYSQLI_CLIENT_SSL;
+ }
+ foreach ($this->connect_options as $key => $value) {
+ $mysql->options($key, $value);
+ }
+
+ // suppress warnings, since we will check connect_error anyway
+ @$mysql->real_connect(
+ $this->host,
+ $this->user,
+ $this->password,
+ $this->dbName,
+ $this->port,
+ $this->socket,
+ $connect_flags
+ );
+
+ if ($mysql->connect_error) {
+ return $this->nonSQLError(
+ 'Unable to connect to MySQL server! Error: ' . $mysql->connect_error
+ );
+ }
+
+ $mysql->set_charset($this->encoding);
+ $this->internal_mysql = $mysql;
+ $this->server_info = $mysql->server_info;
+ }
+
+ return $mysql;
+ }
+
+ public function disconnect()
+ {
+ $mysqli = $this->internal_mysql;
+ if ($mysqli instanceof MySQLi) {
+ if ($thread_id = $mysqli->thread_id) {
+ $mysqli->kill($thread_id);
+ }
+ $mysqli->close();
+ }
+ $this->internal_mysql = null;
+ }
+
+ public function nonSQLError($message)
+ {
+ if ($this->throw_exception_on_nonsql_error) {
+ $e = new MeekroDBException($message);
+ throw $e;
+ }
+
+ $error_handler = is_callable($this->nonsql_error_handler)
+ ? $this->nonsql_error_handler
+ : 'meekrodb_error_handler';
+
+ call_user_func($error_handler, array(
+ 'type' => 'nonsql',
+ 'error' => $message
+ ));
+ }
+
+ public function debugMode($handler = true)
+ {
+ $this->success_handler = $handler;
+ }
+
+ public function serverVersion()
+ {
+ $this->get();
+ return $this->server_info;
+ }
+ public function transactionDepth()
+ {
+ return $this->nested_transactions_count;
+ }
+ public function insertId()
+ {
+ return $this->insert_id;
+ }
+ public function affectedRows()
+ {
+ return $this->affected_rows;
+ }
+ public function count()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array($this, 'numRows'), $args);
+ }
+ public function numRows()
+ {
+ return $this->num_rows;
+ }
+
+ public function useDB()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array($this, 'setDB'), $args);
+ }
+ public function setDB($dbName)
+ {
+ $db = $this->get();
+ if (!$db->select_db($dbName)) {
+ return $this->nonSQLError("Unable to set database to $dbName");
+ }
+ $this->current_db = $dbName;
+ }
+
+ public function startTransaction()
+ {
+ if ($this->nested_transactions && $this->serverVersion() < '5.5') {
+ return $this->nonSQLError(
+ "Nested transactions are only available on MySQL 5.5 and greater. You are using MySQL " .
+ $this->serverVersion()
+ );
+ }
+
+ if (!$this->nested_transactions || $this->nested_transactions_count == 0) {
+ $this->query('START TRANSACTION');
+ $this->nested_transactions_count = 1;
+ } else {
+ $this->query("SAVEPOINT LEVEL{$this->nested_transactions_count}");
+ $this->nested_transactions_count++;
+ }
+
+ return $this->nested_transactions_count;
+ }
+
+ public function commit($all = false)
+ {
+ if ($this->nested_transactions && $this->serverVersion() < '5.5') {
+ return $this->nonSQLError(
+ "Nested transactions are only available on MySQL 5.5 and greater. You are using MySQL " .
+ $this->serverVersion()
+ );
+ }
+
+ if ($this->nested_transactions && $this->nested_transactions_count > 0) {
+ $this->nested_transactions_count--;
+ }
+
+ if (
+ !$this->nested_transactions ||
+ $all ||
+ $this->nested_transactions_count == 0
+ ) {
+ $this->nested_transactions_count = 0;
+ $this->query('COMMIT');
+ } else {
+ $this->query("RELEASE SAVEPOINT LEVEL{$this->nested_transactions_count}");
+ }
+
+ return $this->nested_transactions_count;
+ }
+
+ public function rollback($all = false)
+ {
+ if ($this->nested_transactions && $this->serverVersion() < '5.5') {
+ return $this->nonSQLError(
+ "Nested transactions are only available on MySQL 5.5 and greater. You are using MySQL " .
+ $this->serverVersion()
+ );
+ }
+
+ if ($this->nested_transactions && $this->nested_transactions_count > 0) {
+ $this->nested_transactions_count--;
+ }
+
+ if (
+ !$this->nested_transactions ||
+ $all ||
+ $this->nested_transactions_count == 0
+ ) {
+ $this->nested_transactions_count = 0;
+ $this->query('ROLLBACK');
+ } else {
+ $this->query(
+ "ROLLBACK TO SAVEPOINT LEVEL{$this->nested_transactions_count}"
+ );
+ }
+
+ return $this->nested_transactions_count;
+ }
+
+ protected function formatTableName($table)
+ {
+ $table = trim($table, '`');
+
+ if (strpos($table, '.')) {
+ return implode(
+ '.',
+ array_map(array($this, 'formatTableName'), explode('.', $table))
+ );
+ } else {
+ return '`' . str_replace('`', '``', $table) . '`';
+ }
+ }
+
+ public function update()
+ {
+ $args = func_get_args();
+ $table = array_shift($args);
+ $params = array_shift($args);
+
+ $update_part = $this->parseQueryParams(
+ str_replace('%', $this->param_char, "UPDATE %b SET %hc"),
+ $table,
+ $params
+ );
+
+ $where_part = call_user_func_array(array($this, 'parseQueryParams'), $args);
+ $query = $update_part . ' WHERE ' . $where_part;
+ return $this->query($query);
+ }
+
+ public function insertOrReplace($which, $table, $datas, $options = array())
+ {
+ $datas = unserialize(serialize($datas)); // break references within array
+ $keys = $values = array();
+
+ if (isset($datas[0]) && is_array($datas[0])) {
+ $var = '%ll?';
+ foreach ($datas as $datum) {
+ ksort($datum);
+ if (!$keys) {
+ $keys = array_keys($datum);
+ }
+ $values[] = array_values($datum);
+ }
+ } else {
+ $var = '%l?';
+ $keys = array_keys($datas);
+ $values = array_values($datas);
+ }
+
+ if (
+ $which != 'INSERT' &&
+ $which != 'INSERT IGNORE' &&
+ $which != 'REPLACE'
+ ) {
+ return $this->nonSQLError(
+ 'insertOrReplace() must be called with one of: INSERT, INSERT IGNORE, REPLACE'
+ );
+ }
+
+ if (
+ isset($options['update']) &&
+ is_array($options['update']) &&
+ $options['update'] &&
+ $which == 'INSERT'
+ ) {
+ if (array_values($options['update']) !== $options['update']) {
+ return $this->query(
+ str_replace(
+ '%',
+ $this->param_char,
+ "INSERT INTO %b %lb VALUES $var ON DUPLICATE KEY UPDATE %hc"
+ ),
+ $table,
+ $keys,
+ $values,
+ $options['update']
+ );
+ } else {
+ $update_str = array_shift($options['update']);
+ $query_param = array(
+ str_replace(
+ '%',
+ $this->param_char,
+ "INSERT INTO %b %lb VALUES $var ON DUPLICATE KEY UPDATE "
+ ) . $update_str,
+ $table,
+ $keys,
+ $values
+ );
+ $query_param = array_merge($query_param, $options['update']);
+ return call_user_func_array(array($this, 'query'), $query_param);
+ }
+ }
+
+ return $this->query(
+ str_replace('%', $this->param_char, "%l INTO %b %lb VALUES $var"),
+ $which,
+ $table,
+ $keys,
+ $values
+ );
+ }
+
+ public function insert($table, $data)
+ {
+ return $this->insertOrReplace('INSERT', $table, $data);
+ }
+ public function insertIgnore($table, $data)
+ {
+ return $this->insertOrReplace('INSERT IGNORE', $table, $data);
+ }
+ public function replace($table, $data)
+ {
+ return $this->insertOrReplace('REPLACE', $table, $data);
+ }
+
+ public function insertUpdate()
+ {
+ $args = func_get_args();
+ $table = array_shift($args);
+ $data = array_shift($args);
+
+ if (!isset($args[0])) {
+ // update will have all the data of the insert
+ if (isset($data[0]) && is_array($data[0])) {
+ //multiple insert rows specified -- failing!
+ return $this->nonSQLError(
+ "Badly formatted insertUpdate() query -- you didn't specify the update component!"
+ );
+ }
+
+ $args[0] = $data;
+ }
+
+ if (is_array($args[0])) {
+ $update = $args[0];
+ } else {
+ $update = $args;
+ }
+
+ return $this->insertOrReplace('INSERT', $table, $data, array(
+ 'update' => $update
+ ));
+ }
+
+ public function delete()
+ {
+ $args = func_get_args();
+ $table = $this->formatTableName(array_shift($args));
+
+ $where = call_user_func_array(array($this, 'parseQueryParams'), $args);
+ $query = "DELETE FROM {$table} WHERE {$where}";
+ return $this->query($query);
+ }
+
+ public function sqleval()
+ {
+ $args = func_get_args();
+ $text = call_user_func_array(array($this, 'parseQueryParams'), $args);
+ return new MeekroDBEval($text);
+ }
+
+ public function columnList($table)
+ {
+ return $this->queryOneColumn('Field', "SHOW COLUMNS FROM %b", $table);
+ }
+
+ public function tableList($db = null)
+ {
+ if ($db) {
+ $olddb = $this->current_db;
+ $this->useDB($db);
+ }
+
+ $result = $this->queryFirstColumn('SHOW TABLES');
+ if (isset($olddb)) {
+ $this->useDB($olddb);
+ }
+ return $result;
+ }
+
+ protected function preparseQueryParams()
+ {
+ $args = func_get_args();
+ $sql = trim(strval(array_shift($args)));
+ $args_all = $args;
+
+ if (count($args_all) == 0) {
+ return array($sql);
+ }
+
+ $param_char_length = strlen($this->param_char);
+ $named_seperator_length = strlen($this->named_param_seperator);
+
+ $types = array(
+ $this->param_char . 'll', // list of literals
+ $this->param_char . 'ls', // list of strings
+ $this->param_char . 'l', // literal
+ $this->param_char . 'li', // list of integers
+ $this->param_char . 'ld', // list of decimals
+ $this->param_char . 'lb', // list of backticks
+ $this->param_char . 'lt', // list of timestamps
+ $this->param_char . 's', // string
+ $this->param_char . 'i', // integer
+ $this->param_char . 'd', // double / decimal
+ $this->param_char . 'b', // backtick
+ $this->param_char . 't', // timestamp
+ $this->param_char . '?', // infer type
+ $this->param_char . 'l?', // list of inferred types
+ $this->param_char . 'll?', // list of lists of inferred types
+ $this->param_char . 'hc', // hash `key`='value' pairs separated by commas
+ $this->param_char . 'ha', // hash `key`='value' pairs separated by and
+ $this->param_char . 'ho', // hash `key`='value' pairs separated by or
+ $this->param_char . 'ss', // search string (like string, surrounded with %'s)
+ $this->param_char . 'ssb', // search string (like, begins with)
+ $this->param_char . 'sse' // search string (like, ends with)
+ );
+
+ // generate list of all MeekroDB variables in our query, and their position
+ // in the form "offset => variable", sorted by offsets
+ $posList = array();
+ foreach ($types as $type) {
+ $lastPos = 0;
+ while (($pos = strpos($sql, $type, $lastPos)) !== false) {
+ $lastPos = $pos + 1;
+ if (isset($posList[$pos]) && strlen($posList[$pos]) > strlen($type)) {
+ continue;
+ }
+ $posList[$pos] = $type;
+ }
+ }
+
+ ksort($posList);
+
+ // for each MeekroDB variable, substitute it with array(type: i, value: 53) or whatever
+ $chunkyQuery = array(); // preparsed query
+ $pos_adj = 0; // how much we've added or removed from the original sql string
+ foreach ($posList as $pos => $type) {
+ $type = substr($type, $param_char_length); // variable, without % in front of it
+ $length_type = strlen($type) + $param_char_length; // length of variable w/o %
+
+ $new_pos = $pos + $pos_adj; // position of start of variable
+ $new_pos_back = $new_pos + $length_type; // position of end of variable
+ $arg_number_length = 0; // length of any named or numbered parameter addition
+
+ // handle numbered parameters
+ if ($arg_number_length = strspn($sql, '0123456789', $new_pos_back)) {
+ $arg_number = substr($sql, $new_pos_back, $arg_number_length);
+ if (!array_key_exists($arg_number, $args_all)) {
+ return $this->nonSQLError(
+ "Non existent argument reference (arg $arg_number): $sql"
+ );
+ }
+
+ $arg = $args_all[$arg_number];
+
+ // handle named parameters
+ } elseif (
+ substr($sql, $new_pos_back, $named_seperator_length) ==
+ $this->named_param_seperator
+ ) {
+ $arg_number_length =
+ strspn(
+ $sql,
+ 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_',
+ $new_pos_back + $named_seperator_length
+ ) + $named_seperator_length;
+
+ $arg_number = substr(
+ $sql,
+ $new_pos_back + $named_seperator_length,
+ $arg_number_length - $named_seperator_length
+ );
+ if (count($args_all) != 1 || !is_array($args_all[0])) {
+ return $this->nonSQLError(
+ "If you use named parameters, the second argument must be an array of parameters"
+ );
+ }
+ if (!array_key_exists($arg_number, $args_all[0])) {
+ return $this->nonSQLError(
+ "Non existent argument reference (arg $arg_number): $sql"
+ );
+ }
+
+ $arg = $args_all[0][$arg_number];
+ } else {
+ $arg_number = 0;
+ $arg = array_shift($args);
+ }
+
+ if ($new_pos > 0) {
+ $chunkyQuery[] = substr($sql, 0, $new_pos);
+ }
+
+ if (is_object($arg) && $arg instanceof WhereClause) {
+ list($clause_sql, $clause_args) = $arg->textAndArgs();
+ array_unshift($clause_args, $clause_sql);
+ $preparsed_sql = call_user_func_array(
+ array($this, 'preparseQueryParams'),
+ $clause_args
+ );
+ $chunkyQuery = array_merge($chunkyQuery, $preparsed_sql);
+ } else {
+ $chunkyQuery[] = array('type' => $type, 'value' => $arg);
+ }
+
+ $sql = substr($sql, $new_pos_back + $arg_number_length);
+ $pos_adj -= $new_pos_back + $arg_number_length;
+ }
+
+ if (strlen($sql) > 0) {
+ $chunkyQuery[] = $sql;
+ }
+
+ return $chunkyQuery;
+ }
+
+ public function escape($str)
+ {
+ return "'" . $this->get()->real_escape_string(strval($str)) . "'";
+ }
+
+ public function sanitize($value, $type = 'basic', $hashjoin = ', ')
+ {
+ if ($type == 'basic') {
+ if (is_object($value)) {
+ if ($value instanceof MeekroDBEval) {
+ return $value->text;
+ } elseif ($value instanceof DateTime) {
+ return $this->escape($value->format('Y-m-d H:i:s'));
+ } else {
+ return $this->escape($value);
+ } // use __toString() value for objects, when possible
+ }
+
+ if (is_null($value)) {
+ return $this->usenull ? 'NULL' : "''";
+ } elseif (is_bool($value)) {
+ return $value ? 1 : 0;
+ } elseif (is_int($value)) {
+ return $value;
+ } elseif (is_float($value)) {
+ return $value;
+ } elseif (is_array($value)) {
+ return "''";
+ } else {
+ return $this->escape($value);
+ }
+ } elseif ($type == 'list') {
+ if (is_array($value)) {
+ $value = array_values($value);
+ return '(' .
+ implode(', ', array_map(array($this, 'sanitize'), $value)) .
+ ')';
+ } else {
+ return $this->nonSQLError(
+ "Expected array parameter, got something different!"
+ );
+ }
+ } elseif ($type == 'doublelist') {
+ if (
+ is_array($value) &&
+ array_values($value) === $value &&
+ is_array($value[0])
+ ) {
+ $cleanvalues = array();
+ foreach ($value as $subvalue) {
+ $cleanvalues[] = $this->sanitize($subvalue, 'list');
+ }
+ return implode(', ', $cleanvalues);
+ } else {
+ return $this->nonSQLError(
+ "Expected double array parameter, got something different!"
+ );
+ }
+ } elseif ($type == 'hash') {
+ if (is_array($value)) {
+ $pairs = array();
+ foreach ($value as $k => $v) {
+ $pairs[] = $this->formatTableName($k) . '=' . $this->sanitize($v);
+ }
+
+ return implode($hashjoin, $pairs);
+ } else {
+ return $this->nonSQLError(
+ "Expected hash (associative array) parameter, got something different!"
+ );
+ }
+ } else {
+ return $this->nonSQLError("Invalid type passed to sanitize()!");
+ }
+ }
+
+ protected function parseTS($ts)
+ {
+ if (is_string($ts)) {
+ return date('Y-m-d H:i:s', strtotime($ts));
+ } elseif (is_object($ts) && $ts instanceof DateTime) {
+ return $ts->format('Y-m-d H:i:s');
+ }
+ }
+
+ protected function intval($var)
+ {
+ if (PHP_INT_SIZE == 8) {
+ return intval($var);
+ }
+ return floor(doubleval($var));
+ }
+
+ public function parseQueryParams()
+ {
+ $args = func_get_args();
+ $chunkyQuery = call_user_func_array(
+ array($this, 'preparseQueryParams'),
+ $args
+ );
+
+ $query = '';
+ $array_types = array(
+ 'ls',
+ 'li',
+ 'ld',
+ 'lb',
+ 'll',
+ 'lt',
+ 'l?',
+ 'll?',
+ 'hc',
+ 'ha',
+ 'ho'
+ );
+
+ foreach ($chunkyQuery as $chunk) {
+ if (is_string($chunk)) {
+ $query .= $chunk;
+ continue;
+ }
+
+ $type = $chunk['type'];
+ $arg = $chunk['value'];
+ $result = '';
+
+ $is_array_type = in_array($type, $array_types, true);
+ if ($is_array_type && !is_array($arg)) {
+ return $this->nonSQLError(
+ "Badly formatted SQL query: Expected array, got scalar instead!"
+ );
+ } elseif (!$is_array_type && is_array($arg)) {
+ $arg = '';
+ }
+
+ if ($type == 's') {
+ $result = $this->escape($arg);
+ } elseif ($type == 'i') {
+ $result = $this->intval($arg);
+ } elseif ($type == 'd') {
+ $result = doubleval($arg);
+ } elseif ($type == 'b') {
+ $result = $this->formatTableName($arg);
+ } elseif ($type == 'l') {
+ $result = $arg;
+ } elseif ($type == 'ss') {
+ $result = $this->escape(
+ "%" . str_replace(array('%', '_'), array('\%', '\_'), $arg) . "%"
+ );
+ } elseif ($type == 'ssb') {
+ $result = $this->escape(
+ str_replace(array('%', '_'), array('\%', '\_'), $arg) . "%"
+ );
+ } elseif ($type == 'sse') {
+ $result = $this->escape(
+ "%" . str_replace(array('%', '_'), array('\%', '\_'), $arg)
+ );
+ } elseif ($type == 't') {
+ $result = $this->escape($this->parseTS($arg));
+ } elseif ($type == 'ls') {
+ $result = array_map(array($this, 'escape'), $arg);
+ } elseif ($type == 'li') {
+ $result = array_map(array($this, 'intval'), $arg);
+ } elseif ($type == 'ld') {
+ $result = array_map('doubleval', $arg);
+ } elseif ($type == 'lb') {
+ $result = array_map(array($this, 'formatTableName'), $arg);
+ } elseif ($type == 'll') {
+ $result = $arg;
+ } elseif ($type == 'lt') {
+ $result = array_map(
+ array($this, 'escape'),
+ array_map(array($this, 'parseTS'), $arg)
+ );
+ } elseif ($type == '?') {
+ $result = $this->sanitize($arg);
+ } elseif ($type == 'l?') {
+ $result = $this->sanitize($arg, 'list');
+ } elseif ($type == 'll?') {
+ $result = $this->sanitize($arg, 'doublelist');
+ } elseif ($type == 'hc') {
+ $result = $this->sanitize($arg, 'hash');
+ } elseif ($type == 'ha') {
+ $result = $this->sanitize($arg, 'hash', ' AND ');
+ } elseif ($type == 'ho') {
+ $result = $this->sanitize($arg, 'hash', ' OR ');
+ } else {
+ return $this->nonSQLError(
+ "Badly formatted SQL query: Invalid MeekroDB param $type"
+ );
+ }
+
+ if (is_array($result)) {
+ $result = '(' . implode(',', $result) . ')';
+ }
+
+ $query .= $result;
+ }
+
+ return $query;
+ }
+
+ protected function prependCall($function, $args, $prepend)
+ {
+ array_unshift($args, $prepend);
+ return call_user_func_array($function, $args);
+ }
+ public function query()
+ {
+ $args = func_get_args();
+ return $this->prependCall(array($this, 'queryHelper'), $args, 'assoc');
+ }
+ public function queryAllLists()
+ {
+ $args = func_get_args();
+ return $this->prependCall(array($this, 'queryHelper'), $args, 'list');
+ }
+ public function queryFullColumns()
+ {
+ $args = func_get_args();
+ return $this->prependCall(array($this, 'queryHelper'), $args, 'full');
+ }
+
+ public function queryRaw()
+ {
+ $args = func_get_args();
+ return $this->prependCall(array($this, 'queryHelper'), $args, 'raw_buf');
+ }
+ public function queryRawUnbuf()
+ {
+ $args = func_get_args();
+ return $this->prependCall(array($this, 'queryHelper'), $args, 'raw_unbuf');
+ }
+
+ protected function queryHelper()
+ {
+ $args = func_get_args();
+ $type = array_shift($args);
+ $db = $this->get();
+
+ $is_buffered = true;
+ $row_type = 'assoc'; // assoc, list, raw
+ $full_names = false;
+
+ switch ($type) {
+ case 'assoc':
+ break;
+ case 'list':
+ $row_type = 'list';
+ break;
+ case 'full':
+ $row_type = 'list';
+ $full_names = true;
+ break;
+ case 'raw_buf':
+ $row_type = 'raw';
+ break;
+ case 'raw_unbuf':
+ $is_buffered = false;
+ $row_type = 'raw';
+ break;
+ default:
+ return $this->nonSQLError('Error -- invalid argument to queryHelper!');
+ }
+
+ $sql = call_user_func_array(array($this, 'parseQueryParams'), $args);
+
+ if (
+ $this->pre_sql_handler !== false &&
+ is_callable($this->pre_sql_handler)
+ ) {
+ $sql = call_user_func($this->pre_sql_handler, $sql);
+ }
+
+ if ($this->success_handler) {
+ $starttime = microtime(true);
+ }
+ $result = $db->query(
+ $sql,
+ $is_buffered ? MYSQLI_STORE_RESULT : MYSQLI_USE_RESULT
+ );
+ if ($this->success_handler) {
+ $runtime = microtime(true) - $starttime;
+ } else {
+ $runtime = 0;
+ }
+
+ // ----- BEGIN ERROR HANDLING
+ if (!$sql || $db->error) {
+ if ($this->error_handler) {
+ $error_handler = is_callable($this->error_handler)
+ ? $this->error_handler
+ : 'meekrodb_error_handler';
+
+ call_user_func($error_handler, array(
+ 'type' => 'sql',
+ 'query' => $sql,
+ 'error' => $db->error,
+ 'code' => $db->errno
+ ));
+ }
+
+ if ($this->throw_exception_on_error) {
+ $e = new MeekroDBException($db->error, $sql, $db->errno);
+ throw $e;
+ }
+ } elseif ($this->success_handler) {
+ $runtime = sprintf('%f', $runtime * 1000);
+ $success_handler = is_callable($this->success_handler)
+ ? $this->success_handler
+ : 'meekrodb_debugmode_handler';
+
+ call_user_func($success_handler, array(
+ 'query' => $sql,
+ 'runtime' => $runtime,
+ 'affected' => $db->affected_rows
+ ));
+ }
+
+ // ----- END ERROR HANDLING
+
+ $this->insert_id = $db->insert_id;
+ $this->affected_rows = $db->affected_rows;
+
+ // mysqli_result->num_rows won't initially show correct results for unbuffered data
+ if ($is_buffered && $result instanceof MySQLi_Result) {
+ $this->num_rows = $result->num_rows;
+ } else {
+ $this->num_rows = null;
+ }
+
+ if ($row_type == 'raw' || !($result instanceof MySQLi_Result)) {
+ return $result;
+ }
+
+ $return = array();
+
+ if ($full_names) {
+ $infos = array();
+ foreach ($result->fetch_fields() as $info) {
+ if (strlen($info->table)) {
+ $infos[] = $info->table . '.' . $info->name;
+ } else {
+ $infos[] = $info->name;
+ }
+ }
+ }
+
+ while (
+ $row =
+ $row_type == 'assoc' ? $result->fetch_assoc() : $result->fetch_row()
+ ) {
+ if ($full_names) {
+ $row = array_combine($infos, $row);
+ }
+ $return[] = $row;
+ }
+
+ // free results
+ $result->free();
+ while ($db->more_results()) {
+ $db->next_result();
+ if ($result = $db->use_result()) {
+ $result->free();
+ }
+ }
+
+ return $return;
+ }
+
+ public function queryOneRow()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array($this, 'queryFirstRow'), $args);
+ }
+ public function queryFirstRow()
+ {
+ $args = func_get_args();
+ $result = call_user_func_array(array($this, 'query'), $args);
+ if (!$result || !is_array($result)) {
+ return null;
+ }
+ return reset($result);
+ }
+
+ public function queryOneList()
+ {
+ $args = func_get_args();
+ return call_user_func_array(array($this, 'queryFirstList'), $args);
+ }
+ public function queryFirstList()
+ {
+ $args = func_get_args();
+ $result = call_user_func_array(array($this, 'queryAllLists'), $args);
+ if (!$result || !is_array($result)) {
+ return null;
+ }
+ return reset($result);
+ }
+
+ public function queryFirstColumn()
+ {
+ $args = func_get_args();
+ $results = call_user_func_array(array($this, 'queryAllLists'), $args);
+ $ret = array();
+
+ if (!count($results) || !count($results[0])) {
+ return $ret;
+ }
+
+ foreach ($results as $row) {
+ $ret[] = $row[0];
+ }
+
+ return $ret;
+ }
+
+ public function queryOneColumn()
+ {
+ $args = func_get_args();
+ $column = array_shift($args);
+ $results = call_user_func_array(array($this, 'query'), $args);
+ $ret = array();
+
+ if (!count($results) || !count($results[0])) {
+ return $ret;
+ }
+ if ($column === null) {
+ $keys = array_keys($results[0]);
+ $column = $keys[0];
+ }
+
+ foreach ($results as $row) {
+ $ret[] = $row[$column];
+ }
+
+ return $ret;
+ }
+
+ public function queryFirstField()
+ {
+ $args = func_get_args();
+ $row = call_user_func_array(array($this, 'queryFirstList'), $args);
+ if ($row == null) {
+ return null;
+ }
+ return $row[0];
+ }
+
+ public function queryOneField()
+ {
+ $args = func_get_args();
+ $column = array_shift($args);
+
+ $row = call_user_func_array(array($this, 'queryOneRow'), $args);
+ if ($row == null) {
+ return null;
+ } elseif ($column === null) {
+ $keys = array_keys($row);
+ $column = $keys[0];
+ }
+
+ return $row[$column];
+ }
+}
+
+class WhereClause
+{
+ public $type = 'and'; //AND or OR
+ public $negate = false;
+ public $clauses = array();
+
+ function __construct($type)
+ {
+ $type = strtolower($type);
+ if ($type !== 'or' && $type !== 'and') {
+ return DB::nonSQLError(
+ 'you must use either WhereClause(and) or WhereClause(or)'
+ );
+ }
+ $this->type = $type;
+ }
+
+ function add()
+ {
+ $args = func_get_args();
+ $sql = array_shift($args);
+
+ if ($sql instanceof WhereClause) {
+ $this->clauses[] = $sql;
+ } else {
+ $this->clauses[] = array('sql' => $sql, 'args' => $args);
+ }
+ }
+
+ function negateLast()
+ {
+ $i = count($this->clauses) - 1;
+ if (!isset($this->clauses[$i])) {
+ return;
+ }
+
+ if ($this->clauses[$i] instanceof WhereClause) {
+ $this->clauses[$i]->negate();
+ } else {
+ $this->clauses[$i]['sql'] = 'NOT (' . $this->clauses[$i]['sql'] . ')';
+ }
+ }
+
+ function negate()
+ {
+ $this->negate = !$this->negate;
+ }
+
+ function addClause($type)
+ {
+ $r = new WhereClause($type);
+ $this->add($r);
+ return $r;
+ }
+
+ function count()
+ {
+ return count($this->clauses);
+ }
+
+ function textAndArgs()
+ {
+ $sql = array();
+ $args = array();
+
+ if (count($this->clauses) == 0) {
+ return array('(1)', $args);
+ }
+
+ foreach ($this->clauses as $clause) {
+ if ($clause instanceof WhereClause) {
+ list($clause_sql, $clause_args) = $clause->textAndArgs();
+ } else {
+ $clause_sql = $clause['sql'];
+ $clause_args = $clause['args'];
+ }
+
+ $sql[] = "($clause_sql)";
+ $args = array_merge($args, $clause_args);
+ }
+
+ if ($this->type == 'and') {
+ $sql = sprintf('(%s)', implode(' AND ', $sql));
+ } else {
+ $sql = sprintf('(%s)', implode(' OR ', $sql));
+ }
+
+ if ($this->negate) {
+ $sql = '(NOT ' . $sql . ')';
+ }
+ return array($sql, $args);
+ }
+
+ // backwards compatability
+ // we now return full WhereClause object here and evaluate it in preparseQueryParams
+ function text()
+ {
+ return $this;
+ }
+}
+
+class DBTransaction
+{
+ private $committed = false;
+
+ function __construct()
+ {
+ DB::startTransaction();
+ }
+ function __destruct()
+ {
+ if (!$this->committed) {
+ DB::rollback();
+ }
+ }
+ function commit()
+ {
+ DB::commit();
+ $this->committed = true;
+ }
+}
+
+class MeekroDBException extends Exception
+{
+ protected $query = '';
+
+ function __construct($message = '', $query = '', $code = 0)
+ {
+ parent::__construct($message);
+ $this->query = $query;
+ $this->code = $code;
+ }
+
+ public function getQuery()
+ {
+ return $this->query;
+ }
+}
+
+class DBHelper
+{
+ /*
+ verticalSlice
+ 1. For an array of assoc rays, return an array of values for a particular key
+ 2. if $keyfield is given, same as above but use that hash key as the key in new array
+ */
+
+ public static function verticalSlice($array, $field, $keyfield = null)
+ {
+ $array = (array) $array;
+
+ $R = array();
+ foreach ($array as $obj) {
+ if (!array_key_exists($field, $obj)) {
+ die("verticalSlice: array doesn't have requested field\n");
+ }
+
+ if ($keyfield) {
+ if (!array_key_exists($keyfield, $obj)) {
+ die("verticalSlice: array doesn't have requested field\n");
+ }
+ $R[$obj[$keyfield]] = $obj[$field];
+ } else {
+ $R[] = $obj[$field];
+ }
+ }
+ return $R;
+ }
+
+ /*
+ reIndex
+ For an array of assoc rays, return a new array of assoc rays using a certain field for keys
+ */
+
+ public static function reIndex()
+ {
+ $fields = func_get_args();
+ $array = array_shift($fields);
+ $array = (array) $array;
+
+ $R = array();
+ foreach ($array as $obj) {
+ $target = &$R;
+
+ foreach ($fields as $field) {
+ if (!array_key_exists($field, $obj)) {
+ die("reIndex: array doesn't have requested field\n");
+ }
+
+ $nextkey = $obj[$field];
+ $target = &$target[$nextkey];
+ }
+ $target = $obj;
+ }
+ return $R;
+ }
+}
+
+function meekrodb_error_handler($params)
+{
+ if (isset($params['query'])) {
+ $out[] = "QUERY: " . $params['query'];
+ }
+ if (isset($params['error'])) {
+ $out[] = "ERROR: " . $params['error'];
+ }
+ $out[] = "";
+
+ if (php_sapi_name() == 'cli' && empty($_SERVER['REMOTE_ADDR'])) {
+ echo implode("\n", $out);
+ } else {
+ echo implode("
\n", $out);
+ }
+
+ die();
+}
+
+function meekrodb_debugmode_handler($params)
+{
+ echo "QUERY: " . $params['query'] . " [" . $params['runtime'] . " ms]";
+ if (php_sapi_name() == 'cli' && empty($_SERVER['REMOTE_ADDR'])) {
+ echo "\n";
+ } else {
+ echo "
\n";
+ }
+}
+
+class MeekroDBEval
+{
+ public $text = '';
+
+ function __construct($text)
+ {
+ $this->text = $text;
+ }
+}
+
+?>