@webmux/agent 0.1.4 → 0.2.1

package/dist/cli.js

@@ -1,10 +1,19 @@
 #!/usr/bin/env node
+import {
+  AGENT_PACKAGE_NAME,
+  AGENT_VERSION,
+  SERVICE_NAME,
+  installService,
+  readInstalledServiceConfig,
+  servicePath,
+  uninstallService,
+  upgradeService
+} from "./chunk-INUNCXBM.js";
 
 // src/cli.ts
-import fs2 from "fs";
 import os2 from "os";
-import path2 from "path";
-import { execSync as execSync2 } from "child_process";
+import { execFileSync } from "child_process";
+import { pathToFileURL } from "url";
 import { Command } from "commander";
 
 // src/credentials.ts
@@ -43,694 +52,202 @@ function saveCredentials(creds) {
   });
 }
 
-// src/tmux.ts
-import { execFile } from "child_process";
-import { promisify } from "util";
-var execFileAsync = promisify(execFile);
-var FIELD_SEPARATOR = "";
-var SESSION_NAME_PATTERN = /^[a-zA-Z0-9][a-zA-Z0-9._-]{0,31}$/;
-var TMUX_EMPTY_STATE_MARKERS = [
-  "error connecting to",
-  "failed to connect to server",
-  "no server running",
-  "no sessions"
-];
-var TmuxClient = class {
-  socketName;
-  workspaceRoot;
-  constructor(options) {
-    this.socketName = options.socketName;
-    this.workspaceRoot = options.workspaceRoot;
-  }
-  async listSessions() {
-    const stdout = await this.run(
-      [
-        "list-sessions",
-        "-F",
-        [
-          "#{session_name}",
-          "#{session_windows}",
-          "#{session_attached}",
-          "#{session_created}",
-          "#{session_activity}",
-          "#{session_path}",
-          "#{pane_current_command}"
-        ].join(FIELD_SEPARATOR)
-      ],
-      { allowEmptyState: true }
-    );
-    const sessions = parseSessionList(stdout);
-    const enriched = await Promise.all(
-      sessions.map(async (session) => ({
-        ...session,
-        preview: await this.getPreview(session.name)
-      }))
-    );
-    return enriched.sort((left, right) => {
-      if (left.lastActivityAt !== right.lastActivityAt) {
-        return right.lastActivityAt - left.lastActivityAt;
-      }
-      return left.name.localeCompare(right.name);
-    });
-  }
-  async createSession(name) {
-    assertValidSessionName(name);
-    if (await this.hasSession(name)) {
-      return;
-    }
-    await this.run(["new-session", "-d", "-s", name, "-c", this.workspaceRoot]);
-  }
-  async killSession(name) {
-    assertValidSessionName(name);
-    await this.run(["kill-session", "-t", name]);
-  }
-  async readSession(name) {
-    const sessions = await this.listSessions();
-    return sessions.find((session) => session.name === name) ?? null;
-  }
-  async hasSession(name) {
-    try {
-      await this.run(["has-session", "-t", name]);
-      return true;
-    } catch (error) {
-      const message = String(
-        error.stderr ?? error.message
-      );
-      if (isTmuxEmptyStateMessage(message)) {
-        return false;
-      }
-      return false;
-    }
-  }
-  async getPreview(name) {
-    try {
-      const stdout = await this.run(
-        ["capture-pane", "-p", "-J", "-S", "-18", "-E", "-", "-t", `${name}:`],
-        { allowEmptyState: true }
-      );
-      return formatPreview(stdout);
-    } catch {
-      return ["Session available. Tap to attach."];
-    }
-  }
-  async run(args, options = {}) {
-    try {
-      const { stdout } = await execFileAsync(
-        "tmux",
-        ["-L", this.socketName, ...args],
-        {
-          cwd: this.workspaceRoot,
-          env: {
-            ...process.env,
-            TERM: "xterm-256color"
-          }
-        }
-      );
-      return stdout;
-    } catch (error) {
-      const message = String(
-        error.stderr ?? error.message
-      );
-      if (options.allowEmptyState && TMUX_EMPTY_STATE_MARKERS.some((marker) => message.includes(marker))) {
-        return "";
-      }
-      throw error;
-    }
-  }
-};
-function assertValidSessionName(name) {
-  if (!SESSION_NAME_PATTERN.test(name)) {
-    throw new Error(
-      "Invalid session name. Use up to 32 letters, numbers, dot, dash, or underscore."
-    );
-  }
-}
-function parseSessionList(stdout) {
-  return stdout.split("\n").map((line) => line.trim()).filter(Boolean).flatMap((line) => {
-    const parts = line.split(FIELD_SEPARATOR);
-    const [name, windows, attachedClients, createdAt, lastActivityAt, path3] = parts;
-    const currentCommand = parts[6] ?? "";
-    if (!name || !windows || !attachedClients || !createdAt || !lastActivityAt || !path3) {
-      return [];
-    }
-    return [
-      {
-        name,
-        windows: Number(windows),
-        attachedClients: Number(attachedClients),
-        createdAt: Number(createdAt),
-        lastActivityAt: Number(lastActivityAt),
-        path: path3,
-        currentCommand
-      }
-    ];
-  });
-}
-function formatPreview(stdout) {
-  const lines = stdout.replaceAll("\r", "").split("\n").map((line) => line.trimEnd()).filter((line) => line.length > 0).slice(-3);
-  if (lines.length > 0) {
-    return lines;
-  }
-  return ["Fresh session. Nothing has run yet."];
-}
-function isTmuxEmptyStateMessage(message) {
-  return TMUX_EMPTY_STATE_MARKERS.some((marker) => message.includes(marker));
-}
-
-// src/connection.ts
-import { execSync } from "child_process";
-import WebSocket from "ws";
-
-// src/terminal.ts
-import { spawn } from "node-pty";
-
-// ../shared/src/contracts.ts
-var DEFAULT_TERMINAL_SIZE = {
-  cols: 120,
-  rows: 36
-};
-
-// src/terminal.ts
-async function createTerminalBridge(options) {
-  const {
-    tmux,
-    sessionName,
-    cols = DEFAULT_TERMINAL_SIZE.cols,
-    rows = DEFAULT_TERMINAL_SIZE.rows,
-    onData,
-    onExit
-  } = options;
-  assertValidSessionName(sessionName);
-  await tmux.createSession(sessionName);
-  const ptyProcess = spawn(
-    "tmux",
-    ["-L", tmux.socketName, "attach-session", "-t", sessionName],
-    {
-      cols,
-      rows,
-      cwd: tmux.workspaceRoot,
-      env: {
-        ...process.env,
-        TERM: "xterm-256color"
-      },
-      name: "xterm-256color"
-    }
-  );
-  ptyProcess.onData(onData);
-  ptyProcess.onExit(({ exitCode }) => {
-    onExit(exitCode);
-  });
-  return {
-    write(data) {
-      ptyProcess.write(data);
-    },
-    resize(nextCols, nextRows) {
-      ptyProcess.resize(nextCols, nextRows);
-    },
-    dispose() {
-      ptyProcess.kill();
-    }
-  };
+// src/cli.ts
+async function loadAgentRuntime() {
+  const [{ AgentConnection }, { TmuxClient }] = await Promise.all([
+    import("./connection-RJY775NL.js"),
+    import("./tmux-QIB4H3UA.js")
+  ]);
+  return { AgentConnection, TmuxClient };
 }
-
-// src/connection.ts
-var AGENT_VERSION = "0.1.4";
-var HEARTBEAT_INTERVAL_MS = 3e4;
-var SESSION_SYNC_INTERVAL_MS = 15e3;
-var INITIAL_RECONNECT_DELAY_MS = 1e3;
-var MAX_RECONNECT_DELAY_MS = 3e4;
-var AgentConnection = class {
-  serverUrl;
-  agentId;
-  agentSecret;
-  tmux;
-  ws = null;
-  heartbeatTimer = null;
-  sessionSyncTimer = null;
-  reconnectTimer = null;
-  reconnectDelay = INITIAL_RECONNECT_DELAY_MS;
-  bridges = /* @__PURE__ */ new Map();
-  stopped = false;
-  constructor(serverUrl, agentId, agentSecret, tmux) {
-    this.serverUrl = serverUrl;
-    this.agentId = agentId;
-    this.agentSecret = agentSecret;
-    this.tmux = tmux;
-  }
-  start() {
-    this.stopped = false;
-    this.connect();
-  }
-  stop() {
-    this.stopped = true;
-    if (this.reconnectTimer) {
-      clearTimeout(this.reconnectTimer);
-      this.reconnectTimer = null;
-    }
-    this.stopHeartbeat();
-    this.stopSessionSync();
-    this.disposeAllBridges();
-    if (this.ws) {
-      this.ws.close(1e3, "agent shutting down");
-      this.ws = null;
+function createProgram() {
+  const program = new Command();
+  program.name("webmux-agent").description("Webmux agent \u2014 connects your machine to the webmux server").version(AGENT_VERSION);
+  program.command("register").description("Register this agent with a webmux server").requiredOption("--server <url>", "Server URL (e.g. https://webmux.example.com)").requiredOption("--token <token>", "One-time registration token from the server").option("--name <name>", "Display name for this agent (defaults to hostname)").action(async (opts) => {
+    const serverUrl = opts.server.replace(/\/+$/, "");
+    const agentName = opts.name ?? os2.hostname();
+    console.log(`[agent] Registering with server ${serverUrl}...`);
+    console.log(`[agent] Agent name: ${agentName}`);
+    const body = {
+      token: opts.token,
+      name: agentName
+    };
+    let response;
+    try {
+      response = await fetch(`${serverUrl}/api/agents/register`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify(body)
+      });
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      console.error(`[agent] Failed to connect to server: ${message}`);
+      process.exit(1);
     }
-  }
-  connect() {
-    const wsUrl = buildWsUrl(this.serverUrl);
-    console.log(`[agent] Connecting to ${wsUrl}`);
-    const ws = new WebSocket(wsUrl);
-    this.ws = ws;
-    ws.on("open", () => {
-      console.log("[agent] WebSocket connected, authenticating...");
-      this.reconnectDelay = INITIAL_RECONNECT_DELAY_MS;
-      this.sendMessage({ type: "auth", agentId: this.agentId, agentSecret: this.agentSecret, version: AGENT_VERSION });
-    });
-    ws.on("message", (raw) => {
-      let msg;
+    if (!response.ok) {
+      let errorMessage;
       try {
-        msg = JSON.parse(raw.toString());
+        const errorBody = await response.json();
+        errorMessage = errorBody.error ?? response.statusText;
       } catch {
-        console.error("[agent] Failed to parse server message:", raw.toString());
-        return;
+        errorMessage = response.statusText;
       }
-      this.handleMessage(msg);
-    });
-    ws.on("close", (code, reason) => {
-      console.log(`[agent] WebSocket closed: code=${code} reason=${reason.toString()}`);
-      this.onDisconnect();
+      console.error(`[agent] Registration failed: ${errorMessage}`);
+      process.exit(1);
+    }
+    const result = await response.json();
+    saveCredentials({
+      serverUrl,
+      agentId: result.agentId,
+      agentSecret: result.agentSecret,
+      name: agentName
     });
-    ws.on("error", (err) => {
-      console.error("[agent] WebSocket error:", err.message);
+    console.log(`[agent] Registration successful!`);
+    console.log(`[agent] Agent ID: ${result.agentId}`);
+    console.log(`[agent] Credentials saved to ${credentialsPath()}`);
+    console.log(``);
+    console.log(`Next steps:`);
+    console.log(`  pnpm dlx @webmux/agent start         # run once`);
+    console.log(`  pnpm dlx @webmux/agent service install    # install as managed systemd service`);
+  });
+  program.command("start").description("Start the agent and connect to the server").action(async () => {
+    const creds = loadCredentials();
+    if (!creds) {
+      console.error(
+        `[agent] No credentials found at ${credentialsPath()}. Run "npx @webmux/agent register" first.`
+      );
+      process.exit(1);
+    }
+    const { AgentConnection, TmuxClient } = await loadAgentRuntime();
+    console.log(`[agent] Starting agent "${creds.name}"...`);
+    console.log(`[agent] Server: ${creds.serverUrl}`);
+    console.log(`[agent] Agent ID: ${creds.agentId}`);
+    const tmux = new TmuxClient({
+      socketName: "webmux",
+      workspaceRoot: os2.homedir()
     });
-  }
-  handleMessage(msg) {
-    switch (msg.type) {
-      case "auth-ok":
-        console.log("[agent] Authenticated successfully");
-        if (msg.latestVersion && msg.latestVersion !== AGENT_VERSION) {
-          console.log(`[agent] Update available: ${AGENT_VERSION} \u2192 ${msg.latestVersion}`);
-          this.selfUpdate(msg.latestVersion);
-          return;
-        }
-        this.startHeartbeat();
-        this.startSessionSync();
-        this.syncSessions();
-        break;
-      case "auth-fail":
-        console.error(`[agent] Authentication failed: ${msg.message}`);
-        this.stopped = true;
-        if (this.ws) {
-          this.ws.close();
-          this.ws = null;
-        }
-        process.exit(1);
-        break;
-      case "sessions-list":
-        this.syncSessions();
-        break;
-      case "terminal-attach":
-        this.handleTerminalAttach(msg.browserId, msg.sessionName, msg.cols, msg.rows);
-        break;
-      case "terminal-detach":
-        this.handleTerminalDetach(msg.browserId);
-        break;
-      case "terminal-input":
-        this.handleTerminalInput(msg.browserId, msg.data);
-        break;
-      case "terminal-resize":
-        this.handleTerminalResize(msg.browserId, msg.cols, msg.rows);
-        break;
-      case "session-create":
-        this.handleSessionCreate(msg.requestId, msg.name);
-        break;
-      case "session-kill":
-        this.handleSessionKill(msg.requestId, msg.name);
-        break;
-      default:
-        console.warn("[agent] Unknown message type:", msg.type);
-    }
-  }
-  async syncSessions() {
-    try {
-      const sessions = await this.tmux.listSessions();
-      this.sendMessage({ type: "sessions-sync", sessions });
-      return sessions;
-    } catch (err) {
-      console.error("[agent] Failed to list sessions:", err);
-      this.sendMessage({ type: "error", message: "Failed to list sessions" });
-      return [];
-    }
-  }
-  async handleTerminalAttach(browserId, sessionName, cols, rows) {
-    const existing = this.bridges.get(browserId);
-    if (existing) {
-      existing.dispose();
-      this.bridges.delete(browserId);
-    }
+    const connection = new AgentConnection(
+      creds.serverUrl,
+      creds.agentId,
+      creds.agentSecret,
+      tmux
+    );
+    const shutdown = () => {
+      console.log("\n[agent] Shutting down...");
+      connection.stop();
+      process.exit(0);
+    };
+    process.on("SIGINT", shutdown);
+    process.on("SIGTERM", shutdown);
+    connection.start();
+  });
+  program.command("status").description("Show agent status and credentials info").action(() => {
+    const creds = loadCredentials();
+    if (!creds) {
+      console.log(`[agent] Not registered. No credentials found at ${credentialsPath()}.`);
+      process.exit(0);
+    }
+    console.log(`Agent Name:       ${creds.name}`);
+    console.log(`Agent Version:    ${AGENT_VERSION}`);
+    console.log(`Server URL:       ${creds.serverUrl}`);
+    console.log(`Agent ID:         ${creds.agentId}`);
+    console.log(`Credentials File: ${credentialsPath()}`);
+    const installedService = readInstalledServiceConfig();
     try {
-      const bridge = await createTerminalBridge({
-        tmux: this.tmux,
-        sessionName,
-        cols,
-        rows,
-        onData: (data) => {
-          this.sendMessage({ type: "terminal-output", browserId, data });
-        },
-        onExit: (exitCode) => {
-          this.bridges.delete(browserId);
-          this.sendMessage({ type: "terminal-exit", browserId, exitCode });
-          void this.syncSessions();
-        }
-      });
-      this.bridges.set(browserId, bridge);
-      this.sendMessage({ type: "terminal-ready", browserId, sessionName });
-      await this.syncSessions();
-    } catch (err) {
-      const message = err instanceof Error ? err.message : String(err);
-      console.error(`[agent] Failed to attach terminal for browser ${browserId}:`, message);
-      this.sendMessage({ type: "error", browserId, message: `Failed to attach: ${message}` });
+      const result = execFileSync("systemctl", ["--user", "is-active", SERVICE_NAME], { encoding: "utf-8" }).trim();
+      console.log(`Service:          ${result}`);
+    } catch {
+      console.log(`Service:          not installed`);
     }
-  }
-  handleTerminalDetach(browserId) {
-    const bridge = this.bridges.get(browserId);
-    if (bridge) {
-      bridge.dispose();
-      this.bridges.delete(browserId);
-      void this.syncSessions();
+    if (installedService?.version) {
+      console.log(`Service Version:  ${installedService.version}`);
     }
-  }
-  handleTerminalInput(browserId, data) {
-    const bridge = this.bridges.get(browserId);
-    if (bridge) {
-      bridge.write(data);
-    }
-  }
-  handleTerminalResize(browserId, cols, rows) {
-    const bridge = this.bridges.get(browserId);
-    if (bridge) {
-      bridge.resize(cols, rows);
+  });
+  const service = program.command("service").description("Manage the systemd service");
+  service.command("install").description("Install and start the agent as a managed systemd user service").option("--no-auto-upgrade", "Disable automatic upgrades for the managed service").action((opts) => {
+    const creds = loadCredentials();
+    if (!creds) {
+      console.error(`[agent] Not registered. Run "npx @webmux/agent register" first.`);
+      process.exit(1);
     }
-  }
-  async handleSessionCreate(requestId, name) {
     try {
-      await this.tmux.createSession(name);
-      const sessions = await this.syncSessions();
-      const session = sessions.find((item) => item.name === name);
-      if (!session) {
-        throw new Error("Created session was not returned by tmux");
-      }
-      this.sendMessage({ type: "command-result", requestId, ok: true, session });
+      installService({
+        agentName: creds.name,
+        packageName: AGENT_PACKAGE_NAME,
+        version: AGENT_VERSION,
+        autoUpgrade: opts.autoUpgrade
+      });
+      console.log(``);
+      console.log(`[agent] Service installed and started!`);
+      console.log(`[agent] Managed version: ${AGENT_VERSION}`);
+      console.log(`[agent] It will auto-start on boot.`);
+      console.log(``);
+      console.log(`Useful commands:`);
+      console.log(`  systemctl --user status ${SERVICE_NAME}`);
+      console.log(`  journalctl --user -u ${SERVICE_NAME} -f`);
+      console.log(`  pnpm dlx @webmux/agent service upgrade --to <version>`);
+      console.log(`  pnpm dlx @webmux/agent service uninstall`);
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
-      console.error(`[agent] Failed to create session "${name}":`, message);
-      this.sendMessage({ type: "command-result", requestId, ok: false, error: message });
+      console.error(`[agent] Failed to install managed service: ${message}`);
+      console.error(`[agent] Service file path: ${servicePath()}`);
+      process.exit(1);
     }
-  }
-  async handleSessionKill(requestId, name) {
+  });
+  service.command("uninstall").description("Stop and remove the systemd user service").action(() => {
     try {
-      await this.tmux.killSession(name);
-      await this.syncSessions();
-      this.sendMessage({ type: "command-result", requestId, ok: true });
+      uninstallService();
+      console.log(`[agent] Service file removed: ${servicePath()}`);
+      console.log(`[agent] Service uninstalled.`);
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
-      console.error(`[agent] Failed to kill session "${name}":`, message);
-      this.sendMessage({ type: "command-result", requestId, ok: false, error: message });
+      console.error(`[agent] Failed to uninstall service: ${message}`);
+      process.exit(1);
     }
-  }
-  selfUpdate(targetVersion) {
-    console.log(`[agent] Installing @webmux/agent@${targetVersion}...`);
+  });
+  service.command("status").description("Show systemd service status").action(() => {
     try {
-      execSync(`npm install -g @webmux/agent@${targetVersion}`, { stdio: "inherit" });
-      console.log("[agent] Update installed. Restarting...");
-    } catch (err) {
-      console.error("[agent] Update failed:", err instanceof Error ? err.message : err);
-      console.log("[agent] Continuing with current version");
-      this.startHeartbeat();
-      this.startSessionSync();
-      this.syncSessions();
-      return;
-    }
-    this.stop();
-    process.exit(0);
-  }
-  sendMessage(msg) {
-    if (this.ws && this.ws.readyState === WebSocket.OPEN) {
-      this.ws.send(JSON.stringify(msg));
-    }
-  }
-  startHeartbeat() {
-    this.stopHeartbeat();
-    this.heartbeatTimer = setInterval(() => {
-      this.sendMessage({ type: "heartbeat" });
-    }, HEARTBEAT_INTERVAL_MS);
-  }
-  startSessionSync() {
-    this.stopSessionSync();
-    this.sessionSyncTimer = setInterval(() => {
-      void this.syncSessions();
-    }, SESSION_SYNC_INTERVAL_MS);
-  }
-  stopHeartbeat() {
-    if (this.heartbeatTimer) {
-      clearInterval(this.heartbeatTimer);
-      this.heartbeatTimer = null;
-    }
-  }
-  stopSessionSync() {
-    if (this.sessionSyncTimer) {
-      clearInterval(this.sessionSyncTimer);
-      this.sessionSyncTimer = null;
-    }
-  }
-  disposeAllBridges() {
-    for (const [browserId, bridge] of this.bridges) {
-      bridge.dispose();
-      this.bridges.delete(browserId);
+      execFileSync("systemctl", ["--user", "status", SERVICE_NAME], { stdio: "inherit" });
+    } catch {
+      console.log(`[agent] Service is not installed or not running.`);
     }
-  }
-  onDisconnect() {
-    this.stopHeartbeat();
-    this.stopSessionSync();
-    this.disposeAllBridges();
-    this.ws = null;
-    if (this.stopped) {
-      return;
+  });
+  service.command("upgrade").description("Switch the managed service to a specific agent version and restart it").requiredOption("--to <version>", "Target agent version (for example 0.1.5)").action((opts) => {
+    const creds = loadCredentials();
+    if (!creds) {
+      console.error(`[agent] Not registered. Run "npx @webmux/agent register" first.`);
+      process.exit(1);
     }
-    console.log(`[agent] Reconnecting in ${this.reconnectDelay}ms...`);
-    this.reconnectTimer = setTimeout(() => {
-      this.reconnectTimer = null;
-      this.connect();
-    }, this.reconnectDelay);
-    this.reconnectDelay = Math.min(this.reconnectDelay * 2, MAX_RECONNECT_DELAY_MS);
-  }
-};
-function buildWsUrl(serverUrl) {
-  const url = new URL("/ws/agent", serverUrl);
-  url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
-  return url.toString();
-}
-
-// src/cli.ts
-var SERVICE_NAME = "webmux-agent";
-var program = new Command();
-program.name("webmux-agent").description("Webmux agent \u2014 connects your machine to the webmux server").version("0.0.0");
-program.command("register").description("Register this agent with a webmux server").requiredOption("--server <url>", "Server URL (e.g. https://webmux.example.com)").requiredOption("--token <token>", "One-time registration token from the server").option("--name <name>", "Display name for this agent (defaults to hostname)").action(async (opts) => {
-  const serverUrl = opts.server.replace(/\/+$/, "");
-  const agentName = opts.name ?? os2.hostname();
-  console.log(`[agent] Registering with server ${serverUrl}...`);
-  console.log(`[agent] Agent name: ${agentName}`);
-  const body = {
-    token: opts.token,
-    name: agentName
-  };
-  let response;
-  try {
-    response = await fetch(`${serverUrl}/api/agents/register`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify(body)
-    });
-  } catch (err) {
-    const message = err instanceof Error ? err.message : String(err);
-    console.error(`[agent] Failed to connect to server: ${message}`);
-    process.exit(1);
-  }
-  if (!response.ok) {
-    let errorMessage;
+    console.log(`[agent] Switching managed service to ${opts.to}...`);
     try {
-      const errorBody = await response.json();
-      errorMessage = errorBody.error ?? response.statusText;
-    } catch {
-      errorMessage = response.statusText;
+      upgradeService({
+        agentName: creds.name,
+        packageName: AGENT_PACKAGE_NAME,
+        version: opts.to
+      });
+      console.log("[agent] Managed service updated successfully.");
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      console.error(`[agent] Failed to upgrade managed service: ${message}`);
+      process.exit(1);
     }
-    console.error(`[agent] Registration failed: ${errorMessage}`);
-    process.exit(1);
-  }
-  const result = await response.json();
-  saveCredentials({
-    serverUrl,
-    agentId: result.agentId,
-    agentSecret: result.agentSecret,
-    name: agentName
   });
-  console.log(`[agent] Registration successful!`);
-  console.log(`[agent] Agent ID: ${result.agentId}`);
-  console.log(`[agent] Credentials saved to ${credentialsPath()}`);
-  console.log(``);
-  console.log(`Next steps:`);
-  console.log(`  npx @webmux/agent start              # run once`);
-  console.log(`  npx @webmux/agent service install     # install as systemd service`);
-});
-program.command("start").description("Start the agent and connect to the server").action(() => {
-  const creds = loadCredentials();
-  if (!creds) {
-    console.error(
-      `[agent] No credentials found at ${credentialsPath()}. Run "npx @webmux/agent register" first.`
-    );
-    process.exit(1);
-  }
-  console.log(`[agent] Starting agent "${creds.name}"...`);
-  console.log(`[agent] Server: ${creds.serverUrl}`);
-  console.log(`[agent] Agent ID: ${creds.agentId}`);
-  const tmux = new TmuxClient({
-    socketName: "webmux",
-    workspaceRoot: os2.homedir()
-  });
-  const connection = new AgentConnection(
-    creds.serverUrl,
-    creds.agentId,
-    creds.agentSecret,
-    tmux
-  );
-  const shutdown = () => {
-    console.log("\n[agent] Shutting down...");
-    connection.stop();
-    process.exit(0);
-  };
-  process.on("SIGINT", shutdown);
-  process.on("SIGTERM", shutdown);
-  connection.start();
-});
-program.command("status").description("Show agent status and credentials info").action(() => {
-  const creds = loadCredentials();
-  if (!creds) {
-    console.log(`[agent] Not registered. No credentials found at ${credentialsPath()}.`);
-    process.exit(0);
-  }
-  console.log(`Agent Name:       ${creds.name}`);
-  console.log(`Server URL:       ${creds.serverUrl}`);
-  console.log(`Agent ID:         ${creds.agentId}`);
-  console.log(`Credentials File: ${credentialsPath()}`);
-  try {
-    const result = execSync2(`systemctl --user is-active ${SERVICE_NAME} 2>/dev/null`, { encoding: "utf-8" }).trim();
-    console.log(`Service:          ${result}`);
-  } catch {
-    console.log(`Service:          not installed`);
-  }
-});
-var service = program.command("service").description("Manage the systemd service");
-service.command("install").description("Install and start the agent as a systemd user service").action(() => {
-  const creds = loadCredentials();
-  if (!creds) {
-    console.error(`[agent] Not registered. Run "npx @webmux/agent register" first.`);
-    process.exit(1);
-  }
-  const npxPath = findBinary("npx");
-  if (!npxPath) {
-    console.error(`[agent] Cannot find npx. Make sure Node.js is installed.`);
-    process.exit(1);
-  }
-  const serviceDir = path2.join(os2.homedir(), ".config", "systemd", "user");
-  const servicePath = path2.join(serviceDir, `${SERVICE_NAME}.service`);
-  const npmPath = findBinary("npm") ?? "npm";
-  const unit = `[Unit]
-Description=Webmux Agent (${creds.name})
-After=network-online.target
-Wants=network-online.target
-
-[Service]
-Type=simple
-ExecStartPre=${npmPath} install -g @webmux/agent@latest
-ExecStart=${findBinary("webmux-agent") ?? `${npxPath} -y @webmux/agent`} start
-Restart=always
-RestartSec=10
-Environment=HOME=${os2.homedir()}
-Environment=PATH=${process.env.PATH}
-WorkingDirectory=${os2.homedir()}
-
-[Install]
-WantedBy=default.target
-`;
-  fs2.mkdirSync(serviceDir, { recursive: true });
-  fs2.writeFileSync(servicePath, unit);
-  console.log(`[agent] Service file created: ${servicePath}`);
-  try {
-    execSync2("systemctl --user daemon-reload", { stdio: "inherit" });
-    execSync2(`systemctl --user enable ${SERVICE_NAME}`, { stdio: "inherit" });
-    execSync2(`systemctl --user start ${SERVICE_NAME}`, { stdio: "inherit" });
-    execSync2(`loginctl enable-linger ${os2.userInfo().username}`, { stdio: "inherit" });
-    console.log(``);
-    console.log(`[agent] Service installed and started!`);
-    console.log(`[agent] It will auto-start on boot.`);
-    console.log(``);
-    console.log(`Useful commands:`);
-    console.log(`  systemctl --user status ${SERVICE_NAME}`);
-    console.log(`  journalctl --user -u ${SERVICE_NAME} -f`);
-    console.log(`  npx @webmux/agent service uninstall`);
-  } catch (err) {
-    const message = err instanceof Error ? err.message : String(err);
-    console.error(`[agent] Failed to enable service: ${message}`);
-    console.error(`[agent] Service file was written to ${servicePath}`);
-    console.error(`[agent] You can try manually: systemctl --user enable --now ${SERVICE_NAME}`);
-    process.exit(1);
-  }
-});
-service.command("uninstall").description("Stop and remove the systemd user service").action(() => {
-  const servicePath = path2.join(os2.homedir(), ".config", "systemd", "user", `${SERVICE_NAME}.service`);
-  try {
-    execSync2(`systemctl --user stop ${SERVICE_NAME} 2>/dev/null`, { stdio: "inherit" });
-    execSync2(`systemctl --user disable ${SERVICE_NAME} 2>/dev/null`, { stdio: "inherit" });
-  } catch {
-  }
-  if (fs2.existsSync(servicePath)) {
-    fs2.unlinkSync(servicePath);
-    console.log(`[agent] Service file removed: ${servicePath}`);
-  }
-  try {
-    execSync2("systemctl --user daemon-reload", { stdio: "inherit" });
-  } catch {
-  }
-  console.log(`[agent] Service uninstalled.`);
-});
-service.command("status").description("Show systemd service status").action(() => {
-  try {
-    execSync2(`systemctl --user status ${SERVICE_NAME}`, { stdio: "inherit" });
-  } catch {
-    console.log(`[agent] Service is not installed or not running.`);
-  }
-});
-service.command("upgrade").description("Upgrade agent to latest version and restart service").action(() => {
-  console.log("[agent] Upgrading @webmux/agent to latest...");
-  try {
-    execSync2("npm install -g @webmux/agent@latest", { stdio: "inherit" });
-  } catch {
-    console.error("[agent] Failed to upgrade. Try manually: npm install -g @webmux/agent@latest");
-    process.exit(1);
-  }
-  console.log("[agent] Restarting service...");
-  try {
-    execSync2(`systemctl --user restart ${SERVICE_NAME}`, { stdio: "inherit" });
-    console.log("[agent] Upgrade complete!");
-  } catch {
-    console.log("[agent] Package upgraded. Service not installed or restart failed.");
-    console.log("[agent] If running manually, restart with: npx @webmux/agent@latest start");
-  }
-});
-function findBinary(name) {
-  try {
-    return execSync2(`which ${name} 2>/dev/null`, { encoding: "utf-8" }).trim();
-  } catch {
-    return null;
+  return program;
+}
+async function run(argv = process.argv) {
+  await createProgram().parseAsync(argv);
+}
+function isDirectExecution() {
+  const entryPath = process.argv[1];
+  if (!entryPath) {
+    return false;
   }
+  return import.meta.url === pathToFileURL(entryPath).href;
 }
-program.parse();
+if (isDirectExecution()) {
+  void run();
+}
+export {
+  createProgram,
+  run
+};