Implement lists module and dev login setup
This commit is contained in:
@@ -0,0 +1,58 @@
|
||||
import postgres from "postgres";
|
||||
import { getCurrentSession } from "@/lib/session";
|
||||
import { canAccessList } from "@/modules/lists/server/queries";
|
||||
import { listChannel } from "@/modules/lists/server/realtime";
|
||||
|
||||
export const runtime = "nodejs";
|
||||
export const dynamic = "force-dynamic";
|
||||
|
||||
export async function GET(request: Request, { params }: { params: Promise<{ id: string }> }) {
|
||||
const { id } = await params;
|
||||
const { household } = await getCurrentSession();
|
||||
if (!(await canAccessList(id, household.id))) return new Response("Forbidden", { status: 403 });
|
||||
|
||||
const encoder = new TextEncoder();
|
||||
const sql = postgres(process.env["DATABASE_URL"]!, { max: 1 });
|
||||
let heartbeat: ReturnType<typeof setInterval> | undefined;
|
||||
let listener: { unlisten(): Promise<void> } | undefined;
|
||||
let closed = false;
|
||||
|
||||
async function cleanup() {
|
||||
if (closed) return;
|
||||
closed = true;
|
||||
if (heartbeat) clearInterval(heartbeat);
|
||||
await listener?.unlisten().catch(() => undefined);
|
||||
await sql.end({ timeout: 1 }).catch(() => undefined);
|
||||
}
|
||||
|
||||
const stream = new ReadableStream<Uint8Array>({
|
||||
async start(controller) {
|
||||
controller.enqueue(encoder.encode(": connected\n\n"));
|
||||
|
||||
listener = await sql.listen(listChannel(id), (payload) => {
|
||||
if (closed) return;
|
||||
controller.enqueue(encoder.encode(`data: ${payload}\n\n`));
|
||||
});
|
||||
|
||||
heartbeat = setInterval(() => {
|
||||
if (closed) return;
|
||||
controller.enqueue(encoder.encode(": heartbeat\n\n"));
|
||||
}, 25_000);
|
||||
|
||||
request.signal.addEventListener("abort", () => {
|
||||
void cleanup();
|
||||
});
|
||||
},
|
||||
cancel() {
|
||||
return cleanup();
|
||||
},
|
||||
});
|
||||
|
||||
return new Response(stream, {
|
||||
headers: {
|
||||
"Cache-Control": "no-cache, no-transform",
|
||||
Connection: "keep-alive",
|
||||
"Content-Type": "text/event-stream",
|
||||
},
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user