simple-wkd/src/management.rs

use crate::pending_path;
use crate::settings::SETTINGS;
use crate::utils::{get_user_file_path, key_exists};
use crate::{errors::Error, utils::get_filename};

use chrono::Utc;
use log::{debug, error, warn};
use serde::{Deserialize, Serialize};
use std::{fmt::Display, fs, path::Path};

#[derive(Serialize, Deserialize, Debug, Clone, Copy)]
pub enum Action {
    Add,
    Delete,
}

impl Display for Action {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{self:?}")
    }
}

#[derive(Serialize, Deserialize, Debug)]
pub struct Pending {
    action: Action,
    data: String,
    timestamp: i64,
}
impl Pending {
    pub fn build_add(pem: String) -> Self {
        let timestamp = Utc::now().timestamp();
        Self {
            action: Action::Add,
            data: pem,
            timestamp,
        }
    }
    pub fn build_delete(email: String) -> Self {
        let timestamp = Utc::now().timestamp();
        Self {
            action: Action::Delete,
            data: email,
            timestamp,
        }
    }
    pub const fn action(&self) -> &Action {
        &self.action
    }
    pub fn data(&self) -> &str {
        &self.data
    }
    pub const fn timestamp(&self) -> i64 {
        self.timestamp
    }
}

fn store_pending(pending: &Pending, token: &str) -> Result<(), Error> {
    let serialized = match serde_json::to_string(pending) {
        Ok(serialized) => serialized,
        Err(_) => return Err(Error::SerializeData),
    };
    match fs::write(pending_path!().join(token), serialized) {
        Ok(_) => Ok(()),
        Err(_) => Err(Error::Inaccessible),
    }
}

pub fn store_pending_addition(pem: String, email: &str, token: &str) -> Result<(), Error> {
    let pending = Pending::build_add(pem);
    store_pending(&pending, token)?;
    debug!("Stored submission from {} with token {}", email, token);
    Ok(())
}

pub fn store_pending_deletion(email: String, token: &str) -> Result<(), Error> {
    if let Err(error) = key_exists(&email) {
        match error {
            Error::PathGeneration => debug!("Error while generating path for user {}", email),
            Error::MissingKey => debug!("There is no key for user {}", email),
            _ => error!("An unexpected error occoured!"),
        }
        return Err(error);
    }
    let pending = Pending::build_delete(email.clone());
    store_pending(&pending, token)?;
    debug!(
        "Stored deletion request from {} with token {}",
        email, token
    );
    Ok(())
}

pub fn clean_stale(max_age: i64) {
    for path in fs::read_dir(pending_path!()).unwrap().flatten() {
        let file_path = path.path();
        if file_path.is_file() {
            let content = match fs::read_to_string(&file_path) {
                Ok(content) => content,
                Err(_) => {
                    warn!(
                        "Could not read contents of token {} to string",
                        get_filename(&file_path).unwrap()
                    );
                    continue;
                }
            };
            let key = match serde_json::from_str::<Pending>(&content) {
                Ok(key) => key,
                Err(_) => {
                    warn!(
                        "Could not deserialize token {}",
                        get_filename(&file_path).unwrap()
                    );
                    continue;
                }
            };
            let now = Utc::now().timestamp();
            if now - key.timestamp() > max_age {
                if fs::remove_file(&file_path).is_err() {
                    {
                        warn!(
                            "Could not delete stale token {}",
                            get_filename(&file_path).unwrap()
                        );
                        continue;
                    };
                }
                debug!("Deleted stale token {}", get_filename(&file_path).unwrap())
            }
        }
    }
}

pub fn delete_key(email: &str) -> Result<(), Error> {
    let path = Path::new(&SETTINGS.root_folder).join(get_user_file_path(email)?);
    match fs::remove_file(path) {
        Ok(_) => Ok(()),
        Err(_) => Err(Error::Inaccessible),
    }
}