@blaxel/core 0.2.71-dev.101 → 0.2.71-dev.106

package/dist/esm/sandbox/sandbox.js

@@ -21,6 +21,8 @@ export class SandboxInstance {
     codegen;
     system;
     drives;
+    /* eslint-disable @typescript-eslint/no-explicit-any */
+    h2Session;
     constructor(sandbox) {
         this.sandbox = sandbox;
         this.process = new SandboxProcess(sandbox);
@@ -31,6 +33,7 @@ export class SandboxInstance {
         this.codegen = new SandboxCodegen(sandbox);
         this.system = new SandboxSystem(sandbox);
         this.drives = new SandboxDrive(sandbox);
+        this.h2Session = null;
     }
     get metadata() {
         return this.sandbox.metadata;
@@ -47,6 +50,27 @@ export class SandboxInstance {
     get lastUsedAt() {
         return this.sandbox.lastUsedAt;
     }
+    /**
+     * Warm and attach an H2 session based on the sandbox's region.
+     * Shared by create(), get(), list(), and update helpers.
+     */
+    static async attachH2Session(instance) {
+        const region = instance.spec?.region;
+        if (!region)
+            return instance;
+        const edgeSuffix = settings.env === "prod" ? "bl.run" : "runv2.blaxel.dev";
+        const edgeDomain = `any.${region}.${edgeSuffix}`;
+        try {
+            const { h2Pool } = await import("../common/h2pool.js");
+            const h2Session = await h2Pool.get(edgeDomain);
+            instance.h2Session = h2Session;
+            instance.sandbox.h2Session = h2Session;
+        }
+        catch {
+            // H2 warming is best-effort; fall back to regular fetch
+        }
+        return instance;
+    }
     get expiresIn() {
         return this.sandbox.expiresIn;
     }
@@ -128,11 +152,24 @@ export class SandboxInstance {
         }
         sandbox.spec.runtime.image = sandbox.spec.runtime.image || defaultImage;
         sandbox.spec.runtime.memory = sandbox.spec.runtime.memory || defaultMemory;
-        const { data } = await createSandbox({
-            body: sandbox,
-            throwOnError: true,
-        });
-        const instance = new SandboxInstance(data);
+        const edgeSuffix = settings.env === "prod" ? "bl.run" : "runv2.blaxel.dev";
+        const edgeDomain = sandbox.spec?.region ? `any.${sandbox.spec.region}.${edgeSuffix}` : null;
+        // Kick off warming so h2Pool.get() can join it during the API call
+        if (edgeDomain) {
+            import("../common/h2pool.js").then(({ h2Pool }) => h2Pool.warm(edgeDomain)).catch(() => { });
+        }
+        const [{ data }, h2Session] = await Promise.all([
+            createSandbox({
+                body: sandbox,
+                throwOnError: true,
+            }),
+            edgeDomain ? import("../common/h2pool.js").then(({ h2Pool }) => h2Pool.get(edgeDomain)).catch(() => null) : Promise.resolve(null),
+        ]);
+        // Inject the H2 session into the config so subsystems can use it
+        const config = { ...data, h2Session };
+        const instance = new SandboxInstance(config);
+        instance.h2Session = h2Session;
+        // Note: H2 session already attached via Promise.all above, no need for attachH2Session()
         // TODO remove this part once we have a better way to handle this
         if (safe) {
             try {
@@ -149,11 +186,13 @@ export class SandboxInstance {
             },
             throwOnError: true,
         });
-        return new SandboxInstance(data);
+        const instance = new SandboxInstance(data);
+        return SandboxInstance.attachH2Session(instance);
     }
     static async list() {
         const { data } = await listSandboxes({ throwOnError: true });
-        return data.map((sandbox) => new SandboxInstance(sandbox));
+        const instances = data.map((sandbox) => new SandboxInstance(sandbox));
+        return Promise.all(instances.map((instance) => SandboxInstance.attachH2Session(instance)));
     }
     static async delete(sandboxName) {
         const { data } = await deleteSandbox({
@@ -165,6 +204,8 @@ export class SandboxInstance {
         return data;
     }
     async delete() {
+        // Don't close the H2 session — it's shared via h2Pool
+        this.h2Session = null;
         return await SandboxInstance.delete(this.metadata.name);
     }
     static async updateMetadata(sandboxName, metadata) {
@@ -176,7 +217,7 @@ export class SandboxInstance {
             throwOnError: true,
         });
         const instance = new SandboxInstance(data);
-        return instance;
+        return SandboxInstance.attachH2Session(instance);
     }
     static async updateTtl(sandboxName, ttl) {
         const sandbox = await SandboxInstance.get(sandboxName);
@@ -187,7 +228,7 @@ export class SandboxInstance {
             throwOnError: true,
         });
         const instance = new SandboxInstance(data);
-        return instance;
+        return SandboxInstance.attachH2Session(instance);
     }
     static async updateLifecycle(sandboxName, lifecycle) {
         const sandbox = await SandboxInstance.get(sandboxName);
@@ -198,7 +239,7 @@ export class SandboxInstance {
             throwOnError: true,
         });
         const instance = new SandboxInstance(data);
-        return instance;
+        return SandboxInstance.attachH2Session(instance);
     }
     static async createIfNotExists(sandbox) {
         try {