up

app/middleware/Admin.php

@@ -0,0 +1,54 @@
+<?php
+declare (strict_types = 1);
+
+namespace app\middleware;
+
+use think\facade\Session;
+
+/**
+ * 管理员权限验证中间件
+ */
+class Admin
+{
+    /**
+     * 处理请求
+     *
+     * @param \think\Request $request
+     * @param \Closure       $next
+     * @return mixed
+     */
+    public function handle($request, \Closure $next)
+    {
+        // 检查是否已登录
+        $username = Session::get('username');
+        
+        if (empty($username)) {
+            // 如果是AJAX请求，返回JSON
+            if ($request->isAjax()) {
+                return json([
+                    'code' => 0,
+                    'msg' => '请先登录',
+                ]);
+            }
+            // 否则跳转到登录页
+            return redirect('/login');
+        }
+        
+        // 检查是否为管理员
+        $isAdmin = Session::get('is_admin', false);
+        
+        if (!$isAdmin) {
+            // 如果是AJAX请求，返回JSON
+            if ($request->isAjax()) {
+                return json([
+                    'code' => 0,
+                    'msg' => '无权限访问，需要管理员权限',
+                ]);
+            }
+            // 否则跳转到爬虫页面
+            return redirect('/crawler');
+        }
+        
+        return $next($request);
+    }
+}