PkmnLib_rs/tests/common/library_loader.rs

use hashbrown::HashSet;
use num_traits::PrimInt;
use pkmn_lib::defines::LevelInt;
use pkmn_lib::dynamic_data::libraries::battle_stat_calculator::BattleStatCalculator;
use pkmn_lib::dynamic_data::libraries::damage_library::Gen7DamageLibrary;
use pkmn_lib::dynamic_data::libraries::dynamic_library::DynamicLibrary;
use pkmn_lib::dynamic_data::libraries::misc_library::Gen7MiscLibrary;
use pkmn_lib::static_data::{
    Ability, AbilityLibrary, BattleItemCategory, DataLibrary, EffectParameter, Form, GrowthRateLibrary, Item,
    ItemLibrary, LearnableMoves, LibrarySettings, LookupGrowthRate, MoveData, MoveLibrary, Nature, NatureLibrary,
    SecondaryEffect, Species, StaticData, StaticStatisticSet, Statistic, TypeLibrary,
};
use pkmn_lib::StringKey;
use project_root::get_project_root;
use serde_json::Value;
use std::convert::TryFrom;
use std::fmt::Debug;
use std::fs::File;
use std::io::Read;

pub fn load_library() -> DynamicLibrary {
    let mut path = get_project_root().unwrap();
    path.push("tests/data/");
    let path = path.to_str().unwrap().to_string();
    let mut data = StaticData::new(LibrarySettings::new(100));
    load_types(&path, data.types_mut());
    load_natures(&path, data.natures_mut());
    load_items(&path, data.items_mut());
    load_growth_rates(&path, data.growth_rates_mut());
    load_abilities(&path, data.abilities_mut());
    load_moves(&path, &mut data);
    load_species(&path, &mut data);
    let dynamic = DynamicLibrary::new(
        data,
        BattleStatCalculator {},
        Box::new(Gen7DamageLibrary::new(false)),
        Box::new(Gen7MiscLibrary::new()),
    );
    dynamic
}

pub fn load_types(path: &String, type_library: &mut TypeLibrary) {
    let mut reader = csv::ReaderBuilder::new()
        .delimiter(b'|')
        .from_path(path.to_string() + "Types.csv")
        .unwrap();
    let headers = reader.headers().unwrap();
    for header in headers.iter().skip(1) {
        type_library.register_type(&StringKey::new(header));
    }

    for record in reader.records() {
        let record = record.unwrap();
        let offensive_type = record.get(0).unwrap();
        let offensive_type_id = type_library.get_type_id(&StringKey::new(offensive_type));

        for (i, v) in record.iter().skip(1).enumerate() {
            let effectiveness = v.parse::<f32>().unwrap();
            type_library.set_effectiveness(offensive_type_id, i as u8, effectiveness);
        }
    }
}

pub fn load_natures(path: &String, nature_library: &mut NatureLibrary) {
    let mut reader = csv::ReaderBuilder::new()
        .delimiter(b'|')
        .from_path(path.to_string() + "Natures.csv")
        .unwrap();

    for record in reader.records() {
        let record = record.unwrap();
        let nature_name = StringKey::new(record.get(0).unwrap());
        let increased_statistic_str = record.get(1).unwrap();
        let decreased_statistic_str = record.get(2).unwrap();
        if increased_statistic_str.is_empty() || decreased_statistic_str.is_empty() {
            nature_library.load_nature(nature_name, Nature::new(Statistic::HP, Statistic::HP, 1.0, 1.0));
        } else {
            let increased_statistic = serde_plain::from_str(increased_statistic_str).unwrap();
            let decreased_statistic = serde_plain::from_str(decreased_statistic_str).unwrap();
            nature_library.load_nature(
                nature_name,
                Nature::new(increased_statistic, decreased_statistic, 1.1, 0.9),
            );
        }
    }
}

pub fn load_items(path: &String, lib: &mut ItemLibrary) {
    let mut file = File::open(path.to_string() + "Items.json").unwrap();
    let mut data = String::new();
    file.read_to_string(&mut data).unwrap();
    let json: Value = serde_json::from_str(&data).unwrap();
    let json_array = json.as_array().unwrap();

    for v in json_array {
        let name = StringKey::new(v["name"].as_str().unwrap());
        let category = serde_json::from_value(v["itemType"].clone()).unwrap();
        let mut battle_category = BattleItemCategory::None;
        if let Some(c) = v.get("battleType") {
            battle_category = serde_json::from_value(c.clone()).unwrap();
        }
        let price = v["price"].as_i64().unwrap();
        let mut flags = HashSet::new();
        if let Some(f) = v.get("flags") {
            let a = f.as_array().unwrap();
            for flag in a {
                flags.insert(StringKey::new(flag.as_str().unwrap()));
            }
        }

        lib.add(
            &name,
            Box::new(Item::new(&name, category, battle_category, price as i32, flags)),
        );
    }
}

