miniflux-legacy/vendor/PicoDb/Table.php

413 lines
9.5 KiB
PHP
Raw Normal View History

2013-02-18 03:48:21 +01:00
<?php
namespace PicoDb;
class Table
{
private $table_name = '';
private $sql_limit = '';
private $sql_offset = '';
private $sql_order = '';
private $joins = array();
private $conditions = array();
private $or_conditions = array();
private $is_or_condition = false;
private $columns = array();
private $values = array();
2013-07-13 02:26:47 +02:00
private $distinct = false;
private $group_by = array();
2013-02-18 03:48:21 +01:00
private $db;
public function __construct(Database $db, $table_name)
{
$this->db = $db;
$this->table_name = $table_name;
return $this;
}
public function save(array $data)
{
if (! empty($this->conditions)) {
return $this->update($data);
}
else {
return $this->insert($data);
}
}
public function update(array $data)
{
$columns = array();
$values = array();
foreach ($data as $column => $value) {
$columns[] = $this->db->escapeIdentifier($column).'=?';
$values[] = $value;
}
foreach ($this->values as $value) {
$values[] = $value;
}
$sql = sprintf(
'UPDATE %s SET %s %s',
$this->db->escapeIdentifier($this->table_name),
implode(', ', $columns),
$this->conditions()
);
2013-04-23 04:43:38 +02:00
$result = $this->db->execute($sql, $values);
if ($result !== false && $result->rowCount() > 0) {
return true;
}
return false;
2013-02-18 03:48:21 +01:00
}
public function insert(array $data)
{
$columns = array();
foreach ($data as $column => $value) {
$columns[] = $this->db->escapeIdentifier($column);
}
$sql = sprintf(
'INSERT INTO %s (%s) VALUES (%s)',
$this->db->escapeIdentifier($this->table_name),
implode(', ', $columns),
implode(', ', array_fill(0, count($data), '?'))
);
return false !== $this->db->execute($sql, array_values($data));
}
public function remove()
{
$sql = sprintf(
'DELETE FROM %s %s',
$this->db->escapeIdentifier($this->table_name),
$this->conditions()
);
return false !== $this->db->execute($sql, $this->values);
}
public function listing($key, $value)
{
$this->columns($key, $value);
$listing = array();
$results = $this->findAll();
if ($results) {
foreach ($results as $result) {
$listing[$result[$key]] = $result[$value];
}
}
return $listing;
}
public function findAll()
{
2013-05-26 19:07:45 +02:00
$rq = $this->db->execute($this->buildSelectQuery(), $this->values);
if (false === $rq) return false;
2013-02-18 03:48:21 +01:00
return $rq->fetchAll(\PDO::FETCH_ASSOC);
}
2013-07-13 02:26:47 +02:00
public function findAllByColumn($column)
{
$rq = $this->db->execute($this->buildSelectQuery(), $this->values);
if (false === $rq) return false;
return $rq->fetchAll(\PDO::FETCH_COLUMN, 0);
}
2013-02-18 03:48:21 +01:00
public function findOne()
{
$this->limit(1);
$result = $this->findAll();
return isset($result[0]) ? $result[0] : null;
}
2013-05-26 19:07:45 +02:00
public function findOneColumn($column)
{
$this->limit(1);
$this->columns = array($column);
$rq = $this->db->execute($this->buildSelectQuery(), $this->values);
if (false === $rq) return false;
return $rq->fetchColumn();
}
public function buildSelectQuery()
{
return sprintf(
2013-07-13 02:26:47 +02:00
'SELECT %s %s FROM %s %s %s %s %s %s %s',
$this->distinct ? 'DISTINCT' : '',
2013-05-26 19:07:45 +02:00
empty($this->columns) ? '*' : implode(', ', $this->columns),
$this->db->escapeIdentifier($this->table_name),
implode(' ', $this->joins),
$this->conditions(),
2013-07-13 02:26:47 +02:00
empty($this->group_by) ? '' : 'GROUP BY '.implode(', ', $this->group_by),
2013-05-26 19:07:45 +02:00
$this->sql_order,
$this->sql_limit,
$this->sql_offset
);
}
2013-02-18 03:48:21 +01:00
public function count()
{
$sql = sprintf(
2013-05-26 19:07:45 +02:00
'SELECT COUNT(*) FROM %s'.$this->conditions().$this->sql_order.$this->sql_limit.$this->sql_offset,
2013-02-18 03:48:21 +01:00
$this->db->escapeIdentifier($this->table_name)
);
$rq = $this->db->execute($sql, $this->values);
2013-05-26 19:07:45 +02:00
if (false === $rq) return false;
2013-02-18 03:48:21 +01:00
2013-05-26 19:07:45 +02:00
$result = $rq->fetchColumn();
return $result ? (int) $result : 0;
2013-02-18 03:48:21 +01:00
}
public function join($table, $foreign_column, $local_column)
{
$this->joins[] = sprintf(
'LEFT JOIN %s ON %s=%s',
$this->db->escapeIdentifier($table),
$this->db->escapeIdentifier($table).'.'.$this->db->escapeIdentifier($foreign_column),
$this->db->escapeIdentifier($this->table_name).'.'.$this->db->escapeIdentifier($local_column)
);
return $this;
}
public function conditions()
{
if (! empty($this->conditions)) {
return ' WHERE '.implode(' AND ', $this->conditions);
}
else {
return '';
}
}
public function addCondition($sql)
{
if ($this->is_or_condition) {
$this->or_conditions[] = $sql;
}
else {
$this->conditions[] = $sql;
}
}
public function beginOr()
{
$this->is_or_condition = true;
$this->or_conditions = array();
return $this;
}
public function closeOr()
{
$this->is_or_condition = false;
if (! empty($this->or_conditions)) {
$this->conditions[] = '('.implode(' OR ', $this->or_conditions).')';
}
return $this;
}
public function asc($column)
{
2013-07-06 04:37:19 +02:00
if ($this->sql_order === '') {
$this->sql_order = ' ORDER BY '.$this->db->escapeIdentifier($column).' ASC';
}
else {
$this->sql_order .= ', '.$this->db->escapeIdentifier($column).' ASC';
}
2013-02-18 03:48:21 +01:00
return $this;
}
public function desc($column)
{
2013-07-06 04:37:19 +02:00
if ($this->sql_order === '') {
$this->sql_order = ' ORDER BY '.$this->db->escapeIdentifier($column).' DESC';
}
else {
$this->sql_order .= ', '.$this->db->escapeIdentifier($column).' DESC';
}
2013-02-18 03:48:21 +01:00
return $this;
}
public function limit($value)
{
2013-07-06 04:37:19 +02:00
if (! is_null($value)) $this->sql_limit = ' LIMIT '.(int) $value;
2013-02-18 03:48:21 +01:00
return $this;
}
public function offset($value)
{
2013-07-06 04:37:19 +02:00
if (! is_null($value)) $this->sql_offset = ' OFFSET '.(int) $value;
2013-02-18 03:48:21 +01:00
return $this;
}
2013-07-13 02:26:47 +02:00
public function groupBy()
{
$this->group_by = \func_get_args();
return $this;
}
2013-02-18 03:48:21 +01:00
public function columns()
{
$this->columns = \func_get_args();
return $this;
}
2013-07-13 02:26:47 +02:00
public function distinct()
2013-02-18 03:48:21 +01:00
{
2013-07-13 02:26:47 +02:00
$this->columns = \func_get_args();
$this->distinct = true;
return $this;
}
2013-02-18 03:48:21 +01:00
2013-07-13 02:26:47 +02:00
public function __call($name, array $arguments)
{
2013-02-18 03:48:21 +01:00
$column = $arguments[0];
$sql = '';
2013-07-13 02:26:47 +02:00
switch (strtolower($name)) {
2013-02-18 03:48:21 +01:00
case 'in':
2013-07-13 02:26:47 +02:00
if (isset($arguments[1]) && is_array($arguments[1])) {
2013-02-18 03:48:21 +01:00
$sql = sprintf(
'%s IN (%s)',
$this->db->escapeIdentifier($column),
implode(', ', array_fill(0, count($arguments[1]), '?'))
);
}
break;
2013-04-23 04:43:38 +02:00
case 'notin':
2013-07-13 02:26:47 +02:00
if (isset($arguments[1]) && is_array($arguments[1])) {
2013-04-23 04:43:38 +02:00
$sql = sprintf(
'%s NOT IN (%s)',
$this->db->escapeIdentifier($column),
implode(', ', array_fill(0, count($arguments[1]), '?'))
);
}
break;
2013-02-18 03:48:21 +01:00
case 'like':
$sql = sprintf('%s LIKE ?', $this->db->escapeIdentifier($column));
break;
case 'eq':
case 'equal':
case 'equals':
$sql = sprintf('%s = ?', $this->db->escapeIdentifier($column));
break;
case 'gt':
2013-07-13 02:26:47 +02:00
case 'greaterthan':
2013-02-18 03:48:21 +01:00
$sql = sprintf('%s > ?', $this->db->escapeIdentifier($column));
break;
case 'lt':
2013-07-13 02:26:47 +02:00
case 'lowerthan':
2013-02-18 03:48:21 +01:00
$sql = sprintf('%s < ?', $this->db->escapeIdentifier($column));
break;
case 'gte':
2013-07-13 02:26:47 +02:00
case 'greaterthanorequals':
2013-02-18 03:48:21 +01:00
$sql = sprintf('%s >= ?', $this->db->escapeIdentifier($column));
break;
case 'lte':
2013-07-13 02:26:47 +02:00
case 'lowerthanorequals':
2013-02-18 03:48:21 +01:00
$sql = sprintf('%s <= ?', $this->db->escapeIdentifier($column));
break;
2013-07-13 02:26:47 +02:00
case 'isnull':
$sql = sprintf('%s IS NULL', $this->db->escapeIdentifier($column));
break;
case 'notnull':
$sql = sprintf('%s IS NOT NULL', $this->db->escapeIdentifier($column));
break;
2013-02-18 03:48:21 +01:00
}
2013-07-13 02:26:47 +02:00
if ($sql !== '') {
2013-02-18 03:48:21 +01:00
$this->addCondition($sql);
2013-07-13 02:26:47 +02:00
if (isset($arguments[1])) {
2013-02-18 03:48:21 +01:00
2013-07-13 02:26:47 +02:00
if (is_array($arguments[1])) {
2013-02-18 03:48:21 +01:00
2013-07-13 02:26:47 +02:00
foreach ($arguments[1] as $value) {
$this->values[] = $value;
}
2013-02-18 03:48:21 +01:00
}
2013-07-13 02:26:47 +02:00
else {
2013-02-18 03:48:21 +01:00
2013-07-13 02:26:47 +02:00
$this->values[] = $arguments[1];
}
2013-02-18 03:48:21 +01:00
}
}
return $this;
}
}