added loot/harvest

2 files changed, 44 insertions, 1 deletions

diff --git a/src/systems/droploot.rs b/src/systems/droploot.rs
new file mode 100644
index 0000000..616d38d
--- /dev/null
+++ b/src/systems/droploot.rs
@@ -0,0 +1,41 @@
+
+use rand::Rng;
+
+use specs::{
+	ReadStorage,
+	WriteStorage,
+	System,
+	Join,
+	Write
+};
+
+use crate::{
+	components::{
+		Position,
+		Loot,
+		Dead
+	},
+	resources::{NewEntities}
+};
+
+
+pub struct DropLoot;
+impl <'a> System<'a> for DropLoot{
+	type SystemData = (
+		WriteStorage<'a, Position>,
+		Write<'a, NewEntities>,
+		ReadStorage<'a, Dead>,
+		ReadStorage<'a, Loot>
+	);
+	
+	fn run(&mut self, (positions, mut new, deads, loots): Self::SystemData) {
+		for (position, _, loot) in (&positions, &deads, &loots).join(){
+			for (template, chance) in &loot.loot {
+				if *chance > rand::thread_rng().gen_range(0.0, 1.0) {
+					// todo: better error handling
+					new.create(position.pos, template.clone()).unwrap();
+				}
+			}
+		}
+	}
+}
diff --git a/src/systems/mod.rs b/src/systems/mod.rs
index 887a1f9..1735df9 100644
--- a/src/systems/mod.rs
+++ b/src/systems/mod.rs
@@ -18,6 +18,7 @@ mod controlai;
 mod die;
 mod spawn;
 mod interact;
+mod droploot;
 
 pub use self::{
 	controlinput::ControlInput,
@@ -38,5 +39,6 @@ pub use self::{
 	controlai::ControlAI,
 	die::Die,
 	spawn::Spawn,
-	interact::Interact
+	interact::Interact,
+	droploot::DropLoot
 };