<?php
/**
 * Created by PhpStorm.
 * User: whwyy
 * Date: 2018/3/30 0030
 * Time: 14:56
 */
declare(strict_types=1);

namespace Database\Traits;


use Closure;
use Database\Base\ActiveQueryInterface;
use Database\Command;
use Database\ISqlBuilder;
use Database\ModelInterface;
use Database\Query;
use Database\SqlBuilder;
use Exception;
use Kiri;
use Kiri\Abstracts\Component;

/**
 * Trait QueryTrait
 * @package Database\Traits
 */
abstract class QueryTrait extends Component implements ActiveQueryInterface, ISqlBuilder
{
    protected array      $where  = [];
    protected array      $select = ['*'];
    protected array      $join   = [];
    protected array      $order  = [];
    protected int        $offset = -1;
    protected int        $limit  = -1;
    protected string     $group  = '';
    protected string     $from   = '';
    protected string     $alias  = 't1';
    protected array      $filter = [];
    protected bool       $lock   = false;
    protected SqlBuilder $builder;
    protected array      $params = [];
    protected array      $_alias = ['t1'];


    /**
     * @var ModelInterface|string|null
     */
    protected ModelInterface|string|null $modelClass;


    /**
     * Comply constructor.
     * @throws
     */
    public function __construct($model = null)
    {
        if (!is_null($model)) {
            $this->modelClass = $model;
        }
        $this->builder = SqlBuilder::builder($this);
        parent::__construct();
    }


    /**
     * @param string $name
     * @return mixed
     * @throws Exception
     */
    public function __get(string $name)
    {
        if (property_exists($this, $name)) {
            return $this->$name;
        }
        return parent::__get($name); // TODO: Change the autogenerated stub
    }


    /**
     * @param string $column
     * @param callable $callable
     * @return $this
     */
    public function when(string $column, callable $callable): static
    {
        $caseWhen = new When($column, $this);

        call_user_func($callable, $caseWhen);

        $this->where[] = $caseWhen->end();

        return $this;
    }


    /**
     * @param bool $lock
     * @return $this
     */
    public function lock(bool $lock): static
    {
        $this->lock = $lock;
        return $this;
    }


    /**
     * @param string $whereRaw
     * @return QueryTrait
     */
    public function whereRaw(string $whereRaw): static
    {
        if ($whereRaw == '') {
            return $this;
        }
        $this->where[] = $whereRaw;
        return $this;
    }


    /**
     * @return string
     * @throws
     */
    public function getTable(): string
    {
        return $this->modelClass->getTable();
    }


    /**
     * @param string $column
     * @param string $value
     * @return $this
     */
    public function whereLocate(string $column, string $value): static
    {
        $this->where[] = 'LOCATE(' . $column . ',\'' . addslashes($value) . '\') > 0';
        return $this;
    }


    /**
     * @param string $column
     * @return $this
     */
    public function whereNull(string $column): static
    {
        $this->where[] = $column . ' IS NULL';
        return $this;
    }


    /**
     * @param string $column
     * @return $this
     */
    public function whereEmpty(string $column): static
    {
        $this->where[] = $column . ' = \'\'';
        return $this;
    }


    /**
     * @param string $column
     * @return $this
     */
    public function whereNotEmpty(string $column): static
    {
        $this->where[] = $column . ' <> \'\'';
        return $this;
    }


    /**
     * @param string $column
     * @return $this
     */
    public function whereNotNull(string $column): static
    {
        $this->where[] = $column . ' IS NOT NULL';
        return $this;
    }


    /**
     * @param string $alias
     *
     * @return $this
     *
     * select * from tableName as t1
     */
    public function alias(string $alias = 't1'): static
    {
        $this->alias  = $alias;
        $this->_alias = [$alias];
        return $this;
    }


    /**
     * @param string|Closure $tableName
     *
     * @return $this
     */
    public function from(string|Closure $tableName): static
    {
        if ($tableName instanceof Closure) {
            $tableName = call_user_func($tableName, $this->queryInstance());
        }
        $this->from = $tableName;
        return $this;
    }


