43 lines
1.7 KiB
PHP
43 lines
1.7 KiB
PHP
<?php
|
|
namespace app\middleware;
|
|
|
|
use support\Container;
|
|
use Webman\MiddlewareInterface;
|
|
use Webman\Http\Response;
|
|
use Webman\Http\Request;
|
|
use Webman\Route;
|
|
|
|
class ActionHook implements MiddlewareInterface
|
|
{
|
|
public function process(Request $request, callable $next) : Response
|
|
{
|
|
if ($request->controller) {
|
|
// 禁止直接访问beforeAction afterAction
|
|
if (substr($request->action,0,9) === '__before_' || substr($request->action,0,8) === '__after_') {
|
|
$callback = Route::getFallback() ?? function () {
|
|
return new Response(404, [], \file_get_contents(public_path() . '/404.html'));
|
|
};
|
|
$reponse = $callback($request);
|
|
return $reponse instanceof Response ? $reponse : \response($reponse);
|
|
}
|
|
$controller = Container::get($request->controller);
|
|
$beforeAction = '__before_'.$request->action.'__';
|
|
if (method_exists($controller, $beforeAction)) {
|
|
$before_response = call_user_func([$controller, $beforeAction], $request);
|
|
if ($before_response instanceof Response) {
|
|
return $before_response;
|
|
}
|
|
}
|
|
$response = $next($request);
|
|
$afterAction = '__after_'.$request->action.'__';
|
|
if (method_exists($controller, $afterAction)) {
|
|
$after_response = call_user_func([$controller, $afterAction], $request, $response);
|
|
if ($after_response instanceof Response) {
|
|
return $after_response;
|
|
}
|
|
}
|
|
return $response;
|
|
}
|
|
return $next($request);
|
|
}
|
|
} |