Add files via upload

core/super-memory/manager.js

@@ -0,0 +1,203 @@
+import { extension_settings, getContext } from "/scripts/extensions.js";
+import { extensionName } from "../../utils/settings.js";
+import { amilyHelper } from "../tavern-helper/main.js";
+import { generateIndex } from "./smart-indexer.js";
+import { syncToLorebook, ensureMemoryBook, updateTransientHint } from "./lorebook-bridge.js";
+import { getMemoryState, loadMemoryState, saveMemoryState } from "../table-system/manager.js";
+import { eventSource, event_types } from "/script.js";
+
+let isInitialized = false;
+let updateQueue = [];
+let isProcessing = false;
+let lastChatId = null;
+
+const METADATA_KEY = 'Amily2_Memory_Data';
+
+export async function initializeSuperMemory() {
+    const settings = extension_settings[extensionName] || {};
+    if (settings.super_memory_enabled === false) {
+        console.log('[Amily2-SuperMemory] 功能已禁用 (super_memory_enabled = false)。');
+        if (window.$) $('#sm-system-status').text('已禁用').css('color', 'gray');
+        return;
+    }
+
+    if (isInitialized) return;
+    console.log('[Amily2-SuperMemory] 初始化核心管理器...');
+    
+    if (!amilyHelper) {
+        console.error('[Amily2-SuperMemory] 致命错误：AmilyHelper 未就绪。');
+        return;
+    }
+
+    document.addEventListener('AMILY2_TABLE_UPDATED', handleTableUpdate);
+    
+    eventSource.on(event_types.CHAT_CHANGED, async () => {
+        const settings = extension_settings[extensionName] || {};
+        if (settings.super_memory_enabled === false) return;
+
+        console.log('[Amily2-SuperMemory] 检测到聊天切换，正在刷新记忆状态...');
+        await checkWorldBookStatus();
+        
+        await tryRestoreStateFromMetadata();
+        
+        await forceSyncAll();
+    });
+    
+    await checkWorldBookStatus();
+    
+    await tryRestoreStateFromMetadata();
+    
+    await forceSyncAll(); 
+
+    isInitialized = true;
+    console.log('[Amily2-SuperMemory] 核心管理器初始化完成。');
+    
+    if (window.$) {
+        $('#sm-system-status').text('运行中').css('color', '#4caf50');
+    }
+}
+
+async function checkWorldBookStatus() {
+    try {
+        await ensureMemoryBook();
+    } catch (error) {
+        console.error('[Amily2-SuperMemory] 检查世界书状态失败:', error);
+    }
+}
+
+function handleTableUpdate(event) {
+    const settings = extension_settings[extensionName] || {};
+    if (settings.super_memory_enabled === false) return;
+
+    const { tableName, data, role, hint, headers, rowStatuses } = event.detail; 
+    console.log(`[Amily2-SuperMemory] 检测到表格更新: ${tableName} (Role: ${role})`);
+    
+    updateQueue.push({ tableName, data, role, hint, headers, rowStatuses });
+    processQueue();
+}
+
+async function processQueue() {
+    if (isProcessing || updateQueue.length === 0) return;
+    isProcessing = true;
+
+    try {
+        while (updateQueue.length > 0) {
+            const task = updateQueue.shift();
+            await processUpdateTask(task);
+        }
+        
+        await saveStateToMetadata();
+        
+    } catch (error) {
+        console.error('[Amily2-SuperMemory] 处理更新队列失败:', error);
+    } finally {
+        isProcessing = false;
+    }
+}
+
+async function processUpdateTask(task) {
+    const { tableName, data, role, hint, headers, rowStatuses } = task;
+    
+    const activeData = data.filter((_, i) => !rowStatuses || rowStatuses[i] !== 'pending-deletion');
+    const indexText = generateIndex(activeData, role, tableName);
+    
+    const allTables = getMemoryState();
+    const tableIndex = allTables.findIndex(t => t.name === tableName);
+    const depth = 8001 + (tableIndex >= 0 ? tableIndex : 99);
+
+    await syncToLorebook(tableName, data, indexText, role, headers, rowStatuses, depth);
+
+    if (hint) {
+        console.log(`[Amily2-SuperMemory] 应用主动记忆提示: ${hint}`);
+        await updateTransientHint(hint);
+    }
+    
+    console.log(`[Amily2-SuperMemory] 任务完成: ${tableName}`);
+    
+    updateDashboardCounters();
+}
+
+async function saveStateToMetadata() {
+    const context = getContext();
+    if (!context.chat || context.chat.length === 0) return;
+
+    const lastMsgIndex = context.chat.length - 1;
+    const lastMsg = context.chat[lastMsgIndex];
+    
+    const currentState = getMemoryState();
+    
+    if (!lastMsg.metadata) lastMsg.metadata = {};
+    
+    lastMsg.metadata[METADATA_KEY] = JSON.parse(JSON.stringify(currentState));
+    
+    if (context.saveChat) {
+        await context.saveChat(); 
+    }
+    
+    console.log(`[Amily2-SuperMemory] 状态已保存至消息 #${lastMsgIndex}`);
+}
+
+export async function tryRestoreStateFromMetadata() {
+    const context = getContext();
+    if (!context.chat || context.chat.length === 0) return;
+
+    let foundState = null;
+    let foundIndex = -1;
+
+    for (let i = context.chat.length - 1; i >= 0; i--) {
+        const msg = context.chat[i];
+        if (msg.metadata && msg.metadata[METADATA_KEY]) {
+            foundState = msg.metadata[METADATA_KEY];
+            foundIndex = i;
+            break;
+        }
+    }
+
+    if (foundState) {
+        console.log(`[Amily2-SuperMemory] 发现历史状态 (Msg #${foundIndex})，正在恢复...`);
+        if (typeof loadMemoryState === 'function') {
+            loadMemoryState(foundState);
+            await forceSyncAll();
+        } else {
+            console.warn('[Amily2-SuperMemory] table-system 缺少 loadMemoryState 方法，无法恢复状态。');
+        }
+    } else {
+        console.log('[Amily2-SuperMemory] 未在聊天记录中发现历史状态，使用默认/当前状态。');
+    }
+}
+
+function updateDashboardCounters() {
+    const tables = getMemoryState();
+    if (tables && window.$) {
+        $('#sm-index-count').text(`${tables.length} 个索引`);
+        const totalRows = tables.reduce((acc, t) => acc + (t.rows ? t.rows.length : 0), 0);
+        $('#sm-detail-count').text(`${totalRows} 个详情`);
+    }
+}
+
+export async function forceSyncAll() {
+    console.log('[Amily2-SuperMemory] 正在执行全量同步...');
+    const tables = getMemoryState();
+    
+    if (!tables || tables.length === 0) {
+        console.warn('[Amily2-SuperMemory] 没有可同步的表格数据。');
+        return;
+    }
+
+    for (const table of tables) {
+        let role = 'database';
+        if (table.name.includes('时空') || table.name.includes('世界钟')) role = 'anchor';
+        if (table.name.includes('日志') || table.name.includes('Log')) role = 'log';
+
+        updateQueue.push({
+            tableName: table.name,
+            data: table.rows,
+            headers: table.headers, 
+            rowStatuses: table.rowStatuses || [], 
+            role: role
+        });
+    }
+    
+    await processQueue();
+    console.log('[Amily2-SuperMemory] 全量同步完成。');
+}