agent-browser-loop 0.3.0 → 0.3.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-browser-loop",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Let your AI coding agent drive a browser to verify its own work",
   "license": "MIT",
   "author": "Jason Silberman",

package/src/cli.ts

@@ -13,6 +13,7 @@ import {
   string,
   subcommands,
 } from "cmd-ts";
+import { VERSION } from "./version";
 import type { AgentBrowserOptions } from "./browser";
 import type { StepAction } from "./commands";
 import { parseBrowserConfig } from "./config";
@@ -89,6 +90,8 @@ async function loadConfig(configPath: string): Promise<BrowserCliConfig> {
   return parseBrowserConfig(exported);
 }
 
+// ============================================================================
+// Shared CLI Options
 // ============================================================================
 // Shared CLI Options
 // ============================================================================
@@ -579,13 +582,9 @@ const waitCommand = command({
       process.exit(1);
     }
 
-    const client = new DaemonClient(args.session);
-    if (!(await client.ping())) {
-      console.error(
-        "Daemon not running. Use 'agent-browser open <url>' first.",
-      );
-      process.exit(1);
-    }
+    const client = await ensureDaemon(args.session ?? "default", undefined, {
+      createIfMissing: false,
+    });
 
     const response = await client.wait(condition, { timeoutMs: args.timeout });
 
@@ -612,13 +611,9 @@ const stateCommand = command({
     json: jsonFlag,
   },
   handler: async (args) => {
-    const client = new DaemonClient(args.session);
-    if (!(await client.ping())) {
-      console.error(
-        "Daemon not running. Use 'agent-browser open <url>' first.",
-      );
-      process.exit(1);
-    }
+    const client = await ensureDaemon(args.session ?? "default", undefined, {
+      createIfMissing: false,
+    });
 
     const response = await client.state({
       format: args.json ? "json" : "text",
@@ -656,13 +651,9 @@ const screenshotCommand = command({
     }),
   },
   handler: async (args) => {
-    const client = new DaemonClient(args.session);
-    if (!(await client.ping())) {
-      console.error(
-        "Daemon not running. Use 'agent-browser open <url>' first.",
-      );
-      process.exit(1);
-    }
+    const client = await ensureDaemon(args.session ?? "default", undefined, {
+      createIfMissing: false,
+    });
 
     const response = await client.screenshot({
       fullPage: args.fullPage,
@@ -702,13 +693,9 @@ const resizeCommand = command({
     json: jsonFlag,
   },
   handler: async (args) => {
-    const client = new DaemonClient(args.session);
-    if (!(await client.ping())) {
-      console.error(
-        "Daemon not running. Use 'agent-browser open <url>' first.",
-      );
-      process.exit(1);
-    }
+    const client = await ensureDaemon(args.session ?? "default", undefined, {
+      createIfMissing: false,
+    });
 
     const response = await client.command({
       type: "resize",
@@ -1143,14 +1130,9 @@ const profileSaveCommand = command({
     }),
   },
   handler: async (args) => {
-    const client = new DaemonClient(args.session);
-
-    if (!(await client.ping())) {
-      console.error(
-        "Daemon not running. Use 'agent-browser open <url>' first to start a session.",
-      );
-      process.exit(1);
-    }
+    const client = await ensureDaemon(args.session ?? "default", undefined, {
+      createIfMissing: false,
+    });
 
     // Get storage state from session via command
     const response = await client.command({
@@ -1352,6 +1334,7 @@ const profileCommand = subcommands({
 
 const cli = subcommands({
   name: "agent-browser",
+  version: VERSION,
   cmds: {
     // Primary CLI commands (daemon-based)
     open: openCommand,
@@ -1378,10 +1361,11 @@ const cli = subcommands({
 });
 
 run(cli, process.argv.slice(2)).catch((error) => {
-  log
-    .withError(error)
-    .withMetadata({ argv: process.argv.slice(2) })
-    .error("CLI failed");
-  console.error(error);
+  // Print clean error message for user-facing errors
+  if (error instanceof Error) {
+    console.error(`Error: ${error.message}`);
+  } else {
+    console.error(error);
+  }
   process.exit(1);
 });

package/src/daemon.ts

@@ -3,6 +3,7 @@ import * as net from "node:net";
 import * as os from "node:os";
 import * as path from "node:path";
 import { z } from "zod";
+import { VERSION } from "./version";
 import { type AgentBrowserOptions, createBrowser } from "./browser";
 import {
   type Command,
@@ -157,6 +158,25 @@ export function getConfigPath(): string {
   return path.join(DAEMON_DIR, "daemon.config.json");
 }
 
+export function getVersionPath(): string {
+  return path.join(DAEMON_DIR, "daemon.version");
+}
+
+/**
+ * Get the version of the currently running daemon (if any)
+ */
+export function getDaemonVersion(): string | null {
+  const versionPath = getVersionPath();
+  if (!fs.existsSync(versionPath)) {
+    return null;
+  }
+  try {
+    return fs.readFileSync(versionPath, "utf-8").trim();
+  } catch {
+    return null;
+  }
+}
+
 // ============================================================================
 // Daemon Status
 // ============================================================================
@@ -183,6 +203,7 @@ export function cleanupDaemonFiles(): void {
   const socketPath = getSocketPath();
   const pidPath = getPidPath();
   const configPath = getConfigPath();
+  const versionPath = getVersionPath();
 
   try {
     if (fs.existsSync(socketPath)) fs.unlinkSync(socketPath);
@@ -193,6 +214,9 @@ export function cleanupDaemonFiles(): void {
   try {
     if (fs.existsSync(configPath)) fs.unlinkSync(configPath);
   } catch {}
+  try {
+    if (fs.existsSync(versionPath)) fs.unlinkSync(versionPath);
+  } catch {}
 }
 
 // ============================================================================
@@ -210,10 +234,14 @@ export async function startDaemon(options: DaemonOptions = {}): Promise<void> {
   const socketPath = getSocketPath();
   const pidPath = getPidPath();
   const configPath = getConfigPath();
+  const versionPath = getVersionPath();
 
   ensureDaemonDir();
   cleanupDaemonFiles();
 
+  // Write version file so CLI can detect version mismatch
+  fs.writeFileSync(versionPath, VERSION);
+
   // Multi-session state
   const sessions = new Map<string, DaemonSession>();
   const idGenerator = createIdGenerator();
@@ -814,6 +842,34 @@ export class DaemonClient {
 // Daemon Spawner
 // ============================================================================
 
+/**
+ * Force restart the daemon by shutting down any existing one
+ */
+async function forceRestartDaemon(
+  browserOptions?: AgentBrowserOptions,
+): Promise<void> {
+  const client = new DaemonClient();
+
+  // Try to gracefully shutdown existing daemon
+  if (isDaemonRunning()) {
+    try {
+      if (await client.ping()) {
+        await client.shutdown();
+        // Wait a bit for shutdown
+        await new Promise((r) => setTimeout(r, 200));
+      }
+    } catch {
+      // Ignore errors during shutdown
+    }
+  }
+
+  // Clean up any stale files
+  cleanupDaemonFiles();
+
+  // Spawn fresh daemon
+  await spawnDaemon(browserOptions);
+}
+
 /**
  * Ensure daemon is running and return a client.
  * If sessionId is provided, set the client to use that session.
@@ -829,6 +885,41 @@ export async function ensureDaemon(
 
   // Check if daemon is already running
   if (isDaemonRunning()) {
+    // Check for version mismatch - force restart if versions don't match
+    const daemonVersion = getDaemonVersion();
+    if (daemonVersion && daemonVersion !== VERSION) {
+      // If we're not allowed to create sessions, tell user to re-open
+      if (!createIfMissing) {
+        throw new Error(
+          `Daemon was upgraded (${daemonVersion} -> ${VERSION}). Please run 'agent-browser open <url>' to start a new session.`,
+        );
+      }
+
+      console.log(
+        `Daemon version mismatch (daemon: ${daemonVersion}, cli: ${VERSION}), restarting...`,
+      );
+      await forceRestartDaemon(browserOptions);
+
+      // Wait for new daemon to be ready
+      const maxAttempts = 50;
+      for (let i = 0; i < maxAttempts; i++) {
+        await new Promise((r) => setTimeout(r, 100));
+        if (await client.ping()) {
+          if (sessionId === "default") {
+            const createResp = await client.create({
+              sessionId: "default",
+              browserOptions,
+            });
+            if (!createResp.success) {
+              throw new Error(`Failed to create session: ${createResp.error}`);
+            }
+          }
+          return client;
+        }
+      }
+      throw new Error("Failed to restart daemon after version mismatch");
+    }
+
     // Verify it's responsive
     if (await client.ping()) {
       // Daemon is running, check if session exists or create default
@@ -893,6 +984,32 @@ export async function ensureDaemonNewSession(
 
   // Check if daemon is already running
   if (isDaemonRunning()) {
+    // Check for version mismatch - force restart if versions don't match
+    const daemonVersion = getDaemonVersion();
+    if (daemonVersion && daemonVersion !== VERSION) {
+      console.log(
+        `Daemon version mismatch (daemon: ${daemonVersion}, cli: ${VERSION}), restarting...`,
+      );
+      await forceRestartDaemon(browserOptions);
+
+      // Wait for new daemon to be ready and create session
+      const maxAttempts = 50;
+      for (let i = 0; i < maxAttempts; i++) {
+        await new Promise((r) => setTimeout(r, 100));
+        if (await client.ping()) {
+          const createResp = await client.create({ browserOptions });
+          if (!createResp.success) {
+            throw new Error(`Failed to create session: ${createResp.error}`);
+          }
+          const newSessionId = (createResp.data as { sessionId: string })
+            .sessionId;
+          client.setSession(newSessionId);
+          return client;
+        }
+      }
+      throw new Error("Failed to restart daemon after version mismatch");
+    }
+
     if (await client.ping()) {
       // Create new session with auto-generated ID
       const createResp = await client.create({ browserOptions });

package/src/index.ts

@@ -30,11 +30,15 @@ export {
   DaemonClient,
   type DaemonOptions,
   ensureDaemon,
+  getDaemonVersion,
   getPidPath,
   getSocketPath,
+  getVersionPath,
   isDaemonRunning,
   startDaemon,
 } from "./daemon";
+// Version
+export { VERSION } from "./version";
 // Profiles
 export {
   deleteProfile,

package/src/version.ts

@@ -0,0 +1,19 @@
+// Version is read from package.json at build/runtime
+// This provides a single source of truth for the package version
+
+import { readFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+
+function loadVersion(): string {
+  try {
+    // Try to read from package.json relative to this file
+    const packagePath = join(dirname(import.meta.dirname), "package.json");
+    const pkg = JSON.parse(readFileSync(packagePath, "utf-8"));
+    return pkg.version;
+  } catch {
+    // Fallback if package.json can't be read
+    return "0.0.0";
+  }
+}
+
+export const VERSION = loadVersion();