# A desktop environment selector.
from typing import Any, TYPE_CHECKING

import archinstall
from archinstall import log, Menu
from archinstall.lib.menu.menu import MenuSelectionType

if TYPE_CHECKING:
	_: Any

is_top_level_profile = True

__description__ = str(_('Provides a selection of desktop environments and tiling window managers, e.g. gnome, kde, sway'))

# New way of defining packages for a profile, which is iterable and can be used out side
# of the profile to get a list of "what packages will be installed".
__packages__ = [
	'nano',
	'vim',
	'openssh',
	'htop',
	'wget',
	'iwd',
	'wireless_tools',
	'wpa_supplicant',
	'smartmontools',
	'xdg-utils',
]

__supported__ = [
	'gnome',
	'kde',
	'awesome',
	'sway',
	'cinnamon',
	'xfce4',
	'lxqt',
	'i3',
	'bspwm',
	'budgie',
	'mate',
	'deepin',
	'enlightenment',
	'qtile'
]


def _prep_function(*args, **kwargs) -> bool:
	"""
	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.
	"""
	choice = Menu(str(_('Select your desired desktop environment')), __supported__).run()

	if choice.type_ != MenuSelectionType.Selection:
		return False

	if choice.value:
		# Temporarily store the selected desktop profile
		# in a session-safe location, since this module will get reloaded
		# the next time it gets executed.
		if not archinstall.storage.get('_desktop_profile', None):
			archinstall.storage['_desktop_profile'] = choice.value
		if not archinstall.arguments.get('desktop-environment', None):
			archinstall.arguments['desktop-environment'] = choice.value
		profile = archinstall.Profile(None, choice.value)
		# Loading the instructions with a custom namespace, ensures that a __name__ comparison is never triggered.
		with profile.load_instructions(namespace=f"{choice.value}.py") as imported:
			if hasattr(imported, '_prep_function'):
				return imported._prep_function()
			else:
				log(f"Deprecated (??): {choice.value} profile has no _prep_function() anymore")
				exit(1)

	return False


if __name__ == 'desktop':
	"""
	This "profile" is a meta-profile.
	There are no desktop-specific steps, it simply routes
	the installer to whichever desktop environment/window manager was chosen.

	Maybe in the future, a network manager or similar things *could* be added here.
	We should honor that Arch Linux does not officially endorse a desktop-setup, nor is
	it trying to be a turn-key desktop distribution.

	There are plenty of desktop-turn-key-solutions based on Arch Linux,
	this is therefore just a helper to get started
	"""

	# Install common packages for all desktop environments
	archinstall.storage['installation_session'].add_additional_packages(__packages__)

	archinstall.storage['installation_session'].install_profile(archinstall.storage['_desktop_profile'])