feat: sqlite v2

Author: jog1t

.gitignore

@@ -184,3 +184,5 @@ Cargo.lock
 **/.wrangler
 **/.DS_Store
 .aider*
+
+packages/**/tsup.config.bundled_*.mjs

examples/drizzle/drizzle/migrations.js

@@ -1,10 +1,9 @@
-import journal from './meta/_journal.json';
-import m0000 from './0000_wonderful_iron_patriot.sql';
+import journal from "./meta/_journal.json";
+import m0000 from "./0000_wonderful_iron_patriot.sql";
 
-  export default {
-    journal,
-    migrations: {
-      m0000,
-    }
-  }
-  
+export default {
+	journal,
+	migrations: {
+		m0000,
+	},
+};

examples/drizzle/hooks.js

@@ -4,12 +4,13 @@
   "private": true,
   "type": "module",
   "scripts": {
-    "dev": "tsx --watch src/server.ts",
+    "dev": "tsx --loader @rivetkit/sql-loader --watch src/server.ts",
     "check-types": "tsc --noEmit"
   },
   "devDependencies": {
     "@types/node": "^22.13.9",
-    "rivetkit": "workspace:*",
+    "@rivetkit/sql-loader": "workspace:*",
+    "@rivetkit/actor": "workspace:*",
     "tsx": "^3.12.7",
     "typescript": "^5.5.2"
   },

examples/drizzle/package.json

@@ -1,9 +1,9 @@
-// import { int, sqliteTable, text } from "@rivetkit/db/drizzle";
+import { int, sqliteTable, text } from "@rivetkit/db/drizzle";
 
-// export const usersTable = sqliteTable("users_table", {
-// 	id: int().primaryKey({ autoIncrement: true }),
-// 	name: text().notNull(),
-// 	age: int().notNull(),
-// 	email: text().notNull().unique(),
-// 	email2: text().notNull().unique(),
-// });
+export const usersTable = sqliteTable("users_table", {
+	id: int().primaryKey({ autoIncrement: true }),
+	name: text().notNull(),
+	age: int().notNull(),
+	email: text().notNull().unique(),
+	email2: text().notNull().unique(),
+});

examples/drizzle/register.js

@@ -1,27 +1,33 @@
-// import { actor, setup } from "rivetkit";
-// import { db } from "@rivetkit/db/drizzle";
-// import * as schema from "./db/schema";
-// import migrations from "../drizzle/migrations";
+import { actor, setup } from "@rivetkit/actor";
+import { db } from "@rivetkit/db/drizzle";
+import * as schema from "./db/schema";
+import migrations from "../drizzle/migrations";
 
-// export const counter = actor({
-// 	db: db({ schema, migrations }),
-// 	state: {
-// 		count: 0,
-// 	},
-// 	onAuth: () => {
-// 		// Configure auth here
-// 	},
-// 	actions: {
-// 		increment: (c, x: number) => {
-// 			// createState or state fix fix fix
-// 			c.db.c.state.count += x;
-// 			return c.state.count;
-// 		},
-// 	},
-// });
+export const counter = actor({
+	db: db({ schema, migrations }),
+	state: {
+		count: 0,
+	},
+	onAuth: () => {
+		return "hello";
+		// Configure auth here
+	},
+	actions: {
+		increment: (c, x: number) => {
+			// createState or state fix fix fix
+			const y = c.db;
 
-// export const registry = setup({
-// 	use: { counter },
-// });
+			// Here's DB is unknown
+			// c.db.usersTable;
+			return c.state.count;
+		},
+	},
+});
+// but here, the type is known
+// (await counter.config.db("hello")).client.query.usersTable.findFirst;
 
-// export type Registry = typeof registry;
+export const registry = setup({
+	use: { counter },
+});
+
+export type Registry = typeof registry;

examples/drizzle/src/db/schema.ts

@@ -1,7 +1,3 @@
-// import { registry } from "./registry";
-// import { createMemoryDriver } from "@rivetkit/memory";
-// import { serve } from "@rivetkit/nodejs";
+import { registry } from "./registry";
 
-// serve(registry, {
-// 	driver: createMemoryDriver(),
-// });
+await registry.runServer();

examples/drizzle/src/registry.ts

@@ -135,6 +135,16 @@
         "types": "./dist/test/mod.d.cts",
         "default": "./dist/test/mod.cjs"
       }
+    },
+    "./db": {
+      "import": {
+        "types": "./dist/db/mod.d.ts",
+        "default": "./dist/db/mod.js"
+      },
+      "require": {
+        "types": "./dist/db/mod.d.cts",
+        "default": "./dist/bd/mod.cjs"
+      }
     }
   },
   "engines": {
@@ -143,7 +153,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/actor/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",
+    "build": "tsup src/mod.ts src/client/mod.ts src/common/log.ts src/common/websocket.ts src/actor/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 src/db/mod.ts",
     "check-types": "tsc --noEmit",
     "boop": "tsc --outDir dist/test -d",
     "test": "vitest run",

examples/drizzle/src/server.ts

@@ -5,8 +5,8 @@ import type { ConnId } from "./connection";
 import type { ActorContext } from "./context";
 import type { SaveStateOptions } from "./instance";
 import type { Schedule } from "./schedule";
-import { Registry } from "@/registry/mod";
-import { Client } from "@/client/client";
+import type { Registry } from "@/registry/mod";
+import type { Client } from "@/client/client";
 
 /**
  * Context for a remote procedure call.