nia-wizard 0.1.16 → 0.1.18

package/README.md

@@ -11,6 +11,7 @@ curl -fsSL https://app.trynia.ai/cli | sh
 ```
 
 This will:
+
 1. Install Node.js if needed
 2. Run the wizard which:
    - Opens your browser for authentication
@@ -63,7 +64,10 @@ npx nia-wizard
 # Then select "Install Nia Skill"
 ```
 
-This runs `npx skills add nozomio-labs/nia-skill` and stores your API key at `~/.config/nia/api_key`.
+The wizard stores your API key at `~/.config/nia/api_key`.
+
+- Select `Install Nia Skill` to use the existing `skills` CLI flow.
+- Select `Install Nia CLI` to auto-install `nia-cli`, run `nia auth login --api-key ...`, and then run `nia skill`.
 
 For deterministic agent/CI usage:
 
@@ -103,19 +107,19 @@ npx nia-wizard agent-guide
 
 ## Options
 
-| Option | Description |
-|--------|-------------|
-| `--local` | Use local mode (runs on your machine, requires pipx) |
-| `--remote` | Use remote mode (connects to Nia cloud) |
-| `--debug` | Enable debug logging |
-| `--ci` | CI mode (skip prompts, use defaults) |
-| `--api-key`, `-k` | Nia API key |
-| `--non-interactive` | Fail fast instead of waiting for prompts (skill command) |
-| `--target` | Target coding agent for skill installation (skill command) |
-| `--all-agents` | Install to all detected agents in global scope (skill command) |
-| `--global` | Install to global user skills directories (skill command) |
-| `--source` | Skill source path/repo (skill command) |
-| `--json` | Print machine-readable result (skill command) |
+| Option              | Description                                                    |
+| ------------------- | -------------------------------------------------------------- |
+| `--local`           | Use local mode (runs on your machine, requires pipx)           |
+| `--remote`          | Use remote mode (connects to Nia cloud)                        |
+| `--debug`           | Enable debug logging                                           |
+| `--ci`              | CI mode (skip prompts, use defaults)                           |
+| `--api-key`, `-k`   | Nia API key                                                    |
+| `--non-interactive` | Fail fast instead of waiting for prompts (skill command)       |
+| `--target`          | Target coding agent for skill installation (skill command)     |
+| `--all-agents`      | Install to all detected agents in global scope (skill command) |
+| `--global`          | Install to global user skills directories (skill command)      |
+| `--source`          | Skill source path/repo (skill command)                         |
+| `--json`            | Print machine-readable result (skill command)                  |
 
 ## Authentication
 
@@ -135,6 +139,7 @@ This is the fastest way to get started - no manual copying required!
 ### 2. Manual API Key Entry
 
 If you prefer, you can:
+
 - Get your API key from [app.trynia.ai](https://app.trynia.ai)
 - Pass it directly: `npx @nia/wizard nk_xxx`
 - Or select "Enter API key manually" when prompted
@@ -143,51 +148,51 @@ If you prefer, you can:
 
 ### Full Support (Local + Remote)
 
-| Agent | Config Location |
-|-------|-----------------|
-| Cursor | `~/.cursor/mcp.json` |
-| VS Code | `~/.vscode/mcp.json` |
-| Windsurf | `~/.codeium/windsurf/mcp_config.json` |
-| Cline | `~/.cline/mcp_settings.json` |
-| Continue | `~/.continue/config.json` |
-| Antigravity | `~/.gemini/antigravity/mcp_config.json` |
-| Trae | `~/Library/Application Support/Trae/User/mcp.json` |
-| Roo Code | `~/.roo-code/mcp.json` |
-| Kilo Code | `~/.kilocode/mcp.json` |
-| Gemini CLI | `~/.gemini/settings.json` |
-| OpenCode | `~/.opencode/config.json` |
-| Qodo Gen | `~/.qodo/mcp.json` |
-| Qwen Coder | `~/.qwen/settings.json` |
-| Visual Studio | `~/.vs/mcp.json` |
-| Crush | `~/.crush/config.json` |
-| Copilot Agent | `.github/copilot-mcp.json` (per-repo) |
-| Copilot CLI | `~/.copilot/mcp-config.json` |
-| Factory | CLI: `droid mcp add` |
+| Agent         | Config Location                                    |
+| ------------- | -------------------------------------------------- |
+| Cursor        | `~/.cursor/mcp.json`                               |
+| VS Code       | `~/.vscode/mcp.json`                               |
+| Windsurf      | `~/.codeium/windsurf/mcp_config.json`              |
+| Cline         | `~/.cline/mcp_settings.json`                       |
+| Continue      | `~/.continue/config.json`                          |
+| Antigravity   | `~/.gemini/antigravity/mcp_config.json`            |
+| Trae          | `~/Library/Application Support/Trae/User/mcp.json` |
+| Roo Code      | `~/.roo-code/mcp.json`                             |
+| Kilo Code     | `~/.kilocode/mcp.json`                             |
+| Gemini CLI    | `~/.gemini/settings.json`                          |
+| OpenCode      | `~/.opencode/config.json`                          |
+| Qodo Gen      | `~/.qodo/mcp.json`                                 |
+| Qwen Coder    | `~/.qwen/settings.json`                            |
+| Visual Studio | `~/.vs/mcp.json`                                   |
+| Crush         | `~/.crush/config.json`                             |
+| Copilot Agent | `.github/copilot-mcp.json` (per-repo)              |
+| Copilot CLI   | `~/.copilot/mcp-config.json`                       |
+| Factory       | CLI: `droid mcp add`                               |
 
 ### Remote Only
 
-| Agent | Notes |
-|-------|-------|
-| Claude Code | CLI: `claude mcp add` |
-| Amp | CLI: `amp mcp add` |
-| Vibe | TOML config at `~/.vibe/config.toml` |
+| Agent       | Notes                                |
+| ----------- | ------------------------------------ |
+| Claude Code | CLI: `claude mcp add`                |
+| Amp         | CLI: `amp mcp add`                   |
+| Vibe        | TOML config at `~/.vibe/config.toml` |
 
 ### Local Only
 
-| Agent | Config Location |
-|-------|-----------------|
+| Agent          | Config Location                                                   |
+| -------------- | ----------------------------------------------------------------- |
 | Claude Desktop | `~/Library/Application Support/Claude/claude_desktop_config.json` |
-| Codex CLI | CLI: `codex mcp add` |
-| Codex App | `~/.codex/config.toml` |
-| Zed | `~/.config/zed/settings.json` |
-| Augment | VS Code settings |
-| JetBrains | `~/.jetbrains/mcp.json` |
-| Kiro | `~/.kiro/mcp.json` |
-| LM Studio | `~/.lmstudio/mcp.json` |
-| Bolt AI | `~/Library/Application Support/BoltAI/mcp.json` |
-| Perplexity | `~/Library/Application Support/Perplexity/mcp.json` |
-| Warp | `~/.warp/mcp.json` |
-| Amazon Q | `~/.aws/amazonq/mcp.json` |
+| Codex CLI      | CLI: `codex mcp add`                                              |
+| Codex App      | `~/.codex/config.toml`                                            |
+| Zed            | `~/.config/zed/settings.json`                                     |
+| Augment        | VS Code settings                                                  |
+| JetBrains      | `~/.jetbrains/mcp.json`                                           |
+| Kiro           | `~/.kiro/mcp.json`                                                |
+| LM Studio      | `~/.lmstudio/mcp.json`                                            |
+| Bolt AI        | `~/Library/Application Support/BoltAI/mcp.json`                   |
+| Perplexity     | `~/Library/Application Support/Perplexity/mcp.json`               |
+| Warp           | `~/.warp/mcp.json`                                                |
+| Amazon Q       | `~/.aws/amazonq/mcp.json`                                         |
 
 ### Manual Setup Required
 

package/dist/bin.js

@@ -6,7 +6,7 @@ import {
   runMCPRemove,
   runSkillAdd,
   runWizard
-} from "./chunk-JKF3ZFD5.js";
+} from "./chunk-KAGHMAER.js";
 
 // src/bin.ts
 import yargs from "yargs";

package/dist/{chunk-JKF3ZFD5.js → chunk-KAGHMAER.js}

