From bc2267fb7769e2a52a5518af97a26f39274744fa Mon Sep 17 00:00:00 2001
From: Manish Goregaokar <manishsmail@gmail.com>
Date: Wed, 1 Jul 2020 17:52:57 -0700
Subject: [PATCH] Add MediaStreamAudioSourceNode

---
 .../script/dom/mediastreamaudiosourcenode.rs  | 77 +++++++++++++++++++
 components/script/dom/mod.rs                  |  1 +
 .../webidls/MediaStreamAudioSourceNode.webidl | 17 ++++
 3 files changed, 95 insertions(+)
 create mode 100644 components/script/dom/mediastreamaudiosourcenode.rs
 create mode 100644 components/script/dom/webidls/MediaStreamAudioSourceNode.webidl

diff --git a/components/script/dom/mediastreamaudiosourcenode.rs b/components/script/dom/mediastreamaudiosourcenode.rs
new file mode 100644
index 00000000000..dc0803adec3
--- /dev/null
+++ b/components/script/dom/mediastreamaudiosourcenode.rs
@@ -0,0 +1,77 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+use crate::dom::audiocontext::AudioContext;
+use crate::dom::audionode::AudioNode;
+use crate::dom::bindings::codegen::Bindings::MediaStreamAudioSourceNodeBinding::{
+    MediaStreamAudioSourceNodeMethods, MediaStreamAudioSourceOptions,
+};
+use crate::dom::bindings::error::{Error, Fallible};
+use crate::dom::bindings::inheritance::Castable;
+use crate::dom::bindings::reflector::reflect_dom_object;
+use crate::dom::bindings::root::{Dom, DomRoot};
+use crate::dom::mediastream::MediaStream;
+use crate::dom::window::Window;
+use dom_struct::dom_struct;
+use servo_media::audio::node::AudioNodeInit;
+use servo_media::streams::MediaStreamType;
+
+#[dom_struct]
+pub struct MediaStreamAudioSourceNode {
+    node: AudioNode,
+    stream: Dom<MediaStream>,
+}
+
+impl MediaStreamAudioSourceNode {
+    #[allow(unrooted_must_root)]
+    pub fn new_inherited(
+        context: &AudioContext,
+        options: &MediaStreamAudioSourceOptions,
+    ) -> Fallible<MediaStreamAudioSourceNode> {
+        let track = options
+            .mediaStream
+            .get_tracks()
+            .iter()
+            .find(|t| t.ty() == MediaStreamType::Audio)
+            .ok_or(Error::InvalidState)?
+            .id();
+        let node = AudioNode::new_inherited(
+            AudioNodeInit::MediaStreamSourceNode(track),
+            &context.upcast(),
+            Default::default(),
+            0, // inputs
+            1, // outputs
+        )?;
+        Ok(MediaStreamAudioSourceNode {
+            node,
+            stream: Dom::from_ref(&options.mediaStream),
+        })
+    }
+
+    #[allow(unrooted_must_root)]
+    pub fn new(
+        window: &Window,
+        context: &AudioContext,
+        options: &MediaStreamAudioSourceOptions,
+    ) -> Fallible<DomRoot<MediaStreamAudioSourceNode>> {
+        let node = MediaStreamAudioSourceNode::new_inherited(context, options)?;
+        Ok(reflect_dom_object(Box::new(node), window))
+    }
+
+    #[allow(non_snake_case)]
+    pub fn Constructor(
+        window: &Window,
+        context: &AudioContext,
+        options: &MediaStreamAudioSourceOptions,
+    ) -> Fallible<DomRoot<MediaStreamAudioSourceNode>> {
+        MediaStreamAudioSourceNode::new(window, context, options)
+    }
+}
+
+impl MediaStreamAudioSourceNodeMethods for MediaStreamAudioSourceNode {
+    /// https://webaudio.github.io/web-audio-api/#dom-MediaStreamAudioSourceNode-stream
+    fn MediaStream(&self) -> DomRoot<MediaStream> {
+        DomRoot::from_ref(&self.stream)
+    }
+}
diff --git a/components/script/dom/mod.rs b/components/script/dom/mod.rs
index c9f9cf64221..7b814df775f 100644
--- a/components/script/dom/mod.rs
+++ b/components/script/dom/mod.rs
@@ -436,6 +436,7 @@ pub mod mediaquerylistevent;
 pub mod mediasession;
 pub mod mediastream;
 pub mod mediastreamaudiodestinationnode;
+pub mod mediastreamaudiosourcenode;
 pub mod mediastreamtrack;
 pub mod messagechannel;
 pub mod messageevent;
diff --git a/components/script/dom/webidls/MediaStreamAudioSourceNode.webidl b/components/script/dom/webidls/MediaStreamAudioSourceNode.webidl
new file mode 100644
index 00000000000..655e2443081
--- /dev/null
+++ b/components/script/dom/webidls/MediaStreamAudioSourceNode.webidl
@@ -0,0 +1,17 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+/*
+ * The origin of this IDL file is
+ * https://webaudio.github.io/web-audio-api/#mediastreamaudiosourcenode
+ */
+
+dictionary MediaStreamAudioSourceOptions {
+  required MediaStream mediaStream;
+};
+
+[Exposed=Window]
+interface MediaStreamAudioSourceNode : AudioNode {
+  [Throws] constructor (AudioContext context, MediaStreamAudioSourceOptions options);
+  [SameObject] readonly attribute MediaStream mediaStream;
+};