Selectize.php

<?php
/**
 * Copyright (c) 2014 Petr Olišar (http://olisar.eu)
 *
 * For the full copyright and license information, please view
 * the file LICENSE.md that was distributed with this source code.
 */

namespace App\Form\Control;

use Doctrine\ORM\PersistentCollection;
use Kdyby\Doctrine\Collections\ReadOnlyCollectionWrapper;
use Nette;
use Nette\Forms\Form;
use Nette\Utils;

/**
 * Description of Selectize
 *
 * @author Petr Olišar <petr.olisar@gmail.com>
 * @author <brouwer.p@gmail.com>
 * @author Tomáš Hofman <hofman.tomas@gmail.com>
 */
class Selectize extends Nette\Forms\Controls\BaseControl
{
	private $entity = [];
	private $labelName;
	private $selectize;
	private $selectizeBack;
	private $options;
	private $prompt = null;

	public function __construct(?string $label = null, ?array $entity = null, ?array $options = null)
	{
		parent::__construct($label);
		$this->entity = is_null($entity) ? [] : $entity;
		$this->labelName = $label;
		$this->options = $options;
	}

	public function setOptions(array $options)
	{
		foreach($options as $key => $value)
		{
			$this->options[$key] = $value;
		}
		return $this;
	}

	public function setMode($mode)
	{
		$this->options['mode'] = $mode;
		return $this;
	}

	public function setCreate($create)
	{
		$this->options['create'] = $create;
		return $this;
	}

	public function maxItems($items)
	{
		$this->options['maxItems'] = $items;
		return $this;
	}

	public function setDelimiter($delimiter)
	{
		$this->options['delimiter'] = $delimiter;
		return $this;
	}

	public function setPlugins(array $plugins)
	{
		$this->options['plugins'] = $plugins;
		return $this;
	}

	public function setValueField($valueField)
	{
		$this->options['valueField'] = $valueField;
		return $this;
	}

	public function setLabelField($labelField)
	{
		$this->options['labelField'] = $labelField;
		return $this;
	}

	public function setSearchField($searchField)
	{
		$this->options['searchField'] = $searchField;
		return $this;
	}

	public function setClass($class)
	{
		$this->options['class'] = $class;
		return $this;
	}

	public function setAjaxURL($ajaxURL)
	{
		$this->options['ajaxURL'] = $ajaxURL;
		return $this;
	}

	/**
	 * Sets first prompt item in select box.
	 * @param  string
	 * @return self
	 */
	public function setPrompt($prompt)
	{
		$this->prompt = $prompt;
		return $this;
	}

	/**
	 * Returns first prompt item?
	 * @return mixed
	 */
	public function getPrompt()
	{
		return $this->prompt;
	}

    /**
     * Sets options and option groups from which to choose.
     * @param array $items
     * @return array
     */
    public function setItems(array $items)
    {
        return $this->entity = $items;
    }

	/**
	* Gets items
	* @return array
	*/
	public function getItems() {
        return $this->entity;
	}

	public function setValue($value)
	{
		if(!is_null($value))
		{
			if ($value instanceof Nette\Database\Table\Selection)
			{
				throw new Nette\InvalidArgumentException("Type must be array, instance of Nette\\Database\\Table\\Selection was given. Try Selection::fetchAssoc(\$key)");
			}

			if(is_array($value) || $value instanceof ReadOnlyCollectionWrapper || $value instanceof PersistentCollection)
			{
				$i = 0;
				foreach($value as $key => $slug)
				{
					$i++;
					$idName = $this->options['valueField'];
					$this->selectizeBack .= isset($slug->$idName) ? $slug->$idName : $key;

					if($i < count($value))
					{
						$this->selectizeBack .= $this->options['delimiter'];
					}
				}
			} else
			{
				$this->selectizeBack = $value;
			}
		}

		$this->selectize = $this->selectizeBack;
	}

	public function getValue()
	{
		if (is_array($this->selectize) && count($this->selectize) === 0) {
			return null;
		}
		return $this->selectize;
	}

