72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import { createFileRoute } from "@tanstack/react-router";
|
|
import { serverEvents, EVENT_TYPES, type ServerEvent } from "@/lib/events/emitter";
|
|
import { logger } from "@/lib/logger";
|
|
import { superTokensRequestMiddleware } from "@/utils/supertokens";
|
|
|
|
let activeConnections = 0;
|
|
const encoder = new TextEncoder();
|
|
|
|
export const Route = createFileRoute("/api/events/$")({
|
|
server: {
|
|
middleware: [superTokensRequestMiddleware],
|
|
handlers: {
|
|
GET: ({ request }) => {
|
|
activeConnections++;
|
|
const connectionId = `conn_${Date.now()}_${Math.random().toString(36).substring(2, 11)}`;
|
|
logger.info(`ServerEvents | New connection ${connectionId}. Active: ${activeConnections}`);
|
|
|
|
let cleanedUp = false;
|
|
let cleanup = () => {};
|
|
|
|
const stream = new ReadableStream({
|
|
start(controller) {
|
|
const send = (payload: unknown) => {
|
|
try {
|
|
controller.enqueue(encoder.encode(`data: ${JSON.stringify(payload)}\n\n`));
|
|
} catch (error) {
|
|
logger.error("ServerEvents | Error sending SSE message", error);
|
|
cleanup();
|
|
}
|
|
};
|
|
|
|
const handleEvent = (event: ServerEvent) => send(event);
|
|
for (const type of EVENT_TYPES) {
|
|
serverEvents.on(type, handleEvent);
|
|
}
|
|
|
|
const pingInterval = setInterval(() => {
|
|
send({ type: "ping", timestamp: Date.now() });
|
|
}, 15000);
|
|
|
|
cleanup = () => {
|
|
if (cleanedUp) return;
|
|
cleanedUp = true;
|
|
activeConnections--;
|
|
for (const type of EVENT_TYPES) {
|
|
serverEvents.off(type, handleEvent);
|
|
}
|
|
clearInterval(pingInterval);
|
|
logger.info(`ServerEvents | Connection ${connectionId} cleanup completed. Active: ${activeConnections}`);
|
|
};
|
|
|
|
request.signal?.addEventListener("abort", cleanup);
|
|
|
|
send({ type: "connected" });
|
|
},
|
|
cancel() {
|
|
cleanup();
|
|
},
|
|
});
|
|
|
|
return new Response(stream, {
|
|
headers: {
|
|
"Content-Type": "text/event-stream",
|
|
"Cache-Control": "no-cache, no-store, must-revalidate",
|
|
"X-Accel-Buffering": "no",
|
|
},
|
|
});
|
|
},
|
|
},
|
|
},
|
|
});
|