@arbidocs/cli 0.3.42 → 0.3.44

package/dist/index.js

@@ -3641,7 +3641,7 @@ function getLatestVersion(skipCache = false) {
   }
 }
 function getCurrentVersion() {
-  return "0.3.42";
+  return "0.3.44";
 }
 function readChangelog(fromVersion, toVersion) {
   try {
@@ -3694,17 +3694,17 @@ function showChangelog(fromVersion, toVersion) {
 async function checkForUpdates(autoUpdate) {
   try {
     const latest = getLatestVersion();
-    if (!latest || latest === "0.3.42") return;
+    if (!latest || latest === "0.3.44") return;
     if (autoUpdate) {
       warn(`
-Your arbi version is out of date (${"0.3.42"} \u2192 ${latest}). Updating...`);
+Your arbi version is out of date (${"0.3.44"} \u2192 ${latest}). Updating...`);
       child_process.execSync("npm install -g @arbidocs/cli@latest", { stdio: "inherit" });
-      showChangelog("0.3.42", latest);
+      showChangelog("0.3.44", latest);
       console.log(`Updated to ${latest}.`);
     } else {
       warn(
         `
-Your arbi version is out of date (${"0.3.42"} \u2192 ${latest}).
+Your arbi version is out of date (${"0.3.44"} \u2192 ${latest}).
 Run "arbi update" to upgrade, or "arbi update auto" to always stay up to date.`
       );
     }
@@ -3714,9 +3714,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.42") {
+    if (cached && cached.latest !== "0.3.44") {
       warn(
-        `Your arbi version is out of date (${"0.3.42"} \u2192 ${cached.latest}). Run "arbi update".`
+        `Your arbi version is out of date (${"0.3.44"} \u2192 ${cached.latest}). Run "arbi update".`
       );
     }
   } catch {
@@ -4101,14 +4101,29 @@ function registerListenCommand(program2) {
 
 // src/commands/login.ts
 function registerLoginCommand(program2) {
-  program2.command("login").description("Log in to ARBI").option("-e, --email <email>", "Email address (or ARBI_EMAIL env var)").option("-p, --password <password>", "Password (or ARBI_PASSWORD env var)").option("-k, --signing-key <key>", "Signing key (base64) for agent recovery login").option("-w, --workspace <id>", "Workspace ID to select after login").option("--sso", "Log in with Auth0 SSO (device flow)").option("--agent <name>", "Agent backend to configure (claude, openclaw)").option("--listen", "Start DM listener after login").action(
+  program2.command("login").description("Log in to ARBI").option("-e, --email <email>", "Email address (or ARBI_EMAIL env var)").option("-p, --password <password>", "Password (or ARBI_PASSWORD env var)").option("-k, --signing-key <key>", "Signing key (base64) for agent recovery login").option("-w, --workspace <id>", "Workspace ID to select after login").option("--sso", "Log in with Auth0 SSO (device flow)").option("--agent <name>", "Agent backend to configure (claude, openclaw)").option("--listen", "Start DM listener after login").option("--json", "Emit a single JSON object on success (scripting)").action(
     (opts) => runAction(async () => {
       const config = store.requireConfig();
-      const email = opts.email || process.env.ARBI_EMAIL || await promptInput("Email");
+      const isTty = process.stdin.isTTY;
+      const emailFromFlagOrEnv = opts.email || process.env.ARBI_EMAIL;
+      if (!isTty && !emailFromFlagOrEnv) {
+        const msg = "Email is required when stdin is not a TTY. Use --email <email> or set ARBI_EMAIL.";
+        if (opts.json) console.log(JSON.stringify({ ok: false, error: msg }));
+        else error(msg);
+        process.exit(1);
+      }
+      const passwordFromFlagOrEnv = opts.password || process.env.ARBI_PASSWORD;
+      if (!isTty && !passwordFromFlagOrEnv && !opts.signingKey) {
+        const msg = "Password is required when stdin is not a TTY. Use --password <password> or set ARBI_PASSWORD.";
+        if (opts.json) console.log(JSON.stringify({ ok: false, error: msg }));
+        else error(msg);
+        process.exit(1);
+      }
+      const email = emailFromFlagOrEnv || await promptInput("Email");
       try {
         let ssoPolling = false;
         const { arbi } = opts.signingKey ? await sdk.performSigningKeyLogin(config, email, opts.signingKey, store) : opts.sso ? await (async () => {
-          const pw = opts.password || process.env.ARBI_PASSWORD || await promptPassword("Password");
+          const pw = passwordFromFlagOrEnv || await promptPassword("Password");
           const result = await sdk.performSsoDeviceFlowLogin(config, email, pw, store, {
             onUserCode: (userCode, verificationUri) => {
               console.log(`
@@ -4127,9 +4142,10 @@ Open this URL in your browser:
           if (ssoPolling) console.log("\n");
           return result;
         })() : await (async () => {
-          const pw = opts.password || process.env.ARBI_PASSWORD || await promptPassword("Password");
+          const pw = passwordFromFlagOrEnv || await promptPassword("Password");
           return sdk.performPasswordLogin(config, email, pw, store);
         })();
+        if (!opts.json) success(`Logged in as ${email}`);
         clearChatSession();
         const { data: workspaces3 } = await arbi.fetch.GET("/v1/user/workspaces");
         const wsList = workspaces3 || [];
@@ -4137,26 +4153,39 @@ Open this URL in your browser:
         const memberWorkspaces = wsList.filter(
           (ws) => ws.users?.some((u) => u.user.email === email)
         );
+        let selectedWorkspace;
         if (memberWorkspaces.length === 0) {
-          console.log("No workspaces found. Create one with: arbi workspace create <name>");
+          if (!opts.json) {
+            console.log("No workspaces found. Create one with: arbi workspace create <name>");
+          }
         } else if (opts.workspace) {
           const ws = memberWorkspaces.find((w) => w.external_id === opts.workspace);
           if (!ws) {
-            error(`Workspace ${opts.workspace} not found or you don't have access.`);
+            const msg = `Workspace ${opts.workspace} not found or you don't have access.`;
+            if (opts.json) console.log(JSON.stringify({ ok: false, error: msg }));
+            else error(msg);
             process.exit(1);
           }
           updateConfig({ selectedWorkspaceId: ws.external_id });
-          success(`Workspace: ${ws.name} (${ref(ws.external_id)})`);
+          selectedWorkspace = { external_id: ws.external_id, name: ws.name };
+          if (!opts.json) success(`Workspace: ${ws.name} (${ref(ws.external_id)})`);
         } else if (memberWorkspaces.length === 1) {
           updateConfig({ selectedWorkspaceId: memberWorkspaces[0].external_id });
-          success(
-            `Workspace: ${memberWorkspaces[0].name} (${ref(memberWorkspaces[0].external_id)})`
-          );
+          selectedWorkspace = {
+            external_id: memberWorkspaces[0].external_id,
+            name: memberWorkspaces[0].name
+          };
+          if (!opts.json)
+            success(
+              `Workspace: ${memberWorkspaces[0].name} (${ref(memberWorkspaces[0].external_id)})`
+            );
+        } else if (opts.json) {
         } else {
           const choices = sdk.formatWorkspaceChoices(memberWorkspaces);
           const selected = await promptSelect("Select workspace", choices);
           updateConfig({ selectedWorkspaceId: selected });
           const ws = memberWorkspaces.find((w) => w.external_id === selected);
+          selectedWorkspace = { external_id: selected, name: ws.name };
           success(`Workspace: ${ws.name} (${ref(selected)})`);
           dim('\nTip: Run "arbi config alias" to use A as a shortcut for "arbi ask"');
         }
@@ -4166,8 +4195,21 @@ Open this URL in your browser:
             await startListening(opts.agent, config);
           }
         }
+        if (opts.json) {
+          console.log(
+            JSON.stringify({
+              ok: true,
+              email,
+              server: config.baseUrl,
+              workspace: selectedWorkspace ?? null,
+              workspaces_available: memberWorkspaces.length
+            })
+          );
+        }
       } catch (err) {
-        error(`Login failed: ${formatCliError(err)}`);
+        const msg = formatCliError(err);
+        if (opts.json) console.log(JSON.stringify({ ok: false, error: msg }));
+        else error(`Login failed: ${msg}`);
         process.exit(1);
       } finally {
         await checkForUpdates(getConfig()?.autoUpdate);
@@ -4176,7 +4218,19 @@ Open this URL in your browser:
   );
 }
 var CENTRAL_API_URL = "https://central.arbi.work";
