amai 0.0.9 → 0.0.10

package/dist/cli.cjs

@@ -1254,30 +1254,19 @@ async function pollForTokens({
 async function login() {
   try {
     const device = await authorizeDevice();
-    console.log(pc5__default.default.bold("To sign in, follow these steps:"));
+    console.log("");
     if (device.verification_uri_complete) {
-      console.log(
-        `  1. Open this URL in your browser: ${pc5__default.default.cyan(
-          device.verification_uri_complete
-        )}`
-      );
+      console.log(pc5__default.default.cyan(`open: ${device.verification_uri_complete}`));
     } else {
-      console.log(
-        `  1. Open this URL in your browser: ${pc5__default.default.cyan(
-          device.verification_uri
-        )}`
-      );
-      console.log(
-        `  2. Enter this code when prompted: ${pc5__default.default.bold(device.user_code)}`
-      );
+      console.log(pc5__default.default.cyan(`open: ${device.verification_uri}`));
+      console.log(pc5__default.default.gray(`code: ${device.user_code}`));
     }
-    console.log("  3. Come back here; we will detect when you finish logging in.");
-    console.log();
+    console.log("");
     console.log(
       pc5__default.default.gray(
-        `Waiting for authorization (expires in ~${Math.round(
+        `waiting for authorization (~${Math.round(
           device.expires_in / 60
-        )} minutes)...`
+        )} min)...`
       )
     );
     const tokens = await pollForTokens({
@@ -1286,11 +1275,11 @@ async function login() {
       expiresIn: device.expires_in,
       interval: device.interval
     });
-    console.log(pc5__default.default.green("Successfully authenticated!"));
+    console.log(pc5__default.default.cyan("authenticated"));
     saveTokens(tokens);
     return tokens;
   } catch (error) {
-    console.error(pc5__default.default.red(error.message || "Login failed"));
+    console.error(pc5__default.default.red(error.message || "login failed"));
     throw error;
   }
 }
@@ -1984,7 +1973,19 @@ var rpcHandlers = {
     return { success: true, diff };
   }
 };
+var INITIAL_RECONNECT_DELAY = 1e3;
+var MAX_RECONNECT_DELAY = 6e4;
+var BACKOFF_MULTIPLIER = 2;
 var reconnectTimeout = null;
+var reconnectAttempts = 0;
+function getReconnectDelay() {
+  const delay = Math.min(
+    INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
+    MAX_RECONNECT_DELAY
+  );
+  const jitter = delay * 0.25 * (Math.random() * 2 - 1);
+  return Math.floor(delay + jitter);
+}
 var connectToUserStreams = async (serverUrl) => {
   const userId = getUserId();
   if (!userId?.userId) {
@@ -2005,7 +2006,8 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("open", () => {
-    console.log(pc5__default.default.green("CLI connected to user-streams"));
+    reconnectAttempts = 0;
+    console.log(pc5__default.default.cyan("connected to user streams"));
     if (reconnectTimeout) {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
@@ -2016,7 +2018,7 @@ var connectToUserStreams = async (serverUrl) => {
       const message = JSON.parse(event.toString());
       if (message._tag === "rpc_call") {
         const { requestId, method, input } = message;
-        console.log(pc5__default.default.gray(`RPC call: ${method}`));
+        console.log(pc5__default.default.gray(`> ${method}`));
         const handler = rpcHandlers[method];
         if (!handler) {
           ws.send(JSON.stringify({
@@ -2027,7 +2029,6 @@ var connectToUserStreams = async (serverUrl) => {
               message: `Unknown RPC method: ${method}`
             }
           }));
-          console.log(pc5__default.default.yellow(`Unknown RPC method: ${method}`));
           return;
         }
         try {
@@ -2037,7 +2038,6 @@ var connectToUserStreams = async (serverUrl) => {
             requestId,
             data: result
           }));
-          console.log(pc5__default.default.green(`RPC completed: ${method}`));
         } catch (error) {
           const rpcError = error._tag ? error : {
             _tag: "RpcError",
@@ -2048,7 +2048,7 @@ var connectToUserStreams = async (serverUrl) => {
             requestId,
             data: rpcError
           }));
-          console.log(pc5__default.default.red(`RPC failed: ${method} - ${rpcError.message}`));
+          console.log(pc5__default.default.red(`  ${method} failed`));
         }
         return;
       }
@@ -2067,25 +2067,38 @@ var connectToUserStreams = async (serverUrl) => {
         }
       }
     } catch (parseError) {
-      console.error(pc5__default.default.red(`Failed to parse message: ${parseError}`));
+      console.error(pc5__default.default.red(`parse error`));
     }
   });
   ws.on("close", (code, reason) => {
-    console.log(pc5__default.default.yellow(`CLI disconnected from user-streams (code: ${code})`));
-    console.log(pc5__default.default.gray("Reconnecting in 5 seconds..."));
+    const delay = getReconnectDelay();
+    reconnectAttempts++;
+    console.log(pc5__default.default.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
     reconnectTimeout = setTimeout(() => {
       connectToUserStreams(serverUrl).catch((err) => {
-        console.error(pc5__default.default.red(`Reconnection failed: ${err.message}`));
+        console.error(pc5__default.default.red(`reconnection failed`));
       });
-    }, 5e3);
+    }, delay);
   });
   ws.on("error", (error) => {
-    console.error(pc5__default.default.red(`User streams WebSocket error: ${error.message}`));
+    console.error(pc5__default.default.red(`stream error: ${error.message}`));
   });
   return ws;
 };
 
 // src/server.ts
