From 2bcc0292de473e1c1081c8eb2780a604ce455082 Mon Sep 17 00:00:00 2001
From: Alula <julekonopinska@gmail.com>
Date: Tue, 22 Sep 2020 00:07:13 +0200
Subject: [PATCH] add item count value, which according to people might be used
 by switch version?

---
 src/inventory.rs | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/inventory.rs b/src/inventory.rs
index 7607676..bd00f92 100644
--- a/src/inventory.rs
+++ b/src/inventory.rs
@@ -3,7 +3,8 @@ use crate::shared_game_state::SharedGameState;
 use crate::weapon::{Weapon, WeaponLevel, WeaponType};
 
 #[derive(Clone, Copy)]
-pub struct Item(u16);
+/// (id, amount)
+pub struct Item(u16, u16);
 
 #[derive(Clone)]
 pub struct Inventory {
@@ -33,14 +34,29 @@ impl Inventory {
 
     pub fn add_item(&mut self, item_id: u16) {
         if !self.has_item(item_id) {
-            self.items.push(Item(item_id));
+            self.items.push(Item(item_id, 1));
         }
     }
 
+    pub fn consume_item(&mut self, item_id: u16) {
+        if let Some(item) = self.get_item(item_id) {
+            if item.1 > 1 {
+                item.1 -= 1;
+                return;
+            }
+        }
+
+        self.remove_item(item_id);
+    }
+
     pub fn remove_item(&mut self, item_id: u16) {
         self.items.retain(|item| item.0 != item_id);
     }
 
+    pub fn get_item(&mut self, item_id: u16) -> Option<&mut Item> {
+        self.items.iter_mut().by_ref().find(|item| item.0 == item_id)
+    }
+
     pub fn has_item(&self, item_id: u16) -> bool {
         self.items.iter().any(|item| item.0 == item_id)
     }