esetres 0.1.1

A self hosted file storage server.
Documentation 
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

use std::collections::HashMap;

use serde_json::Value;

use crate::config::{self};

const MIME_TYPES_DB_URL: &str = "https://cdn.jsdelivr.net/gh/jshttp/mime-db@1.52.0/db.json";

/// Returns a map of file extensions mapped to the correct MIME type
/// Uses the local file if exists else it will retrieve the types from the database
pub async fn get<'a>(
    config: &config::Object,
) -> Result<HashMap<String, String>, Box<dyn std::error::Error>> {
    let mut file_types = HashMap::new();

    let file = if let Ok(str) = tokio::fs::read_to_string(&config.mime_types.local_path).await {
        str
    } else {
        println!("Getting mime types from {MIME_TYPES_DB_URL}...");

        // get file from remote
        let client = reqwest::Client::new();
        let res = client
            .get(MIME_TYPES_DB_URL)
            .header("User-Agent", "esetres/0.1.0")
            .send()
            .await?;

        let text = res.text().await?;

        tokio::fs::write(&config.mime_types.local_path, &text).await?;

        text
    };

    let v: Value = serde_json::from_str(&file)?;

    for types in v.as_object().into_iter() {
        for (k, value) in types {
            let extensions = value["extensions"].as_array();
            if let Some(exts) = extensions {
                for ext in exts {
                    let key = k.to_owned();
                    let extension = ext.as_str().unwrap().to_string();
                    file_types.insert(extension, key);
                }
            }
        }
    }

    Ok(file_types)
}