esp-generate 1.3.0

use crate::Chip;

#[derive(Clone, Debug)]
pub struct Module {
    pub name: &'static str,
    pub display_name: &'static str,
    pub chip: Chip,
    pub remove_pins: &'static [&'static str],
}

// TODO: the module data was taken from https://www.espressif.com/en/products/modules and
// will need to be double checked by actual data sheet information. Also, different modules
// may not expose otherwise available pins, we should consider listing them as well.

pub const ESP32_MODULES: &[Module] = &[
    Module {
        name: "esp32-wroom-32e",
        display_name: "ESP32-WROOM-32E/32UE (4/8/16MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-wrover-e",
        display_name: "ESP32-WROVER-E/IE (4/8/16MB flash, 8MB PSRAM)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "spi_psram", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-mini-1",
        display_name: "ESP32-MINI-1/1U (4MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-pico-mini-01",
        display_name: "ESP32-PICO-MINI-02/02U (8MB flash, 2MB PSRAM)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32-pico-v3-zero",
        display_name: "ESP32-PICO-V3-ZERO (4MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32-wroom-32d",
        display_name: "ESP32-WROOM-32D/32U (4/8/16MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "esp32_pico_v3"],
    },
    // ESP32-SOLO-1 omitted, weird single-core ESP32 variant
    Module {
        name: "esp32-wroom-32",
        display_name: "ESP32-WROOM-32 (4MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-wrover-b",
        display_name: "ESP32-WROVER-B/IB (4/8/16MB flash, 8MB PSRAM)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "spi_psram", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-wroom-da",
        display_name: "ESP32-WROOM-DA (4/8/16MB flash)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "esp32_pico_v3"],
    },
    Module {
        name: "esp32-du1906",
        display_name: "ESP32-DU1906/DU1906-U (8MB flash, 8MB PSRAM)",
        chip: Chip::Esp32,
        remove_pins: &["spi_flash", "spi_psram", "esp32_pico_v3"],
    },
];

pub const ESP32C2_MODULES: &[Module] = &[
    Module {
        name: "esp32c2-mini-1",
        display_name: "ESP8684-MINI-1/1U (1/2/4MB flash)",
        chip: Chip::Esp32c2,
        remove_pins: &["spi_flash"],
    },
    // TODO: these have different pins exposed, maybe separate them?
    Module {
        name: "esp32c2-wroom",
        display_name: "ESP8684-WROOM-01C/02C/02UC/03/04C/05/06C/07 (2/4MB flash)",
        chip: Chip::Esp32c2,
        remove_pins: &["spi_flash"],
    },
];

pub const ESP32C3_MODULES: &[Module] = &[
    Module {
        name: "esp32c3-mini-1",
        display_name: "ESP32-C3-MINI-1/1U (4MB embedded flash)",
        chip: Chip::Esp32c3,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32c3-wroom-02",
        display_name: "ESP32-C3-WROOM-02/02U (4MB flash)",
        chip: Chip::Esp32c3,
        remove_pins: &["spi_flash"],
    },
    // TODO: these have different pins exposed, maybe separate them?
    Module {
        name: "esp32c3-wroom-03",
        display_name: "ESP8685-WROOM-03/04/05/06/07 (2/4MB flash)",
        chip: Chip::Esp32c3,
        remove_pins: &["spi_flash"],
    },
];

pub const ESP32C5_MODULES: &[Module] = &[
    Module {
        name: "esp32c5-wroom-1",
        display_name: "ESP32-C5-WROOM-1/1U (Without PSRAM)",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32c5-wroom-1-psram",
        display_name: "ESP32-C5-WROOM-1/1U (8MB PSRAM)",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32c5-mini-1",
        display_name: "ESP32-C5-MINI-1/1U (Without PSRAM)",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32c5-mini-1-psram",
        display_name: "ESP32-C5-MINI-1/1U (8MB PSRAM)",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash", "spi_psram"],
    },
];

pub const ESP32C6_MODULES: &[Module] = &[
    Module {
        name: "esp32c6-mini-1",
        display_name: "ESP32-C6-MINI-1/1U (4/8MB flash)",
        chip: Chip::Esp32c6,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32c6-wroom-1",
        display_name: "ESP32-C6-WROOM-1/1U (4/8/16MB flash)",
        chip: Chip::Esp32c6,
        remove_pins: &["spi_flash"],
    },
];

