kovi 0.13.0-rc1

use dialoguer::Input;
use dialoguer::theme::ColorfulTheme;
use serde::{Deserialize, Serialize};
use std::io::Write as _;
use std::path::Path;
use std::{env, fs};

use crate::error::BotBuildError;
use crate::event::id::ID;

/// 配置
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct KoviConf {
    pub config: Config,
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Config {
    pub main_admin: ID,
    pub admins: Vec<ID>,
    pub debug: bool,
}

impl KoviConf {
    pub fn new(main_admin: ID, admins: Option<Vec<ID>>, debug: bool) -> Self {
        KoviConf {
            config: Config {
                main_admin,
                admins: admins.unwrap_or_default(),
                debug,
            },
        }
    }
}

impl AsRef<KoviConf> for KoviConf {
    fn as_ref(&self) -> &KoviConf {
        self
    }
}

/// 将配置文件写入磁盘
fn read_from_path_config_write_and_return(file_path: &Path) -> Result<KoviConf, std::io::Error> {
    let main_admin: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("What is the ID of the main administrator? (Not used yet)")
        .allow_empty(true)
        .interact_text()
        .expect("unreachable");

    let main_admin_try_parse_int = main_admin.parse::<i64>();

    let main_admin = match main_admin_try_parse_int {
        Err(_) => ID::new(main_admin),
        Ok(v) => {
            enum IDType {
                Int,
                String,
            }
            let id_type: IDType = {
                let items = ["Int", "String"];
                let select = dialoguer::Select::with_theme(&ColorfulTheme::default())
                    .with_prompt("What is the type of the ID?")
                    .items(&items)
                    .default(0)
                    .interact()
                    .expect("unreachable");

                match select {
                    0 => IDType::Int,
                    1 => IDType::String,
                    _ => panic!(), // 不可能的事情
                }
            };

            match id_type {
                IDType::Int => ID::new(v),
                IDType::String => ID::new(main_admin),
            }
        }
    };

    let config = KoviConf::new(
        main_admin, None,
        // Server::new(host, port, access_token, secure, path, all_in_one),
        false,
    );

    let mut doc = match fs::read_to_string(file_path) {
        Ok(content) => match content.parse::<toml_edit::DocumentMut>() {
            Ok(d) => d,
            Err(err) => {
                eprintln!(
                    "Failed to parse existing config, creating new document: {}",
                    err
                );
                toml_edit::DocumentMut::new()
            }
        },
        Err(_) => toml_edit::DocumentMut::new(),
    };

    // Ensure we have a config table and set values
    doc["config"] = toml_edit::table();
    doc["config"]["main_admin"] = toml_edit::value(config.config.main_admin.clone());
    doc["config"]["admins"] = toml_edit::Item::Value(toml_edit::Value::Array(
        config
            .config
            .admins
            .iter()
            .map(|x| toml_edit::Value::from(x.clone()))
            .collect(),
    ));
    doc["config"]["debug"] = toml_edit::value(config.config.debug);

    let file = fs::File::create(file_path)?;
    let mut writer = std::io::BufWriter::new(file);
    writer.write_all(doc.to_string().as_bytes())?;

    Ok(config)
}

/// 读取本地Kovi.conf.toml文件
pub fn load_local_conf() -> Result<KoviConf, BotBuildError> {
    let path = Path::new("kovi.conf.toml");
    //检测文件kovi.conf.toml
    let kovi_conf_file_exist = fs::metadata(path).is_ok();

    #[derive(Debug, Clone, Deserialize, Serialize)]
    struct TempKoviConf {
        config: Option<Config>,
    }

    let conf_json: KoviConf = if kovi_conf_file_exist {
        match fs::read_to_string(path) {
            Ok(v) => match toml::from_str::<TempKoviConf>(&v) {
                Ok(conf) => match conf.config {
                    Some(config) => KoviConf { config },
                    None => read_from_path_config_write_and_return(path)
                        .map_err(|e| BotBuildError::FileCreateError(e.to_string()))?,
                },
                Err(err) => {
                    eprintln!("Configuration file parsing error: {err}");
                    read_from_path_config_write_and_return(path)
                        .map_err(|e| BotBuildError::FileCreateError(e.to_string()))?
                }
            },
            Err(err) => {
                return Err(BotBuildError::FileReadError(err.to_string()));
            }
        }
    } else {
        read_from_path_config_write_and_return(path)
            .map_err(|e| BotBuildError::FileCreateError(e.to_string()))?
    };

    unsafe {
        if env::var("RUST_LOG").is_err() {
            if conf_json.config.debug {
                env::set_var("RUST_LOG", "debug");
            } else {
                env::set_var("RUST_LOG", "info");
            }
        }
    }

    Ok(conf_json)
}