kiri-core/System/Di/Service.php

<?php
/**
 * Created by PhpStorm.
 * User: whwyy
 * Date: 2018/4/25 0025
 * Time: 18:29
 */
declare(strict_types=1);

namespace Snowflake\Di;


use ReflectionException;
use Snowflake\Exception\ComponentException;
use Snowflake\Abstracts\Component;
use Exception;
use Snowflake\Exception\NotFindClassException;
use Snowflake\Snowflake;

/**
 * Class Service
 * @package Snowflake\Snowflake\Di
 */
class Service extends Component
{

    private array $_components = [];


    private array $_definition = [];


    protected array $_alias = [];

    /**
     * @param $id
     *
     * @return mixed
     * @throws
     */
    public function get($id): mixed
    {
        if (isset($this->_components[$id])) {
            return $this->_components[$id];
        }
        if (!isset($this->_definition[$id]) && !isset($this->_alias[$id])) {
            throw new ComponentException("Unknown component ID: $id");
        }
        if (isset($this->_definition[$id])) {
            $config = $this->_definition[$id];
            if (is_object($config)) {
                return $this->_components[$id] = $config;
            }
            $object = Snowflake::createObject($config);
        } else {
            $config = $this->_alias[$id];

            $object = Snowflake::createObject($config);
        }
        return $this->_components[$id] = $object;
    }

    /**
     * @param string $className
     * @param string $alias
     */
    public function setAlias(string $className, string $alias)
    {
        $this->_alias[$className] = $alias;
    }

    /**
     * @param $id
     * @param $definition
     *
     * @return mixed
     * @throws ComponentException
     * @throws ReflectionException
     * @throws NotFindClassException
     */
    public function set($id, $definition): mixed
    {
        if ($definition === NULL) {
            return $this->remove($id);
        }
        unset($this->_components[$id]);
        if (is_object($definition) || is_callable($definition, TRUE)) {
            return $this->_definition[$id] = $definition;
        } else if (!is_array($definition)) {
            throw new ComponentException("Unexpected configuration type for the \"$id\" component: " . gettype($definition));
        }
        if (!isset($definition['class'])) {
            throw new ComponentException("The configuration for the \"$id\" component must contain a \"class\" element.");
        } else {
            $this->_definition[$id] = $definition;
        }
        return $this->get($id);
    }

    /**
     * @param $id
     * @return bool
     */
    public function has($id): bool
    {
        return isset($this->_definition[$id]) || isset($this->_components[$id]) || isset($this->_alias[$id]);
    }

    /**
     * @param array $data
     * @throws Exception
     */
    public function setComponents(array $data)
    {
        foreach ($data as $key => $val) {
            $this->set($key, $val);
        }
    }


    /**
     * @param $name
     * @return mixed
     * @throws Exception
     */
    public function __get($name): mixed
    {
        if ($this->has($name)) {
            return $this->get($name);
        }

        return parent::__get($name);
    }

    /**
     * @param $id
     * @return bool
     */
    public function remove($id): bool
    {
        unset($this->_components[$id]);
        unset($this->_definition[$id]);
        if (isset($this->_alias[$id])) {
            unset($this->_components[$this->_alias[$id]]);
            unset($this->_definition[$this->_alias[$id]]);
            unset($this->_alias[$id]);
        }
        return $this->has($id);
    }
}