ST-Amily2-Chat-Optimisation/core/table-system/executor.js

import { log } from './logger.js';

function insertRow(state, tableIndex, data) {
    if (!state[tableIndex]) {
        log(`AI指令错误：尝试在不存在的表格索引 ${tableIndex} 中插入行。`, 'error');
        return { state, changes: [] };
    }
    const table = state[tableIndex];
    const colCount = table.headers.length;
    const newRow = Array(colCount).fill('');
    const changes = [];
    const newRowIndex = table.rows.length;

    for (const colIndex in data) {
        const cIndex = parseInt(colIndex, 10);
        if (cIndex < colCount) {
            newRow[cIndex] = data[colIndex];
            changes.push({ type: 'update', tableIndex, rowIndex: newRowIndex, colIndex: cIndex });
        }
    }
    table.rows.push(newRow);
    return { state, changes };
}

function updateRow(state, tableIndex, rowIndex, data) {
    if (!state[tableIndex]) {
        log(`AI指令错误：尝试更新不存在的表格 ${tableIndex}。`, 'error');
        return { state, changes: [] };
    }

    const table = state[tableIndex];

    if (rowIndex >= table.rows.length) {
        log(`AI指令修正：updateRow 的行索引 ${rowIndex} 超出范围，自动转换为 insertRow。`, 'warn');
        return insertRow(state, tableIndex, data);
    }

    const row = table.rows[rowIndex];
    const changes = [];
    for (const colIndex in data) {
        const cIndex = parseInt(colIndex, 10);
        if (cIndex < row.length) {
            row[cIndex] = data[colIndex];
            changes.push({ type: 'update', tableIndex, rowIndex, colIndex: cIndex });
        }
    }
    return { state, changes };
}


function deleteRow(state, tableIndex, rowIndex) {
    const table = state[tableIndex];
    if (!table || !table.rows[rowIndex]) {
        log(`AI指令错误：尝试删除不存在的表格 ${tableIndex} 或行 ${rowIndex}。`, 'error');
        return { state, changes: [] };
    }

    if (!table.rowStatuses) {
        table.rowStatuses = Array(table.rows.length).fill('normal');
    }
    
    if (table.rowStatuses[rowIndex] !== 'pending-deletion') {
        table.rowStatuses[rowIndex] = 'pending-deletion';
        const changes = [{ type: 'delete', tableIndex, rowIndex }];
        return { state, changes };
    }

    return { state, changes: [] };
}


const allowedFunctions = {
    insertRow,
    updateRow,
    deleteRow,
};

function parseFunctionCall(callString) {
    const match = callString.trim().match(/(\w+)\((.*)\)/);
    if (!match) {
        log(`指令格式错误，无法解析: "${callString}"`, 'error');
        return null;
    }

    const functionName = match[1];
    const argsString = match[2];

    if (!allowedFunctions[functionName]) {
        log(`检测到非法函数调用: "${functionName}"。已阻止执行。`, 'error');
        return null;
    }

    try {

        const args = [];
        let currentArg = '';
        let inQuote = false;
        let quoteChar = '';
        let braceDepth = 0; 
        
        for (let i = 0; i < argsString.length; i++) {
            const char = argsString[i];
            
            if ((char === '"' || char === "'") && (i === 0 || argsString[i-1] !== '\\')) {
                if (!inQuote) {
                    inQuote = true;
                    quoteChar = char;
                } else if (char === quoteChar) {
                    inQuote = false;
                }
                currentArg += char;
            } else if (!inQuote) {
                if (char === '{' || char === '[') {
                    braceDepth++;
                    currentArg += char;
                } else if (char === '}' || char === ']') {
                    braceDepth--;
                    currentArg += char;
                } else if (char === ',' && braceDepth === 0) {
                    args.push(parseValue(currentArg));
                    currentArg = '';
                } else {
                    currentArg += char;
                }
            } else {
                currentArg += char;
            }
        }
        if (currentArg.trim()) {
            args.push(parseValue(currentArg));
        }

        return { name: functionName, args: args };
    } catch (e) {
        log(`解析函数 "${functionName}" 的参数时出错: ${e.message}`, 'error');
        return null;
    }
}

function parseValue(val) {
    val = val.trim();
    if (val === 'true') return true;
    if (val === 'false') return false;
    if (val === 'null') return null;
    if (val === 'undefined') return undefined;
    if (!isNaN(Number(val)) && val !== '') return Number(val);
    
    if (val.startsWith('"') && val.endsWith('"')) {
         try { return JSON.parse(val); } catch (e) { return val.slice(1, -1); }
    }
    if (val.startsWith("'") && val.endsWith("'")) {
        return val.slice(1, -1);
    }

    if ((val.startsWith('{') && val.endsWith('}')) || (val.startsWith('[') && val.endsWith(']'))) {
        try {
            return JSON.parse(val);
        } catch (e) {
            try {
                let fixedVal = val.replace(/([{,]\s*)(\d+)(\s*:)/g, '$1"$2"$3');
                fixedVal = fixedVal.replace(/'/g, '"');
                
                return JSON.parse(fixedVal);
            } catch (e2) {
                return val;
            }
        }
    }
    return val;
}


export function executeCommands(aiResponseText, initialState) {
    const commandBlockRegex = /<Amily2Edit>([\s\S]*?)<\/Amily2Edit>/;
    const match = aiResponseText.match(commandBlockRegex);

    if (!match) {
        return { finalState: initialState, hasChanges: false, changes: [] };
    }

    log('检测到AI指令块，开始推演...', 'info');
    const commandBlock = match[1].replace(/<!--|-->/g, '').trim();
    if (!commandBlock) {
        return { finalState: initialState, hasChanges: false, changes: [] };
    }

    const commands = commandBlock.split('\n').filter(line => line.trim() !== '');
    if (commands.length === 0) {
        return { finalState: initialState, hasChanges: false, changes: [] };
    }

    let currentState = JSON.parse(JSON.stringify(initialState));
    let allChanges = [];

    commands.forEach(commandString => {
        const trimmedCommand = commandString.trim();
        if (trimmedCommand.startsWith('insertRow(') || 
            trimmedCommand.startsWith('deleteRow(') || 
            trimmedCommand.startsWith('updateRow(')) 
        {
            const parsed = parseFunctionCall(trimmedCommand);
            if (parsed) {
                try {
                    const result = allowedFunctions[parsed.name](currentState, ...parsed.args);
                    currentState = result.state;
                    if (result.changes && result.changes.length > 0) {
                        allChanges = allChanges.concat(result.changes);
                    }
                    log(`成功推演指令: ${commandString}`, 'success');
                } catch (e) {
                    log(`推演指令 "${commandString}" 时发生运行时错误: ${e.message}`, 'error');
                }
            }
        }
    });

    const hasChanges = allChanges.length > 0;
    return { finalState: currentState, hasChanges, changes: allChanges };
}