UserTokenMiddleware.php 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. <?php
  2. // +----------------------------------------------------------------------
  3. // | CRMEB [ CRMEB赋能开发者,助力企业发展 ]
  4. // +----------------------------------------------------------------------
  5. // | Copyright (c) 2016~2020 https://www.crmeb.com All rights reserved.
  6. // +----------------------------------------------------------------------
  7. // | Licensed CRMEB并不是自由软件,未经许可不能去掉CRMEB相关版权
  8. // +----------------------------------------------------------------------
  9. // | Author: CRMEB Team <admin@crmeb.com>
  10. // +----------------------------------------------------------------------
  11. namespace app\common\middleware;
  12. use app\common\repositories\user\UserRepository;
  13. use app\Request;
  14. use crmeb\exceptions\AuthException;
  15. use crmeb\services\JwtTokenService;
  16. use Firebase\JWT\ExpiredException;
  17. use think\exception\ValidateException;
  18. use think\Response;
  19. use Throwable;
  20. class UserTokenMiddleware extends BaseMiddleware
  21. {
  22. /**
  23. * @param Request $request
  24. * @throws Throwable
  25. * @author xaboy
  26. * @day 2020-04-10
  27. */
  28. public function before(Request $request)
  29. {
  30. $force = $this->getArg(0, true);
  31. try {
  32. $token = trim($request->header('X-Token'));
  33. $lonlat = trim($request->header('LatLon'));
  34. // $longitude = explode(',', $lonlat)[0] ?? 0;
  35. // $latitude = explode(',', $lonlat)[1] ?? 0;
  36. if (strpos($token, 'Bearer') === 0)
  37. $token = trim(substr($token, 6));
  38. if (!$token)
  39. throw new ValidateException('请登录');
  40. /**
  41. * @var UserRepository $repository
  42. */
  43. $repository = app()->make(UserRepository::class);
  44. $service = new JwtTokenService();
  45. try {
  46. $payload = $service->parseToken($token);
  47. } catch (ExpiredException $e) {
  48. $repository->checkToken($token);
  49. $payload = $service->decode($token);
  50. } catch (Throwable $e) {//Token 过期
  51. throw new AuthException('token 已过期');
  52. }
  53. if ('user' != $payload->jti[1])
  54. throw new AuthException('无效的 token');
  55. $user = $repository->get($payload->jti[0]);
  56. if (!$user)
  57. throw new AuthException('账号不存在');
  58. if (!$user['status'])
  59. throw new AuthException('账号已被禁用');
  60. $user['now_local'] = $lonlat;
  61. } catch (Throwable $e) {
  62. if ($force)
  63. throw $e;
  64. $request->macro('isLogin', function () {
  65. return false;
  66. });
  67. $request->macros(['tokenInfo', 'uid', 'userInfo', 'token'], function () {
  68. throw new AuthException('请登录');
  69. });
  70. return;
  71. }
  72. $repository->updateToken($token);
  73. $request->macro('isLogin', function () {
  74. return true;
  75. });
  76. $request->macro('tokenInfo', function () use (&$payload) {
  77. return $payload;
  78. });
  79. $request->macro('token', function () use (&$token) {
  80. return $token;
  81. });
  82. $request->macro('uid', function () use (&$user) {
  83. return $user->uid;
  84. });
  85. $request->macro('userInfo', function () use (&$user) {
  86. return $user;
  87. });
  88. }
  89. public function after(Response $response)
  90. {
  91. // TODO: Implement after() method.
  92. }
  93. }