<?php


namespace Database;


use Database\Traits\Builder;
use Exception;
use JetBrains\PhpStorm\Pure;
use Snowflake\Abstracts\Component;


/**
 * Class SqlBuilder
 * @package Database
 */
class SqlBuilder extends Component
{

	use Builder;


	public ActiveQuery|Sql|null $query;


	/**
	 * @param $query
	 * @return $this
	 */
	public static function builder(ISqlBuilder|null $query): static
	{
		return new static(['query' => $query]);
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function getCondition(): string
	{
		return $this->conditionToString();
	}


	/**
	 * @param array $attributes
	 * @return bool|array
	 * @throws Exception
	 */
	public function update(array $attributes): bool|array
	{
		[$string, $array] = $this->builderParams($attributes);
		if (empty($string) || empty($array)) {
			return $this->addError('None data update.');
		}

		$condition = $this->conditionToString();
		if (!empty($condition)) {
			$condition = ' WHERE ' . $condition;
		}

		$update = 'UPDATE ' . $this->tableName() . ' SET ' . implode(',', $string) . $condition;
		$update .= $this->builderLimit($this->query);

		return [$update, $array];
	}


	/**
	 * @param array $attributes
	 * @param string $opera
	 * @return bool|array
	 * @throws Exception
	 */
	public function mathematics(array $attributes, $opera = '+'): bool|array
	{
		$string = $array = [];

		foreach ($attributes as $attribute => $value) {
			$string[] = $attribute . '=' . $attribute . $opera . $value;
		}

		if (empty($string)) {
			return $this->addError('None data update.');
		}

		$condition = $this->conditionToString();
		if (!empty($condition)) {
			$condition = ' WHERE ' . $condition;
		}

		$update = 'UPDATE ' . $this->tableName() . ' SET ' . implode(',', $string) . $condition;
		$update .= $this->builderLimit($this->query);

		return [$update, []];
	}


	/**
	 * @param array $attributes
	 * @param false $isBatch
	 * @return array
	 * @throws Exception
	 */
	public function insert(array $attributes, $isBatch = false): array
	{
		$update = sprintf('INSERT INTO %s', $this->tableName());
		if ($isBatch === false) {
			$attributes = [$attributes];
		}
		$update .= '(' . implode(',', $this->getFields($attributes)) . ') VALUES ';

		$order = 0;
		$keys = $params = [];
		foreach ($attributes as $attribute) {
			[$_keys, $params] = $this->builderParams($attribute, true, $params, $order);

			$keys[] = implode(',', $_keys);
			$order++;
		}
		return [$update . '(' . implode('),(', $keys) . ')', $params];
	}


	/**
	 * @param $attributes
	 * @return array
	 */
	#[Pure] private function getFields($attributes): array
	{
		return array_keys(current($attributes));
	}


	/**
	 * @param array $attributes
	 * @param bool $isInsert
	 * @param array $params
	 * @param int $order
	 * @return array[]
	 * a=:b,
	 */
	private function builderParams(array $attributes, bool $isInsert = false, $params = [], $order = 0): array
	{
		$keys = [];
		foreach ($attributes as $key => $value) {
			if ($isInsert === true) {
				$keys[] = ':' . $key . $order;
			} else {
				$keys[] = $key . '=:' . $key . $order;
			}
			$params[$key . $order] = $value;
		}
		return [$keys, $params];
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function one(): string
	{
		$this->query->limit(0, 1);
		return $this->_prefix(true);
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function all(): string
	{
		return $this->_prefix(true);
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function count(): string
	{
		return $this->_prefix();
	}


	/**
	 * @param $table
	 * @return string
	 */
	public function columns($table): string
	{
		return 'SHOW FULL FIELDS FROM ' . $table;
	}


	/**
	 * @param bool $hasOrder
	 * @return string
	 * @throws Exception
	 */
	private function _prefix($hasOrder = false): string
	{
		$select = $this->builderSelect($this->query->select) . ' FROM ' . $this->tableName();
		if (!empty($this->query->alias)) {
			$select .= $this->builderAlias($this->query->alias);
		}
		if (!empty($this->query->join)) {
			$select .= $this->builderJoin($this->query->join);
		}
		if (!empty($condition = $this->conditionToString())) {
			$select = sprintf('%s WHERE %s', $select, $condition);
		}
		if (!empty($this->query->group)) {
			$select .= $this->builderGroup($this->query->group);
		}
		if ($hasOrder === true && !empty($this->query->order)) {
			$select .= $this->builderOrder($this->query->order);
		}
		return $select . $this->builderLimit($this->query);
	}


	/**
	 * @param false $isCount
	 * @return string
	 * @throws Exception
	 */
	public function get($isCount = false): string
	{
		if ($isCount === false) {
			return $this->all();
		}
		return $this->count();
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function truncate(): string
	{
		return sprintf('TRUNCATE %s', $this->tableName());
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	private function conditionToString(): string
	{
		return $this->where($this->query->where);
	}


	/**
	 * @return string
	 * @throws Exception
	 */
	public function tableName(): string
	{
		if ($this->query->from instanceof \Closure) {
			call_user_func($this->query->from, $activeQuery = new Sql());
			$this->query->from = '(' . $activeQuery->getSql() . ')';
		}
		if ($this->query->from instanceof ActiveQuery) {
			$this->query->from = '(' . SqlBuilder::builder($this->query->from)->get($this->query->from) . ')';
		}
		if (empty($this->query->from)) {
			return $this->query->modelClass::getTable();
		}
		return $this->query->from;
	}

}