ImgURL是一个简单、纯粹的图床程序,让个人图床多一个选择。
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1123 lines
23 KiB

<?php
/*!
* Medoo database framework
* http://medoo.in
* Version 1.1.3
*
* Copyright 2016, Angel Lai
* Released under the MIT license
*/
class medoo
{
// General
protected $database_type;
protected $charset;
protected $database_name;
// For MySQL, MariaDB, MSSQL, Sybase, PostgreSQL, Oracle
protected $server;
protected $username;
protected $password;
// For SQLite
protected $database_file;
// For MySQL or MariaDB with unix_socket
protected $socket;
// Optional
protected $port;
protected $prefix;
protected $option = array();
// Variable
protected $logs = array();
protected $debug_mode = false;
public function __construct($options = null)
{
try {
$commands = array();
$dsn = '';
if (is_array($options))
{
foreach ($options as $option => $value)
{
$this->$option = $value;
}
}
else
{
return false;
}
if (
isset($this->port) &&
is_int($this->port * 1)
)
{
$port = $this->port;
}
$type = strtolower($this->database_type);
$is_port = isset($port);
if (isset($options[ 'prefix' ]))
{
$this->prefix = $options[ 'prefix' ];
}
switch ($type)
{
case 'mariadb':
$type = 'mysql';
case 'mysql':
if ($this->socket)
{
$dsn = $type . ':unix_socket=' . $this->socket . ';dbname=' . $this->database_name;
}
else
{
$dsn = $type . ':host=' . $this->server . ($is_port ? ';port=' . $port : '') . ';dbname=' . $this->database_name;
}
// Make MySQL using standard quoted identifier
$commands[] = 'SET SQL_MODE=ANSI_QUOTES';
break;
case 'pgsql':
$dsn = $type . ':host=' . $this->server . ($is_port ? ';port=' . $port : '') . ';dbname=' . $this->database_name;
break;
case 'sybase':
$dsn = 'dblib:host=' . $this->server . ($is_port ? ':' . $port : '') . ';dbname=' . $this->database_name;
break;
case 'oracle':
$dbname = $this->server ?
'//' . $this->server . ($is_port ? ':' . $port : ':1521') . '/' . $this->database_name :
$this->database_name;
$dsn = 'oci:dbname=' . $dbname . ($this->charset ? ';charset=' . $this->charset : '');
break;
case 'mssql':
$dsn = strstr(PHP_OS, 'WIN') ?
'sqlsrv:server=' . $this->server . ($is_port ? ',' . $port : '') . ';database=' . $this->database_name :
'dblib:host=' . $this->server . ($is_port ? ':' . $port : '') . ';dbname=' . $this->database_name;
// Keep MSSQL QUOTED_IDENTIFIER is ON for standard quoting
$commands[] = 'SET QUOTED_IDENTIFIER ON';
break;
case 'sqlite':
$dsn = $type . ':' . $this->database_file;
$this->username = null;
$this->password = null;
break;
}
if (
in_array($type, array('mariadb', 'mysql', 'pgsql', 'sybase', 'mssql')) &&
$this->charset
)
{
$commands[] = "SET NAMES '" . $this->charset . "'";
}
$this->pdo = new PDO(
$dsn,
$this->username,
$this->password,
$this->option
);
foreach ($commands as $value)
{
$this->pdo->exec($value);
}
}
catch (PDOException $e) {
throw new Exception($e->getMessage());
}
}
public function query($query)
{
if ($this->debug_mode)
{
echo $query;
$this->debug_mode = false;
return false;
}
$this->logs[] = $query;
return $this->pdo->query($query);
}
public function exec($query)
{
if ($this->debug_mode)
{
echo $query;
$this->debug_mode = false;
return false;
}
$this->logs[] = $query;
return $this->pdo->exec($query);
}
public function quote($string)
{
return $this->pdo->quote($string);
}
protected function table_quote($table)
{
return '"' . $this->prefix . $table . '"';
}
protected function column_quote($string)
{
preg_match('/(\(JSON\)\s*|^#)?([a-zA-Z0-9_]*)\.([a-zA-Z0-9_]*)/', $string, $column_match);
if (isset($column_match[ 2 ], $column_match[ 3 ]))
{
return '"' . $this->prefix . $column_match[ 2 ] . '"."' . $column_match[ 3 ] . '"';
}
return '"' . $string . '"';
}
protected function column_push(&$columns)
{
if ($columns == '*')
{
return $columns;
}
if (is_string($columns))
{
$columns = array($columns);
}
$stack = array();
foreach ($columns as $key => $value)
{
if (is_array($value))
{
$stack[] = $this->column_push($value);
}
else
{
preg_match('/([a-zA-Z0-9_\-\.]*)\s*\(([a-zA-Z0-9_\-]*)\)/i', $value, $match);
if (isset($match[ 1 ], $match[ 2 ]))
{
$stack[] = $this->column_quote( $match[ 1 ] ) . ' AS ' . $this->column_quote( $match[ 2 ] );
$columns[ $key ] = $match[ 2 ];
}
else
{
$stack[] = $this->column_quote( $value );
}
}
}
return implode($stack, ',');
}
protected function array_quote($array)
{
$temp = array();
foreach ($array as $value)
{
$temp[] = is_int($value) ? $value : $this->pdo->quote($value);
}
return implode($temp, ',');
}
protected function inner_conjunct($data, $conjunctor, $outer_conjunctor)
{
$haystack = array();
foreach ($data as $value)
{
$haystack[] = '(' . $this->data_implode($value, $conjunctor) . ')';
}
return implode($outer_conjunctor . ' ', $haystack);
}
protected function fn_quote($column, $string)
{
return (strpos($column, '#') === 0 && preg_match('/^[A-Z0-9\_]*\([^)]*\)$/', $string)) ?
$string :
$this->quote($string);
}
protected function data_implode($data, $conjunctor, $outer_conjunctor = null)
{
$wheres = array();
foreach ($data as $key => $value)
{
$type = gettype($value);
if (
preg_match("/^(AND|OR)(\s+#.*)?$/i", $key, $relation_match) &&
$type == 'array'
)
{
$wheres[] = 0 !== count(array_diff_key($value, array_keys(array_keys($value)))) ?
'(' . $this->data_implode($value, ' ' . $relation_match[ 1 ]) . ')' :
'(' . $this->inner_conjunct($value, ' ' . $relation_match[ 1 ], $conjunctor) . ')';
}
else
{
preg_match('/(#?)([\w\.\-]+)(\[(\>|\>\=|\<|\<\=|\!|\<\>|\>\<|\!?~)\])?/i', $key, $match);
$column = $this->column_quote($match[ 2 ]);
if (isset($match[ 4 ]))
{
$operator = $match[ 4 ];
if ($operator == '!')
{
switch ($type)
{
case 'NULL':
$wheres[] = $column . ' IS NOT NULL';
break;
case 'array':
$wheres[] = $column . ' NOT IN (' . $this->array_quote($value) . ')';
break;
case 'integer':
case 'double':
$wheres[] = $column . ' != ' . $value;
break;
case 'boolean':
$wheres[] = $column . ' != ' . ($value ? '1' : '0');
break;
case 'string':
$wheres[] = $column . ' != ' . $this->fn_quote($key, $value);
break;
}
}
if ($operator == '<>' || $operator == '><')
{
if ($type == 'array')
{
if ($operator == '><')
{
$column .= ' NOT';
}
if (is_numeric($value[ 0 ]) && is_numeric($value[ 1 ]))
{
$wheres[] = '(' . $column . ' BETWEEN ' . $value[ 0 ] . ' AND ' . $value[ 1 ] . ')';
}
else
{
$wheres[] = '(' . $column . ' BETWEEN ' . $this->quote($value[ 0 ]) . ' AND ' . $this->quote($value[ 1 ]) . ')';
}
}
}
if ($operator == '~' || $operator == '!~')
{
if ($type != 'array')
{
$value = array($value);
}
$like_clauses = array();
foreach ($value as $item)
{
$item = strval($item);
if (preg_match('/^(?!(%|\[|_])).+(?<!(%|\]|_))$/', $item))
{
$item = '%' . $item . '%';
}
$like_clauses[] = $column . ($operator === '!~' ? ' NOT' : '') . ' LIKE ' . $this->fn_quote($key, $item);
}
$wheres[] = implode(' OR ', $like_clauses);
}
if (in_array($operator, array('>', '>=', '<', '<=')))
{
$condition = $column . ' ' . $operator . ' ';
if (is_numeric($value))
{
$condition .= $value;
}
elseif (strpos($key, '#') === 0)
{
$condition .= $this->fn_quote($key, $value);
}
else
{
$condition .= $this->quote($value);
}
$wheres[] = $condition;
}
}
else
{
switch ($type)
{
case 'NULL':
$wheres[] = $column . ' IS NULL';
break;
case 'array':
$wheres[] = $column . ' IN (' . $this->array_quote($value) . ')';
break;
case 'integer':
case 'double':
$wheres[] = $column . ' = ' . $value;
break;
case 'boolean':
$wheres[] = $column . ' = ' . ($value ? '1' : '0');
break;
case 'string':
$wheres[] = $column . ' = ' . $this->fn_quote($key, $value);
break;
}
}
}
}
return implode($conjunctor . ' ', $wheres);
}
protected function where_clause($where)
{
$where_clause = '';
if (is_array($where))
{
$where_keys = array_keys($where);
$where_AND = preg_grep("/^AND\s*#?$/i", $where_keys);
$where_OR = preg_grep("/^OR\s*#?$/i", $where_keys);
$single_condition = array_diff_key($where, array_flip(
array('AND', 'OR', 'GROUP', 'ORDER', 'HAVING', 'LIMIT', 'LIKE', 'MATCH')
));
if ($single_condition != array())
{
$condition = $this->data_implode($single_condition, '');
if ($condition != '')
{
$where_clause = ' WHERE ' . $condition;
}
}
if (!empty($where_AND))
{
$value = array_values($where_AND);
$where_clause = ' WHERE ' . $this->data_implode($where[ $value[ 0 ] ], ' AND');
}
if (!empty($where_OR))
{
$value = array_values($where_OR);
$where_clause = ' WHERE ' . $this->data_implode($where[ $value[ 0 ] ], ' OR');
}
if (isset($where[ 'MATCH' ]))
{
$MATCH = $where[ 'MATCH' ];
if (is_array($MATCH) && isset($MATCH[ 'columns' ], $MATCH[ 'keyword' ]))
{
$where_clause .= ($where_clause != '' ? ' AND ' : ' WHERE ') . ' MATCH ("' . str_replace('.', '"."', implode($MATCH[ 'columns' ], '", "')) . '") AGAINST (' . $this->quote($MATCH[ 'keyword' ]) . ')';
}
}
if (isset($where[ 'GROUP' ]))
{
$where_clause .= ' GROUP BY ' . $this->column_quote($where[ 'GROUP' ]);
if (isset($where[ 'HAVING' ]))
{
$where_clause .= ' HAVING ' . $this->data_implode($where[ 'HAVING' ], ' AND');
}
}
if (isset($where[ 'ORDER' ]))
{
$ORDER = $where[ 'ORDER' ];
if (is_array($ORDER))
{
$stack = array();
foreach ($ORDER as $column => $value)
{
if (is_array($value))
{
$stack[] = 'FIELD(' . $this->column_quote($column) . ', ' . $this->array_quote($value) . ')';
}
else if ($value === 'ASC' || $value === 'DESC')
{
$stack[] = $this->column_quote($column) . ' ' . $value;
}
else if (is_int($column))
{
$stack[] = $this->column_quote($value);
}
}
$where_clause .= ' ORDER BY ' . implode($stack, ',');
}
else
{
$where_clause .= ' ORDER BY ' . $this->column_quote($ORDER);
}
}
if (isset($where[ 'LIMIT' ]))
{
$LIMIT = $where[ 'LIMIT' ];
if (is_numeric($LIMIT))
{
$where_clause .= ' LIMIT ' . $LIMIT;
}
if (
is_array($LIMIT) &&
is_numeric($LIMIT[ 0 ]) &&
is_numeric($LIMIT[ 1 ])
)
{
if ($this->database_type === 'pgsql')
{
$where_clause .= ' OFFSET ' . $LIMIT[ 0 ] . ' LIMIT ' . $LIMIT[ 1 ];
}
else
{
$where_clause .= ' LIMIT ' . $LIMIT[ 0 ] . ',' . $LIMIT[ 1 ];
}
}
}
}
else
{
if ($where != null)
{
$where_clause .= ' ' . $where;
}
}
return $where_clause;
}
protected function select_context($table, $join, &$columns = null, $where = null, $column_fn = null)
{
preg_match('/([a-zA-Z0-9_\-]*)\s*\(([a-zA-Z0-9_\-]*)\)/i', $table, $table_match);
if (isset($table_match[ 1 ], $table_match[ 2 ]))
{
$table = $this->table_quote($table_match[ 1 ]);
$table_query = $this->table_quote($table_match[ 1 ]) . ' AS ' . $this->table_quote($table_match[ 2 ]);
}
else
{
$table = $this->table_quote($table);
$table_query = $table;
}
$join_key = is_array($join) ? array_keys($join) : null;
if (
isset($join_key[ 0 ]) &&
strpos($join_key[ 0 ], '[') === 0
)
{
$table_join = array();
$join_array = array(
'>' => 'LEFT',
'<' => 'RIGHT',
'<>' => 'FULL',
'><' => 'INNER'
);
foreach($join as $sub_table => $relation)
{
preg_match('/(\[(\<|\>|\>\<|\<\>)\])?([a-zA-Z0-9_\-]*)\s?(\(([a-zA-Z0-9_\-]*)\))?/', $sub_table, $match);
if ($match[ 2 ] != '' && $match[ 3 ] != '')
{
if (is_string($relation))
{
$relation = 'USING ("' . $relation . '")';
}
if (is_array($relation))
{
// For ['column1', 'column2']
if (isset($relation[ 0 ]))
{
$relation = 'USING ("' . implode($relation, '", "') . '")';
}
else
{
$joins = array();
foreach ($relation as $key => $value)
{
$joins[] = (
strpos($key, '.') > 0 ?
// For ['tableB.column' => 'column']
$this->column_quote($key) :
// For ['column1' => 'column2']
$table . '."' . $key . '"'
) .
' = ' .
$this->table_quote(isset($match[ 5 ]) ? $match[ 5 ] : $match[ 3 ]) . '."' . $value . '"';
}
$relation = 'ON ' . implode($joins, ' AND ');
}
}
$table_name = $this->table_quote($match[ 3 ]) . ' ';
if (isset($match[ 5 ]))
{
$table_name .= 'AS ' . $this->table_quote($match[ 5 ]) . ' ';
}
$table_join[] = $join_array[ $match[ 2 ] ] . ' JOIN ' . $table_name . $relation;
}
}
$table_query .= ' ' . implode($table_join, ' ');
}
else
{
if (is_null($columns))
{
if (is_null($where))
{
if (
is_array($join) &&
isset($column_fn)
)
{
$where = $join;
$columns = null;
}
else
{
$where = null;
$columns = $join;
}
}
else
{
$where = $join;
$columns = null;
}
}
else
{
$where = $columns;
$columns = $join;
}
}
if (isset($column_fn))
{
if ($column_fn == 1)
{
$column = '1';
if (is_null($where))
{
$where = $columns;
}
}
else
{
if (empty($columns))
{
$columns = '*';
$where = $join;
}
$column = $column_fn . '(' . $this->column_push($columns) . ')';
}
}
else
{
$column = $this->column_push($columns);
}
return 'SELECT ' . $column . ' FROM ' . $table_query . $this->where_clause($where);
}
protected function data_map($index, $key, $value, $data, &$stack)
{
if (is_array($value))
{
$sub_stack = array();
foreach ($value as $sub_key => $sub_value)
{
if (is_array($sub_value))
{
$current_stack = $stack[ $index ][ $key ];
$this->data_map(false, $sub_key, $sub_value, $data, $current_stack);
$stack[ $index ][ $key ][ $sub_key ] = $current_stack[ 0 ][ $sub_key ];
}
else
{
$this->data_map(false, preg_replace('/^[\w]*\./i', "", $sub_value), $sub_key, $data, $sub_stack);
$stack[ $index ][ $key ] = $sub_stack;
}
}
}
else
{
if ($index !== false)
{
$stack[ $index ][ $value ] = $data[ $value ];
}
else
{
if (preg_match('/[a-zA-Z0-9_\-\.]*\s*\(([a-zA-Z0-9_\-]*)\)/i', $key, $key_match))
{
$key = $key_match[ 1 ];
}
$stack[ $key ] = $data[ $key ];
}
}
}
public function select($table, $join, $columns = null, $where = null)
{
$column = $where == null ? $join : $columns;
$is_single_column = (is_string($column) && $column !== '*');
$query = $this->query($this->select_context($table, $join, $columns, $where));
$stack = array();
$index = 0;
if (!$query)
{
return false;
}
if ($columns === '*')
{
return $query->fetchAll(PDO::FETCH_ASSOC);
}
if ($is_single_column)
{
return $query->fetchAll(PDO::FETCH_COLUMN);
}
while ($row = $query->fetch(PDO::FETCH_ASSOC))
{
foreach ($columns as $key => $value)
{
if (is_array($value))
{
$this->data_map($index, $key, $value, $row, $stack);
}
else
{
$this->data_map($index, $key, preg_replace('/^[\w]*\./i', "", $value), $row, $stack);
}
}
$index++;
}
return $stack;
}
public function insert($table, $datas)
{
$lastId = array();
// Check indexed or associative array
if (!isset($datas[ 0 ]))
{
$datas = array($datas);
}
foreach ($datas as $data)
{
$values = array();
$columns = array();
foreach ($data as $key => $value)
{
$columns[] = $this->column_quote(preg_replace("/^(\(JSON\)\s*|#)/i", "", $key));
switch (gettype($value))
{
case 'NULL':
$values[] = 'NULL';
break;
case 'array':
preg_match("/\(JSON\)\s*([\w]+)/i", $key, $column_match);
$values[] = isset($column_match[ 0 ]) ?
$this->quote(json_encode($value)) :
$this->quote(serialize($value));
break;
case 'boolean':
$values[] = ($value ? '1' : '0');
break;
case 'integer':
case 'double':
case 'string':
$values[] = $this->fn_quote($key, $value);
break;
}
}
$this->exec('INSERT INTO ' . $this->table_quote($table) . ' (' . implode(', ', $columns) . ') VALUES (' . implode($values, ', ') . ')');
$lastId[] = $this->pdo->lastInsertId();
}
return count($lastId) > 1 ? $lastId : $lastId[ 0 ];
}
public function update($table, $data, $where = null)
{
$fields = array();
foreach ($data as $key => $value)
{
preg_match('/([\w]+)(\[(\+|\-|\*|\/)\])?/i', $key, $match);
if (isset($match[ 3 ]))
{
if (is_numeric($value))
{
$fields[] = $this->column_quote($match[ 1 ]) . ' = ' . $this->column_quote($match[ 1 ]) . ' ' . $match[ 3 ] . ' ' . $value;
}
}
else
{
$column = $this->column_quote(preg_replace("/^(\(JSON\)\s*|#)/i", "", $key));
switch (gettype($value))
{
case 'NULL':
$fields[] = $column . ' = NULL';
break;
case 'array':
preg_match("/\(JSON\)\s*([\w]+)/i", $key, $column_match);
$fields[] = $column . ' = ' . $this->quote(
isset($column_match[ 0 ]) ? json_encode($value) : serialize($value)
);
break;
case 'boolean':
$fields[] = $column . ' = ' . ($value ? '1' : '0');
break;
case 'integer':
case 'double':
case 'string':
$fields[] = $column . ' = ' . $this->fn_quote($key, $value);
break;
}
}
}
return $this->exec('UPDATE ' . $this->table_quote($table) . ' SET ' . implode(', ', $fields) . $this->where_clause($where));
}
public function delete($table, $where)
{
return $this->exec('DELETE FROM ' . $this->table_quote($table) . $this->where_clause($where));
}
public function replace($table, $columns, $search = null, $replace = null, $where = null)
{
if (is_array($columns))
{
$replace_query = array();
foreach ($columns as $column => $replacements)
{
foreach ($replacements as $replace_search => $replace_replacement)
{
$replace_query[] = $column . ' = REPLACE(' . $this->column_quote($column) . ', ' . $this->quote($replace_search) . ', ' . $this->quote($replace_replacement) . ')';
}
}
$replace_query = implode(', ', $replace_query);
$where = $search;
}
else
{
if (is_array($search))
{
$replace_query = array();
foreach ($search as $replace_search => $replace_replacement)
{
$replace_query[] = $columns . ' = REPLACE(' . $this->column_quote($columns) . ', ' . $this->quote($replace_search) . ', ' . $this->quote($replace_replacement) . ')';
}
$replace_query = implode(', ', $replace_query);
$where = $replace;
}
else
{
$replace_query = $columns . ' = REPLACE(' . $this->column_quote($columns) . ', ' . $this->quote($search) . ', ' . $this->quote($replace) . ')';
}
}
return $this->exec('UPDATE ' . $this->table_quote($table) . ' SET ' . $replace_query . $this->where_clause($where));
}
public function get($table, $join = null, $columns = null, $where = null)
{
$column = $where == null ? $join : $columns;
$is_single_column = (is_string($column) && $column !== '*');
$query = $this->query($this->select_context($table, $join, $columns, $where) . ' LIMIT 1');
if ($query)
{
$data = $query->fetchAll(PDO::FETCH_ASSOC);
if (isset($data[ 0 ]))
{
if ($is_single_column)
{
return $data[ 0 ][ preg_replace('/^[\w]*\./i', "", $column) ];
}
if ($column === '*')
{
return $data[ 0 ];
}
$stack = array();
foreach ($columns as $key => $value)
{
if (is_array($value))
{
$this->data_map(0, $key, $value, $data[ 0 ], $stack);
}
else
{
$this->data_map(0, $key, preg_replace('/^[\w]*\./i', "", $value), $data[ 0 ], $stack);
}
}
return $stack[ 0 ];
}
else
{
return false;
}
}
else
{
return false;
}
}
public function has($table, $join, $where = null)
{
$column = null;
$query = $this->query('SELECT EXISTS(' . $this->select_context($table, $join, $column, $where, 1) . ')');
if ($query)
{
return $query->fetchColumn() === '1';
}
else
{
return false;
}
}
public function count($table, $join = null, $column = null, $where = null)
{
$query = $this->query($this->select_context($table, $join, $column, $where, 'COUNT'));
return $query ? 0 + $query->fetchColumn() : false;
}
public function max($table, $join, $column = null, $where = null)
{
$query = $this->query($this->select_context($table, $join, $column, $where, 'MAX'));
if ($query)
{
$max = $query->fetchColumn();
return is_numeric($max) ? $max + 0 : $max;
}
else
{
return false;
}
}
public function min($table, $join, $column = null, $where = null)
{
$query = $this->query($this->select_context($table, $join, $column, $where, 'MIN'));
if ($query)
{
$min = $query->fetchColumn();
return is_numeric($min) ? $min + 0 : $min;
}
else
{
return false;
}
}
public function avg($table, $join, $column = null, $where = null)
{
$query = $this->query($this->select_context($table, $join, $column, $where, 'AVG'));
return $query ? 0 + $query->fetchColumn() : false;
}
public function sum($table, $join, $column = null, $where = null)
{
$query = $this->query($this->select_context($table, $join, $column, $where, 'SUM'));
return $query ? 0 + $query->fetchColumn() : false;
}
public function action($actions)
{
if (is_callable($actions))
{
$this->pdo->beginTransaction();
$result = $actions($this);
if ($result === false)
{
$this->pdo->rollBack();
}
else
{
$this->pdo->commit();
}
}
else
{
return false;
}
}
public function debug()
{
$this->debug_mode = true;
return $this;
}
public function error()
{
return $this->pdo->errorInfo();
}
public function last_query()
{
return end($this->logs);
}
public function log()
{
return $this->logs;
}
public function info()
{
$output = array(
'server' => 'SERVER_INFO',
'driver' => 'DRIVER_NAME',
'client' => 'CLIENT_VERSION',
'version' => 'SERVER_VERSION',
'connection' => 'CONNECTION_STATUS'
);
foreach ($output as $key => $value)
{
$output[ $key ] = $this->pdo->getAttribute(constant('PDO::ATTR_' . $value));
}
return $output;
}
}
?>