Initial commit with CC BY-NC-ND 4.0 license

2026-06-06 06:55:51 +00:00 · 2026-02-13 09:59:19 +08:00
commit 2c31e1cbc8
140 changed files with 44625 additions and 0 deletions
--- a/core/table-system/injector.js
+++ b/core/table-system/injector.js
@@ -0,0 +1,128 @@
+
+import { setExtensionPrompt, saveChat } from '/script.js';
+import { extension_settings, getContext } from '/scripts/extensions.js';
+import { getBatchFillerFlowTemplate, convertTablesToCsvString, convertTablesToCsvStringForContentOnly, commitPendingDeletions, getMemoryState, saveStateToMessage } from './manager.js';
+import { tableSystemDefaultSettings } from './settings.js';
+import { extensionName } from '../../utils/settings.js';
+import { log } from './logger.js';
+import { renderTables } from '../../ui/table-bindings.js';
+import { updateOrInsertTableInChat } from '../../ui/message-table-renderer.js';
+
+const INJECTION_KEY = 'AMILY2_TABLE_SYSTEM';
+
+export function generateTableContent() {
+    const settings = extension_settings[extensionName] || {};
+    let injectionContent = '';
+
+    if (!settings.table_injection_enabled) {
+        return '';
+    }
+
+    try {
+
+        const fillingMode = settings.filling_mode || 'main-api'; 
+
+        if (fillingMode === 'secondary-api') {
+            const contentOnlyTemplate = "##以下内容是故事发生的剧情中提取出的内容，已经转化为表格形式呈现给你，请将以下内容作为后续剧情的一部分参考：\n{{{Amily2TableDataContent}}}";
+            const dataString = convertTablesToCsvStringForContentOnly();
+            if (dataString.trim()) {
+                injectionContent = contentOnlyTemplate.replace('{{{Amily2TableDataContent}}}', dataString);
+            }
+        } else if (fillingMode === 'optimized') {
+            const contentOnlyTemplate = "##以下内容是故事发生的剧情中提取出的内容，已经转化为表格形式呈现给你，请将以下内容作为后续剧情的一部分参考：\n{{{Amily2TableDataContent}}}";
+            const dataString = convertTablesToCsvStringForContentOnly();
+            if (dataString.trim()) {
+                injectionContent = contentOnlyTemplate.replace('{{{Amily2TableDataContent}}}', dataString);
+            }
+        }
+        else { 
+            const flowTemplate = getBatchFillerFlowTemplate();
+            const dataString = convertTablesToCsvString();
+            if (flowTemplate && dataString.trim()) {
+                injectionContent = flowTemplate.replace('{{{Amily2TableData}}}', dataString);
+            }
+        }
+
+        if (injectionContent.trim() && window.MiZheSi_Global?.isEnabled()) {
+            injectionContent = `%%AMILY2_TABLE_INJECTION%%${injectionContent}`;
+        }
+    } catch (error) {
+        console.error('[Amily2-表格内容生成器] 生成表格内容时发生错误:', error);
+        return ''; 
+    }
+
+    return injectionContent;
+}
+
+
+
+export async function injectTableData(chat, contextSize, abort, type) {
+    // 【V15.3 核心修正】将提交删除的逻辑移至此处，确保在用户发送消息时立即触发
+    try {
+        const hasDeletions = commitPendingDeletions();
+        if (hasDeletions) {
+            const context = getContext();
+            if (context.chat && context.chat.length > 0) {
+                const currentState = getMemoryState();
+                const lastMessage = context.chat[context.chat.length - 1];
+                if (saveStateToMessage(currentState, lastMessage)) {
+                    await saveChat();
+                    log('【延迟删除】已在注入前提交待删除行并永久保存状态。', 'info');
+                    renderTables();
+                    updateOrInsertTableInChat();
+                }
+            }
+        }
+    } catch (error) {
+        console.error('[Amily2-延迟删除] 在注入前提交待删除行时发生错误:', error);
+    }
+
+    if (window.AMILY2_MACRO_REPLACED === true) {
+        console.log('[Amily2-表格注入器] 检测到宏已替换，跳过传统注入。');
+        window.AMILY2_MACRO_REPLACED = false; 
+        setExtensionPrompt(INJECTION_KEY, '', 0, 0, false, 'SYSTEM'); 
+        return;
+    }
+
+    const settings = extension_settings[extensionName] || {};
+
+    if (type === 'quiet') {
+        return;
+    }
+
+
+
+    try {
+        let injectionContent = generateTableContent();
+
+        if (!settings.table_injection_enabled) {
+            setExtensionPrompt(INJECTION_KEY, '', 0, 0, false, 'SYSTEM');
+            return;
+        }
+
+        if (!injectionContent || injectionContent.trim() === '') {
+             // 理论上不会走到这里，除非宏都没了
+            setExtensionPrompt(INJECTION_KEY, '', 0, 0, false, 'SYSTEM');
+            return;
+        }
+
+        const injectionSettings = settings.injection || tableSystemDefaultSettings.injection;
+        const position = parseInt(injectionSettings.position, 10);
+        const depth = parseInt(injectionSettings.depth, 10);
+        const role = parseInt(injectionSettings.role, 10);
+
+        setExtensionPrompt(
+            INJECTION_KEY,
+            injectionContent,
+            position,
+            depth,
+            false, 
+            role
+        );
+
+        console.log(`[Amily2-表格注入器] 已成功注入表格数据 (位置: ${position}, 深度: ${depth}, 角色: ${role})。`);
+
+    } catch (error) {
+        console.error('[Amily2-表格注入器] 注入表格数据时发生错误:', error);
+    }
+}