    /**
     * @param string $tableName
     * @param string $alias
     * @param array $on
     * @param array $param
     * @return $this
     * $query->join([$tableName, ['userId'=>'uuvOd']], $param)
     * $query->join([$tableName, ['userId'=>'uuvOd'], $param])
     * $query->join($tableName, ['userId'=>'uuvOd',$param])
     */
    private function join(string $tableName, string $alias, array $on, array $param = []): static
    {
        if (empty($on)) {
            return $this;
        }

        $this->_alias[] = $alias;

        $join[] = $tableName . ' AS ' . $alias;
        $join[] = 'ON ' . $this->onCondition($on);
        if (empty($join)) {
            return $this;
        }

        $this->join[] = implode(' ', $join);
        if (!empty($param)) {
            $this->bindParams($param);
        }

        return $this;
    }


    /**
     * @param array $params
     * @return void
     */
    public function bindParams(array $params): void
    {
        foreach ($params as $param) {
            $this->pushParam($param);
        }
    }


    /**
     * @param array $condition
     * @return string
     */
    private function onCondition(array $condition): string
    {
        $array = [];
        foreach ($condition as $key => $item) {
            if (is_numeric($item) || !$this->isAliasField($item)) {
                $array[] = $key . '= ?';
                $this->pushParam($item);
            } else {
                $array[] = $key . '=' . $item;
            }
        }
        return implode(' AND ', $array);
    }


    /**
     * @param string $value
     * @return bool
     */
    private function isAliasField(string $value): bool
    {
        foreach ($this->_alias as $alias) {
            if (str_starts_with($value, $alias . '.')) {
                return true;
            }
        }
        return false;
    }


    /**
     * @param string $tableName
     * @param string $alias
     * @param array $onCondition
     * @param array $param
     * @return $this
     * @throws Exception
     */
    public function leftJoin(string $tableName, string $alias, array $onCondition, array $param = []): static
    {
        if (class_exists($tableName)) {
            $model = Kiri::getDi()->get($tableName);
            if (!($model instanceof ModelInterface)) {
                throw new Exception('Model must implement ' . ModelInterface::class);
            }
            $tableName = $model->getTable();
        }
        return $this->join("LEFT JOIN " . $tableName, $alias, $onCondition, $param);
    }


    /**
     * @param string $tableName
     * @param string $alias
     * @param array $onCondition
     * @param array $param
     * @return $this
     * @throws Exception
     */
    public function rightJoin(string $tableName, string $alias, array $onCondition, array $param = []): static
    {
        if (class_exists($tableName)) {
            $model = Kiri::getDi()->get($tableName);
            if (!($model instanceof ModelInterface)) {
                throw new Exception('Model must implement ' . ModelInterface::class);
            }
            $tableName = $model->getTable();
        }
        return $this->join("RIGHT JOIN " . $tableName, $alias, $onCondition, $param);
    }


    /**
     * @param string $tableName
     * @param string $alias
     * @param array $onCondition
     * @param array $param
     * @return $this
     * @throws Exception
     */
    public function innerJoin(string $tableName, string $alias, array $onCondition, array $param = []): static
    {
        if (class_exists($tableName)) {
            $model = Kiri::getDi()->get($tableName);
            if (!($model instanceof ModelInterface)) {
                throw new Exception('Model must implement ' . ModelInterface::class);
            }
            $tableName = $model->getTable();
        }
        return $this->join("INNER JOIN " . $tableName, $alias, $onCondition, $param);
    }


    /**
     * @param string $field
     *
     * @return $this
     */
    public function sum(string $field): static
    {
        $this->select[] = 'SUM(' . $field . ') AS ' . $field;
        return $this;
    }


    /**
     * @param string $field
     * @return $this
     */
    public function max(string $field): static
    {
        $this->select[] = 'MAX(' . $field . ') AS ' . $field;
        return $this;
    }


