From 477a4f97c5c8910ce13ee326287577d5a197c311 Mon Sep 17 00:00:00 2001
From: "Eshan Roy (Eshanized)" <m.eshanized@gmail.com>
Date: Tue, 30 Apr 2024 22:24:52 +0530
Subject: [PATCH] chore: update header

---
 snigdhaosassistant.h | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/snigdhaosassistant.h b/snigdhaosassistant.h
index 8cac536..1fd22f4 100644
--- a/snigdhaosassistant.h
+++ b/snigdhaosassistant.h
@@ -8,6 +8,8 @@
 #define SNIGDHAOSASSISTANT_H
 
 #include <QMainWindow>
+#include <QAbstractButton>
+#include <QtNetwork/QNetworkAccessManager>
 
 QT_BEGIN_NAMESPACE
 namespace Ui {
@@ -20,10 +22,42 @@ class SnigdhaOSAssistant : public QMainWindow
     Q_OBJECT
 
 public:
-    SnigdhaOSAssistant(QWidget *parent = nullptr);
+    enum class State {
+        QUIT,
+        WELCOME,
+        INTERNET,
+        UPDATE,
+        UPDATE_RETRY,
+        NVIDIA_CHECK,
+        NVIDIA,
+        NVIDIA_APPLY,
+        SELECT,
+        APPLY,
+        APPLY_RETRY,
+        SUCCESS
+    };
+
+    SnigdhaOSAssistant(QWidget* parent = nullptr, QString state = "WELCOME");
     ~SnigdhaOSAssistant();
 
+private slots:
+    void on_textWidget_buttonBox_clicked(QAbstractButton* button);
+    void on_selectWidget_buttonBox_clicked(QAbstractButton* button);
+
+
 private:
     Ui::SnigdhaOSAssistant *ui;
+    QDateTime executable_modify_date;
+    State currentState;
+    void doInternetUpRequest();
+    void doUpdate();
+    void doApply();
+    void doNvidiaCheck();
+    void doNvidiaApply();
+    void populateSelectWidget();
+    void populateSelectWidget(QString filename, QString label);
+    void updateState(State state);
+    void updateState(QString state);
+    void relaunchSelf(QString param);
 };
 #endif // SNIGDHAOSASSISTANT_H