use std::{collections::HashSet, env};

fn features_enabled() -> HashSet<String> {
    env::vars()
        .map(|(key, _)| key)
        .flat_map(|feature| {
            feature
                .strip_prefix("CARGO_FEATURE_")
                .map(str::to_lowercase)
        })
        .collect()
}

fn features_10xx() -> HashSet<String> {
    [
        "imxrt1010",
        "imxrt1015",
        "imxrt1020",
        "imxrt1050",
        "imxrt1060",
        "imxrt1064",
    ]
    .iter()
    .map(ToString::to_string)
    .collect()
}

fn features_11xx() -> HashSet<String> {
    ["imxrt1160", "imxrt1170"]
        .iter()
        .map(ToString::to_string)
        .collect()
}

fn features_chip() -> HashSet<String> {
    features_10xx().into_iter().chain(features_11xx()).collect()
}

fn emit_cfg_checks<F>(cfg: &str, values: impl IntoIterator<Item = F>)
where
    F: std::fmt::Display,
{
    let quoted: Vec<String> = values
        .into_iter()
        .map(|value| format!("\"{}\"", value))
        .collect();
    let joined = quoted.join(", ");
    // Single ":" permitted for backwards compatibility.
    println!("cargo:rustc-check-cfg=cfg({cfg}, values({joined}))");
}

fn main() {
    let all_features = features_enabled();
    let feat_chip: HashSet<_> = features_chip();

    emit_cfg_checks("chip", feat_chip.iter());
    emit_cfg_checks("family", ["none", "imxrt10xx", "imxrt11xx"]);

    let enabled_chip: Vec<_> = all_features.intersection(&feat_chip).collect();
    assert!(
        enabled_chip.len() < 2,
        "Too many chip features! Select one of {:?}",
        enabled_chip
    );

    if let Some(chip) = enabled_chip.first() {
        let feat_10xx = features_10xx();
        let feat_11xx = features_11xx();

        let family = if feat_10xx.contains(*chip) {
            assert!(!feat_11xx.contains(*chip));
            "imxrt10xx"
        } else if feat_11xx.contains(*chip) {
            assert!(!feat_10xx.contains(*chip));
            "imxrt11xx"
        } else {
            "none"
        };

        println!("cargo:rustc-cfg=family=\"{family}\"");
        println!("cargo:rustc-cfg=chip=\"{chip}\"");
    } else {
        println!("cargo:rustc-cfg=family=\"none\"");
        println!("cargo:rustc-cfg=chip=\"none\"");
    }
}