AuthTokenMiddleware.php 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. <?php
  2. namespace app\http\middleware;
  3. use app\models\user\Achievement;
  4. use app\models\user\User;
  5. use app\models\user\UserToken;
  6. use app\Request;
  7. use crmeb\exceptions\AuthException;
  8. use crmeb\interfaces\MiddlewareInterface;
  9. use crmeb\repositories\UserRepository;
  10. use think\db\exception\DataNotFoundException;
  11. use think\db\exception\ModelNotFoundException;
  12. use think\exception\DbException;
  13. /**
  14. * token验证中间件
  15. * Class AuthTokenMiddleware
  16. * @package app\http\middleware
  17. */
  18. class AuthTokenMiddleware implements MiddlewareInterface
  19. {
  20. public function handle(Request $request, \Closure $next, bool $force = true)
  21. {
  22. $authInfo = null;
  23. $token = trim(ltrim($request->header('Authori-zation'), 'Bearer'));
  24. if (!$token) $token = trim(ltrim($request->header('Authorization'), 'Bearer'));//正式版,删除此行,某些服务器无法获取到token调整为 Authori-zation
  25. try {
  26. $authInfo = UserRepository::parseToken($token);
  27. $authInfo['user']['achievement'] = Achievement::achievement($authInfo['user']['uid']);
  28. } catch (AuthException $e) {
  29. if ($force)
  30. return app('json')->make($e->getCode(), $e->getMessage());
  31. }
  32. if (!is_null($authInfo)) {
  33. Request::macro('user', function () use (&$authInfo) {
  34. return $authInfo['user'];
  35. });
  36. Request::macro('tokenData', function () use (&$authInfo) {
  37. return $authInfo['tokenData'];
  38. });
  39. }
  40. Request::macro('isLogin', function () use (&$authInfo) {
  41. return !is_null($authInfo);
  42. });
  43. Request::macro('uid', function () use (&$authInfo) {
  44. return is_null($authInfo) ? 0 : $authInfo['user']->uid;
  45. });
  46. return $next($request);
  47. }
  48. }