feat: v0.4.0 四阶段迭代 — 安全加固 + 工程基线 + 架构重构 + 双 Provider 扩展
P0 安全修复: - API Key 加密存储（safeStorage 密钥链，版本化前缀，历史明文平滑兼容） - 间接提示注入防护（SecurityScanHook 工具结果深扫描，网络工具脱敏/本地工具警示分级） - error:report IPC 断链修复（渲染进程错误上报落 electron-log + 审计） - abort 信号贯通工具层（run_command/dev-tools 子进程随会话中断终止） - run_command 沙箱加固（cd 系统目录/敏感文件读取拦截 + chcp 前缀剥离防解析退化） - .env 真实生效（dotenv 回退加载，应用内配置优先） P1 工程基础: - ESLint 9 flat config + 全部 34 条存量 warnings 清零（零容忍基线） - 测试基线 118 用例 11 文件（token/文件防护/权限/沙箱/注入/命令/引擎/注册表/审计链/摘要分层） - test:electron 双模式（ELECTRON_RUN_AS_NODE 跑 Electron ABI，SQLite 套件全执行） - SessionRecorder 多会话隔离 + 9 种 TRACE 事件补全（含最终轮 iteration_end） - Provider 故障转移（重试耗尽/不可重试一次性切换 fallback + 前端通知） - MCP 真就绪（等待全部连接完成再广播 tools:ready） - SLO/HealthChecker 真实接入（60s 巡检 + 托盘状态） - CONFIG_DEFAULTS 单一来源（消除 SEED 双源漂移） P2 架构升级: - handlers.ts 1940 行拆分为 13 个 IPC 域模块（防重入注册 + 多窗口广播） - AgentEngineManager 每会话独立引擎（LRU 30 + adapter 工厂隔离 abort 信号） - TaskOrchestrator EngineProvider 改造 + abortByParent 联动中断 SubAgent - 会话摘要分层上下文（session_summaries 滚动摘要 + 截断游标清理防因果污染） - 消息编辑重发/重新生成（truncateAfter IPC + store 动作 + UI） - Markdown 导出 / WebSearch 并行抓取（并发 3）/ 记忆 TF 缓存 / 版本构建期注入 P3 能力扩展: - OpenAI Adapter（o 系列推理模型 reasoning_effort/max_completion_tokens） - Anthropic Adapter（原生 Messages API：tool_use 块/角色合并/thinking budget/图片 base64/SSE 事件机） - 设置页/Onboarding 六 Provider 全链路接入
This commit is contained in:
@@ -0,0 +1,146 @@
|
||||
/**
|
||||
* IPC Tool Handlers — 工具与确认域（P2-9 从 handlers.ts 拆分）
|
||||
*
|
||||
* 工具列表/开关、工具就绪查询（P2-9: 从 main.ts beforeLoad 迁入）、
|
||||
* 工具确认（单条/批量/拉取 pending）、自动执行设置。
|
||||
*/
|
||||
|
||||
import { ipcMain } from 'electron';
|
||||
import type { IPCContext } from './context';
|
||||
import log from 'electron-log';
|
||||
|
||||
export function registerToolHandlers(ctx: IPCContext): void {
|
||||
const { toolRegistry, confirmationHook, agentEngineManager } = ctx;
|
||||
|
||||
ipcMain.handle('tools:list', async () => {
|
||||
return toolRegistry.listAllTools().map((t) => ({
|
||||
name: t.name,
|
||||
description: t.description,
|
||||
category: t.category,
|
||||
riskLevel: t.riskLevel,
|
||||
requiresPermission: t.requiresPermission,
|
||||
enabled: t.enabled,
|
||||
}));
|
||||
});
|
||||
|
||||
ipcMain.handle('tools:toggle', async (_event, toolName: string, enabled: boolean) => {
|
||||
// M-35 修复: 校验 toolName 合法性，防止配置 key 污染
|
||||
if (typeof toolName !== 'string' || !toolName || typeof enabled !== 'boolean') {
|
||||
return { success: false, error: 'Invalid parameters' };
|
||||
}
|
||||
// 校验 toolName 在已注册工具列表中（防止写入任意配置 key）
|
||||
const allTools = toolRegistry.listAllTools();
|
||||
if (!allTools.some((t) => t.name === toolName)) {
|
||||
log.warn(`[IPC] tools:toggle rejected: unknown tool "${toolName}"`);
|
||||
return { success: false, error: `Unknown tool: ${toolName}` };
|
||||
}
|
||||
// 工具开关通过配置持久化
|
||||
ctx.configService.set(`tools.${toolName}.enabled`, enabled);
|
||||
// 同步到 ToolRegistry（立即生效）
|
||||
toolRegistry.setToolEnabled(toolName, enabled);
|
||||
// 同步到所有会话引擎的工具列表
|
||||
agentEngineManager.setToolsAll(toolRegistry.listTools());
|
||||
log.info(`Tool ${toolName} ${enabled ? 'enabled' : 'disabled'}`);
|
||||
return { success: true };
|
||||
});
|
||||
|
||||
// ===== 工具就绪查询（P2-9: 从 main.ts beforeLoad 迁入，解决事件竞态） =====
|
||||
ipcMain.handle('tools:isReady', () => ({
|
||||
ready: ctx.toolsReadyRef.ready,
|
||||
toolCount: ctx.toolsReadyRef.toolCount,
|
||||
}));
|
||||
|
||||
// ===== v0.2.0: 工具确认响应（ConfirmationDialog → 主进程） =====
|
||||
ipcMain.on('tool:confirmationResponse', (_event, data: unknown) => {
|
||||
// M-43 修复: 校验 data 结构，防止 undefined/null 导致 TypeError
|
||||
if (!data || typeof data !== 'object') {
|
||||
log.warn('[IPC] tool:confirmationResponse rejected: invalid data');
|
||||
return;
|
||||
}
|
||||
const req = data as { toolCallId?: unknown; approved?: unknown; remember?: unknown; autoExecute?: unknown };
|
||||
if (typeof req.toolCallId !== 'string' || !req.toolCallId) {
|
||||
log.warn('[IPC] tool:confirmationResponse rejected: invalid toolCallId');
|
||||
return;
|
||||
}
|
||||
if (typeof req.approved !== 'boolean') {
|
||||
log.warn('[IPC] tool:confirmationResponse rejected: invalid approved');
|
||||
return;
|
||||
}
|
||||
const remember = typeof req.remember === 'boolean' ? req.remember : false;
|
||||
const autoExecute = typeof req.autoExecute === 'boolean' ? req.autoExecute : false;
|
||||
confirmationHook.resolveConfirmation(req.toolCallId, req.approved, remember, autoExecute);
|
||||
log.info(`[CONFIRM] Tool ${req.toolCallId} ${req.approved ? 'approved' : 'denied'}${remember ? ' (remembered)' : ''}${autoExecute ? ' (autoExecute)' : ''}`);
|
||||
});
|
||||
|
||||
// ===== v0.3.2: 批量工具确认响应（并行工具调用一次性审批） =====
|
||||
ipcMain.on('tool:confirmationResponseBatch', (_event, data: unknown) => {
|
||||
if (!data || typeof data !== 'object') {
|
||||
log.warn('[IPC] tool:confirmationResponseBatch rejected: invalid data');
|
||||
return;
|
||||
}
|
||||
const req = data as {
|
||||
toolCallIds?: unknown;
|
||||
approved?: unknown;
|
||||
remember?: unknown;
|
||||
autoExecute?: unknown;
|
||||
};
|
||||
// 严格校验 toolCallIds 数组
|
||||
if (!Array.isArray(req.toolCallIds) || req.toolCallIds.length === 0) {
|
||||
log.warn('[IPC] tool:confirmationResponseBatch rejected: toolCallIds must be non-empty array');
|
||||
return;
|
||||
}
|
||||
// 每个元素必须是字符串
|
||||
for (const id of req.toolCallIds) {
|
||||
if (typeof id !== 'string' || !id) {
|
||||
log.warn('[IPC] tool:confirmationResponseBatch rejected: invalid toolCallId in array');
|
||||
return;
|
||||
}
|
||||
}
|
||||
if (typeof req.approved !== 'boolean') {
|
||||
log.warn('[IPC] tool:confirmationResponseBatch rejected: invalid approved');
|
||||
return;
|
||||
}
|
||||
const remember = typeof req.remember === 'boolean' ? req.remember : false;
|
||||
const autoExecute = typeof req.autoExecute === 'boolean' ? req.autoExecute : false;
|
||||
const resolved = confirmationHook.resolveConfirmationsBatch(
|
||||
req.toolCallIds as string[],
|
||||
req.approved,
|
||||
remember,
|
||||
autoExecute,
|
||||
);
|
||||
log.info(`[CONFIRM] Batch ${req.approved ? 'approved' : 'denied'}: ${resolved.length}/${req.toolCallIds.length} resolved${remember ? ' (remembered)' : ''}${autoExecute ? ' (autoExecute)' : ''}`);
|
||||
});
|
||||
|
||||
// ===== v0.3.2: 拉取当前所有 pending 确认 =====
|
||||
ipcMain.handle('tool:getPendingConfirmations', async () => {
|
||||
return { success: true, data: confirmationHook.getPendingConfirmations() };
|
||||
});
|
||||
|
||||
// ===== v0.2.0: 持久化自动执行设置 =====
|
||||
ipcMain.handle('tool:setAutoExecute', async (_event, toolName: unknown, enabled: unknown) => {
|
||||
// M-44 修复: 校验 toolName 合法性和 enabled 类型，防止配置 key 污染
|
||||
if (typeof toolName !== 'string' || !toolName) {
|
||||
return { success: false, error: 'Invalid toolName' };
|
||||
}
|
||||
if (typeof enabled !== 'boolean') {
|
||||
return { success: false, error: 'Invalid enabled (must be boolean)' };
|
||||
}
|
||||
// 校验 toolName 在已注册工具列表中（与 tools:toggle 保持一致）
|
||||
const allTools = toolRegistry.listAllTools();
|
||||
if (!allTools.some((t) => t.name === toolName)) {
|
||||
log.warn(`[IPC] tool:setAutoExecute rejected: unknown tool "${toolName}"`);
|
||||
return { success: false, error: `Unknown tool: ${toolName}` };
|
||||
}
|
||||
try {
|
||||
confirmationHook.setAutoExecute(toolName, enabled);
|
||||
log.info(`[CONFIRM] Tool ${toolName} autoExecute set to ${enabled}`);
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
return { success: false, error: (error as Error).message };
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle('tool:getAutoExecuteList', async () => {
|
||||
return { success: true, data: confirmationHook.getAutoExecuteList() };
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user