WebMiddleware.php 3.3 KB

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