use crate::shared::config::planet::{Planet, SpecialSpriteProperties};
use crate::prelude::*;
use crate::shared::ecs::{MAIN_STAR_LAYERS};

pub fn incoming_planets_plugin(app: &mut App) {
    app.add_systems(Update, (handle_incoming_planets, handle_updated_planets));
}

fn build_planet_sprite(planet: &Planet, asset_server: &AssetServer) -> Sprite {
    let mut sprite = Sprite::from_image(asset_server.load(&planet.sprite));
    sprite.custom_size = Some(Vec2::splat(planet.radius as f32 * 2.0));
    if let Some(SpecialSpriteProperties::ForceColor(c)) = planet.special_sprite_properties {
        sprite.color = c;
    }
    sprite
}

fn handle_incoming_planets(
    mut commands: Commands,
    new_planets: Query<(Entity, &Planet), Added<Planet>>,
    asset_server: Res<AssetServer>,
) {
    for (new_entity, new_planet) in new_planets.iter() {
        commands.entity(new_entity)
            .insert(MAIN_STAR_LAYERS.clone())
            .insert(build_planet_sprite(new_planet, &asset_server));
        trace!(?new_planet, "prepared new planet");
    }
}

fn handle_updated_planets(
    mut commands: Commands,
    updated_planets: Query<(Entity, &Planet), Changed<Planet>>,
    asset_server: Res<AssetServer>,
) {
    for (updated_entity, updated_planet) in updated_planets.iter() {
        commands.entity(updated_entity)
            .remove::<Sprite>()
            .insert(build_planet_sprite(updated_planet, &asset_server));
        trace!(?updated_planet, "updated planet");
    }
}