85 lines
3.0 KiB
PHP
85 lines
3.0 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace App\Middleware;
|
|
|
|
use App\Model\Online;
|
|
use App\Utils\Token;
|
|
use Hyperf\Context\ApplicationContext;
|
|
use Hyperf\Context\Context;
|
|
use Psr\Container\ContainerInterface;
|
|
use Psr\Http\Message\ResponseInterface;
|
|
use Psr\Http\Server\MiddlewareInterface;
|
|
use Psr\Http\Message\ServerRequestInterface;
|
|
use Psr\Http\Server\RequestHandlerInterface;
|
|
use Psr\SimpleCache\CacheInterface;
|
|
use function Hyperf\Config\config;
|
|
|
|
class JWTMiddleware implements MiddlewareInterface
|
|
{
|
|
/**
|
|
* @var ContainerInterface
|
|
*/
|
|
protected ContainerInterface $container;
|
|
|
|
/**
|
|
* @var CacheInterface
|
|
*/
|
|
protected CacheInterface $cache;
|
|
|
|
public function __construct(ContainerInterface $container, CacheInterface $cache)
|
|
{
|
|
$this->container = $container;
|
|
$this->cache = $cache;
|
|
}
|
|
|
|
/**
|
|
* 登录状态校验
|
|
* @param ServerRequestInterface $request
|
|
* @param RequestHandlerInterface $handler
|
|
* @return ResponseInterface
|
|
*/
|
|
public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
|
|
{
|
|
$container = ApplicationContext::getContainer();
|
|
// 管理端 和 商户端
|
|
$request = $request->withAttribute("isLogin", false);
|
|
$request = $request->withAttribute("account", []);
|
|
$request = $request->withAttribute("account_id", 0);
|
|
try {
|
|
$token = $request->getHeaderLine("Authorization", "");
|
|
$result = Token::parseToken(str_replace("Bearer ", "", $token));
|
|
if (!empty($result)) {
|
|
$user = $this->cache->get("USER:" . $result['uuid']);
|
|
if (!empty($user)) {
|
|
$account = json_decode($user, true);
|
|
}
|
|
}
|
|
// 判断登录状态是否强退
|
|
if (!empty($account)) {
|
|
// 是否登录
|
|
$request = $request->withAttribute("isLogin", true);
|
|
$account = json_decode($this->cache->get("USER:" . $result['uuid']), true);
|
|
// 账号ID
|
|
$request = $request->withAttribute("account_id", $account['account_id']);
|
|
// 基础信息
|
|
$request = $request->withAttribute("account", $account);
|
|
// 如果账户离线恢复在线
|
|
Online::reLive($result['uuid']);
|
|
}
|
|
} catch (\Exception $exception) {
|
|
}
|
|
Context::set(ServerRequestInterface::class, $request);
|
|
$response = $handler->handle($request);
|
|
// 续签逻辑
|
|
if (!empty($result) && $result['time'] - time() < 600 && !empty($account)) {
|
|
$newToken = Token::buildToken(['uuid' => $result['uuid'], 'time' => time() + config("jwt.ttl")], config("jwt.ttl"));
|
|
$response = $response->withHeader('X-Token-Refresh', $newToken);
|
|
$response = $response->withHeader('X-Token-Expire', config("jwt.ttl"));
|
|
$this->cache->set("USER:" . $result['uuid'], json_encode($account), config("jwt.ttl"));
|
|
}
|
|
return $response;
|
|
}
|
|
}
|