diff options
Diffstat (limited to 'packages/function')
| -rw-r--r-- | packages/function/package.json | 10 | ||||
| -rw-r--r-- | packages/function/src/api.ts | 167 | ||||
| -rw-r--r-- | packages/function/sst-env.d.ts | 25 | ||||
| -rw-r--r-- | packages/function/tsconfig.json | 9 |
4 files changed, 211 insertions, 0 deletions
diff --git a/packages/function/package.json b/packages/function/package.json new file mode 100644 index 000000000..46c83e840 --- /dev/null +++ b/packages/function/package.json @@ -0,0 +1,10 @@ +{ + "name": "@opencode/function", + "version": "0.0.1", + "$schema": "https://json.schemastore.org/package.json", + "private": true, + "type": "module", + "devDependencies": { + "@cloudflare/workers-types": "^4.20250522.0" + } +} diff --git a/packages/function/src/api.ts b/packages/function/src/api.ts new file mode 100644 index 000000000..3e2ce9eca --- /dev/null +++ b/packages/function/src/api.ts @@ -0,0 +1,167 @@ +import { DurableObject } from "cloudflare:workers" +import { randomUUID } from "node:crypto" + +type Env = { + SYNC_SERVER: DurableObjectNamespace<SyncServer> + Bucket: R2Bucket +} + +export class SyncServer extends DurableObject<Env> { + constructor(ctx: DurableObjectState, env: Env) { + super(ctx, env) + } + async fetch() { + console.log("SyncServer subscribe") + + const webSocketPair = new WebSocketPair() + const [client, server] = Object.values(webSocketPair) + + this.ctx.acceptWebSocket(server) + + const data = await this.ctx.storage.list() + for (const [key, content] of data.entries()) { + server.send(JSON.stringify({ key, content })) + } + + return new Response(null, { + status: 101, + webSocket: client, + }) + } + + async webSocketMessage(ws, message) {} + + async webSocketClose(ws, code, reason, wasClean) { + ws.close(code, "Durable Object is closing WebSocket") + } + + async publish(secret: string, key: string, content: any) { + if (secret !== (await this.getSecret())) throw new Error("Invalid secret") + const sessionID = await this.getSessionID() + if ( + !key.startsWith(`session/info/${sessionID}`) && + !key.startsWith(`session/message/${sessionID}/`) + ) + return new Response("Error: Invalid key", { status: 400 }) + + // store message + await this.env.Bucket.put(`share/${key}.json`, JSON.stringify(content), { + httpMetadata: { + contentType: "application/json", + }, + }) + await this.ctx.storage.put(key, content) + const clients = this.ctx.getWebSockets() + console.log("SyncServer publish", key, "to", clients.length, "subscribers") + for (const client of clients) { + client.send(JSON.stringify({ key, content })) + } + } + + public async share(sessionID: string) { + let secret = await this.getSecret() + if (secret) return secret + secret = randomUUID() + + await this.ctx.storage.put("secret", secret) + await this.ctx.storage.put("sessionID", sessionID) + + return secret + } + + private async getSecret() { + return this.ctx.storage.get<string>("secret") + } + + private async getSessionID() { + return this.ctx.storage.get<string>("sessionID") + } + + async clear(secret: string) { + await this.assertSecret(secret) + await this.ctx.storage.deleteAll() + } + + private async assertSecret(secret: string) { + if (secret !== (await this.getSecret())) throw new Error("Invalid secret") + } + + static shortName(id: string) { + return id.substring(id.length - 8) + } +} + +export default { + async fetch(request: Request, env: Env, ctx: ExecutionContext) { + const url = new URL(request.url) + const splits = url.pathname.split("/") + const method = splits[1] + + if (request.method === "GET" && method === "") { + return new Response("Hello, world!", { + headers: { "Content-Type": "text/plain" }, + }) + } + + if (request.method === "POST" && method === "share_create") { + const body = await request.json<any>() + const sessionID = body.sessionID + const short = SyncServer.shortName(sessionID) + const id = env.SYNC_SERVER.idFromName(short) + const stub = env.SYNC_SERVER.get(id) + const secret = await stub.share(sessionID) + return new Response( + JSON.stringify({ + secret, + url: "https://dev.opencode.ai/s?id=" + short, + }), + { + headers: { "Content-Type": "application/json" }, + }, + ) + } + + if (request.method === "POST" && method === "share_delete") { + const body = await request.json<any>() + const sessionID = body.sessionID + const secret = body.secret + const id = env.SYNC_SERVER.idFromName(SyncServer.shortName(sessionID)) + const stub = env.SYNC_SERVER.get(id) + await stub.clear(secret) + return new Response(JSON.stringify({}), { + headers: { "Content-Type": "application/json" }, + }) + } + + if (request.method === "POST" && method === "share_sync") { + const body = await request.json<{ + sessionID: string + secret: string + key: string + content: any + }>() + const name = SyncServer.shortName(body.sessionID) + const id = env.SYNC_SERVER.idFromName(name) + const stub = env.SYNC_SERVER.get(id) + await stub.publish(body.secret, body.key, body.content) + return new Response(JSON.stringify({}), { + headers: { "Content-Type": "application/json" }, + }) + } + + if (request.method === "GET" && method === "share_poll") { + const upgradeHeader = request.headers.get("Upgrade") + if (!upgradeHeader || upgradeHeader !== "websocket") { + return new Response("Error: Upgrade header is required", { + status: 426, + }) + } + const id = url.searchParams.get("id") + console.log("share_poll", id) + if (!id) + return new Response("Error: Share ID is required", { status: 400 }) + const stub = env.SYNC_SERVER.get(env.SYNC_SERVER.idFromName(id)) + return stub.fetch(request) + } + }, +} diff --git a/packages/function/sst-env.d.ts b/packages/function/sst-env.d.ts new file mode 100644 index 000000000..41727ee9d --- /dev/null +++ b/packages/function/sst-env.d.ts @@ -0,0 +1,25 @@ +/* This file is auto-generated by SST. Do not edit. */ +/* tslint:disable */ +/* eslint-disable */ +/* deno-fmt-ignore-file */ + +import "sst" +declare module "sst" { + export interface Resource { + "Web": { + "type": "sst.cloudflare.StaticSite" + "url": string + } + } +} +// cloudflare +import * as cloudflare from "@cloudflare/workers-types"; +declare module "sst" { + export interface Resource { + "Api": cloudflare.Service + "Bucket": cloudflare.R2Bucket + } +} + +import "sst" +export {}
\ No newline at end of file diff --git a/packages/function/tsconfig.json b/packages/function/tsconfig.json new file mode 100644 index 000000000..0faf16aab --- /dev/null +++ b/packages/function/tsconfig.json @@ -0,0 +1,9 @@ +{ + "$schema": "https://json.schemastore.org/tsconfig", + "extends": "@tsconfig/node22/tsconfig.json", + "compilerOptions": { + "module": "ESNext", + "moduleResolution": "bundler", + "types": ["@cloudflare/workers-types", "node"] + } +} |