import archinstall, os

# TODO: Remove hard dependency of bash (due to .bash_profile)

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.
	"""

	__builtins__['_gfx_driver_packages'] = archinstall.select_driver()

	return True

def _post_install(*args, **kwargs):
	choice = input("Would you like to autostart sway on login [Y/n]: ")
	if choice.lower == "y":
		with open(f"{installation.mountpoint}/etc/profile", "a") as f:
			x = """
			if [ -z $DISPLAY ] && [ "$(tty)" == "/dev/tty1" ]; then
  				exec sway
			fi
			"""
			f.write(x)
			f.close()
	else:
		installation.log("To start Sway, run the 'sway' command after logging in.")

# Ensures that this code only gets executed if executed
# through importlib.util.spec_from_file_location("kde", "/somewhere/kde.py")
# or through conventional import kde
if __name__ == 'sway':

	installation.add_additional_packages(_gfx_driver_packages)

	# Install dependency profiles
    if _gfx_driver_packages == 'nvidia':
		# NOTE: This is technically runnable with the --my-next-gpu-wont-be-nvidia option
		raise archinstall.lib.exceptions.HardwareIncompatibilityError("Sway does not officially support the proprietary Nvidia driver, you may have to use nouveau.")

	# Install the application kde from the template under /applications/
	sway = archinstall.Application(installation, 'sway')
	sway.install()