Kir\MySQL\Builder\Insert

Complexity

Total Complexity

39

Size/Duplication

Total Lines	297
Duplicated Lines	0 %

Coupling/Cohesion

Components	2
Dependencies	5

Importance

Changes	19
Bugs	3	Features	6

20 Methods

Rating	Name	Duplication	Size	Complexity
A	into()	0	4	1
A	setIgnore()	0	4	1
A	setKey()	0	4	1
A	add()	0	4	1
A	update()	0	4	1
A	addOrUpdate()	0	5	1
A	addExpr()	0	4	1
A	updateExpr()	0	4	1
A	addOrUpdateExpr()	0	5	1
A	addAll()	0	6	1
A	updateAll()	0	8	2
A	addOrUpdateAll()	0	5	1
A	from()	0	4	1
B	__toString()	0	33	6
A	addTo()	0	9	2
B	addAllTo()	0	16	6
A	buildUpdate()	0	14	3
A	isFieldNameValid()	0	3	2
A	clearValues()	0	17	4
A	formatExtraArgs()	0	7	2

<?php
namespace Kir\MySQL\Builder;

use Kir\MySQL\Tools\AliasReplacer;
use UnexpectedValueException;

class Insert extends InsertUpdateStatement {
	/** @var array */
	private $fields = array();
	/** @var array */
	private $update = array();
	/** @var string */
	private $table = null;
	/** @var string */
	private $keyField = null;
	/** @var bool */
	private $ignore = false;
	/** @var Select */
	private $from = null;

	/**
	 * @param string $table
	 * @return $this
	 */
	public function into($table) {
		$this->table = $table;
		return $this;
	}

	/**
	 * @param bool $value
	 * @return $this
	 */
	public function setIgnore($value = true) {
		$this->ignore = $value;
		return $this;
	}

	/**
	 * Legt den Primaerschluessel fest.
	 * Wenn bei einem Insert der Primaerschluessel mitgegeben wird, dann wird dieser statt der LastInsertId zurueckgegeben
	 *
	 * @param string $field
	 * @return $this
	 */
	public function setKey($field) {
		$this->keyField = $field;
		return $this;
	}

	/**
	 * @param string $field
	 * @param bool|int|float|string $value
	 * @throws UnexpectedValueException
	 * @return $this
	 */
	public function add($field, $value) {
		$this->fields = $this->addTo($this->fields, $field, $value);
		return $this;
	}

	/**
	 * @param string $field
	 * @param bool|int|float|string $value
	 * @throws UnexpectedValueException
	 * @return $this
	 */
	public function update($field, $value) {
		$this->update = $this->addTo($this->update, $field, $value);
		return $this;
	}

	/**
	 * @param string $field
	 * @param bool|int|float|string $value
	 * @throws UnexpectedValueException
	 * @return $this
	 */
	public function addOrUpdate($field, $value) {
		$this->add($field, $value);
		$this->update($field, $value);
		return $this;
	}

	/**
	 * @param string $str
	 * @param string ...$params
	 * @return $this
	 */
	public function addExpr($str) {

The parameter $str is not used and could be removed.

		$this->fields[] = func_get_args();
		return $this;
	}

	/**
	 * @param string $str
	 * @param string ...$params
	 * @return $this
	 */
	public function updateExpr($str) {

The parameter $str is not used and could be removed.

		$this->update[] = func_get_args();
		return $this;
	}

	/**
	 * @param string $str
	 * @return $this
	 */
	public function addOrUpdateExpr($str) {

The parameter $str is not used and could be removed.

		$this->fields[] = func_get_args();
		$this->update[] = func_get_args();
		return $this;
	}

	/**
	 * @param array $data
	 * @param array $mask
	 * @param array $excludeFields
	 * @return $this
	 */
	public function addAll(array $data, array $mask = null, array $excludeFields = null) {
		$this->addAllTo($data, $mask, $excludeFields, function ($field, $value) {
			$this->add($field, $value);
		});
		return $this;
	}

	/**
	 * @param array $data
	 * @param array $mask
	 * @param array $excludeFields
	 * @return $this
	 */
	public function updateAll(array $data, array $mask = null, array $excludeFields = null) {
		$this->addAllTo($data, $mask, $excludeFields, function ($field, $value) {
			if ($field !== $this->keyField) {
				$this->update($field, $value);
			}
		});
		return $this;
	}