pub fn load_growth_rates(path: &String, growth_rate_library: &mut GrowthRateLibrary) {
    let mut file = File::open(path.to_string() + "GrowthRates.json").unwrap();
    let mut data = String::new();
    file.read_to_string(&mut data).unwrap();
    let json: Value = serde_json::from_str(&data).unwrap();
    let o = json.as_object().unwrap();

    for (key, value) in o {
        let name = StringKey::new(key);
        let experience_required_json = value.as_array().unwrap();
        let mut experience_required = Vec::with_capacity(experience_required_json.len());
        for v in experience_required_json {
            experience_required.push(v.as_i64().unwrap() as u32);
        }

        growth_rate_library.add_growth_rate(&name, Box::new(LookupGrowthRate::new(experience_required)));
    }
}

pub fn load_abilities(path: &String, ability_library: &mut AbilityLibrary) {
    let mut file = File::open(path.to_string() + "Abilities.json").unwrap();
    let mut data = String::new();
    file.read_to_string(&mut data).unwrap();
    let json: Value = serde_json::from_str(&data).unwrap();
    let o = json.as_object().unwrap();

    for (key, value) in o {
        let name = StringKey::new(key);
        let mut effect = StringKey::empty();
        if let Some(e) = value.get("effect") {
            effect = StringKey::new(e.as_str().unwrap());
        }
        let mut parameters = Vec::new();
        if let Some(p) = value.get("parameters") {
            for par in p.as_array().unwrap() {
                parameters.push(parse_effect_parameter(par));
            }
        }

        ability_library.add(&name, Box::new(Ability::new(&name, &effect, parameters)));
    }
}

pub fn load_moves(path: &String, lib: &mut StaticData) {
    let mut file = File::open(path.to_string() + "Moves.json").unwrap();
    let mut data = String::new();
    file.read_to_string(&mut data).unwrap();
    let json: Value = serde_json::from_str(&data).unwrap();
    let data = json.as_object().unwrap().get("data").unwrap().as_array().unwrap();
    for move_data in data {
        let move_data = move_data.as_object().unwrap();
        let move_name = StringKey::new(move_data["name"].as_str().unwrap());
        let move_type = StringKey::new(move_data["type"].as_str().unwrap());
        let move_type_id = lib.types().get_type_id(&move_type);
        let move_category = serde_json::from_value(move_data["category"].clone()).unwrap();
        let base_power = move_data["power"].as_i64().unwrap() as u8;
        let accuracy = move_data["accuracy"].as_i64().unwrap() as u8;
        let pp = move_data["pp"].as_i64().unwrap() as u8;
        let target = serde_json::from_value(move_data["target"].clone()).unwrap();
        let priority = move_data["priority"].as_i64().unwrap() as i8;
        let secondary_effect = if let Some(v) = move_data.get("effect") {
            let mut chance = -1.0;
            if let Some(chance_value) = v.get("chance") {
                chance = chance_value.as_f64().unwrap() as f32;
            }
            let mut parameters = Vec::new();
            if let Some(pars) = v.get("parameters") {
                let pars = pars.as_array().unwrap();
                for par in pars {
                    parameters.push(parse_effect_parameter(par));
                }
            }

            SecondaryEffect::new(chance, StringKey::new(v["name"].as_str().unwrap()), parameters)
        } else {
            SecondaryEffect::empty()
        };

        let mut flags = HashSet::new();
        if let Some(f) = move_data.get("flags") {
            let f = f.as_array().unwrap();
            for flag in f {
                flags.insert(StringKey::new(flag.as_str().unwrap()));
            }
        }

        lib.moves_mut().add(
            &move_name,
            MoveData::new(
                &move_name.clone(),
                move_type_id,
                move_category,
                base_power,
                accuracy,
                pp,
                target,
                priority,
                secondary_effect,
                flags,
            ),
        );
    }
}

pub fn load_species(path: &String, library: &mut StaticData) {
    let mut file = File::open(path.to_string() + "Pokemon.json").unwrap();
    let mut data = String::new();
    file.read_to_string(&mut data).unwrap();
    let json: Value = serde_json::from_str(&data).unwrap();
    let o = json.as_object().unwrap();

    for (key, value) in o {
        if key.starts_with('$') {
            continue;
        }
        let name = StringKey::new(key);
        let id = value["id"].as_i64().unwrap();
        let gender_rate = value["genderRatio"].as_f64().unwrap();
        let growth_rate_name = StringKey::new(value["growthRate"].as_str().unwrap());
        let _base_happiness = value["baseHappiness"].as_i64().unwrap();
        let catch_rate = value["catchRate"].as_i64().unwrap();
        let _color = value["color"].as_str().unwrap();
        // let egg_groups = value["eggGroups"]
        //     .as_array()
        //     .unwrap()
        //     .iter()
        //     .map(|&a| a.as_str().unwrap())
        //     .collect();
        let _egg_cycle = value["eggCycles"].as_i64().unwrap();
        // TODO: tags
        // TODO: evolutions

        let forms = value["formes"].as_object().unwrap();
        let default_form_value = &forms["default"];
        let default_form = parse_form("default".into(), default_form_value, library);

        let species = Box::new(Species::new(
            id as u16,
            &name,
            gender_rate as f32,
            &growth_rate_name,
            catch_rate as u8,
            default_form,
            Default::default(),
        ));
        library.species_mut().add(&name, species);
    }
}

