Kir\MySQL\Builder\Traits\OrderByBuilder

Complexity

Total Complexity

13

Size/Duplication

Total Lines	80
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	orderBy()	0	10	3
A	orderByValues()	0	8	2
A	buildOrder()	0	12	3
A	addOrder()	0	14	3
A	fixDirection()	0	3	2

<?php
namespace Kir\MySQL\Builder\Traits;

use Kir\MySQL\Builder\Expr\OrderBySpecification;

trait OrderByBuilder {
	use AbstractDB;

	/** @var array */
	private $orderBy = array();

	/**
	 * @param string|OrderBySpecification $expression
	 * @param string $direction
	 * @return $this
	 */
	public function orderBy($expression, $direction = 'asc') {
		if($expression instanceof OrderBySpecification) {
			foreach($expression->getFields() as $field) {
				$this->addOrder($field[0], $field[1]);
			}
			return $this;
		}
		$this->addOrder($expression, $direction);
		return $this;
	}
	
	/**
	 * @param string $fieldName
	 * @param array $values
	 * @return $this
	 */
	public function orderByValues($fieldName, array $values) {
		$expr = [];
		foreach(array_values($values) as $idx => $value) {
			$expr[] = $this->db()->quoteExpression("WHEN ? THEN ?", array($value, $idx));
		}
		$this->orderBy[] = array(sprintf("CASE %s\n\t\t%s\n\tEND", $this->db()->quoteField($fieldName), join("\n\t\t", $expr)), 'ASC');
		return $this;
	}

	/**
	 * @param string $query
	 * @return string
	 */
	protected function buildOrder($query) {
		if(!count($this->orderBy)) {
			return $query;
		}
		$query .= "ORDER BY\n";
		$arr = array();
		foreach($this->orderBy as $order) {
			list($expression, $direction) = $order;
			$arr[] = sprintf("\t%s %s", $expression, strtoupper($direction));
		}
		return $query.join(",\n", $arr)."\n";
	}
	
	/**
	 * @param string $expression
	 * @param string $direction
	 */
	private function addOrder($expression, $direction) {
		$direction = $this->fixDirection($direction);
		if(is_array($expression)) {
			if(!count($expression)) {
				return;
			}
			$arguments = array(
				$expression[0],
				array_slice($expression, 1)
			);
			$expression = call_user_func_array(array($this->db(), 'quoteExpression'), $arguments);
		}
		$this->orderBy[] = array($expression, $direction);
	}
	
	/**
	 * @param string $direction
	 * @return string
	 */
	private function fixDirection($direction) {
		return strtoupper($direction) !== 'ASC' ? 'DESC' : 'ASC';
	}
}