Sway profile - select seat (#1661)

* sway profile choose seat

1 files changed, 37 insertions, 2 deletions

diff --git a/profiles/sway.py b/profiles/sway.py
index b7266da3..5fbd3365 100644
--- a/profiles/sway.py
+++ b/profiles/sway.py
@@ -1,6 +1,12 @@
 # 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
 
@@ -22,7 +28,7 @@ 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?'
+		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():
@@ -30,6 +36,18 @@ def _check_driver() -> bool:
 
 	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):
 	"""
@@ -38,6 +56,9 @@ def _prep_function(*args, **kwargs):
 	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:
@@ -49,15 +70,29 @@ def _prep_function(*args, **kwargs):
 	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.")
+		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))}")