fix(sandbox): hydrate exposed ports in Sandbox.get() so domain() works after reconnect

Author: jamie-at-bunny

.changeset/sandbox-get-hydrates-ports.md

@@ -0,0 +1,5 @@
+---
+"@bunny.net/sandbox": patch
+---
+
+fix(sandbox): Sandbox.get() recovers exposed port mappings from CDN endpoints so domain() works after reconnect

packages/sandbox/src/provision.ts

@@ -20,7 +20,11 @@ type McClient = ReturnType<typeof createMcClient>;
 interface AppContainer {
   id?: string;
   environmentVariables?: Array<{ name?: string; value?: string }>;
-  endpoints?: Array<{ type?: string; publicHost?: string }>;
+  endpoints?: Array<{
+    type?: string;
+    publicHost?: string;
+    portMappings?: Array<{ containerPort?: number }>;
+  }>;
 }
 
 interface App {
@@ -101,6 +105,20 @@ export async function setContainerEnv(
   }
 }
 
+/** Map each CDN-exposed container port to its public host. */
+export function extractCdnPorts(app: App): Record<number, string> {
+  const ports: Record<number, string> = {};
+  for (const ct of app.containerTemplates ?? []) {
+    for (const ep of ct.endpoints ?? []) {
+      if (ep.type !== "cdn" || !ep.publicHost) continue;
+      for (const pm of ep.portMappings ?? []) {
+        if (pm.containerPort != null) ports[pm.containerPort] = ep.publicHost;
+      }
+    }
+  }
+  return ports;
+}
+
 export function firstContainerId(app: App): string | null {
   return app.containerTemplates?.[0]?.id ?? null;
 }

packages/sandbox/src/sandbox.test.ts

@@ -2,6 +2,7 @@ import { describe, expect, test } from "bun:test";
 import {
   extractAgentToken,
   extractAnycastHost,
+  extractCdnPorts,
   firstContainerId,
   splitHost,
 } from "./provision.ts";
@@ -82,7 +83,20 @@ describe("app extraction", () => {
       {
         id: "ct-1",
         environmentVariables: [{ name: "AGENT_TOKEN", value: "secret" }],
-        endpoints: [{ type: "anycast", publicHost: "1.2.3.4:8023" }],
+        endpoints: [
+          { type: "anycast", publicHost: "1.2.3.4:8023" },
+          {
+            type: "cdn",
+            publicHost: "app-3000.b-cdn.net",
+            portMappings: [{ containerPort: 3000 }],
+          },
+          {
+            type: "cdn",
+            publicHost: "app-8080.b-cdn.net",
+            portMappings: [{ containerPort: 8080 }],
+          },
+          { type: "cdn", portMappings: [{ containerPort: 9999 }] },
+        ],
       },
     ],
   };
@@ -96,9 +110,19 @@ describe("app extraction", () => {
   test("returns the first container id", () => {
     expect(firstContainerId(app)).toBe("ct-1");
   });
+  test("maps CDN-exposed ports to their public hosts", () => {
+    expect(extractCdnPorts(app)).toEqual({
+      3000: "app-3000.b-cdn.net",
+      8080: "app-8080.b-cdn.net",
+    });
+  });
+  test("skips CDN endpoints whose public host is still provisioning", () => {
+    expect(extractCdnPorts(app)[9999]).toBeUndefined();
+  });
   test("returns null when fields are absent", () => {
     expect(extractAnycastHost({})).toBeNull();
     expect(extractAgentToken({})).toBeNull();
     expect(firstContainerId({})).toBeNull();
+    expect(extractCdnPorts({})).toEqual({});
   });
 });

packages/sandbox/src/sandbox.ts

@@ -8,6 +8,7 @@ import {
   deleteApp,
   extractAgentToken,
   extractAnycastHost,
+  extractCdnPorts,
   extractEnv,
   firstContainerId,
   getApp,
@@ -131,7 +132,7 @@ export class Sandbox {
     return sandbox;
   }
 
-  /** Retrieve an existing sandbox by app ID, recovering its connection details. */
+  /** Retrieve an existing sandbox by app ID, recovering its connection details and exposed ports. */
   static async get(options: GetOptions): Promise<Sandbox> {
     const client = mcClient(options);
     const app = await getApp(client, options.appId);
@@ -146,7 +147,13 @@ export class Sandbox {
     const name = (app as { name?: string }).name ?? options.appId;
 
     return new Sandbox(
-      { appId: options.appId, name, agentToken, sshHost },
+      {
+        appId: options.appId,
+        name,
+        agentToken,
+        sshHost,
+        ports: extractCdnPorts(app),
+      },
       () => client,
       transportFor(sshHost, agentToken),
     );