add room persistence

server/api.ts

@@ -99,9 +99,12 @@ const updateFunctionByUUID = new Map<string, {
 }>()
 
 //server is starting up... get rooms list
-const roomByUUID = new Map<string, Room>()
+const roomByUUID = db.getRoomsByUUID() ?? new Map<string, Room>()
 const connectionByUUID = db.getConnectionsByUUID() ?? new Map<string, Connection>()
 
+console.log('got rooms',roomByUUID)
+console.log('got connections', connectionByUUID)
+
 export function validateConnectionByUUID(uuid: string) {
 	return connectionByUUID.has(uuid)
 }
@@ -178,6 +181,7 @@ function deleteRoom(room: Room) {
 		}
 	});
 
+	db.deleteRoom(room)
 	roomByUUID.delete(room.id)
 	notifyAllConnections(sseEvent.delete_room, room)
 }
@@ -264,6 +268,7 @@ api.post(`/${apiRoute.room}`, async (ctx) => {
 	}
 
 	con.roomId = room.id
+	db.persistRoom(room)
 	roomByUUID.set(room.id, room)
 
 	//TODO: persist room?
@@ -336,6 +341,9 @@ api.post(`/${apiRoute.clear}/:key`, async (ctx) => {
 	console.log("CLEAR", oldData, roomByUUID)
 	ctx.response.body = oldData
 
+	roomByUUID.forEach(room => db.deleteRoom(room))
+	connectionByUUID.forEach((con, uuid) => db.deleteConnection(uuid))
+
 	//reinit with empty everything
 	connectionByUUID.clear()
 	roomByUUID.clear()

server/db.ts

@@ -1,6 +1,6 @@
 import { Database } from "jsr:@db/sqlite";
-import { Connection, type Identity } from "./api.ts";
 import { assertEquals } from "jsr:@std/assert";
+import type { Connection, Identity, Room } from "./api.ts";
 
 //const db = new Database("test.db");
 
@@ -36,7 +36,19 @@ createTableIdentity.run()
 createTableRoom.run()
 createTableConnection.run()
 
+const upsertRoom = db.prepare(`
+	INSERT
+	INTO room ( id, ownerId )
+	VALUES ( :id, :ownerId )
+	ON CONFLICT(id)
+	DO UPDATE SET 
+		ownerId = excluded.ownerId
+	RETURNING *;
+`)
 
+const deleteConnectionByUUID = db.prepare(`DELETE FROM connection WHERE uuid = :uuid`)
+const deleteRoomByIds = db.prepare(`DELETE FROM room WHERE id = :id AND ownerId = :ownerId;`)
+const selectRooms = db.prepare(`SELECT * FROM room;`)
 
 const upsertConnection = db.prepare(`
 	INSERT 
@@ -66,15 +78,27 @@ const upsertIdentity = db.prepare(`INSERT
 	RETURNING *;`
 )
 
-const selectConnections = db.prepare(`SELECT * FROM connection`)
-const selectIdentityById = db.prepare(`SELECT * FROM identity WHERE id = :id`)
+const selectConnections = db.prepare(`SELECT * FROM connection;`)
+const selectIdentityById = db.prepare(`SELECT * FROM identity WHERE id = :id;`)
 const selectIdentityBySource = db.prepare(
 	`SELECT * 
 	FROM identity 
 	WHERE source = :source 
 	AND source_id = :source_id;`
 )
 
+export function persistRoom(room: Room) {
+	return upsertRoom.get({...room})
+}
+
+export function deleteRoom(room: Room) {
+	return deleteRoomByIds.get({...room})
+}
+
+export function deleteConnection(uuid: string) {
+	return deleteConnectionByUUID.get({uuid})
+}
+
 export function persistConnection(uuid: string, con: Connection) {
 
 	let idResult: Identity | undefined = undefined
@@ -122,6 +146,13 @@ export function getConnectionsByUUID() {
 	return result
 }
 
+export function getRoomsByUUID() {
+	const rooms = selectRooms.all()
+	const result = new Map<string, Room>()
+	rooms.forEach(r => result.set(r.id, r as Room))
+	return result
+}
+
 function removeNullFields(obj: any) {
 	for (const prop in obj) {
 		if (obj[prop] == null)