HomeDisk/server/src/fs/download.rs

use std::fs;

use axum::{extract::Query, Extension};
use axum_auth::AuthBearer;
use homedisk_database::Database;
use homedisk_types::{
    config::Config,
    errors::{FsError, ServerError},
    fs::upload::Pagination,
};

use crate::middleware::{find_user, validate_jwt, validate_path};

pub async fn handle(
    Extension(db): Extension<Database>,
    Extension(config): Extension<Config>,
    AuthBearer(token): AuthBearer,
    query: Query<Pagination>,
) -> Result<Vec<u8>, ServerError> {
    // validate user token
    let token = validate_jwt(config.jwt.secret.as_bytes(), &token)?;

    // validate the `path` can be used
    validate_path(&query.path)?;

    // search for a user by UUID from a token
    let user = find_user(&db, &token.claims.sub).await?;

    // directory where the file will be placed
    let path = format!(
        "{user_dir}/{req_dir}",
        user_dir = user.user_dir(&config.storage.path),
        req_dir = query.path
    );

    // read file content
    let content =
        fs::read(path).map_err(|err| ServerError::FsError(FsError::ReadFile(err.to_string())))?;

    // send file content in response
    Ok(content)
}