From 667a7827622fbea0e74efa33c45c300c437970da Mon Sep 17 00:00:00 2001
From: yyxyz0120 <295786908+yyxyz0120@users.noreply.github.com>
Date: Sat, 1 Aug 2026 00:33:09 +0800
Subject: [PATCH] feat(story-summary): add per-chat enable toggle

---
 modules/story-summary/data/chat-toggle.js     |  21 +++
 modules/story-summary/story-summary-ui.js     |  35 +++++
 modules/story-summary/story-summary.html      |   4 +
 modules/story-summary/story-summary.js        | 134 ++++++++++++++++--
 .../story-summary/tests/chat-toggle.test.js   |  35 +++++
 package.json                                  |   1 +
 settings.html                                 |   2 +-
 7 files changed, 216 insertions(+), 16 deletions(-)
 create mode 100644 modules/story-summary/data/chat-toggle.js
 create mode 100644 modules/story-summary/tests/chat-toggle.test.js

diff --git a/modules/story-summary/data/chat-toggle.js b/modules/story-summary/data/chat-toggle.js
new file mode 100644
index 00000000..cca9c32d
--- /dev/null
+++ b/modules/story-summary/data/chat-toggle.js
@@ -0,0 +1,21 @@
+export const STORY_SUMMARY_CHAT_ENABLED_KEY = "storySummaryEnabled";
+
+export function getChatStorySummaryEnabled(chatMetadata, extensionId) {
+    if (!chatMetadata || typeof chatMetadata !== "object") return true;
+    const extensionMetadata = chatMetadata.extensions?.[extensionId];
+    return extensionMetadata?.[STORY_SUMMARY_CHAT_ENABLED_KEY] !== false;
+}
+
+export function setChatStorySummaryEnabled(chatMetadata, extensionId, enabled) {
+    if (!chatMetadata || typeof chatMetadata !== "object") {
+        throw new TypeError("chatMetadata must be an object");
+    }
+    chatMetadata.extensions ||= {};
+    chatMetadata.extensions[extensionId] ||= {};
+    chatMetadata.extensions[extensionId][STORY_SUMMARY_CHAT_ENABLED_KEY] = Boolean(enabled);
+    return chatMetadata.extensions[extensionId][STORY_SUMMARY_CHAT_ENABLED_KEY];
+}
+
+export function resolveStorySummaryEnabled(globalEnabled, chatEnabled) {
+    return Boolean(globalEnabled) && chatEnabled !== false;
+}
diff --git a/modules/story-summary/story-summary-ui.js b/modules/story-summary/story-summary-ui.js
index 5e6a3beb..3ee4ca3f 100644
--- a/modules/story-summary/story-summary-ui.js
+++ b/modules/story-summary/story-summary-ui.js
@@ -341,6 +341,7 @@
     let timelineHasRenderedEvents = false;
     let currentTimelineChatId = '';
     let settingsSaveTimeoutId = null;
+    let currentChatSummaryEnabled = true;
     let panelConfigLoadedFromServer = false;
     let settingsOpenedWithServerConfig = false;
     const pendingConfigSaveRequests = new Map();
@@ -444,6 +445,27 @@
         }
     }
 
