vscode-pi-companion 0.0.10

package/.github/workflows/publish.yml

@@ -0,0 +1,62 @@
+name: Publish to VS Code Marketplace & npm
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  publish-vscode:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+    
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: true
+      
+      - uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          cache: 'npm'
+      
+      - name: Install dependencies
+        run: npm ci
+      
+      - name: Install vsce
+        run: npm install -g @vscode/vsce
+      
+      - name: Publish to VS Code Marketplace
+        env:
+          VSCE_TOKEN: ${{ secrets.VSCE_TOKEN }}
+        run: |
+          git config --local user.name "github-actions"
+          git config --local user.email "github-actions@github.com"
+          
+          # Package and publish with version bump
+          npx vsce package
+          vsce publish patch --pat $VSCE_TOKEN --skip-duplicate
+          
+          # Push the tag created by vsce
+          git push --follow-tags origin main
+
+  publish-npm:
+    runs-on: ubuntu-latest
+    needs: publish-vscode
+    if: startsWith(github.ref, 'refs/tags/v')
+    permissions:
+      id-token: write  # Required for OIDC Trusted Publishing
+      
+    steps:
+      - uses: actions/checkout@v4
+      
+      - uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          cache: 'npm'
+      
+      - name: Install dependencies
+        run: npm ci
+      
+      - name: Publish to npm
+        run: npm publish --access public --skip-duplicate

package/.github/workflows/release.yml

@@ -0,0 +1,19 @@
+name: Release
+
+on:
+  push:
+    branches: [main]
+
+permissions:
+  contents: write
+  pull-requests: write
+
+jobs:
+  release:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: googleapis/release-please-action@v4
+        with:
+          release-type: node
+          package-name: vscode-pi-companion
+          target-branch: main

package/.vscode/launch.json

@@ -0,0 +1,17 @@
+{
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "Run Extension",
+      "type": "extensionHost",
+      "request": "launch",
+      "args": [
+        "--extensionDevelopmentPath=${workspaceFolder}"
+      ],
+      "outFiles": [
+        "${workspaceFolder}/out/**/*.js"
+      ],
+      "preLaunchTask": "${defaultBuildTask}"
+    }
+  ]
+}

package/.vscode/tasks.json

@@ -0,0 +1,12 @@
+{
+  "version": "2.0.0",
+  "tasks": [
+    {
+      "type": "npm",
+      "script": "compile",
+      "problemMatcher": [],
+      "label": "npm: compile",
+      "detail": "tsc -p ./"
+    }
+  ]
+}

package/.vscodeignore

@@ -0,0 +1,12 @@
+src/
+.vscode/
+.github/
+.gitignore
+*.ts
+*.map
+!out/**/*.js
+!out/**/*.d.ts
+node_modules/
+extensions/node_modules/
+*.vsix
+tsconfig.json

package/LICENSE

@@ -0,0 +1 @@
+{"name":"MIT","url":"https://opensource.org/licenses/MIT"}

package/README.md

