Can download files

moneymgr_backend/src/extractors/file_id_extractor.rs

@@ -0,0 +1,61 @@
+use crate::extractors::auth_extractor::AuthExtractor;
+use crate::models::files::{File, FileID};
+use crate::services::files_service;
+use actix_web::dev::Payload;
+use actix_web::{FromRequest, HttpRequest};
+use serde::Deserialize;
+
+#[derive(Deserialize)]
+struct FileIdInPath {
+    file_id: FileID,
+}
+
+#[derive(thiserror::Error, Debug)]
+enum FileIdExtractorError {
+    #[error("Current user does not own the file!")]
+    UserDoesNotOwnFile,
+}
+
+pub struct FileIdExtractor(File);
+
+impl FileIdExtractor {
+    pub async fn load_file_from_path(auth: &AuthExtractor, id: FileID) -> anyhow::Result<Self> {
+        let file = files_service::get_file_with_id(id)?;
+
+        if file.user_id() != auth.user_id() {
+            return Err(FileIdExtractorError::UserDoesNotOwnFile.into());
+        }
+
+        Ok(Self(file))
+    }
+}
+
+impl FromRequest for FileIdExtractor {
+    type Error = actix_web::Error;
+    type Future = futures_util::future::LocalBoxFuture<'static, Result<Self, Self::Error>>;
+
+    fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
+        let req = req.clone();
+        Box::pin(async move {
+            let auth = AuthExtractor::extract(&req).await?;
+
+            let file_id =
+                actix_web::web::Path::<FileIdInPath>::from_request(&req, &mut Payload::None)
+                    .await?
+                    .file_id;
+
+            Self::load_file_from_path(&auth, file_id)
+                .await
+                .map_err(|e| {
+                    log::error!("Failed to extract file ID from URL! {}", e);
+                    actix_web::error::ErrorNotFound("Could not fetch file information!")
+                })
+        })
+    }
+}
+
+impl AsRef<File> for FileIdExtractor {
+    fn as_ref(&self) -> &File {
+        &self.0
+    }
+}

moneymgr_backend/src/extractors/mod.rs

@@ -1,4 +1,5 @@
 pub mod account_extractor;
 pub mod auth_extractor;
 pub mod file_extractor;
+pub mod file_id_extractor;
 pub mod money_session;