Merge branch 'feat/back-to-front' into feat/plugin-front-end

src/utils/messageHandler.ts

@@ -27,6 +27,9 @@ let useBackendService = true;
 /** 当前对话会话 */
 let currentSession: DialogSession | null = null;
 
+/** 最后一个活跃的 taskId（用于压缩等操作） */
+let lastTaskId: string | null = null;
+
 /** 待执行的计划（Plan 模式确认后自动执行） */
 let pendingPlanExecution: {
   panel: vscode.WebviewPanel;
@@ -127,6 +130,8 @@ async function handleUserMessageWithBackend(
   // 创建或复用会话
   if (!currentSession || !currentSession.active) {
     currentSession = dialogManager.createSession(extensionPath, reuseTaskId);
+    // 保存 taskId 用于后续操作（如压缩）
+    lastTaskId = currentSession.getTaskId();
     if (reuseTaskId) {
       console.log("[MessageHandler] 复用 taskId 创建会话:", reuseTaskId);
     }
@@ -273,6 +278,16 @@ async function handleUserMessageWithBackend(
         onNotification: (message) => {
           vscode.window.showInformationMessage(message);
         },
+
+        onContextUsage: (data) => {
+          // 发送上下文使用量到 WebView
+          panel.webview.postMessage({
+            command: "contextUsage",
+            currentTokens: data.currentTokens,
+            maxTokens: data.maxTokens,
+            percentage: data.percentage,
+          });
+        },
       },
       mode
     );
@@ -295,7 +310,35 @@ export async function handleUserAnswer(
 /**
  * 中止当前对话
  */
-export function abortCurrentDialog(): void {
+export async function abortCurrentDialog(): Promise<void> {
+  if (currentSession) {
+    // 保存当前已有的对话内容
+    const segments = currentSession.getSegments();
+    if (segments && segments.length > 0) {
+      try {
+        const historyManager = ChatHistoryManager.getInstance();
+        const textContent = segments
+          .filter((s) => s.type === "text" && s.content)
+          .map((s) => s.content)
+          .join("\n");
+
+        // 添加中止标记
+        const abortedContent = textContent + "\n\n[对话已被用户中止]";
+        await historyManager.addAiMessage(abortedContent, undefined, segments);
+        console.log("[MessageHandler] 已保存中止前的对话内容");
+      } catch (error) {
+        console.warn("[MessageHandler] 保存中止对话失败:", error);
+      }
+    }
+  }
+
+  // 通知 WebView 重置分段消息容器
+  const panel = userInteractionManager.getWebviewPanel();
+  if (panel) {
+    panel.webview.postMessage({ command: "resetSegmentedMessage" });
+    console.log("[MessageHandler] 已发送重置分段消息命令");
+  }
+
   dialogManager.abortCurrentSession();
   currentSession = null;
 }
@@ -304,7 +347,15 @@ export function abortCurrentDialog(): void {
  * 获取当前会话的 taskId
  */
 export function getCurrentTaskId(): string | null {
-  return currentSession?.getTaskId() || null;
+  return currentSession?.getTaskId() || lastTaskId;
+}
+
+/**
+ * 设置最后的 taskId（加载历史会话时调用）
+ */
+export function setLastTaskId(taskId: string): void {
+  lastTaskId = taskId;
+  console.log("[MessageHandler] 设置 lastTaskId:", taskId);
 }
 
 /**