AuthTokenMiddleware.php 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  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. $authInfo['user']['recommend_achievement'] = Achievement::recommendeAchievement($authInfo['user']['uid']);
  29. } catch (AuthException $e) {
  30. if ($force)
  31. return app('json')->make($e->getCode(), $e->getMessage());
  32. }
  33. if (!is_null($authInfo)) {
  34. Request::macro('user', function () use (&$authInfo) {
  35. return $authInfo['user'];
  36. });
  37. Request::macro('tokenData', function () use (&$authInfo) {
  38. return $authInfo['tokenData'];
  39. });
  40. }
  41. Request::macro('isLogin', function () use (&$authInfo) {
  42. return !is_null($authInfo);
  43. });
  44. Request::macro('uid', function () use (&$authInfo) {
  45. return is_null($authInfo) ? 0 : $authInfo['user']->uid;
  46. });
  47. return $next($request);
  48. }
  49. }