public function __construct(Connection $connection)
{
$this->connection = $connection;
$this->driver = $connection->getSupplementalDriver();
$this->arrayModes = array(
'INSERT' => $this->driver->isSupported(ISupplementalDriver::SUPPORT_MULTI_INSERT_AS_SELECT) ? 'select' : 'values',
'REPLACE' => 'values',
'UPDATE' => 'assoc',
'WHERE' => 'and',
'HAVING' => 'and',
'ORDER BY' => 'order',
'GROUP BY' => 'order',
);
}
protected function addWhereComposition(array $columns, array $parameters)
{
if ($this->driver->isSupported(ISupplementalDriver::SUPPORT_MULTI_COLUMN_AS_OR_COND)) {
$conditionFragment = '(' . implode(' = ? AND ', $columns) . ' = ?) OR ';
$condition = substr(str_repeat($conditionFragment, count($parameters)), 0, -4);
return $this->addWhere($condition, Nette\Utils\Arrays::flatten($parameters));
} else {
return $this->addWhere('(' . implode(', ', $columns) . ') IN', $parameters);
}
}
/**
* Returns SQL query.
* @param list of columns
* @return string
*/
public function buildSelectQuery($columns = NULL)
{
$join = $this->buildJoins(implode(',', $this->conditions), TRUE);
$join += $this->buildJoins(implode(',', $this->select) . ",{$this->group},{$this->having}," . implode(',', $this->order));
$prefix = $join ? "{$this->delimitedTable}." : '';
if ($this->select) {
$cols = $this->tryDelimite($this->removeExtraTables(implode(', ', $this->select)));
} elseif ($columns) {
$cols = array_map(array($this->driver, 'delimite'), $columns);
$cols = $prefix . implode(', ' . $prefix, $cols);
} elseif ($this->group && !$this->driver->isSupported(ISupplementalDriver::SUPPORT_SELECT_UNGROUPED_COLUMNS)) {
$cols = $this->tryDelimite($this->removeExtraTables($this->group));
} else {
$cols = $prefix . '*';
}
return "SELECT{$this->buildTopClause()} {$cols} FROM {$this->delimitedTable}" . implode($join) . $this->buildConditions();
}
private function formatValue($value, $mode = NULL)
{
if (!$mode || $mode === 'auto') {
if (is_string($value)) {
if (strlen($value) > 20) {
$this->remaining[] = $value;
return '?';
} else {
return $this->connection->quote($value);
}
} elseif (is_int($value)) {
return (string) $value;
} elseif (is_float($value)) {
return rtrim(rtrim(number_format($value, 10, '.', ''), '0'), '.');
} elseif (is_bool($value)) {
return $this->driver->formatBool($value);
} elseif ($value === NULL) {
return 'NULL';
} elseif ($value instanceof Table\IRow) {
return $value->getPrimary();
} elseif ($value instanceof SqlLiteral) {
$prep = clone $this;
list($res, $params) = $prep->process(array_merge([$value->__toString()], $value->getParameters()));
$this->remaining = array_merge($this->remaining, $params);
return $res;
} elseif ($value instanceof \DateTimeInterface) {
return $this->driver->formatDateTime($value);
} elseif ($value instanceof \DateInterval) {
return $this->driver->formatDateInterval($value);
} elseif (is_object($value) && method_exists($value, '__toString')) {
return $this->formatValue((string) $value);
} elseif (is_resource($value)) {
$this->remaining[] = $value;
return '?';
}
} elseif ($mode === 'name') {
if (!is_string($value)) {
$type = gettype($value);
throw new Nette\InvalidArgumentException("Placeholder ?{$mode} expects string, {$type} given.");
}
return $this->delimite($value);
}
if ($value instanceof \Traversable && !$value instanceof Table\IRow) {
$value = iterator_to_array($value);
}
if (is_array($value)) {
$vx = $kx = [];
if ($mode === 'auto') {
$mode = $this->arrayMode;
}
if ($mode === 'values') {
// (key, key, ...) VALUES (value, value, ...)
if (array_key_exists(0, $value)) {
// multi-insert
foreach ($value[0] as $k => $v) {
$kx[] = $this->delimite($k);
}
foreach ($value as $val) {
$vx2 = [];
foreach ($val as $v) {
$vx2[] = $this->formatValue($v);
}
$vx[] = implode(', ', $vx2);
}
$select = $this->driver->isSupported(ISupplementalDriver::SUPPORT_MULTI_INSERT_AS_SELECT);
return '(' . implode(', ', $kx) . ($select ? ') SELECT ' : ') VALUES (') . implode($select ? ' UNION ALL SELECT ' : '), (', $vx) . ($select ? '' : ')');
}
foreach ($value as $k => $v) {
$kx[] = $this->delimite($k);
$vx[] = $this->formatValue($v);
}
return '(' . implode(', ', $kx) . ') VALUES (' . implode(', ', $vx) . ')';
} elseif (!$mode || $mode === 'set') {
foreach ($value as $k => $v) {
if (is_int($k)) {
// value, value, ... OR (1, 2), (3, 4)
$vx[] = is_array($v) ? '(' . $this->formatValue($v) . ')' : $this->formatValue($v);
} elseif (substr($k, -1) === '=') {
// key+=value, key-=value, ...
$k2 = $this->delimite(substr($k, 0, -2));
$vx[] = $k2 . '=' . $k2 . ' ' . substr($k, -2, 1) . ' ' . $this->formatValue($v);
} else {
// key=value, key=value, ...
$vx[] = $this->delimite($k) . '=' . $this->formatValue($v);
}
}
return implode(', ', $vx);
} elseif ($mode === 'and' || $mode === 'or') {
// (key [operator] value) AND ...
foreach ($value as $k => $v) {
if (is_int($k)) {
$vx[] = $this->formatValue($v);
continue;
}
list($k, $operator) = explode(' ', $k . ' ');
$k = $this->delimite($k);
if (is_array($v)) {
if ($v) {
$vx[] = $k . ' ' . ($operator ? $operator . ' ' : '') . 'IN (' . $this->formatValue(array_values($v)) . ')';
} elseif ($operator === 'NOT') {
} else {
$vx[] = '1=0';
}
} else {
$v = $this->formatValue($v);
$vx[] = $k . ' ' . ($operator ?: ($v === 'NULL' ? 'IS' : '=')) . ' ' . $v;
}
}
return $value ? '(' . implode(') ' . strtoupper($mode) . ' (', $vx) . ')' : '1=1';
} elseif ($mode === 'order') {
// key, key DESC, ...
foreach ($value as $k => $v) {
$vx[] = $this->delimite($k) . ($v > 0 ? '' : ' DESC');
}
return implode(', ', $vx);
} else {
throw new Nette\InvalidArgumentException("Unknown placeholder ?{$mode}.");
}
} elseif (in_array($mode, ['and', 'or', 'set', 'values', 'order'], TRUE)) {
$type = gettype($value);
throw new Nette\InvalidArgumentException("Placeholder ?{$mode} expects array or Traversable object, {$type} given.");
} elseif ($mode && $mode !== 'auto') {
throw new Nette\InvalidArgumentException("Unknown placeholder ?{$mode}.");
} else {
throw new Nette\InvalidArgumentException('Unexpected type of parameter: ' . (is_object($value) ? get_class($value) : gettype($value)));
}
}
