Filter crew events to only be visible for crew

server/streams.ts

@@ -1,4 +1,6 @@
 import { Schedule } from "~/shared/types/schedule"
+import { readAccounts } from "~/server/database";
+import { canSeeCrew } from "./utils/schedule";
 
 function sendMessage(
 	stream: WritableStream<string>,
@@ -12,15 +14,32 @@ function sendMessage(
 	;
 }
 
-const streams = new Set<WritableStream<string>>();
+function sendMessageAndClose(
+	stream: WritableStream<string>,
+	message: string,
+) {
+	const writer = stream.getWriter();
+	writer.ready
+		.then(() => {
+			writer.write(message);
+			writer.close();
+		}).catch(console.error)
+		.finally(() => writer.releaseLock())
+	;
+}
+
+const streams = new Map<WritableStream<string>, { sessionId?: number, accountId?: number }>();
 
 let keepaliveInterval: ReturnType<typeof setInterval> | null = null
-export function addStream(stream: WritableStream<string>) {
+export function addStream(stream: WritableStream<string>, sessionId?: number, accountId?: number) {
 	if (streams.size === 0) {
 		console.log("Starting keepalive")
 		keepaliveInterval = setInterval(sendKeepalive, 4000)
 	}
-	streams.add(stream);
+	streams.set(stream, { sessionId, accountId });
+	 // Produce a response immediately to avoid the reply waiting for content.
+	const message = `data: connected sid:${sessionId ?? '-'} aid:${accountId ?? '-'}\n\n`;
+	sendMessage(stream, message);
 }
 export function deleteStream(stream: WritableStream<string>) {
 	streams.delete(stream);
@@ -30,18 +49,43 @@ export function deleteStream(stream: WritableStream<string>) {
 	}
 }
 
-export function broadcastUpdate(schedule: Schedule) {
+export function cancelAccountStreams(accountId: number) {
+	for (const [stream, data] of streams) {
+		if (data.accountId === accountId) {
+			sendMessageAndClose(stream, `data: cancelled\n\n`);
+		}
+	}
+}
+
+export function cancelSessionStreams(sessionId: number) {
+	for (const [stream, data] of streams) {
+		if (data.sessionId === sessionId) {
+			sendMessageAndClose(stream, `data: cancelled\n\n`);
+		}
+	}
+}
+
+export async function broadcastUpdate(schedule: Schedule) {
 	const id = Date.now();
-	const data = JSON.stringify(schedule);
-	const message = `id: ${id}\nevent: update\ndata: ${data}\n\n`
-	console.log(`broadcasting update from ${process.pid} to ${streams.size} clients`);
-	for (const stream of streams) {
+	console.log(`broadcasting update to ${streams.size} clients`);
+	if (!streams.size) {
+		return;
+	}
+	const accounts = await readAccounts();
+	const filteredSchedule = filterSchedule(schedule);
+	for (const [stream, streamData] of streams) {
+		let accountType: string | undefined;
+		if (streamData.accountId !== undefined) {
+			accountType = accounts.find(a => a.id === streamData.accountId)?.type
+		}
+		const data = JSON.stringify(canSeeCrew(accountType) ? schedule : filteredSchedule);
+		const message = `id: ${id}\nevent: update\ndata: ${data}\n\n`
 		sendMessage(stream, message);
 	}
 }
 
 function sendKeepalive() {
-	for (const stream of streams) {
+	for (const stream of streams.keys()) {
 		sendMessage(stream, "data: keepalive\n\n");
 	}
 }