use std::env;
use std::path::PathBuf;

use stm32_metapac_gen::*;

fn parse_chip_core(chip_and_core: &str) -> (String, Option<String>) {
    let mut s = chip_and_core.split('-');
    let chip_name: String = s.next().unwrap().to_string();
    if let Some(c) = s.next() {
        if c.starts_with("cm") {
            return (chip_name, Some(c.to_ascii_lowercase()));
        }
    }

    (chip_and_core.to_string(), None)
}

fn main() {
    let out_dir = PathBuf::from(env::var_os("OUT_DIR").unwrap());
    let data_dir = PathBuf::from("../stm32-data/data");

    let chip_core_name = env::vars_os()
        .map(|(a, _)| a.to_string_lossy().to_string())
        .find(|x| x.starts_with("CARGO_FEATURE_STM32"))
        .expect("No stm32xx Cargo feature enabled")
        .strip_prefix("CARGO_FEATURE_")
        .unwrap()
        .to_ascii_lowercase()
        .replace('_', "-");

    let (chip_name, _) = parse_chip_core(&chip_core_name);

    let opts = Options {
        out_dir: out_dir.clone(),
        data_dir: data_dir.clone(),
        chips: vec![chip_name.to_ascii_uppercase()],
    };
    Gen::new(opts).gen();

    println!(
        "cargo:rustc-link-search={}/src/chips/{}",
        out_dir.display(),
        chip_core_name,
    );

    #[cfg(feature = "memory-x")]
    println!(
        "cargo:rustc-link-search={}/src/chips/{}/memory_x/",
        out_dir.display(),
        chip_core_name
    );
    println!(
        "cargo:rustc-env=STM32_METAPAC_PAC_PATH={}/src/chips/{}/pac.rs",
        out_dir.display(),
        chip_core_name
    );
    println!(
        "cargo:rustc-env=STM32_METAPAC_METADATA_PATH={}/src/chips/{}/metadata.rs",
        out_dir.display(),
        chip_core_name
    );
    println!(
        "cargo:rustc-env=STM32_METAPAC_COMMON_PATH={}/src/common.rs",
        out_dir.display(),
    );

    println!("cargo:rerun-if-changed=build.rs");

    // When the stm32-data chip's JSON changes, we must rebuild
    println!(
        "cargo:rerun-if-changed={}/chips/{}.json",
        data_dir.display(),
        chip_name.to_uppercase()
    );

    println!("cargo:rerun-if-changed={}/registers", data_dir.display());
}
