diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/audio.rs | 86 | ||||
| -rw-r--r-- | src/config.rs | 3 | ||||
| -rw-r--r-- | src/graphql/types/alert.rs | 40 | ||||
| -rw-r--r-- | src/main.rs | 4 |
4 files changed, 132 insertions, 1 deletions
diff --git a/src/audio.rs b/src/audio.rs new file mode 100644 index 0000000..64ee223 --- /dev/null +++ b/src/audio.rs @@ -0,0 +1,86 @@ +use crate::config::CONFIG; +use axum::{ + extract::Path, + http::{HeaderMap, HeaderName, HeaderValue, StatusCode}, +}; +use reqwest::header::AUTHORIZATION; +use std::{ + fs::{self, File}, + io::Write, + path::Path as StdPath, +}; + +/// Create a new sound from a text +pub async fn tts(text: String, filename: String) -> Result<(), String> { + let url = "https://api.v7.unrealspeech.com/stream"; + let api_key = format!("Bearer {}", CONFIG.unrealspeech_token); + let filepath = format!("./assets/sounds/{}", filename); + + // Request JSON body + let body = serde_json::json!({ + "Text": text, + "VoiceId": "Will", + "Bitrate": "192k", + "Speed": "0", + "Pitch": "0.92", + "Codec": "libmp3lame", + }); + + // Send POST request + let client = reqwest::Client::new(); + let response = client + .post(url) + .header(AUTHORIZATION, api_key) + .json(&body) + .send() + .await + .unwrap(); + + // Check for successful response + if response.status().is_success() { + let mut file = File::create(filepath).unwrap(); + let content = response.bytes().await.unwrap(); + let _ = file.write_all(&content); + Ok(()) + } else { + Err(format!("Failed to fetch the audio: {}", response.status())) + } +} + +/// Axum endpoint which shows files +pub async fn show_file( + Path(id): Path<String>, +) -> Result<(HeaderMap, Vec<u8>), (StatusCode, String)> { + let index = id.find('.').unwrap_or(usize::MAX); + + let mut ext_name = "xxx"; + if index != usize::MAX { + ext_name = &id[index + 1..]; + } + + let mut headers = HeaderMap::new(); + + if ["mp3"].contains(&ext_name) { + let content_type = "audio/mpeg"; + headers.insert( + HeaderName::from_static("content-type"), + HeaderValue::from_str(content_type).unwrap(), + ); + } + + let file_name = format!("./assets/sounds/{}", id); + let file_path = StdPath::new(&file_name); + + if !file_path.exists() { + return Err((StatusCode::NOT_FOUND, "File not found".to_string())); + } + + // Read the file and return its content + match fs::read(file_path) { + Ok(file_content) => Ok((headers, file_content)), + Err(_) => Err(( + StatusCode::INTERNAL_SERVER_ERROR, + "Failed to read file".to_string(), + )), + } +} diff --git a/src/config.rs b/src/config.rs index a703010..763b852 100644 --- a/src/config.rs +++ b/src/config.rs @@ -19,6 +19,9 @@ pub struct Configuration { /// Token used by Expo API to send a notification pub expo_access_token: String, + + /// Token used for text-to-speach API + pub unrealspeech_token: String, } impl Configuration { diff --git a/src/graphql/types/alert.rs b/src/graphql/types/alert.rs index 6be723c..2ee0087 100644 --- a/src/graphql/types/alert.rs +++ b/src/graphql/types/alert.rs @@ -132,6 +132,8 @@ pub mod query { } pub mod mutations { + use crate::audio; + use super::*; /// Create a new alert @@ -330,6 +332,44 @@ pub mod mutations { .await .unwrap(); + if let Err(e) = audio::tts( + alert.text1.clone(), + format!("alert-{}-text-1.mp3", alert.id), + ) + .await + { + eprintln!( + "Error for `{}`: {}", + format!("alert-{}-text-1.mp3", alert.id), + e + ); + } + + if let Err(e) = audio::tts( + alert.text2.clone(), + format!("alert-{}-text-2.mp3", alert.id), + ) + .await + { + eprintln!( + "Error for `{}`: {}", + format!("alert-{}-text-2.mp3", alert.id), + e + ); + } + if let Err(e) = audio::tts( + alert.text3.clone(), + format!("alert-{}-text-3.mp3", alert.id), + ) + .await + { + eprintln!( + "Error for `{}`: {}", + format!("alert-{}-text-3.mp3", alert.id), + e + ); + } + Ok(alert) } } diff --git a/src/main.rs b/src/main.rs index 33718c1..940bb5c 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,4 +1,5 @@ #![doc = include_str!("../README.md")] +mod audio; mod config; mod db; mod errors; @@ -13,7 +14,7 @@ use crate::config::CONFIG; use async_graphql::{EmptySubscription, Schema}; use axum::{ http::{header, Method, Request}, - routing::post, + routing::{get, post}, Extension, Router, }; use tokio::net::TcpListener; @@ -45,6 +46,7 @@ async fn create_app() -> Router { .finish(); Router::new() + .route("/assets/sounds/:id", get(audio::show_file)) .route( "/graphql", post(graphql::routes::graphql_handler).layer(Extension(schema.clone())), |