ucui 0.1.2

use std::{path::PathBuf, str::FromStr, sync::OnceLock};

use clap::{Parser, Subcommand, ValueEnum};
use log::LevelFilter;
use shakmaty::{fen::Fen, Chess, Color, FromSetup};

#[derive(Parser)]
#[command(version, about, long_about = None)]
pub struct Config {
    /// Path to a UCI engine
    #[arg(short, long, value_name = "ENGINE")]
    engine: Option<PathBuf>,

    /// White time in seconds
    #[arg(short, long, value_name = "TIME", default_value = "600")]
    white_time: i64,

    /// Black time in seconds
    #[arg(short, long, value_name = "TIME", default_value = "600")]
    black_time: i64,

    /// set engine color
    #[arg(short = 'c', long, value_name = "COLOR", default_value = "black")]
    engine_color: EngineColor,

    /// Optional starting position in FEN format
    #[arg(short, long, value_name = "FEN")]
    fen: Option<String>,

    /// Optional arguments to pass to the engine (separated by ";")
    ///
    /// example: --engine-args '--uci;--quiet'
    #[arg(long, value_name = "ARGS", allow_hyphen_values = true)]
    engine_args: Option<String>,

    /// set log level
    #[arg(long, value_name = "LOG_LEVEL", default_value = "info")]
    log_level: LogLevel,

    // /// Command: [play; ...]
    // #[command(subcommand)]
    // pub command: Option<Commands>,
    #[arg(long)]
    uci_debug_log: Option<String>,
    // #[arg(long)]
    // uci_contempt: Option<String>,
    // #[arg(long)]
    // uci_analysis_contempt: Option<String>,
    #[arg(long)]
    uci_threads: Option<String>,
    #[arg(long)]
    uci_hash: Option<String>,
    // #[arg(long)]
    // uci_clear_hash: Option<String>,
    // #[arg(long)]
    // uci_ponder: Option<String>,
    // #[arg(long)]
    // uci_multi_pv: Option<String>,
    #[arg(long)]
    uci_skill_level: Option<String>,
    #[arg(long)]
    uci_move_overhead: Option<String>,
    #[arg(long)]
    uci_slow_mover: Option<String>,
    #[arg(long)]
    uci_nodestime: Option<String>,
    // #[arg(long)]
    // uci_uci_chess960: Option<String>,
    // #[arg(long)]
    // uci_uci_analyse_mode: Option<String>,
    #[arg(long)]
    uci_uci_limit_strength: Option<String>,
    #[arg(long)]
    uci_uci_elo: Option<String>,
    // #[arg(long)]
    // uci_uci_show_wdl: Option<String>,
    #[arg(long)]
    uci_syzygy_path: Option<String>,
    #[arg(long)]
    uci_syzygy_probe_depth: Option<String>,
    #[arg(long)]
    uci_syzygy50_move_rule: Option<String>,
    #[arg(long)]
    uci_syzygy_probe_limit: Option<String>,
    #[arg(long)]
    uci_use_nnue: Option<String>,
    #[arg(long)]
    uci_eval_file: Option<String>,
}

#[derive(Subcommand)]
pub enum Commands {
    Play,
}

#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
enum EngineColor {
    /// Engine takes white
    White,
    /// Engine takes black
    Black,
}

#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
pub enum LogLevel {
    /// A level lower than all log levels.
    Off,
    /// Corresponds to the `Error` log level.
    Error,
    /// Corresponds to the `Warn` log level.
    Warn,
    /// Corresponds to the `Info` log level.
    Info,
    /// Corresponds to the `Debug` log level.
    Debug,
    /// Corresponds to the `Trace` log level.
    Trace,
}

#[derive(Clone)]
pub enum UciOption {
    DebugLog(String),
    // Contempt(String),
    // AnalysisContempt(String),
    Threads(String),
    Hash(String),
    // ClearHash(String),
    // Ponder(String),
    // MultiPV(String),
    SkillLevel(String),
    MoveOverhead(String),
    SlowMover(String),
    Nodestime(String),
    // UCIChess960(String),
    // UCIAnalyseMode(String),
    UCILimitStrength(String),
    UCIElo(String),
    // UCIShowWDL(String),
    SyzygyPath(String),
    SyzygyProbeDepth(String),
    Syzygy50MoveRule(String),
    SyzygyProbeLimit(String),
    UseNNUE(String),
    EvalFile(String),
}

