Files
wh-api/app/Middleware/CheckTokenMiddleware.php
ykxiao 8a3ab17b25
Some checks failed
Build Docker / build (push) Has been cancelled
增加基础配置
2025-07-12 11:59:33 +08:00

77 lines
2.1 KiB
PHP

<?php
/**
* Author: ykxiao
* Date: 2025/6/4
* Time: 下午2:40
* Description:
*
* (c) ykxiao <yk_9001@hotmail.com>
*
* This source file is subject to the MIT license that is bundled
* with this source code in the file LICENSE.
*/
declare(strict_types=1);
namespace App\Middleware;
use App\Context\UserContext;
use App\Exception\ApiException;
use App\JsonRpc\UserAuthServiceInterface;
use Exception;
use Hyperf\Di\Annotation\Inject;
use Psr\Container\ContainerInterface;
use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\MiddlewareInterface;
use Psr\Http\Server\RequestHandlerInterface;
/**
* Author: ykxiao
* Date: 2025/6/4
* Time: 下午2:43
* Description: 验证Token中间件.
*
* (c) ykxiao <yk_9001@hotmail.com>
*
* This source file is subject to the MIT license that is bundled
* with this source code in the file LICENSE.
*/
class CheckTokenMiddleware implements MiddlewareInterface
{
#[Inject]
protected UserAuthServiceInterface $userAuthServiceInterface;
public function __construct(protected ContainerInterface $container)
{
}
/**
* @throws Exception
*/
public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
{
$token = $this->extractToken($request);
$rpcResult = $this->userAuthServiceInterface->userInfoByToken(['token' => $token]);
$userInfo = $rpcResult['result'] ?? [];
if (!$userInfo) {
throw new ApiException('用户信息不存在', 401);
}
UserContext::setCurrentToken($token);
UserContext::setCurrentUser($userInfo);
return $handler->handle($request);
}
protected function extractToken(ServerRequestInterface $request): string
{
$authorizationHeader = $request->getHeaderLine('Authorization') ?? '';
if (!$authorizationHeader) {
throw new ApiException('授权标头无效', 401);
}
return str_starts_with($authorizationHeader, 'Bearer ') ? substr($authorizationHeader, 7) : '';
}
}