use batch_run::Batch;
use serde::Deserialize;
use std::{
    collections::HashMap,
    fs::{create_dir, read_to_string, File},
    io::Write,
    path::{Path, PathBuf},
};
use toml::from_str;

#[derive(Deserialize, Default)]
struct Data {
    main_type: String,
    support_types: Option<String>,
    definition: String,
    value: String,
}

impl Data {
    fn write(&self, name: &str, path: impl AsRef<Path>) {
        let mut path = path.as_ref().to_owned();
        path.push(name);
        if !path.exists() {
            create_dir(&path).unwrap();
        }
        path.push("dummy"); // a hack, so that folder isn't overwritten with file name
        write!(
            File::create(&path.with_file_name(format!("{}-main.rs", name))).unwrap(),
            include_str!("main.tpl"),
            name = name,
            value = self.value
        )
        .unwrap();
        write!(
            File::create(&path.with_file_name("definition.rs")).unwrap(),
            include_str!("definition.tpl"),
            definition = self.definition
        )
        .unwrap();
        write!(
            File::create(&path.with_file_name(format!("{}-user.rs", name))).unwrap(),
            include_str!("user.tpl"),
            types = self
                .support_types
                .as_ref()
                .map_or(self.main_type.clone(), |types| format!(
                    "{},{}",
                    self.main_type, types
                )),
            ser_type = self.main_type,
            value = self.value
        )
        .unwrap();
        write!(
            File::create(&path.with_file_name(format!("{}-main.snapshot", name))).unwrap(),
            include_str!("main.snapshot.tpl"),
            name = name
        )
        .unwrap();
        write!(
            File::create(&path.with_file_name(format!("{}-user.snapshot", name))).unwrap(),
            include_str!("user.snapshot.tpl"),
        )
        .unwrap();
    }
}

#[test]
fn main() {
    let toml = read_to_string("test_fixtures/data.toml").unwrap();
    let data: HashMap<String, Data> = from_str(&toml).unwrap();
    let path: PathBuf = "test_fixtures".into();
    data.into_iter()
        .for_each(|(key, value)| value.write(&key, &path));

    let b = Batch::new();
    b.run_match("test_fixtures/**/*-main.rs");
    b.run().unwrap().assert_all_ok();
}