fix(engine): prevent ignored mouse button input when main loop is slow

1 files changed, 11 insertions, 4 deletions

diff --git a/engine/src/windowing/mouse.rs b/engine/src/windowing/mouse.rs
index 93a0d93..f6087f4 100644
--- a/engine/src/windowing/mouse.rs
+++ b/engine/src/windowing/mouse.rs
@@ -80,11 +80,18 @@ impl Buttons
         button_data.current_state = button_state;
     }
 
-    pub(crate) fn make_states_previous(&mut self)
+    pub fn set_previous(&mut self, button: Button, button_state: ButtonState)
     {
-        for button_data in self.map.values_mut() {
-            button_data.previous_state = button_data.current_state;
-        }
+        let button_data = self.map.entry(button).or_default();
+
+        button_data.previous_state = button_state;
+    }
+
+    pub fn set_previous_to_current(&mut self, button: Button)
+    {
+        let button_data = self.map.entry(button).or_default();
+
+        button_data.previous_state = button_data.current_state;
     }
 }