TTS upload job status

src/http_server/endpoints/tts/get_tts_upload_model_job_status.rs

@@ -0,0 +1,164 @@
+use actix_http::Error;
+use actix_http::http::header;
+use actix_web::cookie::Cookie;
+use actix_web::dev::HttpResponseBuilder;
+use actix_web::error::ResponseError;
+use actix_web::http::StatusCode;
+use actix_web::web::Path;
+use actix_web::{Responder, web, HttpResponse, error, HttpRequest, HttpMessage};
+use chrono::{DateTime, Utc};
+use crate::AnyhowResult;
+use crate::common_queries::sessions::create_session_for_user;
+use crate::http_server::endpoints::users::create_account::CreateAccountError::{BadInput, ServerError, UsernameTaken, EmailTaken};
+use crate::http_server::endpoints::users::login::LoginSuccessResponse;
+use crate::http_server::web_utils::ip_address::get_request_ip;
+use crate::http_server::web_utils::session_checker::SessionRecord;
+use crate::server_state::ServerState;
+use crate::util::random_crockford_token::random_crockford_token;
+use crate::validations::passwords::validate_passwords;
+use crate::validations::username::validate_username;
+use derive_more::{Display, Error};
+use log::{info, warn, log};
+use regex::Regex;
+use sqlx::MySqlPool;
+use sqlx::error::DatabaseError;
+use sqlx::error::Error::Database;
+use sqlx::mysql::MySqlDatabaseError;
+use std::sync::Arc;
+
+/// For the URL PathInfo
+#[derive(Deserialize)]
+pub struct GetTtsUploadModelStatusPathInfo {
+  token: String,
+}
+
+#[derive(Serialize)]
+pub struct TtsUploadModelJobStatusForResponse {
+  pub job_token: String,
+
+  pub status: String,
+  pub maybe_model_token: Option<String>,
+
+  pub created_at: DateTime<Utc>,
+  pub updated_at: DateTime<Utc>,
+}
+
+#[derive(Serialize)]
+pub struct GetTtsUploadModelStatusSuccessResponse {
+  pub success: bool,
+  pub state: TtsUploadModelJobStatusForResponse,
+}
+
+#[derive(Serialize)]
+pub struct ErrorResponse {
+  pub success: bool,
+  pub error_reason: String,
+}
+
+#[derive(Debug, Display)]
+pub enum GetTtsUploadModelStatusError {
+  ServerError,
+}
+
+#[derive(Serialize)]
+pub struct TtsUploadModelJobStatusRecord {
+  pub job_token: String,
+
+  pub status: String,
+  pub maybe_model_token: Option<String>,
+
+  pub created_at: DateTime<Utc>,
+  pub updated_at: DateTime<Utc>,
+}
+
+impl ResponseError for GetTtsUploadModelStatusError {
+  fn status_code(&self) -> StatusCode {
+    match *self {
+      GetTtsUploadModelStatusError::ServerError=> StatusCode::INTERNAL_SERVER_ERROR,
+    }
+  }
+
+  fn error_response(&self) -> HttpResponse {
+    let error_reason = match self {
+      GetTtsUploadModelStatusError::ServerError => "server error".to_string(),
+    };
+
+    let response = ErrorResponse {
+      success: false,
+      error_reason,
+    };
+
+    let body = match serde_json::to_string(&response) {
+      Ok(json) => json,
+      Err(_) => "{}".to_string(),
+    };
+
+    HttpResponseBuilder::new(self.status_code())
+        .set_header(header::CONTENT_TYPE, "application/json")
+        .body(body)
+  }
+}
+
+pub async fn get_tts_upload_model_job_status_handler(
+  http_request: HttpRequest,
+  path: Path<GetTtsUploadModelStatusPathInfo>,
+  server_state: web::Data<Arc<ServerState>>) -> Result<HttpResponse, GetTtsUploadModelStatusError>
+{
+  // NB: Lookup failure is Err(RowNotFound).
+  // NB: Since this is publicly exposed, we don't query sensitive data.
+  let maybe_status = sqlx::query_as!(
+      TtsUploadModelJobStatusRecord,
+        r#"
+SELECT
+    jobs.token as job_token,
+
+    jobs.status,
+    jobs.on_success_result_token as maybe_model_token,
+
+    jobs.created_at,
+    jobs.updated_at
+
+FROM tts_model_upload_jobs as jobs
+
+WHERE jobs.token = ?
+        "#,
+      &path.token
+    )
+      .fetch_one(&server_state.mysql_pool)
+      .await; // TODO: This will return error if it doesn't exist
+
+  let record : TtsUploadModelJobStatusRecord = match maybe_status {
+    Ok(record) => record,
+    Err(err) => {
+      match err {
+        RowNotFound => {
+          return Err(GetTtsUploadModelStatusError::ServerError);
+        },
+        _ => {
+          warn!("tts template query error: {:?}", err);
+          return Err(GetTtsUploadModelStatusError::ServerError);
+        }
+      }
+    }
+  };
+
+  let model_for_response = TtsUploadModelJobStatusForResponse {
+    job_token: record.job_token.clone(),
+    status: record.status.clone(),
+    maybe_model_token: record.maybe_model_token.clone(),
+    created_at: record.created_at.clone(),
+    updated_at: record.updated_at.clone(),
+  };
+
+  let response = GetTtsUploadModelStatusSuccessResponse {
+    success: true,
+    state: model_for_response,
+  };
+
+  let body = serde_json::to_string(&response)
+      .map_err(|e| GetTtsUploadModelStatusError::ServerError)?;
+
+  Ok(HttpResponse::Ok()
+      .content_type("application/json")
+      .body(body))
+}

src/http_server/endpoints/tts/mod.rs

@@ -3,4 +3,5 @@ pub mod enqueue_upload_tts_model;
 pub mod get_tts_inference_job_status;
 pub mod get_tts_model;
 pub mod get_tts_result;
+pub mod get_tts_upload_model_job_status;
 pub mod list_tts_models;

src/main.rs

@@ -64,6 +64,7 @@ use crate::http_server::endpoints::tts::get_tts_result::get_tts_inference_result
 use crate::http_server::endpoints::tts::get_tts_model::get_tts_model_handler;
 use crate::http_server::endpoints::tts::get_tts_inference_job_status::get_tts_inference_job_status_handler;
 use crate::http_server::endpoints::w2l::get_w2l_inference_job_status::get_w2l_inference_job_status_handler;
+use crate::http_server::endpoints::tts::get_tts_upload_model_job_status::get_tts_upload_model_job_status_handler;
 
 // TODO TODO TODO TODO
 // TODO TODO TODO TODO
@@ -296,6 +297,11 @@ pub async fn serve(server_state: ServerState) -> AnyhowResult<()>
               .route(web::get().to(get_tts_inference_job_status_handler))
               .route(web::head().to(|| HttpResponse::Ok()))
           )
+          .service(
+          web::resource("/upload_model_job/{token}")
+              .route(web::get().to(get_tts_upload_model_job_status_handler))
+              .route(web::head().to(|| HttpResponse::Ok()))
+          )
       )
       // ==================== WAV2LIP ====================
       .service(