Revert "feat: migrate data persistence from Supabase to Postgres"

This reverts commit 43419581be.

apps/dotcom-worker/src/TLDrawDurableObject.ts

@@ -1,7 +1,7 @@
 /// <reference no-default-lib="true"/>
 /// <reference types="@cloudflare/workers-types" />
 
-import { Client as PostgresClient } from 'pg'
+import { SupabaseClient } from '@supabase/supabase-js'
 import {
 	READ_ONLY_LEGACY_PREFIX,
 	READ_ONLY_PREFIX,
@@ -11,18 +11,19 @@ import {
 } from '@tldraw/dotcom-shared'
 import {
 	RoomSnapshot,
+	TLCloseEventCode,
 	TLSocketRoom,
 	type PersistedRoomSnapshotForSupabase,
 } from '@tldraw/sync-core'
 import { TLRecord } from '@tldraw/tlschema'
-import { assertExists, exhaustiveSwitchError } from '@tldraw/utils'
+import { assert, assertExists, exhaustiveSwitchError } from '@tldraw/utils'
 import { createPersistQueue, createSentry } from '@tldraw/worker-shared'
 import { IRequest, Router } from 'itty-router'
 import { AlarmScheduler } from './AlarmScheduler'
 import { PERSIST_INTERVAL_MS } from './config'
 import { getR2KeyForRoom } from './r2'
 import { Analytics, DBLoadResult, Environment, TLServerEvent } from './types'
-import { createPostgresClient } from './utils/createPostgresClient'
+import { createSupabaseClient } from './utils/createSupabaseClient'
 import { getSlug } from './utils/roomOpenMode'
 import { throttle } from './utils/throttle'
 
@@ -118,7 +119,7 @@ export class TLDrawDurableObject {
 	storage: DurableObjectStorage
 
 	// For persistence
-    postgresClient: PostgresClient | null
+	supabaseClient: SupabaseClient | void
 
 	// For analytics
 	measure: Analytics | undefined
@@ -126,7 +127,7 @@ export class TLDrawDurableObject {
 	// For error tracking
 	sentryDSN: string | undefined
 
-    readonly postgresTable: string
+	readonly supabaseTable: string
 	readonly r2: {
 		readonly rooms: R2Bucket
 		readonly versionCache: R2Bucket
@@ -142,9 +143,9 @@ export class TLDrawDurableObject {
 		this.storage = state.storage
 		this.sentryDSN = env.SENTRY_DSN
 		this.measure = env.MEASURE
-        this.postgresClient = createPostgresClient(env)
+		this.supabaseClient = createSupabaseClient(env)
 
-        this.postgresTable = env.TLDRAW_ENV === 'production' ? 'drawings' : 'drawings_staging'
+		this.supabaseTable = env.TLDRAW_ENV === 'production' ? 'drawings' : 'drawings_staging'
 		this.r2 = {
 			rooms: env.ROOMS,
 			versionCache: env.ROOMS_HISTORY_EPHEMERAL,
@@ -348,7 +349,7 @@ export class TLDrawDurableObject {
 		}
 	}
 
-    // Load the room's drawing data. First we check the R2 bucket, then we fallback to Postgres (legacy).
+	// Load the room's drawing data. First we check the R2 bucket, then we fallback to supabase (legacy).
 	async loadFromDatabase(persistenceKey: string): Promise<DBLoadResult> {
 		try {
 			const key = getR2KeyForRoom(persistenceKey)
@@ -358,18 +359,26 @@ export class TLDrawDurableObject {
 				return { type: 'room_found', snapshot: await roomFromBucket.json() }
 			}
 
-            // if we don't have a room in the bucket, try to load from Postgres
-            if (!this.postgresClient) return { type: 'room_not_found' }
-            await this.postgresClient.connect()
-            const result = await this.postgresClient.query('SELECT * FROM ' + this.postgresTable + ' WHERE slug = $1', [persistenceKey])
-            await this.postgresClient.end()
+			// if we don't have a room in the bucket, try to load from supabase
+			if (!this.supabaseClient) return { type: 'room_not_found' }
+			const { data, error } = await this.supabaseClient
+				.from(this.supabaseTable)
+				.select('*')
+				.eq('slug', persistenceKey)
 
-            if (result.rows.length === 0) {
+			if (error) {
+				this.logEvent({ type: 'room', roomId: persistenceKey, name: 'failed_load_from_db' })
+
+				console.error('failed to retrieve document', persistenceKey, error)
+				return { type: 'error', error: new Error(error.message) }
+			}
+			// if it didn't find a document, data will be an empty array
+			if (data.length === 0) {
 				return { type: 'room_not_found' }
 			}
 
-            const roomFromPostgres = result.rows[0] as PersistedRoomSnapshotForSupabase
-            return { type: 'room_found', snapshot: roomFromPostgres.drawing }
+			const roomFromSupabase = data[0] as PersistedRoomSnapshotForSupabase
+			return { type: 'room_found', snapshot: roomFromSupabase.drawing }
 		} catch (error) {
 			this.logEvent({ type: 'room', roomId: persistenceKey, name: 'failed_load_from_db' })
 
@@ -400,7 +409,7 @@ export class TLDrawDurableObject {
 		// just in case there's any possibility of setting up a neverending queue
 	}, PERSIST_INTERVAL_MS / 2)
 
-    // Save the room to Postgres
+	// Save the room to supabase
 	async persistToDatabase() {
 		await this._persistQueue()
 	}

apps/dotcom-worker/src/routes/getRoomSnapshot.ts

@@ -3,7 +3,9 @@ import { notFound } from '@tldraw/worker-shared'
 import { IRequest } from 'itty-router'
 import { getR2KeyForSnapshot } from '../r2'
 import { Environment } from '../types'
-import { createPostgresClient, noPostgresSorry } from '../utils/createPostgresClient'
+import { createSupabaseClient, noSupabaseSorry } from '../utils/createSupabaseClient'
+import { getSnapshotsTable } from '../utils/getSnapshotsTable'
+import { R2Snapshot } from './createRoomSnapshot'
 
 function generateReponse(roomId: string, data: RoomSnapshot) {
 	return new Response(
@@ -37,23 +39,22 @@ export async function getRoomSnapshot(request: IRequest, env: Environment): Prom
 		}
 	}
 
-    // If we can't find the snapshot in R2 then fallback to Postgres
-    // Create a Postgres client
-    const postgresClient = createPostgresClient(env)
-    if (!postgresClient) return noPostgresSorry()
+	// If we can't find the snapshot in R2 then fallback to Supabase
+	// Create a supabase client
+	const supabase = createSupabaseClient(env)
+	if (!supabase) return noSupabaseSorry()
 
-    try {
-        await postgresClient.connect()
-        const result = await postgresClient.query('SELECT drawing FROM snapshots WHERE slug = $1 LIMIT 1', [roomId])
-        await postgresClient.end()
+	// Get the snapshot from the table
+	const supabaseTable = getSnapshotsTable(env)
+	const result = await supabase
+		.from(supabaseTable)
+		.select('drawing')
+		.eq('slug', roomId)
+		.maybeSingle()
+	const data = result.data?.drawing as RoomSnapshot
 
-        if (result.rows.length === 0) return notFound()
-        const data = result.rows[0].drawing as RoomSnapshot
+	if (!data) return notFound()
 
 	// Send back the snapshot!
 	return generateReponse(roomId, data)
-    } catch (err) {
-        console.error('Error querying Postgres', err)
-        return new Response(JSON.stringify({ error: true, message: 'Error querying Postgres' }), { status: 500 })
-    }
 }

apps/dotcom-worker/src/utils/createPostgresClient.ts

@@ -1,35 +0,0 @@
-import { Client } from 'pg'
-import { Environment } from '../types'
-
-export function createPostgresClient(env: Environment) {
-    if (env.POSTGRES_HOST && env.POSTGRES_USER && env.POSTGRES_PASSWORD && env.POSTGRES_DB) {
-        var client = new Client({
-            host: env.POSTGRES_HOST,
-            port: env.POSTGRES_PORT ? parseInt(env.POSTGRES_PORT) : 5432,
-            user: env.POSTGRES_USER,
-            password: env.POSTGRES_PASSWORD,
-            database: env.POSTGRES_DB,
-        })
-
-        client.connect()
-
-        client.query(`
-            CREATE TABLE IF NOT EXISTS snapshots (
-                id SERIAL PRIMARY KEY,
-                slug VARCHAR(255) UNIQUE NOT NULL,
-                drawing JSONB NOT NULL,
-                created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
-            );
-        `)
-
-        return client
-
-    } else {
-        console.warn('No Postgres credentials, loading from Postgres disabled')
-        return null
-    }
-}
-
-export function noPostgresSorry() {
-    return new Response(JSON.stringify({ error: true, message: 'Could not create Postgres client' }))
-}