fix(rivetkit): setup database before lifecycle hooks for c.db availability (#4494)

# Description

Please include a summary of the changes and the related issue. Please also include relevant motivation and context.

## Type of change

- [ ] Bug fix (non-breaking change which fixes an issue)
- [ ] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
- [ ] This change requires a documentation update

## How Has This Been Tested?

Please describe the tests that you ran to verify your changes.

## Checklist:

- [ ] My code follows the style guidelines of this project
- [ ] I have performed a self-review of my code
- [ ] I have commented my code, particularly in hard-to-understand areas
- [ ] I have made corresponding changes to the documentation
- [ ] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my feature works
- [ ] New and existing unit tests pass locally with my changes

Author: NathanFlurry

.github/workflows/pkg-pr-new.yaml

@@ -16,4 +16,12 @@ jobs:
       - run: pnpm install
       - run: pnpm build -F rivetkit -F '@rivetkit/*' -F '!@rivetkit/shared-data' -F '!@rivetkit/engine-frontend' -F '!@rivetkit/mcp-hub'
       - run: npx turbo build:pack-inspector -F rivetkit
+      # Strip sourcemaps from inspector tarball to stay under pkg-pr-new 20MB limit
+      - run: |
+          cd rivetkit-typescript/packages/rivetkit/dist
+          mkdir -p /tmp/inspector-repack
+          tar xzf inspector.tar.gz -C /tmp/inspector-repack
+          find /tmp/inspector-repack -name '*.map' -delete
+          tar czf inspector.tar.gz -C /tmp/inspector-repack .
+          rm -rf /tmp/inspector-repack
       - run: pnpm dlx pkg-pr-new publish 'shared/typescript/*' 'engine/sdks/typescript/runner/' 'engine/sdks/typescript/runner-protocol/' 'rivetkit-typescript/packages/*' --packageManager pnpm --template './examples/*'

pnpm-lock.yaml

@@ -351,6 +351,7 @@
 		"cli-table3": "^0.6.5",
 		"commander": "^12.1.0",
 		"dockerode": "^4.0.9",
+		"drizzle-orm": "^0.44.2",
 		"eventsource": "^4.0.0",
 		"local-pkg": "^0.5.1",
 		"tsup": "^8.4.0",

rivetkit-typescript/packages/rivetkit/package.json

@@ -4,9 +4,8 @@ import type { ManagerDriver } from "@/manager/driver";
 import { type AnyConn } from "./conn/mod";
 import type { AnyActorInstance } from "./instance/mod";
 import type { RegistryConfig } from "@/registry/config";
-import type { RawDatabaseClient } from "@/db/config";
+import type { RawDatabaseClient, DrizzleDatabaseClient } from "@/db/config";
 import type { ISqliteVfs } from "@rivetkit/sqlite-vfs";
-import { BaseSQLiteDatabase } from "drizzle-orm/sqlite-core";
 
 export type ActorDriverBuilder = (
 	config: RegistryConfig,
@@ -86,7 +85,7 @@ export interface ActorDriver {
 	 */
 	overrideDrizzleDatabaseClient?(
 		actorId: string,
-	): Promise<BaseSQLiteDatabase<any, any, any, any> | undefined>;
+	): Promise<DrizzleDatabaseClient | undefined>;
 
 	/**
 	 * Creates a SQLite VFS instance for creating KV-backed databases.

rivetkit-typescript/packages/rivetkit/src/actor/driver.ts

@@ -451,6 +451,7 @@ export class ActorInstance<
 
 	get db(): InferDatabaseClient<DB> {
 		if (!this.#db) {
+			console.trace("[DEBUG] database not loaded");
 			throw new errors.DatabaseNotEnabled();
 		}
 		return this.#db;
@@ -511,6 +512,10 @@ export class ActorInstance<
 			}
 		}
 
+		// Setup database before lifecycle hooks so c.db is available in
+		// createState, onCreate, createVars, and onWake.
+		await this.#setupDatabase(preload);
+
 		// Create a write collector to batch new-actor init writes into a
 		// single kvBatchPut.
 		const writeCollector = new WriteCollector(actorDriver, actorId);
@@ -545,10 +550,6 @@ export class ActorInstance<
 		await this.#measureStartup("onWakeMs", () =>
 			this.#callOnStart(),
 		{ pauseKvGuard: true });
-
-		// Setup database.
-		await this.#setupDatabase(preload);
-
 		// Initialize alarms
 		await this.#measureStartup("initAlarmsMs", () =>
 			this.#scheduleManager.initializeAlarms(),