vscode-pi-companion 0.2.4 → 0.2.6

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.2.4",
+  "version": "0.2.6",
   "publisher": "ravshansbox",
   "keywords": [
     "pi-package"
@@ -21,7 +21,7 @@
     "watch": "tsc -watch -p ./",
     "package": "vsce package",
     "vscode:prepublish": "npm run compile",
-    "test": "node --import ./test/setup.mjs --test test/*.test.mjs"
+    "test": "npm run compile && node --import ./test/setup.mjs --test test/*.test.mjs"
   },
   "devDependencies": {
     "@types/cors": "^2.8.19",

package/src/ide-server.ts

@@ -122,7 +122,7 @@ export class IDEServer {
           }
         };
 
-        await this.mcpServer!.connect(transport);
+        await this.createMcpServer().connect(transport);
       } else {
         res.status(400).json({
           jsonrpc: '2.0',
@@ -359,6 +359,7 @@ export class IDEServer {
     if (this.httpServer) {
       await new Promise<void>((resolve) => {
         this.httpServer!.close(() => resolve());
+        this.httpServer!.closeAllConnections();
       });
       this.httpServer = undefined;
     }

package/src/open-files-manager.ts

@@ -187,10 +187,10 @@ export class OpenFilesManager {
 
     // Update selection boundaries
     if (!editor.selection.isEmpty) {
-      const startLine = Math.min(editor.selection.anchor.line, editor.selection.active.line) + 1;
-      const endLine = Math.max(editor.selection.anchor.line, editor.selection.active.line) + 1;
-      const startChar = Math.min(editor.selection.anchor.character, editor.selection.active.character) + 1;
-      const endChar = Math.max(editor.selection.anchor.character, editor.selection.active.character) + 1;
+      const startLine = editor.selection.start.line + 1;
+      const endLine = editor.selection.end.line + 1;
+      const startChar = editor.selection.start.character + 1;
+      const endChar = editor.selection.end.character + 1;
       
       file.selectionStart = { line: startLine, character: startChar };
       file.selectionEnd = { line: endLine, character: endChar };