marmite 0.4.2

Marmite is the easiest static site generator.
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
58
59
60
61
62
63
64
65
66
67

use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fs;
#[cfg(unix)]
use std::os::unix::fs::PermissionsExt;
use std::path::PathBuf;

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct Credential {
    pub pds_url: String,
    pub identifier: String,
    pub password: String,
}

type CredentialsStore = HashMap<String, Credential>;

fn credentials_dir() -> PathBuf {
    dirs::config_dir()
        .unwrap_or_else(|| PathBuf::from("."))
        .join("marmite")
}

pub fn credentials_path() -> PathBuf {
    credentials_dir().join("credentials.json")
}

fn load_store() -> CredentialsStore {
    let path = credentials_path();
    if !path.exists() {
        return HashMap::new();
    }
    let Ok(content) = fs::read_to_string(&path) else {
        return HashMap::new();
    };
    serde_json::from_str(&content).unwrap_or_default()
}

fn save_store(store: &CredentialsStore) -> Result<(), Box<dyn std::error::Error>> {
    let dir = credentials_dir();
    fs::create_dir_all(&dir)?;
    let path = credentials_path();
    let json = serde_json::to_string_pretty(store)?;
    fs::write(&path, json)?;
    #[cfg(unix)]
    fs::set_permissions(&path, fs::Permissions::from_mode(0o600))?;
    Ok(())
}

pub fn save(cred: &Credential) -> Result<(), Box<dyn std::error::Error>> {
    let mut store = load_store();
    store.insert(cred.identifier.clone(), cred.clone());
    save_store(&store)
}

pub fn load(identifier: &str) -> Option<Credential> {
    let mut store = load_store();
    store.remove(identifier)
}

pub fn load_any() -> Option<Credential> {
    let store = load_store();
    if store.len() == 1 {
        store.into_values().next()
    } else {
        None
    }
}