1 files changed, 51 insertions, 0 deletions
diff --git a/dom/media/MediaStreamWindowCapturer.h b/dom/media/MediaStreamWindowCapturer.h
new file mode 100644
index 0000000000..8a6695ed43
--- /dev/null
+++ b/dom/media/MediaStreamWindowCapturer.h
@@ -0,0 +1,51 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
+/* 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 http://mozilla.org/MPL/2.0/. */
+
+#ifndef MediaStreamWindowCapturer_h
+#define MediaStreamWindowCapturer_h
+
+#include "DOMMediaStream.h"
+
+namespace mozilla {
+namespace dom {
+class AudioStreamTrack;
+class MediaStreamTrack;
+}  // namespace dom
+
+class MediaInputPort;
+
+/**
+ * Given a DOMMediaStream and a window id, this class will pipe the audio from
+ * all live audio tracks in the stream to the MediaTrackGraph's window capture
+ * mechanism.
+ */
+class MediaStreamWindowCapturer : public DOMMediaStream::TrackListener {
+ public:
+  MediaStreamWindowCapturer(DOMMediaStream* aStream, uint64_t aWindowId);
+  ~MediaStreamWindowCapturer();
+
+  void NotifyTrackAdded(const RefPtr<dom::MediaStreamTrack>& aTrack) override;
+  void NotifyTrackRemoved(const RefPtr<dom::MediaStreamTrack>& aTrack) override;
+
+  struct CapturedTrack {
+    CapturedTrack(dom::MediaStreamTrack* aTrack, uint64_t aWindowID);
+    ~CapturedTrack();
+
+    const WeakPtr<dom::MediaStreamTrack> mTrack;
+    const RefPtr<MediaInputPort> mPort;
+  };
+
+  const WeakPtr<DOMMediaStream> mStream;
+  const uint64_t mWindowId;
+
+ protected:
+  void AddTrack(dom::AudioStreamTrack* aTrack);
+  void RemoveTrack(dom::AudioStreamTrack* aTrack);
+
+  nsTArray<UniquePtr<CapturedTrack>> mTracks;
+};
+}  // namespace mozilla
+
+#endif /* MediaStreamWindowCapturer_h */