Jagepard/Rudra-Router

<?php

declare(strict_types=1);

/**
 * @author    : Jagepard <jagepard@yandex.ru">
 * @license   https://mit-license.org/ MIT
 */

namespace Rudra\Router\Traits;

use Rudra\Annotation\Annotation;
use Rudra\Container\Interfaces\RudraInterface;

trait RouterAnnotationTrait
{
    /**
     * @param  array   $controllers
     * @param  boolean $getter
     * @param  boolean $attributes
     * @return void
     */
    public function annotationCollector(array $controllers, bool $getter = false, bool $attributes = false)
    {
        $annotations = [];

        foreach ($controllers as $controller) {
            $actions = get_class_methods($controller);

            foreach ($actions as $action) {
                $annotation = ($attributes)
                    ? $this->rudra->get(Annotation::class)->getAttributes($controller, $action)
                    : $this->rudra->get(Annotation::class)->getAnnotations($controller, $action);

                $middleware = [];

                if (isset($annotation["Middleware"])) {
                    $middleware = array_merge($middleware, ['before' => $this->handleAnnotationMiddleware($annotation["Middleware"])]);
                }

                if (isset($annotation["AfterMiddleware"])) {
                    $middleware = array_merge($middleware, ['after' => $this->handleAnnotationMiddleware($annotation["AfterMiddleware"])]);
                }

                if (isset($annotation["Routing"])) {
                    foreach ($annotation["Routing"] as $route) {

                        $route['controller'] = $controller;
                        $route['action']     = $action;
                        $route['middleware'] = $middleware;
                        $route['method']     = $route['method'] ?? "GET";

                        if ($getter) {
                            $annotations[] = [$route];
                        } else {
                            $this->set($route);
                        }
                    }
                }
            }
        }

        if ($getter) {
            return $annotations;
        }
    }

    /**
     * @param array $annotation
     * @return array
     */
    protected function handleAnnotationMiddleware(array $annotation): array
    {
        $middleware = [];
        $count      = count($annotation);

        for ($i = 0; $i < $count; $i++) {
            $middleware[$i][] = $annotation[$i]["name"];

            if (isset($annotation[$i]["params"])) {
                $middleware[$i][] = $annotation[$i]["params"];
            }
        }

        return $middleware;
    }

    abstract public function rudra(): RudraInterface;
}