vscode-pi-companion 0.1.5 → 0.2.0

package/extensions/index.ts

@@ -47,6 +47,7 @@ let lastWidgetKey = "";
 let widgetDebounceTimer: NodeJS.Timeout | undefined;
 let reconnectTimer: NodeJS.Timeout | undefined;
 let sseRequest: http.ClientRequest | undefined;
+let isConnected = false;
 
 function isRunningInVSCode(): boolean {
   return process.env.TERM_PROGRAM === "vscode";
@@ -138,7 +139,9 @@ function updateWidget(): void {
   if (!currentCtx) return;
 
   if (!currentContext || !currentContext.workspaceState.openFiles.length) {
-    currentCtx.ui.setWidget("pi-companion", ["PI Companion: No files open"]);
+    // Show "Connected" only while waiting for file context, or "Disconnected" if not connected
+    const status = isConnected ? "VS Code: Connected" : "VS Code: Disconnected";
+    currentCtx.ui.setWidget("pi-companion", [status]);
     return;
   }
 
@@ -159,7 +162,7 @@ function updateWidget(): void {
 
   const otherCount = workspaceState.openFiles.length - 1;
   if (otherCount > 0) {
-    lines.push(`   +${otherCount} more file${otherCount > 1 ? "s" : ""} open`);
+    lines.push(`   +${otherCount} more file${otherCount > 1 ? "s" : ""}`);
   }
 
   const widgetKey = lines.join("|");
@@ -190,6 +193,8 @@ async function connectSSE(): Promise<void> {
       path: "/context/stream",
     },
     (res) => {
+      isConnected = true;
+      updateWidget();
       let buffer = "";
 
       res.on("data", (chunk: string) => {
@@ -260,7 +265,6 @@ export default function (pi: ExtensionAPI) {
 
     const port = await getLivePort();
     if (port) {
-      ctx.ui.notify("PI Companion: Connected to VS Code", "info");
       await connectSSE();
     } else {
       ctx.ui.notify("PI Companion: VS Code companion server not running. Run 'PI Companion: Start Server' in VS Code.", "warning");

package/package.json

@@ -3,7 +3,7 @@
   "displayName": "PI Companion",
   "description": "Share VS Code context with pi coding agent",
   "repository": "https://github.com/ravshansbox/vscode-pi-companion",
-  "version": "0.1.5",
+  "version": "0.2.0",
   "publisher": "ravshansbox",
   "keywords": [
     "pi-package"