ec81bcb13c
* wip: better env var reading * move most env vars to env.rs * migrate more env vars * more migration * more migrations * More migration * 🦀 dotenvy is gone (almost) * 🦀 dotenvy is gone 🦀 * Fix mural source account env var handling * Remove defaults from admin key vars * dummy commit to update github pr * fix ci
90 lines
2.0 KiB
Rust
90 lines
2.0 KiB
Rust
use std::str::FromStr;
|
|
|
|
use async_trait::async_trait;
|
|
use thiserror::Error;
|
|
|
|
mod mock;
|
|
mod s3_host;
|
|
|
|
use bytes::Bytes;
|
|
pub use mock::MockHost;
|
|
pub use s3_host::{S3BucketConfig, S3Host};
|
|
|
|
#[derive(Error, Debug)]
|
|
pub enum FileHostingError {
|
|
#[error("S3 error when {0}: {1}")]
|
|
S3Error(&'static str, s3::error::S3Error),
|
|
#[error("File system error in file hosting: {0}")]
|
|
FileSystemError(#[from] std::io::Error),
|
|
#[error("Invalid Filename")]
|
|
InvalidFilename,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct UploadFileData {
|
|
pub file_name: String,
|
|
pub file_publicity: FileHostPublicity,
|
|
pub content_length: u32,
|
|
pub content_sha512: String,
|
|
pub content_sha1: String,
|
|
pub content_md5: Option<String>,
|
|
pub content_type: String,
|
|
pub upload_timestamp: u64,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct DeleteFileData {
|
|
pub file_name: String,
|
|
}
|
|
|
|
#[derive(Debug, Copy, Clone)]
|
|
pub enum FileHostPublicity {
|
|
Public,
|
|
Private,
|
|
}
|
|
|
|
#[async_trait]
|
|
pub trait FileHost {
|
|
async fn upload_file(
|
|
&self,
|
|
content_type: &str,
|
|
file_name: &str,
|
|
file_publicity: FileHostPublicity,
|
|
file_bytes: Bytes,
|
|
) -> Result<UploadFileData, FileHostingError>;
|
|
|
|
async fn get_url_for_private_file(
|
|
&self,
|
|
file_name: &str,
|
|
expiry_secs: u32,
|
|
) -> Result<String, FileHostingError>;
|
|
|
|
async fn delete_file(
|
|
&self,
|
|
file_name: &str,
|
|
file_publicity: FileHostPublicity,
|
|
) -> Result<DeleteFileData, FileHostingError>;
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
|
pub enum FileHostKind {
|
|
S3,
|
|
Local,
|
|
}
|
|
|
|
#[derive(Debug, Error)]
|
|
#[error("invalid file host kind")]
|
|
pub struct InvalidFileHostKind;
|
|
|
|
impl FromStr for FileHostKind {
|
|
type Err = InvalidFileHostKind;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
Ok(match s {
|
|
"s3" => Self::S3,
|
|
"local" => Self::Local,
|
|
_ => return Err(InvalidFileHostKind),
|
|
})
|
|
}
|
|
}
|