fix(rivetkit): export database provider types

Author: NathanFlurry

rivetkit-typescript/packages/rivetkit/src/db/mod.ts

@@ -1,5 +1,6 @@
 export { db } from "@/common/database/mod";
 export type {
+	AnyDatabaseProvider,
 	DatabaseProvider,
 	DatabaseProviderContext,
 	NativeDatabaseProvider,

rivetkit-typescript/packages/rivetkit/tests/package-surface.test.ts

@@ -22,6 +22,11 @@ import type {
 	WakeContextOf,
 	WebSocketContextOf,
 } from "rivetkit";
+import type {
+	AnyDatabaseProvider,
+	DatabaseProvider,
+	RawAccess,
+} from "rivetkit/db";
 import { db as rawDb } from "rivetkit/db";
 import { db as drizzleDb, defineConfig } from "rivetkit/db/drizzle";
 import { decodeWorkflowHistoryTransport } from "rivetkit/inspector";
@@ -73,6 +78,11 @@ type RestoredContextTypeSmoke = [
 	WebSocketContextOf<typeof contextTypeSmokeActor>,
 ];
 
+type RestoredDatabaseSurfaceTypes = [
+	DatabaseProvider<RawAccess>,
+	AnyDatabaseProvider,
+];
+
 describe("package surface", () => {
 	test("restores supported package entrypoints", () => {
 		expect(packageJson.exports).toHaveProperty("./test");
@@ -101,6 +111,11 @@ describe("package surface", () => {
 		expect(contextTypes).toBeNull();
 	});
 
+	test("restores database provider type exports", () => {
+		const databaseTypes: RestoredDatabaseSurfaceTypes | null = null;
+		expect(databaseTypes).toBeNull();
+	});
+
 	test("keeps database helpers on dedicated subpaths", () => {
 		expect(rivetkit).not.toHaveProperty("db");
 		expect(rivetkit).not.toHaveProperty("defineConfig");