-async function getVerificationCode(email, apiKey) {
+var RegisterHintError = class extends Error {
+  constructor(message, kind) {
+    super(message);
+    this.kind = kind;
+    this.name = "RegisterHintError";
+  }
+  kind;
+};
+function looksLikeAgentEmail(email) {
+  const local = email.split("@")[0] ?? "";
+  return local.toLowerCase().startsWith("agent-");
+}
+async function getVerificationCode(email, apiKey, deploymentDomain) {
   const params = new URLSearchParams({ email });
   const res = await fetch(`${CENTRAL_API_URL}/license-management/verify-ci?${params.toString()}`, {
     method: "GET",
@@ -4184,6 +4238,24 @@ async function getVerificationCode(email, apiKey) {
   });
   if (!res.ok) {
     const body = await res.text().catch(() => "");
+    if (res.status === 403 && /agent-\*/i.test(body)) {
+      throw new RegisterHintError(
+        `This SUPPORT_API_KEY is a deployment key that can only register agent-*@${deploymentDomain} emails. Prefix your email with "agent-" (e.g. agent-<id>@${deploymentDomain}) or use a superuser key.`,
+        "agent-email-required"
+      );
+    }
+    if (res.status === 403 && /Invalid API Key/i.test(body)) {
+      throw new RegisterHintError(
+        "SUPPORT_API_KEY was rejected by central. Check the key value.",
+        "bad-api-key"
+      );
+    }
+    if (res.status === 404 && /No verification record/i.test(body)) {
+      throw new RegisterHintError(
+        `No pending verification for ${email}. The email may already be registered \u2014 try: arbi login --email ${email}`,
+        "already-registered"
+      );
+    }
     throw new Error(`Failed to get verification code: ${res.status} ${body}`);
   }
   const data = await res.json();
@@ -4192,7 +4264,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").action(
+  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("-y, --yes", 'Skip the post-register "Log in now?" confirmation').option("--json", "Emit a single JSON object on success (scripting)").action(
     (opts) => runAction(async () => {
       const config = requireConfig();
       if (opts.nonInteractive) {
@@ -4207,7 +4279,7 @@ async function smartRegister(config, opts) {
   let email = opts.email || process.env.ARBI_EMAIL || await promptInput("Email");
   if ((opts.email || process.env.ARBI_EMAIL) && !email.includes("@")) {
     email = `${email}@${config.deploymentDomain}`;
-    console.log(`Using email: ${email}`);
+    if (!opts.json) console.log(`Using email: ${email}`);
   }
   const arbi = client.createArbiClient({
     baseUrl: config.baseUrl,
@@ -4261,16 +4333,22 @@ async function smartRegister(config, opts) {
       firstName,
       lastName
     });
-    success(`
+    if (!opts.json) success(`
 Registered successfully as ${email}`);
   } catch (err) {
-    error(`Registration failed: ${formatCliError(err)}`);
+    if (opts.json) {
+      console.log(JSON.stringify({ ok: false, stage: "register", error: formatCliError(err) }));
+    } else {
+      error(`Registration failed: ${formatCliError(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?");
+  const doLogin = allFlagsProvided || opts.yes || await promptConfirm("Log in now?");
   if (doLogin) {
-    await loginAfterRegister(config, email, pw);
+    await loginAfterRegister(config, email, pw, { json: opts.json });
+  } else if (opts.json) {
+    console.log(JSON.stringify({ ok: true, email, server: config.baseUrl, logged_in: false }));
   }
 }
 async function nonInteractiveRegister(config, opts) {
@@ -4283,12 +4361,18 @@ async function nonInteractiveRegister(config, opts) {
   }
   if (!email.includes("@")) {
     email = `${email}@${config.deploymentDomain}`;
-    console.log(`Using email: ${email}`);
+    if (!opts.json) console.log(`Using email: ${email}`);
   }
   if (!password2) {
     error("Password required. Use --password <password> or set ARBI_PASSWORD");
     process.exit(1);
   }
+  if (supportApiKey && !opts.verificationCode && !looksLikeAgentEmail(email)) {
+    const hint = `Email "${email}" does not start with "agent-". SUPPORT_API_KEY deployment keys only accept agent-*@${config.deploymentDomain} emails. If your key is a superuser key this may still succeed \u2014 proceeding...`;
+    if (!opts.json) {
+      console.error(`Warning: ${hint}`);
+    }
+  }
   const arbi = client.createArbiClient({
     baseUrl: config.baseUrl,
     deploymentDomain: config.deploymentDomain,
@@ -4309,10 +4393,37 @@ async function nonInteractiveRegister(config, opts) {
       body: { email }
     });
     if (verifyResponse.error) {
-      error(`verify-email failed: ${JSON.stringify(verifyResponse.error)}`);
+      if (opts.json) {
+        console.log(
+          JSON.stringify({
+            ok: false,
+            stage: "verify-email",
+            error: JSON.stringify(verifyResponse.error)
+          })
+        );
+      } else {
+        error(`verify-email failed: ${JSON.stringify(verifyResponse.error)}`);
+      }
+      process.exit(1);
+    }
+    try {
+      verificationCode = await getVerificationCode(email, supportApiKey, config.deploymentDomain);
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      if (opts.json) {
+        console.log(
+          JSON.stringify({
+            ok: false,
+            stage: "verify-ci",
+            error: msg,
+            kind: err instanceof RegisterHintError ? err.kind : void 0
+          })
+        );
+      } else {
+        error(`Error: ${msg}`);
+      }
       process.exit(1);
     }
-    verificationCode = await getVerificationCode(email, supportApiKey);
   }
   try {
     await arbi.auth.register({
@@ -4322,60 +4433,93 @@ async function nonInteractiveRegister(config, opts) {
       firstName: opts.firstName ?? "Test",
       lastName: opts.lastName ?? "User"
     });
-    success(`Registered: ${email}`);
+    if (!opts.json) success(`Registered: ${email}`);
   } catch (err) {
-    error(`Registration failed: ${formatCliError(err)}`);
+    if (opts.json) {
+      console.log(JSON.stringify({ ok: false, stage: "register", error: formatCliError(err) }));
+    } else {
+      error(`Registration failed: ${formatCliError(err)}`);
+    }
     process.exit(1);
   }
-  await loginAfterRegister(config, email, password2);
+  await loginAfterRegister(config, email, password2, { json: opts.json });
 }
-async function loginAfterRegister(config, email, password2) {
+async function loginAfterRegister(config, email, password2, { json = false } = {}) {
   try {
     const { arbi, loginResult } = await sdk.performPasswordLogin(config, email, password2, store);
-    success(`Logged in as ${email}`);
+    if (!json) success(`Logged in as ${email}`);
     const { data: workspaces3 } = await arbi.fetch.GET("/v1/user/workspaces");
     const wsList = workspaces3 || [];
     updateCompletionCache(wsList);
-    const memberWorkspaces = wsList.filter((ws2) => ws2.users?.some((u) => u.user.email === email));
+    const memberWorkspaces = wsList.filter((ws) => ws.users?.some((u) => u.user.email === email));
+    let selectedWorkspace;
     if (memberWorkspaces.length === 0) {
-      console.log("Creating your first workspace...");
+      if (!json) console.log("Creating your first workspace...");
       const userProjects = await sdk.projects.listProjects(arbi);
       const defaultProjectExtId = userProjects[0]?.external_id;
       if (!defaultProjectExtId) throw new Error("No projects found for user");
       const encryptedKey = await sdk.generateNewWorkspaceKey(arbi, loginResult.serverSessionKey);
-      const ws2 = await sdk.workspaces.createWorkspace(
+      const ws = await sdk.workspaces.createWorkspace(
         arbi,
         "My First Workspace",
         encryptedKey,
         defaultProjectExtId
       );
-      updateConfig({ selectedWorkspaceId: ws2.external_id });
-      success(`Workspace: ${ws2.name} (${ref(ws2.external_id)})`);
-      return;
-    }
-    if (memberWorkspaces.length === 1) {
+      updateConfig({ selectedWorkspaceId: ws.external_id });
+      selectedWorkspace = { external_id: ws.external_id, name: ws.name };
+      if (!json) success(`Workspace: ${ws.name} (${ref(ws.external_id)})`);
+    } else if (memberWorkspaces.length === 1) {
       updateConfig({ selectedWorkspaceId: memberWorkspaces[0].external_id });
-      success(`Workspace: ${memberWorkspaces[0].name} (${ref(memberWorkspaces[0].external_id)})`);
-      return;
+      selectedWorkspace = {
+        external_id: memberWorkspaces[0].external_id,
+        name: memberWorkspaces[0].name
+      };
+      if (!json)
+        success(`Workspace: ${memberWorkspaces[0].name} (${ref(memberWorkspaces[0].external_id)})`);
+    } else if (json) {
+    } else {
+      const choices = sdk.formatWorkspaceChoices(memberWorkspaces);
+      const selected = await promptSelect("Select workspace", choices);
+      updateConfig({ selectedWorkspaceId: selected });
+      const ws = memberWorkspaces.find((w) => w.external_id === selected);
+      selectedWorkspace = { external_id: selected, name: ws.name };
+      success(`Workspace: ${ws.name} (${ref(selected)})`);
+    }
+    if (json) {
+      console.log(
+        JSON.stringify({
+          ok: true,
+          email,
+          server: config.baseUrl,
+          logged_in: true,
+          workspace: selectedWorkspace ?? null,
+          workspaces_available: memberWorkspaces.length
+        })
+      );
     }
-    const choices = sdk.formatWorkspaceChoices(memberWorkspaces);
-    const selected = await promptSelect("Select workspace", choices);
-    updateConfig({ selectedWorkspaceId: selected });
-    const ws = memberWorkspaces.find((w) => w.external_id === selected);
-    success(`Workspace: ${ws.name} (${ref(selected)})`);
   } catch (err) {
-    error(`Login failed: ${formatCliError(err)}`);
-    error("You can log in later with: arbi login");
+    if (json) {
+      console.log(
+        JSON.stringify({ ok: false, stage: "post-register-login", error: formatCliError(err) })
+      );
+    } else {
+      error(`Login failed: ${formatCliError(err)}`);
+      error("You can log in later with: arbi login");
+    }
   }
 }
 
 // src/commands/logout.ts
 function registerLogoutCommand(program2) {
-  program2.command("logout").description("Log out of ARBI").action(() => {
+  program2.command("logout").description("Log out of ARBI").option("--json", "Emit a single JSON object on success (scripting)").action((opts) => {
     deleteCredentials();
     clearChatSession();
     updateConfig({ selectedWorkspaceId: void 0 });
-    success("Logged out.");
+    if (opts.json) {
+      console.log(JSON.stringify({ ok: true }));
+    } else {
+      success("Logged out.");
+    }
   });
 }
 
@@ -7766,145 +7910,160 @@ Updated to ${latest}.`);
     success("Auto-update enabled. ARBI CLI will update automatically on login.");
   });
 }
+function isExitPromptError(err) {
+  return err instanceof Error && err.name === "ExitPromptError";
+}
 function registerQuickstartCommand(program2) {
   program2.command("quickstart").description("Interactive setup wizard \u2014 configure, register/login, and select a workspace").argument("[url]", "Server URL (auto-detected if omitted)").action(async (url) => {
-    console.log("\nWelcome to ARBI CLI setup!\n");
-    const { config, source } = store.resolveConfigWithFallbacks();
-    if (url) {
-      const domain = new URL(url).hostname;
-      config.baseUrl = url.replace(/\/$/, "");
+    try {
+      await runQuickstart(url);
+    } catch (err) {
+      if (isExitPromptError(err)) {
+        console.log("\nSetup cancelled.");
+        process.exit(130);
+      }
+      error(`Quickstart failed: ${sdk.getErrorMessage(err)}`);
+      process.exit(1);
+    }
+  });
+}
+async function runQuickstart(url) {
+  console.log("\nWelcome to ARBI CLI setup!\n");
+  const { config, source } = store.resolveConfigWithFallbacks();
+  if (url) {
+    const domain = new URL(url).hostname;
+    config.baseUrl = url.replace(/\/$/, "");
+    config.deploymentDomain = domain;
+    store.saveConfig(config);
+  }
+  const useDetected = await promptConfirm(`Use server ${config.baseUrl}?`, true);
+  if (!useDetected) {
+    const customUrl = await promptInput("Server URL");
+    try {
+      const domain = new URL(customUrl).hostname;
+      config.baseUrl = customUrl.replace(/\/$/, "");
       config.deploymentDomain = domain;
       store.saveConfig(config);
+    } catch {
+      error("Invalid URL");
+      process.exit(1);
     }
-    const useDetected = await promptConfirm(`Use server ${config.baseUrl}?`, true);
-    if (!useDetected) {
-      const customUrl = await promptInput("Server URL");
-      try {
-        const domain = new URL(customUrl).hostname;
-        config.baseUrl = customUrl.replace(/\/$/, "");
-        config.deploymentDomain = domain;
-        store.saveConfig(config);
-      } catch {
-        error("Invalid URL");
-        process.exit(1);
-      }
-    } else if (source !== "config") {
-      store.saveConfig(config);
+  } else if (source !== "config") {
+    store.saveConfig(config);
+  }
+  dim(`Server: ${config.baseUrl}`);
+  const action = await promptSelect("Do you have an account?", [
+    { name: "Yes, log me in", value: "login" },
+    { name: "No, register a new account", value: "register" }
+  ]);
+  const arbiClient = client.createArbiClient({
+    baseUrl: config.baseUrl,
+    deploymentDomain: config.deploymentDomain,
+    credentials: "omit"
+  });
+  await arbiClient.crypto.initSodium();
+  let email;
+  let password2;
+  if (action === "register") {
+    email = await promptInput("Email");
+    password2 = await promptPassword("Password");
+    const confirmPw = await promptPassword("Confirm password");
+    if (password2 !== confirmPw) {
+      error("Passwords do not match.");
+      process.exit(1);
     }
-    dim(`Server: ${config.baseUrl}`);
-    const action = await promptSelect("Do you have an account?", [
-      { name: "Yes, log me in", value: "login" },
-      { name: "No, register a new account", value: "register" }
+    const codeMethod = await promptSelect("Verification method", [
+      { name: "I have an invitation code", value: "code" },
+      { name: "Send me a verification email", value: "email" }
     ]);
-    const arbiClient = client.createArbiClient({
-      baseUrl: config.baseUrl,
-      deploymentDomain: config.deploymentDomain,
-      credentials: "omit"
-    });
-    await arbiClient.crypto.initSodium();
-    let email;
-    let password2;
-    if (action === "register") {
-      email = await promptInput("Email");
-      password2 = await promptPassword("Password");
-      const confirmPw = await promptPassword("Confirm password");
-      if (password2 !== confirmPw) {
-        error("Passwords do not match.");
-        process.exit(1);
-      }
-      const codeMethod = await promptSelect("Verification method", [
-        { name: "I have an invitation code", value: "code" },
-        { name: "Send me a verification email", value: "email" }
-      ]);
-      let verificationCode;
-      if (codeMethod === "code") {
-        verificationCode = await promptInput("Invitation code");
-      } else {
-        console.log("Sending verification email...");
-        const verifyResponse = await arbiClient.fetch.POST("/v1/user/verify-email", {
-          body: { email }
-        });
-        if (verifyResponse.error) {
-          error(`Failed to send verification email: ${JSON.stringify(verifyResponse.error)}`);
-          process.exit(1);
-        }
-        success("Verification email sent. Check your inbox.");
-        verificationCode = await promptInput("Verification code");
-      }
-      try {
-        const firstName = await promptInput("First name", false) || "User";
-        const lastName = await promptInput("Last name", false) || "";
-        await arbiClient.auth.register({
-          email,
-          password: password2,
-          verificationCode,
-          firstName,
-          lastName
-        });
-        success(`Registered as ${email}`);
-      } catch (err) {
-        error(`Registration failed: ${sdk.getErrorMessage(err)}`);
+    let verificationCode;
+    if (codeMethod === "code") {
+      verificationCode = await promptInput("Invitation code");
+    } else {
+      console.log("Sending verification email...");
+      const verifyResponse = await arbiClient.fetch.POST("/v1/user/verify-email", {
+        body: { email }
+      });
+      if (verifyResponse.error) {
+        error(`Failed to send verification email: ${JSON.stringify(verifyResponse.error)}`);
         process.exit(1);
       }
-    } else {
-      email = await promptInput("Email");
-      password2 = await promptPassword("Password");
+      success("Verification email sent. Check your inbox.");
+      verificationCode = await promptInput("Verification code");
     }
     try {
-      const { arbi, loginResult } = await sdk.performPasswordLogin(config, email, password2, store);
-      success(`Logged in as ${email}`);
-      const { data: workspaces3 } = await arbi.fetch.GET("/v1/user/workspaces");
-      const wsList = workspaces3 || [];
-      const memberWorkspaces = wsList.filter(
-        (ws) => ws.users?.some((u) => u.user.email === email)
+      const firstName = await promptInput("First name", false) || "User";
+      const lastName = await promptInput("Last name", false) || "";
+      await arbiClient.auth.register({
+        email,
+        password: password2,
+        verificationCode,
+        firstName,
+        lastName
+      });
+      success(`Registered as ${email}`);
+    } catch (err) {
+      if (isExitPromptError(err)) throw err;
+      error(`Registration failed: ${sdk.getErrorMessage(err)}`);
+      process.exit(1);
+    }
+  } else {
+    email = await promptInput("Email");
+    password2 = await promptPassword("Password");
+  }
+  try {
+    const { arbi, loginResult } = await sdk.performPasswordLogin(config, email, password2, store);
+    success(`Logged in as ${email}`);
+    const { data: workspaces3 } = await arbi.fetch.GET("/v1/user/workspaces");
+    const wsList = workspaces3 || [];
+    const memberWorkspaces = wsList.filter((ws) => ws.users?.some((u) => u.user.email === email));
+    const userProjects = await sdk.projects.listProjects(arbi);
+    const defaultProjectExtId = userProjects[0]?.external_id;
+    if (!defaultProjectExtId) throw new Error("No projects found for user");
+    let workspaceId;
+    let workspaceName;
+    if (memberWorkspaces.length === 0) {
+      console.log("Creating your first workspace...");
+      const encryptedKey = await sdk.generateNewWorkspaceKey(arbi, loginResult.serverSessionKey);
+      const ws = await sdk.workspaces.createWorkspace(
+        arbi,
+        "My Workspace",
+        encryptedKey,
+        defaultProjectExtId
       );
-      const userProjects = await sdk.projects.listProjects(arbi);
-      const defaultProjectExtId = userProjects[0]?.external_id;
-      if (!defaultProjectExtId) throw new Error("No projects found for user");
-      let workspaceId;
-      let workspaceName;
-      if (memberWorkspaces.length === 0) {
-        console.log("Creating your first workspace...");
-        const encryptedKey = await sdk.generateNewWorkspaceKey(arbi, loginResult.serverSessionKey);
-        const ws = await sdk.workspaces.createWorkspace(
-          arbi,
-          "My Workspace",
-          encryptedKey,
-          defaultProjectExtId
-        );
+      workspaceId = ws.external_id;
+      workspaceName = ws.name;
+    } else {
+      const choices = [
+        ...sdk.formatWorkspaceChoices(memberWorkspaces),
+        { name: "+ Create new workspace", value: "__new__", description: "" }
+      ];
+      const selected = await promptSelect("Select workspace", choices);
+      if (selected === "__new__") {
+        const name = await promptInput("Workspace name", false) || "My Workspace";
+        const encKey = await sdk.generateNewWorkspaceKey(arbi, loginResult.serverSessionKey);
+        const ws = await sdk.workspaces.createWorkspace(arbi, name, encKey, defaultProjectExtId);
         workspaceId = ws.external_id;
         workspaceName = ws.name;
       } else {
-        const choices = [
-          ...sdk.formatWorkspaceChoices(memberWorkspaces),
-          { name: "+ Create new workspace", value: "__new__", description: "" }
-        ];
-        const selected = await promptSelect("Select workspace", choices);
-        if (selected === "__new__") {
-          const name = await promptInput("Workspace name", false) || "My Workspace";
-          const encKey = await sdk.generateNewWorkspaceKey(arbi, loginResult.serverSessionKey);
-          const ws = await sdk.workspaces.createWorkspace(arbi, name, encKey, defaultProjectExtId);
-          workspaceId = ws.external_id;
-          workspaceName = ws.name;
-        } else {
-          workspaceId = selected;
-          workspaceName = memberWorkspaces.find((w) => w.external_id === selected)?.name || "";
-        }
+        workspaceId = selected;
+        workspaceName = memberWorkspaces.find((w) => w.external_id === selected)?.name || "";
       }
-      updateConfig({ selectedWorkspaceId: workspaceId });
-      console.log("");
-      success("Setup complete!");
-      console.log("");
-      console.log(`  Server:    ${ref(config.baseUrl)}`);
-      console.log(`  Account:   ${ref(email)}`);
-      console.log(`  Workspace: ${workspaceName} (${ref(workspaceId)})`);
-      console.log("");
-      dim('Try: arbi ask "hello"');
-    } catch (err) {
-      error(`Login failed: ${sdk.getErrorMessage(err)}`);
-      process.exit(1);
     }
-  });
+    updateConfig({ selectedWorkspaceId: workspaceId });
+    console.log("");
+    success("Setup complete!");
+    console.log("");
+    console.log(`  Server:    ${ref(config.baseUrl)}`);
+    console.log(`  Account:   ${ref(email)}`);
+    console.log(`  Workspace: ${workspaceName} (${ref(workspaceId)})`);
+    console.log("");
+    dim('Try: arbi ask "hello"');
+  } catch (err) {
+    if (isExitPromptError(err)) throw err;
+    error(`Login failed: ${sdk.getErrorMessage(err)}`);
+    process.exit(1);
+  }
 }
 var CENTRAL_API_URL2 = "https://central.arbi.work";
 var DEFAULT_PASSWORD = "agent-dev-1234";
@@ -8462,7 +8621,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.42");
+program.name("arbi").description("ARBI CLI \u2014 interact with ARBI from the terminal").version("0.3.44");
 registerConfigCommand(program);
 registerLoginCommand(program);
 registerRegisterCommand(program);