fn parse_form(name: StringKey, value: &Value, library: &mut StaticData) -> Form {
    let mut abilities = Vec::new();
    for a in value["abilities"].as_array().unwrap() {
        abilities.push(StringKey::new(a.as_str().unwrap()));
    }
    let mut hidden_abilities = Vec::new();
    for a in value["hiddenAbilities"].as_array().unwrap() {
        hidden_abilities.push(StringKey::new(a.as_str().unwrap()));
    }
    let base_stats = parse_statistics(&value["baseStats"]);
    // TODO: ev reward
    let height = value["height"].as_f64().unwrap();
    let weight = value["weight"].as_f64().unwrap();
    let base_experience = value["baseExp"].as_u64().unwrap();
    let types = value["types"]
        .as_array()
        .unwrap()
        .iter()
        .map(|a| library.types().get_type_id(&StringKey::new(a.as_str().unwrap())))
        .collect();

    let moves = parse_moves(&value["moves"], library.moves());

    Form::new(
        &name,
        height as f32,
        weight as f32,
        base_experience as u32,
        types,
        base_stats,
        abilities,
        hidden_abilities,
        moves,
        Default::default(),
    )
}

fn parse_statistics<T>(value: &Value) -> StaticStatisticSet<T>
where
    T: PrimInt + TryFrom<u64>,
    <T as TryFrom<u64>>::Error: Debug,
{
    StaticStatisticSet::new(
        <T as TryFrom<u64>>::try_from(value.get("hp").unwrap_or(&Value::Number(0.into())).as_u64().unwrap()).unwrap(),
        <T as TryFrom<u64>>::try_from(
            value
                .get("attack")
                .unwrap_or(&Value::Number(0.into()))
                .as_u64()
                .unwrap(),
        )
        .unwrap(),
        <T as TryFrom<u64>>::try_from(
            value
                .get("defense")
                .unwrap_or(&Value::Number(0.into()))
                .as_u64()
                .unwrap(),
        )
        .unwrap(),
        <T as TryFrom<u64>>::try_from(
            value
                .get("specialAttack")
                .unwrap_or(&Value::Number(0.into()))
                .as_u64()
                .unwrap(),
        )
        .unwrap(),
        <T as TryFrom<u64>>::try_from(
            value
                .get("specialDefense")
                .unwrap_or(&Value::Number(0.into()))
                .as_u64()
                .unwrap(),
        )
        .unwrap(),
        <T as TryFrom<u64>>::try_from(value.get("speed").unwrap_or(&Value::Number(0.into())).as_u64().unwrap())
            .unwrap(),
    )
}

fn parse_moves(value: &Value, move_library: &MoveLibrary) -> LearnableMoves {
    let mut moves = LearnableMoves::default();

    let level_moves = value["levelMoves"].as_array().unwrap();
    for level_move in level_moves {
        let name = StringKey::new(level_move["name"].as_str().unwrap());
        let level = level_move["level"].as_u64().unwrap() as LevelInt;
        assert!(move_library.get(&name).is_some());
        moves.add_level_move(level, &name);
    }

    moves
}

fn parse_effect_parameter(value: &Value) -> EffectParameter {
    match value {
        Value::Null => {
            panic!("Unexpected type")
        }
        Value::Bool(b) => EffectParameter::Bool(*b),
        Value::Number(n) => {
            if n.is_f64() {
                EffectParameter::Float(n.as_f64().unwrap() as f32)
            } else {
                EffectParameter::Int(n.as_i64().unwrap())
            }
        }
        Value::String(s) => EffectParameter::String(s.clone()),
        Value::Array(_) => {
            panic!("Unexpected type")
        }
        Value::Object(_) => {
            panic!("Unexpected type")
        }
    }
}

#[test]
#[cfg_attr(miri, ignore)]
fn test_type_library_loaded() {
    let mut path = get_project_root().unwrap();
    path.push("tests/data/");
    let mut lib = TypeLibrary::new(18);
    load_types(&path.to_str().unwrap().to_string(), &mut lib);

    assert_eq!(
        lib.get_effectiveness(
            lib.get_type_id(&StringKey::new("fire")),
            &[lib.get_type_id(&StringKey::new("grass"))],
        ),
        2.0
    );
}