import { spawn } from 'child_process';
import { join } from 'path';
import { schemas } from './tools.js';
export class ToolHandlers {
constructor(scriptsPath) {
this.scriptsPath = scriptsPath;
}
executePython(scriptName, args = [], timeout = 30000) {
return new Promise((resolve) => {
const scriptPath = join(this.scriptsPath, scriptName);
const pythonCmd = process.env.EXCEL_MCP_PYTHON || 'python';
const python = spawn(pythonCmd, [scriptPath, ...args], {
env: { ...process.env, PYTHONIOENCODING: 'utf-8' }
});
let output = '';
let error = '';
let isResolved = false;
// Set timeout for long-running operations
const timeoutId = setTimeout(() => {
if (!isResolved) {
isResolved = true;
python.kill('SIGTERM');
resolve({
content: [{ type: 'text', text: `{"error": "Script execution timeout after ${timeout}ms"}` }]
});
}
}, timeout);
python.stdout.setEncoding('utf8');
python.stderr.setEncoding('utf8');
python.stdout.on('data', (data) => output += data);
python.stderr.on('data', (data) => error += data);
python.on('close', (code) => {
if (!isResolved) {
isResolved = true;
clearTimeout(timeoutId);
const result = output.trim() || (code !== 0 ? `{"error": "${error || 'Script failed'}"}` : '{"error": "No output"}');
resolve({
content: [{ type: 'text', text: result }]
});
}
});
python.on('error', (err) => {
if (!isResolved) {
isResolved = true;
clearTimeout(timeoutId);
resolve({
content: [{ type: 'text', text: `{"error": "Python process error: ${err.message}"}` }]
});
}
});
});
}
async getOpenWorkbooks() {
return this.executePython('get_open_workbooks.py');
}
async setActiveWorkbook(args) {
const validated = schemas.setActiveWorkbook.parse(args);
return this.executePython('set_active_workbook.py', ['--workbook', validated.workbookName]);
}
async executeVBA(args) {
const validated = schemas.fallbackExecuteVBA.parse(args);
const { vbaCode, workbookName, moduleName = 'TempModule', procedureName = 'Main', sheetName } = validated;
const pythonArgs = [
'--code', vbaCode,
'--module', moduleName,
'--procedure', procedureName,
'--filename', workbookName
];
if (sheetName) pythonArgs.push('--sheet', sheetName);
return this.executePython('execute_vba.py', pythonArgs);
}
async getExcelStatus() {
const result = await this.executePython('check_excel.py');
// Add AI agent instructions to successful responses
if (result.content && result.content[0] && result.content[0].text) {
try {
const jsonResponse = JSON.parse(result.content[0].text);
// If no error in response and status is success/ready, add AI instructions
if (!jsonResponse.error && (jsonResponse.status === 'ready' || jsonResponse.status === 'success')) {
jsonResponse.ai_instructions = "Actively use essential_inspect_excel_data and essential_check_excel_format tools for comprehensive Excel analysis and validation. When editing sheets, always check with essential_inspect_excel_data first to understand the current structure and content.";
}
result.content[0].text = JSON.stringify(jsonResponse);
} catch (e) {
// If not valid JSON, return as is
}
}
return result;
}
async getAllSheetNames(args) {
const validated = schemas.getAllSheetNames.parse(args);
const result = await this.executePython('get_sheet_names.py', ['--filename', validated.workbookName]);
// Add AI agent instructions to successful responses
if (result.content && result.content[0] && result.content[0].text) {
try {
const jsonResponse = JSON.parse(result.content[0].text);
// If no error in response and status is success, add AI instructions
if (!jsonResponse.error && jsonResponse.status === 'success') {
jsonResponse.ai_instructions = "Actively use essential_inspect_excel_data and essential_check_excel_format tools for comprehensive Excel analysis and validation.";
}
result.content[0].text = JSON.stringify(jsonResponse);
} catch (e) {
// If not valid JSON, return as is
}
}
return result;
}
async navigateToSheet(args) {
const validated = schemas.navigateToSheet.parse(args);
const pythonArgs = ['--sheet', validated.sheetName, '--filename', validated.workbookName];
return this.executePython('navigate_to_sheet.py', pythonArgs);
}
async editCells(args) {
const validated = schemas.editCells.parse(args);
const pythonArgs = ['--range', validated.range];
const valueStr = typeof validated.value === 'object' ? JSON.stringify(validated.value) : String(validated.value);
pythonArgs.push('--value', valueStr);
pythonArgs.push('--filename', validated.workbookName);
if (validated.sheetName) pythonArgs.push('--sheet', validated.sheetName);
// Use longer timeout for large data operations
return this.executePython('edit_cells.py', pythonArgs, 60000);
}
async getCellFormats(args) {
const validated = schemas.finalVerifyLayoutFormats.parse(args);
const pythonArgs = [];
if (validated.startRow) pythonArgs.push('--start-row', String(validated.startRow));
if (validated.startCol) pythonArgs.push('--start-col', String(validated.startCol));
if (validated.endRow) pythonArgs.push('--end-row', String(validated.endRow));
if (validated.endCol) pythonArgs.push('--end-col', String(validated.endCol));
pythonArgs.push('--filename', validated.workbookName);
if (validated.sheetName) pythonArgs.push('--sheet', validated.sheetName);
return this.executePython('get_cell_formats.py', pythonArgs);
}
async analyzeExcelData(args) {
const validated = schemas.firstAnalyzeExcelData.parse(args);
const pythonArgs = [];
// Add file path or workbook name
if (validated.filePath) {
pythonArgs.push('--file', validated.filePath);
} else if (validated.workbookName) {
pythonArgs.push('--workbook', validated.workbookName);
}
// Add optional parameters
if (validated.sheetName) pythonArgs.push('--sheet', validated.sheetName);
if (validated.startRow) pythonArgs.push('--start-row', String(validated.startRow));
if (validated.endRow) pythonArgs.push('--end-row', String(validated.endRow));
if (validated.mode) pythonArgs.push('--mode', validated.mode);
return this.executePython('analyze_excel_data.py', pythonArgs);
}
async setCellBorders(args) {
const validated = schemas.setCellBorders.parse(args);
const pythonArgs = ['--range', validated.range];
pythonArgs.push('--borders', JSON.stringify(validated.borders));
pythonArgs.push('--filename', validated.workbookName);
if (validated.sheetName) pythonArgs.push('--sheet', validated.sheetName);
return this.executePython('set_cell_borders.py', pythonArgs);
}
async setCellFormats(args) {
const validated = schemas.setCellFormats.parse(args);
const pythonArgs = ['--range', validated.range];
pythonArgs.push('--format', JSON.stringify(validated.format));
pythonArgs.push('--filename', validated.workbookName);
if (validated.sheetName) pythonArgs.push('--sheet', validated.sheetName);
return this.executePython('set_cell_formats.py', pythonArgs);
}
}
