use bevy::prelude::*;
use bevy_rapier2d::prelude::{AdditionalMassProperties, ReadMassProperties, RigidBody};
use crate::config::planet::Planet;
use crate::ecs::Part;

pub fn incoming_parts_plugin(app: &mut App) {
    app.add_systems(Update, (handle_incoming_parts, handle_updated_parts));
}

fn handle_incoming_parts(mut commands: Commands, mut new_parts: Query<(Entity, &Part), Added<Part>>, asset_server: Res<AssetServer>) {
    for (new_entity, new_part) in new_parts.iter() {
        let mut sprite = Sprite::from_image(asset_server.load(&new_part.sprite));
        sprite.custom_size = Some(Vec2::new(new_part.width, new_part.height));

        commands.entity(new_entity)
            .insert(sprite)
            .insert(AdditionalMassProperties::Mass(new_part.mass))
            .insert(ReadMassProperties::default())
            .insert(RigidBody::Dynamic);
        info!(?new_part, "prepared new part");
    }
}
fn handle_updated_parts(mut commands: Commands, mut updated_parts: Query<(Entity, &Part), Changed<Part>>, asset_server: Res<AssetServer>) {
    for (updated_entity, updated_part) in updated_parts.iter() {
        let mut sprite = Sprite::from_image(asset_server.load(&updated_part.sprite));
        sprite.custom_size = Some(Vec2::new(updated_part.width, updated_part.height));

        commands.entity(updated_entity)
            .remove::<Sprite>()
            .remove::<AdditionalMassProperties>()
            .insert(sprite)
            .insert(AdditionalMassProperties::Mass(updated_part.mass));
        info!(?updated_part, "updated part");
    }
}