AstralRinth/src/file_hosting/mock.rs

use super::{DeleteFileData, FileHost, FileHostingError, UploadFileData};
use async_trait::async_trait;
use bytes::Bytes;
use sha2::Digest;
use time::OffsetDateTime;

pub struct MockHost(());

impl MockHost {
    pub fn new() -> Self {
        MockHost(())
    }
}

#[async_trait]
impl FileHost for MockHost {
    async fn upload_file(
        &self,
        content_type: &str,
        file_name: &str,
        file_bytes: Bytes,
    ) -> Result<UploadFileData, FileHostingError> {
        let path =
            std::path::Path::new(&dotenv::var("MOCK_FILE_PATH").unwrap())
                .join(file_name.replace("../", ""));
        std::fs::create_dir_all(
            path.parent().ok_or(FileHostingError::InvalidFilename)?,
        )?;
        let content_sha1 = sha1::Sha1::from(&*file_bytes).hexdigest();
        let content_sha512 =
            format!("{:x}", sha2::Sha512::digest(&*file_bytes));

        std::fs::write(path, &*file_bytes)?;
        Ok(UploadFileData {
            file_id: String::from("MOCK_FILE_ID"),
            file_name: file_name.to_string(),
            content_length: file_bytes.len() as u32,
            content_sha512,
            content_sha1,
            content_md5: None,
            content_type: content_type.to_string(),
            upload_timestamp: OffsetDateTime::now_utc().unix_timestamp() as u64,
        })
    }

    async fn delete_file_version(
        &self,
        file_id: &str,
        file_name: &str,
    ) -> Result<DeleteFileData, FileHostingError> {
        let path =
            std::path::Path::new(&dotenv::var("MOCK_FILE_PATH").unwrap())
                .join(file_name.replace("../", ""));
        std::fs::remove_file(path)?;

        Ok(DeleteFileData {
            file_id: file_id.to_string(),
            file_name: file_name.to_string(),
        })
    }
}