@taodigital/agent-cli 1.2.0 → 1.2.1

package/dist/index.js

@@ -7583,6 +7583,19 @@ async function runPostman(message, send, signal) {
 var _pkg = (0, import_node_module2.createRequire)(__filename)("../package.json");
 var PKG_VERSION = _pkg.version;
 var API_URL = process.env.ZETA_API_URL ?? "https://zeta.taodigitalsolutions.com/api";
+var PROFILE_API_URLS = {
+  default: "https://zeta.taodigitalsolutions.com/api",
+  prod: "https://zeta.taodigitalsolutions.com/api",
+  dev: "https://dev.zeta.taodigitalsolutions.com/api",
+  preview: "https://pre.zeta.taodigitalsolutions.com/api",
+  pre: "https://pre.zeta.taodigitalsolutions.com/api",
+  local: "http://localhost:3001/api"
+};
+function resolveApiUrl(explicitFlag, profile) {
+  if (explicitFlag)
+    return explicitFlag;
+  return PROFILE_API_URLS[profile] ?? API_URL;
+}
 var DEFAULT_MAX_SCREENS = process.env.ZETA_DEFAULT_MAX_SCREENS ?? "50";
 var DEFAULT_STRATEGY = process.env.ZETA_DEFAULT_STRATEGY ?? "BFS";
 var LOGIN_TIMEOUT_MS = parseInt(process.env.ZETA_LOGIN_TIMEOUT_MS ?? "") || 1e4;
@@ -7615,17 +7628,79 @@ var TERMINAL_JOB_STATUSES = /* @__PURE__ */ new Set([
 var IDENTITY_PATH = "/workspace/settings/identity";
 var CREDS_DIR = (0, import_node_path2.join)((0, import_node_os2.homedir)(), ".config", "zeta");
 var CREDS_FILE = (0, import_node_path2.join)(CREDS_DIR, "credentials.json");
-async function loadCreds() {
+function resolveProfile(explicit) {
+  return explicit ?? process.env.ZETA_PROFILE ?? "default";
+}
+async function loadCreds(profile) {
   try {
     const raw = await (0, import_promises2.readFile)(CREDS_FILE, "utf-8");
-    return JSON.parse(raw);
+    const data = JSON.parse(raw);
+    if (data._version === 2) {
+      const name = resolveProfile(profile) === "default" && !profile && !process.env.ZETA_PROFILE ? data._default ?? "default" : resolveProfile(profile);
+      return data.profiles?.[name] ?? null;
+    }
+    if (profile && profile !== "default")
+      return null;
+    return data;
   } catch {
     return null;
   }
 }
-async function saveCreds(creds) {
+async function listAllProfiles() {
+  try {
+    const raw = await (0, import_promises2.readFile)(CREDS_FILE, "utf-8");
+    const data = JSON.parse(raw);
+    if (data._version === 2)
+      return data.profiles ?? {};
+    return { default: data };
+  } catch {
+    return {};
+  }
+}
+async function saveCreds(creds, profile) {
   await (0, import_promises2.mkdir)(CREDS_DIR, { recursive: true });
-  await (0, import_promises2.writeFile)(CREDS_FILE, JSON.stringify(creds, null, 2), { encoding: "utf-8", mode: 384 });
+  let data = { _version: 2, _default: "default", profiles: {} };
+  try {
+    const existing = JSON.parse(await (0, import_promises2.readFile)(CREDS_FILE, "utf-8"));
+    if (existing._version === 2) {
+      data = existing;
+    } else {
+      data.profiles.default = existing;
+    }
+  } catch {
+  }
+  const name = resolveProfile(profile);
+  data.profiles[name] = creds;
+  if (!data._default)
+    data._default = name;
+  await (0, import_promises2.writeFile)(CREDS_FILE, JSON.stringify(data, null, 2), { encoding: "utf-8", mode: 384 });
+}
+async function deleteCredsProfile(profile) {
+  try {
+    const raw = await (0, import_promises2.readFile)(CREDS_FILE, "utf-8");
+    const data = JSON.parse(raw);
+    if (data._version === 2) {
+      const name = resolveProfile(profile);
+      if (!data.profiles[name])
+        return false;
+      delete data.profiles[name];
+      if (data._default === name) {
+        const remaining = Object.keys(data.profiles);
+        data._default = remaining[0] ?? "default";
+      }
+      if (Object.keys(data.profiles).length === 0) {
+        await (0, import_promises2.unlink)(CREDS_FILE).catch(() => {
+        });
+      } else {
+        await (0, import_promises2.writeFile)(CREDS_FILE, JSON.stringify(data, null, 2), { encoding: "utf-8", mode: 384 });
+      }
+      return true;
+    }
+    await (0, import_promises2.unlink)(CREDS_FILE);
+    return true;
+  } catch {
+    return false;
+  }
 }
 function executeCommand(command, logger, timeoutMs) {
   return new Promise((resolve, reject) => {
@@ -7821,13 +7896,15 @@ function parseEnvironmentOptions(value) {
 }
 var program2 = new Command();
 program2.version(PKG_VERSION).description("ZeTa-AI Local Agent \u2014 run tests and crawls on private networks").enablePositionalOptions();
-program2.command("login").description("Authenticate with ZeTa and store credentials locally (~/.config/zeta/credentials.json)").argument("[token]", "Agent token (zeta_...) \u2014 skip interactive prompt").option("-a, --api <url>", "ZeTa API URL").option("-t, --token <token>", "ZeTa Agent Token (zeta_...)").action(async (tokenArg, options) => {
+program2.command("login").description("Authenticate with ZeTa and store credentials locally (~/.config/zeta/credentials.json)").argument("[token]", "Agent token (zeta_...) \u2014 skip interactive prompt").option("-a, --api <url>", "ZeTa API URL").option("-t, --token <token>", "ZeTa Agent Token (zeta_...)").option("--profile <name>", 'Credential profile to save under (default: $ZETA_PROFILE or "default")').action(async (tokenArg, options) => {
+  const profile = resolveProfile(options.profile);
   const iface = rl.createInterface({ input: process.stdin, output: process.stdout });
+  const profileDefault = resolveApiUrl(void 0, profile);
   let apiUrl = options.api ?? "";
   if (!apiUrl) {
-    apiUrl = (await iface.question(`API URL [${API_URL}]: `)).trim();
+    apiUrl = (await iface.question(`API URL [${profileDefault}]: `)).trim();
     if (!apiUrl)
-      apiUrl = API_URL;
+      apiUrl = profileDefault;
   }
   let token = tokenArg ?? options.token ?? "";
   if (!token) {
@@ -7871,9 +7948,9 @@ program2.command("login").description("Authenticate with ZeTa and store credenti
     }
   } catch {
   }
-  await saveCreds({ apiUrl, token, workspaceId, workspaceName, projects });
+  await saveCreds({ apiUrl, token, workspaceId, workspaceName, projects }, profile);
   console.log(`
-[ZeTa] \u2713 Logged in to "${workspaceName ?? workspaceId}"`);
+[ZeTa] \u2713 Logged in to "${workspaceName ?? workspaceId}" (profile: ${profile})`);
   console.log(`[ZeTa] Credentials stored: ${CREDS_FILE}`);
   if (projects.length > 0) {
     console.log(`
@@ -7883,32 +7960,64 @@ Projects (${projects.length}):`);
   const serviceProject = projects[0];
   if (serviceProject) {
     console.log(`
-[ZeTa] Installing background service for "${serviceProject.name}"...`);
+[ZeTa] Installing background service for "${serviceProject.name}" (profile: ${profile})...`);
     try {
-      await serviceInstall(serviceProject.id);
+      await serviceInstall(serviceProject.id, profile);
       if (projects.length > 1) {
         console.log(`[ZeTa] \u2139  To switch project: zeta service uninstall && zeta service install -p <project-id>`);
       }
     } catch (e) {
       console.warn(`[ZeTa] \u26A0  Background service install failed: ${e instanceof Error ? e.message : String(e)}`);
-      console.warn(`[ZeTa]    Start manually: zeta -p ${serviceProject.id}`);
+      console.warn(`[ZeTa]    Start manually: zeta --profile ${profile} -p ${serviceProject.id}`);
     }
   }
 });
-program2.command("logout").description("Remove stored ZeTa credentials").action(async () => {
-  try {
-    await (0, import_promises2.unlink)(CREDS_FILE);
-    console.log("[ZeTa] \u2713 Logged out. Credentials removed.");
-  } catch {
-    console.log("[ZeTa] No stored credentials found.");
+program2.command("logout").description("Remove stored ZeTa credentials").option("--profile <name>", 'Remove only this profile (default: $ZETA_PROFILE or "default"); use --all to remove everything').option("--all", "Remove all profiles and the credentials file").action(async (options) => {
+  if (options.all) {
+    try {
+      await (0, import_promises2.unlink)(CREDS_FILE);
+      console.log("[ZeTa] \u2713 All credentials removed.");
+    } catch {
+      console.log("[ZeTa] No stored credentials found.");
+    }
+    return;
+  }
+  const profile = resolveProfile(options.profile);
+  const removed = await deleteCredsProfile(profile);
+  if (removed) {
+    console.log(`[ZeTa] \u2713 Profile "${profile}" removed.`);
+  } else {
+    console.log(`[ZeTa] Profile "${profile}" not found.`);
   }
 });
-program2.command("whoami").description("Show currently stored credentials").action(async () => {
-  const creds = await loadCreds();
+program2.command("whoami").description("Show currently stored credentials").option("--profile <name>", 'Profile to inspect (default: $ZETA_PROFILE or "default")').option("--all", "Show all configured profiles").action(async (options) => {
+  if (options.all) {
+    const all = await listAllProfiles();
+    const names = Object.keys(all);
+    if (names.length === 0) {
+      console.log("[ZeTa] No profiles configured. Run `zeta login`.");
+      process.exit(1);
+    }
+    names.forEach((name) => {
+      const c = all[name];
+      console.log(`
+Profile: ${name}`);
+      console.log(`  Workspace : ${c.workspaceName ?? c.workspaceId ?? "(unknown)"}`);
+      console.log(`  API       : ${c.apiUrl}`);
+      console.log(`  Token     : ${c.token.slice(0, 10)}...`);
+      if (c.projects?.length) {
+        console.log(`  Projects  : ${c.projects.map((p) => p.name).join(", ")}`);
+      }
+    });
+    return;
+  }
+  const profile = resolveProfile(options.profile);
+  const creds = await loadCreds(profile);
   if (!creds) {
-    console.log("[ZeTa] Not logged in. Run `zeta login`.");
+    console.log(`[ZeTa] Profile "${profile}" not found. Run \`zeta login --profile ${profile}\`.`);
     process.exit(1);
   }
+  console.log(`Profile   : ${profile}`);
   console.log(`Workspace : ${creds.workspaceName ?? creds.workspaceId ?? "(unknown)"}`);
   console.log(`API       : ${creds.apiUrl}`);
   console.log(`Token     : ${creds.token.slice(0, 10)}...`);
@@ -7918,6 +8027,20 @@ Projects (${creds.projects.length}):`);
     creds.projects.forEach((p) => console.log(`  ${p.id}  ${p.name}`));
   }
 });
+program2.command("profiles").description("List all configured credential profiles").action(async () => {
+  const all = await listAllProfiles();
+  const names = Object.keys(all);
+  if (names.length === 0) {
+    console.log("[ZeTa] No profiles configured. Run `zeta login`.");
+    return;
+  }
+  console.log("Configured profiles:");
+  names.forEach((name) => {
+    const c = all[name];
+    console.log(`  ${name.padEnd(12)} ${c.apiUrl}  (${c.workspaceName ?? c.workspaceId ?? "unknown"})`);
+  });
+  console.log("\nSwitch profiles with --profile <name> or ZETA_PROFILE=<name>");
+});
 program2.command("crawl").description("Trigger a crawl for a ZeTa project").requiredOption("-p, --project <id>", "ZeTa Project ID").option("-t, --token <token>", "ZeTa API Token (falls back to stored credentials)").option("-a, --api <url>", "ZeTa API URL (falls back to stored credentials)").option("-u, --url <url>", "App URL to crawl (overrides project setting)").option("--max-screens <n>", "Maximum screens to crawl", DEFAULT_MAX_SCREENS).option("--strategy <s>", "Traversal strategy: BFS | DFS | BEST_FIRST", DEFAULT_STRATEGY).option("--url-filter <regex>", "Regex \u2014 only crawl matching URLs").option("--no-wait", "Start crawl and exit without waiting for completion").action(async (options) => {
   const creds = await loadCreds();
   const token = options.token ?? creds?.token;
@@ -8117,8 +8240,11 @@ var LOG_FILE = (0, import_node_path2.join)(CREDS_DIR, "agent.log");
 function resolveZetaBinary() {
   return { node: process.execPath, script: process.argv[1] };
 }
-function getMacPlist(projectId) {
+function getMacPlist(projectId, profile) {
   const { node, script } = resolveZetaBinary();
+  const profileEntry = profile && profile !== "default" ? `
+        <key>ZETA_PROFILE</key>
+        <string>${profile}</string>` : "";
   return `<?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
@@ -8145,13 +8271,15 @@ function getMacPlist(projectId) {
     <key>EnvironmentVariables</key>
     <dict>
         <key>PATH</key>
-        <string>/opt/homebrew/bin:/usr/local/bin:/usr/bin:/bin</string>
+        <string>/opt/homebrew/bin:/usr/local/bin:/usr/bin:/bin</string>${profileEntry}
     </dict>
 </dict>
 </plist>`;
 }
-function getLinuxUnit(projectId) {
+function getLinuxUnit(projectId, profile) {
   const { node, script } = resolveZetaBinary();
+  const profileEnv = profile && profile !== "default" ? `
+Environment=ZETA_PROFILE=${profile}` : "";
   return `[Unit]
 Description=ZeTa AI Local Agent (project ${projectId})
 After=network-online.target
@@ -8163,7 +8291,7 @@ ExecStart=${node} ${script} -p ${projectId}
 Restart=always
 RestartSec=10
 StandardOutput=append:${LOG_FILE}
-StandardError=append:${LOG_FILE}
+StandardError=append:${LOG_FILE}${profileEnv}
 
 [Install]
 WantedBy=default.target
@@ -8175,12 +8303,12 @@ async function getMacPlistPath() {
 function getLinuxUnitPath() {
   return (0, import_node_path2.join)((0, import_node_os2.homedir)(), ".config", "systemd", "user", `${SERVICE_LABEL.replace(/\./g, "-")}.service`);
 }
-async function serviceInstall(projectId) {
+async function serviceInstall(projectId, profile) {
   if (process.platform === "darwin") {
     const plistPath = await getMacPlistPath();
     await (0, import_promises2.mkdir)((0, import_node_path2.join)((0, import_node_os2.homedir)(), "Library", "LaunchAgents"), { recursive: true });
     await (0, import_promises2.mkdir)(CREDS_DIR, { recursive: true });
-    await (0, import_promises2.writeFile)(plistPath, getMacPlist(projectId), { encoding: "utf-8", mode: 420 });
+    await (0, import_promises2.writeFile)(plistPath, getMacPlist(projectId, profile), { encoding: "utf-8", mode: 420 });
     await new Promise((resolve, reject) => {
       (0, import_node_child_process2.exec)(`launchctl unload "${plistPath}" 2>/dev/null; launchctl load -w "${plistPath}"`, (err) => {
         if (err)
@@ -8197,7 +8325,7 @@ async function serviceInstall(projectId) {
     const unitPath = getLinuxUnitPath();
     await (0, import_promises2.mkdir)((0, import_node_path2.join)((0, import_node_os2.homedir)(), ".config", "systemd", "user"), { recursive: true });
     await (0, import_promises2.mkdir)(CREDS_DIR, { recursive: true });
-    await (0, import_promises2.writeFile)(unitPath, getLinuxUnit(projectId), { encoding: "utf-8", mode: 420 });
+    await (0, import_promises2.writeFile)(unitPath, getLinuxUnit(projectId, profile), { encoding: "utf-8", mode: 420 });
     await new Promise((resolve, reject) => {
       (0, import_node_child_process2.exec)("systemctl --user daemon-reload", (err) => {
         if (err)
@@ -8306,12 +8434,13 @@ async function serviceStatus() {
   }
 }
 program2.command("service").description("Manage the ZeTa agent as a background system service").addCommand(
-  new Command("install").description("Install and start the agent as a service (auto-starts on login)").option("-p, --project <id>", "ZeTa Project ID (uses stored credentials if omitted)").action(async (opts) => {
-    const creds = await loadCreds();
+  new Command("install").description("Install and start the agent as a service (auto-starts on login)").option("-p, --project <id>", "ZeTa Project ID (uses stored credentials if omitted)").option("--profile <name>", 'Credential profile to use (default: $ZETA_PROFILE or "default")').action(async (opts) => {
+    const profile = resolveProfile(opts.profile);
+    const creds = await loadCreds(profile);
     let projectId = opts.project;
     if (!projectId) {
       if (!creds?.projects?.length) {
-        console.error("[ZeTa Service] No project ID. Run `zeta <api-key>` first, then `zeta service install -p <project-id>`");
+        console.error(`[ZeTa Service] No project ID. Run \`zeta <api-key> --profile ${profile}\` first, then \`zeta service install -p <project-id>\``);
         process.exit(1);
       }
       if (creds.projects.length === 1) {
@@ -8324,11 +8453,11 @@ program2.command("service").description("Manage the ZeTa agent as a background s
       }
     }
     if (!creds?.token) {
-      console.error("[ZeTa Service] No credentials. Run `zeta <api-key>` first.");
+      console.error(`[ZeTa Service] No credentials. Run \`zeta <api-key> --profile ${profile}\` first.`);
       process.exit(1);
     }
     try {
-      await serviceInstall(projectId);
+      await serviceInstall(projectId, profile);
     } catch (e) {
       console.error(`[ZeTa Service] Install failed: ${e instanceof Error ? e.message : String(e)}`);
       process.exitCode = 1;
@@ -8437,9 +8566,10 @@ program2.command("run_postman").description("Run a local Postman collection and
       process.exitCode = 1;
   }
 });
-program2.argument("[apiKey]", "API key from the ZeTa dashboard \u2014 saves credentials and exits (run `zeta -p <id>` afterwards)").option("-p, --project <id>", "ZeTa Project ID").option("-t, --token <token>", "ZeTa Agent Token (starts with zeta_)").option("-a, --api <url>", "ZeTa API base URL", process.env.ZETA_API_URL ?? "").action(async (apiKey, opts) => {
+program2.argument("[apiKey]", "API key from the ZeTa dashboard \u2014 saves credentials and exits (run `zeta -p <id>` afterwards)").option("-p, --project <id>", "ZeTa Project ID").option("-t, --token <token>", "ZeTa Agent Token (starts with zeta_)").option("-a, --api <url>", "ZeTa API base URL", process.env.ZETA_API_URL ?? "").option("--profile <name>", 'Credential profile to use/save (default: $ZETA_PROFILE or "default")').action(async (apiKey, opts) => {
+  const profile = resolveProfile(opts.profile);
   if (apiKey) {
-    const api2 = opts.api || API_URL;
+    const api2 = resolveApiUrl(opts.api || void 0, profile);
     const token2 = apiKey;
     console.log("[ZeTa] Validating token...");
     const headers = { Authorization: `Bearer ${token2}` };
@@ -8474,9 +8604,9 @@ program2.argument("[apiKey]", "API key from the ZeTa dashboard \u2014 saves cred
       }
     } catch {
     }
-    await saveCreds({ apiUrl: api2, token: token2, workspaceId, workspaceName, projects });
+    await saveCreds({ apiUrl: api2, token: token2, workspaceId, workspaceName, projects }, profile);
     console.log(`
-[ZeTa] \u2713 API key saved for "${workspaceName ?? workspaceId}"`);
+[ZeTa] \u2713 API key saved for "${workspaceName ?? workspaceId}" (profile: ${profile})`);
     console.log(`[ZeTa] Credentials stored: ${CREDS_FILE}`);
     if (projects.length > 0) {
       console.log(`
@@ -8486,16 +8616,16 @@ Projects (${projects.length}):`);
     const serviceProject = projects[0];
     if (serviceProject) {
       console.log(`
-[ZeTa] Installing background service for "${serviceProject.name}"...`);
+[ZeTa] Installing background service for "${serviceProject.name}" (profile: ${profile})...`);
       try {
-        await serviceInstall(serviceProject.id);
+        await serviceInstall(serviceProject.id, profile);
         if (projects.length > 1) {
           console.log(`[ZeTa] \u2139  Multiple projects found. Service running for "${serviceProject.name}".`);
           console.log(`[ZeTa]    To switch: zeta service uninstall && zeta service install -p <project-id>`);
         }
       } catch (e) {
         console.warn(`[ZeTa] \u26A0  Background service install failed: ${e instanceof Error ? e.message : String(e)}`);
-        console.warn(`[ZeTa]    Start manually: zeta -p ${serviceProject.id}`);
+        console.warn(`[ZeTa]    Start manually: zeta --profile ${profile} -p ${serviceProject.id}`);
       }
     } else {
       console.log(`
@@ -8504,13 +8634,13 @@ Projects (${projects.length}):`);
     }
     return;
   }
-  const creds = await loadCreds();
+  const creds = await loadCreds(profile);
   const token = opts.token ?? creds?.token;
-  const api = opts.api || creds?.apiUrl || API_URL;
+  const api = resolveApiUrl(opts.api || creds?.apiUrl || void 0, profile);
   const project = opts.project;
   if (!token) {
     console.error("[ZeTa Agent] No API key found.");
-    console.error("  Configure once: zeta <your-api-key>");
+    console.error(`  Configure once: zeta <your-api-key>${profile !== "default" ? ` --profile ${profile}` : ""}`);
     console.error("  Or pass inline: zeta -t <token> -p <project-id>");
     process.exit(1);
   }