Add TTS

Author: rikhuijzer

.github/workflows/Test.yml

@@ -23,4 +23,11 @@ jobs:
         with:
           prefix-key: 'test'
 
+      - run: |
+          echo "OPENAI_KEY=${{ secrets.OPENAI_KEY }}" > test.env
+          echo "DEEPINFRA_KEY=${{ secrets.DEEPINFRA_KEY }}" >> test.env
+
       - run: cargo test --all-features
+
+      - name: Cleanup before Post Run
+        run: rm test.env

.gitignore

@@ -2,3 +2,4 @@
 **/*.rs.bk
 ata.toml
 Cargo.lock
+test.env

Cargo.toml

@@ -11,9 +11,9 @@ tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
 transformrs = { git = "https://github.com/rikhuijzer/transformrs.git", rev = "06b759f" }
 anyhow = "1"
 
-
-
 [dev-dependencies]
 assert_cmd = "2"
 predicates = "3"
 pretty_assertions = "1"
+tempfile = "3"
+toml = "0.8"

src/main.rs

@@ -1,5 +1,7 @@
 use clap::Parser;
+use std::fs::File;
 use std::io::Read;
+use std::io::Write;
 use transformrs::Key;
 use transformrs::Provider;
 
@@ -24,7 +26,6 @@ struct Arguments {
 }
 
 enum Task {
-    Chat,
     TTS,
 }
 
@@ -53,7 +54,6 @@ fn default_voice(provider: &Provider) -> Option<String> {
 fn default_model(provider: &Provider, task: &Task) -> Option<String> {
     match provider {
         Provider::OpenAI => match task {
-            Task::Chat => Some("gpt-4o".to_string()),
             Task::TTS => Some("tts-1".to_string()),
         },
         _ => None,
@@ -67,7 +67,6 @@ async fn main() {
     let mut input = String::new();
     std::io::stdin().read_to_string(&mut input).unwrap();
 
-    // TODO: add logic to transformrs to just get the key without .env file.
     let keys = transformrs::load_keys(".env");
     let key = find_single_key(keys);
     let provider = key.provider.clone();
@@ -82,6 +81,11 @@ async fn main() {
             .unwrap()
             .structured()
             .unwrap();
+        let bytes = resp.audio.clone();
+        let ext = resp.file_format;
+        let filename = format!("output.{ext}");
+        let mut file = File::create(filename).unwrap();
+        file.write_all(&bytes).unwrap();
     } else {
         error_and_exit("No action specified.");
     }

tests/cli.rs

@@ -1,11 +1,15 @@
-use assert_cmd::prelude::*;
+use assert_cmd::Command;
 use predicates::prelude::*;
-use std::process::Command;
+use std::io::BufRead;
+use transformrs::Provider;
+
+fn ata() -> Command {
+    Command::cargo_bin("ata").unwrap()
+}
 
 #[test]
 fn unexpected_argument() -> Result<(), Box<dyn std::error::Error>> {
-    let mut cmd = Command::cargo_bin("ata")?;
-
+    let mut cmd = ata();
     cmd.arg("foobar");
     cmd.assert()
         .failure()
@@ -16,12 +20,54 @@ fn unexpected_argument() -> Result<(), Box<dyn std::error::Error>> {
 
 #[test]
 fn help() -> Result<(), Box<dyn std::error::Error>> {
-    let mut cmd = Command::cargo_bin("ata")?;
-
+    let mut cmd = ata();
     cmd.arg("--help");
     cmd.assert()
         .success()
         .stdout(predicate::str::contains("Usage: ata"));
 
     Ok(())
 }
+
+/// Load a key from the local .env file.
+///
+/// This is used for testing only. Expects the .env file to contain keys for providers in the following format:
+///
+/// ```
+/// DEEPINFRA_KEY="<KEY>"
+/// OPENAI_KEY="<KEY>"
+/// ```
+fn load_key(provider: &Provider) -> String {
+    fn finder(line: &Result<String, std::io::Error>, provider: &Provider) -> bool {
+        line.as_ref()
+            .unwrap()
+            .starts_with(&provider.key_name())
+    }
+    let path = std::path::Path::new("test.env");
+    let file = std::fs::File::open(path).expect("Failed to open .env file");
+    let reader = std::io::BufReader::new(file);
+    let mut lines = reader.lines();
+    let key = lines
+        .find(|line| finder(line, provider))
+        .unwrap()
+        .unwrap();
+    key.split("=").nth(1).unwrap().to_string()
+}
+
+#[test]
+fn tts_no_args() -> Result<(), Box<dyn std::error::Error>> {
+    let dir = tempfile::tempdir().unwrap();
+    let mut cmd = ata();
+    let key = load_key(&Provider::OpenAI);
+    cmd.arg("--tts")
+        .env("OPENAI_KEY", key)
+        .write_stdin("Hello world")
+        .current_dir(&dir)
+        .assert()
+        .success();
+
+    let path = dir.path().join("output.mp3");
+    assert!(path.exists());
+
+    Ok(())
+}