Add MediaStream.addTrack() and MediaStream.removeTrack()

1 files changed, 15 insertions, 0 deletions

diff --git a/components/script/dom/mediastream.rs b/components/script/dom/mediastream.rs
index 395abd95ab6..38a9ce5f652 100644
--- a/components/script/dom/mediastream.rs
+++ b/components/script/dom/mediastream.rs
@@ -83,4 +83,19 @@ impl MediaStreamMethods for MediaStream {
             .find(|x| x.id().id().to_string() == &*id)
             .map(|x| DomRoot::from_ref(&**x))
     }
+
+    /// https://w3c.github.io/mediacapture-main/#dom-mediastream-addtrack
+    fn AddTrack(&self, track: &MediaStreamTrack) {
+        let existing = self.tracks.borrow().iter().find(|x| *x == &track).is_some();
+
+        if existing {
+            return;
+        }
+        self.add_track(track)
+    }
+
+    /// https://w3c.github.io/mediacapture-main/#dom-mediastream-removetrack
+    fn RemoveTrack(&self, track: &MediaStreamTrack) {
+        self.tracks.borrow_mut().retain(|x| *x != track);
+    }
 }