@@ -0,0 +1,83 @@
+# VS Code PI Companion
+
+A VS Code extension that provides real-time IDE context (files, selection) to pi and other agents.
+
+## Features
+
+- **Real-time context**: Monitors open files, cursor position, and selection
+- **Instant updates**: Pushes context changes via SSE
+- **Seamless integration**: Selected lines injected into LLM context automatically
+- **Widget display**: Shows current file/selection in pi UI
+
+## Architecture
+
+**VS Code Extension** → SSE ← **pi Extension**
+
+- **VS Code**: Context tracking, SSE server
+- **pi**: SSE client, Widget, Context injection
+
+## Installation
+
+### 1. Install pi Extension
+
+```bash
+pi install git:github.com/ravshansbox/vscode-pi-companion
+```
+
+Or manually:
+```bash
+mkdir -p ~/.pi/agent/extensions/vscode-pi-companion
+git clone git@github.com:ravshansbox/vscode-pi-companion.git ~/.pi/agent/extensions/vscode-pi-companion
+cd ~/.pi/agent/extensions/vscode-pi-companion
+npm install
+```
+
+### 2. Install VS Code Extension
+
+The pi extension will auto-install the VS Code extension when you run `pi` in a VS Code terminal.
+
+Alternatively, install manually:
+```bash
+# Download the .vsix from releases
+code --install-extension vscode-pi-companion.vsix
+```
+
+Then reload VS Code window.
+
+## Usage
+
+1. Open a folder in VS Code
+2. Open VS Code terminal (`Ctrl+\``)
+3. Run `pi`
+4. The extension auto-connects and shows current file in widget
+5. Select code → widget updates instantly
+6. Selected lines are injected into LLM context automatically
+
+## Widget Display
+
+```
+📄 filename           # no selection
+📄 filename L9        # single line
+📄 filename L9:14     # multi-line selection
+```
+
+## MCP Tools (for LLM)
+
+| Tool | Description |
+|------|-------------|
+| `get_ide_context` | Full IDE context |
+| `get_current_file` | Current file with selection |
+| `list_open_files` | List all open files |
+
+## Development
+
+```bash
+# Build VS Code extension
+npm install
+npm run compile
+
+# Package as .vsix
+npx vsce package
+```
+
+Press F5 to run in Extension Development Host.

package/extensions/index.ts

