@opengeni/runtime 2.4.2-canary.0 → 2.4.2-canary.2

package/dist/{chunk-72SCUP4B.js → chunk-WATBACLT.js}

@@ -14376,9 +14376,10 @@ function notifyDurableOpOwnershipTransferred(opId) {
 // src/sandbox/provider-command-session.ts
 import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
 var admission = new AsyncLocalStorage2();
+var MAX_PROVIDER_COMMAND_HANDLE = 2147483647;
 function withProviderCommandHandle(handle, fn) {
   if (handle === void 0) return fn();
-  if (!Number.isSafeInteger(handle) || handle <= 0 || handle > 2147483647)
+  if (!Number.isSafeInteger(handle) || handle <= 0 || handle > MAX_PROVIDER_COMMAND_HANDLE)
     throw new Error("Invalid admitted provider command handle");
   return admission.run(handle, fn);
 }
@@ -20393,10 +20394,26 @@ function sameExecution(a, b) {
 function installModalCommandSession(session, control) {
   markTypedExecHandleLoss(session);
   const originalExec = session.execCommand?.bind(session);
+  const originalWrite = session.writeStdin?.bind(session);
   const cancelLegacyStart = session.cancelPendingExecCommand?.bind(session);
   const pendingStarts = /* @__PURE__ */ new Set();
   const entries = /* @__PURE__ */ new Map();
+  const setupHandles = /* @__PURE__ */ new Map();
+  let nextSetupHandle = MAX_PROVIDER_COMMAND_HANDLE + 1;
   const receipts = /* @__PURE__ */ new Map();
+  const setupPage = (raw, handle, sdkHandle) => {
+    const banner = parseExecResponseBanner(raw);
+    if (banner.kind === "exited") {
+      setupHandles.delete(handle);
+      return raw;
+    }
+    if (banner.kind !== "running" || banner.sessionId !== sdkHandle)
+      throw new ModalProcessObservationUnavailableError(handle);
+    return raw.replace(
+      /^Process running with session ID \d+(?=\r?$)/mu,
+      `Process running with session ID ${handle}`
+    );
+  };
   const formatPage = (handle, page, maxOutputTokens) => {
     const result = [
       `Provider output receipt: ${randomUUID5()}`,
@@ -20425,7 +20442,13 @@ function installModalCommandSession(session, control) {
     const handle = admittedProviderCommandHandle();
     if (handle === void 0) {
       if (!originalExec) throw new Error("Modal command requires mutation admission");
-      return originalExec(args);
+      if (!Number.isSafeInteger(nextSetupHandle)) throw new Error("Modal setup handles exhausted");
+      const setupHandle = nextSetupHandle++;
+      const raw = await originalExec(args);
+      const banner = parseExecResponseBanner(raw);
+      if (banner.kind !== "running") return raw;
+      setupHandles.set(setupHandle, banner.sessionId);
+      return setupPage(raw, setupHandle, banner.sessionId);
     }
     if (entries.has(handle)) throw new Error("Modal command handle is already bound");
     const cancellation = new AbortController();
@@ -20442,7 +20465,11 @@ function installModalCommandSession(session, control) {
           cancellation.signal
         );
       } catch {
-        return formatPage(handle, { command: entry.command, chunks: [], exitCode: null });
+        return formatPage(handle, {
+          command: entry.command,
+          chunks: [],
+          exitCode: null
+        });
       }
     } finally {
       pendingStarts.delete(cancellation);
@@ -20460,7 +20487,7 @@ function installModalCommandSession(session, control) {
     return entry ? structuredClone(entry.command) : null;
   };
   session.bindProviderCommand = (handle, command, persistence) => {
-    if (!Number.isSafeInteger(handle) || handle <= 0)
+    if (!Number.isSafeInteger(handle) || handle <= 0 || handle > MAX_PROVIDER_COMMAND_HANDLE)
       throw new Error("Invalid retained Modal command handle");
     const existing = entries.get(handle);
     if (existing && !sameExecution(existing.command, command))
@@ -20481,6 +20508,11 @@ function installModalCommandSession(session, control) {
     receipts.delete(result);
   };
   session.writeStdin = async (args) => {
+    const sdkHandle = setupHandles.get(args.sessionId);
+    if (sdkHandle !== void 0 && originalWrite) {
+      const raw = await originalWrite({ ...args, sessionId: sdkHandle });
+      return setupPage(raw, args.sessionId, sdkHandle);
+    }
     const entry = entries.get(args.sessionId);
     if (!entry?.persistence)
       throw new ModalProcessObservationUnavailableError(args.sessionId, {
@@ -20556,9 +20588,9 @@ function decodePage(prior, input, terminal) {
   };
 }
 var ModalCommandControl = class _ModalCommandControl {
-  constructor(client, sandboxId, root, environment = {}) {
+  constructor(client, sandboxIdentity, root, environment = {}) {
     this.client = client;
-    this.sandboxId = sandboxId;
+    this.sandboxIdentity = sandboxIdentity;
     this.root = root;
     this.environment = environment;
   }
@@ -20567,16 +20599,19 @@ var ModalCommandControl = class _ModalCommandControl {
       throw new Error("Modal command control requires the verified 0.9.0 SDK contract");
     return new _ModalCommandControl(commandControlPlane(client), sandboxId, root, environment);
   }
+  get sandboxId() {
+    return typeof this.sandboxIdentity === "function" ? this.sandboxIdentity() : this.sandboxIdentity;
+  }
   async start(args, signal) {
     signal?.throwIfAborted();
     const workdir = posix2.resolve(this.root, args.workdir ?? this.root);
     if (workdir !== this.root && !workdir.startsWith(`${this.root.replace(/\/$/u, "")}/`))
       throw new Error("Command workdir is outside the sandbox workspace");
-    const task = await this.client.sandboxGetTaskId(
-      { sandboxId: this.sandboxId },
-      signal ? { signal } : void 0
-    );
+    const sandboxId = this.sandboxId;
+    const task = await this.client.sandboxGetTaskId({ sandboxId }, signal ? { signal } : void 0);
     if (!task.taskId || task.taskResult) throw new Error("Modal command task is unavailable");
+    if (sandboxId !== this.sandboxId)
+      throw new Error("Modal sandbox changed during command preparation");
     const login = args.shell ? args.login ?? true : false;
     let command = [args.shell ?? "/bin/sh", login ? "-lc" : "-c", args.cmd];
     if (args.runAs) {
@@ -20623,7 +20658,7 @@ var ModalCommandControl = class _ModalCommandControl {
     if (!result.execId) throw new Error("Modal command start returned no execution identity");
     return {
       kind: "modal-control-v1",
-      sandboxId: this.sandboxId,
+      sandboxId,
       taskId: task.taskId,
       execId: result.execId,
       ...args.tty ? { pty: true } : {},
@@ -21038,7 +21073,11 @@ function installOpenGeniModalSnapshotPolicy(session) {
           profile: mutable.modal.profile,
           ...mutable.modal.logger ? { logger: mutable.modal.logger } : {}
         },
-        mutable.state.sandboxId,
+        () => {
+          const sandboxId = mutable.state?.sandboxId;
+          if (!sandboxId) throw new Error("Modal session sandbox identity is unavailable");
+          return sandboxId;
+        },
         mutable.state.manifest.root,
         () => mutable.state?.environment ?? {}
       )
@@ -36083,4 +36122,4 @@ export {
   collectSandboxEnvironment2 as collectSandboxEnvironment,
   parseExposedPorts2 as parseExposedPorts
 };
-//# sourceMappingURL=chunk-72SCUP4B.js.map
+//# sourceMappingURL=chunk-WATBACLT.js.map