@opengeni/runtime 0.6.0 → 0.7.0

package/src/sandbox/selfhosted/testing.ts

@@ -72,7 +72,10 @@ export class MockAgentResponder implements ControlRpc {
     this.hostname = opts.hostname ?? "mock-machine";
     this.execHandler = opts.exec ?? ((req) => defaultEcho(req, this.hostname));
     for (const [path, content] of Object.entries(opts.files ?? {})) {
-      this.files.set(normalize(path), typeof content === "string" ? encoder.encode(content) : content);
+      this.files.set(
+        normalize(path),
+        typeof content === "string" ? encoder.encode(content) : content,
+      );
     }
   }
 
@@ -87,13 +90,27 @@ export class MockAgentResponder implements ControlRpc {
     return bytes ? decoder.decode(bytes) : undefined;
   }
 
-  async request(subject: string, req: ControlRequest, _opts: { timeoutMs: number }): Promise<ControlResponse> {
+  async request(
+    subject: string,
+    req: ControlRequest,
+    _opts: { timeoutMs: number },
+  ): Promise<ControlResponse> {
     this.requests.push({ subject, req });
     if (!this.online) {
-      return errorResponse(req.requestId, ErrorCode.ERROR_CODE_AGENT_OFFLINE, "the enrolled agent is offline", false);
+      return errorResponse(
+        req.requestId,
+        ErrorCode.ERROR_CODE_AGENT_OFFLINE,
+        "the enrolled agent is offline",
+        false,
+      );
     }
     if (this.draining) {
-      return errorResponse(req.requestId, ErrorCode.ERROR_CODE_DRAINING, "the agent is draining", true);
+      return errorResponse(
+        req.requestId,
+        ErrorCode.ERROR_CODE_DRAINING,
+        "the agent is draining",
+        true,
+      );
     }
     const op = req.op;
     if (!op) {
@@ -101,7 +118,10 @@ export class MockAgentResponder implements ControlRpc {
     }
     switch (op.$case) {
       case "ping":
-        return ok(req.requestId, { $case: "ping", ping: { nonce: op.ping.nonce, agentMonotonicMs: "0" } });
+        return ok(req.requestId, {
+          $case: "ping",
+          ping: { nonce: op.ping.nonce, agentMonotonicMs: "0" },
+        });
       case "exec": {
         const res = await this.execHandler(op.exec);
         return ok(req.requestId, { $case: "exec", exec: res });
@@ -109,7 +129,12 @@ export class MockAgentResponder implements ControlRpc {
       case "fsRead": {
         const bytes = this.files.get(normalize(op.fsRead.path));
         if (!bytes) {
-          return errorResponse(req.requestId, ErrorCode.ERROR_CODE_NOT_FOUND, `no such file: ${op.fsRead.path}`, false);
+          return errorResponse(
+            req.requestId,
+            ErrorCode.ERROR_CODE_NOT_FOUND,
+            `no such file: ${op.fsRead.path}`,
+            false,
+          );
         }
         const res: FsReadResponse = { content: bytes, totalSize: String(bytes.length) };
         return ok(req.requestId, { $case: "fsRead", fsRead: res });
@@ -147,16 +172,16 @@ export class MockAgentResponder implements ControlRpc {
         const bytes = this.files.get(normalize(op.fsStat.path));
         const res: FsStatResponse = bytes
           ? {
-            exists: true,
-            entry: {
-              name: normalize(op.fsStat.path).split("/").pop() ?? "",
-              path: op.fsStat.path,
-              kind: FsEntryKind.FS_ENTRY_KIND_FILE,
-              size: String(bytes.length),
-              modifiedMs: "0",
-              mode: 0o644,
-            },
-          }
+              exists: true,
+              entry: {
+                name: normalize(op.fsStat.path).split("/").pop() ?? "",
+                path: op.fsStat.path,
+                kind: FsEntryKind.FS_ENTRY_KIND_FILE,
+                size: String(bytes.length),
+                modifiedMs: "0",
+                mode: 0o644,
+              },
+            }
           : { exists: false, entry: undefined };
         return ok(req.requestId, { $case: "fsStat", fsStat: res });
       }
@@ -170,7 +195,13 @@ export class MockAgentResponder implements ControlRpc {
         return ok(req.requestId, {
           $case: "desktopEnsure",
           desktopEnsure: {
-            channel: { channelId: "mock-desktop", workspaceId: "", agentId: "", kind: 1, port: 6080 },
+            channel: {
+              channelId: "mock-desktop",
+              workspaceId: "",
+              agentId: "",
+              kind: 1,
+              port: 6080,
+            },
             display: { id: ":99", width: 1024, height: 768, virtual: true },
           },
         });
@@ -188,7 +219,12 @@ export class MockAgentResponder implements ControlRpc {
         });
       }
       default:
-        return errorResponse(req.requestId, ErrorCode.ERROR_CODE_UNSUPPORTED, `mock does not implement ${op.$case}`, false);
+        return errorResponse(
+          req.requestId,
+          ErrorCode.ERROR_CODE_UNSUPPORTED,
+          `mock does not implement ${op.$case}`,
+          false,
+        );
     }
   }
 }
@@ -211,7 +247,12 @@ function ok(requestId: string, result: NonNullable<ControlResponse["result"]>):
   return { requestId, error: undefined, result };
 }
 
-function errorResponse(requestId: string, code: ErrorCode, message: string, retryable: boolean): ControlResponse {
+function errorResponse(
+  requestId: string,
+  code: ErrorCode,
+  message: string,
+  retryable: boolean,
+): ControlResponse {
   const error: AgentError = { code, message, retryable, detail: {} };
   return { requestId, error, result: undefined };
 }

package/src/sandbox/stream-port.ts

@@ -48,7 +48,10 @@ type PortResolvableSession = {
  *  `transport:null` (a value, never a crash) — a headless-only provider or a
  *  transient tunnel failure must not fail the whole handshake. */
 export class StreamPortUnavailableError extends Error {
-  constructor(message: string, readonly cause?: unknown) {
+  constructor(
+    message: string,
+    readonly cause?: unknown,
+  ) {
     super(message);
     this.name = "StreamPortUnavailableError";
   }
@@ -103,7 +106,11 @@ const DEFAULT_RESOLUTION: [number, number] = [1280, 800];
  * the holder + validated at the in-box websockify edge).
  */
 export function buildStreamUrl(endpoint: ExposedPortEndpoint): string {
-  if (typeof endpoint.host !== "string" || endpoint.host.length === 0 || typeof endpoint.port !== "number") {
+  if (
+    typeof endpoint.host !== "string" ||
+    endpoint.host.length === 0 ||
+    typeof endpoint.port !== "number"
+  ) {
     throw new StreamPortUnavailableError(
       `provider returned a malformed exposed-port endpoint (host=${String(endpoint.host)}, port=${String(endpoint.port)})`,
     );
@@ -112,13 +119,18 @@ export function buildStreamUrl(endpoint: ExposedPortEndpoint): string {
   const scheme = tls ? "wss" : "ws";
   const defaultPort = tls ? 443 : 80;
   // Bracket a bare IPv6 host (urlForExposedPort parity).
-  const host = endpoint.host.includes(":") && !endpoint.host.startsWith("[") ? `[${endpoint.host}]` : endpoint.host;
+  const host =
+    endpoint.host.includes(":") && !endpoint.host.startsWith("[")
+      ? `[${endpoint.host}]`
+      : endpoint.host;
   // The path: default the root `/` (Modal/Daytona/Blaxel edge), or the
   // provider-supplied path (the selfhosted relay's `/stream` route, M8b). Always
   // leading-slash-normalized.
-  const rawPath = typeof endpoint.path === "string" && endpoint.path.length > 0 ? endpoint.path : "/";
+  const rawPath =
+    typeof endpoint.path === "string" && endpoint.path.length > 0 ? endpoint.path : "/";
   const path = rawPath.startsWith("/") ? rawPath : `/${rawPath}`;
-  const origin = endpoint.port === defaultPort ? `${scheme}://${host}` : `${scheme}://${host}:${endpoint.port}`;
+  const origin =
+    endpoint.port === defaultPort ? `${scheme}://${host}` : `${scheme}://${host}:${endpoint.port}`;
   const authority = `${origin}${path}`;
   const query = endpoint.query ?? "";
   return query ? `${authority}?${query}` : authority;

package/src/sandbox/stream-token.ts

@@ -54,7 +54,10 @@ export type MintStreamTokenInput = {
  * envelope (`ogs_` prefix). The token is RECORDED against the holder row by the
  * caller and is NEVER appended to the data-plane URL as a query param.
  */
-export async function mintStreamToken(secret: string, input: MintStreamTokenInput): Promise<string> {
+export async function mintStreamToken(
+  secret: string,
+  input: MintStreamTokenInput,
+): Promise<string> {
   const nowSeconds = input.nowSeconds ?? Math.floor(Date.now() / 1000);
   const ttlSeconds = input.ttlSeconds ?? STREAM_TOKEN_DEFAULT_TTL_SECONDS;
   const payload: StreamTokenPayloadType = StreamTokenPayload.parse({
@@ -87,4 +90,7 @@ export async function verifyStreamToken(
   return verifyStreamTokenEnvelope(secret, token, nowSeconds);
 }
 
-export { StreamTokenPayload, type StreamTokenPayload as StreamTokenPayloadType } from "@opengeni/contracts";
+export {
+  StreamTokenPayload,
+  type StreamTokenPayload as StreamTokenPayloadType,
+} from "@opengeni/contracts";

package/src/sandbox/terminal-server.ts

@@ -43,7 +43,9 @@ export class TerminalServerError extends Error {
 
   constructor(exitCode: number, output: string) {
     const stage = exitCode === 14 ? "ttyd" : "unknown";
-    super(`terminal server failed at stage "${stage}" (exit ${exitCode})${output ? `:\n${output}` : ""}`);
+    super(
+      `terminal server failed at stage "${stage}" (exit ${exitCode})${output ? `:\n${output}` : ""}`,
+    );
     this.name = "TerminalServerError";
     this.exitCode = exitCode;
     this.stage = stage;
@@ -70,8 +72,16 @@ type ExecResultLike = {
   exitCode?: number | null;
 };
 type ExecCapableSession = {
-  exec?: (args: { cmd: string; yieldTimeMs?: number; maxOutputTokens?: number }) => Promise<ExecResultLike>;
-  execCommand?: (args: { cmd: string; yieldTimeMs?: number; maxOutputTokens?: number }) => Promise<string>;
+  exec?: (args: {
+    cmd: string;
+    yieldTimeMs?: number;
+    maxOutputTokens?: number;
+  }) => Promise<ExecResultLike>;
+  execCommand?: (args: {
+    cmd: string;
+    yieldTimeMs?: number;
+    maxOutputTokens?: number;
+  }) => Promise<string>;
 };
 
 export type EnsureTerminalServerOptions = {
@@ -198,6 +208,10 @@ export async function tearDownTerminalServer(session: unknown): Promise<void> {
     return;
   }
   if (typeof s?.execCommand === "function") {
-    await s.execCommand({ cmd: "opengeni-terminal-down", yieldTimeMs: 10_000, maxOutputTokens: 4_000 });
+    await s.execCommand({
+      cmd: "opengeni-terminal-down",
+      yieldTimeMs: 10_000,
+      maxOutputTokens: 4_000,
+    });
   }
 }