Add preferences for customreboot

arelange · Mar 27, 2024 · 45e962e · 45e962e
1 parent f0e6e64
commit 45e962e
Show file tree

Hide file tree

Showing 4 changed files with 98 additions and 106 deletions.
diff --git a/customreboot/bootloader.js b/customreboot/bootloader.js
@@ -20,7 +20,7 @@ export class Bootloader {
      * @returns BootLoaders type. Can be "EFI", "SYSD", "GRUB", or "UNKNOWN"
      */
     static async GetUseableType() {
-        const settings = Extension.lookupByUUID('customreboot@nova1545').getSettings('org.gnome.shell.extensions.customreboot');
+        const settings = Extension.lookupByUUID('hibernate-status@dromi').getSettings('org.gnome.shell.extensions.hibernate-status-button');
 
         if (await EFIBootManager.IsUseable() && settings.get_boolean('use-efibootmgr')) return BootLoaders.EFI;
         if (await Grub.IsUseable() && settings.get_boolean('use-grub')) return BootLoaders.GRUB;
@@ -38,4 +38,4 @@ export class Bootloader {
         if (type === BootLoaders.GRUB) return Grub;
         return undefined;
     }
-}
+}
diff --git a/customreboot/prefs.js b/customreboot/prefs.js
diff --git a/extension.js b/extension.js
@@ -500,6 +500,11 @@ export default class HibernateButtonExtension extends Extension {
             this._hibernateMenuItem = 0;
         }
 
+        if (this._customRestartMenuItem) {
+            this._customRestartMenuItem.destroy();
+            this._customRestartMenuItem = 0;
+        }
+
         if (this.sourceId) {
             GLib.Source.remove(this.sourceId);
             this.sourceId = null;

diff --git a/prefs.js b/prefs.js
@@ -3,7 +3,9 @@ import Gtk from 'gi://Gtk';
 import Adw from 'gi://Adw';
 // Use __() and N__() for the extension gettext domain, and reuse
 // the shell domain with the default _() and N_()
-
+import { EFIBootManager } from './customreboot/efibootmgr.js';
+import { Grub } from './customreboot/grub.js';
+import { SystemdBoot } from './customreboot/systemdBoot.js';
 import {
     ExtensionPreferences,
     gettext as __,
@@ -164,5 +166,93 @@ export default class Prefs extends ExtensionPreferences {
             Gio.SettingsBindFlags.DEFAULT);
         window._settings.bind('show-suspend-then-hibernate-dialog', suspend_then_hibernate_dialog_row, 'active',
             Gio.SettingsBindFlags.DEFAULT);
+
+        const reboot_page = new Adw.PreferencesPage({
+            title: __('Custom Reboot'),
+            icon_name: 'system-reboot-symbolic',
+        });
+        const reboot_group = new Adw.PreferencesGroup();
+        reboot_page.add(reboot_group);
+
+        // Create row for efibootmgr
+        const efi = new Adw.ActionRow({ title: 'Use EFI Boot Manager (efibootmgr)' });
+
+        // Create row for grub
+        const grub = new Adw.ActionRow({ title: 'Use Grub'});
+
+        // Create row for systemd-boot
+        const sysd = new Adw.ActionRow({ title: 'Use Systemd Boot'});
+
+        // Add rows
+        reboot_group.add(efi);
+        reboot_group.add(grub);
+        reboot_group.add(sysd);
+
+        let settings = window._settings
+
+        // Create switch for efibootmgr
+        const efi_switch = new Gtk.Switch({
+            active: settings.get_boolean('use-efibootmgr'),
+            valign: Gtk.Align.CENTER,
+        });
+
+        // Create switch for grub
+        const grub_switch = new Gtk.Switch({
+            active: settings.get_boolean('use-grub'),
+            valign: Gtk.Align.CENTER,
+        });
+
+        // Create switch for systemd-boot
+        const sysd_switch = new Gtk.Switch({
+            active: settings.get_boolean('use-systemd-boot'),
+            valign: Gtk.Align.CENTER,
+        });
+
+        // Bind settings for efibootmgr
+        settings.bind(
+            'use-efibootmgr',
+            efi_switch,
+            'active',
+            Gio.SettingsBindFlags.DEFAULT
+        );
+
+        // Bind settings for grub
+        settings.bind(
+            'use-grub',
+            grub_switch,
+            'active',
+            Gio.SettingsBindFlags.DEFAULT
+        );
+
+        // Bind settings for systemd-boot
+        settings.bind(
+            'use-systemd-boot',
+            sysd_switch,
+            'active',
+            Gio.SettingsBindFlags.DEFAULT
+        );
+
+        // Add the switch for efibootmgr
+        efi.add_suffix(efi_switch);
+        efi.activatable_widget = efi_switch;
+
+        // Add the switch for grub
+        grub.add_suffix(grub_switch);
+        grub.activatable_widget = grub_switch;
+
+        // Add the switch for systemd-boot
+        sysd.add_suffix(sysd_switch);
+        sysd.activatable_widget = sysd_switch;
+
+        // Add our page to the window
+        window.add(reboot_page);
+
+        (async () => {
+            // Disable/enable switches in accordance to them being usable
+
+            efi_switch.set_sensitive(await EFIBootManager.IsUseable());
+            grub_switch.set_sensitive(await Grub.IsUseable());
+            sysd_switch.set_sensitive(await SystemdBoot.IsUseable());
+        })();
     }
 }