voice_conversion_for_inference

crates/lib/mysql_queries/src/queries/voice_conversion/results/get_voice_conversion_result_for_inference.rs

@@ -0,0 +1,179 @@
+// NB: Incrementally getting rid of build warnings...
+#![forbid(unused_imports)]
+#![forbid(unused_mut)]
+#![forbid(unused_variables)]
+
+use anyhow::anyhow;
+use chrono::{DateTime, Utc};
+use crate::helpers::boolean_converters::{nullable_i8_to_bool};
+use enums::common::visibility::Visibility;
+use errors::AnyhowResult;
+use log::warn;
+use sqlx::MySqlPool;
+use tokens::tokens::voice_conversion_results::VoiceConversionResultToken;
+use tokens::users::user::UserToken;
+
+// TODO(bt, 2023-09-07): I lazily copied this code from `get_tts_result` and didn't validate the query or the fields.
+//  Once this is used in production flows, this should be spot checked.
+
+pub struct VoiceConversionResult {
+  pub voice_conversion_result_token: VoiceConversionResultToken,
+
+  pub maybe_creator_user_token: Option<UserToken>,
+  pub maybe_creator_is_banned: bool,
+
+  pub public_bucket_hash: String,
+
+  pub creator_set_visibility: Visibility,
+
+  pub file_size_bytes: u32,
+  pub duration_millis: u32,
+
+  pub user_deleted_at: Option<DateTime<Utc>>,
+  pub mod_deleted_at: Option<DateTime<Utc>>,
+}
+
+
+#[derive(Serialize)]
+pub struct VoiceConversionResultRecordRaw {
+  pub voice_conversion_result_token: VoiceConversionResultToken,
+
+  pub maybe_creator_is_banned: Option<i8>,
+  pub maybe_creator_user_token: Option<UserToken>,
+
+  pub public_bucket_hash: String,
+
+  pub creator_set_visibility: String,
+
+  pub file_size_bytes: i32,
+  pub duration_millis: i32,
+
+  pub user_deleted_at: Option<DateTime<Utc>>,
+  pub mod_deleted_at: Option<DateTime<Utc>>,
+}
+
+pub async fn get_voice_conversion_result_for_inference(
+  voice_conversion_result_token: &str,
+  can_see_deleted: bool,
+  mysql_pool: &MySqlPool
+) -> AnyhowResult<Option<VoiceConversionResult>> {
+
+  let maybe_record = if can_see_deleted {
+    select_including_deleted(voice_conversion_result_token, mysql_pool).await
+  } else {
+    select_without_deleted(voice_conversion_result_token, mysql_pool).await
+  };
+
+  let ir : VoiceConversionResultRecordRaw = match maybe_record {
+    Ok(inference_result) => inference_result,
+    Err(ref err) => {
+      match err {
+        sqlx::Error::RowNotFound => {
+          warn!("tts result not found: {:?}", &err);
+          return Ok(None);
+        },
+        _ => {
+          warn!("tts result query error: {:?}", &err);
+          return Err(anyhow!("database error"));
+        }
+      }
+    }
+  };
+
+  let ir_for_response = VoiceConversionResult {
+    voice_conversion_result_token: ir.voice_conversion_result_token,
+
+    maybe_creator_user_token: ir.maybe_creator_user_token,
+
+    // NB: Fail open/public since we're already looking at it
+    maybe_creator_is_banned: nullable_i8_to_bool(ir.maybe_creator_is_banned, false),
+    public_bucket_hash: ir.public_bucket_hash,
+
+    creator_set_visibility: Visibility::from_str(&ir.creator_set_visibility)
+        .unwrap_or(Visibility::Public),
+
+    file_size_bytes: if ir.file_size_bytes > 0 { ir.file_size_bytes as u32 } else { 0 },
+    duration_millis: if ir.duration_millis > 0 { ir.duration_millis as u32 } else { 0 },
+
+    user_deleted_at: ir.user_deleted_at,
+    mod_deleted_at: ir.mod_deleted_at,
+  };
+
+  Ok(Some(ir_for_response))
+}
+
+async fn select_including_deleted(
+  voice_conversion_result_token: &str,
+  mysql_pool: &MySqlPool
+) -> Result<VoiceConversionResultRecordRaw, sqlx::Error> {
+  sqlx::query_as!(
+      VoiceConversionResultRecordRaw,
+        r#"
+SELECT
+    voice_conversion_results.token as `voice_conversion_result_token: tokens::tokens::voice_conversion_results::VoiceConversionResultToken`,
+
+    users.token as `maybe_creator_user_token: tokens::users::user::UserToken`,
+    users.is_banned as maybe_creator_is_banned,
+
+    voice_conversion_results.public_bucket_hash,
+
+    voice_conversion_results.creator_set_visibility,
+
+    voice_conversion_results.file_size_bytes,
+    voice_conversion_results.duration_millis,
+
+    voice_conversion_results.user_deleted_at,
+    voice_conversion_results.mod_deleted_at
+
+FROM voice_conversion_results
+LEFT OUTER JOIN voice_conversion_models
+  ON voice_conversion_results.model_token = voice_conversion_models.token
+LEFT OUTER JOIN users
+  ON voice_conversion_results.maybe_creator_user_token = users.token
+WHERE
+    voice_conversion_results.token = ?
+        "#,
+      voice_conversion_result_token
+    )
+      .fetch_one(mysql_pool)
+      .await // TODO: This will return error if it doesn't exist
+}
+
+async fn select_without_deleted(
+  voice_conversion_result_token: &str,
+  mysql_pool: &MySqlPool
+) -> Result<VoiceConversionResultRecordRaw, sqlx::Error> {
+  sqlx::query_as!(
+      VoiceConversionResultRecordRaw,
+        r#"
+SELECT
+    voice_conversion_results.token as `voice_conversion_result_token: tokens::tokens::voice_conversion_results::VoiceConversionResultToken`,
+
+    users.token as `maybe_creator_user_token: tokens::users::user::UserToken`,
+    users.is_banned as maybe_creator_is_banned,
+
+    voice_conversion_results.public_bucket_hash,
+
+    voice_conversion_results.creator_set_visibility,
+
+    voice_conversion_results.file_size_bytes,
+    voice_conversion_results.duration_millis,
+
+    voice_conversion_results.user_deleted_at,
+    voice_conversion_results.mod_deleted_at
+
+FROM voice_conversion_results
+LEFT OUTER JOIN voice_conversion_models
+  ON voice_conversion_results.model_token = voice_conversion_models.token
+LEFT OUTER JOIN users
+  ON voice_conversion_results.maybe_creator_user_token = users.token
+WHERE
+    voice_conversion_results.token = ?
+    AND voice_conversion_results.user_deleted_at IS NULL
+    AND voice_conversion_results.mod_deleted_at IS NULL
+        "#,
+      voice_conversion_result_token
+    )
+      .fetch_one(mysql_pool)
+      .await // TODO: This will return error if it doesn't exist
+}

crates/lib/mysql_queries/src/queries/voice_conversion/results/mod.rs

@@ -1,2 +1,3 @@
 pub mod get_voice_conversion_result;
+pub mod get_voice_conversion_result_for_inference;
 pub mod insert_voice_conversion_result;