WebMiddleware.php 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  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 WebMiddleware 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['origin']) && $header['origin'][0]!='null' && isset($header['origin'][0])){
  38. $origin = $header['origin'][0];
  39. $logindevice = explode("//", $origin);
  40. $data = [
  41. 'website_url'=>$logindevice[1]
  42. ];
  43. $result = $this->websiteServiceClient->getWebsiteId($data);
  44. if(!isset($result['data']['id']) || !$result['data']['id']){
  45. return $this->response->json(
  46. [
  47. 'code' => -1,
  48. 'data' => [],
  49. 'message' => '网站不存在',
  50. ]
  51. );
  52. }
  53. var_dump("获取站点id:",$result);
  54. Context::set("SiteId",$result['data']['id']);
  55. if ($result) {
  56. return $handler->handle($request);
  57. }
  58. }else{
  59. return $this->response->json(
  60. [
  61. 'code' => -1,
  62. 'data' => [],
  63. 'message' => 'origin:必填',
  64. ]
  65. );
  66. }
  67. }catch (\Exception $e){
  68. var_dump("错误消息:",$e->getMessage(),$e->getCode());
  69. return $this->response->json(
  70. [
  71. 'code' => $e->getCode(),
  72. 'data' => [],
  73. 'message' => '网址必填:'.$e->getMessage(),
  74. ]
  75. );
  76. }
  77. return false;
  78. }
  79. }