added monster ai

1 files changed, 5 insertions, 3 deletions

diff --git a/src/systems/updatecooldowns.rs b/src/systems/updatecooldowns.rs
index 73ca770..705561d 100644
--- a/src/systems/updatecooldowns.rs
+++ b/src/systems/updatecooldowns.rs
@@ -6,16 +6,18 @@ use specs::{
 	Join
 };
 
-use crate::components::ControlCooldown;
+use crate::components::{Controller, ControlCooldown};
 
 
 pub struct UpdateCooldowns;
 impl <'a> System<'a> for UpdateCooldowns {
 	type SystemData = (
 		Entities<'a>,
-		WriteStorage<'a, ControlCooldown>
+		WriteStorage<'a, ControlCooldown>,
+		WriteStorage<'a, Controller>
 	);
-	fn run(&mut self, (entities, mut cooldowns): Self::SystemData) {
+	fn run(&mut self, (entities, mut cooldowns, mut controllers): Self::SystemData) {
+		controllers.clear();
 		let mut to_remove = Vec::new();
 		for (entity, cooldown) in (&entities, &mut cooldowns).join() {
 			if cooldown.amount > 0 {