1 files changed, 12 insertions, 3 deletions
diff --git a/ui/qt5/horizonwizard.hh b/ui/qt5/horizonwizard.hh
index 8af9df7..ce5c22d 100644
--- a/ui/qt5/horizonwizard.hh
+++ b/ui/qt5/horizonwizard.hh
@@ -122,6 +122,13 @@ public:
         Custom
     };
 
+    enum DesktopType {
+        Plasma,
+        LXQt,
+        MATE,
+        XFCE
+    };
+
     enum BinShProvider {
         Dash,
         Bash
@@ -132,9 +139,9 @@ public:
         SysVInit
     };
 
-    HorizonWizard(QWidget *parent = nullptr);
-    void accept();
-    void reject();
+    explicit HorizonWizard(QWidget *parent = nullptr);
+    void accept() override;
+    void reject() override;
     /*! Emit a HorizonScript file with the user's choices. */
     QString toHScript();
     QShortcut *f1, *f3, *f6, *f8, *f10;
@@ -190,6 +197,8 @@ public:
     InitSystem sbininit;
     /*! Determines the packages to install. */
     PackageType pkgtype;
+    /*! If pkgtype is Standard, which desktop environment to install. */
+    DesktopType desktopType;
     /*! If pkgtype is Custom, a list of packages to install. */
     QStringList packages;
     /*! Determines the kernel to install. */