From ba070124f9c38a67fb80cbc357d1d2bba8832cb9 Mon Sep 17 00:00:00 2001
From: mirkobrombin <send@mirko.pm>
Date: Sat, 10 Sep 2022 15:06:58 +0200
Subject: [PATCH] ask to reboot at final step

---
 vanilla_first_setup/gtk/window.ui         | 6 +++---
 vanilla_first_setup/utils/configurator.py | 9 ++++++++-
 vanilla_first_setup/window.py             | 9 +++++----
 3 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/vanilla_first_setup/gtk/window.ui b/vanilla_first_setup/gtk/window.ui
index 87c07cb..e7b4d61 100644
--- a/vanilla_first_setup/gtk/window.ui
+++ b/vanilla_first_setup/gtk/window.ui
@@ -311,13 +311,13 @@
               <object class="AdwStatusPage">
                 <property name="icon-name">emblem-default-symbolic</property>
                 <property name="title">Done!</property>
-                <property name="description">You can now enjoy your Vanila OS experience.</property>
+                <property name="description">Restart your PC to enjoy your Vanila OS experience.</property>
                 <property name="halign">fill</property>
                 <property name="valign">fill</property>
                 <property name="hexpand">true</property>
                 <child>
-                  <object class="GtkButton" id="btn_close">
-                    <property name="label">Close</property>
+                  <object class="GtkButton" id="btn_reboot">
+                    <property name="label">Restart</property>
                     <property name="halign">center</property>
                     <style>
                       <class name="pill" />
diff --git a/vanilla_first_setup/utils/configurator.py b/vanilla_first_setup/utils/configurator.py
index 5006617..1b35a75 100644
--- a/vanilla_first_setup/utils/configurator.py
+++ b/vanilla_first_setup/utils/configurator.py
@@ -32,7 +32,7 @@ class Configurator:
     def __fake(self, msg: str):
         time.sleep(1)
         logger.info(f"Fake: {msg}")
-
+    
     def __enable_snap(self):
         if self.fake:
             return self.__fake("Fake: Snap enabled")
@@ -120,3 +120,10 @@ class Configurator:
         autostart_file = os.path.expanduser("~/.config/autostart/vanilla-first-setup.desktop")
         if os.path.exists(autostart_file):
             os.remove(autostart_file)
+
+    @staticmethod
+    def reboot():
+        if self.fake:
+            return self.__fake("Fake: Reboot")
+
+        subprocess.run(['gnome-session-quit', '--reboot'])
diff --git a/vanilla_first_setup/window.py b/vanilla_first_setup/window.py
index dcd5d3e..698117f 100644
--- a/vanilla_first_setup/window.py
+++ b/vanilla_first_setup/window.py
@@ -21,6 +21,7 @@ from vanilla_first_setup.models.preset import Preset
 from vanilla_first_setup.models.config import Config
 from vanilla_first_setup.utils.processor import Processor
 from vanilla_first_setup.utils.run_async import RunAsync
+from vanilla_first_setup.utils.configurator import Configurator
 from vanilla_first_setup.utils.welcome_langs import welcome
 from vanilla_first_setup.utils.checks import has_nvidia_gpu
 from vanilla_first_setup.dialogs.subsystem import SubSystemDialog
@@ -35,7 +36,7 @@ class FirstSetupWindow(Adw.ApplicationWindow):
     btn_start = Gtk.Template.Child()
     btn_next = Gtk.Template.Child()
     btn_save = Gtk.Template.Child()
-    btn_close = Gtk.Template.Child()
+    btn_reboot = Gtk.Template.Child()
     btn_no_subsystem = Gtk.Template.Child()
     btn_use_subsystem = Gtk.Template.Child()
     btn_info_subsystem = Gtk.Template.Child()
@@ -85,7 +86,7 @@ class FirstSetupWindow(Adw.ApplicationWindow):
         self.btn_start.connect('clicked', self.__on_btn_start_clicked)
         self.btn_next.connect('clicked', self.__on_btn_next_clicked)
         self.btn_save.connect('clicked', self.on_btn_save_clicked)
-        self.btn_close.connect('clicked', self.on_btn_close_clicked)
+        self.btn_reboot.connect('clicked', self.on_btn_reboot_clicked)
 
         self.btn_no_subsystem.connect('clicked', self.on_btn_subsystem_clicked, False)
         self.btn_use_subsystem.connect('clicked', self.on_btn_subsystem_clicked, True)
@@ -133,8 +134,8 @@ class FirstSetupWindow(Adw.ApplicationWindow):
     def __on_switch_distrobox_state_set(self, widget, state):
         self.__config.set_val('distrobox', state)
 
-    def on_btn_close_clicked(self, widget):
-        self.get_application().quit()
+    def on_btn_reboot_clicked(self, widget):
+        Configurator.reboot()
 
     def on_btn_subsystem_clicked(self, widget, state):
         self.__config.set_val('distrobox', state)