feat:codeToChat

src/extension.ts

@@ -10,10 +10,42 @@ import { initCreditsService } from "./services/creditsService";
 import { isTokenExpired } from "./utils/jwtUtils";
 import { NotificationService } from "./services/notificationService";
 import { InvitationService } from "./services/invitationService";
+import { ICCoderCodeActionProvider } from "./providers/codeActionProvider";
 
 export async function activate(context: vscode.ExtensionContext) {
   console.log("🎉 IC Coder 插件已激活！");
 
+  // 创建装饰类型（代码旁边的提示）
+  const decorationType = vscode.window.createTextEditorDecorationType({
+    after: {
+      contentText: ' Ctrl+L 添加到 IC Coder 对话',
+      color: '#888',
+      fontStyle: 'italic',
+      margin: '0 0 0 1em'
+    }
+  });
+
+  // 更新装饰
+  const updateDecorations = () => {
+    const editor = vscode.window.activeTextEditor;
+    if (!editor) return;
+
+    if (!editor.selection.isEmpty) {
+      const range = new vscode.Range(editor.selection.end, editor.selection.end);
+      const decoration = { range };
+      editor.setDecorations(decorationType, [decoration]);
+    } else {
+      editor.setDecorations(decorationType, []);
+    }
+  };
+
+  context.subscriptions.push(
+    vscode.window.onDidChangeTextEditorSelection(updateDecorations),
+    vscode.window.onDidChangeActiveTextEditor(updateDecorations)
+  );
+
+  updateDecorations();
+
   // 初始化通知服务
   const notificationService = NotificationService.getInstance(context);
   console.log('[Extension] 通知服务已初始化');
@@ -250,6 +282,48 @@ export async function activate(context: vscode.ExtensionContext) {
     }
   );
 
+  // 注册命令：将选中代码添加到对话
+  const addCodeToChat = vscode.commands.registerCommand(
+    "ic-coder.addCodeToChat",
+    async () => {
+      const editor = vscode.window.activeTextEditor;
+      if (!editor) return;
+
+      const selection = editor.selection;
+      const selectedText = editor.document.getText(selection);
+
+      if (!selectedText) {
+        vscode.window.showWarningMessage("请先选择代码");
+        return;
+      }
+
+      const fileName = editor.document.fileName;
+      const startLine = selection.start.line + 1;
+      const endLine = selection.end.line + 1;
+
+      // 检查是否已有打开的面板
+      let panel = (global as any).currentICHelperPanel;
+      if (!panel || panel._isDisposed) {
+        await showICHelperPanel(context);
+        panel = (global as any).currentICHelperPanel;
+      }
+
+      // 发送代码上下文
+      setTimeout(() => {
+        if (panel?.webview) {
+          panel.webview.postMessage({
+            command: 'addCodeContext',
+            fileName,
+            startLine,
+            endLine,
+            code: selectedText,
+            languageId: editor.document.languageId
+          });
+        }
+      }, 300);
+    }
+  );
+
   // 注册命令：查看会话历史
   // TODO: 这些命令需要根据新的任务架构重新实现
   // 暂时注释掉，等待重新实现
@@ -312,6 +386,13 @@ export async function activate(context: vscode.ExtensionContext) {
   // 注册 VCD 自定义编辑器
   const vcdEditorProvider = VCDViewerEditorProvider.register(context, vcdFileServer);
 
+  // 注册 Code Action Provider
+  const codeActionProvider = vscode.languages.registerCodeActionsProvider(
+    { scheme: 'file' },
+    new ICCoderCodeActionProvider(),
+    { providedCodeActionKinds: [vscode.CodeActionKind.RefactorRewrite] }
+  );
+
   // 添加到订阅
   context.subscriptions.push(
     openPanelCommand,
@@ -322,6 +403,7 @@ export async function activate(context: vscode.ExtensionContext) {
     logoutCommand,
     changeInvitationCodeCommand,
     testNotificationCommand,
+    addCodeToChat,
     // testTrialUserCommand,
     // testExpiredUserCommand,
     // TODO: 等待重新实现这些命令
@@ -332,7 +414,8 @@ export async function activate(context: vscode.ExtensionContext) {
     // clearHistoryCommand,
     // searchSessionCommand,
     viewRegistration,
-    vcdEditorProvider
+    vcdEditorProvider,
+    codeActionProvider
   );
 }