	/**
	 * @param array $data
	 * @param array $mask
	 * @param array $excludeFields
	 * @return $this
	 */
	public function addOrUpdateAll(array $data, array $mask = null, array $excludeFields = null) {
		$this->addAll($data, $mask, $excludeFields);
		$this->updateAll($data, $mask, $excludeFields);
		return $this;
	}

	/**
	 * @param Select $select
	 * @return $this
	 */
	public function from(Select $select) {
		$this->from = $select;
		return $this;
	}

	/**
	 * @throws Exception
	 * @return string
	 */
	public function __toString() {
		if ($this->table === null) {
			throw new Exception('Specify a table-name');
		}

		$tableName = (new AliasReplacer($this->db()->getAliasRegistry()))->replace($this->table);

		$queryArr = array();
		$ignoreStr = $this->ignore ? ' IGNORE' : '';
		$queryArr[] = "INSERT{$ignoreStr} INTO\n\t{$tableName}\n";

		if($this->from !== null) {
			$fields = $this->from->getFields();
			$queryArr[] = sprintf("\t(%s)\n", join(', ', array_keys($fields)));
			$queryArr[] = $this->from;
		} else {
			$fields = $this->fields;
			$insertData = $this->buildFieldList($fields);
			if (!count($insertData)) {
				throw new Exception('No field-data found');
			}
			$queryArr[] = sprintf("SET\n%s\n", join(",\n", $insertData));
		}

		$updateData = $this->buildUpdate();
		if($updateData) {
			$queryArr[] = "{$updateData}\n";
		}

		$query = join('', $queryArr);

		return $query;
	}

	/**
	 * @param array $fields
	 * @param string $field
	 * @param bool|int|float|string $value
	 * @return array
	 */
	private function addTo(array $fields, $field, $value) {
		if ($this->isFieldNameValid($field)) {
			throw new UnexpectedValueException('Field name is invalid');
		}
		$sqlField = $field;
		$sqlValue = $this->db()->quote($value);
		$fields[$sqlField] = $sqlValue;
		return $fields;
	}

	/**
	 * @param array $data
	 * @param array $mask
	 * @param array $excludeFields
	 * @param callable $fn
	 * @return $this
	 */
	private function addAllTo(array $data, array $mask = null, array $excludeFields = null, $fn) {
		if($mask !== null) {
			$data = array_intersect_key($data, array_combine($mask, $mask));
		}
		if($excludeFields !== null) {
			foreach($excludeFields as $excludeField) {
				if(array_key_exists($excludeField, $data)) {
					unset($data[$excludeField]);
				}
			}
		}
		$data = $this->clearValues($data);
		foreach ($data as $field => $value) {
			call_user_func($fn, $field, $value);
		}
	}

	/**
	 * @return string
	 */
	private function buildUpdate() {
		$queryArr = array();
		if(!empty($this->update)) {
			$queryArr[] = "ON DUPLICATE KEY UPDATE\n";
			$updateArr = array();
			if($this->keyField !== null) {
				$updateArr[] = "\t`{$this->keyField}` = LAST_INSERT_ID({$this->keyField})";
			}
			$updateArr = $this->buildFieldList($this->update, $updateArr);

			$queryArr[] = join(",\n", $updateArr);
		}
		return join('', $queryArr);
	}

	/**
	 * @param string $fieldName
	 * @return bool
	 */
	private function isFieldNameValid($fieldName) {
		return is_numeric($fieldName) || !is_scalar($fieldName);
	}

	/**
	 * @param array $values
	 * @return array
	 * @throws Exception
	 */
	private function clearValues(array $values) {
		if(!count($values)) {
			return [];
		}

		$tableName = (new AliasReplacer($this->db()->getAliasRegistry()))->replace($this->table);
		$fields = $this->db()->getTableFields($tableName);
		$result = array();

		foreach ($values as $fieldName => $fieldValue) {
			if(in_array($fieldName, $fields)) {
				$result[$fieldName] = $fieldValue;
			}
		}

		return $result;
	}

	/**
	 * @param string $expression
	 * @param array $args
	 * @return string
	 */
	private function formatExtraArgs($expression, $args) {
		if(count($args) > 1) {
			$args = array_slice($args, 1);
			$expression = $this->db()->quoteExpression($expression, $args);
		}
		return $expression;
	}
}