config.rs

use crate::commands;

use log::info;
use std::collections::HashMap;
use std::env;
use std::fs::File;
use std::io::prelude::*;
use std::sync::Mutex;
use toml;
use yansi::Paint;

#[derive(Debug, Deserialize, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum Feature {
    ExternalPr,
    Commands,
}

#[derive(Debug, Deserialize)]
pub struct Config {
    pub github: Site,
    pub gitlab: Site,
    pub mappings: Vec<Mapping>,
    pub features: Vec<Feature>,
    pub commands: Commands,
    pub pull_request_event_trigger_actions: Actions,
}

pub fn feature_enabled(feature: &Feature) -> bool {
    CONFIG.features.contains(&feature)
}

pub fn command_enabled(command: &commands::CommandAction) -> bool {
    feature_enabled(&Feature::Commands) && CONFIG.commands.enabled_commands.contains(&command)
}

pub fn action_enabled(action: &String) -> bool {
    CONFIG.pull_request_event_trigger_actions.enabled_trigger_actions.contains(&action)
}

#[derive(Debug, Deserialize)]
pub struct Actions {
    pub enabled_trigger_actions: Vec<String>,
}

#[derive(Debug, Deserialize)]
pub struct Commands {
    pub enabled_commands: Vec<commands::CommandAction>,
}

#[derive(Debug, Deserialize)]
pub struct Site {
    pub webhook_secret: String,
    pub username: String,
    pub ssh_key: String,
    pub api_token: String,
    pub hostname: Option<String>,
}

#[derive(Debug, Deserialize)]
pub struct Mapping {
    pub github_repo: String,
    pub gitlab_repo: String,
}

lazy_static! {
    pub static ref HUB_TO_LAB: Mutex<HashMap<String, String>> = {
        let m: HashMap<String, String> = HashMap::new();
        Mutex::new(m)
    };
}

lazy_static! {
    pub static ref LAB_TO_HUB: Mutex<HashMap<String, String>> = {
        let m: HashMap<String, String> = HashMap::new();
        Mutex::new(m)
    };
}

fn get_labhub_toml_path() -> String {
    env::var("LABHUB_TOML").unwrap_or_else(|_| "LabHub.toml".to_string())
}

lazy_static! {
    pub static ref CONFIG: Config = {
        let labhub_toml_path = get_labhub_toml_path();
        let config: Config = toml::from_str(&read_file_to_string(&labhub_toml_path)).unwrap();
        config
    };
}

fn read_file_to_string(filename: &str) -> String {
    let mut file = File::open(filename).expect("Unable to open the file");
    let mut contents = String::new();
    file.read_to_string(&mut contents)
        .expect("Unable to read the file");
    contents
}

pub fn load_config() {
    info!(
        "Loaded LabHub configuration values from {}",
        get_labhub_toml_path()
    );
    info!("CONFIG => {:#?}", Paint::red(&*CONFIG));

    for mapping in CONFIG.mappings.iter() {
        let mut hub_to_lab_lock = HUB_TO_LAB.lock();
        let hub_to_lab = hub_to_lab_lock.as_mut().unwrap();
        hub_to_lab.insert(mapping.github_repo.clone(), mapping.gitlab_repo.clone());

        let mut lab_to_hub_lock = LAB_TO_HUB.lock();
        let lab_to_hub = lab_to_hub_lock.as_mut().unwrap();
        lab_to_hub.insert(mapping.gitlab_repo.clone(), mapping.github_repo.clone());
    }
    info!(
        "HUB_TO_LAB => {:#?}",
        Paint::red(HUB_TO_LAB.lock().unwrap())
    );
    info!(
        "LAB_TO_HUB => {:#?}",
        Paint::red(LAB_TO_HUB.lock().unwrap())
    );
}