pi-msgpack-rpc 0.1.2 → 0.2.0

package/index.ts

@@ -1,5 +1,5 @@
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
 
 import { msgpackRpcCall } from "./rpc";
@@ -8,7 +8,7 @@ export default function (pi: ExtensionAPI) {
   pi.registerTool({
     name: "msgpack_rpc_call",
     label: "MessagePack RPC Call",
-    description: "Generic MessagePack RPC call (based on your working talk.ts script)",
+    description: "Generic MessagePack RPC Calls Interface",
     parameters: Type.Object({
       server: Type.String({ description: "Exact socket path from :echo v:servername inside Neovim" }),
       method: Type.String({ description: "e.g. nvim_get_api_info" }),
@@ -28,9 +28,9 @@ export default function (pi: ExtensionAPI) {
         throw new Error("server and method are required");
       }
 
-      ctx.ui.notify(`[RPC] Starting call to ${server} → ${method}`, "info");
+      ctx.ui.notify(`[Tool] Starting call to ${server} → ${method}`, "info");
 
-      const result = await msgpackRpcCall(ctx, server, method, args, timeout);
+      const result = await msgpackRpcCall(server, method, args, timeout, ctx);
 
       return {
         content: [{ type: "text", text: result }],
@@ -38,4 +38,20 @@ export default function (pi: ExtensionAPI) {
       };
     },
   });
+
+  pi.events.on("msgpack:rpc:call", async (payload: any, ctx?: any) => {
+    const { server, method, params = [], timeout = 30000, correlationId } = payload || {};
+
+    if (!server || !method || !correlationId) {
+      pi.events.emit("msgpack:rpc:response", { correlationId, success: false, error: "Missing required fields" });
+      return;
+    }
+
+    try {
+      const result = await msgpackRpcCall(server, method, params, timeout, ctx);
+      pi.events.emit("msgpack:rpc:response", { correlationId, success: true, result });
+    } catch (err: any) {
+      pi.events.emit("msgpack:rpc:response", { correlationId, success: false, error: err.message });
+    }
+  })
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-msgpack-rpc",
   "description": "A MessagePack RPC tool for Pi",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "main": "index.ts",
   "repository": {
     "type": "git",
@@ -11,5 +11,13 @@
   "dependencies": {
     "msgpackr": "^1.11.10"
   },
-  "pi": { "extensions": ["./index.ts"] }
+  "devDependencies": {
+    "@earendil-works/pi-coding-agent": "latest",
+    "@types/node": "^25.6.2"
+  },
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  }
 }

package/rpc.ts

@@ -4,30 +4,30 @@ import { pipeline } from 'node:stream';
 import { pack, UnpackrStream } from 'msgpackr';
 
 export async function msgpackRpcCall(
-    ctx: any,
     server: string,
     method: string,
     args : any[] = [],
-    timeout = 5000
+    timeout = 30000,
+    ctx?: any
   ) {
   return new Promise((resolve, reject) => {
     const unpackStream = new UnpackrStream();
 
     const client = net.createConnection(server, () => {
-      ctx.ui.notify(`[RPC] Connected to ${server}, calling ${method}`, "info");
+      if (ctx) ctx.ui.notify(`[RPC] Connected to ${server}, calling ${method}`, "info");
       client.write(pack([0, Date.now() % 1000000, method, args]));
     })
 
     pipeline(client, unpackStream, (err) => {
-      if (err) ctx.ui.notify(`[RPC] Pipeline error: ${err.message}`, "error");
+      if (err && ctx) ctx.ui.notify(`[RPC] Pipeline error: ${err.message}`, "error");
     });
 
     unpackStream.on('data', (msg: any) => {
-      ctx.ui.notify(`[RPC] Received response`, "success");
+      if (ctx) ctx.ui.notify(`[RPC] Received response`, "success");
       client.end();
 
       if (Array.isArray(msg) && msg[0] === 1) {
-        const error = msg[2]; const result = msg[3];
+        const error = msg[2]; const result = msg[3] || "n/a";
         error !== null && error !== undefined ? reject(error) : resolve(result);
       } else {
         resolve(msg);