use std::collections::HashMap;

use bevy::prelude::Component;
use starkingdoms_common::PlanetType;

#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
pub enum MaterialType {
    Plasma,    // Sun
    Composite, // Mercury
    Sulfur,    // Venus
    Silicon,   // Moon
    Iron,      // Mars
    Hydrogen,  // Jupiter
    Helium,    // Saturn
    Rubber,    // Uranus
    Methane,   // Neptune
    Ice,       // Pluto
}
impl TryFrom<PlanetType> for MaterialType {
    type Error = ();
    fn try_from(value: PlanetType) -> Result<Self, ()> {
        match value {
            PlanetType::Sun => Ok(Self::Plasma),
            PlanetType::Mercury => Ok(Self::Composite),
            PlanetType::Venus => Ok(Self::Sulfur),
            PlanetType::Moon => Ok(Self::Silicon),
            PlanetType::Mars => Ok(Self::Iron),
            PlanetType::Jupiter => Ok(Self::Hydrogen),
            PlanetType::Saturn => Ok(Self::Helium),
            PlanetType::Uranus => Ok(Self::Rubber),
            PlanetType::Neptune => Ok(Self::Methane),
            PlanetType::Pluto => Ok(Self::Ice),
            _ => Err(()),
        }
    }
}

#[derive(Component, Debug, Clone)]
pub struct MaterialStorage {
    pub material_type: MaterialType,
    pub stored: u32,
    pub capacity: u32,
}

#[derive(Component, Debug, Clone, Default)]
pub struct VarietyMaterialStorage {
    pub materials: HashMap<MaterialType, u32>,
    pub capacity: u32,
}

#[derive(Component, Debug, Clone, Default)]
pub struct IsMining(pub bool);