Add file service

src/controllers/files.rs

@@ -2,30 +2,14 @@ use axum::{
     extract::{Path, State},
     response::IntoResponse,
 };
-use reqwest::StatusCode;
-use std::path::PathBuf;
-use tracing::error;
 
 use crate::{
-    constants::CDN_ROOT,
-    env::state::AppState,
-    utils::{
-        fetch::fetch_and_cache,
-        http::{response_error, response_file},
-    },
+    env::state::AppState, services::file::process_file_request, utils::http::response_error,
 };
 
 pub async fn get(State(state): State<AppState>, Path(path): Path<String>) -> impl IntoResponse {
-    let file_path = PathBuf::from(format!("{}/files/{}", CDN_ROOT, path));
-
-    if file_path.exists() {
-        error!("File exists but respond with Rust: {:?}", file_path);
-        return response_file(&file_path).await;
-    }
-
-    if let Err(_) = fetch_and_cache(state.host, &file_path, &path).await {
-        return response_error(StatusCode::NOT_FOUND);
+    match process_file_request(&state, &path).await {
+        Ok(response) => response,
+        Err(status) => response_error(status),
     }
-
-    response_file(&file_path).await
 }

src/services/file.rs

@@ -0,0 +1,25 @@
+use axum::response::Response;
+use reqwest::StatusCode;
+use std::path::PathBuf;
+use tracing::error;
+
+use crate::{
+    constants::CDN_ROOT,
+    env::state::AppState,
+    utils::{fetch::fetch_and_cache, http::response_file},
+};
+
+pub async fn process_file_request(state: &AppState, path: &str) -> Result<Response, StatusCode> {
+    let file_path = PathBuf::from(format!("{}/files/{}", CDN_ROOT, path));
+
+    if file_path.exists() {
+        error!("File exists but respond with Rust: {:?}", file_path);
+        return Ok(response_file(&file_path).await);
+    }
+
+    if let Err(_) = fetch_and_cache(state.host.clone(), &file_path, &path).await {
+        return Err(StatusCode::NOT_FOUND);
+    }
+
+    Ok(response_file(&file_path).await)
+}

src/services/mod.rs

@@ -1 +1,2 @@
+pub mod file;
 pub mod image;