Added Prisma TypedSQL

apps/backend/package.json

@@ -13,9 +13,9 @@
     "build-self-host-seed-script": "tsup --config prisma/tsup.config.ts",
     "analyze-bundle": "ANALYZE_BUNDLE=1 pnpm run build",
     "start": "next start --port 8102",
-    "codegen-prisma": "pnpm run prisma generate",
-    "codegen-prisma:watch": "pnpm run prisma generate --watch",
-    "codegen": "pnpm run with-env bash -c 'if [ \"$STACK_ACCELERATE_ENABLED\" = \"true\" ]; then pnpm run prisma generate --no-engine && pnpm run generate-docs; else pnpm run codegen-prisma && pnpm run generate-docs; fi'",
+    "codegen-prisma": "concurrently -k -s first \"tsx scripts/mock-postgres.mjs\" \"node -e \\\"setTimeout(() => process.exit(0), 2000)\\\" && cross-env STACK_DATABASE_CONNECTION_STRING=\\\"postgres://postgres@localhost:8118/postgres?pgbouncer=true&connection_limit=1\\\" STACK_DIRECT_DATABASE_CONNECTION_STRING=\\\"postgres://postgres@localhost:8118/postgres?connection_limit=1\\\" STACK_SEED_INTERNAL_PROJECT_PUBLISHABLE_CLIENT_KEY=\\\"test_key\\\" STACK_SEED_INTERNAL_PROJECT_SECRET_SERVER_KEY=\\\"test_secret\\\" STACK_SEED_INTERNAL_PROJECT_SUPER_SECRET_ADMIN_KEY=\\\"test_admin\\\" sh -c \\\"pnpm run prisma migrate reset --force && pnpm run prisma generate --sql\\\"\"",
+    "codegen-prisma:watch": "pnpm run codegen-prisma && chokidar \"prisma/**/*\" -c \"pnpm run codegen-prisma\"",
+    "codegen": "pnpm run with-env bash -c 'if [ \"$STACK_ACCELERATE_ENABLED\" = \"true\" ]; then pnpm run prisma generate --sql --no-engine && pnpm run generate-docs; else pnpm run codegen-prisma && pnpm run generate-docs; fi'",
     "codegen:watch": "concurrently -n \"prisma,docs\" -k \"pnpm run codegen-prisma:watch\" \"pnpm run watch-docs\"",
     "psql-inner": "psql $STACK_DATABASE_CONNECTION_STRING",
     "psql": "pnpm run with-env pnpm run psql-inner",
