ST-Amily2-Chat-Optimisation/core/super-memory/lorebook-bridge.js

import { amilyHelper } from "../tavern-helper/main.js";
import { extension_settings, getContext } from "/scripts/extensions.js";
import { extensionName } from "../../utils/settings.js";
import { this_chid, characters } from "/script.js";
import { withLoreLock } from "../lore-service.js";

export function getMemoryBookName() {
    let charName = "Global";
    const context = getContext();
    
    if (this_chid !== undefined && characters[this_chid]) {
        charName = characters[this_chid].name;
    } else if (context.characterId !== undefined && characters[context.characterId]) {
        charName = characters[context.characterId].name;
    }
    
    const safeCharName = charName.replace(/[<>:"/\\|?*]/g, '_');
    return `Amily2_Memory_${safeCharName}`;
}

/** 无锁内核：在已持有写锁时调用（避免嵌套死锁） */
async function _doEnsureBook(bookName) {
    const books = await amilyHelper.getLorebooks();
    if (!books.includes(bookName)) {
        console.log(`[Amily2-Bridge] 创建角色专用世界书: ${bookName}`);
        await amilyHelper.createLorebook(bookName);
    }
    const settings = extension_settings[extensionName] || {};
    const shouldBind = settings.superMemory_autoBind === true;
    if (shouldBind && bookName.startsWith("Amily2_Memory_") && bookName !== "Amily2_Memory_Global") {
        console.log(`[Amily2-Bridge] 自动绑定世界书到当前角色...`);
        await amilyHelper.bindLorebookToCharacter(bookName);
    } else if (!shouldBind) {
        console.log(`[Amily2-Bridge] 跳过自动绑定 (设置已禁用)。请手动在世界书管理中激活: ${bookName}`);
    }
}

/**
 * @param {Object} [opts]
 * @param {number}  [opts.depth=100]           详情条目的注入深度
 * @param {boolean} [opts.isIndexConstant=true] 索引条目是否常驻（蓝灯）
 * @param {boolean} [opts.pinFirstRow=false]    首行详情条目升为常驻（总调/全局定义行）
 */
export async function syncToLorebook(tableName, data, indexText, role, headers, rowStatuses, opts = {}) {
    const {
        depth = 100,
        isIndexConstant = true,
        pinFirstRow = false,
    } = opts;
    console.log(`[Amily2-Bridge] 开始同步表格: ${tableName} (Depth: ${depth}, IndexConstant: ${isIndexConstant}, PinFirstRow: ${pinFirstRow})`);
    return withLoreLock(`syncToLorebook(${tableName})`, async () => {
    await _doEnsureBook(getMemoryBookName());

    const bookName = getMemoryBookName();

    let entries = await amilyHelper.getLorebookEntries(bookName);
    if (!entries) entries = [];

    const entriesToUpdate = [];
    const entriesToCreate = [];

    const arraysEqual = (a, b) => {
        if (a === b) return true;
        if (a == null || b == null) return false;
        if (a.length !== b.length) return false;
        const sA = [...a].sort();
        const sB = [...b].sort();
        return sA.every((val, index) => val === sB[index]);
    };

    const processEntry = (comment, keys, content, type = 'selective', enabled = true, excludeRecursion = false, specificOrder = null, specificDepth = null) => {
        const existingEntry = entries.find(e => e.comment === comment);
        if (existingEntry) {
            let isChanged = false;
            
            if (existingEntry.content !== content) isChanged = true;
            if (!arraysEqual(existingEntry.key, keys)) isChanged = true;
            if (existingEntry.enabled !== enabled) isChanged = true;
            
            const shouldBeConstant = (type === 'constant');
            if (!!existingEntry.constant !== shouldBeConstant) isChanged = true;
            
            if (!!existingEntry.exclude_recursion !== excludeRecursion) isChanged = true;
            
            if (specificOrder !== null && existingEntry.order !== specificOrder) isChanged = true;
            if (specificDepth !== null && existingEntry.depth !== specificDepth) isChanged = true;

            if (isChanged) {
                existingEntry.content = content;
                existingEntry.key = keys;

                existingEntry.exclude_recursion = excludeRecursion; 
                existingEntry.prevent_recursion = excludeRecursion;
                existingEntry.excludeRecursion = excludeRecursion; 
                existingEntry.preventRecursion = excludeRecursion; 

                if (specificOrder !== null) {
                    existingEntry.order = specificOrder;
                    existingEntry.position = 4; 
                }
                if (specificDepth !== null) {
                    existingEntry.depth = specificDepth; 
                }
                
                if (type === 'constant') {
                    existingEntry.constant = true;
                } else {
                    existingEntry.constant = false;
                }
                existingEntry.enabled = enabled;
                delete existingEntry.disable;
                delete existingEntry.disabled;
                
                entriesToUpdate.push(existingEntry);
            }
        } else {
            entriesToCreate.push({
                comment: comment,
                keys: keys, 
                content: content,
                type: type, 
                position: 4,
                order: specificOrder !== null ? specificOrder : depth, 
                depth: specificDepth !== null ? specificDepth : depth,
                enabled: enabled,
                exclude_recursion: excludeRecursion, 
                prevent_recursion: excludeRecursion,
                excludeRecursion: excludeRecursion, 
                preventRecursion: excludeRecursion 
            });
        }
    };

    if (role === 'database') {
        const indexKey = [`Amily_Index_${tableName}`,'索引'];
        const indexComment = `[Amily2] Index for ${tableName}`;
        let indexContent = `【${tableName} 索引】\n${indexText.trim()}`;

        if (tableName.includes("大纲")) {
            indexContent += `\n<system_note>\n【Amily2-前瞻性预测指令】\n为了实现“提前一轮触发绿灯世界书”的高级记忆回流，请在生成本轮回复时，根据当前剧情走向，**大胆预测并自然提及**下一轮可能出现的索引编码。可自行增加一个标签并使用<!--到-->html注释掉该编码。这将激活系统的预加载机制，确保剧情发展的连贯性。\n</system_note>`;
        }

        const indexType = isIndexConstant ? 'constant' : 'selective';
        processEntry(indexComment, indexKey, indexContent, indexType, true, true, 0, 0);
    }

    const settings = extension_settings[extensionName] || {};
    const optimizationEnabled = settings.context_optimization_enabled !== false;

    const renderRowContent = (row) => {
        let finalHeaders = headers;
        if (!finalHeaders || finalHeaders.length < row.length) {
            finalHeaders = [];
            for (let i = 0; i < row.length; i++) {
                finalHeaders.push((headers && headers[i]) ? headers[i] : `Col_${i}`);
            }
        }

        let entryContent;
        if (optimizationEnabled) {
            const primaryVal = row[0] || 'Unknown';
            entryContent = `【${tableName}档案: ${primaryVal}】\n`;
            for (let i = 0; i < row.length; i++) {
                const key = finalHeaders[i] || `Col_${i}`;
                const val = row[i] || '';
                entryContent += `- ${key}: ${val}\n`;
            }
        } else {
            let textContent = `【${tableName} 详情】\n`;
            for (let i = 0; i < row.length; i++) {
                const key = finalHeaders[i] || `Col_${i}`;
                const val = row[i] || '';
                textContent += `- ${key}: ${val}\n`;
            }
            entryContent = textContent.trim();
        }
        return entryContent.trim();
    };

    data.forEach((row, index) => {
        if (!row || row.length === 0) return;

        const rawVal = row[0];
        if (rawVal === undefined || rawVal === null) return;

        const primaryVal = String(rawVal).trim();
        if (primaryVal === '') return;

        const isPendingDeletion = rowStatuses && rowStatuses[index] === 'pending-deletion';
        const isEnabled = !isPendingDeletion;

        const triggerKeys = [primaryVal];
        const entryComment = `[Amily2] Detail: ${tableName} - ${primaryVal}`;

        // 首行常驻：第一行通常是总调/全局定义（基调、主线目标等），
        // 绿灯模式下没人提到其主键就永远不注入；开启后升为蓝灯常驻。
        const entryType = (pinFirstRow && index === 0) ? 'constant' : 'selective';

        processEntry(entryComment, triggerKeys, renderRowContent(row), entryType, isEnabled);
    });

    const entriesToDelete = [];
    const tablePrefix = `[Amily2] Detail: ${tableName} -`;

    const activeKeys = new Set();
    for(const row of data) {
        if(row && row.length > 0) {
            const rVal = row[0];
            if (rVal !== undefined && rVal !== null) {
                const sVal = String(rVal).trim();
                if (sVal !== '') {
                    activeKeys.add(sVal);
                }
            }
        }
    }

    console.log(`[Amily2-Bridge-GC] ${tableName} 的活跃主键 (Active Keys):`, Array.from(activeKeys));

    for (const entry of entries) {
        if (entry.comment && entry.comment.startsWith(tablePrefix)) {
            const entryKey = entry.comment.substring(tablePrefix.length).trim();
            
            if (!activeKeys.has(entryKey)) {
                console.log(`[Amily2-Bridge-GC] 发现残留条目 (将删除): ${entry.comment} (Key: ${entryKey})`);
                entriesToDelete.push(entry.uid);
            }
        }
    }

    if (entriesToDelete.length > 0) {
        console.log(`[Amily2-Bridge] 清理 ${entriesToDelete.length} 个废弃条目...`);
        await amilyHelper.deleteLorebookEntries(bookName, entriesToDelete);
    }

    if (entriesToUpdate.length > 0) {
        console.log(`[Amily2-Bridge] 更新 ${entriesToUpdate.length} 个条目...`);
        await amilyHelper.setLorebookEntries(bookName, entriesToUpdate);
    }
    
    if (entriesToCreate.length > 0) {
        console.log(`[Amily2-Bridge] 创建 ${entriesToCreate.length} 个新条目...`);
        await amilyHelper.createLorebookEntries(bookName, entriesToCreate);
    }

    if (entriesToDelete.length === 0 && entriesToUpdate.length === 0 && entriesToCreate.length === 0) {
        console.log(`[Amily2-Bridge] ${tableName} 无需变更 (数据一致)。`);
    }

    console.log(`[Amily2-Bridge] 同步完成: ${tableName}`);
    }); // end withLoreLock
}

export async function ensureMemoryBook() {
    const bookName = getMemoryBookName();
    return withLoreLock(`ensureMemoryBook(${bookName})`, () => _doEnsureBook(bookName));
}

function createEntryTemplate() {
    return {
        uid: Date.now() + Math.floor(Math.random() * 1000),
        key: [],
        keysecondary: [],
        comment: "",
        content: "",
        constant: false,
        selective: true,
        order: 100,
        position: 1, 
        enabled: true
    };
}

export async function updateTransientHint(hint) {
    console.log('[Amily2-Bridge] 更新瞬时记忆提示...');
    await ensureMemoryBook();
    const bookName = getMemoryBookName();

    const comment = "[Amily2] Active Memory Hint";
    const content = hint ? `\n<system_note>\n【重要记忆回响】\n${hint}\n</system_note>\n` : "";
    const enabled = !!hint;

    let entries = await amilyHelper.getLorebookEntries(bookName);
    if (!entries) entries = [];

    const existingEntry = entries.find(e => e.comment === comment);

    if (existingEntry) {
        existingEntry.content = content;
        existingEntry.enabled = enabled;
        existingEntry.order = 0; 
        existingEntry.constant = true;
        
        await amilyHelper.setLorebookEntries(bookName, [existingEntry]);
    } else if (hint) {
        const newEntry = {
            comment: comment,
            keys: [],
            content: content,
            constant: true,
            selective: false,
            order: 0,
            position: 0, 
            enabled: true
        };
        await amilyHelper.createLorebookEntries(bookName, [newEntry]);
    }
    
    console.log(`[Amily2-Bridge] 瞬时记忆提示已${enabled ? '启用' : '清除'}。`);
}