impl UciOption {
    pub fn id(&self) -> &'static str {
        match self {
            UciOption::DebugLog(_) => "Debug Log File",
            // UciOption::Contempt(_) => "Contempt",
            // UciOption::AnalysisContempt(_) => "Analysis Contempt",
            UciOption::Threads(_) => "Threads",
            UciOption::Hash(_) => "Hash",
            // UciOption::ClearHash(_) => "Clear Hash",
            // UciOption::Ponder(_) => "Ponder",
            // UciOption::MultiPV(_) => "MultiPV",
            UciOption::SkillLevel(_) => "Skill Level",
            UciOption::MoveOverhead(_) => "Move Overhead",
            UciOption::SlowMover(_) => "Slow Mover",
            UciOption::Nodestime(_) => "nodestime",
            // UciOption::UCIChess960(_) => "UCI_Chess960",
            // UciOption::UCIAnalyseMode(_) => "UCI_AnalyseMode",
            UciOption::UCILimitStrength(_) => "UCI_LimitStrength",
            UciOption::UCIElo(_) => "UCI_Elo",
            // UciOption::UCIShowWDL(_) => "UCI_ShowWDL",
            UciOption::SyzygyPath(_) => "SyzygyPath",
            UciOption::SyzygyProbeDepth(_) => "SyzygyProbeDepth",
            UciOption::Syzygy50MoveRule(_) => "Syzygy50MoveRule",
            UciOption::SyzygyProbeLimit(_) => "SyzygyProbeLimit",
            UciOption::UseNNUE(_) => "Use NNUE",
            UciOption::EvalFile(_) => "EvalFil",
        }
    }
    pub fn value(&self) -> &String {
        match self {
            UciOption::DebugLog(value) => value,
            // UciOption::Contempt(value) => value,
            // UciOption::AnalysisContempt(value) => value,
            UciOption::Threads(value) => value,
            UciOption::Hash(value) => value,
            // UciOption::ClearHash(value) => value,
            // UciOption::Ponder(value) => value,
            // UciOption::MultiPV(value) => value,
            UciOption::SkillLevel(value) => value,
            UciOption::MoveOverhead(value) => value,
            UciOption::SlowMover(value) => value,
            UciOption::Nodestime(value) => value,
            // UciOption::UCIChess960(value) => value,
            // UciOption::UCIAnalyseMode(value) => value,
            UciOption::UCILimitStrength(value) => value,
            UciOption::UCIElo(value) => value,
            // UciOption::UCIShowWDL(value) => value,
            UciOption::SyzygyPath(value) => value,
            UciOption::SyzygyProbeDepth(value) => value,
            UciOption::Syzygy50MoveRule(value) => value,
            UciOption::SyzygyProbeLimit(value) => value,
            UciOption::UseNNUE(value) => value,
            UciOption::EvalFile(value) => value,
        }
    }
}

static CONFIG: OnceLock<Config> = OnceLock::new();

fn init_table() -> Config {
    Config::parse()
}

pub fn config() -> &'static Config {
    CONFIG.get_or_init(init_table)
}

pub fn get_engine() -> Option<String> {
    config()
        .engine
        .clone()
        .and_then(|path| path.as_os_str().to_str().map(String::from))
}

pub fn get_engine_args() -> Option<Vec<String>> {
    config()
        .engine_args
        .clone()
        .map(|args| args.split(";").map(|arg| arg.to_string()).collect())
}

pub fn get_engine_color() -> Color {
    match config().engine_color {
        EngineColor::Black => Color::Black,
        EngineColor::White => Color::White,
    }
}

pub fn get_time_white() -> i64 {
    std::cmp::max(0, config().white_time)
}

