Tldraw/apps/dotcom-worker/src/lib/routes/createRoom.ts

46 wiersze
1.6 KiB
TypeScript

import { SerializedSchema, SerializedStore } from '@tldraw/store'
import { TLRecord } from '@tldraw/tlschema'
import { RoomSnapshot, schema } from '@tldraw/tlsync'
import { IRequest } from 'itty-router'
import { nanoid } from 'nanoid'
import { getR2KeyForRoom } from '../r2'
import { Environment } from '../types'
import { validateSnapshot } from '../utils/validateSnapshot'
type SnapshotRequestBody = {
schema: SerializedSchema
snapshot: SerializedStore<TLRecord>
}
// Sets up a new room based on a provided snapshot, e.g. when a user clicks the "Share" buttons or the "Fork project" buttons.
export async function createRoom(request: IRequest, env: Environment): Promise<Response> {
// The data sent from the client will include the data for the new room
const data = (await request.json()) as SnapshotRequestBody
// There's a chance the data will be invalid, so we check it first
const snapshotResult = validateSnapshot(data)
if (!snapshotResult.ok) {
return Response.json({ error: true, message: snapshotResult.error }, { status: 400 })
}
// Create a new slug for the room
const slug = nanoid()
// Create the new snapshot
const snapshot: RoomSnapshot = {
schema: schema.serialize(),
clock: 0,
documents: Object.values(snapshotResult.value).map((r) => ({
state: r,
lastChangedClock: 0,
})),
tombstones: {},
}
// Bang that snapshot into the database
await env.ROOMS.put(getR2KeyForRoom(slug), JSON.stringify(snapshot))
// Send back the slug so that the client can redirect to the new room
return new Response(JSON.stringify({ error: false, slug }))
}