feat(system): Allow player to only have one weapon

This makes the game a bit more challenging!
yngtdd · Apr 10, 2022 · eb4fc7b · eb4fc7b
1 parent 356113d
commit eb4fc7b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/systems/player_input.rs b/src/systems/player_input.rs
@@ -6,6 +6,7 @@ use crate::prelude::*;
 #[read_component(Enemy)]
 #[read_component(Item)]
 #[read_component(Carried)]
+#[read_component(Weapon)]
 #[write_component(Health)]
 pub fn player_input(
     ecs: &mut SubWorld,
@@ -34,6 +35,17 @@ pub fn player_input(
                     .for_each(|(entity, _item, _item_pos)| {
                         commands.remove_component::<Point>(*entity);
                         commands.add_component(*entity, Carried(player));
+
+                        if let Ok(e) = ecs.entry_ref(*entity) {
+                            if e.get_component::<Weapon>().is_ok() {
+                                <(Entity, &Carried, &Weapon)>::query()
+                                    .iter(ecs)
+                                    .filter(|(_, c, _)| c.0 == player)
+                                    .for_each(|(e, _, _)| {
+                                        commands.remove(*e);
+                                    })
+                            }
+                        }
                     });
                 Point::new(0, 0)
             }