@rivetkit/engine-test-runner 2.0.21

package/.turbo/turbo-build.log

@@ -0,0 +1,17 @@
+
+> @rivetkit/engine-test-runner@2.0.21 build /Users/nathan/rivet/engine/sdks/typescript/test-runner
+> tsup src/index.ts
+
+CLI Building entry: src/index.ts
+CLI Using tsconfig: tsconfig.json
+CLI tsup v8.5.0
+CLI Using tsup config: /Users/nathan/rivet/engine/sdks/typescript/test-runner/tsup.config.ts
+CLI Target: node16
+CLI Cleaning output folder
+ESM Build start
+ESM dist/index.js     9.89 KB
+ESM dist/index.js.map 19.40 KB
+ESM ⚡️ Build success in 9ms
+DTS Build start
+DTS ⚡️ Build success in 720ms
+DTS dist/index.d.ts 175.00 B

package/.turbo/turbo-check-types.log

@@ -0,0 +1,4 @@
+
+> @rivetkit/engine-test-runner@2.0.21 check-types /Users/nathan/rivet/engine/sdks/typescript/test-runner
+> tsc --noEmit
+

package/Dockerfile

@@ -0,0 +1,26 @@
+FROM node:22-slim
+
+WORKDIR /app
+
+# Install pnpm (match workspace version) and set CI for non-interactive behavior
+RUN corepack enable && corepack prepare pnpm@10.13.1 --activate
+ENV CI=true
+
+# Copy and build SDK
+COPY package.json pnpm-lock.yaml pnpm-workspace.yaml tsup.base.ts tsconfig.base.json turbo.json .
+COPY sdks/typescript/runner-protocol ./sdks/typescript/runner-protocol/
+COPY sdks/typescript/runner ./sdks/typescript/runner/
+
+# Install dependencies
+COPY sdks/typescript/test-runner/package.json ./sdks/typescript/test-runner/
+RUN pnpm install --no-frozen-lockfile
+
+# Build the application
+COPY sdks/typescript/test-runner/ ./sdks/typescript/test-runner/
+RUN pnpm build -F @rivetkit/engine-test-runner
+
+# Expose the HTTP server port
+EXPOSE 5050
+
+# Run the application
+CMD ["node", "sdks/typescript/test-runner/dist/index.js"]

package/LICENSE

@@ -0,0 +1,203 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2023 Rivet Gaming, Inc.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+

package/dist/index.d.ts

@@ -0,0 +1,6 @@
+import * as hono_types from 'hono/types';
+import { Hono } from 'hono';
+
+declare const app: Hono<hono_types.BlankEnv, hono_types.BlankSchema, "/">;
+
+export { app as default };

package/dist/index.js

