@rotriz/pi-web-ui 1.2.0 → 1.3.0

package/extension.mjs

@@ -1,153 +1,49 @@
 // pi-web-ui extension entry point for the Pi coding agent.
-// Registers the /web command and starts the HTTP/SSE gateway.
+// Runs the HTTP/SSE gateway in-process (no subprocess needed).
 
-import { homedir } from "node:os";
-import { dirname, join } from "node:path";
-import { exec, execFile } from "node:child_process";
-import { fileURLToPath } from "node:url";
-import { existsSync } from "node:fs";
-import { request } from "node:http";
+import { exec } from "node:child_process";
 
 const PORT = Number(process.env.PI_WEB_PORT || process.env.PORT || 3123);
 
-// HTTP GET helper — works on Node 14+ without global fetch
-function httpGet(url) {
-  return new Promise((resolve) => {
-    const req = request(url, (res) => {
-      let data = "";
-      res.on("data", (chunk) => (data += chunk));
-      res.on("end", () => resolve({ ok: res.statusCode >= 200 && res.statusCode < 300, status: res.statusCode, data }));
-    });
-    req.on("error", () => resolve({ ok: false, status: 0, data: "" }));
-    req.setTimeout(3000, () => { req.destroy(); resolve({ ok: false, status: 0, data: "" }); });
-    req.end();
-  });
-}
-
 function openBrowser(url) {
   const cmd = process.platform === "darwin" ? "open" : process.platform === "win32" ? "start" : "xdg-open";
   exec(`${cmd} ${JSON.stringify(url)}`, () => {});
 }
 
-// Resolve the pi-coding-agent module path from within Pi's process context
-function findPiModulePath() {
-  try {
-    // When running inside Pi, the module is already loaded — find it via require.resolve or import.meta
-    const candidates = [];
-    // Check if pi's entry script reveals the path
-    const entry = process.argv[1] || "";
-    if (entry) {
-      const dir = dirname(entry);
-      candidates.push(join(dir, "..", "dist", "index.js"));
-      candidates.push(join(dir, "index.js"));
-      // Go up to find the package
-      let d = dir;
-      for (let i = 0; i < 5; i++) {
-        const candidate = join(d, "dist", "index.js");
-        if (existsSync(candidate)) candidates.push(candidate);
-        const pkgCandidate = join(d, "node_modules", "@earendil-works", "pi-coding-agent", "dist", "index.js");
-        if (existsSync(pkgCandidate)) candidates.push(pkgCandidate);
-        d = dirname(d);
-      }
-    }
-    // Check common global locations
-    const globalDirs = [
-      join(homedir(), ".local", "share", "fnm"),
-      "/usr/local/lib/node_modules",
-      "/usr/lib/node_modules",
-      join(homedir(), ".nvm", "versions", "node"),
-      join(homedir(), ".volta", "tools", "image", "packages"),
-    ];
-    for (const base of globalDirs) {
-      if (!existsSync(base)) continue;
-      // Recursively look for the package
-      const piPkg = "@earendil-works/pi-coding-agent/dist/index.js";
-      // Check direct path
-      const direct = join(base, piPkg);
-      if (existsSync(direct)) { candidates.push(direct); continue; }
-      // For version managers, search one level deep
-    }
-    // Also check NODE_PATH
-    if (process.env.NODE_PATH) {
-      for (const p of process.env.NODE_PATH.split(":")) {
-        const candidate = join(p, "@earendil-works", "pi-coding-agent", "dist", "index.js");
-        if (existsSync(candidate)) candidates.push(candidate);
-      }
-    }
-    for (const c of candidates) {
-      if (existsSync(c)) return c;
-    }
-  } catch {}
-  return null;
-}
-
 export default function (pi) {
-  let serverProcess = null;
-  let origin = `http://localhost:${PORT}`;
-
-  const extDir = (() => {
-    try {
-      return dirname(fileURLToPath(import.meta.url));
-    } catch {
-      return join(homedir(), ".pi", "agent", "extensions", "pi-web-ui");
-    }
-  })();
-
-  function findNode() {
-    return process.execPath;
-  }
+  let serverStarted = false;
+  let startPromise = null;
+  const origin = `http://localhost:${PORT}`;
 
   async function ensureServer() {
-    // Check if server is already running (from a previous session or manual start)
-    try {
-      const res = await httpGet(`${origin}/api/tabs`);
-      if (res.ok) return;
-    } catch {}
-
-    if (serverProcess && !serverProcess.killed) return;
-    const serverPath = join(extDir, "server.mjs");
-    if (!existsSync(serverPath)) {
-      throw new Error(`pi-web-ui server not found at ${serverPath}`);
-    }
-
-    // Resolve the pi module path and pass it to the server
-    const piModulePath = findPiModulePath();
-    const env = { ...process.env, PORT: String(PORT) };
-    if (piModulePath) env.PI_WEB_PI_MODULE = piModulePath;
-
-    // Also pass NODE_PATH so the subprocess can find global modules
-    const nodePath = [];
-    if (process.env.NODE_PATH) nodePath.push(process.env.NODE_PATH);
-    // Add the global node_modules from the current node binary
-    const nodeDir = dirname(dirname(process.execPath));
-    const globalNM = join(nodeDir, "lib", "node_modules");
-    if (existsSync(globalNM)) nodePath.push(globalNM);
-    if (nodePath.length) env.NODE_PATH = nodePath.join(":");
-
-    serverProcess = execFile(findNode(), [serverPath], {
-      cwd: process.cwd(),
-      env,
-      stdio: "ignore",
-    });
-    serverProcess.unref();
-    serverProcess.on("exit", () => { serverProcess = null; });
-
-    // Wait for the server to be ready (up to 30 seconds)
-    for (let i = 0; i < 60; i++) {
-      await new Promise((r) => setTimeout(r, 500));
+    if (serverStarted) return;
+    if (startPromise) return startPromise;
+    startPromise = (async () => {
       try {
-        const res = await httpGet(`${origin}/api/tabs`);
-        if (res.ok) return;
-      } catch {}
-    }
-    throw new Error("pi-web-ui server did not start in time");
+        const { startServer } = await import("./server.mjs");
+        await startServer(PORT);
+        serverStarted = true;
+      } catch (err) {
+        // Server may already be listening (port in use = already running)
+        if (err?.code === "EADDRINUSE") {
+          serverStarted = true;
+          return;
+        }
+        throw err;
+      } finally {
+        startPromise = null;
+      }
+    })();
+    return startPromise;
   }
 
