eva-ics 3.99.0

use colored::Colorize;
use eva::{tools::get_eva_dir, BUILD, VERSION};
use serde::Serialize;
use serde_json::Value;
use std::collections::BTreeMap;
use std::fs;
use yedb::Database;

const PFX: &str = "eva";

#[derive(Serialize)]
struct Info<'a> {
    version: &'a str,
    build: u64,
}

macro_rules! imported {
    ($k: expr) => {
        println!("{}", format!(" + {}", $k).green());
    };
}

macro_rules! skipped {
    ($k: expr) => {
        println!("{}", format!(" [skipped] {}", $k).dimmed());
    };
}

fn main() {
    let dir_eva = get_eva_dir();
    let mut db = Database::new();
    let db_path = format!("{dir_eva}/runtime/registry");
    db.set_db_path(&db_path).unwrap();
    db.open().expect("Unable to init/open the registry");
    println!("Importing registry schema...");
    let schema: BTreeMap<String, Value> = serde_yaml::from_str(
        &fs::read_to_string(&format!("{dir_eva}/lib/registry/schema.yml")).unwrap(),
    )
    .unwrap();
    for (k, v) in schema {
        let key_name = format!(".schema/{PFX}/{k}");
        imported!(key_name);
        db.key_set(&key_name, v).unwrap();
    }
    println!("Importing registry defaults...");
    let dir_defaults = format!("{dir_eva}/lib/registry/defaults");
    for entry in glob::glob(&format!("{dir_defaults}/**/*.yml"))
        .unwrap()
        .flatten()
    {
        let fname = &entry.as_os_str().to_string_lossy();
        let key_name = format!("{PFX}{}", &fname[dir_defaults.len()..fname.len() - 4]);
        let mut key_content: Value =
            serde_yaml::from_str(&fs::read_to_string(entry).unwrap()).unwrap();
        if let Err(e) = db.key_get(&key_name) {
            if e.kind() == yedb::ErrorKind::KeyNotFound {
                if key_name.ends_with("/config/core") {
                    let mut config: BTreeMap<String, Value> =
                        serde_json::from_value(key_content).unwrap();
                    config.insert(
                        "system_name".to_owned(),
                        Value::String(hostname::get().unwrap().to_string_lossy().to_string()),
                    );
                    key_content = serde_json::to_value(&config).unwrap();
                }
                db.key_set(&key_name, key_content).unwrap();
                imported!(key_name);
            } else {
                panic!("{}", e);
            }
        } else {
            skipped!(key_name);
        }
    }
    db.key_set(
        &format!("{PFX}/data/info"),
        serde_json::to_value(Info {
            version: VERSION,
            build: BUILD,
        })
        .unwrap(),
    )
    .unwrap();
}