aboutsummaryrefslogtreecommitdiff
path: root/source/client/local_user.ts
blob: 7f2327e8be27347860c966acb6e7221505859236 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
import { parameter_bool, parameter_number, parameter_string } from "./helper";
import { log } from "./logger";
import { RemoteUser } from "./remote_user";
import { get_rnnoise_node } from "./rnnoise";
import { Room } from "./room";
import { User } from "./user";


export class LocalUser extends User {

    private audio_track?: MediaStreamTrack
    private video_track?: MediaStreamTrack
    private audio_disable_cleanup?: () => void

    mic_gain?: GainNode
    default_gain: number = parameter_number("mic_gain", 1)

    controls?: { audio: HTMLElement, video: HTMLElement, mute: HTMLElement }

    constructor(room: Room, name: string) {
        super(room, name)
        this.el.classList.add("local")
        this.local = true
        this.create_controls()
        if (parameter_bool("audio_enabled", false)) this.enable_audio()
        if (parameter_bool("video_enabled", false)) this.enable_video()
    }

    create_controls() {
        const audio_toggle = document.createElement("input")
        const video_toggle = document.createElement("input")
        const mute_toggle = document.createElement("input")
        audio_toggle.type = video_toggle.type = mute_toggle.type = "button"
        audio_toggle.value = "Audio"
        video_toggle.value = "Video"
        mute_toggle.value = "Mute"
        let audio = parameter_bool("audio_enabled", false),
            video = parameter_bool("video_enabled", false),
            mute = parameter_bool("video_enabled", false)


        audio_toggle.addEventListener("click", () => {
            audio = !audio
            if (audio) this.enable_audio()
            else this.disable_audio()
        })
        video_toggle.addEventListener("click", () => {
            video = !video
            if (video) this.enable_video()
            else this.disable_video()
        })
        mute_toggle.addEventListener("click", () => {
            mute = !mute
            this.mic_gain?.gain?.setValueAtTime(mute ? 0 : this.default_gain, 0)
            if (mute) this.controls?.mute.classList.add("enabled")
            else this.controls?.mute.classList.remove("enabled")
        })

        const el = document.createElement("div")
        el.classList.add("local-controls")
        el.append(audio_toggle, video_toggle, mute_toggle)
        this.controls = { video: video_toggle, audio: audio_toggle, mute: mute_toggle }
        document.body.append(el)
    }

    update_view_w() {
        this.update_view()
        if (this.stream.getAudioTracks().length > 0)
            this.controls?.audio.classList.add("enabled")
        else this.controls?.audio.classList.remove("enabled")

        if (this.stream.getVideoTracks().length > 0)
            this.controls?.video.classList.add("enabled")
        else this.controls?.video.classList.remove("enabled")
    }

    async add_initial_to_remote(ru: RemoteUser) {
        if (this.audio_track) ru.peer.addTrack(this.audio_track)
        if (this.video_track) ru.peer.addTrack(this.video_track)
    }

    async enable_video() {
        if (this.video_track) return
        log("media", "requesting user media (video)")
        const user_media = await window.navigator.mediaDevices.getUserMedia({ video: true })
        const t = this.video_track = user_media.getVideoTracks()[0]
        this.room.remote_users.forEach(u => u.peer.addTrack(t))
        this.stream.addTrack(t)
        this.update_view_w()
    }
    async enable_audio() {
        if (this.audio_track) return
        log("media", "requesting user media (audio)")

        const use_rnnoise = parameter_bool("rnnoise", true)
        const audio_contraints = use_rnnoise ? {
            channelCount: { ideal: 1 },
            noiseSuppression: { ideal: false },
            echoCancellation: { ideal: true },
            autoGainControl: { ideal: false },
        } : 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 (use_rnnoise) {
            rnnoise = await get_rnnoise_node(context)
            source.connect(rnnoise)
            rnnoise.connect(gain)
        } else {
            source.connect(gain)
        }
        gain.connect(destination)

        this.audio_disable_cleanup = () => {
            source.disconnect()
            if (rnnoise) rnnoise.disconnect()
            gain.disconnect()
            destination.disconnect()
            this.mic_gain = undefined
        }

        const t = destination.stream.getAudioTracks()[0]
        this.audio_track = t
        this.room.remote_users.forEach(u => u.peer.addTrack(t))
        this.stream.addTrack(t)
        this.update_view_w()
    }

    async disable_video() {
        if (!this.video_track) return
        this.room.remote_users.forEach(u => {
            u.peer.getSenders().forEach(s => {
                if (s.track == this.video_track) u.peer.removeTrack(s)
            })
        })
        this.stream.removeTrack(this.video_track)
        this.update_view_w()
        this.video_track = undefined
    }
    async disable_audio() {
        if (!this.audio_track) return
        if (this.audio_disable_cleanup) this.audio_disable_cleanup()
        this.room.remote_users.forEach(u => {
            u.peer.getSenders().forEach(s => {
                if (s.track == this.audio_track) u.peer.removeTrack(s)
            })
        })
        this.stream.removeTrack(this.audio_track)
        this.update_view_w()
        this.audio_track = undefined
    }
}