phug-php/phug

<?php

namespace Phug\Util\Partial;

use InvalidArgumentException;
use Phug\EventManagerTrait;
use Phug\Util\ModuleContainerInterface;
use Phug\Util\ModuleInterface;

/**
 * Class ModuleContainerTrait.
 */
trait ModuleContainerTrait
{
    use EventManagerTrait;
    use OptionTrait;
    /**
     * @var array<ModuleInterface>
     */
    private $modules = [];

    /**
     * @param string|ModuleInterface $module
     *
     * @return bool
     */
    public function hasModule($module)
    {
        return $module instanceof ModuleInterface
            ? in_array($module, $this->modules, true)
            : isset($this->modules[$module]);
    }

    /**
     * @param string|ModuleInterface $module
     *
     * @return ModuleInterface
     */
    public function getModule($module)
    {
        return $module instanceof ModuleInterface
            ? $module
            : $this->modules[$module];
    }

    /**
     * @return array<ModuleInterface>
     */
    public function getModules()
    {
        return array_values($this->modules);
    }

    /**
     * @return array<string>
     */
    public function getStaticModules()
    {
        return array_filter(array_keys($this->modules), function ($key) {
            return is_string($key);
        });
    }

    /**
     * @param string|ModuleInterface $module
     *
     * @return $this
     */
    public function addModule($module)
    {
        if ($module instanceof ModuleInterface) {
            if (in_array($module, $this->modules, true)) {
                throw new InvalidArgumentException(
                    'This occurrence of '.get_class($module).' is already registered.'
                );
            }

            if ($module->getContainer() !== $this) {
                throw new InvalidArgumentException(
                    'This occurrence of '.get_class($module).' is already registered in another module container.'
                );
            }

            $module->attachEvents();
            $this->modules[] = $module;

            return $this;
        }

        /** @var string $className */
        $className = $module;

        if (!is_subclass_of($className, $this->getModuleBaseClassName(), true)
            || !is_subclass_of($className, ModuleInterface::class)) {
            throw new InvalidArgumentException(
                'Passed module class name '.$className.
                ' needs to be a class extending '.$this->getModuleBaseClassName()
                .' and/or '.ModuleInterface::class
            );
        }

        if (isset($this->modules[$className])) {
            throw new InvalidArgumentException(
                'Module '.$className.' is already registered.'
            );
        }

        if (!($this instanceof ModuleContainerInterface)) {
            throw new \RuntimeException(
                'Current module container uses the ModuleContainerTrait, but doesn\'t implement '
                .ModuleContainerInterface::class.', please implement it.'
            );
        }

        /** @var ModuleInterface $module */
        $module = new $className($this);
        $module->attachEvents();
        $this->modules[$className] = $module;

        return $this;
    }

    /**
     * @param array<string|ModuleInterface> $modules
     *
     * @return $this
     */
    public function addModules(array $modules)
    {
        foreach ($modules as $module) {
            $this->addModule($module);
        }

        return $this;
    }

    /**
     * @param string|ModuleInterface $module
     *
     * @return $this
     */
    public function removeModule($module)
    {
        if ($module instanceof ModuleInterface) {
            if (!in_array($module, $this->modules, true)) {
                throw new InvalidArgumentException(
                    'This occurrence of '.get_class($module).' is not registered.'
                );
            }

            $this->modules = array_filter($this->modules, function ($instance) use ($module) {
                return $instance !== $module;
            });

            return $this;
        }

        /** @var string $className */
        $className = $module;

        if (!$this->hasModule($className)) {
            throw new InvalidArgumentException(
                'The container doesn\'t contain a '.$className.' module'
            );
        }

        $this->modules[$className]->detachEvents();
        unset($this->modules[$className]);

        return $this;
    }

    /**
     * @return string
     */
    public function getModuleBaseClassName()
    {
        return ModuleInterface::class;
    }
}