@@ -15,13 +15,15 @@ function debug(...args) {
 var BACKEND_URL = process.env.NIA_BACKEND_URL || "https://apigcp.trynia.ai";
 var APP_URL = process.env.NIA_APP_URL || "";
 var SESSION_TTL_MS = 15 * 60 * 1e3;
+var FETCH_TIMEOUT_MS = 1e4;
 async function startDeviceSession() {
   debug("Starting device session...");
   const response = await fetch(`${BACKEND_URL}/public/mcp-device/start`, {
     method: "POST",
     headers: {
       "Content-Type": "application/json"
-    }
+    },
+    signal: AbortSignal.timeout(FETCH_TIMEOUT_MS)
   });
   if (!response.ok) {
     const errorText = await response.text().catch(() => "Unknown error");
@@ -62,7 +64,8 @@ async function exchangeForApiKey(session) {
     body: JSON.stringify({
       authorization_session_id: session.authorization_session_id,
       user_code: session.user_code
-    })
+    }),
+    signal: AbortSignal.timeout(FETCH_TIMEOUT_MS)
   });
   if (!response.ok) {
     const errorData = await response.json().catch(() => ({ detail: "Unknown error" }));
@@ -376,68 +379,81 @@ async function runDeviceFlow() {
   console.log("");
   clack_default.log.step(chalk.yellow("Complete these steps in your browser:"));
   console.log("  1. Sign in or create an account");
-  console.log("  2. The CLI will be authorized automatically");
+  console.log("  2. Complete any setup steps shown");
+  console.log("");
+  clack_default.log.message(chalk.dim("The CLI will detect when you're done and continue automatically."));
   console.log("");
   return await waitForAuthorizationAndExchange(session);
 }
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
 async function waitForAuthorizationAndExchange(session) {
+  const spinner = clack_default.spinner();
+  spinner.start("Waiting for browser authorization...");
+  const POLL_INTERVAL_MS = 2e3;
+  const MAX_NETWORK_ERRORS = 5;
+  let consecutiveNetworkErrors = 0;
   while (true) {
     if (!isSessionValid(session)) {
+      spinner.stop("Session expired");
       clack_default.log.error("Session has expired. Please start over.");
       return abort("Session expired", 1);
     }
-    const shouldContinue = await abortIfCancelled(
-      clack_default.confirm({
-        message: "Press Enter once you've signed in (or N to enter key manually)",
-        initialValue: true
-      })
-    );
-    if (!shouldContinue) {
-      return await promptForManualApiKey();
-    }
-    const spinner = clack_default.spinner();
-    spinner.start("Checking authorization...");
     try {
       const apiKey = await exchangeForApiKey(session);
       spinner.stop(chalk.green("\u2713 Authorized!"));
       clack_default.log.success("API key obtained successfully!");
       return apiKey;
     } catch (error) {
-      spinner.stop("Not ready yet");
       if (isDeviceFlowError(error)) {
         switch (error.type) {
           case "not_ready":
-            clack_default.log.warn("Browser authorization not complete yet.");
-            console.log("");
-            clack_default.log.message(
-              chalk.dim("Make sure you have:\n") + "  \u2022 Signed in to your Nia account\n  \u2022 Completed any setup steps in the browser"
-            );
-            console.log("");
+            consecutiveNetworkErrors = 0;
             break;
           case "expired":
+            spinner.stop("Session expired");
             clack_default.log.error("Session has expired.");
             clack_default.log.info("Please run the wizard again to start a new session.");
             return abort("Session expired", 1);
           case "consumed":
+            spinner.stop("Session already used");
             clack_default.log.error("This session was already used.");
             clack_default.log.info("Please run the wizard again to start a new session.");
             return abort("Session already used", 1);
           case "invalid":
+            spinner.stop("Invalid session");
             clack_default.log.error(error.message);
             clack_default.log.info("Please run the wizard again to start a new session.");
             return abort("Invalid session", 1);
+          case "network":
+            consecutiveNetworkErrors++;
+            if (consecutiveNetworkErrors >= MAX_NETWORK_ERRORS) {
+              spinner.stop("Connection failed");
+              clack_default.log.error(`Failed to reach Nia servers after ${MAX_NETWORK_ERRORS} attempts.`);
+              clack_default.log.info("Falling back to manual API key entry.");
+              return await promptForManualApiKey();
+            }
+            break;
           default:
+            spinner.stop("Error");
             clack_default.log.error(error.message);
             clack_default.log.info("Falling back to manual API key entry.");
             return await promptForManualApiKey();
         }
       } else {
-        clack_default.log.error("An unexpected error occurred.");
-        debug(`Exchange error: ${error}`);
-        clack_default.log.info("Falling back to manual API key entry.");
-        return await promptForManualApiKey();
+        consecutiveNetworkErrors++;
+        if (consecutiveNetworkErrors >= MAX_NETWORK_ERRORS) {
+          spinner.stop("Connection failed");
+          clack_default.log.error("Lost connection to Nia servers.");
+          debug(`Exchange error: ${error}`);
+          clack_default.log.info("Falling back to manual API key entry.");
+          return await promptForManualApiKey();
+        }
       }
     }
+    const backoff = consecutiveNetworkErrors > 0 ? POLL_INTERVAL_MS * Math.min(consecutiveNetworkErrors, 4) : POLL_INTERVAL_MS;
+    await sleep(backoff);
   }
 }
 async function promptForManualApiKey() {
@@ -2931,7 +2947,7 @@ async function removeMCPServerFromClientsStep() {
 
 // src/run.ts
 import chalk3 from "chalk";
-import { spawnSync as spawnSync5 } from "child_process";
+import { spawnSync as spawnSync6 } from "child_process";
 
 // src/utils/dependencies.ts
 import { execSync as execSync5, spawnSync as spawnSync4 } from "child_process";
@@ -3147,10 +3163,99 @@ function storeApiKey(apiKey) {
   fs33.writeFileSync(NIA_KEY_PATH, apiKey, { mode: 384 });
 }
 
+// src/utils/nia-cli.ts
+import { spawnSync as spawnSync5 } from "child_process";
+var NIA_CLI_PACKAGE = "nia-cli";
+function npmCommand() {
+  return process.platform === "win32" ? "npm.cmd" : "npm";
+}
+function npxCommand() {
+  return process.platform === "win32" ? "npx.cmd" : "npx";
+}
+function niaCommand() {
+  return process.platform === "win32" ? "nia.cmd" : "nia";
+}
+function isNiaCliInstalled() {
+  const result = spawnSync5(niaCommand(), ["--version"], {
+    stdio: "pipe",
+    encoding: "utf-8",
+    shell: false
+  });
+  debug("nia --version status", result.status);
+  return result.status === 0;
+}
+function ensureNiaCliInstalled() {
+  if (isNiaCliInstalled()) {
+    return true;
+  }
+  clack_default.log.info("Installing nia-cli globally with npm...");
+  const installSpinner = clack_default.spinner();
+  installSpinner.start("Installing nia-cli...");
+  const installResult = spawnSync5(npmCommand(), ["install", "-g", NIA_CLI_PACKAGE], {
+    stdio: "pipe",
+    encoding: "utf-8",
+    shell: false
+  });
+  if (installResult.status !== 0) {
+    installSpinner.stop("Failed to install nia-cli");
+    const installOutput = [installResult.stdout, installResult.stderr].filter(Boolean).join("\n").trim();
+    if (installOutput) {
+      clack_default.log.error(installOutput);
+    }
+    clack_default.log.error("Could not install nia-cli automatically.");
+    clack_default.log.info("Install manually: npm install -g nia-cli");
+    return false;
+  }
+  if (!isNiaCliInstalled()) {
+    installSpinner.stop("Installed nia-cli, but `nia` is not available");
+    clack_default.log.error("The `nia` command is still unavailable in your PATH.");
+    clack_default.log.info("Open a new terminal or run: npm install -g nia-cli");
+    return false;
+  }
+  installSpinner.stop("nia-cli installed!");
+  return true;
+}
+function runNiaSkill() {
+  const runResult = spawnSync5(niaCommand(), ["skill"], {
+    stdio: "inherit",
+    shell: false
+  });
+  if (runResult.status === 0) {
+    return true;
+  }
+  debug("nia skill failed, falling back to npx nia-cli skill", runResult.status);
+  clack_default.log.warn("`nia skill` failed. Trying `npx -y nia-cli skill`...");
+  const fallbackResult = spawnSync5(npxCommand(), ["-y", NIA_CLI_PACKAGE, "skill"], {
+    stdio: "inherit",
+    shell: false
+  });
+  return fallbackResult.status === 0;
+}
+function runNiaAuthLogin(apiKey) {
+  const authResult = spawnSync5(niaCommand(), ["auth", "login", "--api-key", apiKey], {
+    stdio: "inherit",
+    shell: false
+  });
+  if (authResult.status === 0) {
+    return true;
+  }
+  debug("nia auth login failed, falling back to npx nia-cli auth login", authResult.status);
+  clack_default.log.warn("`nia auth login` failed. Trying `npx -y nia-cli auth login`...");
+  const fallbackResult = spawnSync5(
+    npxCommand(),
+    ["-y", NIA_CLI_PACKAGE, "auth", "login", "--api-key", apiKey],
+    {
+      stdio: "inherit",
+      shell: false
+    }
+  );
+  return fallbackResult.status === 0;
+}
+
 // src/run.ts
 async function runAddMcpInstall(apiKey) {
   clack_default.log.info("Launching add-mcp installer...\n");
-  const result = spawnSync5(
+  const result = spawnSync6(
     "npx",
     [
       "add-mcp",
@@ -3170,12 +3275,24 @@ async function runAddMcpInstall(apiKey) {
 }
 async function runSkillsInstall() {
   clack_default.log.info("Launching Nia skill installer...\n");
-  const result = spawnSync5("npx", ["skills", "add", "nozomio-labs/nia-skill"], {
+  const result = spawnSync6("npx", ["skills", "add", "nozomio-labs/nia-skill"], {
     stdio: "inherit",
     shell: true
   });
   return result.status === 0;
 }
+async function runNiaCliSkillInstall(apiKey) {
+  clack_default.log.info("Launching Nia CLI skill installer...\n");
+  if (!ensureNiaCliInstalled()) {
+    return false;
+  }
+  clack_default.log.info("Configuring nia-cli authentication...");
+  if (!runNiaAuthLogin(apiKey)) {
+    clack_default.log.warn("nia-cli authentication failed");
+    return false;
+  }
+  return runNiaSkill();
+}
 async function runManualMode() {
   const allClients = getAllClients();
   const selectedName = await abortIfCancelled(
@@ -3270,6 +3387,11 @@ async function runWizard(options) {
     clack_default.multiselect({
       message: "What would you like to do? (space to select, enter to confirm)",
       options: [
+        {
+          value: "nia-cli",
+          label: "Install Nia CLI",
+          hint: "Installs nia CLI + skill"
+        },
         {
           value: "add-mcp",
           label: "Install via add-mcp",
@@ -3304,6 +3426,7 @@ async function runWizard(options) {
   let installedAddMcp = false;
   let installedMcp = false;
   let installedSkills = false;
+  let installedNiaCliSkill = false;
   if (actions.includes("add-mcp")) {
     console.log("");
     const success = await runAddMcpInstall(apiKey);
@@ -3365,10 +3488,26 @@ async function runWizard(options) {
       clack_default.log.warn("Skills installation may have failed");
     }
   }
-  if (installedAddMcp || installedMcp || installedSkills) {
+  if (actions.includes("nia-cli")) {
+    console.log("");
+    const success = await runNiaCliSkillInstall(apiKey);
+    if (success) {
+      clack_default.log.success("Nia CLI skill installed!");
+      clack_default.log.message(chalk3.dim("Run `nia skill` to manage your Nia CLI skill."));
+      installedNiaCliSkill = true;
+    } else {
+      clack_default.log.warn("Nia CLI skill installation may have failed");
+    }
+  }
+  if (installedAddMcp || installedMcp || installedSkills || installedNiaCliSkill) {
+    const niaSkillManagementHint = installedNiaCliSkill ? `
+${chalk3.cyan("Nia CLI skill management:")} ${chalk3.yellow("Run `nia skill` any time to manage your skill.")}
+` : "";
     const outroMessage = `
 ${chalk3.green("\u2713 Nia installed!")}
 
+${niaSkillManagementHint}
+
 ${chalk3.cyan("Get started:")}
   \u2022 Browse pre-indexed sources: ${chalk3.cyan("https://app.trynia.ai/explore")}
   \u2022 Or index your own repos, docs, and papers
@@ -3456,7 +3595,7 @@ async function runMCPRemove(options = {}) {
 
 // src/skill.ts
 import chalk5 from "chalk";
-import { spawnSync as spawnSync6 } from "child_process";
+import { spawnSync as spawnSync7 } from "child_process";
 var DEFAULT_SKILL_SOURCE = "nozomio-labs/nia-skill";
 var NON_INTERACTIVE_TIMEOUT_MS = 3e4;
 async function runSkillAdd(options) {
@@ -3528,7 +3667,7 @@ async function runSkillAdd(options) {
   }
 }
 function detectSkillsCapabilities() {
-  const result = spawnSync6("npx", ["skills", "add", "--help"], {
+  const result = spawnSync7("npx", ["skills", "add", "--help"], {
     shell: false,
     stdio: "pipe",
     encoding: "utf-8",
@@ -3628,7 +3767,7 @@ function runSkillsInstall2(params) {
   }
   const command = `npx ${args.join(" ")}`;
   debug("running", command);
-  const result = spawnSync6("npx", args, {
+  const result = spawnSync7("npx", args, {
     shell: false,
     stdio: params.nonInteractive ? "pipe" : "inherit",
     encoding: params.nonInteractive ? "utf-8" : void 0,
@@ -3766,4 +3905,4 @@ export {
   runSkillAdd,
   printAgentGuide
 };
-//# sourceMappingURL=chunk-JKF3ZFD5.js.map
+//# sourceMappingURL=chunk-KAGHMAER.js.map