slvccans/ST-Amily2-Chat-Optimisation-SecFix
1
0
Fork 0
mirror of https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git synced 2026-06-06 17:05:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update index.js

Browse Source
This commit is contained in:
Wx-2025
2025-10-19 01:39:27 +08:00
committed by GitHub
parent 6a5168f8ed
commit ea3e82cd6b
1 changed files with 9 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
glossary/index.js
View File

@@ -1,18 +1,7 @@
import { executeNovelProcessing } from './executor.js'; import { executeNovelProcessing } from './executor.js';
let novelText = null;
let recognizedChaptersList = [];
const getNovelFileInput = () => document.getElementById('novel-file-input');
const getChapterRegexInput = () => document.getElementById('novel-chapter-regex');
const getRecognizeBtn = () => document.getElementById('novel-recognize-chapters');
const getProcessBtn = () => document.getElementById('novel-confirm-and-process'); const getProcessBtn = () => document.getElementById('novel-confirm-and-process');
const getChapterPreview = () => document.getElementById('novel-chapter-preview');
const getChapterCount = () => document.getElementById('novel-chapter-count');
const getStatusDisplay = () => document.getElementById('novel-process-status'); const getStatusDisplay = () => document.getElementById('novel-process-status');
const getPresetSelect = () => document.getElementById('novel-preset-select');
const getBatchSizeInput = () => document.getElementById('novel-batch-size');
const getForceNewCheckbox = () => document.getElementById('novel-force-new');
export function updateStatus(message, type = 'info') { export function updateStatus(message, type = 'info') {
const statusDisplay = getStatusDisplay(); const statusDisplay = getStatusDisplay();
@@ -22,105 +11,31 @@ export function updateStatus(message, type = 'info') {
} }
} }
function resetChapterUI() { export function handleFileUpload(file, callback) {
const preview = getChapterPreview();
const count = getChapterCount();
const processBtn = getProcessBtn();
if (preview) preview.innerHTML = '<small>请先上传文件并识别章节...</small>';
if (count) count.textContent = '0';
if (processBtn) processBtn.disabled = true;
recognizedChaptersList = [];
}
export function handleFileUpload(file) {
if (!file || !file.type.startsWith('text/')) { if (!file || !file.type.startsWith('text/')) {
updateStatus('请选择一个有效的 .txt 文件。', 'error'); updateStatus('请选择一个有效的 .txt 文件。', 'error');
return; return;
} }
const reader = new FileReader(); const reader = new FileReader();
reader.onload = (event) => { reader.onload = (event) => {
novelText = event.target.result; const content = event.target.result;
updateStatus(`文件 "${file.name}" 已成功加载。请点击“识别章节”。`, 'success'); updateStatus(`文件 "${file.name}" 已成功加载。`, 'success');
resetChapterUI(); if (callback) {
callback(content);
}
}; };
reader.onerror = () => { reader.onerror = () => {
updateStatus(`读取文件 "${file.name}" 时发生错误。`, 'error'); updateStatus(`读取文件 "${file.name}" 时发生错误。`, 'error');
novelText = null;
}; };
reader.readAsText(file); reader.readAsText(file);
} }
export function recognizeChapters() { export async function processNovel(processingState) {
if (!novelText) {
updateStatus('请先上传一个小说文件。', 'error');
return;
}
const regexInput = getChapterRegexInput();
const customRegex = regexInput.value.trim();
const defaultRegex = '(^\\s*(?:(?:第|卷)\\s*[一二三四五六七八九十百千万零〇\\d]+\\s*[章回节部篇]|Chapter\\s+\\d+|\\d+\\s*[.、]|序章|楔子|引子|序幕|尾声|终章|后记|番外)\\s*.*)';
let finalRegex;
try { try {
finalRegex = new RegExp(customRegex || defaultRegex, 'gm'); return await executeNovelProcessing(processingState, updateStatus);
} catch (e) {
updateStatus('无效的正则表达式。', 'error');
return;
}
updateStatus('正在识别章节...', 'info');
recognizedChaptersList = [];
const matches = [...novelText.matchAll(finalRegex)];
if (matches.length > 0) {
for (let i = 0; i < matches.length; i++) {
const currentMatch = matches[i];
const nextMatch = matches[i + 1];
const title = currentMatch[0].trim();
const startIndex = currentMatch.index + currentMatch[0].length;
const endIndex = nextMatch ? nextMatch.index : novelText.length;
const content = novelText.substring(startIndex, endIndex).trim();
if (title) {
recognizedChaptersList.push({ title, content });
}
}
}
const preview = getChapterPreview();
const count = getChapterCount();
const processBtn = getProcessBtn();
if (preview) {
preview.innerHTML = recognizedChaptersList.map((chap, index) => `<div>${index + 1}. ${chap.title}</div>`).join('');
}
if (count) {
count.textContent = recognizedChaptersList.length;
}
if (recognizedChaptersList.length > 0) {
processBtn.disabled = false;
updateStatus(`成功识别 ${recognizedChaptersList.length} 个章节。请预览并确认。`, 'success');
} else {
updateStatus('未能识别出章节。请尝试调整正则表达式或检查文件内容。', 'error');
processBtn.disabled = true;
}
}
export async function processNovel() {
const processBtn = getProcessBtn();
processBtn.disabled = true;
try {
const batchSize = parseInt(getBatchSizeInput().value, 10);
const forceNew = getForceNewCheckbox().checked;
await executeNovelProcessing(recognizedChaptersList, batchSize, forceNew, updateStatus);
} catch (error) { } catch (error) {
console.error('处理小说时发生UI层错误:', error); console.error('处理小说时发生UI层错误:', error);
updateStatus(`处理失败: ${error.message}`, 'error'); updateStatus(`处理失败: ${error.message}`, 'error');
} finally { throw error;
processBtn.disabled = false;
} }
} }