<?php

namespace Leenooks\Traits;

use Illuminate\Database\Eloquent\Builder;

/**
 * Trait CompositeKeys
 * Enable Models to have multiple primary keys
 *
 * Need to add to the Model...
 * 	public $incrementing = false;
 *  protected $primaryKey = [<ARRAY_OF_KEYS>];
 *
 * @package Leenooks\Traits
 */
trait CompositeKeys {
	/**
	 * Set the keys for a save update query.
	 *
	 * @param  \Illuminate\Database\Eloquent\Builder  $query
	 * @return \Illuminate\Database\Eloquent\Builder
	 */
	protected function setKeysForSaveQuery(Builder $query)
	{
		$keys = $this->getKeyName();

		if (! is_array($keys)) {
			return parent::setKeysForSaveQuery($query);
		}

		foreach($keys as $keyName) {
			$query->where($keyName, '=', $this->getKeyForSaveQuery($keyName));
		}

		return $query;
	}

	/**
	 * Get the primary key value for a save query.
	 *
	 * @param mixed $keyName
	 * @return mixed
	 */
	protected function getKeyForSaveQuery($keyName = null)
	{
		if (is_null($keyName)) {
			$keyName = $this->getKeyName();
		}

		if (isset($this->original[$keyName])) {
			return $this->original[$keyName];
		}

		return $this->getAttribute($keyName);
	}
}