From c48d57df1430acc1579ec110b2a52831b1aa4876 Mon Sep 17 00:00:00 2001
From: Marcus <1922576605@qq.com>
Date: Fri, 22 May 2026 23:10:40 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=8C=E5=96=84=E9=89=B4=E6=9D=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env.example                          |  3 --
 app/components/CopyCom.vue            | 10 ++++--
 app/pages/dashboard.vue               | 10 +++---
 app/services/answer_service.ts        |  2 +-
 eslint.config.mjs                     |  1 +
 server/api/cache/clear.post.ts        | 18 +----------
 server/api/records.get.ts             | 21 ++----------
 server/api/stats.get.ts               | 19 +----------
 server/api/user/refresh-token.post.ts | 10 +-----
 server/middleware/api-auth.ts         | 35 ++++++++++++++++++++
 server/types.d.ts                     | 16 ++++++++++
 server/utils/accessToken.ts           | 42 ------------------------
 server/utils/api-auth-rules.ts        | 46 +++++++++++++++++++++++++++
 server/utils/env.ts                   |  2 --
 14 files changed, 118 insertions(+), 117 deletions(-)
 create mode 100644 server/middleware/api-auth.ts
 create mode 100644 server/types.d.ts
 delete mode 100644 server/utils/accessToken.ts
 create mode 100644 server/utils/api-auth-rules.ts

diff --git a/.env.example b/.env.example
index 962e612..5bd8813 100644
--- a/.env.example
+++ b/.env.example
@@ -8,9 +8,6 @@ MAX_TOKENS=500
 TEMPERATURE=0.7
 OPENAI_STREAM_MAX_SSE_LINE_BYTES=1048576
 
-# 安全设置，可选
-# ACCESS_TOKEN=your-access-token
-
 # 缓存配置
 ENABLE_CACHE=true
 CACHE_EXPIRATION=86400
diff --git a/app/components/CopyCom.vue b/app/components/CopyCom.vue
index e9eebb3..c3044dc 100644
--- a/app/components/CopyCom.vue
+++ b/app/components/CopyCom.vue
@@ -1,4 +1,4 @@
-<!-- app/components/CopyCom.vue - 通用复制按钮，带图标切换动画 -->
+<!-- app/components/CopyCom.vue - 通用复制按钮 -->
 <script lang="ts" setup>
 import { Check, Copy } from "lucide-vue-next";
 import type { HTMLAttributes } from "vue";
