1 files changed, 131 insertions, 0 deletions
diff --git a/client-web/source/user/local.ts b/client-web/source/user/local.ts
new file mode 100644
index 0000000..38bcfb9
--- /dev/null
+++ b/client-web/source/user/local.ts
@@ -0,0 +1,131 @@
+/// <reference lib="dom" />
+
+import { log } from "../logger.ts";
+import { PREFS } from "../preferences.ts";
+import { RemoteUser } from "./remote.ts";
+import { get_rnnoise_node } from "../rnnoise.ts";
+import { Room } from "../room.ts";
+import { TrackHandle } from "../track_handle.ts";
+import { User } from "./mod.ts";
+
+export class LocalUser extends User {
+    mic_gain?: GainNode
+    default_gain: number = PREFS.microphone_gain
+
+    constructor(room: Room, id: number) {
+        super(room, id)
+        this.el.classList.add("local")
+        this.local = true
+        this.create_controls()
+        this.add_initial_tracks()
+        log("usermodel", `added local user: ${this.display_name}`)
+    }
+
+    async add_initial_tracks() {
+        if (PREFS.microphone_enabled) this.publish_track(await this.create_mic_track())
+        if (PREFS.camera_enabled) this.publish_track(await this.create_camera_track())
+        if (PREFS.screencast_enabled) this.publish_track(await this.create_screencast_track())
+    }
+
+    publish_track(t: TrackHandle) {
+        this.room.remote_users.forEach(u => u.peer.addTrack(t.track))
+        this.add_track(t)
+        t.addEventListener("ended", () => {
+            this.room.remote_users.forEach(u => {
+                u.peer.getSenders().forEach(s => {
+                    if (s.track == t.track) u.peer.removeTrack(s)
+                })
+            })
+        })
+    }
+
+    add_initial_to_remote(u: RemoteUser) {
+        this.tracks.forEach(t => u.peer.addTrack(t.track))
+    }
+
+    create_controls() {
+        const mic_toggle = document.createElement("input")
+        const camera_toggle = document.createElement("input")
+        const screen_toggle = document.createElement("input")
+        mic_toggle.type = camera_toggle.type = screen_toggle.type = "button"
+        mic_toggle.value = "Microphone"
+        camera_toggle.value = "Camera"
+        screen_toggle.value = "Screencast"
+
+        const create = async (_e: HTMLElement, tp: Promise<TrackHandle>) => {
+            log("media", "awaiting track")
+            const t = await tp
+            log("media", "got track")
+            this.publish_track(t)
+        }
+
+        mic_toggle.addEventListener("click", () => create(mic_toggle, this.create_mic_track()))
+        camera_toggle.addEventListener("click", () => create(camera_toggle, this.create_camera_track()))
+        screen_toggle.addEventListener("click", () => create(screen_toggle, this.create_screencast_track()))
+
+        const el = document.createElement("div")
+        el.classList.add("local-controls")
+        el.append(mic_toggle, camera_toggle, screen_toggle)
+        document.body.append(el)
+    }
+
+    async create_camera_track() {
+        log("media", "requesting user media (camera)")
+        const user_media = await window.navigator.mediaDevices.getUserMedia({
+            video: { facingMode: { ideal: PREFS.camera_facing_mode } }
+        })
+        return new TrackHandle(user_media.getVideoTracks()[0], true)
+    }
+
+    async create_screencast_track() {
+        log("media", "requesting user media (screen)")
+        const user_media = await window.navigator.mediaDevices.getDisplayMedia({ video: true })
+        return new TrackHandle(user_media.getVideoTracks()[0], true)
+    }
+
+    async create_mic_track() {
+        log("media", "requesting user media (audio)")
+        const audio_contraints = PREFS.rnnoise ? {
+            channelCount: { ideal: 1 },
+            noiseSuppression: { ideal: false },
+            echoCancellation: { ideal: true },
+            autoGainControl: { ideal: true },
+        } : {
+            channelCount: { ideal: 1 },
+            noiseSuppression: { ideal: false },
+            echoCancellation: { ideal: true },
+            autoGainControl: { ideal: true },
+        };
+
+        const user_media = await window.navigator.mediaDevices.getUserMedia({ audio: audio_contraints })
+        const context = new AudioContext()
+        const source = context.createMediaStreamSource(user_media)
+        const destination = context.createMediaStreamDestination()
+        const gain = context.createGain()
+        gain.gain.value = this.default_gain
+        this.mic_gain = gain
+
+        let rnnoise: RNNoiseNode;
+        if (PREFS.rnnoise) {
+            rnnoise = await get_rnnoise_node(context)
+            source.connect(rnnoise)
+            rnnoise.connect(gain)
+        } else {
+            source.connect(gain)
+        }
+        gain.connect(destination)
+
+        const t = new TrackHandle(destination.stream.getAudioTracks()[0], true)
+
+        t.addEventListener("ended", () => {
+            user_media.getTracks().forEach(t => t.stop())
+            source.disconnect()
+            if (rnnoise) rnnoise.disconnect()
+            gain.disconnect()
+            destination.disconnect()
+            this.mic_gain = undefined
+        })
+
+        return t
+    }
+}