@@ -31,6 +31,7 @@
     "seed": "pnpm run db-seed-script"
   },
   "dependencies": {
+    "@electric-sql/pglite": "0.2.7",
     "@next/bundle-analyzer": "15.0.3",
     "@node-oauth/oauth2-server": "^5.1.0",
     "@opentelemetry/api": "^1.9.0",
@@ -44,7 +45,7 @@
     "@opentelemetry/sdk-trace-base": "^1.26.0",
     "@opentelemetry/sdk-trace-node": "^1.26.0",
     "@opentelemetry/semantic-conventions": "^1.27.0",
-    "@prisma/client": "^6.0.1",
+    "@prisma/client": "^6.1.0",
     "@prisma/extension-accelerate": "^1.2.1",
     "@prisma/instrumentation": "^5.19.1",
     "@sentry/nextjs": "^8.40.0",
@@ -55,6 +56,7 @@
     "@vercel/functions": "^1.4.2",
     "@vercel/otel": "^1.10.0",
     "bcrypt": "^5.1.1",
+    "chokidar-cli": "^3.0.0",
     "dotenv-cli": "^7.3.0",
     "jose": "^5.2.2",
     "next": "15.0.3",
@@ -63,6 +65,7 @@
     "openid-client": "^5.6.4",
     "oslo": "^1.2.1",
     "pg": "^8.11.3",
+    "pg-gateway": "0.3.0-beta.3",
     "posthog-node": "^4.1.0",
     "react": "^19.0.0",
     "react-dom": "^19.0.0",
@@ -82,9 +85,10 @@
     "@types/react-dom": "^19.0.0",
     "@types/semver": "^7.5.8",
     "concurrently": "^8.2.2",
+    "cross-env": "^7.0.3",
     "glob": "^10.4.1",
     "import-in-the-middle": "^1.12.0",
-    "prisma": "^6.0.1",
+    "prisma": "^6.1.0",
     "require-in-the-middle": "^7.4.0",
     "rimraf": "^5.0.5",
     "tsup": "^8.3.0",

apps/backend/prisma/schema.prisma

@@ -1,6 +1,6 @@
 generator client {
   provider        = "prisma-client-js"
-  previewFeatures = ["tracing", "relationJoins"]
+  previewFeatures = ["typedSql", "relationJoins"]
 }
 
 datasource db {

apps/backend/prisma/sql/getUsersLastActiveAtMillis.sql

@@ -0,0 +1,4 @@
+SELECT data->>'userId' as "userId", MAX("createdAt") as "lastActiveAt"
+FROM "Event"
+WHERE data->>'userId' = ANY($1)
+GROUP BY data->>'userId'

apps/backend/scripts/mock-postgres.mjs

@@ -0,0 +1,50 @@
+import { PGlite } from '@electric-sql/pglite';
+import net from 'node:net';
+import { fromNodeSocket } from 'pg-gateway/node';
+
+const db = new PGlite();
+
+const server = net.createServer(async (socket) => {
+  let activeDb;
+
+  console.info(`Client connected: ${socket.remoteAddress}:${socket.remotePort}`)
+  await fromNodeSocket(socket, {
+    serverVersion: '16.3',
+
+    auth: {
+      // No password required
+      method: 'trust',
+    },
+
+    async onStartup({ clientParams }) {
+      console.info(`Connecting client to ${clientParams?.database}`)
+      // If the DB is the Prisma shadow DB, create a temp in-memory instance
+      if (clientParams?.database === 'prisma-shadow') {
+        activeDb = new PGlite()
+      } else {
+        activeDb = db
+      }
+
+      // Wait for PGlite to be ready before further processing
+      await activeDb.waitReady
+    },
+
+    async onMessage(data, { isAuthenticated }) {
+      if (!isAuthenticated) {
+        // currently we have no authentication, but let's keep it for the future
+        return
+      }
+
+      // Forward raw message to PGlite and send response to client
+      return await activeDb.execProtocolRaw(data)
+    },
+  })
+
+  socket.on('end', () => {
+    console.info('Client disconnected')
+  })
+})
+
+server.listen(8118, () => {
+  console.info('Postgres server listening on port 8118')
+})

apps/backend/sentry.client.config.ts

@@ -14,6 +14,10 @@ Sentry.init({
 
   enabled: process.env.NODE_ENV !== "development" && !process.env.CI,
 
+  // Sentry and Prisma are not compatible with the new tracing. https://github.com/prisma/prisma/issues/25885
+  // TODO: Remove this once sentry fixes the issue.
+  tracesSampleRate: 0,
+
   // You can remove this option if you're not planning to use the Sentry Session Replay feature:
   integrations: [
     Sentry.replayIntegration({

apps/backend/src/app/api/v1/users/crud.tsx

@@ -5,6 +5,7 @@ import { prismaClient, retryTransaction } from "@/prisma-client";
 import { createCrudHandlers } from "@/route-handlers/crud-handler";
 import { runAsynchronouslyAndWaitUntil } from "@/utils/vercel";
 import { BooleanTrue, Prisma } from "@prisma/client";
+import { getUsersLastActiveAtMillis as getUsersLastActiveAtMillisSql } from "@prisma/client/sql";
 import { KnownErrors } from "@stackframe/stack-shared";
 import { currentUserCrud } from "@stackframe/stack-shared/dist/interface/crud/current-user";
 import { UsersCrud, usersCrud } from "@stackframe/stack-shared/dist/interface/crud/users";
@@ -228,16 +229,11 @@ export const getUsersLastActiveAtMillis = async (userIds: string[], fallbackTo:
     return [];
   }
 
-  const events = await prismaClient.$queryRaw<Array<{ userId: string, lastActiveAt: Date }>>`
-    SELECT data->>'userId' as "userId", MAX("createdAt") as "lastActiveAt"
-    FROM "Event"
-    WHERE data->>'userId' = ANY(${Prisma.sql`ARRAY[${Prisma.join(userIds)}]`})
-    GROUP BY data->>'userId'
-  `;
+  const events = await prismaClient.$queryRawTyped(getUsersLastActiveAtMillisSql(userIds));
 
   return userIds.map((userId, index) => {
     const event = events.find(e => e.userId === userId);
-    return event ? event.lastActiveAt.getTime() : (
+    return event && event.lastActiveAt ? event.lastActiveAt.getTime() : (
       typeof fallbackTo[index] === "number" ? (fallbackTo[index] as number) : (fallbackTo[index] as Date).getTime()
     );
   });