@arbidocs/cli 0.3.3 → 0.3.5

package/dist/{index.cjs → index.js}

@@ -2,11 +2,12 @@
 'use strict';
 
 var commander = require('commander');
-var core = require('@arbidocs/core');
-var prompts = require('@inquirer/prompts');
 var fs = require('fs');
-var path = require('path');
 var os = require('os');
+var path = require('path');
+var core = require('@arbidocs/core');
+var chalk = require('chalk');
+var prompts = require('@inquirer/prompts');
 var child_process = require('child_process');
 var sdk = require('@arbidocs/sdk');
 var module$1 = require('module');
@@ -15,8 +16,9 @@ var _documentCurrentScript = typeof document !== 'undefined' ? document.currentS
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
 var fs__default = /*#__PURE__*/_interopDefault(fs);
-var path__default = /*#__PURE__*/_interopDefault(path);
 var os__default = /*#__PURE__*/_interopDefault(os);
+var path__default = /*#__PURE__*/_interopDefault(path);
+var chalk__default = /*#__PURE__*/_interopDefault(chalk);
 
 var store = new core.FileConfigStore();
 function getConfig() {
@@ -51,8 +53,48 @@ function updateChatSession(updates) {
 function clearChatSession() {
   store.clearChatSession();
 }
+function success(msg) {
+  console.log(chalk__default.default.green(msg));
+}
+function error(msg) {
+  console.error(chalk__default.default.red(msg));
+}
+function warn(msg) {
+  console.error(chalk__default.default.yellow(msg));
+}
+function label(key, value) {
+  console.log(`${chalk__default.default.bold(key)} ${value}`);
+}
+function dim(msg) {
+  console.log(chalk__default.default.dim(msg));
+}
+function bold(msg) {
+  console.log(chalk__default.default.bold(msg));
+}
+function status(s) {
+  const lower = s.toLowerCase();
+  if (["healthy", "completed", "available", "online", "on"].includes(lower)) return chalk__default.default.green(s);
+  if (["failed", "error", "unavailable", "offline"].includes(lower)) return chalk__default.default.red(s);
+  if (["processing", "pending", "degraded", "warning"].includes(lower)) return chalk__default.default.yellow(s);
+  return s;
+}
+function ref(s) {
+  return chalk__default.default.cyan(s);
+}
 
 // src/commands/config-cmd.ts
+var ALIAS_LINE = 'alias A="arbi ask"';
+var ALIAS_MARKER = "# arbi-cli alias";
+function getShellRcPath() {
+  const shell = process.env.SHELL || "";
+  if (shell.includes("zsh")) return path.join(os.homedir(), ".zshrc");
+  return path.join(os.homedir(), ".bashrc");
+}
+function isAliasInstalled(rcPath) {
+  if (!fs.existsSync(rcPath)) return false;
+  const content = fs.readFileSync(rcPath, "utf-8");
+  return content.includes(ALIAS_LINE) || content.includes(ALIAS_MARKER);
+}
 function registerConfigCommand(program2) {
   const config = program2.command("config").description("Manage CLI configuration");
   config.command("set-url <url>").description("Set the ARBI server URL").action((url) => {
@@ -60,25 +102,55 @@ function registerConfigCommand(program2) {
       const parsed = new URL(url);
       const deploymentDomain = parsed.hostname;
       updateConfig({ baseUrl: url.replace(/\/+$/, ""), deploymentDomain });
-      console.log(`Server URL: ${url}`);
-      console.log(`Domain:     ${deploymentDomain}`);
+      label("Server URL:", url);
+      label("Domain:", deploymentDomain);
     } catch {
-      console.error(`Invalid URL: ${url}`);
+      error(`Invalid URL: ${url}`);
       process.exit(1);
     }
   });
   config.command("show").description("Show current configuration").action(() => {
     const cfg = getConfig();
     if (!cfg) {
-      console.log("Not configured. Run: arbi config set-url <url>");
+      dim("Not configured. Run: arbi config set-url <url>");
       return;
     }
-    console.log(`Server URL:  ${cfg.baseUrl}`);
-    console.log(`Domain:      ${cfg.deploymentDomain}`);
-    console.log(`Auto-update: ${cfg.autoUpdate ? "on" : "off"}`);
+    label("Server URL:", cfg.baseUrl);
+    label("Domain:", cfg.deploymentDomain);
+    label("Auto-update:", cfg.autoUpdate ? "on" : "off");
+    label("Notifications:", cfg.notifications ? "on" : "off");
     if (cfg.selectedWorkspaceId) {
-      console.log(`Workspace:   ${cfg.selectedWorkspaceId}`);
+      label("Workspace:", cfg.selectedWorkspaceId);
+    }
+  });
+  config.command("notifications [on|off]").description("Toggle background WebSocket notifications").action((toggle) => {
+    const cfg = getConfig();
+    if (!toggle) {
+      label("Notifications:", cfg?.notifications ? "on" : "off");
+      return;
+    }
+    if (toggle !== "on" && toggle !== "off") {
+      error("Usage: arbi config notifications [on|off]");
+      process.exit(1);
+    }
+    updateConfig({ notifications: toggle === "on" });
+    success(`Notifications: ${toggle}`);
+  });
+  config.command("alias").description('Set up shell alias A for "arbi ask"').action(() => {
+    const rcPath = getShellRcPath();
+    if (isAliasInstalled(rcPath)) {
+      success(`Alias already set up in ${rcPath}`);
+      dim("Usage: A what is the meaning of life");
+      return;
     }
+    fs.appendFileSync(rcPath, `
+${ALIAS_MARKER}
+${ALIAS_LINE}
+`);
+    success(`Added alias A="arbi ask" to ${rcPath}`);
+    dim("");
+    dim(`Run: source ${rcPath}`);
+    dim("Then: A what is the meaning of life");
   });
 }
 
@@ -94,8 +166,8 @@ var messages = {
   TransactionInactiveError: "A request was placed against a transaction which is currently not active, or which is finished.",
   VersionError: "An attempt was made to open a database using a lower version than the existing version."
 };
-var setErrorCode = (error, value) => {
-  Object.defineProperty(error, "code", {
+var setErrorCode = (error2, value) => {
+  Object.defineProperty(error2, "code", {
     value,
     writable: false,
     enumerable: true,
@@ -3429,7 +3501,7 @@ function getLatestVersion() {
   }
 }
 function getCurrentVersion() {
-  return "0.3.3";
+  return "0.3.5";
 }
 async function fetchChangelog(fromVersion, toVersion) {
   try {
@@ -3473,19 +3545,17 @@ async function showChangelog(fromVersion, toVersion) {
 async function checkForUpdates(autoUpdate) {
   try {
     const latest = getLatestVersion();
-    if (!latest || latest === "0.3.3") return;
+    if (!latest || latest === "0.3.5") return;
     if (autoUpdate) {
-      console.log(
-        `
-Your arbi version is out of date (${"0.3.3"} \u2192 ${latest}). Updating...`
-      );
-      await showChangelog("0.3.3", latest);
+      warn(`
+Your arbi version is out of date (${"0.3.5"} \u2192 ${latest}). Updating...`);
+      await showChangelog("0.3.5", latest);
       child_process.execSync("npm install -g @arbidocs/cli@latest", { stdio: "inherit" });
       console.log(`Updated to ${latest}.`);
     } else {
-      console.error(
+      warn(
         `
-Your arbi version is out of date (${"0.3.3"} \u2192 ${latest}).
+Your arbi version is out of date (${"0.3.5"} \u2192 ${latest}).
 Run "arbi update" to upgrade, or "arbi update auto" to always stay up to date.`
       );
     }
@@ -3495,9 +3565,9 @@ Run "arbi update" to upgrade, or "arbi update auto" to always stay up to date.`
 function hintUpdateOnError() {
   try {
     const cached = readCache();
-    if (cached && cached.latest !== "0.3.3") {
-      console.error(
-        `Your arbi version is out of date (${"0.3.3"} \u2192 ${cached.latest}). Run "arbi update".`
+    if (cached && cached.latest !== "0.3.5") {
+      warn(
+        `Your arbi version is out of date (${"0.3.5"} \u2192 ${cached.latest}). Run "arbi update".`
       );
     }
   } catch {
@@ -3514,7 +3584,7 @@ function registerLoginCommand(program2) {
       const { arbi } = await core.performPasswordLogin(config, email, pw, store);
       const { data: workspaces2 } = await arbi.fetch.GET("/api/user/workspaces");
       const wsList = workspaces2 || [];
-      console.log(`Logged in as ${email}`);
+      success(`Logged in as ${email}`);
       if (wsList.length === 0) {
         console.log("No workspaces found.");
         return;
@@ -3522,25 +3592,26 @@ function registerLoginCommand(program2) {
       if (opts.workspace) {
         const ws2 = wsList.find((w) => w.external_id === opts.workspace);
         if (!ws2) {
-          console.error(`Workspace ${opts.workspace} not found.`);
+          error(`Workspace ${opts.workspace} not found.`);
           process.exit(1);
         }
         updateConfig({ selectedWorkspaceId: ws2.external_id });
-        console.log(`Workspace: ${ws2.name} (${ws2.external_id})`);
+        success(`Workspace: ${ws2.name} (${ref(ws2.external_id)})`);
         return;
       }
       if (wsList.length === 1) {
         updateConfig({ selectedWorkspaceId: wsList[0].external_id });
-        console.log(`Workspace: ${wsList[0].name} (${wsList[0].external_id})`);
+        success(`Workspace: ${wsList[0].name} (${ref(wsList[0].external_id)})`);
         return;
       }
       const choices = core.formatWorkspaceChoices(wsList);
       const selected = await promptSelect("Select workspace", choices);
       updateConfig({ selectedWorkspaceId: selected });
       const ws = wsList.find((w) => w.external_id === selected);
-      console.log(`Workspace: ${ws.name} (${selected})`);
+      success(`Workspace: ${ws.name} (${ref(selected)})`);
+      dim('\nTip: Run "arbi config alias" to use A as a shortcut for "arbi ask"');
     } catch (err) {
-      console.error(`Login failed: ${core.getErrorMessage(err)}`);
+      error(`Login failed: ${core.getErrorMessage(err)}`);
       process.exit(1);
     } finally {
       await checkForUpdates(getConfig()?.autoUpdate);
@@ -3564,7 +3635,7 @@ async function getVerificationCode(email, apiKey) {
   return Array.isArray(words) ? words.join(" ") : String(words);
 }
 function registerRegisterCommand(program2) {
-  program2.command("register").description("Register a new ARBI account").option("--non-interactive", "CI/automation mode (requires SUPPORT_API_KEY env var)").option("-e, --email <email>", "Email address (or ARBI_EMAIL env var)").option("-p, --password <password>", "Password (or ARBI_PASSWORD env var)").option("-c, --verification-code <code>", "Verification code (skip prompt)").option("--first-name <name>", "First name").option("--last-name <name>", "Last name").option("-w, --workspace <id>", "Workspace ID to select after login").action(async (opts) => {
+  program2.command("register").description("Register a new ARBI account").option("--non-interactive", "CI/automation mode (requires SUPPORT_API_KEY env var)").option("-e, --email <email>", "Email address (or ARBI_EMAIL env var)").option("-p, --password <password>", "Password (or ARBI_PASSWORD env var)").option("-c, --verification-code <code>", "Verification code (skip prompt)").option("--first-name <name>", "First name").option("--last-name <name>", "Last name").action(async (opts) => {
     const config = requireConfig();
     if (opts.nonInteractive) {
       await nonInteractiveRegister(config, opts);
@@ -3597,10 +3668,10 @@ async function smartRegister(config, opts) {
         body: { email }
       });
       if (verifyResponse.error) {
-        console.error(`Failed to send verification email: ${JSON.stringify(verifyResponse.error)}`);
+        error(`Failed to send verification email: ${JSON.stringify(verifyResponse.error)}`);
         process.exit(1);
       }
-      console.log("Verification email sent. Check your inbox.");
+      success("Verification email sent. Check your inbox.");
       verificationCode = await promptInput("Verification code");
     }
   }
@@ -3612,7 +3683,7 @@ async function smartRegister(config, opts) {
     pw = await promptPassword("Password");
     const confirmPw = await promptPassword("Confirm password");
     if (pw !== confirmPw) {
-      console.error("Passwords do not match.");
+      error("Passwords do not match.");
       process.exit(1);
     }
   }
@@ -3627,16 +3698,16 @@ async function smartRegister(config, opts) {
       firstName,
       lastName
     });
-    console.log(`
+    success(`
 Registered successfully as ${email}`);
   } catch (err) {
-    console.error(`Registration failed: ${core.getErrorMessage(err)}`);
+    error(`Registration failed: ${core.getErrorMessage(err)}`);
     process.exit(1);
   }
   const allFlagsProvided = !!(opts.email || process.env.ARBI_EMAIL) && !!(opts.password || process.env.ARBI_PASSWORD) && !!opts.verificationCode;
   const doLogin = allFlagsProvided || await promptConfirm("Log in now?");
   if (doLogin) {
-    await loginAfterRegister(config, email, pw, opts.workspace);
+    await loginAfterRegister(config, email, pw);
   }
 }
 async function nonInteractiveRegister(config, opts) {
@@ -3644,11 +3715,11 @@ async function nonInteractiveRegister(config, opts) {
   const password2 = opts.password || process.env.ARBI_PASSWORD;
   const supportApiKey = process.env.SUPPORT_API_KEY;
   if (!email) {
-    console.error("Email required. Use --email <email> or set ARBI_EMAIL");
+    error("Email required. Use --email <email> or set ARBI_EMAIL");
     process.exit(1);
   }
   if (!password2) {
-    console.error("Password required. Use --password <password> or set ARBI_PASSWORD");
+    error("Password required. Use --password <password> or set ARBI_PASSWORD");
     process.exit(1);
   }
   const arbi = sdk.createArbiClient({
@@ -3662,7 +3733,7 @@ async function nonInteractiveRegister(config, opts) {
     verificationCode = opts.verificationCode;
   } else {
     if (!supportApiKey) {
-      console.error(
+      error(
         "Verification code required. Use --verification-code <code> or set SUPPORT_API_KEY for CI mode"
       );
       process.exit(1);
@@ -3671,7 +3742,7 @@ async function nonInteractiveRegister(config, opts) {
       body: { email }
     });
     if (verifyResponse.error) {
-      console.error(`verify-email failed: ${JSON.stringify(verifyResponse.error)}`);
+      error(`verify-email failed: ${JSON.stringify(verifyResponse.error)}`);
       process.exit(1);
     }
     verificationCode = await getVerificationCode(email, supportApiKey);
@@ -3684,42 +3755,38 @@ async function nonInteractiveRegister(config, opts) {
       firstName: opts.firstName ?? "Test",
       lastName: opts.lastName ?? "User"
     });
-    console.log(`Registered: ${email}`);
+    success(`Registered: ${email}`);
   } catch (err) {
-    console.error(`Registration failed: ${core.getErrorMessage(err)}`);
+    error(`Registration failed: ${core.getErrorMessage(err)}`);
     process.exit(1);
   }
 }
-async function loginAfterRegister(config, email, password2, workspaceId) {
+async function loginAfterRegister(config, email, password2) {
   try {
     const { arbi } = await core.performPasswordLogin(config, email, password2, store);
+    success(`Logged in as ${email}`);
     const { data: workspaces2 } = await arbi.fetch.GET("/api/user/workspaces");
     const wsList = workspaces2 || [];
-    console.log(`Logged in as ${email}`);
-    if (wsList.length === 0) return;
-    if (workspaceId) {
-      const ws = wsList.find((w) => w.external_id === workspaceId);
-      if (ws) {
-        updateConfig({ selectedWorkspaceId: ws.external_id });
-        console.log(`Workspace: ${ws.name} (${ws.external_id})`);
-      } else {
-        console.error(`Workspace ${workspaceId} not found.`);
-      }
+    if (wsList.length === 0) {
+      console.log("Creating your first workspace...");
+      const ws2 = await core.workspaces.createWorkspace(arbi, "My First Workspace");
+      updateConfig({ selectedWorkspaceId: ws2.external_id });
+      success(`Workspace: ${ws2.name} (${ref(ws2.external_id)})`);
       return;
     }
     if (wsList.length === 1) {
       updateConfig({ selectedWorkspaceId: wsList[0].external_id });
-      console.log(`Workspace: ${wsList[0].name} (${wsList[0].external_id})`);
-    } else if (wsList.length > 1) {
-      const choices = core.formatWorkspaceChoices(wsList);
-      const selected = await promptSelect("Select workspace", choices);
-      updateConfig({ selectedWorkspaceId: selected });
-      const ws = wsList.find((w) => w.external_id === selected);
-      console.log(`Workspace: ${ws.name} (${selected})`);
+      success(`Workspace: ${wsList[0].name} (${ref(wsList[0].external_id)})`);
+      return;
     }
+    const choices = core.formatWorkspaceChoices(wsList);
+    const selected = await promptSelect("Select workspace", choices);
+    updateConfig({ selectedWorkspaceId: selected });
+    const ws = wsList.find((w) => w.external_id === selected);
+    success(`Workspace: ${ws.name} (${ref(selected)})`);
   } catch (err) {
-    console.error(`Login failed: ${core.getErrorMessage(err)}`);
-    console.error("You can log in later with: arbi login");
+    error(`Login failed: ${core.getErrorMessage(err)}`);
+    error("You can log in later with: arbi login");
   }
 }
 
@@ -3728,7 +3795,7 @@ function registerLogoutCommand(program2) {
   program2.command("logout").description("Log out of ARBI").action(() => {
     deleteCredentials();
     updateConfig({ selectedWorkspaceId: void 0 });
-    console.log("Logged out.");
+    success("Logged out.");
   });
 }
 
@@ -3738,28 +3805,85 @@ function registerStatusCommand(program2) {
     const config = getConfig();
     const creds = getCredentials();
     if (!config?.baseUrl) {
-      console.log("Not configured. Run: arbi config set-url <url>");
+      dim("Not configured. Run: arbi config set-url <url>");
       return;
     }
-    console.log(`Server:    ${config.baseUrl}`);
+    label("Server:", config.baseUrl);
     if (creds) {
-      console.log(`User:      ${creds.email}`);
+      label("User:", creds.email);
     } else {
-      console.log("User:      (not logged in)");
+      label("User:", "(not logged in)");
     }
     if (config.selectedWorkspaceId) {
-      console.log(`Workspace: ${config.selectedWorkspaceId}`);
+      label("Workspace:", config.selectedWorkspaceId);
     } else {
-      console.log("Workspace: (none selected)");
+      label("Workspace:", "(none selected)");
     }
   });
 }
+var activeConnection = null;
+function timestamp() {
+  return (/* @__PURE__ */ new Date()).toLocaleTimeString("en-GB", { hour12: false });
+}
+function colorize(level, text) {
+  if (level === "success") return chalk__default.default.green(text);
+  if (level === "error") return chalk__default.default.red(text);
+  if (level === "warning") return chalk__default.default.yellow(text);
+  return text;
+}
+async function startBackgroundNotifications(baseUrl, accessToken) {
+  if (activeConnection) return;
+  try {
+    activeConnection = await core.connectWithReconnect({
+      baseUrl,
+      accessToken,
+      onMessage: (msg) => {
+        const { text, level } = core.formatWsMessage(msg);
+        process.stderr.write(`
+${colorize(level, `[${timestamp()}] ${text}`)}
+`);
+      },
+      onClose: () => {
+        activeConnection = null;
+      },
+      onReconnecting: (attempt, maxRetries) => {
+        process.stderr.write(
+          `
+${chalk__default.default.yellow(`[${timestamp()}] Reconnecting... (${attempt}/${maxRetries})`)}
+`
+        );
+      },
+      onReconnected: () => {
+        process.stderr.write(`
+${chalk__default.default.green(`[${timestamp()}] Reconnected`)}
+`);
+      },
+      onReconnectFailed: () => {
+        process.stderr.write(`
+${chalk__default.default.red(`[${timestamp()}] Reconnection failed`)}
+`);
+      }
+    });
+  } catch {
+  }
+}
+function stopBackgroundNotifications() {
+  activeConnection?.close();
+  activeConnection = null;
+}
+process.on("exit", stopBackgroundNotifications);
+process.on("SIGINT", () => {
+  stopBackgroundNotifications();
+  process.exit(0);
+});
+
+// src/helpers.ts
 function runAction(fn) {
   return async () => {
     try {
       await fn();
     } catch (err) {
-      console.error(`Error: ${core.getErrorMessage(err)}`);
+      error(`Error: ${core.getErrorMessage(err)}`);
       hintUpdateOnError();
       process.exit(1);
     }
@@ -3770,7 +3894,7 @@ async function resolveAuth() {
     return await core.resolveAuth(store);
   } catch (err) {
     if (err instanceof core.ArbiError) {
-      console.error(err.message);
+      error(err.message);
       process.exit(1);
     }
     throw err;
@@ -3778,17 +3902,22 @@ async function resolveAuth() {
 }
 async function resolveWorkspace(workspaceOpt) {
   try {
-    return await core.resolveWorkspace(store, workspaceOpt);
+    const ctx = await core.resolveWorkspace(store, workspaceOpt);
+    if (getConfig()?.notifications) {
+      startBackgroundNotifications(ctx.config.baseUrl, ctx.accessToken).catch(() => {
+      });
+    }
+    return ctx;
   } catch (err) {
     if (err instanceof core.ArbiError) {
-      console.error(err.message);
+      error(err.message);
       process.exit(1);
     }
     throw err;
   }
 }
 function printTable(columns, rows) {
-  console.log(columns.map((c) => c.header.padEnd(c.width)).join(""));
+  console.log(chalk__default.default.bold(columns.map((c) => c.header.padEnd(c.width)).join("")));
   for (const row of rows) {
     console.log(
       columns.map((c) => {
@@ -3802,7 +3931,7 @@ function parseJsonArg(input2, example) {
   try {
     return JSON.parse(input2);
   } catch {
-    console.error(`Invalid JSON. Example: ${example}`);
+    error(`Invalid JSON. Example: ${example}`);
     process.exit(1);
   }
 }
@@ -3849,9 +3978,9 @@ function registerWorkspacesCommand(program2) {
       if (id) {
         const ws2 = data.find((w) => w.external_id === id);
         if (!ws2) {
-          console.error(`Workspace ${id} not found.`);
-          console.error("Available workspaces:");
-          for (const w of data) console.error(`  ${w.external_id}  ${w.name}`);
+          error(`Workspace ${id} not found.`);
+          error("Available workspaces:");
+          for (const w of data) error(`  ${w.external_id}  ${w.name}`);
           process.exit(1);
         }
         selectedId = id;
@@ -3861,7 +3990,7 @@ function registerWorkspacesCommand(program2) {
       }
       updateConfig({ selectedWorkspaceId: selectedId });
       const ws = data.find((w) => w.external_id === selectedId);
-      console.log(`Selected: ${ws.name} (${selectedId})`);
+      success(`Selected: ${ws.name} (${ref(selectedId)})`);
     })()
   );
   workspace.command("create <name>").description("Create a new workspace").option("-d, --description <text>", "Workspace description").option("--public", "Make workspace public", false).action(
@@ -3873,14 +4002,14 @@ function registerWorkspacesCommand(program2) {
         opts.description,
         opts.public ?? false
       );
-      console.log(`Created: ${data.name} (${data.external_id})`);
+      success(`Created: ${data.name} (${ref(data.external_id)})`);
     })()
   );
   workspace.command("delete <id>").description("Delete a workspace").action(
     (id) => runAction(async () => {
       const { arbi } = await resolveAuth();
       const data = await core.workspaces.deleteWorkspace(arbi, id);
-      console.log(data?.detail ?? `Deleted workspace ${id}`);
+      success(data?.detail ?? `Deleted workspace ${id}`);
     })()
   );
   workspace.command("update <id> <json>").description("Update workspace properties (pass JSON)").action(
@@ -3888,7 +4017,7 @@ function registerWorkspacesCommand(program2) {
       const body = parseJsonArg(json, `arbi workspace update wrk-123 '{"name": "New Name"}'`);
       const { arbi } = await resolveWorkspace(id);
       const data = await core.workspaces.updateWorkspace(arbi, id, body);
-      console.log(`Updated: ${data.name} (${data.external_id})`);
+      success(`Updated: ${data.name} (${ref(data.external_id)})`);
     })()
   );
   workspace.command("users").description("List users in the active workspace").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").action(
@@ -3932,14 +4061,14 @@ function registerWorkspacesCommand(program2) {
       const { arbi, workspaceId } = await resolveWorkspace(opts.workspace);
       const role = opts.role ?? "collaborator";
       const data = await core.workspaces.addWorkspaceUsers(arbi, workspaceId, emails, role);
-      for (const u of data) console.log(`Added: ${u.user.email} as ${u.role}`);
+      for (const u of data) success(`Added: ${u.user.email} as ${u.role}`);
     })()
   );
   workspace.command("remove-user <user-ids...>").description("Remove users from the active workspace").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").action(
     (userIds, opts) => runAction(async () => {
       const { arbi, workspaceId } = await resolveWorkspace(opts.workspace);
       await core.workspaces.removeWorkspaceUsers(arbi, workspaceId, userIds);
-      console.log(`Removed ${userIds.length} user(s).`);
+      success(`Removed ${userIds.length} user(s).`);
     })()
   );
   workspace.command("set-role <role> <user-ids...>").description("Update user roles (owner, collaborator, guest)").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").action(
@@ -3951,19 +4080,19 @@ function registerWorkspacesCommand(program2) {
         userIds,
         role
       );
-      for (const u of data) console.log(`Updated: ${u.user.email} \u2192 ${u.role}`);
+      for (const u of data) success(`Updated: ${u.user.email} \u2192 ${u.role}`);
     })()
   );
   workspace.command("copy <target-workspace-id> <doc-ids...>").description("Copy documents to another workspace").option("-w, --workspace <id>", "Source workspace ID (defaults to selected workspace)").action(
     (targetId, docIds, opts) => runAction(async () => {
       const { arbi, workspaceId } = await resolveWorkspace(opts.workspace);
       const data = await core.workspaces.copyDocuments(arbi, workspaceId, targetId, docIds);
-      console.log(`${data.detail} (${data.documents_copied} document(s) copied)`);
+      success(`${data.detail} (${data.documents_copied} document(s) copied)`);
     })()
   );
 }
-function statusSymbol(status) {
-  switch (status) {
+function statusSymbol(status2) {
+  switch (status2) {
     case "completed":
       return "\u2713";
     // ✓
@@ -4060,7 +4189,7 @@ function registerDocsCommand(program2) {
         docIds = selected;
       }
       await core.documents.deleteDocuments(arbi, docIds);
-      console.log(`Deleted ${docIds.length} document(s).`);
+      success(`Deleted ${docIds.length} document(s).`);
     })()
   );
   doc.command("update [json]").description("Update documents (interactive form if no JSON given)").action(
@@ -4070,7 +4199,7 @@ function registerDocsCommand(program2) {
         const docs = Array.isArray(parsed) ? parsed : parsed.documents;
         const { arbi } = await resolveWorkspace();
         const data = await core.documents.updateDocuments(arbi, docs);
-        console.log(`Updated ${data.length} document(s).`);
+        success(`Updated ${data.length} document(s).`);
       } else {
         const { arbi, workspaceId } = await resolveWorkspace();
         const choices = await fetchDocChoices(arbi, workspaceId);
@@ -4096,7 +4225,7 @@ function registerDocsCommand(program2) {
         const data = await core.documents.updateDocuments(arbi, [
           { external_id: docId, [field]: value }
         ]);
-        console.log(`Updated ${data.length} document(s).`);
+        success(`Updated ${data.length} document(s).`);
       }
     })()
   );
@@ -4104,9 +4233,9 @@ function registerDocsCommand(program2) {
     (urls, opts) => runAction(async () => {
       const { arbi, workspaceId } = await resolveWorkspace(opts.workspace);
       const data = await core.documents.uploadUrl(arbi, urls, workspaceId, opts.shared ?? false);
-      console.log(`Uploaded: ${data.doc_ext_ids.join(", ")}`);
+      success(`Uploaded: ${data.doc_ext_ids.join(", ")}`);
       if (data.duplicates && data.duplicates.length > 0) {
-        console.log(`Duplicates: ${data.duplicates.join(", ")}`);
+        warn(`Duplicates: ${data.duplicates.join(", ")}`);
       }
     })()
   );
@@ -4125,7 +4254,7 @@ function registerDocsCommand(program2) {
           { name: "Marker (raw)", value: "marker" }
         ]);
       } else if (!validStages.includes(stage)) {
-        console.error(`Invalid stage: ${stage}. Must be one of: ${validStages.join(", ")}`);
+        error(`Invalid stage: ${stage}. Must be one of: ${validStages.join(", ")}`);
         process.exit(1);
       }
       const data = await core.documents.getParsedContent(
@@ -4142,7 +4271,7 @@ function registerUploadCommand(program2) {
     (files, opts) => runAction(async () => {
       for (const f of files) {
         if (!fs__default.default.existsSync(f)) {
-          console.error(`File not found: ${f}`);
+          error(`File not found: ${f}`);
           process.exit(1);
         }
       }
@@ -4153,9 +4282,9 @@ function registerUploadCommand(program2) {
       const auth = { baseUrl: config.baseUrl, accessToken, workspaceKeyHeader };
       for (const filePath of files) {
         const result = await core.documents.uploadLocalFile(auth, workspaceId, filePath);
-        console.log(`Uploaded: ${result.fileName} (${result.doc_ext_ids.join(", ")})`);
+        success(`Uploaded: ${result.fileName} (${result.doc_ext_ids.join(", ")})`);
         if (result.duplicates && result.duplicates.length > 0) {
-          console.log(`  Duplicates: ${result.duplicates.join(", ")}`);
+          warn(`  Duplicates: ${result.duplicates.join(", ")}`);
         }
         for (const id of result.doc_ext_ids) uploadedDocs.set(id, result.fileName);
       }
@@ -4170,12 +4299,12 @@ Watching ${pending.size} document(s)...`);
             if (sdk.isMessageType(msg, "task_update")) {
               if (!pending.has(msg.doc_ext_id)) return;
               console.log(
-                `  ${uploadedDocs.get(msg.doc_ext_id) || msg.file_name}: ${msg.status} (${msg.progress}%)`
+                `  ${uploadedDocs.get(msg.doc_ext_id) || msg.file_name}: ${status(msg.status)} (${msg.progress}%)`
               );
               if (msg.status === "completed" || msg.status === "failed") {
                 pending.delete(msg.doc_ext_id);
                 if (pending.size === 0) {
-                  console.log("\nAll documents processed.");
+                  success("\nAll documents processed.");
                   conn.close();
                 }
               }
@@ -4183,7 +4312,7 @@ Watching ${pending.size} document(s)...`);
           },
           onClose: () => {
             if (pending.size > 0)
-              console.log(`
+              warn(`
 Connection closed. ${pending.size} document(s) still processing.`);
           }
         });
@@ -4221,7 +4350,7 @@ function registerDownloadCommand(program2) {
       const outputPath = opts.output || path__default.default.join(process.cwd(), filename);
       const buffer = Buffer.from(await res.arrayBuffer());
       fs__default.default.writeFileSync(outputPath, buffer);
-      console.log(
+      success(
         `Downloaded: ${path__default.default.basename(outputPath)} (${(buffer.length / (1024 * 1024)).toFixed(1)} MB)`
       );
     })()
@@ -4254,12 +4383,12 @@ function registerAskCommand(program2) {
         onAgentStep: (data) => {
           if (opts.verbose) {
             const focus = data.focus || data.status || "";
-            console.error(`
-[agent] ${focus}`);
+            console.error(chalk__default.default.dim(`
+[agent] ${focus}`));
           }
         },
-        onError: (message) => console.error(`
-Error: ${message}`)
+        onError: (message) => console.error(chalk__default.default.red(`
+Error: ${message}`))
       });
       process.stdout.write("\n");
       if (result.assistantMessageExtId) {
@@ -4275,8 +4404,11 @@ Error: ${message}`)
     })()
   );
 }
-function timestamp() {
-  return (/* @__PURE__ */ new Date()).toLocaleTimeString("en-GB", { hour12: false });
+function colorize2(level, text) {
+  if (level === "success") return chalk__default.default.green(text);
+  if (level === "error") return chalk__default.default.red(text);
+  if (level === "warning") return chalk__default.default.yellow(text);
+  return text;
 }
 function registerWatchCommand(program2) {
   program2.command("watch").description("Watch workspace activity in real time").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").action(
@@ -4287,23 +4419,14 @@ function registerWatchCommand(program2) {
         baseUrl: config.baseUrl,
         accessToken,
         onMessage: (msg) => {
-          if (sdk.isMessageType(msg, "task_update")) {
-            console.log(
-              `[${timestamp()}] task_update: ${msg.file_name} - ${msg.status} (${msg.progress}%)`
-            );
-          } else if (sdk.isMessageType(msg, "batch_complete")) {
-            console.log(
-              `[${timestamp()}] batch_complete: ${msg.batch_type} - ${msg.doc_ext_ids.length} docs`
-            );
-          } else if (sdk.isMessageType(msg, "error")) {
-            console.log(`[${timestamp()}] error: ${msg.message}`);
-          } else {
-            console.log(`[${timestamp()}] ${msg.type}: ${JSON.stringify(msg)}`);
-          }
+          const { text, level } = core.formatWsMessage(msg);
+          console.log(colorize2(level, text));
         },
         onClose: (code, reason) => {
-          console.log(`
-Connection closed (code ${code}${reason ? ": " + reason : ""})`);
+          console.log(
+            chalk__default.default.yellow(`
+Connection closed (code ${code}${reason ? ": " + reason : ""})`)
+          );
         }
       });
     })()
@@ -4344,7 +4467,7 @@ function registerContactsCommand(program2) {
       }
       const data = await core.contacts.addContacts(arbi, emails);
       for (const c of data) {
-        console.log(`Added: ${c.email} (${c.external_id}) \u2014 ${c.status}`);
+        success(`Added: ${c.email} (${c.external_id}) \u2014 ${c.status}`);
       }
     })()
   );
@@ -4371,7 +4494,7 @@ function registerContactsCommand(program2) {
         if (contactIds.length === 0) return;
       }
       await core.contacts.removeContacts(arbi, contactIds);
-      console.log(`Removed ${contactIds.length} contact(s).`);
+      success(`Removed ${contactIds.length} contact(s).`);
     })()
   );
   contacts.action(async () => {
@@ -4413,7 +4536,7 @@ function registerDmCommand(program2) {
       if (!recipient) {
         const contacts = await core.contacts.listContacts(arbi);
         if (contacts.length === 0) {
-          console.error("No contacts found. Add contacts first: arbi contacts add <email>");
+          error("No contacts found. Add contacts first: arbi contacts add <email>");
           process.exit(1);
         }
         recipient = await promptSelect(
@@ -4437,15 +4560,15 @@ function registerDmCommand(program2) {
         const contacts = await core.contacts.listContacts(arbi);
         const match = contacts.find((c) => c.email === recipient);
         if (!match) {
-          console.error(`No contact found with email: ${recipient}`);
-          console.error("Add them first: arbi contacts add " + recipient);
+          error(`No contact found with email: ${recipient}`);
+          error("Add them first: arbi contacts add " + recipient);
           process.exit(1);
         }
         recipientExtId = match.user?.external_id ?? match.external_id;
       }
       const data = await core.dm.sendDM(arbi, [{ recipient_ext_id: recipientExtId, content }]);
       for (const n of data) {
-        console.log(`Sent: ${n.external_id} \u2192 ${n.recipient.email}`);
+        success(`Sent: ${n.external_id} \u2192 ${n.recipient.email}`);
       }
     })()
   );
@@ -4474,7 +4597,7 @@ function registerDmCommand(program2) {
         if (msgIds.length === 0) return;
       }
       const data = await core.dm.markRead(arbi, msgIds);
-      console.log(`Marked ${data.length} message(s) as read.`);
+      success(`Marked ${data.length} message(s) as read.`);
     })()
   );
   dm.command("delete [ids...]").description("Delete messages (interactive picker if no IDs given)").action(
@@ -4501,7 +4624,7 @@ function registerDmCommand(program2) {
         if (msgIds.length === 0) return;
       }
       await core.dm.deleteDMs(arbi, msgIds);
-      console.log(`Deleted ${msgIds.length} message(s).`);
+      success(`Deleted ${msgIds.length} message(s).`);
     })()
   );
   dm.action(async () => {
@@ -4579,7 +4702,7 @@ function registerTagsCommand(program2) {
         instruction,
         shared
       });
-      console.log(`Created: ${data.name} (${data.external_id})`);
+      success(`Created: ${data.name} (${ref(data.external_id)})`);
     })()
   );
   tags.command("delete [id]").description("Delete a tag (interactive picker if no ID given)").action(
@@ -4590,7 +4713,7 @@ function registerTagsCommand(program2) {
         id = await promptSelect("Select tag to delete", choices);
       }
       const data = await core.tags.deleteTag(arbi, id);
-      console.log(data?.detail ?? `Deleted tag ${id}`);
+      success(data?.detail ?? `Deleted tag ${id}`);
     })()
   );
   tags.command("update [id] [json]").description("Update a tag (interactive picker + form if no args)").action(
@@ -4616,7 +4739,7 @@ function registerTagsCommand(program2) {
         }
       }
       const data = await core.tags.updateTag(arbi, id, body);