pub fn get_time_black() -> i64 {
    std::cmp::max(0, config().black_time)
}

pub fn get_start_pos() -> Option<Chess> {
    config()
        .fen
        .clone()
        .and_then(|fen| Fen::from_str(&fen).ok())
        .and_then(|fen| {
            Chess::from_setup(fen.as_setup().clone(), shakmaty::CastlingMode::Standard).ok()
        })
}

pub fn get_log_level() -> LevelFilter {
    match config().log_level {
        LogLevel::Off => LevelFilter::Off,
        LogLevel::Error => LevelFilter::Error,
        LogLevel::Warn => LevelFilter::Warn,
        LogLevel::Info => LevelFilter::Info,
        LogLevel::Debug => LevelFilter::Debug,
        LogLevel::Trace => LevelFilter::Trace,
    }
}

// pub fn get_name() -> Option<String> {
//     config().name.clone()
// }

pub fn get_engine_options() -> Vec<UciOption> {
    let mut options: Vec<UciOption> = Vec::new();
    let cfg = config();
    if let Some(value) = cfg.uci_debug_log.clone() {
        options.push(UciOption::DebugLog(value));
    };
    // if let Some(value) = cfg.uci_contempt.clone() {
    //     options.push(UciOption::Contempt(value));
    // };
    // if let Some(value) = cfg.uci_analysis_contempt.clone() {
    //     options.push(UciOption::AnalysisContempt(value));
    // };
    if let Some(value) = cfg.uci_threads.clone() {
        options.push(UciOption::Threads(value));
    };
    if let Some(value) = cfg.uci_hash.clone() {
        options.push(UciOption::Hash(value));
    };
    // if let Some(value) = cfg.uci_clear_hash.clone() {
    //     options.push(UciOption::ClearHash(value));
    // };
    // if let Some(value) = cfg.uci_ponder.clone() {
    //     options.push(UciOption::Ponder(value));
    // };
    // if let Some(value) = cfg.uci_multi_pv.clone() {
    //     options.push(UciOption::MultiPV(value));
    // };
    if let Some(value) = cfg.uci_skill_level.clone() {
        options.push(UciOption::SkillLevel(value));
    };
    if let Some(value) = cfg.uci_move_overhead.clone() {
        options.push(UciOption::MoveOverhead(value));
    };
    if let Some(value) = cfg.uci_slow_mover.clone() {
        options.push(UciOption::SlowMover(value));
    };
    if let Some(value) = cfg.uci_nodestime.clone() {
        options.push(UciOption::Nodestime(value));
    };
    // if let Some(value) = cfg.uci_uci_chess960.clone() {
    //     options.push(UciOption::UCIChess960(value));
    // };
    // if let Some(value) = cfg.uci_uci_analyse_mode.clone() {
    //     options.push(UciOption::UCIAnalyseMode(value));
    // };
    if let Some(value) = cfg.uci_uci_limit_strength.clone() {
        options.push(UciOption::UCILimitStrength(value));
    };
    if let Some(value) = cfg.uci_uci_elo.clone() {
        options.push(UciOption::UCIElo(value));
    };
    // if let Some(value) = cfg.uci_uci_show_wdl.clone() {
    //     options.push(UciOption::UCIShowWDL(value));
    // };
    if let Some(value) = cfg.uci_syzygy_path.clone() {
        options.push(UciOption::SyzygyPath(value));
    };
    if let Some(value) = cfg.uci_syzygy_probe_depth.clone() {
        options.push(UciOption::SyzygyProbeDepth(value));
    };
    if let Some(value) = cfg.uci_syzygy50_move_rule.clone() {
        options.push(UciOption::Syzygy50MoveRule(value));
    };
    if let Some(value) = cfg.uci_syzygy_probe_limit.clone() {
        options.push(UciOption::SyzygyProbeLimit(value));
    };
    if let Some(value) = cfg.uci_use_nnue.clone() {
        options.push(UciOption::UseNNUE(value));
    };
    if let Some(value) = cfg.uci_eval_file.clone() {
        options.push(UciOption::EvalFile(value));
    };
    options
}