feat:支持发送文件到聊天窗口

Author: yefansky

package.json

@@ -129,6 +129,13 @@
         "when": "false",
         "category": "CodeReDesign",
         "icon": "${add}"
+      },
+      {
+        "command": "codeReDesign.sendToChat",
+        "title": "CodeReDesign: Send to Chat",
+        "when": "false",
+        "category": "CodeReDesign",
+        "icon": "${file}"
       }
     ],
     "viewsContainers": {
@@ -206,6 +213,11 @@
         {
           "command": "codeReDesign.packupToCvb",
           "group": "cvb@1"
+        },
+        {
+          "command": "codeReDesign.sendToChat",
+          "group": "cvb@1",
+          "when": "resourceScheme == file"
         }
       ]
     },

src/chatPanel.ts

@@ -5,6 +5,7 @@ import path from 'path';
 import * as fs from "fs";
 import * as apiTools from './apiTools';
 import { getOutputChannel } from './extension';
+import { processFilePlaceholder } from './metaCommand';
 
 // Webview 输出通道实现
 class WebviewOutputChannel implements vscode.OutputChannel {
@@ -240,6 +241,8 @@ export class ChatPanel {
     }
 
     private async handleSendOrEditMessage(message: any, webviewOutputChannel: WebviewOutputChannel): Promise<void> {
+        message.text = await processFilePlaceholder(message.text);
+
         if (message.command === 'editMessage' && message.index < this.conversation.length) {
             this.conversation.splice(message.index + 1);
             this.panel.webview.postMessage({ command: 'clearAfterIndex', index: message.index });
@@ -470,6 +473,18 @@ export class ChatPanel {
         return conversation;
     }
 
+    public static insertFilePathToInput(filePath: string): void {
+        if (ChatPanel.currentPanel) {
+            const formattedPath = `@file:${filePath}`;
+            ChatPanel.currentPanel.panel.webview.postMessage({
+                command: 'insertFilePath',
+                content: formattedPath
+            });
+        } else {
+            vscode.window.showInformationMessage('No active chat panel. Please open a chat panel first.');
+        }
+    }
+
     public dispose(): void {
         ChatPanel.currentPanel = undefined;
         this.panel.dispose();

src/extension.ts

@@ -398,7 +398,13 @@ export function activate(context: vscode.ExtensionContext) {
         }
     });
 
-    context.subscriptions.push(generateCvbCommand, redesignCvbCommand, applyCvbCommand, stopOperation, analyzeCodeCommand, startChatCommand, packupToCvbCommand);
+    let sendFileToChatCommand = vscode.commands.registerCommand('codeReDesign.sendToChat', async (uri: vscode.Uri) => {
+            if (uri && uri.scheme === 'file') {
+                ChatPanel.insertFilePathToInput(uri.fsPath);
+            }
+    });
+
+    context.subscriptions.push(generateCvbCommand, redesignCvbCommand, applyCvbCommand, stopOperation, analyzeCodeCommand, startChatCommand, packupToCvbCommand, sendFileToChatCommand);
 
     setupCvbAsMarkdown(context);
 

src/metaCommand.ts

@@ -0,0 +1,31 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import {readFileAsUtf8} from './utiliti';
+
+// Process @file:path placeholders in the message text
+export async function processFilePlaceholder(messageText: string): Promise<string> {
+    const filePlaceholderRegex = /^@file:([^\s\n]+)(?:\s|\n|$)/m;
+    const match = messageText.match(filePlaceholderRegex);
+
+    if (!match) {
+        return messageText;
+    }
+
+    const filePath = match[1];
+    try {
+        // Read file content
+        const content = await readFileAsUtf8(filePath);
+        const fileName = path.basename(filePath);
+
+        // Replace placeholder with formatted content
+        return messageText.replace(
+            filePlaceholderRegex,
+            `@file:${filePath}\n---------------\n${content}\n---------------\n`
+        );
+    } catch (error) {
+        return messageText.replace(
+            filePlaceholderRegex,
+            `@file:${filePath}\n---------------\n**Error reading file**: ${(error as Error).message}\n---------------\n`
+        );
+    }
+}

src/resources/chatPanelScript.js

@@ -349,6 +349,11 @@ function setupMessageHandlers() {
                     }
                 });
                 break;
+            case 'insertFilePath':
+                // Insert file path at the beginning of the input
+                input.value = data.content + ' ' + input.value;
+                input.focus();
+                break;
         }
     });
 

src/utiliti.ts

@@ -1,6 +1,56 @@
 import { ExtensionContext, ExtensionMode } from 'vscode';
+import * as fs from 'fs/promises';
+import * as iconv from 'iconv-lite';
 
 let isDebugMode: boolean;
 export function activate(context: ExtensionContext) {
     isDebugMode = context.extensionMode === ExtensionMode.Development;
+}
+
+/**
+ * 读取文件并根据其编码（GBK、UTF-8 或带 BOM 的 UTF-8）转换为 UTF-8 字符串
+ * @param filePath 文件路径
+ * @returns 转换后的 UTF-8 字符串
+ * @throws 如果无法读取文件或解码失败，抛出错误
+ */
+export async function readFileAsUtf8(filePath: string): Promise<string> {
+    try {
+        // 读取文件的原始 Buffer
+        const buffer = await fs.readFile(filePath);
+
+        // 检测是否为带 BOM 的 UTF-8
+        const isUtf8WithBom =
+            buffer.length >= 3 &&
+            buffer[0] === 0xEF &&
+            buffer[1] === 0xBB &&
+            buffer[2] === 0xBF;
+
+        if (isUtf8WithBom) {
+            // 移除 BOM 并作为 UTF-8 解码
+            return buffer.slice(3).toString('utf8');
+        }
+
+        // 尝试作为 UTF-8 解码
+        try {
+            // 先验证是否是有效的 UTF-8
+            const utf8Text = buffer.toString('utf8');
+            // 简单的 UTF-8 有效性检查：重新编码后比较
+            if (Buffer.from(utf8Text, 'utf8').equals(buffer)) {
+                return utf8Text;
+            }
+        } catch (utf8Error) {
+            // 如果 UTF-8 解码失败，继续尝试 GBK
+        }
+
+        // 尝试作为 GBK 解码
+        try {
+            const gbkText = iconv.decode(buffer, 'gbk');
+            return gbkText;
+        } catch (gbkError) {
+            throw new Error(`Failed to decode file as GBK: ${(gbkError as Error).message}`);
+        }
+
+    } catch (error) {
+        throw new Error(`Failed to read file: ${(error as Error).message}`);
+    }
 }