From 1e9ccafe158c55e28af1cad8d89be3b7a9a257bc Mon Sep 17 00:00:00 2001
From: Spiros <samantas5855@gmail.com>
Date: Thu, 13 Jun 2024 20:19:54 +0300
Subject: [PATCH] Implement Disk menu

---
 src/mainwindow.cpp | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/mainwindow.cpp b/src/mainwindow.cpp
index 0cee6a4..5b035cf 100644
--- a/src/mainwindow.cpp
+++ b/src/mainwindow.cpp
@@ -276,16 +276,24 @@ void MainWindow::updateUI()
         button->setCheckable(true);
         button->setAutoExclusive(true);
 
-        connect(button, &QPushButton::clicked, this, [=]() {
+        auto populateAndNavigate = [=]() {
             if (isNvme) {
                 populateWindow(localObj, health, nvmeSmartOrdered);
             } else {
                 populateWindow(localObj, health);
             }
             updateNavigationButtons(buttonGroup->buttons().indexOf(button));
+        };
+
+        connect(button, &QPushButton::clicked, this, [=]() {
+            populateAndNavigate();
             disksGroup->actions()[buttonGroup->buttons().indexOf(button)]->setChecked(true);
         });
 
+        connect(diskAction, &QAction::triggered, this, [=]() {
+            populateAndNavigate();
+        });
+
         if (firstTime) {
             globalObj = localObj;
             globalHealth = health;