FooMiddleware.php 3.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. <?php
  2. declare(strict_types=1);
  3. namespace App\Middleware\Auth;
  4. use Hyperf\Di\Annotation\Inject;
  5. use Hyperf\HttpServer\Contract\RequestInterface;
  6. use Hyperf\HttpServer\Contract\ResponseInterface as HttpResponse;
  7. use Psr\Container\ContainerInterface;
  8. use Psr\Http\Message\ResponseInterface;
  9. use Psr\Http\Message\ServerRequestInterface;
  10. use Psr\Http\Server\MiddlewareInterface;
  11. use Psr\Http\Server\RequestHandlerInterface;
  12. use Hyperf\Context\Context;
  13. use App\JsonRpc\WebsiteServiceInterface;
  14. use Phper666\JWTAuth\JWT;
  15. class FooMiddleware implements MiddlewareInterface
  16. {
  17. protected ContainerInterface $container;
  18. protected RequestInterface $request;
  19. protected HttpResponse $response;
  20. protected JWT $JWT;
  21. /**
  22. * @var WebsiteServiceInterface
  23. */
  24. #[Inject]
  25. private $websiteServiceClient;
  26. public function __construct(ContainerInterface $container, HttpResponse $response, RequestInterface $request,Jwt $JWT)
  27. {
  28. $this->container = $container;
  29. $this->response = $response;
  30. $this->request = $request;
  31. $this->JWT = $JWT;
  32. }
  33. public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
  34. {
  35. $header = $request->getHeaders();
  36. try {
  37. if($header && isset($header['token']) && $header['token'][0]!='null' && $header['token'][0]!='' && isset($header['token'][0])){
  38. $ver = $this->JWT->getClaimsByToken($header['token'][0]);
  39. $tokenTime = $this->JWT->getTokenDynamicCacheTime($header['token'][0]);
  40. if($tokenTime==0){
  41. return $this->response->raw(json_encode([
  42. 'code' => -1,
  43. 'data' => [],
  44. 'message' => 'token无效,请重新登录'
  45. ]));
  46. }
  47. if($header['userurl'] && $header['userurl'][0]){
  48. $origin = $header['userurl'][0];
  49. $data = [
  50. 'website_url'=>$origin
  51. ];
  52. $result = $this->websiteServiceClient->getWebsiteId($data);
  53. if(!isset($result['data']['id']) || !$result['data']['id']){
  54. return $this->response->raw(json_encode([
  55. 'code' => -1,
  56. 'data' => [],
  57. 'message' => '网站不存在...'
  58. ]));
  59. }
  60. Context::set("SiteId",$result['data']['id']);
  61. }
  62. Context::set("UserId",$ver['uid']);
  63. Context::set("TypeId",$ver['type_id']);
  64. if ($ver) {
  65. return $handler->handle($request);
  66. }
  67. }
  68. return $this->response->raw(json_encode([
  69. 'code' => -1,
  70. 'data' => [],
  71. 'message' => 'token无效,请重新登录'
  72. ]));
  73. } catch (\Exception $e) {
  74. return $this->response->raw(json_encode([
  75. 'code' => $e->getCode(),
  76. 'data' => [],
  77. 'message' => $e->getMessage()
  78. ]));
  79. }
  80. }
  81. }