1 files changed, 14 insertions, 3 deletions
diff --git a/source/client/track_handle.ts b/source/client/track_handle.ts
index bf3858f..dd96936 100644
--- a/source/client/track_handle.ts
+++ b/source/client/track_handle.ts
@@ -1,14 +1,25 @@
 
 export class TrackHandle extends EventTarget {
-    constructor(public track: MediaStreamTrack) {
+    constructor(
+        public track: MediaStreamTrack,
+        public local = false
+    ) {
         super()
         track.onended = () => this.dispatchEvent(new CustomEvent("ended"))
-        track.onmute = () => this.dispatchEvent(new CustomEvent("mute"))
-        track.onunmute = () => this.dispatchEvent(new CustomEvent("unmute"))
+        // TODO research how onmute and onunmute behave
+        track.onmute = () => this.dispatchEvent(new CustomEvent("ended")) // onmute seems to be called when the remote ends the track
+        track.onunmute = () => this.dispatchEvent(new CustomEvent("started"))
+
+        this.addEventListener("ended", () => {
+            // drop all references to help gc
+            track.onunmute = track.onmute = track.onended = null
+        })
     }
 
     get kind() { return this.track.kind }
     get label() { return this.track.label }
     get muted() { return this.track.muted }
     get id() { return this.track.id }
+
+    end() { this.track.stop(); this.dispatchEvent(new CustomEvent("ended")) }
 }