kiri-core/Annotation/Loader.php

<?php


namespace Annotation;


use Attribute;
use DirectoryIterator;
use Exception;
use ReflectionClass;
use ReflectionMethod;
use ReflectionProperty;
use Snowflake\Abstracts\BaseObject;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;
use Throwable;


/**
 * Class Loader
 * @package Annotation
 */
class Loader extends BaseObject
{


    private array $_classes = [];


    private array $_fileMap = [];


    private FileTree $files;


    public function init()
    {
        $this->files = new FileTree();
    }


    /**
     * @param $path
     * @param $namespace
     * @throws Exception
     */
    public function loader($path, $namespace)
    {
        $this->_scanDir(new DirectoryIterator($path), $namespace);
    }


    /**
     * @return array
     */
    public function getClasses(): array
    {
        return $this->_classes;
    }


    /**
     * @param string $class
     * @param string $property
     * @return mixed
     */
    public function getProperty(string $class, string $property = ''): mixed
    {
        if (!isset($this->_classes[$class])) {
            return null;
        }
        $properties = $this->_classes[$class]['property'];
        if (!empty($property) && isset($properties[$property])) {
            return $properties[$property];
        }
        return $properties;
    }


    /**
     * @param string $class
     * @param mixed $handler
     * @return Loader
     */
    public function injectProperty(string $class, object $handler): static
    {
        $properties = $this->getProperty($class);
        if (empty($properties)) {
            return $this;
        }
        foreach ($properties as $property => $attributes) {
            foreach ($attributes as $attribute) {
                $attribute->execute([$handler, $property]);
            }
        }
        return $this;
    }


    /**
     * @param string $class
     * @param string $method
     * @return mixed
     */
    public function getMethod(string $class, string $method = ''): array
    {
        if (!isset($this->_classes[$class])) {
            return [];
        }
        $properties = $this->_classes[$class]['methods'];
        if (!empty($method) && isset($properties[$method])) {
            return $properties[$method];
        }
        return $properties;
    }


    /**
     * @param string $class
     * @return array
     */
    public function getTarget(string $class): array
    {
        return $this->_classes[$class] ?? [];
    }


    /**
     * @param DirectoryIterator $paths
     * @param $namespace
     * @throws Exception
     */
    public function _scanDir(DirectoryIterator $paths, $namespace)
    {
        foreach ($paths as $path) {
            if ($path->isDot() || str_starts_with($path->getFilename(), '.')) {
                continue;
            }
            if ($path->isDir()) {
                $iterator = new DirectoryIterator($path->getRealPath());
                $this->_scanDir($iterator, $namespace);
            } else {
                $this->readFile($path, $namespace);
            }
        }
    }


    /**
     * @param DirectoryIterator $path
     * @param $namespace
     * @throws Exception
     */
    private function readFile(DirectoryIterator $path, $namespace)
    {
        try {
            if ($path->getExtension() !== 'php') {
                return;
            }

            $replace = Snowflake::getDi()->getReflect($this->explodeFileName($path, $namespace));
            if (empty($replace) || !$replace->getAttributes(Target::class)) {
                return;
            }
            $this->appendFileToDirectory($path->getRealPath(), $replace->getName());

            $_array = ['handler' => $replace->newInstanceWithoutConstructor(), 'target' => [], 'methods' => [], 'property' => []];
            foreach ($replace->getAttributes() as $attribute) {
                if ($attribute->getName() == Attribute::class) {
                    continue;
                }
                $_array['target'][] = $attribute->newInstance();
            }

            $methods = $replace->getMethods(ReflectionMethod::IS_PUBLIC);
            foreach ($methods as $method) {
                $_method = [];
                foreach ($method->getAttributes() as $attribute) {
                    if (!class_exists($attribute->getName())) {
                        continue;
                    }
                    $_method[] = $attribute->newInstance();
                }
                $_array['methods'][$method->getName()] = $_method;
            }

            $methods = $replace->getProperties();
            foreach ($methods as $method) {
                $_property = [];
                if ($method->isStatic()) continue;
                foreach ($method->getAttributes() as $attribute) {
                    if (!class_exists($attribute->getName())) {
                        continue;
                    }
                    $_property[] = $attribute->newInstance();
                }
                $_array['property'][$method->getName()] = $_property;
            }

            $this->_fileMap[$replace->getFileName()] = $replace->getName();

            $this->_classes[$replace->getName()] = $_array;
        } catch (Throwable $throwable) {
            $this->addError($throwable, 'throwable');
        }
    }