    /**
     * @param string $lngField
     * @param string $latField
     * @param int|float $lng1
     * @param int|float $lat1
     *
     * @return $this
     */
    public function distance(string $lngField, string $latField, int|float $lng1, int|float $lat1): static
    {
        $sql            = "ROUND(6378.138 * 2 * ASIN(SQRT(POW(SIN(($lat1 * PI() / 180 - $lat1 * PI() / 180) / 2),2) + COS($lat1 * PI() / 180) * COS($latField * PI() / 180) * POW(SIN(($lng1 * PI() / 180 - $lngField * PI() / 180) / 2),2))) * 1000) AS distance";
        $this->select[] = $sql;
        return $this;
    }


    /**
     * @param string|array $column
     * @param string $sort
     *
     * @return $this
     *
     * [
     *     'addTime',
     *     'descTime desc'
     * ]
     */
    public function orderBy(string|array $column, string $sort = 'DESC'): static
    {
        if (empty($column)) {
            return $this;
        }
        if (is_string($column)) {
            return $this->addOrder(...func_get_args());
        }

        foreach ($column as $val) {
            $this->addOrder($val);
        }

        return $this;
    }


    /**
     * @param string $column
     * @param string $sort
     *
     * @return $this
     *
     */
    private function addOrder(string $column, string $sort = 'DESC'): static
    {
        if (func_num_args() == 1 || str_contains($column, ' ')) {
            $this->order[] = $column;
        } else {
            $this->order[] = "$column $sort";
        }
        return $this;
    }


    /**
     * @param array $column
     *
     * @return $this
     */
    public function select(array $column = ['*']): static
    {
        $this->select = $column;
        return $this;
    }


    /**
     * @return $this
     */
    public function rand(): static
    {
        $this->order[] = 'RAND()';
        return $this;
    }


    /**
     * @param array|Closure|string $conditionArray
     * @param string $opera
     * @param null $value
     * @return QueryTrait
     * @throws
     */
    public function whereOr(array|Closure|string $conditionArray = [], string $opera = '=', $value = null): static
    {
        if ($conditionArray instanceof Closure) {
            $conditionArray = $this->makeClosureFunction($conditionArray);
        }

        if (func_num_args() > 1) {
            [$conditionArray, $opera, $value] = $this->opera(...func_get_args());

            $conditionArray = $this->sprintf($conditionArray, $value, $opera);
        }

        $this->where = ['((' . implode(' AND ', $this->where) . ') OR (' . $conditionArray . '))'];
        return $this;
    }


    /**
     * @param string $column
     * @param string $value
     * @return $this
     */
    public function whereLike(string $column, string $value): static
    {
        $this->pushParam($value);
        $this->where[] = $column . ' LIKE \'%?%\'';
        return $this;
    }


    /**
     * @param string $column
     * @param string $value
     * @return $this
     */
    public function whereLeftLike(string $column, string $value): static
    {
        $this->pushParam($value);
        $this->where[] = $column . ' LLike \'%?\'';
        return $this;
    }


    /**
     * @param string $column
     * @param string $value
     * @return $this
     */
    public function whereRightLike(string $column, string $value): static
    {
        $this->pushParam($value);
        $this->where[] = $column . ' RLike \'?%\'';
        return $this;
    }


    /**
     * @param string $column
     * @param string $value
     * @return $this
     */
    public function whereNotLike(string $column, string $value): static
    {
        $this->pushParam($value);
        $this->where[] = $column . ' NOT LIKE \'%?%\'';
        return $this;
    }


    /**
     * @param string $columns
     * @param array|Closure $value
     * @return $this
     * @throws
     */
    public function whereIn(string $columns, array|Closure $value): static
    {
        if ($value instanceof Closure) {
            $value = $this->makeClosureFunction($value);
        }
        if (count($value) < 1) {
            $value = [-1];
        }
        $this->where[] = $columns . ' IN (' . implode(',', $value) . ')';
        return $this;
    }


    /**
     * @return Query
     * @throws
     */
    public function queryInstance(): Query
    {
        return new Query();
    }


    /**
     * @param string $columns
     * @param array $value
     * @return $this
     */
    public function whereNotIn(string $columns, array $value): static
    {
        if (count($value) < 1) {
            $value = [-1];
        }
        $this->where[] = $columns . ' NOT IN (' . implode(',', $value) . ')';
        return $this;
    }


