From fc43be0777501e69d558c05b16bfe752adaec789 Mon Sep 17 00:00:00 2001
From: Robert Garrett <robertgarrett404@gmail.com>
Date: Wed, 13 Aug 2025 14:22:00 -0500
Subject: [PATCH] Hook up the action selectors to the buttons

---
 src/widgets.rs | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/src/widgets.rs b/src/widgets.rs
index baee080..d7fa640 100644
--- a/src/widgets.rs
+++ b/src/widgets.rs
@@ -141,6 +141,7 @@ fn spawn_gameover_ui(mut commands: Commands) {
         children![
             (
                 Button,
+                GameOverMenuAction::ToMainMenu,
                 Node {
                     width: Val::Px(150.0),
                     height: Val::Px(65.0),
@@ -161,6 +162,7 @@ fn spawn_gameover_ui(mut commands: Commands) {
             ),
             (
                 Button,
+                GameOverMenuAction::Quit,
                 Node {
                     width: Val::Px(150.0),
                     height: Val::Px(65.0),
@@ -216,17 +218,31 @@ fn animate_get_ready_widget(
 /// place to keep all system logic for this plugin.
 fn operate_gameover_ui(
     mut interactions: Query<
-        (&Interaction, &mut BackgroundColor, &mut BorderColor),
+        (
+            &Interaction,
+            &mut BackgroundColor,
+            &mut BorderColor,
+            &GameOverMenuAction,
+        ),
         (Changed<Interaction>, With<Button>),
     >,
+    mut game_state: ResMut<NextState<GameState>>,
+    mut app_exit_events: EventWriter<AppExit>,
 ) {
     // TODO: Better colors. These are taken from the example and they're ugly.
-    // TODO: Read the menu action enum component and take that action.
-    for (interaction, mut color, mut border_color) in &mut interactions {
+    for (interaction, mut color, mut border_color, menu_action) in &mut interactions {
         match *interaction {
             Interaction::Pressed => {
                 *color = UI_BUTTON_PRESSED.into();
                 border_color.0 = RED.into();
+                match menu_action {
+                    GameOverMenuAction::ToMainMenu => {
+                        game_state.set(GameState::TitleScreen);
+                    }
+                    GameOverMenuAction::Quit => {
+                        app_exit_events.write(AppExit::Success);
+                    }
+                }
             }
             Interaction::Hovered => {
                 *color = UI_BUTTON_HOVERED.into();