feat: support ffprobe in audio

Author: LordOfPolls

interactions/api/voice/audio.py

@@ -1,4 +1,5 @@
 import audioop
+import shutil
 import subprocess  # noqa: S404
 import threading
 import time
@@ -13,6 +14,10 @@
     "AudioVolume",
 )
 
+from interactions.client.const import get_logger
+from interactions.api.voice.opus import Encoder
+from interactions.client.utils import FastJson
+
 
 class AudioBuffer:
     def __init__(self) -> None:
@@ -62,6 +67,8 @@ class BaseAudio(ABC):
     """Does this audio data need encoding with opus?"""
     bitrate: Optional[int]
     """Optionally specify a specific bitrate to encode this audio data with"""
+    encoder: Optional[Encoder]
+    """The encoder to use for this audio data"""
 
     def __del__(self) -> None:
         self.cleanup()
@@ -135,10 +142,40 @@ def audio_complete(self) -> bool:
                 return False
         return True
 
-    def _create_process(self, *, block: bool = True) -> None:
+    def _create_process(self, *, block: bool = True, probe: bool = True) -> None:
         before = (
             self.ffmpeg_before_args if isinstance(self.ffmpeg_before_args, list) else self.ffmpeg_before_args.split()
         )
+
+        config = {
+            "sample_rate": 48000,
+            "bitrate": None,
+        }
+
+        if shutil.which("ffprobe") is not None and probe:
+            ffprobe_cmd = [
+                "ffprobe",
+                "-loglevel",
+                "quiet",
+                "-print_format",
+                "json",
+                "-show_streams",
+                "-select_streams",
+                "a:0",
+                self.source,
+            ]
+            raw_output = subprocess.check_output(ffprobe_cmd, stderr=subprocess.DEVNULL)
+            output = FastJson.loads(raw_output)
+
+            config["sample_rate"] = int(output["streams"][0]["sample_rate"])
+            config["bitrate"] = int(output["streams"][0]["bit_rate"])
+
+            get_logger().debug(f"Detected audio data for {self.source} - {config}")
+
+            if getattr(self, "bitrate", None) is None and self.encoder:
+                self.bitrate = int(config["bitrate"] / 1024)
+                self.encoder.set_bitrate(self.bitrate)
+
         after = self.ffmpeg_args if isinstance(self.ffmpeg_args, list) else self.ffmpeg_args.split()
         cmd = [
             "ffmpeg",
@@ -147,7 +184,7 @@ def _create_process(self, *, block: bool = True) -> None:
             "-f",
             "s16le",
             "-ar",
-            "48000",
+            str(config["sample_rate"]),
             "-ac",
             "2",
             "-loglevel",

interactions/api/voice/player.py

@@ -94,6 +94,7 @@ def run(self) -> None:
             # noinspection PyProtectedMember
             self.current_audio.volume = self.state._volume
 
+        self.current_audio.encoder = self._encoder
         self._encoder.set_bitrate(getattr(self.current_audio, "bitrate", self.state.channel.bitrate))
 
         self._stopped.clear()