tankopedia.rs

use std::borrow::Cow;
use std::collections::BTreeMap;
use std::fs;
use std::io::Write;
use std::num::NonZeroU32;
use std::path::Path;

use tracing::instrument;

use crate::opts::ImportTankopediaOpts;
use crate::prelude::*;
use crate::wargaming::models::{Nation, TankId, TankType, Vehicle};
use crate::wargaming::{Tankopedia, WargamingApi};

mod generated;

/// Retrieves a vehicle from the Tankopedia.
pub fn get_vehicle(tank_id: TankId) -> Cow<'static, Vehicle> {
    generated::GENERATED
        .get(&tank_id)
        .map_or_else(|| Cow::Owned(Vehicle::new_hardcoded(tank_id)), Cow::Borrowed)
}

/// Updates the bundled `tankopedia.json` and generates the bundled [`phf::Map`] with the tankopedia.
#[instrument(skip_all)]
pub async fn import(opts: ImportTankopediaOpts) -> Result {
    sentry::configure_scope(|scope| scope.set_tag("app", "import-tankopedia"));

    let api = WargamingApi::new(
        &opts.application_id,
        time::Duration::from_secs(30),
        NonZeroU32::new(10).unwrap(),
    )?;
    let json_path = Path::new(file!())
        .parent()
        .unwrap()
        .join("tankopedia")
        .join("tankopedia.json");
    let mut tankopedia: Tankopedia =
        serde_json::from_str::<Tankopedia>(&fs::read_to_string(&json_path)?)?
            .into_iter()
            .chain(api.get_tankopedia().await?)
            .collect();
    fs::write(&json_path, serde_json::to_string_pretty(&tankopedia)?)?;

    insert_missing_vehicles(&mut tankopedia)?;
    info!(n_vehicles = tankopedia.len(), "finished");

    let mut file = fs::File::create(
        Path::new(file!())
            .parent()
            .unwrap()
            .join("tankopedia")
            .join("generated.rs"),
    )?;
    writeln!(&mut file, "//! @generated")?;
    writeln!(&mut file)?;
    writeln!(&mut file, "use std::borrow::Cow;")?;
    writeln!(&mut file)?;
    writeln!(&mut file, "use crate::wargaming::models::{{Nation, TankType, Vehicle}};")?;
    writeln!(&mut file)?;
    writeln!(&mut file, "pub static GENERATED: phf::Map<u32, Vehicle> = phf::phf_map! {{")?;
    for (_, vehicle) in tankopedia {
        writeln!(&mut file, "    {}_u32 => Vehicle {{", vehicle.tank_id)?;
        writeln!(&mut file, "        tank_id: {:?},", vehicle.tank_id)?;
        writeln!(&mut file, "        name: Cow::Borrowed({:?}),", vehicle.name,)?;
        writeln!(&mut file, "        tier: {:?},", vehicle.tier)?;
        writeln!(&mut file, "        is_premium: {:?},", vehicle.is_premium)?;
        writeln!(&mut file, "        nation: Nation::{:?},", vehicle.nation)?;
        writeln!(&mut file, "        type_: TankType::{:?},", vehicle.type_)?;
        writeln!(&mut file, "    }},")?;
    }
    writeln!(&mut file, "}};")?;

    Ok(())
}

/// Inserts the hand-coded tanks that are somehow missing from the Tankopedia.
fn insert_missing_vehicles(vehicles: &mut BTreeMap<String, Vehicle>) -> Result {
    for vehicle in [
        Vehicle {
            tank_id: 20817,
            name: Cow::Borrowed("Эксплорер"),
            tier: 6,
            is_premium: true,
            nation: Nation::Uk,
            type_: TankType::Medium,
        },
        Vehicle {
            tank_id: 3089,
            name: Cow::Borrowed("Leichttraktor"),
            tier: 1,
            is_premium: false,
            nation: Nation::Germany,
            type_: TankType::Light,
        },
        Vehicle {
            tank_id: 64081,
            name: Cow::Borrowed("Mk I* Heavy Tank"),
            tier: 1,
            is_premium: true,
            nation: Nation::Uk,
            type_: TankType::Heavy,
        },
        Vehicle {
            tank_id: 1329,
            name: Cow::Borrowed("Renault NC-31"),
            tier: 1,
            is_premium: false,
            nation: Nation::China,
            type_: TankType::Light,
        },
        Vehicle {
            tank_id: 609,
            name: Cow::Borrowed("R. Otsu"),
            tier: 1,
            is_premium: false,
            nation: Nation::Japan,
            type_: TankType::Light,
        },
        Vehicle {
            tank_id: 23297,
            name: Cow::Borrowed("Объект 244"),
            tier: 6,
            is_premium: true,
            nation: Nation::Ussr,
            type_: TankType::Heavy,
        },
        Vehicle {
            tank_id: 18241,
            name: Cow::Borrowed("B-C Bourrasque"),
            tier: 8,
            is_premium: true,
            nation: Nation::France,
            type_: TankType::Medium,
        },
        Vehicle {
            tank_id: 577,
            name: Cow::Borrowed("Renault FT"),
            tier: 2,
            is_premium: true,
            nation: Nation::France,
            type_: TankType::AT,
        },
        Vehicle {
            tank_id: 81,
            name: Cow::Borrowed("Vickers Medium Mk. I"),
            tier: 1,
            is_premium: true,
            nation: Nation::Uk,
            type_: TankType::Medium,
        },
        Vehicle {
            tank_id: 545,
            name: Cow::Borrowed("T1 Cunningham"),
            tier: 1,
            is_premium: true,
            nation: Nation::Usa,
            type_: TankType::Light,
        },
        Vehicle {
            tank_id: 24849,
            name: Cow::Borrowed("Kryos"),
            tier: 6,
            is_premium: true,
            nation: Nation::Germany,
            type_: TankType::AT,
        },
        Vehicle {
            tank_id: 9777,
            name: Cow::Borrowed("WZ-114"),
            tier: 9,
            is_premium: true,
            nation: Nation::China,
            type_: TankType::Heavy,
        },
        Vehicle {
            tank_id: 12417,
            name: Cow::Borrowed("Bisonte C45"),
            tier: 8,
            is_premium: true,
            nation: Nation::Europe,
            type_: TankType::Heavy,
        },
        Vehicle {
            tank_id: 10545,
            name: Cow::Borrowed("Wind"),
            tier: 6,
            is_premium: true,
            nation: Nation::China,
            type_: TankType::Light,
        },
    ] {
        match vehicles.get(&vehicle.tank_id.to_string()) {
            Some(_) => bail!("vehicle #{} is already in the tankopedia", vehicle.tank_id),
            None => vehicles.insert(vehicle.tank_id.to_string(), vehicle),
        };
    }

    Ok(())
}