    /**
     * @param string $path
     * @throws Exception
     */
    public function loadByDirectory(string $path, ?string $outPath = null)
    {
        try {
            $this->each($path, $outPath);
        } catch (Throwable $exception) {
            $this->addError($exception, 'throwable');
        }
    }


    /**
     * @param DirectoryIterator $path
     * @param string $namespace
     * @return string
     */
    private function explodeFileName(DirectoryIterator $path, string $namespace): string
    {
        $replace = str_replace(APP_PATH . 'app', '', $path->getRealPath());

        $replace = str_replace('.php', '', $replace);
        $replace = str_replace(DIRECTORY_SEPARATOR, '\\', $replace);
        $explode = explode('\\', $replace);
        array_shift($explode);

        return $namespace . '\\' . implode('\\', $explode);
    }


    /**
     * @param string $filePath
     * @param string $className
     */
    public function appendFileToDirectory(string $filePath, string $className)
    {
        $directory = $this->splitDirectory($filePath);
        array_pop($directory);

        $tree = null;
        foreach ($directory as $value) {
            $tree = $this->getTree($tree, $value);
        }
        if ($tree instanceof FileTree) {
            $tree->addFile($className);
        }
    }


    /**
     * @param string $filePath
     * @return $this
     */
    private function each(string $filePath, ?string $output): static
    {
        $tree = null;
        $directory = $this->splitDirectory($filePath);

        if (!empty($output)) {
            $output = DIRECTORY_SEPARATOR . trim($output, '/');
        }

        $out_path = '';
        foreach ($directory as $key => $value) {
            $out_path .= DIRECTORY_SEPARATOR . $value;
            if ($out_path === $output) {
                break;
            }
            $tree = $this->getTree($tree, $value);
        }
        if ($tree instanceof FileTree) {
            $this->eachNode($tree->getChildes());
            $this->execute($tree->getFiles());
        }
        return $this;
    }


    /**
     * @param string $filePath
     * @return false|string[]
     */
    private function splitDirectory(string $filePath)
    {
        $DIRECTORY = explode(DIRECTORY_SEPARATOR, $filePath);
        return array_filter($DIRECTORY, function ($value) {
            return !empty($value);
        });
    }


    /**
     * @param $tree
     * @param $value
     * @return FileTree
     */
    private function getTree($tree, $value): FileTree
    {
        if ($tree === null) {
            $tree = $this->files->getChild($value);
        } else {
            $tree = $tree->getChild($value);
        }
        return $tree;
    }


    /**
     * @param FileTree[] $nodes
     */
    private function eachNode(array $nodes)
    {
        foreach ($nodes as $node) {
            $childes = $node->getChildes();
            if (!empty($childes)) {
                $this->eachNode($childes);
            }
            $this->execute($node->getFiles());
        }
    }


    /**
     * @param string $filename
     * @return mixed
     */
    public function getClassByFilepath(string $filename): mixed
    {
        if (!isset($this->_fileMap[$filename])) {
            return null;
        }
        return $this->_classes[$this->_fileMap[$filename]];
    }


    /**
     * @param array $classes
     */
    private function execute(array $classes)
    {
        if (empty($classes)) {
            return;
        }
        foreach ($classes as $className) {
            $annotations = $this->_classes[$className] ?? null;
            if ($annotations === null) {
                continue;
            }
            foreach ($annotations['target'] ?? [] as $value) {
                $value->execute([$annotations['handler']]);
            }
            foreach ($annotations['methods'] as $name => $attribute) {
                foreach ($attribute as $value) {
                    if (!($value instanceof \Annotation\Attribute)) {
                        continue;
                    }
                    $value->execute([$annotations['handler'], $name]);
                }
            }
        }
    }

}