82 lines
1.6 KiB
PHP
82 lines
1.6 KiB
PHP
<?php
|
|
|
|
|
|
namespace Http\Handler\Abstracts;
|
|
|
|
|
|
use Closure;
|
|
use Co\Iterator;
|
|
use Kiri\Abstracts\BaseObject;
|
|
|
|
|
|
/**
|
|
* Class MiddlewareManager
|
|
* @package Http\Route
|
|
*/
|
|
class MiddlewareManager extends BaseObject
|
|
{
|
|
|
|
|
|
/**
|
|
* @var array<string, Iterator>
|
|
*/
|
|
private static array $_middlewares = [];
|
|
|
|
|
|
/**
|
|
* @param $class
|
|
* @param $method
|
|
* @param array|string $middlewares
|
|
* @return bool
|
|
*/
|
|
public static function add($class, $method, array|string $middlewares): bool
|
|
{
|
|
if (is_object($class)) {
|
|
$class = $class::class;
|
|
}
|
|
if (!isset(static::$_middlewares[$class])) {
|
|
static::$_middlewares[$class] = [];
|
|
}
|
|
if (!isset(static::$_middlewares[$class][$method])) {
|
|
static::$_middlewares[$class][$method] = new Iterator();
|
|
}
|
|
if (is_string($middlewares)) {
|
|
$middlewares = [$middlewares];
|
|
}
|
|
$source = static::$_middlewares[$class][$method];
|
|
foreach ($middlewares as $middleware) {
|
|
if (isset($source[$middleware])) {
|
|
continue;
|
|
}
|
|
$source[$middleware] = di($middleware);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $handler
|
|
* @return mixed
|
|
*/
|
|
public static function get($handler): Iterator
|
|
{
|
|
if (!($handler instanceof Closure)) {
|
|
if (!isset(static::$_middlewares[$handler[0]])) {
|
|
static::$_middlewares[$handler[0]] = [];
|
|
}
|
|
if (!isset(static::$_middlewares[$handler[0]][$handler[1]])) {
|
|
static::$_middlewares[$handler[0]][$handler[1]] = new Iterator();
|
|
}
|
|
/** @var Iterator $iterator */
|
|
$iterator = static::$_middlewares[$handler[0]][$handler[1]];
|
|
if ($iterator->count() > 0 && !$iterator->valid()) {
|
|
$iterator->rewind();
|
|
}
|
|
return $iterator;
|
|
}
|
|
return di(Iterator::class);
|
|
}
|
|
|
|
|
|
}
|