add speaker support for gtts-cli | mpv (#88)

2 files changed, 41 insertions, 0 deletions

diff --git a/src/epy_reader/speakers/__init__.py b/src/epy_reader/speakers/__init__.py
index 078be31..4628714 100644
--- a/src/epy_reader/speakers/__init__.py
+++ b/src/epy_reader/speakers/__init__.py
@@ -2,8 +2,10 @@ __all__ = [
     "SpeakerBaseModel",
     "SpeakerMimic",
     "SpeakerPico",
+    "SpeakerGttsMPV"
 ]
 
 from epy_reader.speakers.base import SpeakerBaseModel
 from epy_reader.speakers.mimic import SpeakerMimic
 from epy_reader.speakers.pico import SpeakerPico
+from epy_reader.speakers.gtts_mpv import SpeakerGttsMPV
diff --git a/src/epy_reader/speakers/gtts_mpv.py b/src/epy_reader/speakers/gtts_mpv.py
new file mode 100644
index 0000000..a4406dd
--- /dev/null
+++ b/src/epy_reader/speakers/gtts_mpv.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+
+import shutil
+import subprocess
+
+from epy_reader.speakers.base import SpeakerBaseModel
+
+
+class SpeakerGttsMPV(SpeakerBaseModel):
+    cmd = "gtts-mpv"
+    available = bool(shutil.which("gtts-cli") and shutil.which("mpv"))
+
+    def speak(self, text: str) -> None:
+        self._gtts_process = subprocess.Popen(
+            ["gtts-cli", "-", *self.args],
+            text=True,
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+        )
+        self._mpv_process = subprocess.Popen(
+            ["mpv", "-"],
+            stdin=self._gtts_process.stdout,
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+        )
+        assert self._gtts_process.stdin
+        self._gtts_process.stdin.write(text)
+        self._gtts_process.stdin.close()
+
+    def is_done(self) -> bool:
+        return self._mpv_process.poll() is not None
+
+    def stop(self) -> None:
+        self._gtts_process.terminate()
+        self._mpv_process.terminate()
+
+    def cleanup(self) -> None:
+        pass