feat(plugin): VSCode extension (#717)

Co-authored-by: Neko <neko@ayaka.moe>

Author: luoling8192

.vscode/launch.json

@@ -25,6 +25,18 @@
       "presentation": {
         "hidden": true
       }
+    },
+    {
+      "name": "Debug VSCode Extension",
+      "type": "extensionHost",
+      "request": "launch",
+      "args": [
+        "--extensionDevelopmentPath=${workspaceFolder}/plugins/airi-plugin-vscode"
+      ],
+      "outFiles": [
+        "${workspaceFolder}/plugins/airi-plugin-vscode/dist/**/*.js"
+      ],
+      "preLaunchTask": "npm: build - plugins/airi-plugin-vscode"
     }
   ],
   "compounds": [

packages/server-shared/src/types/websocket/events.ts

@@ -59,6 +59,7 @@ export interface WebSocketEvents<C = undefined> {
   'input:voice': {
     audio: ArrayBuffer
   } & Partial<WithInputSource<'browser' | 'discord'>>
+  'vscode:context': C
 }
 
 export type WebSocketEvent<C = undefined> = {

plugins/airi-plugin-vscode/package.json

@@ -0,0 +1,69 @@
+{
+  "publisher": "airi",
+  "name": "@proj-airi/airi-plugin-vscode",
+  "displayName": "Airi",
+  "version": "0.1.0",
+  "description": "VSCode extension that shares your coding context with Airi",
+  "categories": [
+    "Other"
+  ],
+  "main": "./dist/extension.js",
+  "engines": {
+    "vscode": "^1.85.0"
+  },
+  "activationEvents": [
+    "onStartupFinished"
+  ],
+  "contributes": {
+    "commands": [
+      {
+        "command": "airi.enable",
+        "title": "AIRI: Enable"
+      },
+      {
+        "command": "airi.disable",
+        "title": "AIRI: Disable"
+      },
+      {
+        "command": "airi.status",
+        "title": "AIRI: Show Status"
+      }
+    ],
+    "configuration": {
+      "title": "AIRI",
+      "properties": {
+        "airi.enabled": {
+          "type": "boolean",
+          "default": true,
+          "description": "Enable Airi companion"
+        },
+        "airi.contextLines": {
+          "type": "number",
+          "default": 5,
+          "description": "Number of context lines to send (before and after current line)"
+        },
+        "airi.sendInterval": {
+          "type": "number",
+          "default": 3000,
+          "description": "Interval in milliseconds to send updates (0 for real-time)"
+        }
+      }
+    }
+  },
+  "scripts": {
+    "vscode:prepublish": "pnpm run build",
+    "build": "tsdown",
+    "dev": "tsdown --watch",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@guiiai/logg": "catalog:",
+    "@proj-airi/server-sdk": "workspace:*"
+  },
+  "devDependencies": {
+    "@types/node": "^24.10.0",
+    "@types/vscode": "^1.85.0",
+    "tsdown": "^0.15.2",
+    "typescript": "^5.3.0"
+  }
+}

plugins/airi-plugin-vscode/src/airi-client.ts

@@ -0,0 +1,68 @@
+import type { AiriEvent } from './types'
+
+import { useLogger } from '@guiiai/logg'
+import { Client } from '@proj-airi/server-sdk'
+
+/**
+ * Airi Channel Server Client
+ */
+export class AiriClient {
+  private client: Client<AiriEvent> | null = null
+
+  /**
+   * Connect to Channel Server
+   */
+  async connect(): Promise<boolean> {
+    try {
+      this.client = new Client({ name: 'proj-airi:plugin-vscode' })
+
+      useLogger().log('Airi companion connected to Channel Server')
+      return true
+    }
+    catch (error) {
+      useLogger().errorWithError('Failed to connect to Airi Channel Server:', error)
+      return false
+    }
+  }
+
+  /**
+   * Disconnect from Channel Server
+   */
+  disconnect(): void {
+    if (this.client) {
+      this.client.close()
+      this.client = null
+      useLogger().log('Airi companion disconnected')
+    }
+  }
+
+  /**
+   * Send event to Airi
+   */
+  sendEvent(event: AiriEvent): void {
+    if (!this.client) {
+      useLogger().warn('Cannot send event: not connected to Airi Channel Server')
+      return
+    }
+
+    try {
+      // Send event to Airi
+      this.client.send({
+        type: 'vscode:context',
+        data: event,
+      })
+
+      useLogger().log(`Sent event to Airi: ${event.type}`, event)
+    }
+    catch (error) {
+      useLogger().errorWithError('Failed to send event to Airi:', error)
+    }
+  }
+
+  /**
+   * Is connected to Channel Server
+   */
+  isConnected(): boolean {
+    return !!this.client
+  }
+}

plugins/airi-plugin-vscode/src/context-collector.ts

@@ -0,0 +1,133 @@
+import type { CodingContext } from './types'
+
+import { useLogger } from '@guiiai/logg'
+
+import * as vscode from 'vscode'
+
+/**
+ * Collector for coding context in VSCode
+ */
+export class ContextCollector {
+  constructor(
+    private readonly contextLines: number = 5,
+  ) {}
+
+  /**
+   * Collect context from the current active editor
+   */
+  async collect(editor: vscode.TextEditor): Promise<CodingContext | null> {
+    try {
+      const document = editor.document
+      const position = editor.selection.active
+
+      // File information
+      const file = {
+        path: document.uri.fsPath,
+        languageId: document.languageId,
+        fileName: document.fileName,
+        workspaceFolder: this.getWorkspaceFolder(document.uri),
+      }
+
+      // Cursor position
+      const cursor = {
+        line: position.line,
+        character: position.character,
+      }
+
+      // Selected text
+      const selection = editor.selection.isEmpty
+        ? undefined
+        : {
+            text: document.getText(editor.selection),
+            start: {
+              line: editor.selection.start.line,
+              character: editor.selection.start.character,
+            },
+            end: {
+              line: editor.selection.end.line,
+              character: editor.selection.end.character,
+            },
+          }
+
+      // Current line
+      const currentLine = {
+        lineNumber: position.line,
+        text: document.lineAt(position.line).text,
+      }
+
+      // Context (N lines before and after)
+      const context = this.getContext(document, position.line)
+
+      // Git information (simplified, can be extended later)
+      const git = await this.getGitInfo(document.uri)
+
+      return {
+        file,
+        cursor,
+        selection,
+        currentLine,
+        context,
+        git,
+        timestamp: Date.now(),
+      }
+    }
+    catch (error) {
+      useLogger().errorWithError('Failed to collect context:', error)
+      return null
+    }
+  }
+
+  /**
+   * Get context before and after the current line
+   */
+  private getContext(document: vscode.TextDocument, currentLine: number) {
+    const before: string[] = []
+    const after: string[] = []
+
+    // Get preceding lines
+    const startLine = Math.max(0, currentLine - this.contextLines)
+    for (let i = startLine; i < currentLine; i++) {
+      before.push(document.lineAt(i).text)
+    }
+
+    // Get following lines
+    const endLine = Math.min(document.lineCount - 1, currentLine + this.contextLines)
+    for (let i = currentLine + 1; i <= endLine; i++) {
+      after.push(document.lineAt(i).text)
+    }
+
+    return { before, after }
+  }
+
+  /**
+   * Get workspace folder path
+   */
+  private getWorkspaceFolder(uri: vscode.Uri): string | undefined {
+    const folder = vscode.workspace.getWorkspaceFolder(uri)
+    return folder?.uri.fsPath
+  }
+
+  /**
+   * Get Git information (simplified)
+   */
+  private async getGitInfo(uri: vscode.Uri): Promise<{ branch: string, isDirty: boolean } | undefined> {
+    try {
+      const gitExtension = vscode.extensions.getExtension('vscode.git')?.exports
+      if (!gitExtension)
+        return undefined
+
+      const git = gitExtension.getAPI(1)
+      const repo = git.getRepository(uri)
+      if (!repo)
+        return undefined
+
+      return {
+        branch: repo.state.HEAD?.name ?? 'unknown',
+        isDirty: repo.state.workingTreeChanges.length > 0,
+      }
+    }
+    catch {
+      return undefined
+    }
+  }
+}