@allbluecn/web-app 0.19.2 → 0.19.4

package/src/lib/query-client-seam.test.ts

@@ -1,5 +1,4 @@
 import { describe, expect, it, vi } from 'vitest'
-import { QueryClient } from '@tanstack/react-query'
 import { getQueryClient, setQueryClient } from './query-client-seam'
 
 vi.mock('./query-client-seam', async (importOriginal) => {
@@ -7,8 +6,11 @@ vi.mock('./query-client-seam', async (importOriginal) => {
 })
 
 describe('query-client seam', () => {
-  it('未注入时自动创建兜底 QueryClient', () => {
-    expect(getQueryClient()).toBeInstanceOf(QueryClient)
+  it('SSR（vitest 默认）未注入时 fail-fast 抛错，不做跨请求共享兜底', () => {
+    // vitest 下 import.meta.env.SSR === true：懒创建兜底在服务端会跨请求
+    // 共享缓存（数据串染），必须抛错暴露装配缺失；客户端兜底保留（浏览器
+    // 无跨请求问题），由 useQuery 消费方行为回归覆盖
+    expect(() => getQueryClient()).toThrow(/QueryClient 未注入/)
   })
 
   it('注入后返回注入实例', () => {
@@ -16,4 +18,4 @@ describe('query-client seam', () => {
     setQueryClient(fake)
     expect(getQueryClient()).toBe(fake)
   })
-})
+})

package/src/lib/query-client-seam.ts

@@ -8,6 +8,14 @@ export function setQueryClient(client: QueryClient) {
 
 export function getQueryClient(): QueryClient {
   if (!instance) {
+    // SSR 下 fail-fast：模块级单例兜底在服务端会跨请求共享缓存（请求间数据
+    // 串染）。正常装配顺序 router.tsx 的 getRouter() 先 setQueryClient，此
+    // throw 只在「忘记注入」的装配缺失时暴露（客户端兜底保留，行为不变）
+    if (import.meta.env.SSR) {
+      throw new Error(
+        '[query-client-seam] QueryClient 未注入：SSR 禁止懒创建兜底（模块级单例会跨请求共享缓存）。router.tsx 的 getRouter() 必须先 setQueryClient。',
+      )
+    }
     instance = new QueryClient()
   }
   return instance

package/src/lib/redis.ts

@@ -31,13 +31,23 @@ function getRedis(): Redis | null {
   }
 }
 
+// 配置错误/网络断连时缓存静默降级可接受，但零信号会让排障无从下手：
+// 仅在首次失败时打一条 warn（去抖，不刷屏）
+let warnedUnavailable = false
+function warnRedisOnce(stage: string, err: unknown): void {
+  if (warnedUnavailable) return
+  warnedUnavailable = true
+  console.warn(`[redis] ${stage} 失败，缓存静默降级（后续不再提示）:`, err instanceof Error ? err.message : String(err))
+}
+
 export async function cacheGet<T>(key: string): Promise<T | null> {
   const r = getRedis()
   if (!r) return null
   try {
     const val = await r.get(key)
     return val ? (JSON.parse(val) as T) : null
-  } catch {
+  } catch (err) {
+    warnRedisOnce(`cacheGet(${key})`, err)
     return null
   }
 }
@@ -47,5 +57,7 @@ export async function cacheSet(key: string, value: unknown, ttlSeconds = 1800):
   if (!r) return
   try {
     await r.set(key, JSON.stringify(value), "EX", ttlSeconds)
-  } catch { /* noop */ }
+  } catch (err) {
+    warnRedisOnce(`cacheSet(${key})`, err)
+  }
 }

package/src/lib/restore-version.ts

@@ -1 +1,4 @@
+// ⚠ 兼容 shim：lib 反向依赖 components（层级倒置）。仅作旧 import 路径过渡，
+// 唯一消费方 use-page-version-history.ts 应改直连源模块后删除本文件
+//（审计 ⚪ 记录：lib 是被 components 复用的底层，此依赖方向易成环）。
 export * from '@/components/design/editor/versions/lib/restore-version'

package/src/lib/router-context.ts

@@ -17,6 +17,9 @@
 
 import type { QueryClient } from '@tanstack/react-query'
 
+// ⚠ session.user 形状与 lib/session.ts 的 SessionUser 宽严不一（role 此处为
+// 窄联合、彼处为 string|null）：演进时须两处同步（审计 ⚪ 记录；收敛为复用
+// SessionUser 属跨文件类型重构，涉及 router.tsx context 装配，暂记待决策）。
 export interface RouterContext {
   queryClient: QueryClient
   locale: string

package/src/lib/safe-url.ts

@@ -0,0 +1,91 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+// 用户可控 URL 进入 `href` / `<iframe src>` / `window.open` 前的 scheme 白名单（同构，零依赖）。
+//
+// 审计 2026-10-25 主题 3：知识库属性链接、书签卡、工作台 KB 卡、搜索结果、AI citation、
+// anyhub 应用 iframe 等 ~8 处把存库字符串直接当 URL 渲染，与 tiptap 内部的 sanitizeUrl
+// 不对称——存储型 `javascript:` / `data:text/html` 链接点击即在当前源执行。
+// 本模块是全站唯一入口：渲染侧一律 `toSafeHref(raw)`，非法返回 null 由调用方降级
+// （不渲染链接 / 渲染纯文本 / 回落 "#"）。tiptap 自带 sanitizeUrl 服务于编辑器协议配置，
+// 保持独立。
+
+const DEFAULT_ALLOWED_SCHEMES: readonly string[] = ["http:", "https:", "mailto:", "tel:"]
+
+/** 解析相对 URL 用的占位基址；结果只用来读 protocol，不会泄露到返回值 */
+const RELATIVE_BASE = "https://relative.invalid/"
+
+export interface SafeHrefOptions {
+  /** 允许的 scheme（含冒号），默认 http/https/mailto/tel */
+  allowedSchemes?: readonly string[]
+  /** 是否放行站内相对路径（`/x`、`./x`、`?q`、`#h`），默认 true；iframe src 等场景应关掉 */
+  allowRelative?: boolean
+}
+
+function isRelativeReference(value: string): boolean {
+  // 协议相对 `//host` 不算站内相对（会解析为外站 https），交给绝对分支按 scheme 判定
+  if (value.startsWith("//")) return false
+  return (
+    value.startsWith("/") ||
+    value.startsWith("./") ||
+    value.startsWith("../") ||
+    value.startsWith("?") ||
+    value.startsWith("#")
+  )
+}
+
+/**
+ * 返回可安全放入 href 的 URL：绝对 URL 归一化为 `URL#href`，站内相对引用原样返回；
+ * scheme 不在白名单、无法解析、为空一律 null。
+ */
+export function toSafeHref(input: string | null | undefined, options: SafeHrefOptions = {}): string | null {
+  if (typeof input !== "string") return null
+  const raw = input.trim()
+  if (!raw) return null
+
+  const allowed = options.allowedSchemes ?? DEFAULT_ALLOWED_SCHEMES
+  const allowRelative = options.allowRelative ?? true
+
+  if (isRelativeReference(raw)) {
+    return allowRelative ? raw : null
+  }
+
+  let parsed: URL
+  try {
+    // WHATWG 解析会剥掉 scheme 内嵌的制表/换行（`java\tscript:`），由 protocol 统一判定
+    parsed = new URL(raw, RELATIVE_BASE)
+  } catch {
+    return null
+  }
+
+  // 用占位基址解析成功但 input 本身无 scheme（如 `foo/bar`）：按相对引用处理
+  if (parsed.origin === new URL(RELATIVE_BASE).origin && !/^[a-z][a-z0-9+.-]*:/i.test(raw)) {
+    return allowRelative ? raw : null
+  }
+
+  if (!allowed.includes(parsed.protocol)) return null
+  return parsed.href
+}
+
+export function isSafeHref(input: string | null | undefined, options?: SafeHrefOptions): boolean {
+  return toSafeHref(input, options) !== null
+}
+
+/** 渲染侧便捷形态：非法时回落给定值（默认 "#"），避免在 JSX 里写三元 */
+export function safeHrefOr(input: string | null | undefined, fallback = "#", options?: SafeHrefOptions): string {
+  return toSafeHref(input, options) ?? fallback
+}

package/src/lib/sanitize.ts

@@ -53,6 +53,35 @@ const GLOBAL_ATTRS = [
   'aria-hidden', 'aria-label', 'aria-labelledby',
 ]
 
+/** style 声明值黑名单：外链打点（url）、IE 表达式、伪协议一律拒绝 */
+const UNSAFE_CSS_VALUE = /(url\s*\(|expression\s*\(|javascript:)/i
+
+/** 渲染必需的 CSS 属性白名单（富文本 marks / diff 高亮 / SVG 头像内联样式） */
+const CSS_PROPERTY_ALLOWLIST = [
+  'color', 'background', 'background-color', 'opacity',
+  'font-size', 'font-family', 'font-weight', 'font-style',
+  'text-align', 'text-decoration', 'text-decoration-line', 'text-decoration-style',
+  'text-decoration-color', 'text-transform', 'line-height', 'letter-spacing',
+  'white-space', 'word-break', 'overflow-wrap', 'vertical-align',
+  'width', 'min-width', 'max-width', 'height', 'min-height', 'max-height',
+  'margin', 'margin-top', 'margin-right', 'margin-bottom', 'margin-left',
+  'padding', 'padding-top', 'padding-right', 'padding-bottom', 'padding-left',
+  'border', 'border-color', 'border-width', 'border-style', 'border-radius',
+  'display', 'flex', 'flex-direction', 'align-items', 'justify-content', 'gap',
+  'overflow', 'list-style-type', 'box-sizing',
+  'fill', 'stroke', 'stroke-width', 'stop-color', 'stop-opacity',
+  'transform', 'transform-origin',
+]
+
+function buildAllowedStyles(): sanitizeHtml.IOptions['allowedStyles'] {
+  const safeValue = [new RegExp(`^(?!.*(?:${UNSAFE_CSS_VALUE.source})).*$`, 'i')]
+  return {
+    '*': Object.fromEntries(
+      CSS_PROPERTY_ALLOWLIST.map((prop) => [prop, safeValue]),
+    ),
+  }
+}
+
 const OPTIONS: sanitizeHtml.IOptions = {
   allowedTags: ALLOWED_TAGS,
   allowedAttributes: {
@@ -86,8 +115,12 @@ const OPTIONS: sanitizeHtml.IOptions = {
       return { tagName: tagName.toLowerCase(), attribs: lowerAttribs }
     },
   },
-  // style 属性交给上面的白名单后，sanitize-html 会对 CSS 做解析过滤
-  // （javascript: 等危险表达式被清除）；无需 parseStyleAttributes 关闭
+  // style 属性白名单（审计 🟡）：sanitize-html 默认对 CSS 声明值过滤很弱
+  // （主要靠解析重组），position:fixed 全屏钓鱼覆盖、background:url(...) 外链
+  // 打点等仍可行。allowedStyles 显式收敛：属性白名单（渲染必需项）+ 值级
+  // 正则拒绝 url()/expression()/javascript:。未列出的属性整体丢弃。
+  allowedStyles: buildAllowedStyles(),
+  // 无需 parseStyleAttributes 关闭（关闭会与 allowedStyles 互斥直接抛错）
 }
 
 /**

package/src/lib/share-events.ts

@@ -17,7 +17,7 @@
 
 const clients = new Map<string, Set<ReadableStreamDefaultController>>();
 
-const MAX_CLIENTS_PER_Design = 20;
+const MAX_CLIENTS_PER_DESIGN = 20;
 
 const CLEANUP_INTERVAL = 60_000;
 const IDLE_TIMEOUT = 5 * 60_000;
@@ -32,10 +32,20 @@ function ensureCleanupTimer() {
     const now = Date.now();
     for (const [designId, set] of clients) {
       for (const controller of set) {
+        // 空闲超时摘除（审计 🟡：IDLE_TIMEOUT/controllerMeta 原为死代码）：
+        // enqueue 不抛错但不消费的半死连接会永久占用连接配额——最近一次
+        // 真实事件投递（notifyClients 续期）超过 IDLE_TIMEOUT 即摘除
+        const lastActiveAt = controllerMeta.get(controller) ?? 0;
         try {
           controller.enqueue(new TextEncoder().encode(":cleanup-ping\n\n"));
         } catch {
           set.delete(controller);
+          controllerMeta.delete(controller);
+          continue;
+        }
+        if (now - lastActiveAt > IDLE_TIMEOUT) {
+          set.delete(controller);
+          controllerMeta.delete(controller);
         }
       }
       if (set.size === 0) clients.delete(designId);
@@ -49,7 +59,7 @@ const controllerMeta = new WeakMap<ReadableStreamDefaultController, number>();
 export function addClient(designId: string, controller: ReadableStreamDefaultController) {
   ensureCleanupTimer();
   const set = clients.get(designId);
-  if (set && set.size >= MAX_CLIENTS_PER_Design) {
+  if (set && set.size >= MAX_CLIENTS_PER_DESIGN) {
     throw new Error("连接数已达上限");
   }
   if (!set) clients.set(designId, new Set());
@@ -68,6 +78,12 @@ export function removeClient(designId: string, controller: ReadableStreamDefault
 export function notifyClients(designId: string, event: string, data: unknown) {
   const set = clients.get(designId);
   if (!set) return;
+  // SSE event 字段拒绝换行（审计 🟡：换行可注入伪造 SSE 字段/事件行；当前
+  // event 均来自服务端内部调用，此为防未来引入用户可控来源的守卫）
+  if (event.includes("\n") || event.includes("\r")) {
+    console.error(`[share-events] 非法 event 名（含换行，疑似注入）: ${JSON.stringify(event)}`);
+    return;
+  }
   const message = new TextEncoder().encode(`event: ${event}\ndata: ${JSON.stringify(data)}\n\n`);
   for (const controller of set) {
     try {

package/src/lib/sse-transport.ts

@@ -48,11 +48,16 @@ export function createSSETransport(entityIds: string[]): RealtimeTransport {
     eventSource = new EventSource(url)
 
     eventSource.onmessage = (e) => {
+      // 解析与消费分开 try：onUpdate（消费方回调）抛出的异常不得被当作
+      // 「keepalive 解析失败」吞掉（retryCount = 0 也曾一并跳过，退避永不重置）
+      let data: EntityUpdateEvent
       try {
-        const data = JSON.parse(e.data) as EntityUpdateEvent
-        onUpdate(data)
-        retryCount = 0
-      } catch { /* skip keepalive */ }
+        data = JSON.parse(e.data) as EntityUpdateEvent
+      } catch {
+        return // keepalive 等非 JSON 帧
+      }
+      onUpdate(data)
+      retryCount = 0
     }
 
     eventSource.onerror = () => {
@@ -62,10 +67,13 @@ export function createSSETransport(entityIds: string[]): RealtimeTransport {
   }
 
   const subscribe = (
-    _ids: string[],
+    ids: string[],
     onUpdate: (event: EntityUpdateEvent) => void
   ): (() => void) => {
-    const url = `/api/ws?ids=${entityIds.join(",")}`
+    // 真正消费入参（原实现忽略 _ids 用闭包 entityIds，调用方传不同 ids 静默无效）；
+    // id 逐个编码，防 ","/"&"/"#" 破坏 query 语义
+    const effectiveIds = ids.length > 0 ? ids : entityIds
+    const url = `/api/ws?ids=${effectiveIds.map(encodeURIComponent).join(",")}`
     connect(url, onUpdate)
 
     return () => {

package/src/lib/story/description-diff.ts

@@ -84,17 +84,6 @@ export function splitHtmlBlocks(html: string): string[] {
   return raw.filter((b) => !isBlankHtmlBlock(b));
 }
 
-/**
- * 对比新旧 description 字段（DB 原始字符串），返回需要写入活动时间线的差异列表。
- *
- * 算法：
- * 1. 解析两端 HTML
- * 2. 切成块级元素数组（过滤空白块）
- * 3. 按索引逐块对比，长度不等时短的一侧用空字符串补齐
- *
- * 这个算法是「按顺序对位」的简单 diff，不识别块的移动/插入/删除，与历史实现保持
- * 一致。本函数的修复点仅在于过滤空白块，避免编辑器副作用产生噪声记录。
- */
 /**
  * 提取一段 HTML 的纯文本内容（去掉标签、解码实体、折叠空白）。
  *
@@ -114,6 +103,18 @@ export function normalizeBlockText(html: string): string {
     .trim();
 }
 
+/**
+ * 对比新旧 description 字段（DB 原始字符串），返回需要写入活动时间线的差异列表。
+ *
+ * 算法：
+ * 1. 解析两端 HTML
+ * 2. 切成块级元素数组（过滤空白块）
+ * 3. 按索引逐块对比，长度不等时短的一侧用空字符串补齐
+ *
+ * 这个算法是「按顺序对位」的简单 diff，不识别块的移动/插入/删除，与历史实现保持
+ * 一致。本函数的修复点仅在于过滤空白块，避免编辑器副作用产生噪声记录。
+ * （原 JSDoc 错位悬浮在 normalizeBlockText 之前，已归位，审计 🟢）
+ */
 export function diffDescriptionBlocks(
   oldDesc: string | null | undefined,
   newDesc: string,

package/src/lib/story/video-embed.ts

@@ -28,10 +28,15 @@ function getBilibiliPart(rawUrl: string): string | null {
 export function resolveVideoEmbed(rawUrl: string): VideoEmbed | null {
   if (!rawUrl) return null;
 
-  // 直链视频文件：扩展名只看 URL 路径末段，避免 query 中的伪扩展名误判
+  // 直链视频文件：扩展名只看 URL 路径末段，避免 query 中的伪扩展名误判。
+  // 协议守卫（审计 🟡）：javascript:/data: 等任意 scheme 均可被 new URL 解析，
+  // 且路径可伪造 .mp4 后缀——src 会进 <video>/预览链路，仅放行 http(s)
   try {
     const parsed = new URL(rawUrl);
-    if (VIDEO_FILE_RE.test(parsed.pathname)) {
+    if (
+      (parsed.protocol === "http:" || parsed.protocol === "https:") &&
+      VIDEO_FILE_RE.test(parsed.pathname)
+    ) {
       return { kind: "file", src: rawUrl };
     }
   } catch {

package/src/lib/tiptap/__tests__/link-policy.test.ts

@@ -0,0 +1,35 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+
+import { describe, expect, it } from "vitest";
+import { isAllowedLinkUri } from "../link-policy";
+
+describe("isAllowedLinkUri（TipTap Link scheme 白名单）", () => {
+  it("放行 http/https/mailto/tel", () => {
+    expect(isAllowedLinkUri("https://example.com/a?b=1")).toBe(true);
+    expect(isAllowedLinkUri("http://localhost:3000/x")).toBe(true);
+    expect(isAllowedLinkUri("mailto:a@b.com")).toBe(true);
+    expect(isAllowedLinkUri("tel:+8610000000000")).toBe(true);
+  });
+
+  it("拒绝危险 scheme（编辑器/分享视图的 href sink）", () => {
+    expect(isAllowedLinkUri("javascript:alert(1)")).toBe(false);
+    expect(isAllowedLinkUri("JavaScript:alert(1)")).toBe(false);
+    expect(isAllowedLinkUri("data:text/html,<script>alert(1)</script>")).toBe(false);
+    expect(isAllowedLinkUri("vbscript:msgbox(1)")).toBe(false);
+    expect(isAllowedLinkUri("file:///etc/passwd")).toBe(false);
+  });
+
+  it("放行站内相对路径与锚点，且不把端口误判为 scheme", () => {
+    expect(isAllowedLinkUri("/stories/abc")).toBe(true);
+    expect(isAllowedLinkUri("./relative")).toBe(true);
+    expect(isAllowedLinkUri("#anchor")).toBe(true);
+    expect(isAllowedLinkUri("stories/abc")).toBe(true);
+  });
+
+  it("传入 defaultValidate 时对放行 scheme 做二次校验", () => {
+    expect(isAllowedLinkUri("https://ok.example", (u) => u.includes("ok"))).toBe(true);
+    expect(isAllowedLinkUri("https://bad.example", () => false)).toBe(false);
+  });
+});

package/src/lib/tiptap/link-policy.ts

@@ -0,0 +1,25 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+
+// TipTap Link 扩展的 URI scheme 白名单（编辑器与只读分享视图共用单一来源）。
+// 默认校验只拦 javascript:，仍放行 data:/vbscript: 等；协作内容与粘贴 HTML 均可能
+// 夹带，导出/分享视图在非受控展示端渲染时构成 sink（审计 A7x-5 待决策项收口）。
+const ALLOWED_SCHEME = /^(https?:|mailto:|tel:)/i;
+
+/**
+ * @param url 待校验链接
+ * @param defaultValidate TipTap 提供的默认校验（可选，传入时用于放行项的二次校验）
+ */
+export function isAllowedLinkUri(
+  url: string,
+  defaultValidate?: (url: string) => boolean,
+): boolean {
+  const trimmed = url.trim();
+  if (ALLOWED_SCHEME.test(trimmed)) {
+    return defaultValidate ? defaultValidate(url) : true;
+  }
+  // 无 scheme 分隔符 → 站内相对路径/锚点。首段不含「:」可避免把 ":3000" 端口误判为 scheme。
+  const beforeSlash = trimmed.split("/")[0];
+  return !beforeSlash.includes(":");
+}

package/src/lib/vector-store/allowlist.ts

@@ -0,0 +1,33 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+// 向量库白名单单一事实源（审计 🔵/🟠 修复）：
+// - 表名清单收敛到 @allbluecn/database/lib/vec-tables 的 VEC_TABLES
+//   （零依赖常量模块，pg / sqlite 双后端 + desktop-init 共用，防手抄漂移）；
+// - 过滤/删除/插入列白名单此前双后端各抄一份且 insert 缺失（SQL 标识符注入面），
+//   现统一由本模块导出，两个 store 共同消费。
+// ⚠ 新增向量表/列时：VEC_TABLES 与迁移同步（见 vec-tables.ts 头注释）+ 本文件列清单。
+
+import { VEC_TABLES } from "@allbluecn/database/lib/vec-tables";
+
+export const ALLOWED_TABLES = VEC_TABLES;
+
+/** searchByVector / deleteEmbeddings 的 where 列白名单（含全量删除前置查询） */
+export const ALLOWED_FILTER_COLUMNS = ["kbId", "documentId", "designPageId"] as const;
+
+/** insertEmbedding 的 extraFields 键白名单（键名会拼进 SQL 列名，必须白名单） */
+export const ALLOWED_INSERT_COLUMNS = [...ALLOWED_FILTER_COLUMNS, "chunkIndex"] as const;

package/src/lib/vector-store/pg-vector-store.ts

@@ -17,9 +17,11 @@
 
 import { prisma } from "@allbluecn/database";
 import type { VectorStore, VectorSearchResult } from "./types";
-
-const ALLOWED_TABLES = ["design_pages", "components", "blocks", "design_embeddings", "knowledge_embeddings"] as const;
-const ALLOWED_DELETE_COLUMNS = ["kbId", "documentId", "designPageId"];
+import {
+  ALLOWED_TABLES,
+  ALLOWED_FILTER_COLUMNS,
+  ALLOWED_INSERT_COLUMNS,
+} from "./allowlist";
 
 function validateTable(table: string): void {
   if (!(ALLOWED_TABLES as readonly string[]).includes(table)) {
@@ -31,8 +33,7 @@ export const pgVectorStore: VectorStore = {
   async searchByVector(embedding, table, limit = 10, whereColumn, whereValue): Promise<VectorSearchResult[]> {
     validateTable(table);
 
-    const allowedColumns = ["kbId", "designPageId", "documentId"];
-    if (whereColumn && !(allowedColumns as string[]).includes(whereColumn)) {
+    if (whereColumn && !(ALLOWED_FILTER_COLUMNS as readonly string[]).includes(whereColumn)) {
       throw new Error(`非法过滤列: ${whereColumn}`);
     }
 
@@ -70,6 +71,12 @@ export const pgVectorStore: VectorStore = {
     }
 
     const keys = Object.keys(extraFields);
+    // 键名直接拼进 SQL 列名：不在白名单即拒绝（SQL 标识符注入面，值已参数化但键没有）
+    for (const k of keys) {
+      if (!(ALLOWED_INSERT_COLUMNS as readonly string[]).includes(k)) {
+        throw new Error(`非法插入列: ${k}`);
+      }
+    }
     const values = Object.values(extraFields);
     const cols = keys.map(k => `"${k}"`).join(", ");
     const placeholders = values.map((_, i) => `$${i + 1}`).join(", ");
@@ -83,7 +90,7 @@ export const pgVectorStore: VectorStore = {
 
   async deleteEmbeddings(table, whereColumn, whereValue): Promise<void> {
     validateTable(table);
-    if (!(ALLOWED_DELETE_COLUMNS as readonly string[]).includes(whereColumn)) {
+    if (!(ALLOWED_FILTER_COLUMNS as readonly string[]).includes(whereColumn)) {
       throw new Error(`非法删除过滤列: ${whereColumn}`);
     }
     await prisma.$queryRawUnsafe(`DELETE FROM "${table}" WHERE "${whereColumn}" = $1`, whereValue);