-  async function stopServer() {
-    if (serverProcess && !serverProcess.killed) {
-      serverProcess.kill();
-      serverProcess = null;
-    }
+  async function stopServerFn() {
+    if (!serverStarted) return;
+    try {
+      const { stopServer } = await import("./server.mjs");
+      await stopServer();
+      serverStarted = false;
+    } catch {}
   }
 
   // ── Lifecycle ────────────────────────────────────────────────────────
@@ -159,7 +55,7 @@ export default function (pi) {
 
   pi.on("session_shutdown", async (event) => {
     if (event.reason === "reload" || event.reason === "quit") {
-      await stopServer();
+      await stopServerFn();
     }
   });
 
@@ -173,7 +69,7 @@ export default function (pi) {
     handler: async (args, ctx) => {
       const arg = String(args ?? "").trim().toLowerCase();
       if (arg === "stop") {
-        await stopServer();
+        await stopServerFn();
         ctx.ui?.notify?.("pi-web-ui stopped", "info");
         return;
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rotriz/pi-web-ui",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "Web UI extension for the Pi coding agent — browser-based session management, git integration, and task tracking",
   "type": "module",
   "license": "MIT",

package/server.mjs

@@ -1402,5 +1402,24 @@ const server = createServer(async (req, res) => {
     json(res, 500, { error: String(err?.stack ?? err) });
   }
 });
+// When run directly (node server.mjs), start listening immediately.
+// When imported as a module (by extension.mjs), export control functions.
+const isDirectRun = process.argv[1] && resolve(process.argv[1]) === resolve(fileURLToPath(import.meta.url));
 
-server.listen(PORT, () => console.log(`[pi-web-ui] listening on http://localhost:${PORT}`));
+if (isDirectRun) {
+  server.listen(PORT, () => console.log(`[pi-web-ui] listening on http://localhost:${PORT}`));
+}
+
+export { server, PORT, shared };
+export function startServer(port) {
+  return new Promise((res, rej) => {
+    server.listen(port || PORT, () => {
+      console.log(`[pi-web-ui] listening on http://localhost:${port || PORT}`);
+      res();
+    });
+    server.on("error", rej);
+  });
+}
+export function stopServer() {
+  return new Promise((res) => server.close(() => res()));
+}