    /**
     * @param string $column
     * @param int|float $start
     * @param int|float $end
     * @return $this
     */
    public function whereBetween(string $column, int|float $start, int|float $end): static
    {
        if (empty($column) || empty($start) || empty($end)) {
            return $this;
        }

        $this->pushParam($start);
        $this->pushParam($end);
        $this->where[] = $column . ' BETWEEN ? AND ?';

        return $this;
    }


    /**
     * @param string $column
     * @param int|float $start
     * @param int|float $end
     * @return $this
     */
    public function whereNotBetween(string $column, int|float $start, int|float $end): static
    {
        if (empty($column) || empty($start) || empty($end)) {
            return $this;
        }

        $this->pushParam($start);
        $this->pushParam($end);
        $this->where[] = $column . ' NOT BETWEEN ? AND ?';

        return $this;
    }


    /**
     * @param mixed $value
     * @return $this
     */
    public function pushParam(mixed $value): static
    {
        $this->params[] = $value;
        return $this;
    }


    /**
     * @param array|string $column
     * @param mixed $value
     * @return $this
     */
    public function where(array|string $column, mixed $value = null): static
    {
        if (func_num_args() === 2) {
            return $this->addArray([$column => $value]);
        }
        if (is_string($column)) {
            return $this->whereRaw($column);
        } else {
            return $this->addArray($column);
        }
    }


    /**
     * @param string $column
     * @param string $opera
     * @param mixed $value
     * @return $this
     */
    public function whereMath(string $column, string $opera, mixed $value): static
    {
        $this->pushParam($value);
        $this->where[] = $column . ' ' . $opera . ' ?';
        return $this;
    }


    /**
     * @param Closure $closure
     * @return $this
     */
    public function whereClosure(Closure $closure): static
    {
        $this->where[] = $this->makeClosureFunction($closure);
        return $this;
    }


    /**
     * @param Closure|array $closure
     * @return string
     * @throws
     */
    public function makeClosureFunction(Closure|array $closure): string
    {
        $generate = $this->queryInstance();
        if ($closure instanceof Closure) {
            call_user_func($closure, $generate);
        } else {
            $generate->addArray($closure);
        }
        return $generate->build();
    }


    /**
     * @param string $name
     * @param string|null $having
     *
     * @return $this
     */
    public function groupBy(string $name, string $having = NULL): static
    {
        $this->group = $name;
        if (empty($having)) {
            return $this;
        }
        $this->group .= ' HAVING ' . $having;
        return $this;
    }


    /**
     * @param int $limit
     *
     * @return $this
     */
    public function limit(int $limit = 20): static
    {
        $this->limit = $limit;
        return $this;
    }


    /**
     * @param int $offset
     * @return $this
     */
    public function offset(int $offset): static
    {
        $this->offset = $offset;
        return $this;
    }


    /**
     * @return array
     */
    private function opera(): array
    {
        if (func_num_args() == 3) {
            [$column, $opera, $value] = func_get_args();
        } else {
            [$column, $value] = func_get_args();
        }
        if (!isset($opera)) {
            $opera = '=';
        }
        return [$column, $opera, $value];
    }


    /**
     * @param array $array
     * @return $this
     */
    private function addArray(array $array): static
    {
        foreach ($array as $key => $value) {
            if ($value === null) {
                continue;
            }
            $this->where[] = is_numeric($key) ? $value : $this->sprintf($key, $value);
        }
        return $this;
    }


    /**
     * @param string $column
     * @param mixed $value
     * @param string $opera
     * @return string
     */
    private function sprintf(string $column, mixed $value, string $opera = '='): string
    {
        $this->pushParam($value);
        return $column . ' ' . $opera . ' ?';
    }


    /**
     * @param string $querySql
     * @param array $params
     * @return Command
     */
    public function buildCommand(string $querySql, array $params = []): Command
    {
        $connection = $this->modelClass->getConnection();
        if (count($params) > 0) {
            $this->bindParams($params);
        }
        return $connection->createCommand($querySql, $this->params);
    }


    /**
     * @return string
     * @throws
     */
    public function build(): string
    {
        return $this->builder->get();
    }


}