pub const ESP32C61_MODULES: &[Module] = &[
    Module {
        name: "esp32c61-wroom-1",
        display_name: "ESP32-C61-WROOM-1/1U",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32c61-mini-1",
        display_name: "ESP32-C61-MINI-1/1U",
        chip: Chip::Esp32c61,
        remove_pins: &["spi_flash"],
    },
];
pub const ESP32H2_MODULES: &[Module] = &[
    Module {
        name: "esp32h2-mini-1",
        display_name: "ESP32-H2-MINI-1/1U (1/2/4MB flash)",
        chip: Chip::Esp32h2,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32h2-wroom-02c",
        display_name: "ESP32-H2-WROOM-02C (2/4MB flash)",
        chip: Chip::Esp32h2,
        remove_pins: &["spi_flash"],
    },
];

pub const ESP32S2_MODULES: &[Module] = &[
    Module {
        name: "esp32s2-mini-2",
        display_name: "ESP32-S2-MINI-2/2U (4MB embedded flash)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s2-mini-2-psram",
        display_name: "ESP32-S2-MINI-2/2U (4MB embedded flash, 2MB PSRAM)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s2-solo-2",
        display_name: "ESP32-S2-SOLO-2/2U (4MB flash)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s2-solo-2-psram",
        display_name: "ESP32-S2-SOLO-2/2U (4MB embedded flash, 2MB PSRAM)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s2-mini-1",
        display_name: "ESP32-S2-MINI-1/1U (4MB embedded flash)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s2-mini-1-psram",
        display_name: "ESP32-S2-MINI-1/1U (4MB embedded flash, 2MB PSRAM)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s2-solo",
        display_name: "ESP32-S2-SOLO/SOLO-U (4/8/16MB flash)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s2-solo-psram",
        display_name: "ESP32-S2-SOLO/SOLO-U (4/8/16MB embedded flash, 2MB PSRAM)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s2-wrover",
        display_name: "ESP32-S2-WROVER/WROVER-I (4/8/16MB flash, 2MB PSRAM)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s2-wroom",
        display_name: "ESP32-S2-WROOM/WROOM-I (4/8/16MB flash)",
        chip: Chip::Esp32s2,
        remove_pins: &["spi_flash"],
    },
];

pub const ESP32S3_MODULES: &[Module] = &[
    Module {
        name: "esp32s3-wroom-1",
        display_name: "ESP32-S3-WROOM-1/1U (4/8/16MB flash)",
        chip: Chip::Esp32s3,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s3-wroom-1-psram",
        display_name: "ESP32-S3-WROOM-1/1U (4/8/16MB flash, 2MB PSRAM)",
        chip: Chip::Esp32s3,
        remove_pins: &["spi_flash", "spi_psram"],
    },
    Module {
        name: "esp32s3-wroom-1-octal-psram",
        display_name: "ESP32-S3-WROOM-1/1U (4/8/16MB flash, 8/16MB PSRAM)",
        chip: Chip::Esp32s3,
        remove_pins: &["spi_flash", "octal_psram"],
    },
    Module {
        name: "esp32s3-wroom-2",
        display_name: "ESP32-S3-WROOM-2 (16/32MB flash, 8/16MB PSRAM)",
        chip: Chip::Esp32s3,
        remove_pins: &["octal_flash", "octal_psram"],
    },
    Module {
        name: "esp32s3-mini-1",
        display_name: "ESP32-S3-MINI-1/1U (4/8MB embedded flash)",
        chip: Chip::Esp32s3,
        remove_pins: &["spi_flash"],
    },
    Module {
        name: "esp32s3-mini-1-psram",
        display_name: "ESP32-S3-MINI-1/1U (4/8MB embedded flash, 2MB PSRAM)",
        chip: Chip::Esp32s3,
        remove_pins: &["spi_flash", "spi_psram"],
    },
];

use crate::template::GeneratorOptionItem;

/// Populates the module category in the template options with chip-specific modules.
pub fn populate_module_category(chip: Chip, options: &mut [GeneratorOptionItem]) {
    for item in options.iter_mut() {
        let GeneratorOptionItem::Category(category) = item else {
            continue;
        };
        if category.name != "module" {
            continue;
        }

        let template_opt = match category.options.first() {
            Some(GeneratorOptionItem::Option(opt)) => opt.clone(),
            _ => {
                panic!("module category must contain a placeholder !Option");
            }
        };

        category.options.clear();

        for module in chip.modules() {
            let mut opt = template_opt.clone();
            opt.name = module.name.to_string();
            opt.display_name = module.display_name.to_string();
            opt.selection_group = "module".to_string();
            category.options.push(GeneratorOptionItem::Option(opt));
        }

        break;
    }
}