添加注释

app/controller/Passport.php

@@ -61,7 +61,7 @@ class Passport extends BaseController
     }
 
     /**
-     * 用户登录
+     * 用户登陆
      * @return array
      * @throws \think\db\exception\DataNotFoundException
      * @throws \think\db\exception\DbException
@@ -87,7 +87,7 @@ class Passport extends BaseController
             if ($count > $defaultCount) {
                 $this->validate($data,['captcha|验证码'=>'require|captcha']);
             }
-            // 用户登录
+            // 用户登陆
             $userModel = new UserModel();
             $user = $userModel->login($data);
 
@@ -234,7 +234,7 @@ class Passport extends BaseController
                 $this->validate($data,['captcha|验证码'=>'require|captcha']);
             }
 
-            // 管理员登录
+            // 管理员登陆
             $adminUser = AdminUser::login($data);
 
             if ($adminUser['status'] == 1) {
@@ -286,7 +286,7 @@ class Passport extends BaseController
                 $this->validate($data,['captcha|验证码'=>'require|captcha']);
             }
 
-            // 管理员登录
+            // 管理员登陆
             $agentUser = AgentUser::login($data);
 
             if ($agentUser['status'] == 1) {

app/middleware/CheckAdmin.php

@@ -0,0 +1,37 @@
+<?php
+declare (strict_types = 1);
+
+namespace app\middleware;
+
+use think\Exception;
+use think\facade\Request;
+use think\facade\Session;
+
+class CheckAdmin
+{
+    /**
+     * 请求处理
+     * @param $request
+     * @param \Closure $next
+     * @return mixed|\think\response\Json
+     */
+    public function handle($request, \Closure $next)
+    {
+        try {
+            $token = Request::header()['token']??false;
+            if(!$token)
+                throw new Exception('Without Token',403);
+            $userinfo = checkToken($token);
+            if($userinfo['code'] != 200)
+                throw new Exception('Token checked error',403);
+            $request->userInfo = $userinfo['data'];
+            if (!Session::get('login_admin_user_data')) {
+                throw new Exception('管理员未登陆，请先登陆后操作',201);
+            }
+        }
+        catch (\Exception $err){
+            return json(['code'=>$err->getCode(),'msg'=>$err->getMessage()]);
+        }
+        return $next($request);
+    }
+}

app/middleware/CheckAgent.php

@@ -0,0 +1,37 @@
+<?php
+declare (strict_types = 1);
+
+namespace app\middleware;
+
+use think\Exception;
+use think\facade\Request;
+use think\facade\Session;
+
+class CheckAgent
+{
+    /**
+     * 处理请求
+     * @param $request
+     * @param \Closure $next
+     * @return mixed|\think\response\Json
+     */
+    public function handle($request, \Closure $next)
+    {
+        try {
+            $token = Request::header()['token']??false;
+            if(!$token)
+                throw new Exception('Without Token',403);
+            $userinfo = checkToken($token);
+            if($userinfo['code'] != 200)
+                throw new Exception('Token checked error',403);
+            $request->userInfo = $userinfo['data'];
+            if (!Session::get('login_agent_user_data')) {
+                throw new Exception('代理未登陆，请先登陆后操作',201);
+            }
+        }
+        catch (\Exception $err){
+            return json(['code'=>$err->getCode(),'msg'=>$err->getMessage()]);
+        }
+        return $next($request);
+    }
+}

app/middleware/CheckUser.php

@@ -27,7 +27,7 @@ class CheckUser
                 throw new Exception('Token checked error',403);
             $request->userInfo = $userinfo['data'];
             if (!Session::get('login_user_data')) {
-                throw new Exception('用户未登录，请先登陆后操作',201);
+                throw new Exception('用户未登陆，请先登陆后操作',201);
             }
         }
         catch (\Exception $err){

app/model/AdminUser.php

@@ -41,7 +41,7 @@ class AdminUser extends Model
             # 缓存用户信息
             $login_user_data = $user->toArray();
             unset($login_user_data['password'],$login_user_data['salt'],$login_user_data['status']);
-            Session::set('login_user_data',$login_user_data);
+            Session::set('login_admin_user_data',$login_user_data);
 
             return ['status' => 1, 'msg' => '登陆成功', 'data' => $login_user_data];
         } catch (\Exception $e) {

app/model/AgentUser.php

@@ -44,7 +44,7 @@ class AgentUser extends Model
             # 缓存用户信息
             $login_user_data = $user->toArray();
             unset($login_user_data['password'],$login_user_data['salt'],$login_user_data['status']);
-            Session::set('login_user_data',$login_user_data);
+            Session::set('login_agent_user_data',$login_user_data);
 
             return ['status' => 1, 'msg' => '登陆成功', 'data' => $login_user_data];
         } catch (\Exception $e) {

app/model/User.php

@@ -40,7 +40,7 @@ class User extends Model
     }
 
     /**
-     * 用户登录
+     * 用户登陆
      * @param $data
      * @return array
      * @throws \think\db\exception\DataNotFoundException
@@ -64,8 +64,8 @@ class User extends Model
                 $login_user_data = $user->toArray();
                 unset($login_user_data['password'],$login_user_data['salt']);
                 Session::set('login_user_data',$login_user_data);
-                // 登录成功
+                // 登陆成功
-                return ['status' => true, 'msg' => '登录成功', 'data' => $login_user_data];
+                return ['status' => true, 'msg' => '登陆成功', 'data' => $login_user_data];
             } else {
 
                 return ['status' => false, 'msg' => '密码错误'];