<?php

namespace ADT\Forms;

use Exception;
use Nette;
use Nette\Application\UI;
use Nette\Application\UI\Presenter;
use Traversable;

class DynamicContainer extends BaseContainer
{
	const string NEW_PREFIX = '_new_';
	
	private StaticContainerFactory $staticContainerFactory;
	private bool $allowAdding = true;
	private ?StaticContainer $template = null;
	private int $newCount = 0;

	public function __construct()
	{
		$this->monitor(Presenter::class, function() {
			/** @var UI\Form $form */
			$form = $this->getForm();

			if (!$form->isSubmitted()) {
				$form->onRender = array_merge(
					[
						function() {
							if ($this->isRequired() && !$this->count()) {
								$this->createNew();
							}
						}
					],
					$form->onRender
				);
				return;
			}

			if ($this->getHttpData()) {
				foreach (array_keys($this->getHttpData()) as $id) {
					$this->getComponent($id); // eager initialize
				}
			}
		});
	}


	public function validate(?array $controls = NULL): void
	{
		parent::validate($controls);
		
		if (
			$this->isRequired()
			&&
			!iterator_count($this->getComponents())
		) {
			$this->addError($this->getRequiredMessage());
		}
	}

	/**
	 * @param string $name
	 * @return Nette\ComponentModel\IComponent|null
	 */
	protected function createComponent(string $name): ?Nette\ComponentModel\IComponent
	{
		if (str_contains($name, static::NEW_PREFIX)) {
			$this->newCount++;
		}
		return $this[$name] = $this->staticContainerFactory->create();
	}


	public function setStaticContainerFactory($staticContainerFactory): static
	{
		$this->staticContainerFactory = $staticContainerFactory;
		return $this;
	}


	/**
	 * @throws Exception
	 */
	public function getTemplate(): StaticContainer
	{
		if (!$this->isAllowAdding()) {
			throw new Exception('Adding is not allowed.');
		}

		if (!$this->template) {
			$this->template = $this[static::NEW_PREFIX]->setIsTemplate(true);
		}
		return $this->template;
	}


	public function createNew(): StaticContainer
	{
		return $this[static::NEW_PREFIX . $this->newCount++];
	}

	public function setDefaults(object|array $data, bool $erase = false): static
	{
		$form = $this->getForm(throw: false);
		if (!$form || !$form->isAnchored() || !$form->isSubmitted()) {
			return parent::setDefaults($data, $erase);
		}
		return $this;
	}

	public function setValues(object|array $values, bool $erase = false, bool $onlyDisabled = false): static
	{
		foreach ($values as $name => $value) {
			if ((is_array($value) || $value instanceof Traversable) && !$this->getComponent($name, FALSE)) {
				$this->createComponent($name);
			}
		}

		return parent::setValues($values, $erase, $onlyDisabled);
	}


	public function isAllowAdding(): bool
	{
		return $this->allowAdding;
	}


	public function setAllowAdding(bool $allowAdding): self
	{
		$this->allowAdding = $allowAdding;
		return $this;
	}


	private function getHttpData(): ?array
	{
		$path = explode(self::NameSeparator, $this->lookupPath('Nette\Application\UI\Form'));
		$allData = $this->getForm()->getHttpData();
		return Nette\Utils\Arrays::get($allData, $path, NULL);
	}


	/**
	 * @return StaticContainer[]
	 */
	public function getContainers(): array
	{
		return array_filter(
			array_filter($this->getComponents(), fn($item) => $item instanceof StaticContainer),
			fn(StaticContainer $staticContainer) => !$staticContainer->isTemplate()
		);
	}
	
	public function count(): int
	{
		return count($this->getContainers());
	}
}