Files
kiri-core/http-helper/Route/MiddlewareManager.php
T

107 lines
2.4 KiB
PHP
Raw Normal View History

2021-07-27 03:30:55 +08:00
<?php
2021-08-17 16:43:50 +08:00
namespace Http\Route;
2021-07-27 03:30:55 +08:00
2021-07-27 16:08:16 +08:00
use Closure;
2021-08-19 10:57:45 +08:00
use Http\IInterface\MiddlewareInterface;
2021-08-11 01:04:57 +08:00
use Kiri\Abstracts\BaseObject;
2021-07-27 03:30:55 +08:00
/**
2021-07-27 04:16:42 +08:00
* Class MiddlewareManager
2021-08-17 16:43:50 +08:00
* @package Http\Route
2021-07-27 03:30:55 +08:00
*/
2021-07-27 04:16:42 +08:00
class MiddlewareManager extends BaseObject
2021-07-27 03:30:55 +08:00
{
2021-07-27 16:08:16 +08:00
private static array $_middlewares = [];
/**
* @param $class
* @param $method
* @param array|string $middlewares
*/
2021-08-28 02:08:20 +08:00
public static function addMiddlewares($class, $method, array|string $middlewares)
2021-07-27 16:08:16 +08:00
{
if (is_object($class)) {
$class = $class::class;
}
if (!isset(static::$_middlewares[$class . '::' . $method])) {
static::$_middlewares[$class . '::' . $method] = [];
}
if (is_string($middlewares) && !in_array($middlewares, static::$_middlewares[$class . '::' . $method])) {
static::$_middlewares[$class . '::' . $method][] = $middlewares;
return;
}
foreach ($middlewares as $middleware) {
if (in_array($middlewares, static::$_middlewares[$class . '::' . $method])) {
continue;
}
static::$_middlewares[$class . '::' . $method][] = $middleware;
}
}
/**
* @param $class
* @param $method
* @return bool
*/
2021-08-28 02:08:20 +08:00
public static function hasMiddleware($class, $method): bool
2021-07-27 16:08:16 +08:00
{
if (is_object($class)) {
$class = $class::class;
}
return isset(static::$_middlewares[$class . '::' . $method]);
}
/**
* @param $class
* @param $method
* @param $caller
* @return mixed
*/
2021-08-28 02:08:20 +08:00
public static function callerMiddlewares($class, $method, $caller): mixed
2021-07-27 16:08:16 +08:00
{
if (is_object($class)) {
$class = $class::class;
}
$middlewares = static::$_middlewares[$class . '::' . $method] ?? [];
if (empty($middlewares)) {
return $caller;
}
return array_reduce(array_reverse($middlewares), function ($stack, $pipe) {
return function ($passable) use ($stack, $pipe) {
2021-08-19 10:57:45 +08:00
if ($pipe instanceof MiddlewareInterface) {
2021-07-27 16:08:16 +08:00
return $pipe->onHandler($passable, $stack);
}
return call_user_func($pipe, $passable, $stack);
};
}, $caller);
}
/**
* @param $middlewares
* @param Closure $caller
* @return Closure
*/
2021-08-28 02:08:20 +08:00
public static function closureMiddlewares($middlewares, Closure $caller): Closure
2021-07-27 16:08:16 +08:00
{
return array_reduce(array_reverse($middlewares), function ($stack, $pipe) {
return function ($passable) use ($stack, $pipe) {
2021-08-19 10:57:45 +08:00
if ($pipe instanceof MiddlewareInterface) {
2021-07-27 16:08:16 +08:00
return $pipe->onHandler($passable, $stack);
}
return call_user_func($pipe, $passable, $stack);
};
}, $caller);
}
2021-07-27 03:30:55 +08:00
}