From fe12b3c4e2e4315dbded1caa14cf82bf045effa2 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Wed, 4 Jun 2025 21:58:01 -0700
Subject: [PATCH] Add logic to return to DiagMenu

---
 boot_diags/src/app.rs | 1 +
 config/config.json5   | 1 +
 core/src/action.rs    | 1 +
 3 files changed, 3 insertions(+)

diff --git a/boot_diags/src/app.rs b/boot_diags/src/app.rs
index b7ad559..17ccb88 100644
--- a/boot_diags/src/app.rs
+++ b/boot_diags/src/app.rs
@@ -338,6 +338,7 @@ impl App {
                     self.action_tx.send(Action::SetMode(Mode::Failed))?;
                 }
                 Action::BootScan => self.action_tx.send(Action::SetMode(Mode::BootScan))?,
+                Action::DiagMainMenu => self.action_tx.send(Action::SetMode(Mode::DiagMenu))?,
                 Action::InstallDriver => {
                     self.action_tx.send(Action::SetMode(Mode::InstallDrivers))?;
                 }
diff --git a/config/config.json5 b/config/config.json5
index 4d64cda..6fbdb61 100644
--- a/config/config.json5
+++ b/config/config.json5
@@ -109,6 +109,7 @@
       "<Enter>": "Process",
       "<Up>": "KeyUp",
       "<Down>": "KeyDown",
+      "<m>": "DiagMainMenu",
       "<r>": "BootScan",
       "<s>": "ScanDisks",
       "<q>": "Quit",
diff --git a/core/src/action.rs b/core/src/action.rs
index d067140..b956748 100644
--- a/core/src/action.rs
+++ b/core/src/action.rs
@@ -37,6 +37,7 @@ pub enum Action {
     DiagLineStart { text: String },
     DiagLineUpdate { result: DiagResult, text: String },
     DiagLineEnd { text: String },
+    DiagMainMenu,
     // App (Clone)
     Highlight(usize),
     InstallDriver,