continuwuity/src/service/acl/data.rs

use std::collections::HashSet;

use clap::ValueEnum;
use serde::{Deserialize, Serialize};
use url::Host;

pub trait Data: Send + Sync {
    /// check if given host exists in Acls, if so return it
    fn check_acl(&self, host: &Host<String>) -> crate::Result<Option<AclMode>>;

    /// add a given Acl entry to the database
    fn add_acl(&self, acl: AclDatabaseEntry) -> crate::Result<()>;
    /// remove a given Acl entry from the database
    fn remove_acl(&self, host: Host<String>) -> crate::Result<()>;

    /// list all acls
    fn get_all_acls(&self) -> HashSet<AclDatabaseEntry>;
}

#[derive(Serialize, Deserialize, Debug, Clone, Copy, Hash, Eq, PartialEq, ValueEnum)]
pub enum AclMode {
    Block,
    Allow,
}

impl AclMode {
    pub fn to_emoji(self) -> char {
        match self {
            AclMode::Block => '❎',
            AclMode::Allow => '✅',
        }
    }
}
#[derive(Serialize, Deserialize, Debug, Clone, Hash, Eq, PartialEq)]

pub struct AclDatabaseEntry {
    pub(crate) mode: AclMode,
    pub(crate) hostname: Host,
}