From 8143409ada4ad802b191f67423b84fbeace63856 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Mon, 7 Jun 2021 01:19:57 -0400 Subject: [PATCH] moderate endpoint exposed --- src/http_server/endpoints/w2l/delete_w2l_template.rs | 0 src/http_server/endpoints/w2l/mod.rs | 1 + .../endpoints/w2l/set_w2l_template_mod_approval.rs | 6 +++--- src/main.rs | 6 ++++++ 4 files changed, 10 insertions(+), 3 deletions(-) create mode 100644 src/http_server/endpoints/w2l/delete_w2l_template.rs diff --git a/src/http_server/endpoints/w2l/delete_w2l_template.rs b/src/http_server/endpoints/w2l/delete_w2l_template.rs new file mode 100644 index 0000000000..e69de29bb2 diff --git a/src/http_server/endpoints/w2l/mod.rs b/src/http_server/endpoints/w2l/mod.rs index 56b37f8cfb..9eb955a1d0 100644 --- a/src/http_server/endpoints/w2l/mod.rs +++ b/src/http_server/endpoints/w2l/mod.rs @@ -1,3 +1,4 @@ +pub mod delete_w2l_template; pub mod enqueue_infer_w2l; pub mod enqueue_infer_w2l_with_uploads; pub mod enqueue_upload_w2l_template; diff --git a/src/http_server/endpoints/w2l/set_w2l_template_mod_approval.rs b/src/http_server/endpoints/w2l/set_w2l_template_mod_approval.rs index 3d3c85b9a8..08aa71c0f6 100644 --- a/src/http_server/endpoints/w2l/set_w2l_template_mod_approval.rs +++ b/src/http_server/endpoints/w2l/set_w2l_template_mod_approval.rs @@ -25,7 +25,7 @@ use std::sync::Arc; /// For the URL PathInfo #[derive(Deserialize)] pub struct SetW2lTemplateModApprovalPathInfo { - w2l_template_token: String, + slug: String, } #[derive(Deserialize)] @@ -115,7 +115,7 @@ pub async fn set_w2l_template_mod_approval_handler( } let template_query_result = select_w2l_template_by_token( - &path.w2l_template_token, + &path.slug, true, // Only mods can perform this action &server_state.mysql_pool, ).await; @@ -144,7 +144,7 @@ LIMIT 1 &request.is_approved, &user_session.user_token, - &path.w2l_template_token, + &path.slug, ) .execute(&server_state.mysql_pool) .await; diff --git a/src/main.rs b/src/main.rs index 24cc135f82..b7f18bacd5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -68,6 +68,7 @@ use crate::http_server::endpoints::w2l::get_w2l_inference_job_status::get_w2l_in use crate::http_server::endpoints::tts::get_tts_upload_model_job_status::get_tts_upload_model_job_status_handler; use crate::http_server::endpoints::w2l::get_w2l_upload_template_job_status::get_w2l_upload_template_job_status_handler; use crate::http_server::endpoints::users::edit_profile::edit_profile_handler; +use crate::http_server::endpoints::w2l::set_w2l_template_mod_approval::set_w2l_template_mod_approval_handler; // TODO TODO TODO TODO // TODO TODO TODO TODO @@ -329,6 +330,11 @@ pub async fn serve(server_state: ServerState) -> AnyhowResult<()> .route(web::get().to(get_w2l_template_handler)) .route(web::head().to(|| HttpResponse::Ok())) ) + .service( + web::resource("/template/{slug}/moderate") + .route(web::post().to(set_w2l_template_mod_approval_handler)) + .route(web::head().to(|| HttpResponse::Ok())) + ) .service( web::resource("/result/{token}") .route(web::get().to(get_w2l_inference_result_handler))