From 037ef691255a81eef14f1712c7f8f03773abba87 Mon Sep 17 00:00:00 2001
From: Ryan Cao <70191398+ryanccn@users.noreply.github.com>
Date: Sat, 28 Oct 2023 15:41:54 +0800
Subject: [PATCH] feat: allow patching screenshare with any audio device

---
 src/renderer/components/ScreenSharePicker.tsx | 42 +++++++++++++++++++
 src/renderer/patches/screenShareAudio.ts      | 22 +++++++++-
 2 files changed, 62 insertions(+), 2 deletions(-)

diff --git a/src/renderer/components/ScreenSharePicker.tsx b/src/renderer/components/ScreenSharePicker.tsx
index cd2fc6cba..47c0fc475 100644
--- a/src/renderer/components/ScreenSharePicker.tsx
+++ b/src/renderer/components/ScreenSharePicker.tsx
@@ -20,6 +20,7 @@ import {
     useState
 } from "@vencord/types/webpack/common";
 import type { Dispatch, SetStateAction } from "react";
+import { patchAudioWithDevice } from "renderer/patches/screenShareAudio";
 import { addPatch } from "renderer/patches/shared";
 import { isLinux, isWindows } from "renderer/utils";
 
@@ -36,6 +37,7 @@ interface StreamSettings {
     fps: StreamFps;
     audio: boolean;
     audioSource?: string;
+    audioDevice?: string;
 }
 
 export interface StreamPick extends StreamSettings {
@@ -109,6 +111,9 @@ export function openScreenSharePicker(screens: Source[], skipPicker: boolean) {
                                 await VesktopNative.virtmic.start([v.audioSource]);
                             }
                         }
+
+                        patchAudioWithDevice(v.audioDevice);
+
                         resolve(v);
                     }}
                     close={() => {
@@ -211,6 +216,11 @@ function StreamSettings({
                     </section>
                 </div>
 
+                <AudioSourceAnyDevice
+                    audioDevice={settings.audioDevice}
+                    setAudioDevice={source => setSettings(s => ({ ...s, audioDevice: source }))}
+                />
+
                 {isWindows && (
                     <Switch
                         value={settings.audio}
@@ -233,6 +243,38 @@ function StreamSettings({
     );
 }
 
+function AudioSourceAnyDevice({
+    audioDevice,
+    setAudioDevice
+}: {
+    audioDevice?: string;
+    setAudioDevice(s: string): void;
+}) {
+    const [sources, _, loading] = useAwaiter(
+        () =>
+            navigator.mediaDevices
+                .enumerateDevices()
+                .then(devices => devices.filter(device => device.kind === "audioinput")),
+        { fallbackValue: [] }
+    );
+
+    return (
+        <section>
+            <Forms.FormTitle>Audio</Forms.FormTitle>
+            {loading && <Forms.FormTitle>Loading audio devices...</Forms.FormTitle>}
+
+            {sources.length > 0 && (
+                <Select
+                    options={sources.map((s, idx) => ({ label: s.label, value: s.deviceId, default: idx === 0 }))}
+                    isSelected={s => s === audioDevice}
+                    select={setAudioDevice}
+                    serialize={String}
+                />
+            )}
+        </section>
+    );
+}
+
 function AudioSourcePickerLinux({
     audioSource,
     setAudioSource
diff --git a/src/renderer/patches/screenShareAudio.ts b/src/renderer/patches/screenShareAudio.ts
index 3b926c392..ce893d0d5 100644
--- a/src/renderer/patches/screenShareAudio.ts
+++ b/src/renderer/patches/screenShareAudio.ts
@@ -6,9 +6,27 @@
 
 import { isLinux } from "renderer/utils";
 
-if (isLinux) {
-    const original = navigator.mediaDevices.getDisplayMedia;
+const original = navigator.mediaDevices.getDisplayMedia;
+
+export const patchAudioWithDevice = (deviceId?: string) => {
+    if (!deviceId) {
+        navigator.mediaDevices.getDisplayMedia = original;
+        return;
+    }
+
+    navigator.mediaDevices.getDisplayMedia = async function (opts) {
+        const stream = await original.call(this, opts);
+        const audio = await navigator.mediaDevices.getUserMedia({ audio: { deviceId: { exact: deviceId } } });
+        const tracks = audio.getAudioTracks();
 
+        tracks.forEach(t => stream.addTrack(t));
+        console.log(`Patched stream ${stream.id} with ${tracks.length} audio tracks from ${deviceId}`);
+
+        return stream;
+    };
+};
+
+if (isLinux) {
     async function getVirtmic() {
         try {
             const devices = await navigator.mediaDevices.enumerateDevices();