chore: add rivet as a default driver

Author: NathanFlurry

docs/workers/quickstart.mdx

@@ -42,14 +42,18 @@ export const registry = setup({
 import { registry } from "./registry";
 import { Hono } from "hono";
 import { serve } from "@hono/node-server";
+import { createNodeWebSocket } from '@hono/node-ws'
+
+// Setup server
+const app = new Hono();
 
 // Start RivetKit
 //
 // State is stored in memory, this can be configured later
-const { client, hono } = registry.run();
-
-// Setup server
-const app = new Hono();
+const { injectWebSocket, upgradeWebSocket } = createNodeWebSocket({ app })  // TODO: do this before app
+const { client, hono } = registry.run({
+	getUpgradeWebSocket: () => upgradeWebSocket,
+});
 
 // Expose RivetKit to the frontend (optional)
 app.route("/registry", hono);
@@ -67,6 +71,7 @@ app.post("/increment/:name", async (c) => {
 serve({ fetch: app.fetch, port: 8080 }, (x) =>
 	console.log("Listening at http://localhost:8080"),
 );
+injectWebSocket(server)
 ```
 
 ```ts Express.js
@@ -77,6 +82,37 @@ TODO
 TODO
 ```
 
+```ts Hono
+import { registry } from "./registry";
+import { Hono } from "hono";
+
+// Start RivetKit
+//
+// State is stored in memory, this can be configured later
+const { client, serve } = registry.server();
+
+// Setup server
+const app = new Hono();
+
+// Example endpoint
+app.post("/increment/:name", async (c) => {
+	const name = c.req.param("name");
+
+	// Communicate with actor
+	const counter = client.counter.getOrCreate(name);
+	const newCount = await counter.increment(1);
+
+	return c.text(`New Count: ${newCount}`);
+});
+
+// Start server
+serve(app);
+```
+
+TODO: How to serve without registry helper
+
+TODO: Why we need to use our own custom serve fn
+
 </CodeGroup>
 
 <Info>

examples/hono/src/server.ts

@@ -4,7 +4,7 @@ import { serve } from "@hono/node-server";
 import { createMemoryDriver } from "@rivetkit/memory";
 
 // Start RivetKit
-const { client, hono } = registry.run({
+const { client, hono } = registry.server({
 	driver: createMemoryDriver(),
 });
 

examples/rivet/.dockerignore

@@ -14,8 +14,7 @@
     "typescript": "^5.5.2"
   },
   "dependencies": {
-    "@rivetkit/worker": "https://pkg.pr.new/rivet-gg/rivetkit@65c3659",
-    "@rivetkit/rivet": "https://pkg.pr.new/rivet-gg/rivetkit/@rivetkit/rivet@65c3659"
+    "@rivetkit/worker": "workspace:*"
   },
   "example": {
     "platforms": [

examples/rivet/Dockerfile

@@ -17,4 +17,3 @@ export const registry = setup({
 	workers: { counter },
 });
 
-export type Registry = typeof registry;

examples/rivet/package.json

@@ -1,4 +1,3 @@
-import { startManager } from "@rivetkit/rivet/manager";
 import { registry } from "./registry";
 
-startManager(registry);
+registry.runServer();

examples/rivet/src/registry.ts

@@ -66,6 +66,16 @@
         "default": "./dist/utils.cjs"
       }
     },
+    "./drivers/rivet": {
+      "import": {
+        "types": "./dist/drivers/rivet/mod.d.ts",
+        "default": "./dist/drivers/rivet/mod.js"
+      },
+      "require": {
+        "types": "./dist/drivers/rivet/mod.d.cts",
+        "default": "./dist/drivers/rivet/mod.cjs"
+      }
+    },
     "./driver-helpers": {
       "import": {
         "types": "./dist/driver-helpers/mod.d.ts",
@@ -133,7 +143,7 @@
   "sideEffects": false,
   "scripts": {
     "dev": "pnpm build --watch",
-    "build": "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/worker/errors.ts src/topologies/coordinate/mod.ts src/topologies/partition/mod.ts src/utils.ts src/driver-helpers/mod.ts src/driver-test-suite/mod.ts src/test/mod.ts",
+    "build": "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/worker/errors.ts src/topologies/coordinate/mod.ts src/topologies/partition/mod.ts src/utils.ts src/drivers/rivet/mod.ts src/driver-helpers/mod.ts src/driver-test-suite/mod.ts src/test/mod.ts",
     "check-types": "tsc --noEmit",
     "boop": "tsc --outDir dist/test -d",
     "test": "vitest run",
@@ -152,6 +162,7 @@
   "devDependencies": {
     "@hono/node-server": "^1.14.0",
     "@hono/node-ws": "^1.1.1",
+    "@rivet-gg/actor-core": "^25.1.0",
     "@types/invariant": "^2",
     "@types/node": "^22.13.1",
     "@types/ws": "^8",
@@ -165,10 +176,10 @@
     "ws": "^8.18.1"
   },
   "peerDependencies": {
-    "eventsource": "^3.0.5",
-    "ws": "^8.0.0",
     "@hono/node-server": "^1.14.0",
-    "@hono/node-ws": "^1.1.1"
+    "@hono/node-ws": "^1.1.1",
+    "eventsource": "^3.0.5",
+    "ws": "^8.0.0"
   },
   "peerDependenciesMeta": {
     "eventsource": {

examples/rivet/src/server.ts

@@ -1,3 +1,4 @@
+import { getEnvUniversal } from "@/utils";
 import {
 	type LevelIndex,
 	LevelNameMap,
@@ -75,13 +76,7 @@ export class Logger {
 const loggers: Record<string, Logger> = {};
 
 export function getLogger(name = "default"): Logger {
-	let defaultLogLevelEnv: LogLevel | undefined = undefined;
-	if (typeof Deno !== "undefined") {
-		defaultLogLevelEnv = Deno.env.get("_LOG_LEVEL") as LogLevel;
-	} else if (typeof process !== "undefined") {
-		// Do this after Deno since `process` is sometimes polyfilled
-		defaultLogLevelEnv = process.env._LOG_LEVEL as LogLevel;
-	}
+	const defaultLogLevelEnv: LogLevel | undefined = getEnvUniversal("_LOG_LEVEL") as LogLevel | undefined;
 
 	const defaultLogLevel: LogLevel = defaultLogLevelEnv ?? "INFO";
 	if (!loggers[name]) {

packages/core/package.json

@@ -2,6 +2,7 @@ import { z } from "zod";
 import type { ContentfulStatusCode } from "hono/utils/http-status";
 import * as errors from "@/worker/errors";
 import type { Logger } from "./log";
+import { getEnvUniversal } from "@/utils";
 
 // Maximum size of a key component in bytes
 // Set to 128 bytes to allow for separators and escape characters in the full key
@@ -206,7 +207,7 @@ export function stringifyError(error: unknown): string {
 	if (error instanceof Error) {
 		if (
 			typeof process !== "undefined" &&
-			process.env._RIVETKIT_ERROR_STACK === "1"
+			getEnvUniversal("_RIVETKIT_ERROR_STACK") === "1"
 		) {
 			return `${error.name}: ${error.message}${error.stack ? `\n${error.stack}` : ""}`;
 		} else {