use clap::{Parser, Subcommand, ValueEnum};
use factorio_blueprint::{
    belt_finding::{conflict_avoidance::ConflictAvoidance, problems, Problem},
    common::visualize::Visualize,
};
use std::{io, path::PathBuf};

#[derive(ValueEnum, Clone)]
enum Mode {
    Solve,
    ConflictAvoidance,
    ConflictStep,
}

#[derive(Subcommand)]
enum ProblemCase {
    Simple,
    Level1,
    Level2,
    Level3,
    Level5,
    File { filename: PathBuf },
}

impl ProblemCase {
    fn get_problem(&self) -> Problem {
        match self {
            ProblemCase::Simple => problems::simple(),
            ProblemCase::Level1 => problems::belt_madness_level_1(),
            ProblemCase::Level2 => problems::belt_madness_level_2(),
            ProblemCase::Level3 => problems::belt_madness_level_3(),
            ProblemCase::Level5 => problems::belt_madness_level_5(),
            ProblemCase::File { filename } => {
                let file = std::fs::File::open(filename).unwrap();
                serde_json::from_reader(file).unwrap()
            }
        }
    }
}

#[derive(Parser)]
struct Args {
    #[arg(value_enum, default_value = "conflict-avoidance")]
    mode: Mode,
    #[command(subcommand)]
    problem: ProblemCase,
}

fn main() {
    let args = Args::parse();

    let mut p = args.problem.get_problem();

    match args.mode {
        Mode::Solve => {
            p.print_visualization();
            p.find_path();
            p.print_visualization();
        }
        Mode::ConflictAvoidance => {
            p.print_visualization();
            p.find_path();
            p.find_path();
            p.print_visualization();
            let mut c = ConflictAvoidance::new(&p);
            c.print_visualization();
            while c.remove_conflict(None) {
                c.print_visualization();
            }
        }
        Mode::ConflictStep => {
            p.print_visualization();
            p.find_path();
            p.print_visualization();
            let mut c = ConflictAvoidance::new(&p);
            c.print_visualization();
            while c.remove_conflict(None) {
                c.print_visualization();
                let mut s = String::new();
                let _ = io::stdin().read_line(&mut s);
            }
        }
    }
}