nia-wizard 0.1.17 → 0.1.18

package/dist/bin.js

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

package/dist/{chunk-XF23S5X2.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() {
@@ -3889,4 +3905,4 @@ export {
   runSkillAdd,
   printAgentGuide
 };
-//# sourceMappingURL=chunk-XF23S5X2.js.map
+//# sourceMappingURL=chunk-KAGHMAER.js.map