@runuai/host 0.1.0 → 0.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runuai/host",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Uai host — runs ephemeral AI coding tasks in Docker on a machine you control.",
   "license": "MIT",
   "author": "Diogo Perillo <diogo.perillo@gmail.com>",

package/src/cli.ts

@@ -21,8 +21,9 @@ import "./load-env";
 
 import { spawn, spawnSync } from "node:child_process";
 import { createHash, randomBytes } from "node:crypto";
-import { existsSync, readFileSync, writeFileSync } from "node:fs";
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { hostname } from "node:os";
+import { dirname } from "node:path";
 
 import { ulid } from "ulid";
 
@@ -298,9 +299,28 @@ async function cmdSetup(rest: string[]): Promise<void> {
     return;
   }
 
-  upsertEnvLocal("UAI_HOST_ID", hostId);
-  upsertEnvLocal("UAI_HOST_TOKEN", secret);
-  upsertEnvLocal("UAI_CLOUD_URL", cloud);
+  // The host is already created on the cloud and the token is spent. If the
+  // local write fails, don't lose the secret — print it so it can be saved by
+  // hand rather than forcing the user to mint a fresh token.
+  try {
+    upsertEnvLocal("UAI_HOST_ID", hostId);
+    upsertEnvLocal("UAI_HOST_TOKEN", secret);
+    upsertEnvLocal("UAI_CLOUD_URL", cloud);
+  } catch (err) {
+    console.error(
+      red(
+        `enrolled, but could not write ${envLocalPath()}: ${err instanceof Error ? err.message : String(err)}`,
+      ),
+    );
+    console.error(
+      dim("the host exists on the cloud — save these into that file manually, then `uai-host start`:"),
+    );
+    console.error(`  UAI_HOST_ID=${hostId}`);
+    console.error(`  UAI_HOST_TOKEN=${secret}`);
+    console.error(`  UAI_CLOUD_URL=${cloud}`);
+    process.exitCode = 1;
+    return;
+  }
   console.log(green("✓ enrolled") + dim(` — config written to ${envLocalPath()}`));
   console.log("");
   console.log("Start the host:");
@@ -326,6 +346,10 @@ async function cmdPair(token: string | undefined): Promise<void> {
 
 function upsertEnvLocal(key: string, value: string): void {
   const path = envLocalPath();
+  // UAI_HOME (~/.uai on a fresh install) may not exist yet — create it before
+  // the first write, or writeFileSync ENOENTs (which on `setup` would orphan a
+  // just-redeemed host + burn the one-time token).
+  mkdirSync(dirname(path), { recursive: true });
   const existing = existsSync(path) ? readFileSync(path, "utf8") : "";
   const kept = existing
     .split("\n")