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
|
# A desktop environment using "Sway"
from typing import Any, TYPE_CHECKING
import archinstall
from archinstall import Menu
from archinstall.lib.menu.menu import MenuSelectionType
if TYPE_CHECKING:
_: Any
is_top_level_profile = False
__packages__ = [
"sway",
"swaylock",
"swayidle",
"waybar",
"dmenu",
"light",
"grim",
"slurp",
"pavucontrol",
"foot",
]
def _check_driver() -> bool:
packages = archinstall.storage.get("gfx_driver_packages", [])
if packages and "nvidia" in packages:
prompt = _('The proprietary Nvidia driver is not supported by Sway. It is likely that you will run into issues, are you okay with that?')
choice = Menu(prompt, Menu.yes_no(), default_option=Menu.no(), skip=False).run()
if choice.value == Menu.no():
return False
return True
def _get_system_privelege_control_preference():
# need to activate seat service and add to seat group
title = str(_('Sway needs access to your seat (collection of hardware devices i.e. keyboard, mouse, etc)'))
title += str(_('\n\nChoose an option to give Sway access to your hardware'))
choice = Menu(title, ["polkit", "seatd"]).run()
if choice.type_ != MenuSelectionType.Selection:
return False
archinstall.storage['sway_sys_priv_ctrl'] = [choice.value]
archinstall.arguments['sway_sys_priv_ctrl'] = [choice.value]
return True
def _prep_function(*args, **kwargs):
"""
Magic function called by the importing installer
before continuing any further. It also avoids executing any
other code in this stage. So it's a safe way to ask the user
for more input before any other installer steps start.
"""
if not _get_system_privelege_control_preference():
return False
driver = archinstall.select_driver()
if driver:
archinstall.storage["gfx_driver_packages"] = driver
if not _check_driver():
return _prep_function(args, kwargs)
return True
return False
"""
def _post_install(*args, **kwargs):
if "seatd" in sway_sys_priv_ctrl:
print(_('After booting, add user(s) to the `seat` user group and re-login to use Sway'))
return True
"""
# Ensures that this code only gets executed if executed
# through importlib.util.spec_from_file_location("sway", "/somewhere/sway.py")
# or through conventional import sway
if __name__ == "sway":
if not _check_driver():
raise archinstall.lib.exceptions.HardwareIncompatibilityError(_('Sway does not support the proprietary nvidia drivers.'))
# Install the Sway packages
archinstall.storage['installation_session'].add_additional_packages(__packages__)
if "seatd" in archinstall.storage['sway_sys_priv_ctrl']:
archinstall.storage['installation_session'].add_additional_packages(['seatd'])
archinstall.storage['installation_session'].enable_service('seatd')
elif "polkit" in archinstall.storage['sway_sys_priv_ctrl']:
archinstall.storage['installation_session'].add_additional_packages(['polkit'])
else:
raise archinstall.lib.exceptions.ProfileError(_('Sway requires either seatd or polkit to run'))
# Install the graphics driver packages
archinstall.storage['installation_session'].add_additional_packages(f"xorg-server xorg-xinit {' '.join(archinstall.storage.get('gfx_driver_packages', None))}")
|