+var INITIAL_RECONNECT_DELAY2 = 1e3;
+var MAX_RECONNECT_DELAY2 = 6e4;
+var BACKOFF_MULTIPLIER2 = 2;
+var reconnectAttempts2 = 0;
+function getReconnectDelay2() {
+  const delay = Math.min(
+    INITIAL_RECONNECT_DELAY2 * Math.pow(BACKOFF_MULTIPLIER2, reconnectAttempts2),
+    MAX_RECONNECT_DELAY2
+  );
+  const jitter = delay * 0.25 * (Math.random() * 2 - 1);
+  return Math.floor(delay + jitter);
+}
 var toolExecutors = {
   editFile: editFiles,
   deleteFile,
@@ -2108,12 +2121,13 @@ function connectToServer(serverUrl = DEFAULT_SERVER_URL) {
     }
   });
   ws.on("open", () => {
-    console.log(pc5__default.default.green("Connected to server agent streams"));
+    reconnectAttempts2 = 0;
+    console.log(pc5__default.default.cyan("connected to server"));
   });
   ws.on("message", async (data) => {
     const message = JSON.parse(data.toString());
     if (message.type === "tool_call") {
-      console.log(`tool call: ${message.tool}${message.projectCwd ? ` (project: ${message.projectCwd})` : ""}`);
+      console.log(pc5__default.default.gray(`> ${message.tool}`));
       try {
         const executor = toolExecutors[message.tool];
         if (!executor) {
@@ -2125,17 +2139,16 @@ function connectToServer(serverUrl = DEFAULT_SERVER_URL) {
           id: message.id,
           result
         }));
-        console.log(pc5__default.default.green(`tool call completed: ${message.tool}`));
       } catch (error) {
         ws.send(JSON.stringify({
           type: "tool_result",
           id: message.id,
           error: error.message
         }));
-        console.error(pc5__default.default.red(`tool call failed: ${message.tool} ${error.message}`));
+        console.error(pc5__default.default.red(`  ${message.tool} failed: ${error.message}`));
       }
     } else if (message.type === "rpc_call") {
-      console.log(`rpc call: ${message.method}`);
+      console.log(pc5__default.default.gray(`> rpc: ${message.method}`));
       try {
         const handler = rpcHandlers[message.method];
         if (!handler) {
@@ -2147,29 +2160,30 @@ function connectToServer(serverUrl = DEFAULT_SERVER_URL) {
           id: message.id,
           result
         }));
-        console.log(pc5__default.default.green(`rpc call completed: ${message.method}`));
       } catch (error) {
         ws.send(JSON.stringify({
           type: "tool_result",
           id: message.id,
           error: error.message
         }));
-        console.error(pc5__default.default.red(`rpc call failed: ${message.method} ${error.message}`));
+        console.error(pc5__default.default.red(`  rpc failed: ${message.method}`));
       }
     }
   });
   ws.on("close", () => {
-    console.log(pc5__default.default.red("disconnected from server. reconnecting in 5s..."));
-    setTimeout(() => connectToServer(serverUrl), 5e3);
+    const delay = getReconnectDelay2();
+    reconnectAttempts2++;
+    console.log(pc5__default.default.gray(`disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
+    setTimeout(() => connectToServer(serverUrl), delay);
   });
   ws.on("error", (error) => {
-    console.error(pc5__default.default.red(`web socket error: ${error.message}`));
+    console.error(pc5__default.default.red(`connection error: ${error.message}`));
   });
   return ws;
 }
 async function main() {
   const serverUrl = DEFAULT_SERVER_URL;
-  console.log(pc5__default.default.green("starting local amai..."));
+  console.log(pc5__default.default.gray("starting ama..."));
   connectToServer(serverUrl);
   await connectToUserStreams(serverUrl);
   startHttpServer();
@@ -2411,8 +2425,14 @@ function getDaemonPid() {
     return null;
   }
 }
-var VERSION = "0.0.9";
+var VERSION = "0.0.10";
 var PROJECT_DIR = process.cwd();
+var LOGO = `
+   __ _ _ __ ___   __ _ 
+  / _\` | '_ \` _ \\ / _\` |
+ | (_| | | | | | | (_| |
+  \\__,_|_| |_| |_|\\__,_|
+`;
 function promptUser(question) {
   const rl = readline__default.default.createInterface({
     input: process.stdin,
@@ -2427,168 +2447,249 @@ function promptUser(question) {
 }
 async function startWithCodeServer() {
   if (!isCodeServerInstalled()) {
-    console.log(pc5__default.default.cyan("First run detected. Setting up code-server..."));
+    console.log(pc5__default.default.gray("setting up code-server..."));
     try {
       await installCodeServer();
     } catch (error) {
-      console.error(pc5__default.default.red(`Failed to install code-server: ${error.message}`));
-      console.log(pc5__default.default.yellow("Continuing without code-server..."));
+      console.error(pc5__default.default.red(`failed to install code-server: ${error.message}`));
+      console.log(pc5__default.default.gray("continuing without code-server..."));
     }
   }
   if (isCodeServerInstalled()) {
     try {
       await startCodeServer(PROJECT_DIR);
     } catch (error) {
-      console.error(pc5__default.default.red(`Failed to start code-server: ${error.message}`));
+      console.error(pc5__default.default.red(`failed to start code-server: ${error.message}`));
     }
   }
   main();
 }
+async function checkForUpdates() {
+  try {
+    const response = await fetch("https://registry.npmjs.org/amai/latest");
+    const data = await response.json();
+    const latestVersion = data.version;
+    return {
+      current: VERSION,
+      latest: latestVersion,
+      hasUpdate: latestVersion !== VERSION
+    };
+  } catch {
+    return { current: VERSION, latest: VERSION, hasUpdate: false };
+  }
+}
+function runNpmInstall() {
+  return new Promise((resolve, reject) => {
+    const child = child_process.spawn("npm", ["install", "-g", "amai@latest"], {
+      stdio: "inherit",
+      shell: true
+    });
+    child.on("close", (code) => {
+      if (code === 0) resolve();
+      else reject(new Error(`npm install exited with code ${code}`));
+    });
+    child.on("error", reject);
+  });
+}
 var args = process.argv.slice(2);
+if (args[0] === "--version" || args[0] === "-v") {
+  console.log(pc5__default.default.gray(`amai ${VERSION}`));
+  process.exit(0);
+}
 if (args[0] === "--help" || args[0] === "-h") {
-  console.log(`
-${pc5__default.default.bold("amai cli")} ${pc5__default.default.gray(VERSION)}
-
-Usage: amai [command] [options]
-
-Commands:
-  login                 Authorize device
-  logout                Log out and remove credentials
-  start                 Start background daemon (background mode is highly recommended for better performance and stability)
-  stop                  Stop background daemon
-  status                Check daemon status
-  project add <path>    Register a project directory
-  project list          List registered projects
-Options:
-  --help, -h            Show this help message
-  --logout, -l          Log out and remove credentials
-Environment Variables:
-  SERVER_URL            Server URL to connect to
-
-Example:
-  amai login
-  amai start
-  amai project add /path/to/project
-  amai                  Start the agent (will prompt for background mode)
-    `);
+  console.log(pc5__default.default.cyan(LOGO));
+  console.log(pc5__default.default.gray(`  v${VERSION}`));
+  console.log("");
+  console.log(pc5__default.default.cyan("  usage"));
+  console.log(pc5__default.default.gray("    amai [command]"));
+  console.log("");
+  console.log(pc5__default.default.cyan("  commands"));
+  console.log(pc5__default.default.gray("    login           authenticate with amai"));
+  console.log(pc5__default.default.gray("    logout          remove credentials"));
+  console.log(pc5__default.default.gray("    start           start background daemon"));
+  console.log(pc5__default.default.gray("    stop            stop background daemon"));
+  console.log(pc5__default.default.gray("    status          check daemon status"));
+  console.log(pc5__default.default.gray("    update          update to latest version"));
+  console.log(pc5__default.default.gray("    project add     register a project"));
+  console.log(pc5__default.default.gray("    project list    list projects"));
+  console.log("");
+  console.log(pc5__default.default.cyan("  options"));
+  console.log(pc5__default.default.gray("    -h, --help      show help"));
+  console.log(pc5__default.default.gray("    -v, --version   show version"));
+  console.log("");
   process.exit(0);
 }
-if (args[0] === "start") {
-  if (isDaemonRunning()) {
-    console.log(pc5__default.default.yellow("ama is already running"));
+if (args[0] === "update") {
+  (async () => {
+    console.log(pc5__default.default.gray("checking for updates..."));
+    const { current, latest, hasUpdate } = await checkForUpdates();
+    if (!hasUpdate) {
+      console.log(pc5__default.default.cyan(`already on latest version (${current})`));
+      process.exit(0);
+    }
+    console.log(pc5__default.default.cyan(`update available: ${current} -> ${latest}`));
+    const answer = await promptUser(pc5__default.default.gray("install update? (Y/n): "));
+    if (answer === "" || answer.toLowerCase() === "y" || answer.toLowerCase() === "yes") {
+      console.log(pc5__default.default.gray("updating..."));
+      try {
+        await runNpmInstall();
+        console.log(pc5__default.default.cyan(`updated to ${latest}`));
+      } catch (error) {
+        console.error(pc5__default.default.red(`update failed: ${error.message}`));
+        process.exit(1);
+      }
+    }
     process.exit(0);
-  }
-  if (!isAuthenticated()) {
-    console.log(pc5__default.default.yellow("Not authenticated. Please log in first."));
-    login().then(() => {
-      console.log(pc5__default.default.green("starting ama in background mode..."));
-      startDaemon();
-      console.log(pc5__default.default.green("ama started in background mode successfully"));
+  })();
+} else if (args[0] === "start") {
+  (async () => {
+    if (isDaemonRunning()) {
+      console.log(pc5__default.default.gray("amai is already running"));
       process.exit(0);
-    }).catch(() => {
-      console.error(pc5__default.default.red("Login failed. Cannot start ama in background mode."));
-      process.exit(1);
-    });
-  } else {
+    }
+    if (!isAuthenticated()) {
+      console.log(pc5__default.default.gray("not authenticated"));
+      try {
+        await login();
+      } catch {
+        console.error(pc5__default.default.red("login failed"));
+        process.exit(1);
+      }
+    }
     startDaemon();
-    console.log(pc5__default.default.green(pc5__default.default.bold("amai started in background mode")));
-    console.log(pc5__default.default.gray(`Tip: You can check status any time with ${pc5__default.default.bold("amai status")}`));
+    console.log(pc5__default.default.cyan("amai started"));
+    console.log(pc5__default.default.gray(`check status: amai status`));
     process.exit(0);
-  }
-}
-if (args[0] === "stop") {
+  })();
+} else if (args[0] === "stop") {
   if (stopDaemon()) {
-    console.log(pc5__default.default.green("Daemon stopped successfully"));
+    console.log(pc5__default.default.cyan("daemon stopped"));
   } else {
-    console.log(pc5__default.default.yellow("Daemon was not running"));
+    console.log(pc5__default.default.gray("daemon was not running"));
   }
   process.exit(0);
-}
-if (args[0] === "status") {
+} else if (args[0] === "status") {
   const running = isDaemonRunning();
   const pid = getDaemonPid();
+  console.log("");
+  console.log(pc5__default.default.cyan("  amai status"));
+  console.log("");
   if (running && pid) {
-    console.log(pc5__default.default.green(`Daemon is running (PID: ${pid})`));
+    console.log(pc5__default.default.gray(`  status   running`));
+    console.log(pc5__default.default.gray(`  pid      ${pid}`));
   } else {
-    console.log(pc5__default.default.yellow("Daemon is not running"));
+    console.log(pc5__default.default.gray(`  status   stopped`));
   }
+  console.log(pc5__default.default.gray(`  version  ${VERSION}`));
+  console.log("");
   process.exit(0);
-}
-if (args[0] === "project") {
+} else if (args[0] === "project") {
   if (args[1] === "add") {
     const projectPath = args[2];
     if (!projectPath) {
-      console.error(pc5__default.default.red("Please provide a project path"));
-      console.log("Usage: amai project add <path>");
+      console.error(pc5__default.default.red("please provide a project path"));
+      console.log(pc5__default.default.gray("usage: amai project add <path>"));
       process.exit(1);
     }
     const resolvedPath = path10__default.default.resolve(projectPath);
     if (!fs6__default.default.existsSync(resolvedPath)) {
-      console.error(pc5__default.default.red(`Path does not exist: ${resolvedPath}`));
+      console.error(pc5__default.default.red(`path does not exist: ${resolvedPath}`));
       process.exit(1);
     }
     if (!fs6__default.default.statSync(resolvedPath).isDirectory()) {
-      console.error(pc5__default.default.red(`Path is not a directory: ${resolvedPath}`));
+      console.error(pc5__default.default.red(`path is not a directory: ${resolvedPath}`));
       process.exit(1);
     }
     const projectId = path10__default.default.basename(resolvedPath);
     projectRegistry.register(projectId, resolvedPath);
-    console.log(pc5__default.default.green(`Project registered: ${projectId} -> ${resolvedPath}`));
+    console.log(pc5__default.default.cyan(`project registered: ${projectId}`));
+    console.log(pc5__default.default.gray(`  ${resolvedPath}`));
     process.exit(0);
   } else if (args[1] === "list") {
     const projects = projectRegistry.list();
+    console.log("");
+    console.log(pc5__default.default.cyan("  projects"));
+    console.log("");
     if (projects.length === 0) {
-      console.log(pc5__default.default.yellow("No projects registered"));
+      console.log(pc5__default.default.gray("  no projects registered"));
     } else {
-      console.log(pc5__default.default.bold("Registered projects:"));
       projects.forEach((project) => {
-        console.log(`  ${pc5__default.default.cyan(project.id)}: ${project.cwd} ${project.active ? pc5__default.default.green("(active)") : ""}`);
+        const status = project.active ? pc5__default.default.cyan("active") : pc5__default.default.gray("inactive");
+        console.log(pc5__default.default.gray(`  ${project.id} [${status}]`));
+        console.log(pc5__default.default.gray(`    ${project.cwd}`));
       });
     }
+    console.log("");
     process.exit(0);
   } else {
-    console.error(pc5__default.default.red(`Unknown project command: ${args[1]}`));
-    console.log('Use "amai project add <path>" or "amai project list"');
+    console.error(pc5__default.default.red(`unknown project command: ${args[1]}`));
+    console.log(pc5__default.default.gray("usage: amai project add <path> | amai project list"));
     process.exit(1);
   }
-}
-if (args[0] === "login" || args[0] === "--login") {
-  login().then(() => process.exit(0)).catch(() => process.exit(1));
+} else if (args[0] === "login" || args[0] === "--login") {
+  (async () => {
+    try {
+      await login();
+      console.log("");
+      if (isDaemonRunning()) {
+        console.log(pc5__default.default.gray("amai is already running"));
+        process.exit(0);
+      }
+      const answer = await promptUser(pc5__default.default.gray("start amai now? (Y/n): "));
+      const shouldStart = answer === "" || answer.toLowerCase() === "y" || answer.toLowerCase() === "yes";
+      if (shouldStart) {
+        const bgAnswer = await promptUser(pc5__default.default.gray("run in background? (Y/n): "));
+        const runInBackground = bgAnswer === "" || bgAnswer.toLowerCase() === "y" || bgAnswer.toLowerCase() === "yes";
+        if (runInBackground) {
+          console.log(pc5__default.default.gray("starting..."));
+          startDaemon();
+          console.log(pc5__default.default.cyan("amai started"));
+          console.log(pc5__default.default.gray('use "amai status" to check status'));
+        } else {
+          console.log(pc5__default.default.gray("starting in foreground..."));
+          startWithCodeServer();
+          return;
+        }
+      }
+      process.exit(0);
+    } catch {
+      console.error(pc5__default.default.red("login failed"));
+      process.exit(1);
+    }
+  })();
 } else if (args[0] === "logout" || args[0] === "--logout") {
   logout();
-  console.log(pc5__default.default.green("Logged out successfully"));
+  console.log(pc5__default.default.cyan("logged out"));
   process.exit(0);
 } else {
   (async () => {
+    console.log(pc5__default.default.cyan(LOGO));
     if (!isAuthenticated()) {
-      console.log(pc5__default.default.yellow("Not authenticated. Please log in first."));
+      console.log(pc5__default.default.gray("not authenticated"));
       try {
         await login();
+        console.log("");
       } catch {
-        console.error(pc5__default.default.red("Login failed. Cannot start server."));
+        console.error(pc5__default.default.red("login failed"));
         process.exit(1);
       }
     }
     if (isDaemonRunning()) {
-      console.log(pc5__default.default.yellow('Daemon is already running. Use "amai status" to check its status.'));
+      console.log(pc5__default.default.gray("amai is already running"));
+      console.log(pc5__default.default.gray('use "amai status" to check status'));
       process.exit(0);
     }
-    console.log("");
-    console.log(pc5__default.default.bold("How would you like to run amai?"));
-    console.log(pc5__default.default.gray("Background mode is highly recommended for better performance and stability."));
-    const answer = await promptUser(
-      pc5__default.default.cyan("Run in background? (Y/n): ")
-    );
+    const answer = await promptUser(pc5__default.default.gray("run in background? (Y/n): "));
     const runInBackground = answer === "" || answer.toLowerCase() === "y" || answer.toLowerCase() === "yes";
     if (runInBackground) {
-      console.log(pc5__default.default.green("Starting daemon in background..."));
+      console.log(pc5__default.default.gray("starting..."));
       startDaemon();
-      console.log(pc5__default.default.green("Daemon started successfully!"));
-      console.log(pc5__default.default.gray('Use "amai status" to check daemon status.'));
-      console.log(pc5__default.default.gray('Use "amai stop" to stop the daemon.'));
+      console.log(pc5__default.default.cyan("amai started"));
+      console.log(pc5__default.default.gray('use "amai status" to check status'));
+      console.log(pc5__default.default.gray('use "amai stop" to stop'));
       process.exit(0);
     } else {
-      console.log(pc5__default.default.yellow("Starting in foreground mode..."));
+      console.log(pc5__default.default.gray("starting in foreground..."));
       startWithCodeServer();
     }
   })();