@@ -0,0 +1,370 @@
+/**
+ * PI Companion Extension
+ * 
+ * Connects to the VS Code extension via SSE for real-time context updates.
+ * No polling, instant updates, no tool calls visible.
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import * as http from "http";
+import { execSync } from "child_process";
+import { Type } from "@sinclair/typebox";
+
+// Check if running in VS Code integrated terminal
+function isRunningInVSCode(): boolean {
+  return process.env.TERM_PROGRAM === 'vscode';
+}
+
+// Install VS Code extension from marketplace
+function installFromMarketplace(ctx: ExtensionContext): void {
+  try {
+    execSync('code --install-extension ravshansbox.vscode-pi-companion --force', { stdio: 'pipe' });
+    ctx.ui.notify("PI Companion: VS Code extension installed. Reload window to activate.", "info");
+  } catch {
+    // Silently ignore
+  }
+}
+
+// Check if extension is installed
+function isExtensionInstalled(): boolean {
+  try {
+    const result = execSync('code --list-extensions', { encoding: 'utf8' });
+    return result.includes('ravshansbox.vscode-pi-companion');
+  } catch {
+    return false;
+  }
+}
+
+// Install/update VS Code extension if running in VS Code terminal
+async function installVSCodeExtension(ctx: ExtensionContext): Promise<void> {
+  if (!isRunningInVSCode()) {
+    return;
+  }
+  
+  const isInstalled = isExtensionInstalled();
+  
+  if (isInstalled) {
+    // Already installed - check for updates
+    const latestVersion = await getLatestVersion();
+    if (latestVersion) {
+      console.log(`[PI Companion] Update available: v${latestVersion}`);
+      console.log(`[PI Companion] Run: pi install git:github.com/ravshansbox/vscode-pi-companion`);
+    }
+    return;
+  }
+  
+  // Not installed - install from marketplace
+  installFromMarketplace(ctx);
+}
+
+// Get latest version from GitHub
+async function getLatestVersion(): Promise<string | null> {
+  try {
+    const response = await fetch('https://api.github.com/repos/ravshansbox/vscode-pi-companion/releases/latest', {
+      headers: { 'Accept': 'application/vnd.github.v3+json' }
+    });
+    
+    if (!response.ok) {
+      return null;
+    }
+    
+    const release = await response.json() as { tag_name?: string };
+    return release.tag_name?.replace('v', '') || null;
+  } catch {
+    return null;
+  }
+}
+
+interface IdeContext {
+  workspaceState: {
+    openFiles: Array<{
+      path: string;
+      timestamp: number;
+      isActive?: boolean;
+      selectedText?: string;
+      cursor?: { line: number; character: number };
+      selectionStart?: { line: number; character: number };
+      selectionEnd?: { line: number; character: number };
+      content?: string;
+    }>;
+    isTrusted: boolean;
+    workspacePath: string;
+  };
+}
+
+let currentContext: IdeContext | undefined;
+let currentCtx: ExtensionContext | undefined;
+let lastWidgetKey = "";
+let widgetDebounceTimer: NodeJS.Timeout | undefined;
+let reconnectTimer: NodeJS.Timeout | undefined;
+let sseRequest: http.ClientRequest | undefined;
+
+function getPort(): number | undefined {
+  const configDir = path.join(os.tmpdir(), "pi-companion");
+
+  try {
+    const files = fs.readdirSync(configDir);
+    const connectionFiles = files
+      .filter((f) => f.startsWith("connection-") && f.endsWith(".json"))
+      .map((f) => ({
+        name: f,
+        path: path.join(configDir, f),
+        mtime: fs.statSync(path.join(configDir, f)).mtime.getTime(),
+      }))
+      .sort((a, b) => b.mtime - a.mtime);
+
+    if (connectionFiles.length > 0) {
+      const content = fs.readFileSync(connectionFiles[0].path, "utf-8");
+      const info = JSON.parse(content);
+      return info.port;
+    }
+  } catch {
+    // No connection files found
+  }
+  return undefined;
+}
+
+function updateWidget(): void {
+  if (!currentCtx) return;
+  
+  if (!currentContext || !currentContext.workspaceState.openFiles.length) {
+    currentCtx.ui.setWidget("pi-companion", ["PI Companion: No files open"]);
+    return;
+  }
+
+  const { workspaceState } = currentContext;
+  const lines: string[] = [];
+
+  const activeFile = workspaceState.openFiles.find((f) => f.isActive);
+
+  if (activeFile) {
+    const filename = activeFile.path.split("/").pop() || activeFile.path;
+    
+    // Build location info - ONLY show when text is actually selected
+    let location = "";
+    if (activeFile.selectionStart && activeFile.selectionEnd) {
+      const start = activeFile.selectionStart.line;
+      const end = activeFile.selectionEnd.line;
+      if (start === end) {
+        location = ` L${start}`;
+      } else {
+        location = ` L${start}:${end}`;
+      }
+    }
+    // Don't show cursor position when nothing is selected
+    
+    lines.push(`📄 ${filename}${location}`);
+  }
+
+  const otherCount = workspaceState.openFiles.length - 1;
+  if (otherCount > 0) {
+    lines.push(`   +${otherCount} more file${otherCount > 1 ? "s" : ""} open`);
+  }
+
+  const widgetKey = lines.join("|");
+  
+  // Debounce widget updates to avoid flickering during rapid selection changes
+  if (widgetDebounceTimer) {
+    clearTimeout(widgetDebounceTimer);
+  }
+  
+  widgetDebounceTimer = setTimeout(() => {
+    if (widgetKey !== lastWidgetKey) {
+      lastWidgetKey = widgetKey;
+      currentCtx!.ui.setWidget("pi-companion", lines);
+    }
+    widgetDebounceTimer = undefined;
+  }, 50); // Wait 50ms after last update
+}
+
+function connectSSE(): void {
+  const port = getPort();
+  if (!port) return;
+
+  // Close existing connection
+  if (sseRequest) {
+    sseRequest.destroy();
+    sseRequest = undefined;
+  }
+
+  const url = new URL(`http://127.0.0.1:${port}/context/stream`);
+  
+  sseRequest = http.get({
+    hostname: url.hostname,
+    port: url.port,
+    path: url.pathname,
+  }, (res) => {
+    let buffer = "";
+    
+    res.on("data", (chunk: string) => {
+      buffer += chunk;
+      
+      // Process complete SSE messages
+      const lines = buffer.split("\n");
+      buffer = lines.pop() || "";
+      
+      for (const line of lines) {
+        if (line.startsWith("data: ")) {
+          const data = line.slice(6);
+          if (data && data !== "[object Object]") {
+            try {
+              const parsed = JSON.parse(data);
+              currentContext = parsed;
+              updateWidget();
+            } catch {
+              // Ignore parse errors
+            }
+          }
+        }
+      }
+    });
+
+    res.on("end", () => {
+      scheduleReconnect();
+    });
+
+    res.on("error", () => {
+      scheduleReconnect();
+    });
+  });
+
+  sseRequest.on("error", () => {
+    scheduleReconnect();
+  });
+}
+
+function scheduleReconnect(): void {
+  if (reconnectTimer || !currentCtx) return;
+  reconnectTimer = setTimeout(() => {
+    reconnectTimer = undefined;
+    connectSSE();
+  }, 2000);
+}
+
+export default function (pi: ExtensionAPI) {
+  // Inject IDE context before each prompt
+  pi.on("before_agent_start", async (event, ctx) => {
+    if (!currentContext) return;
+    
+    const activeFile = currentContext.workspaceState.openFiles.find((f) => f.isActive);
+    if (!activeFile) return;
+
+    let contextText = "";
+    
+    if (activeFile.selectedText) {
+      // Inject selected text
+      contextText = `Current selection in ${activeFile.path}:\n\`\`\`\n${activeFile.selectedText}\n\`\`\``;
+    } else if (activeFile.content) {
+      // Inject full file content for small files, or first 100 lines
+      const lines = activeFile.content.split("\n");
+      if (lines.length <= 100) {
+        contextText = `Current file ${activeFile.path}:\n\`\`\`\n${activeFile.content}\n\`\`\``;
+      } else {
+        contextText = `Current file ${activeFile.path} (first 100 lines):\n\`\`\`\n${lines.slice(0, 100).join("\n")}\n\`\`\``;
+      }
+    }
+
+    if (contextText) {
+      return {
+        message: {
+          customType: "pi-companion",
+          content: contextText,
+          display: false, // Don't show in chat history
+        },
+      };
+    }
+  });
+
+  pi.on("session_start", async (_event, ctx) => {
+    currentCtx = ctx;
+
+    // Auto-install/update VS Code extension if needed
+    installVSCodeExtension(ctx);
+
+    const port = getPort();
+    if (port) {
+      ctx.ui.notify("PI Companion: Connected to VS Code", "info");
+      connectSSE();
+    } else {
+      ctx.ui.notify("PI Companion: VS Code extension not detected", "warning");
+    }
+  });
+
+  pi.on("session_shutdown", async () => {
+    if (reconnectTimer) {
+      clearTimeout(reconnectTimer);
+      reconnectTimer = undefined;
+    }
+    if (sseRequest) {
+      sseRequest.destroy();
+      sseRequest = undefined;
+    }
+  });
+
+  // Register tools
+  pi.registerTool({
+    name: "get_ide_context",
+    label: "Get IDE Context",
+    description: "Get the current IDE context: open files, cursor position, and selection",
+    parameters: Type.Object({}),
+    async execute() {
+      if (!currentContext) {
+        return {
+          content: [{ type: "text", text: "No IDE context available. Make sure the PI Companion VS Code extension is installed and running." }],
+          details: {},
+        };
+      }
+      return {
+        content: [{ type: "text", text: JSON.stringify(currentContext, null, 2) }],
+        details: {},
+      };
+    },
+  });
+
+  pi.registerTool({
+    name: "get_current_file",
+    label: "Get Current File",
+    description: "Get the content of the currently active file in the IDE",
+    parameters: Type.Object({}),
+    async execute() {
+      if (!currentContext) {
+        return { content: [{ type: "text", text: "No IDE context available" }], details: {} };
+      }
+      const activeFile = currentContext.workspaceState.openFiles.find((f) => f.isActive);
+      if (!activeFile) {
+        return { content: [{ type: "text", text: "No active file" }], details: {} };
+      }
+      return {
+        content: [{ type: "text", text: JSON.stringify({
+          path: activeFile.path,
+          cursor: activeFile.cursor,
+          selection: activeFile.selectedText,
+          content: activeFile.content,
+        }, null, 2) }],
+        details: {},
+      };
+    },
+  });
+
+  pi.registerTool({
+    name: "list_open_files",
+    label: "List Open Files",
+    description: "List all files currently open in the IDE",
+    parameters: Type.Object({}),
+    async execute() {
+      if (!currentContext) {
+        return { content: [{ type: "text", text: "No IDE context available" }], details: {} };
+      }
+      const files = currentContext.workspaceState.openFiles.map((f) => ({
+        path: f.path,
+        isActive: f.isActive,
+        cursor: f.cursor,
+        hasSelection: !!f.selectedText,
+      }));
+      return { content: [{ type: "text", text: JSON.stringify(files, null, 2) }], details: {} };
+    },
+  });
+}

package/extensions/package.json

@@ -0,0 +1,9 @@
+{
+  "name": "vscode-pi-companion-extension",
+  "version": "0.0.1",
+  "type": "module",
+  "dependencies": {
+    "@mariozechner/pi-coding-agent": "*",
+    "@sinclair/typebox": "*"
+  }
+}

package/package.json

@@ -0,0 +1,88 @@
+{
+  "name": "vscode-pi-companion",
+  "displayName": "PI Companion",
+  "description": "Share VS Code context with pi coding agent",
+  "repository": "https://github.com/ravshansbox/vscode-pi-companion",
+  "version": "0.0.10",
+  "publisher": "ravshansbox",
+  "keywords": [
+    "pi-package"
+  ],
+  "engines": {
+    "vscode": "^1.85.0"
+  },
+  "categories": [
+    "AI",
+    "Other"
+  ],
+  "main": "./out/extension.js",
+  "scripts": {
+    "compile": "tsc -p ./",
+    "watch": "tsc -watch -p ./",
+    "package": "vsce package",
+    "vscode:prepublish": "npm run compile"
+  },
+  "devDependencies": {
+    "@types/cors": "^2.8.19",
+    "@types/express": "^5.0.6",
+    "@types/node": "^18.0.0",
+    "@types/vscode": "^1.85.0",
+    "@types/ws": "^8.18.1",
+    "@vscode/vsce": "^3.7.1",
+    "esbuild": "^0.27.4",
+    "typescript": "^5.0.0"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "cors": "^2.8.5",
+    "express": "^4.18.2",
+    "ws": "^8.16.0",
+    "zod": "^3.22.4"
+  },
+  "pi": {
+    "extensions": [
+      "./extensions"
+    ]
+  },
+  "activationEvents": [
+    "onStartupFinished"
+  ],
+  "contributes": {
+    "configuration": {
+      "title": "PI Companion",
+      "properties": {
+        "piCompanion.port": {
+          "type": "number",
+          "default": 8765,
+          "description": "Port for MCP server"
+        }
+      }
+    },
+    "commands": [
+      {
+        "command": "pi-companion.run",
+        "title": "PI Companion: Run PI"
+      },
+      {
+        "command": "pi-companion.start",
+        "title": "PI Companion: Start Server"
+      },
+      {
+        "command": "pi-companion.stop",
+        "title": "PI Companion: Stop Server"
+      },
+      {
+        "command": "pi-companion.status",
+        "title": "PI Companion: Show Status"
+      }
+    ],
+    "keybindings": [
+      {
+        "command": "pi-companion.run",
+        "key": "cmd+escape",
+        "mac": "cmd+escape",
+        "when": "editorTextFocus"
+      }
+    ]
+  }
+}