@@ -0,0 +1,325 @@
+// src/index.ts
+import { serve } from "@hono/node-server";
+import { Runner } from "@rivetkit/engine-runner";
+import { Hono } from "hono";
+import { streamSSE } from "hono/streaming";
+
+// src/log.ts
+import {
+  pino,
+  stdTimeFunctions
+} from "pino";
+import { inspect } from "util";
+var baseLogger;
+var configuredLogLevel;
+var loggerCache = /* @__PURE__ */ new Map();
+function getPinoLevel(logLevel) {
+  if (logLevel) {
+    return logLevel;
+  }
+  if (configuredLogLevel) {
+    return configuredLogLevel;
+  }
+  return (process.env["LOG_LEVEL"] || "warn").toString().toLowerCase();
+}
+function getIncludeTarget() {
+  return process.env["LOG_TARGET"] === "1";
+}
+function customWrite(level, o) {
+  const entries = {};
+  if (process.env["LOG_TIMESTAMP"] === "1" && o.time) {
+    const date = typeof o.time === "number" ? new Date(o.time) : /* @__PURE__ */ new Date();
+    entries.ts = date;
+  }
+  entries.level = level.toUpperCase();
+  if (o.target) {
+    entries.target = o.target;
+  }
+  if (o.msg) {
+    entries.msg = o.msg;
+  }
+  for (const [key, value] of Object.entries(o)) {
+    if (key !== "time" && key !== "level" && key !== "target" && key !== "msg" && key !== "pid" && key !== "hostname") {
+      entries[key] = value;
+    }
+  }
+  const output = inspect(entries, {
+    compact: true,
+    breakLength: Infinity,
+    colors: true
+  });
+  console.log(output);
+}
+async function configureDefaultLogger(logLevel) {
+  if (logLevel) {
+    configuredLogLevel = logLevel;
+  }
+  baseLogger = pino({
+    level: getPinoLevel(logLevel),
+    messageKey: "msg",
+    // Do not include pid/hostname in output
+    base: {},
+    // Keep a string level in the output
+    formatters: {
+      level(_label, number) {
+        return { level: number };
+      }
+    },
+    timestamp: process.env["LOG_TIMESTAMP"] === "1" ? stdTimeFunctions.epochTime : false,
+    browser: {
+      write: {
+        fatal: customWrite.bind(null, "fatal"),
+        error: customWrite.bind(null, "error"),
+        warn: customWrite.bind(null, "warn"),
+        info: customWrite.bind(null, "info"),
+        debug: customWrite.bind(null, "debug"),
+        trace: customWrite.bind(null, "trace")
+      }
+    },
+    hooks: {
+      logMethod(inputArgs, _method, level) {
+        const levelMap = {
+          10: "trace",
+          20: "debug",
+          30: "info",
+          40: "warn",
+          50: "error",
+          60: "fatal"
+        };
+        const levelName = levelMap[level] || "info";
+        const time = process.env["LOG_TIMESTAMP"] === "1" ? Date.now() : void 0;
+        if (inputArgs.length >= 2) {
+          const [objOrMsg, msg] = inputArgs;
+          if (typeof objOrMsg === "object" && objOrMsg !== null) {
+            customWrite(levelName, { ...objOrMsg, msg, time });
+          } else {
+            customWrite(levelName, { msg: String(objOrMsg), time });
+          }
+        } else if (inputArgs.length === 1) {
+          const [objOrMsg] = inputArgs;
+          if (typeof objOrMsg === "object" && objOrMsg !== null) {
+            customWrite(levelName, { ...objOrMsg, time });
+          } else {
+            customWrite(levelName, { msg: String(objOrMsg), time });
+          }
+        }
+      }
+    }
+  });
+  loggerCache.clear();
+}
+function getBaseLogger() {
+  if (!baseLogger) {
+    configureDefaultLogger();
+  }
+  return baseLogger;
+}
+function getLogger(name = "default") {
+  const cached = loggerCache.get(name);
+  if (cached) {
+    return cached;
+  }
+  const base = getBaseLogger();
+  const child = getIncludeTarget() ? base.child({ target: name }) : base;
+  loggerCache.set(name, child);
+  return child;
+}
+
+// src/index.ts
+var INTERNAL_SERVER_PORT = process.env.INTERNAL_SERVER_PORT ? Number(process.env.INTERNAL_SERVER_PORT) : 5051;
+var RIVET_NAMESPACE = process.env.RIVET_NAMESPACE ?? "default";
+var RIVET_RUNNER_NAME = process.env.RIVET_RUNNER_NAME ?? "test-runner";
+var RIVET_RUNNER_KEY = process.env.RIVET_RUNNER_KEY ?? `key-${Math.floor(Math.random() * 1e4)}`;
+var RIVET_RUNNER_VERSION = process.env.RIVET_RUNNER_VERSION ? Number(process.env.RIVET_RUNNER_VERSION) : 1;
+var RIVET_RUNNER_TOTAL_SLOTS = process.env.RIVET_RUNNER_TOTAL_SLOTS ? Number(process.env.RIVET_RUNNER_TOTAL_SLOTS) : 100;
+var RIVET_ENDPOINT = process.env.RIVET_ENDPOINT ?? "http://127.0.0.1:6420";
+var RIVET_TOKEN = process.env.RIVET_TOKEN ?? "dev";
+var AUTOSTART_SERVER = process.env.NO_AUTOSTART_SERVER === void 0;
+var AUTOSTART_RUNNER = process.env.NO_AUTOSTART_RUNNER === void 0;
+var runnerStarted = Promise.withResolvers();
+var runnerStopped = Promise.withResolvers();
+var runner = null;
+var actorWebSockets = /* @__PURE__ */ new Map();
+var app = new Hono();
+function loggerMiddleware(logger) {
+  return async (c, next) => {
+    const method = c.req.method;
+    const path = c.req.path;
+    const startTime = Date.now();
+    await next();
+    const duration = Date.now() - startTime;
+    logger.debug({
+      msg: "http request",
+      method,
+      path,
+      status: c.res.status,
+      dt: `${duration}ms`,
+      reqSize: c.req.header("content-length"),
+      resSize: c.res.headers.get("content-length"),
+      userAgent: c.req.header("user-agent")
+    });
+  };
+}
+app.use("*", loggerMiddleware(getLogger()));
+app.get("/wait-ready", async (c) => {
+  await runnerStarted.promise;
+  return c.json(runner == null ? void 0 : runner.runnerId);
+});
+app.get("/has-actor", async (c) => {
+  const actorIdQuery = c.req.query("actor");
+  const generationQuery = c.req.query("generation");
+  const generation = generationQuery ? Number(generationQuery) : void 0;
+  if (!actorIdQuery || !(runner == null ? void 0 : runner.hasActor(actorIdQuery, generation))) {
+    return c.text("", 404);
+  }
+  return c.text("ok");
+});
+app.get("/shutdown", async (c) => {
+  await (runner == null ? void 0 : runner.shutdown(true));
+  return c.text("ok");
+});
+app.get("/start", async (c) => {
+  return streamSSE(c, async (stream) => {
+    const [runner2, runnerStarted2, runnerStopped2] = await startRunner();
+    c.req.raw.signal.addEventListener("abort", () => {
+      getLogger().debug("SSE aborted, shutting down runner");
+      runner2.shutdown(true);
+    });
+    await runnerStarted2.promise;
+    stream.writeSSE({ data: runner2.getServerlessInitPacket() });
+    await runnerStopped2.promise;
+  });
+});
+await autoConfigureServerless();
+if (AUTOSTART_SERVER) {
+  serve({
+    fetch: app.fetch,
+    port: INTERNAL_SERVER_PORT
+  });
+  getLogger().info(
+    `Internal HTTP server listening on port ${INTERNAL_SERVER_PORT}`
+  );
+}
+if (AUTOSTART_RUNNER)
+  [runner, runnerStarted, runnerStopped] = await startRunner();
+async function autoConfigureServerless() {
+  const res = await fetch(
+    `http://127.0.0.1:6420/runner-configs/${RIVET_RUNNER_NAME}?namespace=${RIVET_NAMESPACE}`,
+    {
+      method: "PUT",
+      headers: {
+        Authorization: `Bearer ${RIVET_TOKEN}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        datacenters: {
+          default: {
+            serverless: {
+              url: `http://localhost:${INTERNAL_SERVER_PORT}`,
+              max_runners: 10,
+              slots_per_runner: 1,
+              request_lifespan: 15
+            }
+          }
+        }
+      })
+    }
+  );
+  if (!res.ok) {
+    throw new Error(
+      `request failed: ${res.statusText} (${res.status}):
+${await res.text()}`
+    );
+  }
+}
+async function startRunner() {
+  getLogger().info("Starting runner");
+  const runnerStarted2 = Promise.withResolvers();
+  const runnerStopped2 = Promise.withResolvers();
+  const config = {
+    logger: getLogger(),
+    version: RIVET_RUNNER_VERSION,
+    endpoint: RIVET_ENDPOINT,
+    token: RIVET_TOKEN,
+    namespace: RIVET_NAMESPACE,
+    runnerName: RIVET_RUNNER_NAME,
+    runnerKey: RIVET_RUNNER_KEY,
+    totalSlots: RIVET_RUNNER_TOTAL_SLOTS,
+    prepopulateActorNames: {},
+    onConnected: () => {
+      runnerStarted2.resolve(void 0);
+    },
+    onDisconnected: () => {
+    },
+    onShutdown: () => {
+      runnerStopped2.resolve(void 0);
+    },
+    fetch: async (runner3, actorId, request) => {
+      getLogger().info(
+        `Fetch called for actor ${actorId}, URL: ${request.url}`
+      );
+      const url = new URL(request.url);
+      if (url.pathname === "/ping") {
+        const responseData = {
+          actorId,
+          status: "ok",
+          timestamp: Date.now()
+        };
+        return new Response(JSON.stringify(responseData), {
+          status: 200,
+          headers: { "Content-Type": "application/json" }
+        });
+      } else if (url.pathname === "/sleep") {
+        runner3.sleepActor(actorId);
+        return new Response("ok", {
+          status: 200,
+          headers: { "Content-Type": "application/json" }
+        });
+      }
+      return new Response("ok", { status: 200 });
+    },
+    onActorStart: async (_actorId, _generation, _config) => {
+      getLogger().info(
+        `Actor ${_actorId} started (generation ${_generation})`
+      );
+    },
+    onActorStop: async (_actorId, _generation) => {
+      getLogger().info(
+        `Actor ${_actorId} stopped (generation ${_generation})`
+      );
+    },
+    websocket: async (_runner, actorId, ws, request) => {
+      getLogger().info(`WebSocket connected for actor ${actorId}`);
+      actorWebSockets.set(actorId, ws);
+      ws.addEventListener("message", (event) => {
+        const data = event.data;
+        getLogger().info({
+          msg: `WebSocket message from actor ${actorId}`,
+          data
+        });
+        ws.send(`Echo: ${data}`);
+      });
+      ws.addEventListener("close", () => {
+        getLogger().info(`WebSocket closed for actor ${actorId}`);
+        actorWebSockets.delete(actorId);
+      });
+      ws.addEventListener("error", (error) => {
+        getLogger().error({
+          msg: `WebSocket error for actor ${actorId}:`,
+          error
+        });
+      });
+    }
+  };
+  const runner2 = new Runner(config);
+  await runner2.start();
+  getLogger().info("Waiting runner start...");
+  await runnerStarted2.promise;
+  getLogger().info("Runner started");
+  return [runner2, runnerStarted2, runnerStopped2];
+}
+var index_default = app;
+export {
+  index_default as default
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/log.ts"],"sourcesContent":["import { serve } from \"@hono/node-server\";\nimport type { ActorConfig, RunnerConfig } from \"@rivetkit/engine-runner\";\nimport { Runner } from \"@rivetkit/engine-runner\";\nimport { Hono, type Context as HonoContext, type Next } from \"hono\";\nimport { streamSSE } from \"hono/streaming\";\nimport type { Logger } from \"pino\";\nimport type WebSocket from \"ws\";\nimport { getLogger } from \"./log\";\n\nconst INTERNAL_SERVER_PORT = process.env.INTERNAL_SERVER_PORT\n\t? Number(process.env.INTERNAL_SERVER_PORT)\n\t: 5051;\nconst RIVET_NAMESPACE = process.env.RIVET_NAMESPACE ?? \"default\";\nconst RIVET_RUNNER_NAME = process.env.RIVET_RUNNER_NAME ?? \"test-runner\";\nconst RIVET_RUNNER_KEY =\n\tprocess.env.RIVET_RUNNER_KEY ?? `key-${Math.floor(Math.random() * 10000)}`;\nconst RIVET_RUNNER_VERSION = process.env.RIVET_RUNNER_VERSION\n\t? Number(process.env.RIVET_RUNNER_VERSION)\n\t: 1;\nconst RIVET_RUNNER_TOTAL_SLOTS = process.env.RIVET_RUNNER_TOTAL_SLOTS\n\t? Number(process.env.RIVET_RUNNER_TOTAL_SLOTS)\n\t: 100;\nconst RIVET_ENDPOINT = process.env.RIVET_ENDPOINT ?? \"http://127.0.0.1:6420\";\nconst RIVET_TOKEN = process.env.RIVET_TOKEN ?? \"dev\";\nconst AUTOSTART_SERVER = process.env.NO_AUTOSTART_SERVER === undefined;\nconst AUTOSTART_RUNNER = process.env.NO_AUTOSTART_RUNNER === undefined;\n\nlet runnerStarted = Promise.withResolvers();\nlet runnerStopped = Promise.withResolvers();\nlet runner: Runner | null = null;\nconst actorWebSockets = new Map<string, WebSocket>();\n\n// Create internal server\nconst app = new Hono();\n\nfunction loggerMiddleware(logger: Logger) {\n\treturn async (c: HonoContext, next: Next) => {\n\t\tconst method = c.req.method;\n\t\tconst path = c.req.path;\n\t\tconst startTime = Date.now();\n\n\t\tawait next();\n\n\t\tconst duration = Date.now() - startTime;\n\t\tlogger.debug({\n\t\t\tmsg: \"http request\",\n\t\t\tmethod,\n\t\t\tpath,\n\t\t\tstatus: c.res.status,\n\t\t\tdt: `${duration}ms`,\n\t\t\treqSize: c.req.header(\"content-length\"),\n\t\t\tresSize: c.res.headers.get(\"content-length\"),\n\t\t\tuserAgent: c.req.header(\"user-agent\"),\n\t\t});\n\t};\n}\napp.use(\"*\", loggerMiddleware(getLogger()));\n\napp.get(\"/wait-ready\", async (c) => {\n\tawait runnerStarted.promise;\n\treturn c.json(runner?.runnerId);\n});\n\napp.get(\"/has-actor\", async (c) => {\n\tconst actorIdQuery = c.req.query(\"actor\");\n\tconst generationQuery = c.req.query(\"generation\");\n\tconst generation = generationQuery ? Number(generationQuery) : undefined;\n\n\tif (!actorIdQuery || !runner?.hasActor(actorIdQuery, generation)) {\n\t\treturn c.text(\"\", 404);\n\t}\n\treturn c.text(\"ok\");\n});\n\napp.get(\"/shutdown\", async (c) => {\n\tawait runner?.shutdown(true);\n\treturn c.text(\"ok\");\n});\n\napp.get(\"/start\", async (c) => {\n\treturn streamSSE(c, async (stream) => {\n\t\tconst [runner, runnerStarted, runnerStopped] = await startRunner();\n\n\t\tc.req.raw.signal.addEventListener(\"abort\", () => {\n\t\t\tgetLogger().debug(\"SSE aborted, shutting down runner\");\n\t\t\trunner.shutdown(true);\n\t\t});\n\n\t\tawait runnerStarted.promise;\n\n\t\tstream.writeSSE({ data: runner.getServerlessInitPacket()! });\n\n\t\tawait runnerStopped.promise;\n\t});\n});\n\nawait autoConfigureServerless();\n\nif (AUTOSTART_SERVER) {\n\tserve({\n\t\tfetch: app.fetch,\n\t\tport: INTERNAL_SERVER_PORT,\n\t});\n\tgetLogger().info(\n\t\t`Internal HTTP server listening on port ${INTERNAL_SERVER_PORT}`,\n\t);\n}\n\nif (AUTOSTART_RUNNER)\n\t[runner, runnerStarted, runnerStopped] = await startRunner();\n\nasync function autoConfigureServerless() {\n\tconst res = await fetch(\n\t\t`http://127.0.0.1:6420/runner-configs/${RIVET_RUNNER_NAME}?namespace=${RIVET_NAMESPACE}`,\n\t\t{\n\t\t\tmethod: \"PUT\",\n\t\t\theaders: {\n\t\t\t\tAuthorization: `Bearer ${RIVET_TOKEN}`,\n\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t},\n\t\t\tbody: JSON.stringify({\n\t\t\t\tdatacenters: {\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tserverless: {\n\t\t\t\t\t\t\turl: `http://localhost:${INTERNAL_SERVER_PORT}`,\n\t\t\t\t\t\t\tmax_runners: 10,\n\t\t\t\t\t\t\tslots_per_runner: 1,\n\t\t\t\t\t\t\trequest_lifespan: 15,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}),\n\t\t},\n\t);\n\n\tif (!res.ok) {\n\t\tthrow new Error(\n\t\t\t`request failed: ${res.statusText} (${res.status}):\\n${await res.text()}`,\n\t\t);\n\t}\n}\n\nasync function startRunner(): Promise<\n\t[Runner, PromiseWithResolvers<unknown>, PromiseWithResolvers<unknown>]\n> {\n\tgetLogger().info(\"Starting runner\");\n\n\tconst runnerStarted = Promise.withResolvers();\n\tconst runnerStopped = Promise.withResolvers();\n\n\tconst config: RunnerConfig = {\n\t\tlogger: getLogger(),\n\t\tversion: RIVET_RUNNER_VERSION,\n\t\tendpoint: RIVET_ENDPOINT,\n\t\ttoken: RIVET_TOKEN,\n\t\tnamespace: RIVET_NAMESPACE,\n\t\trunnerName: RIVET_RUNNER_NAME,\n\t\trunnerKey: RIVET_RUNNER_KEY,\n\t\ttotalSlots: RIVET_RUNNER_TOTAL_SLOTS,\n\t\tprepopulateActorNames: {},\n\t\tonConnected: () => {\n\t\t\trunnerStarted.resolve(undefined);\n\t\t},\n\t\tonDisconnected: () => {},\n\t\tonShutdown: () => {\n\t\t\trunnerStopped.resolve(undefined);\n\t\t},\n\t\tfetch: async (runner: Runner, actorId: string, request: Request) => {\n\t\t\tgetLogger().info(\n\t\t\t\t`Fetch called for actor ${actorId}, URL: ${request.url}`,\n\t\t\t);\n\t\t\tconst url = new URL(request.url);\n\t\t\tif (url.pathname === \"/ping\") {\n\t\t\t\t// Return the actor ID in response\n\t\t\t\tconst responseData = {\n\t\t\t\t\tactorId,\n\t\t\t\t\tstatus: \"ok\",\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t};\n\n\t\t\t\treturn new Response(JSON.stringify(responseData), {\n\t\t\t\t\tstatus: 200,\n\t\t\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t\t\t});\n\t\t\t} else if (url.pathname === \"/sleep\") {\n\t\t\t\trunner.sleepActor(actorId);\n\n\t\t\t\treturn new Response(\"ok\", {\n\t\t\t\t\tstatus: 200,\n\t\t\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t\t\t});\n\t\t\t}\n\n\t\t\treturn new Response(\"ok\", { status: 200 });\n\t\t},\n\t\tonActorStart: async (\n\t\t\t_actorId: string,\n\t\t\t_generation: number,\n\t\t\t_config: ActorConfig,\n\t\t) => {\n\t\t\tgetLogger().info(\n\t\t\t\t`Actor ${_actorId} started (generation ${_generation})`,\n\t\t\t);\n\t\t},\n\t\tonActorStop: async (_actorId: string, _generation: number) => {\n\t\t\tgetLogger().info(\n\t\t\t\t`Actor ${_actorId} stopped (generation ${_generation})`,\n\t\t\t);\n\t\t},\n\t\twebsocket: async (\n\t\t\t_runner: Runner,\n\t\t\tactorId: string,\n\t\t\tws: WebSocket,\n\t\t\trequest: Request,\n\t\t) => {\n\t\t\tgetLogger().info(`WebSocket connected for actor ${actorId}`);\n\t\t\tactorWebSockets.set(actorId, ws);\n\n\t\t\t// Echo server - send back any messages received\n\t\t\tws.addEventListener(\"message\", (event) => {\n\t\t\t\tconst data = event.data;\n\t\t\t\tgetLogger().info({\n\t\t\t\t\tmsg: `WebSocket message from actor ${actorId}`,\n\t\t\t\t\tdata,\n\t\t\t\t});\n\t\t\t\tws.send(`Echo: ${data}`);\n\t\t\t});\n\n\t\t\tws.addEventListener(\"close\", () => {\n\t\t\t\tgetLogger().info(`WebSocket closed for actor ${actorId}`);\n\t\t\t\tactorWebSockets.delete(actorId);\n\t\t\t});\n\n\t\t\tws.addEventListener(\"error\", (error) => {\n\t\t\t\tgetLogger().error({\n\t\t\t\t\tmsg: `WebSocket error for actor ${actorId}:`,\n\t\t\t\t\terror,\n\t\t\t\t});\n\t\t\t});\n\t\t},\n\t};\n\n\tconst runner = new Runner(config);\n\n\t// Start runner\n\tawait runner.start();\n\n\t// Wait for runner to be ready\n\tgetLogger().info(\"Waiting runner start...\");\n\tawait runnerStarted.promise;\n\n\tgetLogger().info(\"Runner started\");\n\n\treturn [runner, runnerStarted, runnerStopped];\n}\n\nexport default app;\n","import {\n\ttype Level,\n\ttype LevelWithSilent,\n\ttype Logger,\n\tpino,\n\tstdTimeFunctions,\n} from \"pino\";\nimport { inspect } from \"util\";\n\nexport type { Logger } from \"pino\";\n\nlet baseLogger: Logger | undefined;\nlet configuredLogLevel: Level | undefined;\n\n/** Cache of child loggers by logger name. */\nconst loggerCache = new Map<string, Logger>();\n\nexport function getPinoLevel(logLevel?: Level): LevelWithSilent {\n\t// Priority: provided > configured > env > default\n\tif (logLevel) {\n\t\treturn logLevel;\n\t}\n\n\tif (configuredLogLevel) {\n\t\treturn configuredLogLevel;\n\t}\n\n\treturn (process.env[\"LOG_LEVEL\"] || \"warn\")\n\t\t.toString()\n\t\t.toLowerCase() as LevelWithSilent;\n}\n\nexport function getIncludeTarget(): boolean {\n\treturn process.env[\"LOG_TARGET\"] === \"1\";\n}\n\n/**\n * Configure a custom base logger.\n */\nexport function configureBaseLogger(logger: Logger): void {\n\tbaseLogger = logger;\n\tloggerCache.clear();\n}\n\n// TODO: This can be simplified in logfmt.ts\nfunction customWrite(level: string, o: any) {\n\tconst entries: any = {};\n\n\t// Add timestamp if enabled\n\tif (process.env[\"LOG_TIMESTAMP\"] === \"1\" && o.time) {\n\t\tconst date = typeof o.time === \"number\" ? new Date(o.time) : new Date();\n\t\tentries.ts = date;\n\t}\n\n\t// Add level\n\tentries.level = level.toUpperCase();\n\n\t// Add target if present\n\tif (o.target) {\n\t\tentries.target = o.target;\n\t}\n\n\t// Add message\n\tif (o.msg) {\n\t\tentries.msg = o.msg;\n\t}\n\n\t// Add other properties\n\tfor (const [key, value] of Object.entries(o)) {\n\t\tif (\n\t\t\tkey !== \"time\" &&\n\t\t\tkey !== \"level\" &&\n\t\t\tkey !== \"target\" &&\n\t\t\tkey !== \"msg\" &&\n\t\t\tkey !== \"pid\" &&\n\t\t\tkey !== \"hostname\"\n\t\t) {\n\t\t\tentries[key] = value;\n\t\t}\n\t}\n\n\tconst output = inspect(entries, {\n\t\tcompact: true,\n\t\tbreakLength: Infinity,\n\t\tcolors: true,\n\t});\n\tconsole.log(output);\n}\n\n/**\n * Configure the default logger with optional log level.\n */\nexport async function configureDefaultLogger(logLevel?: Level): Promise<void> {\n\t// Store the configured log level\n\tif (logLevel) {\n\t\tconfiguredLogLevel = logLevel;\n\t}\n\n\tbaseLogger = pino({\n\t\tlevel: getPinoLevel(logLevel),\n\t\tmessageKey: \"msg\",\n\t\t// Do not include pid/hostname in output\n\t\tbase: {},\n\t\t// Keep a string level in the output\n\t\tformatters: {\n\t\t\tlevel(_label: string, number: number) {\n\t\t\t\treturn { level: number };\n\t\t\t},\n\t\t},\n\t\ttimestamp:\n\t\t\tprocess.env[\"LOG_TIMESTAMP\"] === \"1\"\n\t\t\t\t? stdTimeFunctions.epochTime\n\t\t\t\t: false,\n\t\tbrowser: {\n\t\t\twrite: {\n\t\t\t\tfatal: customWrite.bind(null, \"fatal\"),\n\t\t\t\terror: customWrite.bind(null, \"error\"),\n\t\t\t\twarn: customWrite.bind(null, \"warn\"),\n\t\t\t\tinfo: customWrite.bind(null, \"info\"),\n\t\t\t\tdebug: customWrite.bind(null, \"debug\"),\n\t\t\t\ttrace: customWrite.bind(null, \"trace\"),\n\t\t\t},\n\t\t},\n\t\thooks: {\n\t\t\tlogMethod(inputArgs, _method, level) {\n\t\t\t\t// TODO: This is a hack to not implement our own transport target. We can get better perf if we have our own transport target.\n\n\t\t\t\tconst levelMap: Record<number, string> = {\n\t\t\t\t\t10: \"trace\",\n\t\t\t\t\t20: \"debug\",\n\t\t\t\t\t30: \"info\",\n\t\t\t\t\t40: \"warn\",\n\t\t\t\t\t50: \"error\",\n\t\t\t\t\t60: \"fatal\",\n\t\t\t\t};\n\t\t\t\tconst levelName = levelMap[level] || \"info\";\n\t\t\t\tconst time =\n\t\t\t\t\tprocess.env[\"LOG_TIMESTAMP\"] === \"1\"\n\t\t\t\t\t\t? Date.now()\n\t\t\t\t\t\t: undefined;\n\t\t\t\t// TODO: This can be simplified in logfmt.ts\n\t\t\t\tif (inputArgs.length >= 2) {\n\t\t\t\t\tconst [objOrMsg, msg] = inputArgs;\n\t\t\t\t\tif (typeof objOrMsg === \"object\" && objOrMsg !== null) {\n\t\t\t\t\t\tcustomWrite(levelName, { ...objOrMsg, msg, time });\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcustomWrite(levelName, { msg: String(objOrMsg), time });\n\t\t\t\t\t}\n\t\t\t\t} else if (inputArgs.length === 1) {\n\t\t\t\t\tconst [objOrMsg] = inputArgs;\n\t\t\t\t\tif (typeof objOrMsg === \"object\" && objOrMsg !== null) {\n\t\t\t\t\t\tcustomWrite(levelName, { ...objOrMsg, time });\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcustomWrite(levelName, { msg: String(objOrMsg), time });\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t});\n\n\tloggerCache.clear();\n}\n\n/**\n * Get or initialize the base logger.\n */\nexport function getBaseLogger(): Logger {\n\tif (!baseLogger) {\n\t\tconfigureDefaultLogger();\n\t}\n\treturn baseLogger!;\n}\n\n/**\n * Returns a child logger with `target` bound for the given name.\n */\nexport function getLogger(name = \"default\"): Logger {\n\t// Check cache first\n\tconst cached = loggerCache.get(name);\n\tif (cached) {\n\t\treturn cached;\n\t}\n\n\t// Create\n\tconst base = getBaseLogger();\n\n\t// Add target to log if enabled\n\tconst child = getIncludeTarget() ? base.child({ target: name }) : base;\n\n\t// Cache the logger\n\tloggerCache.set(name, child);\n\n\treturn child;\n}\n"],"mappings":";AAAA,SAAS,aAAa;AAEtB,SAAS,cAAc;AACvB,SAAS,YAAoD;AAC7D,SAAS,iBAAiB;;;ACJ1B;AAAA,EAIC;AAAA,EACA;AAAA,OACM;AACP,SAAS,eAAe;AAIxB,IAAI;AACJ,IAAI;AAGJ,IAAM,cAAc,oBAAI,IAAoB;AAErC,SAAS,aAAa,UAAmC;AAE/D,MAAI,UAAU;AACb,WAAO;AAAA,EACR;AAEA,MAAI,oBAAoB;AACvB,WAAO;AAAA,EACR;AAEA,UAAQ,QAAQ,IAAI,WAAW,KAAK,QAClC,SAAS,EACT,YAAY;AACf;AAEO,SAAS,mBAA4B;AAC3C,SAAO,QAAQ,IAAI,YAAY,MAAM;AACtC;AAWA,SAAS,YAAY,OAAe,GAAQ;AAC3C,QAAM,UAAe,CAAC;AAGtB,MAAI,QAAQ,IAAI,eAAe,MAAM,OAAO,EAAE,MAAM;AACnD,UAAM,OAAO,OAAO,EAAE,SAAS,WAAW,IAAI,KAAK,EAAE,IAAI,IAAI,oBAAI,KAAK;AACtE,YAAQ,KAAK;AAAA,EACd;AAGA,UAAQ,QAAQ,MAAM,YAAY;AAGlC,MAAI,EAAE,QAAQ;AACb,YAAQ,SAAS,EAAE;AAAA,EACpB;AAGA,MAAI,EAAE,KAAK;AACV,YAAQ,MAAM,EAAE;AAAA,EACjB;AAGA,aAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,CAAC,GAAG;AAC7C,QACC,QAAQ,UACR,QAAQ,WACR,QAAQ,YACR,QAAQ,SACR,QAAQ,SACR,QAAQ,YACP;AACD,cAAQ,GAAG,IAAI;AAAA,IAChB;AAAA,EACD;AAEA,QAAM,SAAS,QAAQ,SAAS;AAAA,IAC/B,SAAS;AAAA,IACT,aAAa;AAAA,IACb,QAAQ;AAAA,EACT,CAAC;AACD,UAAQ,IAAI,MAAM;AACnB;AAKA,eAAsB,uBAAuB,UAAiC;AAE7E,MAAI,UAAU;AACb,yBAAqB;AAAA,EACtB;AAEA,eAAa,KAAK;AAAA,IACjB,OAAO,aAAa,QAAQ;AAAA,IAC5B,YAAY;AAAA;AAAA,IAEZ,MAAM,CAAC;AAAA;AAAA,IAEP,YAAY;AAAA,MACX,MAAM,QAAgB,QAAgB;AACrC,eAAO,EAAE,OAAO,OAAO;AAAA,MACxB;AAAA,IACD;AAAA,IACA,WACC,QAAQ,IAAI,eAAe,MAAM,MAC9B,iBAAiB,YACjB;AAAA,IACJ,SAAS;AAAA,MACR,OAAO;AAAA,QACN,OAAO,YAAY,KAAK,MAAM,OAAO;AAAA,QACrC,OAAO,YAAY,KAAK,MAAM,OAAO;AAAA,QACrC,MAAM,YAAY,KAAK,MAAM,MAAM;AAAA,QACnC,MAAM,YAAY,KAAK,MAAM,MAAM;AAAA,QACnC,OAAO,YAAY,KAAK,MAAM,OAAO;AAAA,QACrC,OAAO,YAAY,KAAK,MAAM,OAAO;AAAA,MACtC;AAAA,IACD;AAAA,IACA,OAAO;AAAA,MACN,UAAU,WAAW,SAAS,OAAO;AAGpC,cAAM,WAAmC;AAAA,UACxC,IAAI;AAAA,UACJ,IAAI;AAAA,UACJ,IAAI;AAAA,UACJ,IAAI;AAAA,UACJ,IAAI;AAAA,UACJ,IAAI;AAAA,QACL;AACA,cAAM,YAAY,SAAS,KAAK,KAAK;AACrC,cAAM,OACL,QAAQ,IAAI,eAAe,MAAM,MAC9B,KAAK,IAAI,IACT;AAEJ,YAAI,UAAU,UAAU,GAAG;AAC1B,gBAAM,CAAC,UAAU,GAAG,IAAI;AACxB,cAAI,OAAO,aAAa,YAAY,aAAa,MAAM;AACtD,wBAAY,WAAW,EAAE,GAAG,UAAU,KAAK,KAAK,CAAC;AAAA,UAClD,OAAO;AACN,wBAAY,WAAW,EAAE,KAAK,OAAO,QAAQ,GAAG,KAAK,CAAC;AAAA,UACvD;AAAA,QACD,WAAW,UAAU,WAAW,GAAG;AAClC,gBAAM,CAAC,QAAQ,IAAI;AACnB,cAAI,OAAO,aAAa,YAAY,aAAa,MAAM;AACtD,wBAAY,WAAW,EAAE,GAAG,UAAU,KAAK,CAAC;AAAA,UAC7C,OAAO;AACN,wBAAY,WAAW,EAAE,KAAK,OAAO,QAAQ,GAAG,KAAK,CAAC;AAAA,UACvD;AAAA,QACD;AAAA,MACD;AAAA,IACD;AAAA,EACD,CAAC;AAED,cAAY,MAAM;AACnB;AAKO,SAAS,gBAAwB;AACvC,MAAI,CAAC,YAAY;AAChB,2BAAuB;AAAA,EACxB;AACA,SAAO;AACR;AAKO,SAAS,UAAU,OAAO,WAAmB;AAEnD,QAAM,SAAS,YAAY,IAAI,IAAI;AACnC,MAAI,QAAQ;AACX,WAAO;AAAA,EACR;AAGA,QAAM,OAAO,cAAc;AAG3B,QAAM,QAAQ,iBAAiB,IAAI,KAAK,MAAM,EAAE,QAAQ,KAAK,CAAC,IAAI;AAGlE,cAAY,IAAI,MAAM,KAAK;AAE3B,SAAO;AACR;;;ADxLA,IAAM,uBAAuB,QAAQ,IAAI,uBACtC,OAAO,QAAQ,IAAI,oBAAoB,IACvC;AACH,IAAM,kBAAkB,QAAQ,IAAI,mBAAmB;AACvD,IAAM,oBAAoB,QAAQ,IAAI,qBAAqB;AAC3D,IAAM,mBACL,QAAQ,IAAI,oBAAoB,OAAO,KAAK,MAAM,KAAK,OAAO,IAAI,GAAK,CAAC;AACzE,IAAM,uBAAuB,QAAQ,IAAI,uBACtC,OAAO,QAAQ,IAAI,oBAAoB,IACvC;AACH,IAAM,2BAA2B,QAAQ,IAAI,2BAC1C,OAAO,QAAQ,IAAI,wBAAwB,IAC3C;AACH,IAAM,iBAAiB,QAAQ,IAAI,kBAAkB;AACrD,IAAM,cAAc,QAAQ,IAAI,eAAe;AAC/C,IAAM,mBAAmB,QAAQ,IAAI,wBAAwB;AAC7D,IAAM,mBAAmB,QAAQ,IAAI,wBAAwB;AAE7D,IAAI,gBAAgB,QAAQ,cAAc;AAC1C,IAAI,gBAAgB,QAAQ,cAAc;AAC1C,IAAI,SAAwB;AAC5B,IAAM,kBAAkB,oBAAI,IAAuB;AAGnD,IAAM,MAAM,IAAI,KAAK;AAErB,SAAS,iBAAiB,QAAgB;AACzC,SAAO,OAAO,GAAgB,SAAe;AAC5C,UAAM,SAAS,EAAE,IAAI;AACrB,UAAM,OAAO,EAAE,IAAI;AACnB,UAAM,YAAY,KAAK,IAAI;AAE3B,UAAM,KAAK;AAEX,UAAM,WAAW,KAAK,IAAI,IAAI;AAC9B,WAAO,MAAM;AAAA,MACZ,KAAK;AAAA,MACL;AAAA,MACA;AAAA,MACA,QAAQ,EAAE,IAAI;AAAA,MACd,IAAI,GAAG,QAAQ;AAAA,MACf,SAAS,EAAE,IAAI,OAAO,gBAAgB;AAAA,MACtC,SAAS,EAAE,IAAI,QAAQ,IAAI,gBAAgB;AAAA,MAC3C,WAAW,EAAE,IAAI,OAAO,YAAY;AAAA,IACrC,CAAC;AAAA,EACF;AACD;AACA,IAAI,IAAI,KAAK,iBAAiB,UAAU,CAAC,CAAC;AAE1C,IAAI,IAAI,eAAe,OAAO,MAAM;AACnC,QAAM,cAAc;AACpB,SAAO,EAAE,KAAK,iCAAQ,QAAQ;AAC/B,CAAC;AAED,IAAI,IAAI,cAAc,OAAO,MAAM;AAClC,QAAM,eAAe,EAAE,IAAI,MAAM,OAAO;AACxC,QAAM,kBAAkB,EAAE,IAAI,MAAM,YAAY;AAChD,QAAM,aAAa,kBAAkB,OAAO,eAAe,IAAI;AAE/D,MAAI,CAAC,gBAAgB,EAAC,iCAAQ,SAAS,cAAc,cAAa;AACjE,WAAO,EAAE,KAAK,IAAI,GAAG;AAAA,EACtB;AACA,SAAO,EAAE,KAAK,IAAI;AACnB,CAAC;AAED,IAAI,IAAI,aAAa,OAAO,MAAM;AACjC,SAAM,iCAAQ,SAAS;AACvB,SAAO,EAAE,KAAK,IAAI;AACnB,CAAC;AAED,IAAI,IAAI,UAAU,OAAO,MAAM;AAC9B,SAAO,UAAU,GAAG,OAAO,WAAW;AACrC,UAAM,CAACA,SAAQC,gBAAeC,cAAa,IAAI,MAAM,YAAY;AAEjE,MAAE,IAAI,IAAI,OAAO,iBAAiB,SAAS,MAAM;AAChD,gBAAU,EAAE,MAAM,mCAAmC;AACrD,MAAAF,QAAO,SAAS,IAAI;AAAA,IACrB,CAAC;AAED,UAAMC,eAAc;AAEpB,WAAO,SAAS,EAAE,MAAMD,QAAO,wBAAwB,EAAG,CAAC;AAE3D,UAAME,eAAc;AAAA,EACrB,CAAC;AACF,CAAC;AAED,MAAM,wBAAwB;AAE9B,IAAI,kBAAkB;AACrB,QAAM;AAAA,IACL,OAAO,IAAI;AAAA,IACX,MAAM;AAAA,EACP,CAAC;AACD,YAAU,EAAE;AAAA,IACX,0CAA0C,oBAAoB;AAAA,EAC/D;AACD;AAEA,IAAI;AACH,GAAC,QAAQ,eAAe,aAAa,IAAI,MAAM,YAAY;AAE5D,eAAe,0BAA0B;AACxC,QAAM,MAAM,MAAM;AAAA,IACjB,wCAAwC,iBAAiB,cAAc,eAAe;AAAA,IACtF;AAAA,MACC,QAAQ;AAAA,MACR,SAAS;AAAA,QACR,eAAe,UAAU,WAAW;AAAA,QACpC,gBAAgB;AAAA,MACjB;AAAA,MACA,MAAM,KAAK,UAAU;AAAA,QACpB,aAAa;AAAA,UACZ,SAAS;AAAA,YACR,YAAY;AAAA,cACX,KAAK,oBAAoB,oBAAoB;AAAA,cAC7C,aAAa;AAAA,cACb,kBAAkB;AAAA,cAClB,kBAAkB;AAAA,YACnB;AAAA,UACD;AAAA,QACD;AAAA,MACD,CAAC;AAAA,IACF;AAAA,EACD;AAEA,MAAI,CAAC,IAAI,IAAI;AACZ,UAAM,IAAI;AAAA,MACT,mBAAmB,IAAI,UAAU,KAAK,IAAI,MAAM;AAAA,EAAO,MAAM,IAAI,KAAK,CAAC;AAAA,IACxE;AAAA,EACD;AACD;AAEA,eAAe,cAEb;AACD,YAAU,EAAE,KAAK,iBAAiB;AAElC,QAAMD,iBAAgB,QAAQ,cAAc;AAC5C,QAAMC,iBAAgB,QAAQ,cAAc;AAE5C,QAAM,SAAuB;AAAA,IAC5B,QAAQ,UAAU;AAAA,IAClB,SAAS;AAAA,IACT,UAAU;AAAA,IACV,OAAO;AAAA,IACP,WAAW;AAAA,IACX,YAAY;AAAA,IACZ,WAAW;AAAA,IACX,YAAY;AAAA,IACZ,uBAAuB,CAAC;AAAA,IACxB,aAAa,MAAM;AAClB,MAAAD,eAAc,QAAQ,MAAS;AAAA,IAChC;AAAA,IACA,gBAAgB,MAAM;AAAA,IAAC;AAAA,IACvB,YAAY,MAAM;AACjB,MAAAC,eAAc,QAAQ,MAAS;AAAA,IAChC;AAAA,IACA,OAAO,OAAOF,SAAgB,SAAiB,YAAqB;AACnE,gBAAU,EAAE;AAAA,QACX,0BAA0B,OAAO,UAAU,QAAQ,GAAG;AAAA,MACvD;AACA,YAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,UAAI,IAAI,aAAa,SAAS;AAE7B,cAAM,eAAe;AAAA,UACpB;AAAA,UACA,QAAQ;AAAA,UACR,WAAW,KAAK,IAAI;AAAA,QACrB;AAEA,eAAO,IAAI,SAAS,KAAK,UAAU,YAAY,GAAG;AAAA,UACjD,QAAQ;AAAA,UACR,SAAS,EAAE,gBAAgB,mBAAmB;AAAA,QAC/C,CAAC;AAAA,MACF,WAAW,IAAI,aAAa,UAAU;AACrC,QAAAA,QAAO,WAAW,OAAO;AAEzB,eAAO,IAAI,SAAS,MAAM;AAAA,UACzB,QAAQ;AAAA,UACR,SAAS,EAAE,gBAAgB,mBAAmB;AAAA,QAC/C,CAAC;AAAA,MACF;AAEA,aAAO,IAAI,SAAS,MAAM,EAAE,QAAQ,IAAI,CAAC;AAAA,IAC1C;AAAA,IACA,cAAc,OACb,UACA,aACA,YACI;AACJ,gBAAU,EAAE;AAAA,QACX,SAAS,QAAQ,wBAAwB,WAAW;AAAA,MACrD;AAAA,IACD;AAAA,IACA,aAAa,OAAO,UAAkB,gBAAwB;AAC7D,gBAAU,EAAE;AAAA,QACX,SAAS,QAAQ,wBAAwB,WAAW;AAAA,MACrD;AAAA,IACD;AAAA,IACA,WAAW,OACV,SACA,SACA,IACA,YACI;AACJ,gBAAU,EAAE,KAAK,iCAAiC,OAAO,EAAE;AAC3D,sBAAgB,IAAI,SAAS,EAAE;AAG/B,SAAG,iBAAiB,WAAW,CAAC,UAAU;AACzC,cAAM,OAAO,MAAM;AACnB,kBAAU,EAAE,KAAK;AAAA,UAChB,KAAK,gCAAgC,OAAO;AAAA,UAC5C;AAAA,QACD,CAAC;AACD,WAAG,KAAK,SAAS,IAAI,EAAE;AAAA,MACxB,CAAC;AAED,SAAG,iBAAiB,SAAS,MAAM;AAClC,kBAAU,EAAE,KAAK,8BAA8B,OAAO,EAAE;AACxD,wBAAgB,OAAO,OAAO;AAAA,MAC/B,CAAC;AAED,SAAG,iBAAiB,SAAS,CAAC,UAAU;AACvC,kBAAU,EAAE,MAAM;AAAA,UACjB,KAAK,6BAA6B,OAAO;AAAA,UACzC;AAAA,QACD,CAAC;AAAA,MACF,CAAC;AAAA,IACF;AAAA,EACD;AAEA,QAAMA,UAAS,IAAI,OAAO,MAAM;AAGhC,QAAMA,QAAO,MAAM;AAGnB,YAAU,EAAE,KAAK,yBAAyB;AAC1C,QAAMC,eAAc;AAEpB,YAAU,EAAE,KAAK,gBAAgB;AAEjC,SAAO,CAACD,SAAQC,gBAAeC,cAAa;AAC7C;AAEA,IAAO,gBAAQ;","names":["runner","runnerStarted","runnerStopped"]}

