From a94ee6b50339cabac069638b5ee2f70e1308048a Mon Sep 17 00:00:00 2001
From: Victor Tran <vicr12345@gmail.com>
Date: Sat, 13 Oct 2018 16:18:34 +1100
Subject: Add some checks

---
 installer/mainwindow.h | 54 ++++++++++++++++++++++++++++----------------------
 1 file changed, 30 insertions(+), 24 deletions(-)

(limited to 'installer/mainwindow.h')

diff --git a/installer/mainwindow.h b/installer/mainwindow.h
index 4eb89d2..d5f8bb2 100644
--- a/installer/mainwindow.h
+++ b/installer/mainwindow.h
@@ -23,6 +23,7 @@
 
 #include <windows.h>
 #include <shellapi.h>
+#include <Psapi.h>
 
 namespace Ui {
 class MainWindow;
@@ -32,44 +33,49 @@ class MainWindow : public QMainWindow
 {
     Q_OBJECT
 
-public:
-    explicit MainWindow(QWidget *parent = 0);
-    ~MainWindow();
+    public:
+        explicit MainWindow(QWidget *parent = 0);
+        ~MainWindow();
 
-private slots:
-    void on_installOptions_clicked();
+    private slots:
+        void on_installOptions_clicked();
 
-    void getInstallerMetadata();
+        void getInstallerMetadata();
 
-    void setInstallPath();
+        void setInstallPath();
 
-    void on_cancelMetadataButton_clicked();
+        void on_cancelMetadataButton_clicked();
 
-    void on_retryMetadataButton_clicked();
+        void on_retryMetadataButton_clicked();
 
-    void on_installEveryone_toggled(bool checked);
+        void on_installEveryone_toggled(bool checked);
 
-    void on_installButton_clicked();
+        void on_installButton_clicked();
 
-    void on_installButton_2_clicked();
+        void on_installButton_2_clicked();
 
-    void on_exitButton_clicked();
+        void on_exitButton_clicked();
 
-    void on_retryInstallButton_clicked();
+        void on_retryInstallButton_clicked();
 
-    void on_finishButton_clicked();
+        void on_finishButton_clicked();
 
-private:
-    Ui::MainWindow *ui;
+        void on_cancelInstallButton_clicked();
 
-    void paintEvent(QPaintEvent* event);
-    void showEvent(QShowEvent* event);
+        void on_browseInstallPathButton_clicked();
 
-    QNetworkAccessManager mgr;
-    QPixmap backgroundImage;
-    QJsonObject metadata;
-    QWinTaskbarButton* taskbarButton;
-    bool installDone = false;
+    private:
+        Ui::MainWindow *ui;
+
+        void paintEvent(QPaintEvent* event);
+        void showEvent(QShowEvent* event);
+
+        QNetworkAccessManager mgr;
+        QPixmap backgroundImage;
+        QJsonObject metadata;
+        QWinTaskbarButton* taskbarButton;
+        QLocalSocket* sock = nullptr;
+        bool installDone = false;
 };
 
 #endif // MAINWINDOW_H
-- 
cgit v1.2.3