You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

84 lines
2.2 KiB

  1. <?php
  2. namespace App\Middleware\Auth;
  3. use App\Model\v3\User;
  4. use App\TaskWorker\SSDBTask;
  5. use Hashids\Hashids;
  6. use Hyperf\HttpServer\Contract\RequestInterface as HttpRequest;
  7. use Hyperf\HttpServer\Contract\ResponseInterface as HttpResponse;
  8. use Hyperf\Utils\ApplicationContext;
  9. use Psr\Container\ContainerInterface;
  10. use Psr\Http\Message\ResponseInterface;
  11. use Psr\Http\Message\ServerRequestInterface;
  12. use Psr\Http\Server\MiddlewareInterface;
  13. use Psr\Http\Server\RequestHandlerInterface;
  14. class UserMiddleware implements MiddlewareInterface
  15. {
  16. /**
  17. * @var ContainerInterface
  18. */
  19. protected $container;
  20. /**
  21. * @var HttpResponse
  22. */
  23. protected $response;
  24. /**
  25. * @var HttpRequest
  26. */
  27. protected $request;
  28. public function __construct(ContainerInterface $container, HttpResponse $response, HttpRequest $request)
  29. {
  30. $this->container = $container;
  31. $this->response = $response;
  32. $this->request = $request;
  33. make(Hashids::class, ['secret' => config('auth.user.hash_ids_secret')]);
  34. }
  35. /**
  36. * @inheritDoc
  37. */
  38. public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
  39. {
  40. if (env('APP_ENV') == 'dev' || env('APP_ENV') == 'local') {
  41. return $handler->handle($request);
  42. }
  43. // 校验
  44. if (!$this->checkLogin()) {
  45. $content = [
  46. "status" => 'ok',
  47. "code" => 9002,
  48. "result" => [],
  49. "message" => '用户登录已失效'
  50. ];
  51. return $this->response->json($content);
  52. }
  53. return $handler->handle($request);
  54. }
  55. private function checkLogin()
  56. {
  57. return true;
  58. $userToken = $this->request->input('user_token', '');
  59. $ssdb = ApplicationContext::getContainer()->get(SSDBTask::class);
  60. $exists = $ssdb->exec('exists', $userToken);
  61. if (!$userToken || !$exists) {
  62. return false;
  63. }
  64. $hashIds = ApplicationContext::getContainer()->get(Hashids::class);
  65. $user = $hashIds->decode($userToken);
  66. $this->request->user = User::find($user[0]);
  67. return true;
  68. }
  69. }