aralez/src/utils/parceyaml.rs

use crate::utils::tools::*;
use dashmap::DashMap;
use log::{error, info, warn};
use serde::{Deserialize, Serialize};
use serde_yaml::Error;
use std::collections::HashMap;
use std::fs;
use std::sync::atomic::AtomicUsize;

#[derive(Debug, Serialize, Deserialize)]
struct Config {
    upstreams: HashMap<String, HostConfig>,
}

#[derive(Debug, Serialize, Deserialize)]
struct HostConfig {
    paths: HashMap<String, PathConfig>,
}

#[derive(Debug, Serialize, Deserialize)]
struct PathConfig {
    protocol: String,
    ssl: bool,
    servers: Vec<String>,
}

pub fn load_yaml_to_dashmap(d: &str, kind: &str) -> Option<UpstreamsDashMap> {
    let dashmap = UpstreamsDashMap::new();
    let mut yaml_data = d.to_string();
    match kind {
        "filepath" => {
            let _ = match fs::read_to_string(d) {
                Ok(data) => {
                    info!("Reading upstreams from {}", d);
                    yaml_data = data
                }
                Err(e) => {
                    error!("Reading: {}: {:?}", d, e.to_string());
                    warn!("Running with empty upstreams list, chane it via API");
                    return None;
                }
            };
        }
        "content" => {
            info!("Reading upstreams from API post body");
        }
        _ => error!("Mismatched parameter, only filepath|content is allowed "),
    }

    let p: Result<Config, Error> = serde_yaml::from_str(&yaml_data);
    match p {
        Ok(parsed) => {
            for (hostname, host_config) in parsed.upstreams {
                let path_map = DashMap::new();
                for (path, path_config) in host_config.paths {
                    let mut server_list = Vec::new();
                    for server in path_config.servers {
                        if let Some((ip, port_str)) = server.split_once(':') {
                            if let Ok(port) = port_str.parse::<u16>() {
                                server_list.push((ip.to_string(), port, path_config.ssl, path_config.protocol.clone()));
                            }
                        }
                    }
                    path_map.insert(path, (server_list, AtomicUsize::new(0)));
                }
                dashmap.insert(hostname, path_map);
            }
            Some(dashmap)
        }
        Err(e) => {
            error!("Failed to parse upstreams file: {}", e);
            None
        }
    }
}

pub fn parce_main_config(path: &str) -> DashMap<String, String> {
    info!("Parsing configuration");
    let data = fs::read_to_string(path).unwrap();
    let reply = DashMap::new();
    let cfg: HashMap<String, String> = serde_yaml::from_str(&*data).expect("Failed to parse main config file");
    for (k, v) in cfg {
        reply.insert(k.to_string(), v.to_string());
    }
    reply
}