use bevy::prelude::*;
use bevy_rapier2d::prelude::AdditionalMassProperties;
use crate::config::planet::Planet;
pub fn incoming_planets_plugin(app: &mut App) {
app.add_systems(Update, (handle_incoming_planets, handle_updated_planets));
}
fn handle_incoming_planets(mut commands: Commands, mut new_planets: Query<(Entity, &Planet), Added<Planet>>, asset_server: Res<AssetServer>) {
for (new_entity, new_planet) in new_planets.iter() {
let mut sprite = Sprite::from_image(asset_server.load(&new_planet.sprite));
sprite.custom_size = Some(Vec2::new(new_planet.radius*2.0, new_planet.radius*2.0));
commands.entity(new_entity)
.insert(sprite)
.insert(AdditionalMassProperties::Mass(new_planet.mass));
info!(?new_planet, "prepared new planet");
}
}
fn handle_updated_planets(mut commands: Commands, mut updated_planets: Query<(Entity, &Planet), Changed<Planet>>, asset_server: Res<AssetServer>) {
for (updated_entity, updated_planet) in updated_planets.iter() {
let mut sprite = Sprite::from_image(asset_server.load(&updated_planet.sprite));
sprite.custom_size = Some(Vec2::new(updated_planet.radius*2.0, updated_planet.radius*2.0));
commands.entity(updated_entity)
.remove::<Sprite>()
.remove::<AdditionalMassProperties>()
.insert(sprite)
.insert(AdditionalMassProperties::Mass(updated_planet.mass));
info!(?updated_planet, "updated planet");
}
}