1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
import * as Main from 'resource:///org/gnome/shell/ui/main.js';
import * as PopupMenu from 'resource:///org/gnome/shell/ui/popupMenu.js';
import * as QuickSettings from 'resource:///org/gnome/shell/ui/quickSettings.js';
import { Extension } from "resource:///org/gnome/shell/extensions/extension.js";
import { getDefault } from "resource:///org/gnome/shell/misc/systemActions.js";
import GObject from "gi://GObject";
//import
import { LogWarning, Log } from './utils.js';
import { BootLoaders, Bootloader } from "./bootloader.js";
const RebootQuickMenu = GObject.registerClass(
class RebootQuickMenu extends QuickSettings.QuickMenuToggle {
_init(extension) {
const init_params = { iconName: 'system-reboot-symbolic', toggleMode: false };
init_params.title = "Reboot Into";
super._init(init_params);
// Set toggle to be unchecked
this.checked = false;
// Open menu and set toggle check to true
this.clickedID = this.connect("clicked", () => {
this.menu.open();
this.checked = true;
});
// Connect 'menu-closed' to update checked state
this.menuClosedID = this.menu.connect("menu-closed", () => {
this.checked = false;
});
// Add boot options to menu
try {
this.createBootMenu(extension);
}
catch (e) {
LogWarning(e);
}
}
cleanConns() {
// Disconnect all connections
this.disconnect(this.openStateID);
this.disconnect(this.clickedID);
}
async createBootMenu(extension) {
// Get boot options
const type = await Bootloader.GetUseableType(extension);
const header_title = `Boot Options - ${type}`;
// Set Menu Header
this.menu.setHeader('system-reboot-symbolic', header_title, 'Reboot into the selected entry');
const loader = await Bootloader.GetUseable(type);
if (loader === undefined) {
// Set Menu Header
this.menu.setHeader('system-reboot-symbolic', 'Error', 'The selected boot loader cannot be found...');
// Add reload option, to refresh extension menu without reloading GNOME or the extension
this.menu.addMenuItem(new PopupMenu.PopupSeparatorMenuItem());
this.menu.addAction('Reload', () => {
this.menu.removeAll();
this.createBootMenu();
});
// Add button to open settings
this.menu.addAction('Settings', () => {
extension.openPreferences();
});
return;
}
loader.GetBootOptions().then(([bootOps, defaultOpt]) => {
if (bootOps !== undefined) {
this._itemsSection = new PopupMenu.PopupMenuSection();
for (let [title, id] of bootOps) {
Log(`${title} - ${id}`);
this._itemsSection.addAction(String(title), () => {
// Set boot option
loader.SetBootOption(String(id)).then(result => {
if (result) {
// On success trigger restart dialog
new getDefault().activateRestart();
}
});
}, (title === defaultOpt || id === defaultOpt)? "pan-end-symbolic" : undefined);
}
this.menu.addMenuItem(this._itemsSection);
}
// Add reload option, to refresh extension menu without reloading GNOME or the extension
this.menu.addMenuItem(new PopupMenu.PopupSeparatorMenuItem());
this.menu.addAction('Reload', () => {
this.menu.removeAll();
this.createBootMenu();
});
// Add button to open settings
this.menu.addAction('Settings', () => {
extension.openPreferences();
});
loader.CanQuickReboot().then(async result => {
if (!result) return;
if (!await loader.QuickRebootEnabled()) {
this.menu.addAction('Enable Quick Reboot', async () => {
await loader.EnableQuickReboot(extension);
this.menu.removeAll();
this.createBootMenu();
});
}
else {
this.menu.addAction('Disable Quick Reboot', async () => {
await loader.DisableQuickReboot();
this.menu.removeAll();
this.createBootMenu();
});
}
});
}).catch((error) => {
LogWarning(error);
// Only do this if the current bootloader is grub
if (type === BootLoaders.GRUB)
{
// Only add this if all fails, giving user option to make the config readable
this.menu.addMenuItem(new PopupSeparatorMenuItem());
this.menu.addAction('Fix Grub.conf Perms', async () => {
await loader.SetReadable();
this.menu.removeAll();
this.createBootMenu(extension);
});
}
})
}
});
export default class CustomReboot extends Extension {
enable() {
// Add items to QuickSettingsMenu
this._indicator = new QuickSettings.SystemIndicator();
this.menu = new RebootQuickMenu(this);
this._indicator.quickSettingsItems.push(this.menu);
Main.panel.statusArea.quickSettings.addExternalIndicator(this._indicator, 1);
}
disable() {
this.menu.cleanConns();
this.menu = null;
this._indicator.quickSettingsItems.forEach(item => {
item.destroy();
});
this._indicator.destroy();
this._indicator = null;
}
}
|