Add waitUntilRunning and use before SSH

Introduce waitUntilRunning helper that polls getSandbox until the
sandbox status is RUNNING (default 60s timeout, 2s interval).
Invoke it in create and start before connecting via SSH.
Also unify auth token usage in start to pass it to the helper

Author: tsirysndr

apps/cli/src/cmd/create.ts

@@ -5,6 +5,7 @@ import type { Sandbox } from "../types/sandbox";
 import connectToSandbox from "./ssh";
 import { c } from "../theme";
 import { expandRepo } from "../lib/expandRepo";
+import waitUntilRunning from "../lib/waitUntilRunning";
 
 async function createSandbox(
   name: string,
@@ -52,6 +53,7 @@ async function createSandbox(
       );
       return;
     }
+    await waitUntilRunning(sandbox.data.name, token);
     await connectToSandbox(sandbox.data.name);
   } catch (error) {
     consola.error(`Failed to create sandbox: ${error}`);

apps/cli/src/cmd/start.ts

@@ -5,6 +5,7 @@ import { client } from "../client";
 import { env } from "../lib/env";
 import connectToSandbox from "./ssh";
 import { expandRepo } from "../lib/expandRepo";
+import waitUntilRunning from "../lib/waitUntilRunning";
 
 async function start(
   name: string,
@@ -14,6 +15,8 @@ async function start(
   if (repo) repo = expandRepo(repo);
 
   try {
+    const authToken = env.POCKETENV_TOKEN || token;
+
     await client.post(
       "/xrpc/io.pocketenv.sandbox.startSandbox",
       {
@@ -24,12 +27,13 @@ async function start(
           id: name,
         },
         headers: {
-          Authorization: `Bearer ${env.POCKETENV_TOKEN || token}`,
+          Authorization: `Bearer ${authToken}`,
         },
       },
     );
 
     if (ssh) {
+      await waitUntilRunning(name, authToken);
       await connectToSandbox(name);
       return;
     }

apps/cli/src/lib/waitUntilRunning.ts

@@ -0,0 +1,27 @@
+import { client } from "../client";
+import type { Sandbox } from "../types/sandbox";
+
+async function waitUntilRunning(
+  name: string,
+  authToken: string,
+  timeoutMs = 60_000,
+  intervalMs = 2_000,
+): Promise<void> {
+  const deadline = Date.now() + timeoutMs;
+  while (Date.now() < deadline) {
+    const response = await client.get<{ sandbox: Sandbox | null }>(
+      "/xrpc/io.pocketenv.sandbox.getSandbox",
+      {
+        params: { id: name },
+        headers: { Authorization: `Bearer ${authToken}` },
+      },
+    );
+    if (response.data.sandbox?.status === "RUNNING") return;
+    await new Promise((resolve) => setTimeout(resolve, intervalMs));
+  }
+  throw new Error(
+    `Sandbox ${name} did not reach RUNNING state within ${timeoutMs / 1000}s`,
+  );
+}
+
+export default waitUntilRunning;