desktop audio and switches to turn up/down audio

2026-04-16 17:19:56 -05:00
parent df95ea4771
commit 135091f6fa
15 changed files with 818 additions and 12 deletions
@@ -0,0 +1,117 @@
+import type { ServerWebSocket } from "bun";
+import { v4 as uuidv4 } from "uuid";
+import * as clientManager from "../clientManager";
+import { encodeMessage } from "../protocol";
+import * as sessionManager from "../sessions/sessionManager";
+import type { DesktopAudioViewer, SocketData } from "../sessions/types";
+import { canUserAccessClient } from "../users";
+
+function sendDesktopAudioCommand(
+  clientId: string,
+  commandType: "desktop_audio_start" | "desktop_audio_stop",
+  payload: Record<string, unknown>,
+) {
+  const target = clientManager.getClient(clientId);
+  if (!target) return false;
+  try {
+    target.ws.send(
+      encodeMessage({
+        type: "command",
+        commandType: commandType as any,
+        id: uuidv4(),
+        payload,
+      }),
+    );
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function safeJson(ws: ServerWebSocket<SocketData>, payload: Record<string, unknown>) {
+  try {
+    ws.send(JSON.stringify(payload));
+  } catch {
+  }
+}
+
+export function handleDesktopAudioViewerOpen(ws: ServerWebSocket<SocketData>) {
+  const { clientId, userId, userRole } = ws.data;
+  if (userId !== undefined && userRole && !canUserAccessClient(userId, userRole as any, clientId)) {
+    ws.close(1008, "Forbidden: client access denied");
+    return;
+  }
+  const sessionId = uuidv4();
+  ws.data.sessionId = sessionId;
+  const target = clientManager.getClient(clientId);
+  const session: DesktopAudioViewer = { id: sessionId, clientId, viewer: ws, createdAt: Date.now() };
+  sessionManager.addDesktopAudioSession(session);
+
+  safeJson(ws, { type: "ready", sessionId, clientId, clientOnline: !!target });
+  if (!target) {
+    safeJson(ws, { type: "status", status: "offline", reason: "Client is offline" });
+    return;
+  }
+
+  safeJson(ws, { type: "status", status: "ready" });
+}
+
+export function handleDesktopAudioViewerMessage(ws: ServerWebSocket<SocketData>, raw: string | ArrayBuffer | Uint8Array) {
+  const { clientId } = ws.data;
+  if (typeof raw === "string") {
+    try {
+      const message = JSON.parse(raw);
+      if (message?.type === "start") {
+        const source = typeof message?.source === "string" ? message.source : "system";
+        const started = sendDesktopAudioCommand(clientId, "desktop_audio_start", {
+          source,
+          sessionId: ws.data.sessionId,
+        });
+        safeJson(ws, { type: "status", status: started ? "connected" : "error" });
+        return;
+      }
+      if (message?.type === "stop") {
+        sendDesktopAudioCommand(clientId, "desktop_audio_stop", {});
+        safeJson(ws, { type: "status", status: "disconnected" });
+      }
+    } catch {
+    }
+    return;
+  }
+}
+
+export function handleDesktopAudioUplink(clientId: string, payload: any) {
+  const sessionId = typeof payload?.sessionId === "string" ? payload.sessionId : "";
+  const bytes = payload?.data instanceof Uint8Array
+    ? payload.data
+    : payload?.data instanceof ArrayBuffer
+      ? new Uint8Array(payload.data)
+      : ArrayBuffer.isView(payload?.data)
+        ? new Uint8Array(payload.data.buffer)
+        : null;
+  if (!bytes || bytes.byteLength === 0) return;
+
+  for (const session of sessionManager.getDesktopAudioSessionsByClient(clientId)) {
+    if (sessionId && session.id !== sessionId) continue;
+    try {
+      session.viewer.send(bytes);
+    } catch {
+    }
+  }
+}
+
+export function cleanupDesktopAudioViewer(ws: ServerWebSocket<SocketData>) {
+  let removedClientId = ws.data.clientId;
+  for (const [sid, session] of sessionManager.getAllDesktopAudioSessions().entries()) {
+    if (session.viewer === ws) {
+      removedClientId = session.clientId;
+      sessionManager.deleteDesktopAudioSession(sid);
+      break;
+    }
+  }
+
+  const hasViewers = sessionManager.getDesktopAudioSessionsByClient(removedClientId).length > 0;
+  if (!hasViewers) {
+    sendDesktopAudioCommand(removedClientId, "desktop_audio_stop", {});
+  }
+}