From 14e9d4946fc879d37829a00a116a257b03e1d212 Mon Sep 17 00:00:00 2001
From: Cyberarm <matthewlikesrobots@gmail.com>
Date: Tue, 31 Jan 2023 10:17:41 -0600
Subject: [PATCH] Add GameState#needs_repaint? and Container#remove

---
 lib/cyberarm_engine/game_state.rb            | 4 ++++
 lib/cyberarm_engine/ui/elements/container.rb | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/lib/cyberarm_engine/game_state.rb b/lib/cyberarm_engine/game_state.rb
index 4479625..5e32fa1 100644
--- a/lib/cyberarm_engine/game_state.rb
+++ b/lib/cyberarm_engine/game_state.rb
@@ -34,6 +34,10 @@ module CyberarmEngine
       true
     end
 
+    def needs_repaint?
+      true
+    end
+
     def drop(filename)
     end
 
diff --git a/lib/cyberarm_engine/ui/elements/container.rb b/lib/cyberarm_engine/ui/elements/container.rb
index 47ea0d5..c06bdfa 100644
--- a/lib/cyberarm_engine/ui/elements/container.rb
+++ b/lib/cyberarm_engine/ui/elements/container.rb
@@ -42,6 +42,10 @@ module CyberarmEngine
         root.gui_state.request_recalculate
       end
 
+      def remove(element)
+        root.gui_state.request_recalculate if @children.delete(element)
+      end
+
       def clear(&block)
         @children.clear