reworked entity storage

1 files changed, 16 insertions, 1 deletions

diff --git a/src/entity_button.rs b/src/entity_button.rs
index baa89a4..35f787f 100644
--- a/src/entity_button.rs
+++ b/src/entity_button.rs
@@ -36,6 +36,21 @@ impl<'a> Button<'a> {
     }
 
     pub async fn pressed(&mut self) {
-        self.0.wait_command().await;
+        loop {
+            self.0.wait_command().await;
+            let pressed = self.0.with_data(|data| {
+                let storage = data.storage.as_button_mut();
+                if !storage.consumed && storage.timestamp.is_some() {
+                    storage.consumed = true;
+                    true
+                } else {
+                    false
+                }
+            });
+
+            if pressed {
+                break;
+            }
+        }
     }
 }