package/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "@rivetkit/engine-test-runner",
+  "version": "2.0.21",
+  "type": "module",
+  "dependencies": {
+    "@hono/node-server": "^1.19.1",
+    "hono": "^4.7.0",
+    "pino": "^9.9.5",
+    "ws": "^8.18.3",
+    "@rivetkit/engine-runner": "2.0.21",
+    "@rivetkit/engine-runner-protocol": "2.0.21"
+  },
+  "devDependencies": {
+    "@types/node": "^22.18.1",
+    "@types/ws": "^8.18.1",
+    "tinybench": "^5.0.1",
+    "tsup": "^8.5.0",
+    "tsx": "^4.20.5",
+    "typescript": "^5.9.2",
+    "vitest": "^1.6.1"
+  },
+  "scripts": {
+    "start": "tsx src/index.ts",
+    "build": "tsup src/index.ts",
+    "check-types": "tsc --noEmit"
+  }
+}

package/src/index.ts

@@ -0,0 +1,257 @@
+import { serve } from "@hono/node-server";
+import type { ActorConfig, RunnerConfig } from "@rivetkit/engine-runner";
+import { Runner } from "@rivetkit/engine-runner";
+import { Hono, type Context as HonoContext, type Next } from "hono";
+import { streamSSE } from "hono/streaming";
+import type { Logger } from "pino";
+import type WebSocket from "ws";
+import { getLogger } from "./log";
+
+const INTERNAL_SERVER_PORT = process.env.INTERNAL_SERVER_PORT
+	? Number(process.env.INTERNAL_SERVER_PORT)
+	: 5051;
+const RIVET_NAMESPACE = process.env.RIVET_NAMESPACE ?? "default";
+const RIVET_RUNNER_NAME = process.env.RIVET_RUNNER_NAME ?? "test-runner";
+const RIVET_RUNNER_KEY =
+	process.env.RIVET_RUNNER_KEY ?? `key-${Math.floor(Math.random() * 10000)}`;
+const RIVET_RUNNER_VERSION = process.env.RIVET_RUNNER_VERSION
+	? Number(process.env.RIVET_RUNNER_VERSION)
+	: 1;
+const RIVET_RUNNER_TOTAL_SLOTS = process.env.RIVET_RUNNER_TOTAL_SLOTS
+	? Number(process.env.RIVET_RUNNER_TOTAL_SLOTS)
+	: 100;
+const RIVET_ENDPOINT = process.env.RIVET_ENDPOINT ?? "http://127.0.0.1:6420";
+const RIVET_TOKEN = process.env.RIVET_TOKEN ?? "dev";
+const AUTOSTART_SERVER = process.env.NO_AUTOSTART_SERVER === undefined;
+const AUTOSTART_RUNNER = process.env.NO_AUTOSTART_RUNNER === undefined;
+
+let runnerStarted = Promise.withResolvers();
+let runnerStopped = Promise.withResolvers();
+let runner: Runner | null = null;
+const actorWebSockets = new Map<string, WebSocket>();
+
+// Create internal server
+const app = new Hono();
+
+function loggerMiddleware(logger: Logger) {
+	return async (c: HonoContext, next: Next) => {
+		const method = c.req.method;
+		const path = c.req.path;
+		const startTime = Date.now();
+
+		await next();
+
+		const duration = Date.now() - startTime;
+		logger.debug({
+			msg: "http request",
+			method,
+			path,
+			status: c.res.status,
+			dt: `${duration}ms`,
+			reqSize: c.req.header("content-length"),
+			resSize: c.res.headers.get("content-length"),
+			userAgent: c.req.header("user-agent"),
+		});
+	};
+}
+app.use("*", loggerMiddleware(getLogger()));
+
+app.get("/wait-ready", async (c) => {
+	await runnerStarted.promise;
+	return c.json(runner?.runnerId);
+});
+
+app.get("/has-actor", async (c) => {
+	const actorIdQuery = c.req.query("actor");
+	const generationQuery = c.req.query("generation");
+	const generation = generationQuery ? Number(generationQuery) : undefined;
+
+	if (!actorIdQuery || !runner?.hasActor(actorIdQuery, generation)) {
+		return c.text("", 404);
+	}
+	return c.text("ok");
+});
+
+app.get("/shutdown", async (c) => {
+	await runner?.shutdown(true);
+	return c.text("ok");
+});
+
+app.get("/start", async (c) => {
+	return streamSSE(c, async (stream) => {
+		const [runner, runnerStarted, runnerStopped] = await startRunner();
+
+		c.req.raw.signal.addEventListener("abort", () => {
+			getLogger().debug("SSE aborted, shutting down runner");
+			runner.shutdown(true);
+		});
+
+		await runnerStarted.promise;
+
+		stream.writeSSE({ data: runner.getServerlessInitPacket()! });
+
+		await runnerStopped.promise;
+	});
+});
+
+await autoConfigureServerless();
+
+if (AUTOSTART_SERVER) {
+	serve({
+		fetch: app.fetch,
+		port: INTERNAL_SERVER_PORT,
+	});
+	getLogger().info(
+		`Internal HTTP server listening on port ${INTERNAL_SERVER_PORT}`,
+	);
+}
+
+if (AUTOSTART_RUNNER)
+	[runner, runnerStarted, runnerStopped] = await startRunner();
+
+async function autoConfigureServerless() {
+	const res = await fetch(
+		`http://127.0.0.1:6420/runner-configs/${RIVET_RUNNER_NAME}?namespace=${RIVET_NAMESPACE}`,
+		{
+			method: "PUT",
+			headers: {
+				Authorization: `Bearer ${RIVET_TOKEN}`,
+				"Content-Type": "application/json",
+			},
+			body: JSON.stringify({
+				datacenters: {
+					default: {
+						serverless: {
+							url: `http://localhost:${INTERNAL_SERVER_PORT}`,
+							max_runners: 10,
+							slots_per_runner: 1,
+							request_lifespan: 15,
+						},
+					},
+				},
+			}),
+		},
+	);
+
+	if (!res.ok) {
+		throw new Error(
+			`request failed: ${res.statusText} (${res.status}):\n${await res.text()}`,
+		);
+	}
+}
+
+async function startRunner(): Promise<
+	[Runner, PromiseWithResolvers<unknown>, PromiseWithResolvers<unknown>]
+> {
+	getLogger().info("Starting runner");
+
+	const runnerStarted = Promise.withResolvers();
+	const runnerStopped = Promise.withResolvers();
+
+	const config: RunnerConfig = {
+		logger: getLogger(),
+		version: RIVET_RUNNER_VERSION,
+		endpoint: RIVET_ENDPOINT,
+		token: RIVET_TOKEN,
+		namespace: RIVET_NAMESPACE,
+		runnerName: RIVET_RUNNER_NAME,
+		runnerKey: RIVET_RUNNER_KEY,
+		totalSlots: RIVET_RUNNER_TOTAL_SLOTS,
+		prepopulateActorNames: {},
+		onConnected: () => {
+			runnerStarted.resolve(undefined);
+		},
+		onDisconnected: () => {},
+		onShutdown: () => {
+			runnerStopped.resolve(undefined);
+		},
+		fetch: async (runner: Runner, actorId: string, request: Request) => {
+			getLogger().info(
+				`Fetch called for actor ${actorId}, URL: ${request.url}`,
+			);
+			const url = new URL(request.url);
+			if (url.pathname === "/ping") {
+				// Return the actor ID in response
+				const responseData = {
+					actorId,
+					status: "ok",
+					timestamp: Date.now(),
+				};
+
+				return new Response(JSON.stringify(responseData), {
+					status: 200,
+					headers: { "Content-Type": "application/json" },
+				});
+			} else if (url.pathname === "/sleep") {
+				runner.sleepActor(actorId);
+
+				return new Response("ok", {
+					status: 200,
+					headers: { "Content-Type": "application/json" },
+				});
+			}
+
+			return new Response("ok", { status: 200 });
+		},
+		onActorStart: async (
+			_actorId: string,
+			_generation: number,
+			_config: ActorConfig,
+		) => {
+			getLogger().info(
+				`Actor ${_actorId} started (generation ${_generation})`,
+			);
+		},
+		onActorStop: async (_actorId: string, _generation: number) => {
+			getLogger().info(
+				`Actor ${_actorId} stopped (generation ${_generation})`,
+			);
+		},
+		websocket: async (
+			_runner: Runner,
+			actorId: string,
+			ws: WebSocket,
+			request: Request,
+		) => {
+			getLogger().info(`WebSocket connected for actor ${actorId}`);
+			actorWebSockets.set(actorId, ws);
+
+			// Echo server - send back any messages received
+			ws.addEventListener("message", (event) => {
+				const data = event.data;
+				getLogger().info({
+					msg: `WebSocket message from actor ${actorId}`,
+					data,
+				});
+				ws.send(`Echo: ${data}`);
+			});
+
+			ws.addEventListener("close", () => {
+				getLogger().info(`WebSocket closed for actor ${actorId}`);
+				actorWebSockets.delete(actorId);
+			});
+
+			ws.addEventListener("error", (error) => {
+				getLogger().error({
+					msg: `WebSocket error for actor ${actorId}:`,
+					error,
+				});
+			});
+		},
+	};
+
+	const runner = new Runner(config);
+
+	// Start runner
+	await runner.start();
+
+	// Wait for runner to be ready
+	getLogger().info("Waiting runner start...");
+	await runnerStarted.promise;
+
+	getLogger().info("Runner started");
+
+	return [runner, runnerStarted, runnerStopped];
+}
+
+export default app;