-      console.log(`Updated: ${data.name} (${data.external_id})`);
+      success(`Updated: ${data.name} (${ref(data.external_id)})`);
     })()
   );
   tags.action(async (_opts, cmd) => {
@@ -4662,7 +4785,7 @@ function registerDoctagsCommand(program2) {
         docIds = await pickDocs(arbi, workspaceId, "Select documents to tag");
       if (docIds.length === 0) return;
       const data = await core.doctags.assignDocTags(arbi, tagId, docIds, opts?.note);
-      console.log(`Created ${data.length} doctag(s) for tag ${tagId}.`);
+      success(`Created ${data.length} doctag(s) for tag ${tagId}.`);
     })()
   );
   doctagsCmd.command("delete [tag-id] [doc-ids...]").description("Remove a tag from documents (interactive pickers if no args)").action(
@@ -4673,7 +4796,7 @@ function registerDoctagsCommand(program2) {
         docIds = await pickDocs(arbi, workspaceId, "Select documents to untag");
       if (docIds.length === 0) return;
       await core.doctags.removeDocTags(arbi, tagId, docIds);
-      console.log(`Removed tag ${tagId} from ${docIds.length} document(s).`);
+      success(`Removed tag ${tagId} from ${docIds.length} document(s).`);
     })()
   );
   doctagsCmd.command("generate").description("AI-generate doctags (interactive pickers if no flags)").option("--tags <ids>", "Comma-separated tag IDs").option("--docs <ids>", "Comma-separated document IDs").action(
@@ -4765,7 +4888,7 @@ function registerConversationsCommand(program2) {
           "Select conversation to delete"
         );
       const data = await core.conversations.deleteConversation(arbi, conversationId);
-      console.log(data?.detail ?? `Deleted conversation ${conversationId}`);
+      success(data?.detail ?? `Deleted conversation ${conversationId}`);
     })()
   );
   conv.command("share [conversation-id]").description("Share a conversation (interactive picker if no ID given)").action(
@@ -4774,7 +4897,7 @@ function registerConversationsCommand(program2) {
       if (!conversationId)
         conversationId = await pickConversation(arbi, workspaceId, "Select conversation to share");
       const data = await core.conversations.shareConversation(arbi, conversationId);
-      console.log(data?.detail ?? `Shared conversation ${conversationId}`);
+      success(data?.detail ?? `Shared conversation ${conversationId}`);
     })()
   );
   conv.command("title [conversation-id] [title]").description("Update conversation title (interactive if no args)").action(
@@ -4783,7 +4906,7 @@ function registerConversationsCommand(program2) {
       if (!conversationId) conversationId = await pickConversation(arbi, workspaceId);
       if (!title) title = await promptInput("New title");
       const data = await core.conversations.updateConversationTitle(arbi, conversationId, title);
-      console.log(data?.detail ?? `Updated title to: ${title}`);
+      success(data?.detail ?? `Updated title to: ${title}`);
     })()
   );
   conv.command("add-user [conversation-id] [user-id]").description("Add a user to a conversation (interactive if no args)").action(
@@ -4792,7 +4915,7 @@ function registerConversationsCommand(program2) {
       if (!conversationId) conversationId = await pickConversation(arbi, workspaceId);
       if (!userId) userId = await promptInput("User ext_id or email");
       const data = await core.conversations.addConversationUser(arbi, conversationId, userId);
-      console.log(data?.detail ?? `Added user ${userId} to conversation`);
+      success(data?.detail ?? `Added user ${userId} to conversation`);
     })()
   );
   conv.command("remove-user [conversation-id] [user-id]").description("Remove a user from a conversation (interactive if no args)").action(
@@ -4801,7 +4924,7 @@ function registerConversationsCommand(program2) {
       if (!conversationId) conversationId = await pickConversation(arbi, workspaceId);
       if (!userId) userId = await promptInput("User ext_id");
       const data = await core.conversations.removeConversationUser(arbi, conversationId, userId);
-      console.log(data?.detail ?? `Removed user ${userId} from conversation`);
+      success(data?.detail ?? `Removed user ${userId} from conversation`);
     })()
   );
   conv.command("message <message-id>").description("Get message details").action(
@@ -4815,7 +4938,7 @@ function registerConversationsCommand(program2) {
     (messageId) => runAction(async () => {
       const { arbi } = await resolveAuth();
       const data = await core.conversations.deleteMessage(arbi, messageId);
-      console.log(data?.detail ?? `Deleted message ${messageId}`);
+      success(data?.detail ?? `Deleted message ${messageId}`);
     })()
   );
   conv.action(async (_opts, cmd) => {
@@ -4836,7 +4959,7 @@ function registerSettingsCommand(program2) {
       const body = parseJsonArg(json, `arbi settings set '{"hide_online_status": true}'`);
       const { arbi } = await resolveAuth();
       await core.settings.updateSettings(arbi, body);
-      console.log("Settings updated.");
+      success("Settings updated.");
     })()
   );
   settings.action(async () => {
@@ -5041,7 +5164,7 @@ function registerAgentconfigCommand(program2) {
       if (opts?.tag) body.tag_ext_id = opts.tag;
       if (opts?.message) body.parent_message_ext_id = opts.message;
       const data = await core.agentconfig.saveConfig(arbi, body);
-      console.log(`Saved: ${data.title || "(untitled)"} (${data.external_id})`);
+      success(`Saved: ${data.title || "(untitled)"} (${ref(data.external_id)})`);
     })()
   );
   ac.command("delete [config-id]").description("Delete a configuration (picker if no ID)").action(
@@ -5049,7 +5172,7 @@ function registerAgentconfigCommand(program2) {
       const { arbi } = await resolveAuth();
       if (!configId) configId = await pickConfig(arbi, "Select configuration to delete");
       const data = await core.agentconfig.deleteConfig(arbi, configId);
-      console.log(data.detail);
+      success(data.detail);
     })()
   );
   ac.command("schema").description("Show JSON schema for all configuration models").action(
@@ -5068,20 +5191,20 @@ function registerHealthCommand(program2) {
     runAction(async () => {
       const { arbi } = await resolveAuth();
       const data = await core.health.getHealth(arbi);
-      console.log(`Status: ${data.status}`);
-      if (data.backend_git_hash) console.log(`Backend: ${data.backend_git_hash}`);
-      if (data.frontend_docker_version) console.log(`Frontend: ${data.frontend_docker_version}`);
+      label("Status:", status(data.status));
+      if (data.backend_git_hash) label("Backend:", data.backend_git_hash);
+      if (data.frontend_docker_version) label("Frontend:", data.frontend_docker_version);
       if (data.services.length > 0) {
-        console.log("\nServices:");
+        bold("\nServices:");
         for (const s of data.services) {
-          console.log(`  ${s.name}: ${s.status}${s.detail ? ` \u2014 ${s.detail}` : ""}`);
+          console.log(`  ${s.name}: ${status(s.status)}${s.detail ? ` \u2014 ${s.detail}` : ""}`);
         }
       }
       if (data.models_health) {
-        console.log(`
+        bold(`
 Models (${data.models_health.application}):`);
         for (const m of data.models_health.models) {
-          console.log(`  ${m.model}: ${m.status}${m.detail ? ` \u2014 ${m.detail}` : ""}`);
+          console.log(`  ${m.model}: ${status(m.status)}${m.detail ? ` \u2014 ${m.detail}` : ""}`);
         }
       }
     })
@@ -5107,7 +5230,7 @@ Models (${data.models_health.application}):`);
 }
 function registerTuiCommand(program2) {
   program2.command("tui").description("Launch interactive terminal UI").option("-w, --workspace <id>", "Workspace ID to use").allowUnknownOption(true).action((opts, cmd) => {
-    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('index.cjs', document.baseURI).href)));
+    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('index.js', document.baseURI).href)));
     const entryPoint = require2.resolve("@arbidocs/tui");
     const args = [];
     if (opts.workspace) {
@@ -5124,31 +5247,31 @@ function registerTuiCommand(program2) {
 function registerUpdateCommand(program2) {
   const update = program2.command("update").description("Update ARBI CLI to the latest version").action(async () => {
     const current = getCurrentVersion();
-    console.log(`Current version: ${current}`);
+    label("Current version:", current);
     console.log("Checking for updates...\n");
     try {
       const latest = getLatestVersion() || child_process.execSync("npm view @arbidocs/cli version", { encoding: "utf8" }).trim();
       if (latest === current) {
-        console.log("Already up to date.");
+        success("Already up to date.");
         return;
       }
-      console.log(`New version available: ${latest}`);
+      success(`New version available: ${latest}`);
       await showChangelog(current, latest);
       console.log("\nUpdating...\n");
       child_process.execSync("npm install -g @arbidocs/cli@latest", { stdio: "inherit" });
-      console.log(`
+      success(`
 Updated to ${latest}.`);
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
-      console.error(`Update failed: ${msg}`);
-      console.error("\nYou can update manually with:");
-      console.error("  npm install -g @arbidocs/cli@latest");
+      error(`Update failed: ${msg}`);
+      error("\nYou can update manually with:");
+      dim("  npm install -g @arbidocs/cli@latest");
       process.exit(1);
     }
   });
   update.command("auto").description("Enable automatic updates on login").action(() => {
     updateConfig({ autoUpdate: true });
-    console.log("Auto-update enabled. ARBI CLI will update automatically on login.");
+    success("Auto-update enabled. ARBI CLI will update automatically on login.");
   });
 }
 
@@ -5161,7 +5284,7 @@ console.info = (...args) => {
   _origInfo(...args);
 };
 var program = new commander.Command();
-program.name("arbi").description("ARBI CLI \u2014 interact with ARBI from the terminal").version("0.3.3");
+program.name("arbi").description("ARBI CLI \u2014 interact with ARBI from the terminal").version("0.3.5");
 registerConfigCommand(program);
 registerLoginCommand(program);
 registerRegisterCommand(program);
@@ -5184,5 +5307,5 @@ registerHealthCommand(program);
 registerTuiCommand(program);
 registerUpdateCommand(program);
 program.parse();
-//# sourceMappingURL=index.cjs.map
-//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map