mitsuba 1.10.0

Lightweight 4chan board archive software (like Foolfuuka), in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use s3::bucket::Bucket;
use s3::creds::Credentials;
use s3::region::Region;
use crate::util::bool_from_env;

pub struct ObjectStorage {
    pub region: Region,
    pub credentials: Credentials,
    pub bucket_name: String,
    pub bucket: Bucket,
    pub enabled: bool
}
impl ObjectStorage {
    pub fn new() -> Self {
        if bool_from_env(&"ENABLE_S3_STORAGE".to_string()) {
            Self::new_env()
        } else {
            // if not enabled we give the dummy version
            Self::default()
        }
    }
    pub fn new_env() -> Self {
        let credentials = Credentials::from_env_specific(
            Some("S3_ACCESS_KEY_ID"),
            Some("S3_SECRET_ACCESS_KEY"),
            None,
            None,
        ).unwrap();
        let bucket_name = std::env::var("S3_BUCKET").unwrap().to_string();
        let region = Region::Custom {
            region: std::env::var("S3_REGION").unwrap().into(),
            endpoint: std::env::var("S3_ENDPOINT").unwrap().into(),
        };
        let mut bucket = Bucket::new_with_path_style(&bucket_name, region.clone(), credentials.clone()).unwrap();
        bucket.add_header("x-amz-acl", "public-read");
        bucket.add_header("Content-Disposition", "inline");
        Self {
            credentials,
            bucket_name,
            region,
            bucket,
            enabled: true
        }
    }
}
impl Default for ObjectStorage {
    fn default() -> Self {
        Self {
            // dummy values
            credentials: Credentials::anonymous().unwrap(),
            bucket_name: "bucket".to_string(),
            region: "us-east-1".parse().unwrap(),
            bucket: Bucket::new(&"bucket".to_string(), "us-east-1".parse().unwrap(), Credentials::anonymous().unwrap()).unwrap(),
            enabled: false
        }
    }
}