@@ -14,10 +14,12 @@ const {
   variant = "outline",
   size = "sm",
   timeout = 2000,
-  disableDefaultCopy = false
+  disableDefaultCopy = false,
+  disabled = false
 } = defineProps<{
   /** 要复制的文本 */
   text?: string;
+  // eslint-disable-next-line vue/require-default-prop
   class?: HTMLAttributes["class"] | undefined;
   variant?: ButtonVariants["variant"];
   size?: ButtonVariants["size"];
@@ -25,6 +27,8 @@ const {
   timeout?: number;
   /** 禁用内置复制行为，只触发状态变化（外部自行处理复制） */
   disableDefaultCopy?: boolean;
+  /** 禁用按鈕，不可点击且展示禁用样式 */
+  disabled?: boolean;
 }>();
 
 const emit = defineEmits<{
@@ -34,6 +38,7 @@ const emit = defineEmits<{
 const isCopied = ref(false);
 
 const handleCopy = () => {
+  if (disabled) return;
   if (!disableDefaultCopy) {
     copy(text);
   }
@@ -52,6 +57,7 @@ defineExpose({ isCopied });
     type="button"
     :variant="variant"
     :size="size"
+    :disabled="disabled"
     :class="cn('relative overflow-hidden pl-8', className)"
     @click.prevent.stop="handleCopy"
   >
diff --git a/app/pages/dashboard.vue b/app/pages/dashboard.vue
index db87866..aebc1a7 100644
--- a/app/pages/dashboard.vue
+++ b/app/pages/dashboard.vue
@@ -1,6 +1,6 @@
 <!-- app/pages/dashboard.vue - 运行状态与最近问答记录页面 -->
 <script lang="ts" setup>
-import { ArrowPathIcon } from "@heroicons/vue/24/outline";
+import { RefreshCw } from "lucide-vue-next";
 
 import QaRecordDetail from "@/components/dashboard/QaRecordDetail.vue";
 import QaRecordsTable from "@/components/dashboard/QaRecordsTable.vue";
@@ -64,17 +64,17 @@ const handleRefreshToken = async () => {
           >
             {{ authStore.user.apiToken }}
           </code>
-          <CopyCom :text="authStore.user.apiToken" />
+          <CopyCom :text="authStore.user.apiToken" :disabled="refreshing" />
           <Button
             variant="outline"
             size="icon"
             :disabled="refreshing"
             title="刷新 Token"
+            class="flex items-center gap-1 w-fit px-2.5"
             @click="handleRefreshToken"
           >
-            <ArrowPathIcon
-              :class="['size-4', refreshing ? 'animate-spin' : '']"
-            />
+            <RefreshCw :class="['size-4', refreshing ? 'animate-spin' : '']" />
+            <span>刷新 Token</span>
           </Button>
         </div>
       </CardContent>
diff --git a/app/services/answer_service.ts b/app/services/answer_service.ts
index a52f072..eb780d0 100644
--- a/app/services/answer_service.ts
+++ b/app/services/answer_service.ts
@@ -32,7 +32,7 @@ export class AnswerService {
     );
   }
 
-  /** 读取运行统计；配置 ACCESS_TOKEN 时由服务端校验 */
+  /** 读取运行统计（需登录） */
   public static getStats() {
     return BaseClientService.get<IStatsResponse>(
       `${AnswerService.basePath}/stats`
diff --git a/eslint.config.mjs b/eslint.config.mjs
index 33b5bdc..4958ca7 100644
--- a/eslint.config.mjs
+++ b/eslint.config.mjs
@@ -55,6 +55,7 @@ export default withNuxt(
       'vue/html-self-closing': ['error', { html: { void: 'any' } }],
       'no-unused-vars': 'warn',
       '@typescript-eslint/no-unused-vars': 'warn',
+      "@typescript-eslint/no-extraneous-class": "off",
     },
   },
 )
\ No newline at end of file
diff --git a/server/api/cache/clear.post.ts b/server/api/cache/clear.post.ts
index c3c5c97..9cd27d3 100644
--- a/server/api/cache/clear.post.ts
+++ b/server/api/cache/clear.post.ts
@@ -1,10 +1,4 @@
 // server/api/cache/clear.post.ts - 清空答题缓存接口
-import { setResponseStatus } from "h3";
-
-import {
-  invalidAccessTokenResponse,
-  verifyAccessToken
-} from "~~/server/utils/accessToken";
 import { answerCache } from "~~/server/utils/cache";
 import { serverEnv } from "~~/server/utils/env";
 import { createApiLogger } from "~~/server/utils/logging";
@@ -12,22 +6,12 @@ import { createApiLogger } from "~~/server/utils/logging";
 /**
  * 清空内存缓存
  *
- * 这是一个管理类接口，配置 ACCESS_TOKEN 时必须携带正确 token
+ * 这是一个管理类接口，必须携带有效 session cookie（由 api-auth middleware 统一鉴权）
  * 清理的是当前 Nuxt 进程内缓存；多实例部署时，每个实例都有自己的内存缓存
  */
 export default defineEventHandler((event) => {
   const logger = createApiLogger(event, "api.cache.clear");
 
-  // 没配置 token 时保持旧服务的开放行为；配置后用于简单保护管理接口
-  if (!verifyAccessToken(event)) {
-    setResponseStatus(event, 403);
-    logger.warn("invalid_access_token");
-    return {
-      success: false,
-      message: invalidAccessTokenResponse().msg
-    };
-  }
-
   // 缓存被关闭时不报错，只明确告诉调用方当前没有缓存可清
   if (!serverEnv.enableCache || !answerCache) {
     return {
diff --git a/server/api/records.get.ts b/server/api/records.get.ts
index 303a07b..eaf1fb3 100644
--- a/server/api/records.get.ts
+++ b/server/api/records.get.ts
@@ -1,7 +1,6 @@
 // server/api/records.get.ts - 用户问答记录接口，供 Nuxt Dashboard 表格展示
-import { getQuery, setResponseStatus } from "h3";
+import { getQuery } from "h3";
 
-import { getAuthSession } from "~~/server/utils/auth";
 import { createApiLogger } from "~~/server/utils/logging";
 import { getQaRecords } from "~~/server/utils/runtimeState";
 
@@ -9,27 +8,13 @@ import { getQaRecords } from "~~/server/utils/runtimeState";
  * 用户问答记录接口（分页，最新在前）
  *
  * 安全边界：
- * - 必须携带有效 session cookie（登录用户）
+ * - 必须携带有效 session cookie（由 api-auth middleware 统一鉴权）
  * - 只返回当前登录用户自己的记录，用 userId 严格隔离
  * - 不包含 OpenAI 响应体、API Key 或内部错误
  */
 export default defineEventHandler(async (event) => {
   const logger = createApiLogger(event, "api.records");
 
-  const session = await getAuthSession(event);
-  if (!session) {
-    setResponseStatus(event, 401);
-    logger.warn("unauthorized");
-    return {
-      success: false,
-      message: "未登录",
-      records: [],
-      page: 1,
-      size: 10,
-      total: 0
-    };
-  }
-
   const query = getQuery(event);
   const rawPage = Number.parseInt(query.page?.toString() || "1", 10);
   const rawSize = Number.parseInt(query.size?.toString() || "10", 10);
@@ -38,7 +23,7 @@ export default defineEventHandler(async (event) => {
     : 10;
   const page = Number.isFinite(rawPage) ? Math.max(rawPage, 1) : 1;
 
-  const { records, total } = await getQaRecords(session.user.id, {
+  const { records, total } = await getQaRecords(event.context.auth!.user.id, {
     page,
     size
   });
diff --git a/server/api/stats.get.ts b/server/api/stats.get.ts
index c57a1c4..6fde850 100644
--- a/server/api/stats.get.ts
+++ b/server/api/stats.get.ts
@@ -1,10 +1,4 @@
 // server/api/stats.get.ts - 服务运行统计接口
-import { setResponseStatus } from "h3";
-
-import {
-  invalidAccessTokenResponse,
-  verifyAccessToken
-} from "~~/server/utils/accessToken";
 import { createApiLogger } from "~~/server/utils/logging";
 import { getRuntimeStats } from "~~/server/utils/runtimeState";
 
@@ -12,21 +6,10 @@ import { getRuntimeStats } from "~~/server/utils/runtimeState";
  * 运行统计接口
  *
  * 统计信息包含 uptime、模型名、缓存数量、最近问答记录数量
- * 因为它能反映服务使用情况，所以和旧 Python 服务一样走可选 ACCESS_TOKEN 校验
+ * 必须携带有效 session cookie（由 api-auth middleware 统一鉴权）
  */
 export default defineEventHandler((event) => {
   const logger = createApiLogger(event, "api.stats");
-
-  // 未配置 ACCESS_TOKEN 时 verifyAccessToken 会放行；配置后必须携带正确 token
-  if (!verifyAccessToken(event)) {
-    setResponseStatus(event, 403);
-    logger.warn("invalid_access_token");
-    return {
-      success: false,
-      message: invalidAccessTokenResponse().msg
-    };
-  }
-
   // getRuntimeStats 会实时清理过期缓存，再返回有效缓存数量
   logger.info("finish_success");
   return getRuntimeStats();
diff --git a/server/api/user/refresh-token.post.ts b/server/api/user/refresh-token.post.ts
index 8ca1f15..37ab498 100644
--- a/server/api/user/refresh-token.post.ts
+++ b/server/api/user/refresh-token.post.ts
@@ -4,21 +4,13 @@ import { randomBytes } from "node:crypto";
 
 import { setResponseStatus } from "h3";
 
-import { getAuthSession } from "~~/server/utils/auth";
 import { prisma } from "~~/server/utils/db";
 import { createApiLogger } from "~~/server/utils/logging";
 
 export default defineEventHandler(async (event) => {
   const logger = createApiLogger(event, "api.user.refresh-token");
 
-  const session = await getAuthSession(event);
-  if (!session) {
-    setResponseStatus(event, 401);
-    logger.warn("unauthorized");
-    return { success: false, msg: "未登录" };
-  }
-
-  const userId = session.user.id;
+  const userId = event.context.auth!.user.id;
   const newToken = randomBytes(16).toString("hex");
 
   try {
diff --git a/server/middleware/api-auth.ts b/server/middleware/api-auth.ts
new file mode 100644
index 0000000..05bfc5b
--- /dev/null
+++ b/server/middleware/api-auth.ts
@@ -0,0 +1,35 @@
+// server/middleware/api-auth.ts - 统一 API 鉴权层，所有非公开 /api/* 请求都必须通过此处
+import { createError, getRequestURL } from "h3";
+
+import { isPublicApiRoute } from "~~/server/utils/api-auth-rules";
+import { getAuthSession } from "~~/server/utils/auth";
+
+/**
+ * 统一 API 鉴权 middleware
+ *
+ * 策略：默认所有 /api/* 需要登录，只有 publicApiRoutes 里的路由可以公开访问。
+ * 鉴权通过后将 session 挂到 event.context.auth，handler 直接取用，无需重复查询。
+ *
+ * 不影响页面渲染、静态资源或 Nuxt 内部请求。
+ */
+export default defineEventHandler(async (event) => {
+  const { pathname } = getRequestURL(event);
+  const method = event.method;
+
+  // 只处理 /api/ 路径
+  if (!pathname.startsWith("/api/")) return;
+
+  // OPTIONS 预检请求不需要鉴权，CORS headers 由 nuxt.config routeRules 统一设置
+  if (method === "OPTIONS") return;
+
+  // 公开路由直接放行
+  if (isPublicApiRoute(pathname, method)) return;
+
+  const session = await getAuthSession(event);
+  if (!session) {
+    throw createError({ statusCode: 401, statusMessage: "未登录" });
+  }
+
+  // 挂到 event.context，后续 handler 直接读取，不再重复查询 session
+  event.context.auth = session;
+});
diff --git a/server/types.d.ts b/server/types.d.ts
new file mode 100644
index 0000000..a4486ad
--- /dev/null
+++ b/server/types.d.ts
@@ -0,0 +1,16 @@
+// server/types.d.ts - H3EventContext 扩展，供所有 server handler 使用
+import type { auth } from "~~/server/utils/auth";
+
+type BetterAuthSession = NonNullable<
+  Awaited<ReturnType<typeof auth.api.getSession>>
+>;
+
+declare module "h3" {
+  interface H3EventContext {
+    /**
+     * Better Auth session，由 api-auth middleware 在非公开路由上设置。
+     * 公开路由（publicApiRoutes）不会被赋值，业务接口可安全断言为非空。
+     */
+    auth?: BetterAuthSession;
+  }
+}
diff --git a/server/utils/accessToken.ts b/server/utils/accessToken.ts
deleted file mode 100644
index ff5202e..0000000
--- a/server/utils/accessToken.ts
+++ /dev/null
@@ -1,42 +0,0 @@
-// server/utils/accessToken.ts - OCS API 可选访问令牌校验（用于 stats/cache 等管理接口）
-import { getQuery, type H3Event } from "h3";
-
-import { serverEnv } from "~~/server/utils/env";
-
-/**
- * H3 的 query 值可能是字符串、数组或 undefined
- * 访问令牌只接受第一个值，和多数 Web 框架读取 query 的行为保持一致
- */
-const firstQueryValue = (value: unknown) => {
-  if (Array.isArray(value)) return value[0]?.toString() || "";
-  return value?.toString() || "";
-};
-
-/**
- * 校验可选访问令牌
- *
- * - 没配置 `ACCESS_TOKEN` 时，服务保持旧 Python 项目的开放行为
- * - 配置后，兼容两种传递方式：`X-Access-Token` 请求头或 `?token=...`
- */
-export const verifyAccessToken = (event: H3Event) => {
-  if (!serverEnv.accessToken) return true;
-
-  const headerToken = event.node.req.headers["x-access-token"]?.toString();
-  const queryToken = firstQueryValue(getQuery(event).token);
-
-  return (
-    headerToken === serverEnv.accessToken ||
-    queryToken === serverEnv.accessToken
-  );
-};
-
-/**
- * OCS 搜索接口使用 `{ code, msg }`，而 stats/cache 使用 `{ success, message }`
- * 这里先返回最基础的 OCS 形状，其他接口复用其中的 msg 文案
- */
-export const invalidAccessTokenResponse = () => {
-  return {
-    code: 0,
-    msg: "无效的访问令牌"
-  };
-};
diff --git a/server/utils/api-auth-rules.ts b/server/utils/api-auth-rules.ts
new file mode 100644
index 0000000..b58b18f
--- /dev/null
+++ b/server/utils/api-auth-rules.ts
@@ -0,0 +1,46 @@
+// server/utils/api-auth-rules.ts - API 鉴权路由规则表：声明公开路由，其余默认需要登录
+
+type HttpMethod = "GET" | "POST" | "PUT" | "PATCH" | "DELETE" | "OPTIONS";
+
+type ApiRouteRule = {
+  path: string | RegExp;
+  method?: HttpMethod | HttpMethod[];
+};
+
+/**
+ * 公开 API 路由规则表
+ *
+ * 新增接口默认需要登录，无需任何改动；
+ * 只有明确需要公开访问的路由才加到这里。
+ */
+export const publicApiRoutes: ApiRouteRule[] = [
+  // Better Auth 自己的登录、注册、退出、get-session 等接口必须放行
+  { path: "/api/auth/**" },
+  // 服务健康检查，供部署平台和 Docker healthcheck 使用，不含敏感信息
+  { method: "GET", path: "/api/health" },
+  // OCS 油猴脚本跨域无法携带 cookie，搜索接口有自己的双通道鉴权（session 或 apiToken）
+  { path: "/api/search" }
+];
+
+/** 路径匹配：支持精确匹配、`/**` 前缀通配和 RegExp */
+function matchPath(rulePath: string | RegExp, pathname: string): boolean {
+  if (rulePath instanceof RegExp) return rulePath.test(pathname);
+  if (rulePath.endsWith("/**")) {
+    const prefix = rulePath.slice(0, -3);
+    return pathname === prefix || pathname.startsWith(`${prefix}/`);
+  }
+  return pathname === rulePath;
+}
+
+/** 判断当前请求是否命中公开路由规则 */
+export function isPublicApiRoute(pathname: string, method: string): boolean {
+  return publicApiRoutes.some((rule) => {
+    const methods = Array.isArray(rule.method)
+      ? rule.method
+      : rule.method
+        ? [rule.method]
+        : null;
+    const methodMatched = !methods || methods.includes(method as HttpMethod);
+    return methodMatched && matchPath(rule.path, pathname);
+  });
+}
diff --git a/server/utils/env.ts b/server/utils/env.ts
index 3d7eb0f..7afda2b 100644
--- a/server/utils/env.ts
+++ b/server/utils/env.ts
@@ -73,8 +73,6 @@ export const serverEnv = {
   maxTokens: readInteger("MAX_TOKENS", 500),
   /** 模型采样温度，越低越稳定 */
   temperature: readNumber("TEMPERATURE", 0.7),
-  /** 可选访问令牌；配置后 API 需要 header 或 query 携带 token */
-  accessToken: readOptionalString("ACCESS_TOKEN"),
   /** 是否启用内存缓存 */
   enableCache: readBoolean("ENABLE_CACHE", true),
   /** 缓存过期时间，单位秒 */