feat: 增加中途中断指令

Author: yefansky

.gitignore

@@ -17,3 +17,4 @@ dist/
 Thumbs.db
 tmp/*
 .CodeReDesignWorkSpace/*
+*.vsix

README.md

@@ -39,7 +39,8 @@ ctrl + shift + p 打开指令菜单（Command Palette），有以下几个指令
    如果查看后觉得没问题，可以用这个指令
    将这个 CVB 格式文件展开覆盖本地文件
 
-
+4. **codeReDesign.stopOperation** (中断处理)
+   中断正在执行的uploadCvb操作
 ---
 
 # CodeReDesign
@@ -82,3 +83,6 @@ Press `ctrl + shift + p` to open the Command Palette, where the following comman
 3. **codeReDesign.applyCvb** (Apply CVB file)
    If you review the changes and find them satisfactory, you can use this command.
    It will unpack and overwrite the local files with the changes from the CVB file.
+
+4. **codeReDesign.stopOperation** (stop Upload CVB)
+   stop processing Operation

package-lock.json

@@ -39,6 +39,10 @@
       {
         "command": "codeReDesign.applyThisCvb",
         "title": "CodeReDesign: Apply this CVB to Workspace"
+      },
+      {
+        "command": "codeReDesign.stopOperation",
+        "title": "CodeReDesign: Stop Operation"
       }
     ],
     "views": {
@@ -105,6 +109,6 @@
   "dependencies": {
     "iconv-lite": "^0.6.3",
     "jschardet": "^3.1.4",
-    "openai": "^4.78.1"
+    "openai": "^4.81.0"
   }
 }

package.json

@@ -32,14 +32,16 @@ const apiBaseURL = "https://api.deepseek.com";
  * @param outputChannel 输出通道，用于实时显示流式内容
  * @param streamMode 是否启用流式模式
  * @param endstring 结束字符串，用于检查输出是否包含特定字符串
+ * @param abortSignal 用于中断请求的信号
  * @returns API 返回的完整内容
  */
 async function callDeepSeekApi(
     userContent: string,
     systemContent: string = 'You are a helpful assistant.',
     outputChannel?: vscode.OutputChannel,
     streamMode: boolean = true,
-    endstring?: string
+    endstring?: string,
+    abortSignal?: AbortSignal
 ): Promise<string | null> {
     const apiKey = getDeepSeekApiKey();
     if (!apiKey) {
@@ -85,6 +87,9 @@ async function callDeepSeekApi(
 
             if (streamMode) {
                 for await (const chunk of response as AsyncIterable<OpenAI.Chat.Completions.ChatCompletionChunk>) {
+                    if (abortSignal?.aborted) {
+                        throw new Error('operation stop by user');
+                    }
                     const content = chunk.choices[0]?.delta?.content || '';
                     chunkResponse += content;
                     if (outputChannel) {
@@ -128,6 +133,10 @@ async function callDeepSeekApi(
         return fullResponse;
 
     } catch (error) {
+        if (error instanceof Error && error.message === 'operation stop by user') {
+            vscode.window.showInformationMessage('operation stop by user');
+            return null;
+        }
         vscode.window.showErrorMessage('API调用失败: ' + (error as Error).message);
         return null;
     }
@@ -138,12 +147,14 @@ async function callDeepSeekApi(
  * @param cvbContent CVB 文件内容
  * @param userRequest 用户输入的重构需求
  * @param outputChannel 输出通道，用于实时显示流式内容
+ * @param abortSignal 用于中断请求的信号
  * @returns API 返回的完整 CVB 内容
  */
 export async function queryCodeReDesign(
     cvbContent: string,
     userRequest: string,
-    outputChannel: vscode.OutputChannel
+    outputChannel: vscode.OutputChannel,
+    abortSignal?: AbortSignal
 ): Promise<string | null> {
     const requestContent = `
 
@@ -170,7 +181,7 @@ ${userRequest}
 请输出CVB格式的代码:
 `;
 
-    return callDeepSeekApi(requestContent, undefined, outputChannel, true, '## END_CVB'); // 添加结束字符串
+    return callDeepSeekApi(requestContent, undefined, outputChannel, true, '## END_CVB', abortSignal); // 添加结束字符串
 }
 
 /**

src/deepseekApi.ts

@@ -41,6 +41,9 @@ export function activate(context: vscode.ExtensionContext) {
         vscode.window.showInformationMessage(`CVB file generated at: ${cvbFilePath}`);
     });
 
+    // 用于存储当前的上传操作
+    let currentOperationController: AbortController | null = null;
+
     // 注册命令:上传 CVB 并调用 API
     let uploadCvbCommand = vscode.commands.registerCommand('codeReDesign.uploadCvb', async () => {
         const workspaceFolders = vscode.workspace.workspaceFolders;
@@ -88,13 +91,27 @@ export function activate(context: vscode.ExtensionContext) {
         const cvbFilePath = path.join(tmpDir, selectedCvbFile);
         const cvbContent = fs.readFileSync(cvbFilePath, 'utf-8');
 
-        const apiResponse = await queryCodeReDesign(cvbContent, userPrompt, outputChannel);
+        // 创建新的 AbortController
+        currentOperationController = new AbortController();
+        const apiResponse = await queryCodeReDesign(cvbContent, userPrompt, outputChannel, currentOperationController.signal);
         if (apiResponse) {
             const { cvbContent: newCvbContent, metadata, files } = parseCvb(apiResponse);
             const newCvbFilePath = path.join(tmpDir, fileName);
             fs.writeFileSync(newCvbFilePath, newCvbContent, 'utf-8');
             vscode.window.showInformationMessage(`API response saved as CVB file: ${newCvbFilePath}`);
         }
+        currentOperationController = null;
+    });
+
+    // 注册命令：中断当前的上传操作
+    let stopOperation = vscode.commands.registerCommand('codeReDesign.stopOperation', () => {
+        if (currentOperationController) {
+            currentOperationController.abort();
+            currentOperationController = null;
+            vscode.window.showInformationMessage('Stop operation.');
+        } else {
+            vscode.window.showInformationMessage('No operation in progress.');
+        }
     });
 
     // 注册命令：应用 CVB 到工作目录
@@ -135,7 +152,7 @@ export function activate(context: vscode.ExtensionContext) {
         }
     });
 
-    context.subscriptions.push(generateCvbCommand, uploadCvbCommand, applyCvbCommand, outputChannel);
+    context.subscriptions.push(generateCvbCommand, uploadCvbCommand, applyCvbCommand, stopOperation, outputChannel);
 
     setupCvbAsMarkdown(context);