	public function loadHttpData(): void
	{
		if($this->options['mode'] === 'select')
		{
			$value = $this->getHttpData(Form::DATA_LINE);
			if($value === "")
			{
				$value = NULL;
			}
			$this->selectizeBack = $this->selectize = $value;
		} else
		{
			$this->prepareData();
		}
	}

	public function getControl(): Utils\Html
	{
		$this->setOption('rendered', true);
		$name = $this->getHtmlName();
        $el = clone $this->control;
        if (array_key_exists('ajaxURL', $this->options))
        {
            $this->entity = $this->findActiveValue($this->entity, $this->options['valueField'], $this->selectizeBack);
        }
        if($this->options['mode'] === 'full')
		{
			return $el->addAttributes([
				'id' => $this->getHtmlId(),
				'type' => 'text',
				'name' => $name,
				'class' => array(isset($this->options['class']) ? $this->options['class'] : 'selectize' . ' form-control text'),
				'data-entity' => $this->entity,
				'data-options' => $this->options,
				'value' => $this->selectizeBack
			]);
		} elseif ($this->options['mode'] === 'select')
		{
			$this->entity = empty($this->prompt) ?
				$this->entity : self::arrayUnshiftAssoc($this->entity, '', $this->translate($this->prompt));
			return Nette\Forms\Helpers::createSelectBox($this->entity, [
					'selected?' => $this->selectizeBack
				])
				->id($this->getHtmlId())
				->name($name)
				->data('entity', $this->entity)
				->data('options', $this->options)
				->class(isset($this->options['class']) ? $this->options['class'] : 'selectize' . ' form-control')
				->addAttributes(parent::getControl()->attrs)
                		->setValue($this->selectizeBack);
		}
	}

	private function findActiveValue($array, $key, $value)
	{
		$results = array();

		if (is_array($array)) {
			if (isset($array[$key]) && $array[$key] == $value) {
				$results[] = $array;
			}

			foreach ($array as $subarray) {
				$results = array_merge($results, $this->findActiveValue($subarray, $key, $value));
			}
		}

		return $results;
	}

	private static function arrayUnshiftAssoc(&$arr, $key, $val)
	{
		$arr = array_reverse($arr, true);
		$arr[$key] = $val;
		return array_reverse($arr, true);
	}

	private function prepareData()
	{
		$this->selectize = $this->split($this->getHttpData(Form::DATA_LINE));
		$this->selectizeBack = $this->getHttpData(Form::DATA_LINE);
		$iteration = false;
		foreach($this->selectize as $key => $value)
		{
			if(!$this->myInArray($this->entity, $value, $this->options['valueField']))
			{
				$iteration ?: $this->selectize['new'] = [];
				array_push($this->entity, [$this->options['valueField'] => $value, 'name' => $value]);
				array_push($this->selectize['new'], $value);
				unset($this->selectize[$key]);
				$iteration = true;
			}
		}
	}

	private function split($selectize)
	{
		$return = Nette\Utils\Strings::split(empty($selectize) ?
			'' : $selectize, '~'.$this->options['delimiter'].'\s*~');
		return $return[0] === "" ? [] : $return;
	}

	/**
	 *
	 * @author <brouwer.p@gmail.com>
	 * @param array $array
	 * @param string|int $value
	 * @param string|int $key
	 * @return boolean
	 */
	private function myInArray(array $array, $value, $key)
	{
		if(isset($array[$key]) AND $array[$key]==$value)
		{
			return true;
		}

		foreach ($array as $val)
		{
			if (is_array($val))
			{
				if($this->myInArray($val,$value,$key))
				{
					return true;
				}
			}
		}
		return false;
	}

	public static function register($method = 'addSelectize', $config)
	{
		Nette\Forms\Container::extensionMethod($method, function(Nette\Forms\Container $container, $name, $label, $entity = null, array $options = null) use ($config)
		{
			$container[$name] = new Selectize($label, $entity, is_array($options) ?
				array_replace($config, $options) : $config);
			return $container[$name];
		});
	}
}