@pocketenv/cli 0.3.2 → 0.3.3

package/dist/index.js

@@ -22,7 +22,7 @@ import relativeTime from 'dayjs/plugin/relativeTime.js';
 import { password, editor, input } from '@inquirer/prompts';
 import sodium from 'libsodium-wrappers';
 
-var version = "0.3.2";
+var version = "0.3.3";
 
 async function getAccessToken() {
   const tokenPath = path.join(os.homedir(), ".pocketenv", "token.json");
@@ -418,10 +418,29 @@ function expandRepo(repo) {
   return repo;
 }
 
+async function waitUntilRunning(name, authToken, timeoutMs = 6e4, intervalMs = 2e3) {
+  const deadline = Date.now() + timeoutMs;
+  while (Date.now() < deadline) {
+    const response = await client.get(
+      "/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 / 1e3}s`
+  );
+}
+
 async function start(name, { ssh: ssh$1, repo }) {
   const token = await getAccessToken();
   if (repo) repo = expandRepo(repo);
   try {
+    const authToken = env$1.POCKETENV_TOKEN || token;
     await client.post(
       "/xrpc/io.pocketenv.sandbox.startSandbox",
       {
@@ -432,11 +451,12 @@ async function start(name, { ssh: ssh$1, repo }) {
           id: name
         },
         headers: {
-          Authorization: `Bearer ${env$1.POCKETENV_TOKEN || token}`
+          Authorization: `Bearer ${authToken}`
         }
       }
     );
     if (ssh$1) {
+      await waitUntilRunning(name, authToken);
       await ssh(name);
       return;
     }
@@ -619,6 +639,7 @@ async function createSandbox(name, {
       );
       return;
     }
+    await waitUntilRunning(sandbox.data.name, token);
     await ssh(sandbox.data.name);
   } catch (error) {
     consola.error(`Failed to create sandbox: ${error}`);

package/package.json

@@ -4,7 +4,7 @@
   "bin": {
     "pocketenv": "dist/index.js"
   },
-  "version": "0.3.2",
+  "version": "0.3.3",
   "type": "module",
   "keywords": [
     "sandbox",

package/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}`);

package/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;
     }

package/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;