feat: 集成后端通信和前端交互功能
- 重构消息处理器(src/utils/messageHandler.ts) - 集成 DialogService 实现后端对话管理 - 添加流式消息处理和 SSE 事件监听 - 实现工具执行状态的实时更新 - 支持用户问题的交互处理 - 添加对话中止和错误处理机制 - 更新 ICHelperPanel(src/panels/ICHelperPanel.ts) - 添加 submitAnswer 消息处理,支持用户答案提交 - 添加 abortDialog 消息处理,支持对话中止 - 与后端服务进行双向通信 - 更新 ICViewProvider(src/views/ICViewProvider.ts) - 同步更新消息处理逻辑 - 添加 extensionPath 参数传递 - 支持新的消息类型和事件处理 完成前后端通信的完整集成,实现: - AI 对话的流式响应 - 工具调用的实时反馈 - 用户交互的双向通信 - 错误处理和状态管理
This commit is contained in:
@ -14,6 +14,15 @@ import {
|
||||
checkIverilogAvailable,
|
||||
} from "./iverilogRunner";
|
||||
import { ChatHistoryManager } from "./chatHistoryManager";
|
||||
import { dialogManager, DialogSession } from "../services/dialogService";
|
||||
import { userInteractionManager } from "../services/userInteraction";
|
||||
import { healthCheck } from "../services/apiClient";
|
||||
|
||||
/** 是否使用后端服务(可通过配置控制) */
|
||||
let useBackendService = true;
|
||||
|
||||
/** 当前对话会话 */
|
||||
let currentSession: DialogSession | null = null;
|
||||
|
||||
/**
|
||||
* 处理用户消息
|
||||
@ -25,33 +34,53 @@ export async function handleUserMessage(
|
||||
) {
|
||||
console.log("收到用户消息:", text);
|
||||
|
||||
// 记录用户消息到历史
|
||||
const historyManager = ChatHistoryManager.getInstance();
|
||||
await historyManager.addUserMessage(text);
|
||||
// 记录用户消息到历史(允许失败,不阻塞主流程)
|
||||
try {
|
||||
const historyManager = ChatHistoryManager.getInstance();
|
||||
await historyManager.addUserMessage(text);
|
||||
} catch (error) {
|
||||
console.warn("记录消息历史失败(可能没有打开工作区):", error);
|
||||
}
|
||||
|
||||
// 检查是否是 VCD 生成命令
|
||||
// 设置 WebView 面板用于用户交互
|
||||
userInteractionManager.setWebviewPanel(panel);
|
||||
|
||||
// 检查是否是 VCD 生成命令(本地处理)
|
||||
if (isVCDGenerationCommand(text)) {
|
||||
await handleVCDGeneration(panel, extensionPath || "");
|
||||
return;
|
||||
}
|
||||
|
||||
// 检查是否是文件操作命令
|
||||
// 检查是否是文件操作命令(本地处理)
|
||||
const fileOperation = parseFileOperation(text);
|
||||
|
||||
console.log("解析结果:", fileOperation);
|
||||
|
||||
if (fileOperation) {
|
||||
console.log("执行文件操作:", fileOperation.type, fileOperation.filePath);
|
||||
await handleFileOperation(panel, fileOperation);
|
||||
return;
|
||||
}
|
||||
|
||||
// 普通消息处理
|
||||
console.log("作为普通消息处理");
|
||||
// 尝试使用后端服务
|
||||
if (useBackendService && extensionPath) {
|
||||
try {
|
||||
await handleUserMessageWithBackend(panel, text, extensionPath);
|
||||
return;
|
||||
} catch (error) {
|
||||
console.error("后端服务不可用,回退到本地模式:", error);
|
||||
// 后端不可用时,使用本地模拟回复
|
||||
}
|
||||
}
|
||||
|
||||
// 本地模拟回复(后端不可用时的 fallback)
|
||||
console.log("使用本地模拟回复");
|
||||
const reply = getMockReply(text);
|
||||
|
||||
// 记录助手回复到历史
|
||||
await historyManager.addAiMessage(reply);
|
||||
// 记录AI回复到历史(允许失败)
|
||||
try {
|
||||
const historyManager = ChatHistoryManager.getInstance();
|
||||
await historyManager.addAiMessage(reply);
|
||||
} catch (error) {
|
||||
console.warn("记录AI回复历史失败:", error);
|
||||
}
|
||||
|
||||
setTimeout(() => {
|
||||
panel.webview.postMessage({
|
||||
@ -61,6 +90,122 @@ export async function handleUserMessage(
|
||||
}, 500);
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用后端服务处理用户消息
|
||||
*/
|
||||
async function handleUserMessageWithBackend(
|
||||
panel: vscode.WebviewPanel,
|
||||
text: string,
|
||||
extensionPath: string
|
||||
): Promise<void> {
|
||||
// 创建或复用会话
|
||||
if (!currentSession || !currentSession.active) {
|
||||
currentSession = dialogManager.createSession(extensionPath);
|
||||
}
|
||||
|
||||
const historyManager = ChatHistoryManager.getInstance();
|
||||
|
||||
// 显示加载状态
|
||||
panel.webview.postMessage({
|
||||
command: "showLoading",
|
||||
text: "正在思考...",
|
||||
});
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
currentSession!.sendMessage(text, {
|
||||
onText: (fullText, isStreaming) => {
|
||||
// 暂时只在完成时发送消息(非流式)
|
||||
if (!isStreaming) {
|
||||
console.log('[MessageHandler] 发送最终消息, 文本长度:', fullText.length);
|
||||
panel.webview.postMessage({
|
||||
command: "receiveMessage",
|
||||
text: fullText,
|
||||
});
|
||||
}
|
||||
},
|
||||
|
||||
onToolStart: (toolName) => {
|
||||
panel.webview.postMessage({
|
||||
command: "toolStart",
|
||||
toolName,
|
||||
});
|
||||
},
|
||||
|
||||
onToolComplete: (toolName, result) => {
|
||||
panel.webview.postMessage({
|
||||
command: "toolComplete",
|
||||
toolName,
|
||||
result,
|
||||
});
|
||||
},
|
||||
|
||||
onToolError: (toolName, error) => {
|
||||
panel.webview.postMessage({
|
||||
command: "toolError",
|
||||
toolName,
|
||||
error,
|
||||
});
|
||||
},
|
||||
|
||||
onQuestion: (askId, question, options) => {
|
||||
panel.webview.postMessage({
|
||||
command: "showQuestion",
|
||||
askId,
|
||||
question,
|
||||
options,
|
||||
});
|
||||
},
|
||||
|
||||
onComplete: async () => {
|
||||
// 隐藏加载状态
|
||||
panel.webview.postMessage({
|
||||
command: "hideLoading",
|
||||
});
|
||||
|
||||
// 记录到历史(如果有累积文本)
|
||||
// 注意:实际文本已通过 onText 发送
|
||||
resolve();
|
||||
},
|
||||
|
||||
onError: (message) => {
|
||||
panel.webview.postMessage({
|
||||
command: "hideLoading",
|
||||
});
|
||||
panel.webview.postMessage({
|
||||
command: "receiveMessage",
|
||||
text: `❌ 错误: ${message}`,
|
||||
});
|
||||
reject(new Error(message));
|
||||
},
|
||||
|
||||
onNotification: (message) => {
|
||||
vscode.window.showInformationMessage(message);
|
||||
},
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理用户回答(从 WebView 调用)
|
||||
*/
|
||||
export async function handleUserAnswer(
|
||||
askId: string,
|
||||
selected?: string[],
|
||||
customInput?: string
|
||||
): Promise<void> {
|
||||
if (currentSession) {
|
||||
await currentSession.submitAnswer(askId, selected, customInput);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 中止当前对话
|
||||
*/
|
||||
export function abortCurrentDialog(): void {
|
||||
dialogManager.abortCurrentSession();
|
||||
currentSession = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 解析文件操作命令
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user