spqr/vendor/cakephp/database/Expression/WhenThenExpression.php
2024-11-05 12:10:06 +08:00

351 lines
11 KiB
PHP

<?php
declare(strict_types=1);
/**
* CakePHP(tm) : Rapid Development Framework (https://cakephp.org)
* Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
*
* Licensed under The MIT License
* For full copyright and license information, please see the LICENSE.txt
* Redistributions of files must retain the above copyright notice.
*
* @copyright Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
* @link https://cakephp.org CakePHP(tm) Project
* @since 4.3.0
* @license https://opensource.org/licenses/mit-license.php MIT License
*/
namespace Cake\Database\Expression;
use Cake\Database\ExpressionInterface;
use Cake\Database\Query;
use Cake\Database\Type\ExpressionTypeCasterTrait;
use Cake\Database\TypeMap;
use Cake\Database\ValueBinder;
use Closure;
use InvalidArgumentException;
use LogicException;
use function Cake\Core\getTypeName;
/**
* Represents a SQL when/then clause with a fluid API
*/
class WhenThenExpression implements ExpressionInterface
{
use CaseExpressionTrait;
use ExpressionTypeCasterTrait;
/**
* The names of the clauses that are valid for use with the
* `clause()` method.
*
* @var array<string>
*/
protected $validClauseNames = [
'when',
'then',
];
/**
* The type map to use when using an array of conditions for the
* `WHEN` value.
*
* @var \Cake\Database\TypeMap
*/
protected $_typeMap;
/**
* Then `WHEN` value.
*
* @var \Cake\Database\ExpressionInterface|object|scalar|null
*/
protected $when = null;
/**
* The `WHEN` value type.
*
* @var array|string|null
*/
protected $whenType = null;
/**
* The `THEN` value.
*
* @var \Cake\Database\ExpressionInterface|object|scalar|null
*/
protected $then = null;
/**
* Whether the `THEN` value has been defined, eg whether `then()`
* has been invoked.
*
* @var bool
*/
protected $hasThenBeenDefined = false;
/**
* The `THEN` result type.
*
* @var string|null
*/
protected $thenType = null;
/**
* Constructor.
*
* @param \Cake\Database\TypeMap|null $typeMap The type map to use when using an array of conditions for the `WHEN`
* value.
*/
public function __construct(?TypeMap $typeMap = null)
{
if ($typeMap === null) {
$typeMap = new TypeMap();
}
$this->_typeMap = $typeMap;
}
/**
* Sets the `WHEN` value.
*
* @param \Cake\Database\ExpressionInterface|object|array|scalar $when The `WHEN` value. When using an array of
* conditions, it must be compatible with `\Cake\Database\Query::where()`. Note that this argument is _not_
* completely safe for use with user data, as a user supplied array would allow for raw SQL to slip in! If you
* plan to use user data, either pass a single type for the `$type` argument (which forces the `$when` value to be
* a non-array, and then always binds the data), use a conditions array where the user data is only passed on the
* value side of the array entries, or custom bindings!
* @param array<string, string>|string|null $type The when value type. Either an associative array when using array style
* conditions, or else a string. If no type is provided, the type will be tried to be inferred from the value.
* @return $this
* @throws \InvalidArgumentException In case the `$when` argument is neither a non-empty array, nor a scalar value,
* an object, or an instance of `\Cake\Database\ExpressionInterface`.
* @throws \InvalidArgumentException In case the `$type` argument is neither an array, a string, nor null.
* @throws \InvalidArgumentException In case the `$when` argument is an array, and the `$type` argument is neither
* an array, nor null.
* @throws \InvalidArgumentException In case the `$when` argument is a non-array value, and the `$type` argument is
* neither a string, nor null.
* @see CaseStatementExpression::when() for a more detailed usage explanation.
*/
public function when($when, $type = null)
{
if (
!(is_array($when) && !empty($when)) &&
!is_scalar($when) &&
!is_object($when)
) {
throw new InvalidArgumentException(sprintf(
'The `$when` argument must be either a non-empty array, a scalar value, an object, ' .
'or an instance of `\%s`, `%s` given.',
ExpressionInterface::class,
is_array($when) ? '[]' : getTypeName($when)
));
}
if (
$type !== null &&
!is_array($type) &&
!is_string($type)
) {
throw new InvalidArgumentException(sprintf(
'The `$type` argument must be either an array, a string, or `null`, `%s` given.',
getTypeName($type)
));
}
if (is_array($when)) {
if (
$type !== null &&
!is_array($type)
) {
throw new InvalidArgumentException(sprintf(
'When using an array for the `$when` argument, the `$type` argument must be an ' .
'array too, `%s` given.',
getTypeName($type)
));
}
// avoid dirtying the type map for possible consecutive `when()` calls
$typeMap = clone $this->_typeMap;
if (
is_array($type) &&
count($type) > 0
) {
$typeMap = $typeMap->setTypes($type);
}
$when = new QueryExpression($when, $typeMap);
} else {
if (
$type !== null &&
!is_string($type)
) {
throw new InvalidArgumentException(sprintf(
'When using a non-array value for the `$when` argument, the `$type` argument must ' .
'be a string, `%s` given.',
getTypeName($type)
));
}
if (
$type === null &&
!($when instanceof ExpressionInterface)
) {
$type = $this->inferType($when);
}
}
$this->when = $when;
$this->whenType = $type;
return $this;
}
/**
* Sets the `THEN` result value.
*
* @param \Cake\Database\ExpressionInterface|object|scalar|null $result The result value.
* @param string|null $type The result type. If no type is provided, the type will be inferred from the given
* result value.
* @return $this
*/
public function then($result, ?string $type = null)
{
if (
$result !== null &&
!is_scalar($result) &&
!(is_object($result) && !($result instanceof Closure))
) {
throw new InvalidArgumentException(sprintf(
'The `$result` argument must be either `null`, a scalar value, an object, ' .
'or an instance of `\%s`, `%s` given.',
ExpressionInterface::class,
getTypeName($result)
));
}
$this->then = $result;
if ($type === null) {
$type = $this->inferType($result);
}
$this->thenType = $type;
$this->hasThenBeenDefined = true;
return $this;
}
/**
* Returns the expression's result value type.
*
* @return string|null
* @see WhenThenExpression::then()
*/
public function getResultType(): ?string
{
return $this->thenType;
}
/**
* Returns the available data for the given clause.
*
* ### Available clauses
*
* The following clause names are available:
*
* * `when`: The `WHEN` value.
* * `then`: The `THEN` result value.
*
* @param string $clause The name of the clause to obtain.
* @return \Cake\Database\ExpressionInterface|object|scalar|null
* @throws \InvalidArgumentException In case the given clause name is invalid.
*/
public function clause(string $clause)
{
if (!in_array($clause, $this->validClauseNames, true)) {
throw new InvalidArgumentException(
sprintf(
'The `$clause` argument must be one of `%s`, the given value `%s` is invalid.',
implode('`, `', $this->validClauseNames),
$clause
)
);
}
return $this->{$clause};
}
/**
* @inheritDoc
*/
public function sql(ValueBinder $binder): string
{
if ($this->when === null) {
throw new LogicException('Case expression has incomplete when clause. Missing `when()`.');
}
if (!$this->hasThenBeenDefined) {
throw new LogicException('Case expression has incomplete when clause. Missing `then()` after `when()`.');
}
$when = $this->when;
if (
is_string($this->whenType) &&
!($when instanceof ExpressionInterface)
) {
$when = $this->_castToExpression($when, $this->whenType);
}
if ($when instanceof Query) {
$when = sprintf('(%s)', $when->sql($binder));
} elseif ($when instanceof ExpressionInterface) {
$when = $when->sql($binder);
} else {
$placeholder = $binder->placeholder('c');
if (is_string($this->whenType)) {
$whenType = $this->whenType;
} else {
$whenType = null;
}
$binder->bind($placeholder, $when, $whenType);
$when = $placeholder;
}
$then = $this->compileNullableValue($binder, $this->then, $this->thenType);
return "WHEN $when THEN $then";
}
/**
* @inheritDoc
*/
public function traverse(Closure $callback)
{
if ($this->when instanceof ExpressionInterface) {
$callback($this->when);
$this->when->traverse($callback);
}
if ($this->then instanceof ExpressionInterface) {
$callback($this->then);
$this->then->traverse($callback);
}
return $this;
}
/**
* Clones the inner expression objects.
*
* @return void
*/
public function __clone()
{
if ($this->when instanceof ExpressionInterface) {
$this->when = clone $this->when;
}
if ($this->then instanceof ExpressionInterface) {
$this->then = clone $this->then;
}
}
}