+    function syncCurrentChatSummaryControls(enabled = currentChatSummaryEnabled) {
+        currentChatSummaryEnabled = enabled !== false;
+        const toggle = $('current-chat-enabled');
+        if (toggle) {
+            toggle.checked = currentChatSummaryEnabled;
+            toggle.disabled = false;
+        }
+
+        const generateButton = $('btn-generate');
+        if (generateButton) {
+            generateButton.disabled = !currentChatSummaryEnabled;
+            generateButton.title = currentChatSummaryEnabled ? '' : '请先启用当前聊天的剧情总结';
+        }
+
+        for (const id of ['hide-summarized', 'keep-visible-count', 'use-vector-boundary']) {
+            const control = $(id);
+            if (control) control.disabled = !currentChatSummaryEnabled;
+        }
+        syncVectorBoundaryControl(config.vector?.enabled, currentChatSummaryEnabled && config.ui.hideSummarized);
+    }
+
     function syncAutoSummaryControls() {
         const en = $('trigger-enabled');
         const timing = $('trigger-timing');
@@ -2248,6 +2270,10 @@
         const btn = $('btn-generate');
 
         switch (d.type) {
+            case 'CHAT_SUMMARY_STATE':
+                syncCurrentChatSummaryControls(d.state?.effectiveEnabled !== false);
+                break;
+
             case 'GENERATION_STATE':
                 localGenerating = !!d.isGenerating;
                 btn.textContent = localGenerating ? '停止' : '总结';
@@ -2566,6 +2592,14 @@
             e.target.value = val;
         };
 
+        // Current chat switch (saved immediately in chat metadata)
+        $('current-chat-enabled').onchange = e => {
+            const enabled = e.target.checked;
+            e.target.disabled = true;
+            syncCurrentChatSummaryControls(enabled);
+            postMsg('SET_CURRENT_CHAT_ENABLED', { enabled });
+        };
+
         // Main actions
         $('btn-clear').onclick = async () => {
             const action = await showCleanActionMenu();
@@ -2705,6 +2739,7 @@
         renderFacts([]);
 
         bindEvents();
+        syncCurrentChatSummaryControls(currentChatSummaryEnabled);
 
         // === THEME SWITCHER ===
         (function () {
diff --git a/modules/story-summary/story-summary.html b/modules/story-summary/story-summary.html
index e464ca69..5338fa3c 100644
--- a/modules/story-summary/story-summary.html
+++ b/modules/story-summary/story-summary.html
@@ -88,6 +88,10 @@ <h1 style="user-select: none;">剧情<span>总结</span></h1>
 
         <!-- Controls -->
         <div class="controls">
+            <label class="chk-label current-chat-toggle" title="只影响当前聊天；关闭后停止自动总结和长期记忆注入，但保留已有数据">
+                <input type="checkbox" id="current-chat-enabled" checked>
+                <span>当前聊天启用</span>
+            </label>
             <label class="chk-label">
                 <input type="checkbox" id="hide-summarized">
                 <span>隐藏已总结 · <strong id="summarized-count">0</strong> 楼（保留<input type="number" id="keep-visible-count"
diff --git a/modules/story-summary/story-summary.js b/modules/story-summary/story-summary.js
index fc834263..5cd49437 100644
--- a/modules/story-summary/story-summary.js
+++ b/modules/story-summary/story-summary.js
@@ -8,15 +8,16 @@
 // 4) Prompt 注入：extension_prompts + IN_CHAT + depth（动态计算，最小为2）
 // ═══════════════════════════════════════════════════════════════════════════
 
-import { getContext } from "../../../../../extensions.js";
+import { getContext, saveMetadataDebounced } from "../../../../../extensions.js";
 import {
     event_types,
     extension_prompts,
     extension_prompt_types,
     extension_prompt_roles,
     getRequestHeaders,
+    chat_metadata,
 } from "../../../../../../script.js";
-import { extensionFolderPath } from "../../core/constants.js";
+import { EXT_ID, extensionFolderPath } from "../../core/constants.js";
 import { xbLog, CacheRegistry } from "../../core/debug-core.js";
 import { createModuleEvents } from "../../core/event-manager.js";
 import { postToIframe, isTrustedMessage } from "../../core/iframe-messaging.js";
@@ -39,6 +40,11 @@ import {
     saveSummaryPanelConfigVerified,
     loadConfigFromServer,
 } from "./data/config.js";
+import {
+    getChatStorySummaryEnabled,
+    resolveStorySummaryEnabled,
+    setChatStorySummaryEnabled,
+} from "./data/chat-toggle.js";
 import {
     getSummaryStore,
     saveSummaryStore,
@@ -132,6 +138,48 @@ const MESSAGE_EVENT = "message";
 const SUMMARY_MODEL_FETCH_PROVIDERS = new Set(["openai"]);
 const SUMMARY_MODEL_FETCH_TIMEOUT_MS = 5000;
 
+export function getStorySummaryChatState() {
+    const context = getContext();
+    const globalEnabled = Boolean(getSettings().storySummary?.enabled);
+    const hasChat = Boolean(context?.chatId);
+    const chatEnabled = hasChat
+        ? getChatStorySummaryEnabled(chat_metadata, EXT_ID)
+        : true;
+    return {
+        hasChat,
+        chatId: context?.chatId || null,
+        globalEnabled,
+        chatEnabled,
+        effectiveEnabled: hasChat && resolveStorySummaryEnabled(globalEnabled, chatEnabled),
+    };
+}
+
+export function isStorySummaryEnabledForCurrentChat() {
+    return getStorySummaryChatState().effectiveEnabled;
+}
+
+function notifyStorySummaryChatState() {
+    const state = getStorySummaryChatState();
+    postToFrame({ type: "CHAT_SUMMARY_STATE", state });
+    $(document).trigger("xiaobaix:storySummary:chat-state", [state]);
+}
+
+export async function setStorySummaryEnabledForCurrentChat(enabled) {
+    const context = getContext();
+    if (!context?.chatId) {
+        throw new Error("No active chat");
+    }
+
+    setChatStorySummaryEnabled(chat_metadata, EXT_ID, enabled);
+    saveMetadataDebounced?.();
+    notifyStorySummaryChatState();
+
+    if (events) {
+        await handleChatChanged();
+    }
+    return getStorySummaryChatState();
+}
+
 function compactRecallRuntimeStatsForLog(statsList = getRecallRuntimeStats()) {
     if (!Array.isArray(statsList) || !statsList.length) return "[]";
     return statsList.map((item) => {
@@ -1874,11 +1922,13 @@ async function importSummaryMemoryPackage(rawText) {
 // Compatibility export for ena-planner.
 // Returns a compact plain-text snapshot of story-summary memory.
 export function getStorySummaryForEna() {
+    if (!isStorySummaryEnabledForCurrentChat()) return "";
     const promptText = String(_lastBuiltPromptText || "").trim();
     return promptText || getStorySummaryMemoryText();
 }
 
 export function getStorySummaryMemoryText() {
+    if (!isStorySummaryEnabledForCurrentChat()) return "";
     return formatStorySummaryMemoryText(getSummaryStore());
 }
 
@@ -2040,7 +2090,7 @@ async function getHideBoundaryFloor(store) {
 }
 
 async function applyHideState({ reset = true } = {}) {
-    if (!getSettings().storySummary?.enabled) return;
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     const store = getSummaryStore();
     const ui = getHideUiSettings();
     if (!ui.hideSummarized) return;
@@ -2073,7 +2123,7 @@ function applyHideStateDebounced({ reset = false } = {}) {
     cancelHideApplyTimer();
     hideApplyTimer = setTimeout(() => {
         hideApplyTimer = null;
-        if (!getSettings().storySummary?.enabled) return;
+        if (!isStorySummaryEnabledForCurrentChat()) return;
         if (!getHideUiSettings().hideSummarized) return;
         applyHideState({ reset }).catch((e) => xbLog.warn(MODULE_ID, "applyHideState failed", e));
     }, HIDE_APPLY_DEBOUNCE_MS);
@@ -2167,7 +2217,7 @@ async function clearHideState() {
 async function maybeAutoRunSummary(reason) {
     const { chatId, chat } = getContext();
     if (!chatId || !Array.isArray(chat)) return;
-    if (!getSettings().storySummary?.enabled) return;
+    if (!isStorySummaryEnabledForCurrentChat()) return;
 
     const cfgAll = getSummaryPanelConfig();
     const trig = cfgAll.trigger || {};
@@ -2212,7 +2262,7 @@ async function autoRunSummaryWithRetry(targetMesId, configForRun) {
                         addEventDocuments(allEvents.filter(e => idSet.has(e.id)));
                     }
 
-                    if (getSettings().storySummary?.enabled && getHideUiSettings().hideSummarized) {
+                    if (isStorySummaryEnabledForCurrentChat() && getHideUiSettings().hideSummarized) {
                         applyHideStateDebounced();
                     }
                     await updateFrameStatsAfterSummary(store);
@@ -2261,6 +2311,7 @@ async function handleFrameMessage(event) {
             sendVectorConfigToFrame();
             sendVectorStatsToFrame();
             sendAnchorStatsToFrame();
+            notifyStorySummaryChatState();
             break;
         }
 
@@ -2282,7 +2333,22 @@ async function handleFrameMessage(event) {
             $(".xb-ss-close-btn").show();
             break;
 
+        case "SET_CURRENT_CHAT_ENABLED": {
+            try {
+                await setStorySummaryEnabledForCurrentChat(data.enabled !== false);
+            } catch (error) {
+                xbLog.warn(MODULE_ID, "Failed to update current chat story summary state", error);
+                notifyStorySummaryChatState();
+            }
+            break;
+        }
+
         case "REQUEST_GENERATE": {
+            if (!isStorySummaryEnabledForCurrentChat()) {
+                postToFrame({ type: "SUMMARY_STATUS", statusText: "请先启用当前聊天的剧情总结" });
+                notifyStorySummaryChatState();
+                break;
+            }
             const ctx = getContext();
             currentMesId = (ctx.chat?.length ?? 1) - 1;
             handleManualGenerate(currentMesId, data.config || {});
@@ -2763,12 +2829,19 @@ async function handleChatChanged() {
     lastRecallLogText = "";
     const { chat } = getContext();
     activeChatId = getContext().chatId || null;
+    notifyStorySummaryChatState();
+    initButtonsForAll();
+
+    if (!isStorySummaryEnabledForCurrentChat()) {
+        await deactivateCurrentChatStorySummary();
+        return;
+    }
+
     logRecallRuntimeCheckpoint("chatChanged:before-retain", `chat=${activeChatId || "-"} length=${Array.isArray(chat) ? chat.length : 0}`);
     await retainRecallRuntimeOnly(activeChatId);
     const newLength = Array.isArray(chat) ? chat.length : 0;
 
     await rollbackSummaryIfNeeded();
-    initButtonsForAll();
 
     const store = getSummaryStore();
 
@@ -2800,6 +2873,7 @@ async function handleChatChanged() {
 }
 
 async function handleMessageDeleted(scheduledChatId) {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     if (isChatStale(scheduledChatId)) return;
     const { chat, chatId } = getContext();
     const newLength = chat?.length || 0;
@@ -2823,6 +2897,7 @@ async function handleMessageDeleted(scheduledChatId) {
 }
 
 async function handleMessageSwiped(scheduledChatId) {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     if (isChatStale(scheduledChatId)) return;
     const { chat, chatId } = getContext();
     const lastFloor = (chat?.length || 1) - 1;
@@ -2845,6 +2920,7 @@ async function handleMessageSwiped(scheduledChatId) {
 }
 
 async function handleMessageReceived(scheduledChatId, targetMesId = null) {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     if (isChatStale(scheduledChatId)) return;
     const { chat, chatId } = getContext();
     const lastFloor = (chat?.length || 1) - 1;
@@ -2871,10 +2947,12 @@ async function handleMessageReceived(scheduledChatId, targetMesId = null) {
 function handleMessageSent(scheduledChatId) {
     if (isChatStale(scheduledChatId)) return;
     initButtonForLatestMessage();
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     scheduleAutoSummary("before_user");
 }
 
 async function handleMessageUpdated(scheduledChatId) {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     if (isChatStale(scheduledChatId)) return;
     await rollbackSummaryIfNeeded();
     initButtonsForAll();
@@ -2882,6 +2960,7 @@ async function handleMessageUpdated(scheduledChatId) {
 }
 
 function handleMessageRendered(data) {
+    if (!getSettings().storySummary?.enabled) return;
     const mesId = data?.element ? $(data.element).attr("mesid") : data?.messageId;
     if (mesId != null) addSummaryBtnToMessage(mesId);
     else initButtonsForAll();
@@ -2892,6 +2971,7 @@ function handleMessageRendered(data) {
 // ═══════════════════════════════════════════════════════════════════════════
 
 function handleMessageSentForRecall() {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     const { chat } = getContext();
     const lastMsg = chat?.[chat.length - 1];
     if (lastMsg?.is_user) {
@@ -2910,7 +2990,7 @@ function clearExtensionPrompt() {
 
 async function handleGenerationStarted(type, _params, isDryRun) {
     if (isDryRun) return;
-    if (!getSettings().storySummary?.enabled) return;
+    if (!isStorySummaryEnabledForCurrentChat()) return;
 
     const T0 = performance.now();
     const timing = {
@@ -3048,6 +3128,7 @@ function isChatStale(scheduledChatId) {
 }
 
 function notifyStorySummaryAfterAi(data, source) {
+    if (!isStorySummaryEnabledForCurrentChat()) return;
     const { chatId, chat } = getContext();
     if (!chatId || !Array.isArray(chat) || !chat.length) return;
 
@@ -3071,7 +3152,7 @@ function registerAfterAiGateHandler() {
     initAfterAiGate();
     if (afterAiGateDispose) return;
     afterAiGateDispose = registerAfterAiHandler(MODULE_ID, async ({ chatId, messageId }) => {
-        if (!getSettings().storySummary?.enabled) return;
+        if (!isStorySummaryEnabledForCurrentChat()) return;
         if (activeChatId !== chatId) return;
         scheduleWithChatGuard(handleMessageReceived, 0, messageId);
     });
@@ -3125,6 +3206,7 @@ function registerEvents() {
 
     events.on(event_types.CHAT_CHANGED, () => {
         activeChatId = getContext().chatId || null;
+        notifyStorySummaryChatState();
         scheduleWithChatGuard(handleChatChanged, 80);
     });
     events.on(event_types.MESSAGE_DELETED, () => scheduleWithChatGuard(handleMessageDeleted, 50));
@@ -3187,6 +3269,25 @@ function unregisterEvents() {
     window.removeEventListener("resize", handleViewportChangeForBackground);
     window.visualViewport?.removeEventListener?.("resize", handleViewportChangeForBackground);
     window.visualViewport?.removeEventListener?.("scroll", handleViewportChangeForBackground);
+    notifyStorySummaryChatState();
+}
+
+async function deactivateCurrentChatStorySummary() {
+    clearDeferredBackgroundTasks();
+    cancelHideApplyTimer();
+    clearExtensionPrompt();
+    _lastBuiltPromptText = "";
+    lastRecallLogText = "";
+
+    if (activeChatId) {
+        clearRecallRuntime(activeChatId).catch(() => {});
+    }
+
+    try {
+        await clearHideState();
+    } catch (error) {
+        xbLog.warn(MODULE_ID, "Failed to restore hidden messages while disabling this chat", error);
+    }
 }
 
 // ═══════════════════════════════════════════════════════════════════════════
@@ -3405,7 +3506,7 @@ function showBackupManagerModal(initialFiles) {
 $(document).on("xiaobaix:storySummary:toggle", async (_e, enabled) => {
     if (enabled) {
         registerEvents();
-        initButtonsForAll();
+        await handleChatChanged();
     } else {
         try {
             await clearHideState();
@@ -3414,6 +3515,7 @@ $(document).on("xiaobaix:storySummary:toggle", async (_e, enabled) => {
         }
         unregisterEvents();
     }
+    notifyStorySummaryChatState();
 });
 
 // ═══════════════════════════════════════════════════════════════════════════
@@ -3424,14 +3526,16 @@ jQuery(() => {
     window.registerModuleCleanup?.(MODULE_ID, unregisterEvents);
     if (!getSettings().storySummary?.enabled) return;
     (async () => {
-        await loadConfigFromServer();
+        try {
+            await loadConfigFromServer();
+        } catch (e) {
+            xbLog.warn(MODULE_ID, "Failed to load server config before initialization; using local cache", e);
+        }
         registerEvents();
         initStateIntegration();
         maybePreloadTokenizer();
+        await handleChatChanged();
     })().catch((e) => {
-        xbLog.warn(MODULE_ID, "初始化前加载服务端配置失败，继续使用本地缓存", e);
-        registerEvents();
-        initStateIntegration();
-        maybePreloadTokenizer();
+        xbLog.error(MODULE_ID, "Story summary initialization failed", e);
     });
 });
diff --git a/modules/story-summary/tests/chat-toggle.test.js b/modules/story-summary/tests/chat-toggle.test.js
new file mode 100644
index 00000000..62f90000
--- /dev/null
+++ b/modules/story-summary/tests/chat-toggle.test.js
@@ -0,0 +1,35 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import {
+    getChatStorySummaryEnabled,
+    resolveStorySummaryEnabled,
+    setChatStorySummaryEnabled,
+    STORY_SUMMARY_CHAT_ENABLED_KEY,
+} from "../data/chat-toggle.js";
+
+const EXT_ID = "LittleWhiteBox";
+
+test("new chats inherit the enabled default without mutating metadata", () => {
+    const metadata = {};
+
+    assert.equal(getChatStorySummaryEnabled(metadata, EXT_ID), true);
+    assert.deepEqual(metadata, {});
+});
+
+test("chat-level disabled state is persisted under extension metadata", () => {
+    const metadata = {};
+
+    assert.equal(setChatStorySummaryEnabled(metadata, EXT_ID, false), false);
+    assert.equal(
+        metadata.extensions[EXT_ID][STORY_SUMMARY_CHAT_ENABLED_KEY],
+        false,
+    );
+    assert.equal(getChatStorySummaryEnabled(metadata, EXT_ID), false);
+});
+
+test("global switch remains the master switch", () => {
+    assert.equal(resolveStorySummaryEnabled(false, true), false);
+    assert.equal(resolveStorySummaryEnabled(true, false), false);
+    assert.equal(resolveStorySummaryEnabled(true, true), true);
+});
diff --git a/package.json b/package.json
index ee84bc11..2082d30a 100644
--- a/package.json
+++ b/package.json
@@ -15,6 +15,7 @@
         "build:tavern:app": "vue-tsc --noEmit -p tsconfig.tavern.json && vite build --config vite.tavern.config.mjs",
         "build:tavern": "npm run build:tavern:material-symbols && npm run build:tavern:host && npm run build:tavern:app",
         "test:story-summary:runtime": "node scripts/story-summary-runtime-check.mjs",
+        "test:story-summary:chat-toggle": "node --test modules/story-summary/tests/chat-toggle.test.js",
         "test:story-summary:replay": "node scripts/story-summary-replay-runner.mjs",
         "test:story-summary:bootstrap": "node scripts/story-summary-replay-runner.mjs bootstrap",
         "test:story-summary:recall": "node scripts/story-summary-replay-runner.mjs recall-only",
diff --git a/settings.html b/settings.html
index e0b6937c..c2913eef 100644
--- a/settings.html
+++ b/settings.html
@@ -202,7 +202,7 @@
           <hr class="sysHR" />
           <div class="flex-container littlewhitebox-setting-row">
             <input type="checkbox" id="xiaobaix_story_summary_enabled" />
-            <label for="xiaobaix_story_summary_enabled" class="has-tooltip" data-tooltip="在消息楼层添加总结按钮，点击可打开剧情总结面板，AI分析生成关键词云、时间线、人物关系、角色弧光">剧情总结</label>
+            <label for="xiaobaix_story_summary_enabled" class="has-tooltip" data-tooltip="启用剧情总结模块；可在剧情总结面板中对当前聊天单独开关">剧情总结</label>
           </div>
           <div class="flex-container littlewhitebox-setting-row">
             <input type="checkbox" id="xiaobaix_story_outline_enabled" />