Happyr\DoctrineSpecification\Operand\PlatformFunction

Complexity

Total Complexity

8

Size/Duplication

Total Lines	91
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	11	2
A	transform()	0	17	6

<?php

namespace Happyr\DoctrineSpecification\Operand;

use Doctrine\ORM\QueryBuilder;
use Happyr\DoctrineSpecification\Exception\InvalidArgumentException;

class PlatformFunction implements Operand
{
    /**
     * Doctrine internal functions.
     *
     * @see \Doctrine\ORM\Query\Parser
     *
     * @var string[]
     */
    private static $doctrineFunctions = [
        // String functions
        'concat',
        'substring',
        'trim',
        'lower',
        'upper',
        'identity',
        // Numeric functions
        'length',
        'locate',
        'abs',
        'sqrt',
        'mod',
        'size',
        'date_diff',
        'bit_and',
        'bit_or',
        // Aggregate functions
        'min',
        'max',
        'avg',
        'sum',
        'count',
        // Datetime functions
        'current_date',
        'current_time',
        'current_timestamp',
        'date_add',
        'date_sub',
    ];

    /**
     * @var string
     */
    private $functionName = '';

    /**
     * @var array
     */
    private $arguments = [];

    /**
     * @param string $functionName
     * @param mixed  $arguments
     */
    public function __construct($functionName, $arguments = [])
    {
        if (2 === func_num_args()) {
            $this->functionName = $functionName;
            $this->arguments = (array) $arguments;
        } else {
            $arguments = func_get_args();
            $this->functionName = array_shift($arguments);
            $this->arguments = $arguments;
        }
    }

    /**
     * @param QueryBuilder $qb
     * @param string       $dqlAlias
     *
     * @return string
     */
    public function transform(QueryBuilder $qb, $dqlAlias)
    {
        if (!in_array(strtolower($this->functionName), self::$doctrineFunctions) &&
            !$qb->getEntityManager()->getConfiguration()->getCustomStringFunction($this->functionName) &&
            !$qb->getEntityManager()->getConfiguration()->getCustomNumericFunction($this->functionName) &&
            !$qb->getEntityManager()->getConfiguration()->getCustomDatetimeFunction($this->functionName)
        ) {
            throw new InvalidArgumentException(sprintf('"%s" is not a valid function name.', $this->functionName));
        }

        $arguments = ArgumentToOperandConverter::convert($this->arguments);
        foreach ($arguments as $key => $argument) {
            $arguments[$key] = $argument->transform($qb, $dqlAlias);
        }

        return sprintf('%s(%s)', $this->functionName, implode(', ', $arguments));
    }
}