package/src/log.ts

@@ -0,0 +1,194 @@
+import {
+	type Level,
+	type LevelWithSilent,
+	type Logger,
+	pino,
+	stdTimeFunctions,
+} from "pino";
+import { inspect } from "util";
+
+export type { Logger } from "pino";
+
+let baseLogger: Logger | undefined;
+let configuredLogLevel: Level | undefined;
+
+/** Cache of child loggers by logger name. */
+const loggerCache = new Map<string, Logger>();
+
+export function getPinoLevel(logLevel?: Level): LevelWithSilent {
+	// Priority: provided > configured > env > default
+	if (logLevel) {
+		return logLevel;
+	}
+
+	if (configuredLogLevel) {
+		return configuredLogLevel;
+	}
+
+	return (process.env["LOG_LEVEL"] || "warn")
+		.toString()
+		.toLowerCase() as LevelWithSilent;
+}
+
+export function getIncludeTarget(): boolean {
+	return process.env["LOG_TARGET"] === "1";
+}
+
+/**
+ * Configure a custom base logger.
+ */
+export function configureBaseLogger(logger: Logger): void {
+	baseLogger = logger;
+	loggerCache.clear();
+}
+
+// TODO: This can be simplified in logfmt.ts
+function customWrite(level: string, o: any) {
+	const entries: any = {};
+
+	// Add timestamp if enabled
+	if (process.env["LOG_TIMESTAMP"] === "1" && o.time) {
+		const date = typeof o.time === "number" ? new Date(o.time) : new Date();
+		entries.ts = date;
+	}
+
+	// Add level
+	entries.level = level.toUpperCase();
+
+	// Add target if present
+	if (o.target) {
+		entries.target = o.target;
+	}
+
+	// Add message
+	if (o.msg) {
+		entries.msg = o.msg;
+	}
+
+	// Add other properties
+	for (const [key, value] of Object.entries(o)) {
+		if (
+			key !== "time" &&
+			key !== "level" &&
+			key !== "target" &&
+			key !== "msg" &&
+			key !== "pid" &&
+			key !== "hostname"
+		) {
+			entries[key] = value;
+		}
+	}
+
+	const output = inspect(entries, {
+		compact: true,
+		breakLength: Infinity,
+		colors: true,
+	});
+	console.log(output);
+}
+
+/**
+ * Configure the default logger with optional log level.
+ */
+export async function configureDefaultLogger(logLevel?: Level): Promise<void> {
+	// Store the configured log level
+	if (logLevel) {
+		configuredLogLevel = logLevel;
+	}
+
+	baseLogger = pino({
+		level: getPinoLevel(logLevel),
+		messageKey: "msg",
+		// Do not include pid/hostname in output
+		base: {},
+		// Keep a string level in the output
+		formatters: {
+			level(_label: string, number: number) {
+				return { level: number };
+			},
+		},
+		timestamp:
+			process.env["LOG_TIMESTAMP"] === "1"
+				? stdTimeFunctions.epochTime
+				: false,
+		browser: {
+			write: {
+				fatal: customWrite.bind(null, "fatal"),
+				error: customWrite.bind(null, "error"),
+				warn: customWrite.bind(null, "warn"),
+				info: customWrite.bind(null, "info"),
+				debug: customWrite.bind(null, "debug"),
+				trace: customWrite.bind(null, "trace"),
+			},
+		},
+		hooks: {
+			logMethod(inputArgs, _method, level) {
+				// TODO: This is a hack to not implement our own transport target. We can get better perf if we have our own transport target.
+
+				const levelMap: Record<number, string> = {
+					10: "trace",
+					20: "debug",
+					30: "info",
+					40: "warn",
+					50: "error",
+					60: "fatal",
+				};
+				const levelName = levelMap[level] || "info";
+				const time =
+					process.env["LOG_TIMESTAMP"] === "1"
+						? Date.now()
+						: undefined;
+				// TODO: This can be simplified in logfmt.ts
+				if (inputArgs.length >= 2) {
+					const [objOrMsg, msg] = inputArgs;
+					if (typeof objOrMsg === "object" && objOrMsg !== null) {
+						customWrite(levelName, { ...objOrMsg, msg, time });
+					} else {
+						customWrite(levelName, { msg: String(objOrMsg), time });
+					}
+				} else if (inputArgs.length === 1) {
+					const [objOrMsg] = inputArgs;
+					if (typeof objOrMsg === "object" && objOrMsg !== null) {
+						customWrite(levelName, { ...objOrMsg, time });
+					} else {
+						customWrite(levelName, { msg: String(objOrMsg), time });
+					}
+				}
+			},
+		},
+	});
+
+	loggerCache.clear();
+}
+
+/**
+ * Get or initialize the base logger.
+ */
+export function getBaseLogger(): Logger {
+	if (!baseLogger) {
+		configureDefaultLogger();
+	}
+	return baseLogger!;
+}
+
+/**
+ * Returns a child logger with `target` bound for the given name.
+ */
+export function getLogger(name = "default"): Logger {
+	// Check cache first
+	const cached = loggerCache.get(name);
+	if (cached) {
+		return cached;
+	}
+
+	// Create
+	const base = getBaseLogger();
+
+	// Add target to log if enabled
+	const child = getIncludeTarget() ? base.child({ target: name }) : base;
+
+	// Cache the logger
+	loggerCache.set(name, child);
+
+	return child;
+}

package/tsconfig.json

@@ -0,0 +1,11 @@
+{
+	"extends": "../../../../tsconfig.base.json",
+	"compilerOptions": {
+		"paths": {
+			"@/*": ["./src/*"]
+		},
+		"lib": ["ESNext", "DOM"]
+	},
+	"include": ["src/**/*", "tests/**/*", "benches/**/*"],
+	"exclude": ["node_modules"]
+}

package/tsup.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from "tsup";
+import defaultConfig from "../../../../tsup.base.ts";
+
+export default defineConfig({
+	...defaultConfig,
+	format: "esm",
+});

package/turbo.json

@@ -0,0 +1,4 @@
+{
+	"$schema": "https://turbo.build/schema.json",
+	"extends": ["//"]
+}

package/vitest.config.ts

@@ -0,0 +1,16 @@
+import { resolve } from "node:path";
+import { defineConfig } from "vitest/config";
+import defaultConfig from "../../../../vitest.base.ts";
+
+export default defineConfig({
+	...defaultConfig,
+	resolve: {
+		alias: {
+			"@": resolve(__dirname, "./src"),
+		},
+	},
+	test: {
+		...defaultConfig.test,
+		include: ["tests/**/*.test.ts"],
+	},
+});