Add handling of signal interrupt and EOF at input prompts (#2154)

4 files changed, 23 insertions, 10 deletions

diff --git a/archinstall/lib/interactions/disk_conf.py b/archinstall/lib/interactions/disk_conf.py
index 84a3196c..c18119ec 100644
--- a/archinstall/lib/interactions/disk_conf.py
+++ b/archinstall/lib/interactions/disk_conf.py
@@ -134,7 +134,10 @@ def select_disk_config(
 				output = "You will use whatever drive-setup is mounted at the specified directory\n"
 				output += "WARNING: Archinstall won't check the suitability of this setup\n"
 
-				path = prompt_dir(str(_('Enter the root directory of the mounted devices: ')), output)
+				try:
+					path = prompt_dir(str(_('Enter the root directory of the mounted devices: ')), output)
+				except (KeyboardInterrupt, EOFError):
+					return preset
 				mods = disk.device_handler.detect_pre_mounted_mods(path)
 
 				return disk.DiskLayoutConfiguration(
diff --git a/archinstall/lib/interactions/general_conf.py b/archinstall/lib/interactions/general_conf.py
index a23426d0..b12a6fb8 100644
--- a/archinstall/lib/interactions/general_conf.py
+++ b/archinstall/lib/interactions/general_conf.py
@@ -28,14 +28,15 @@ def ask_ntp(preset: bool = True) -> bool:
 
 
 def ask_hostname(preset: str = '') -> str:
-	while True:
-		hostname = TextInput(
-			str(_('Desired hostname for the installation: ')),
-			preset
-		).run().strip()
+	hostname = TextInput(
+		str(_('Desired hostname for the installation: ')),
+		preset
+	).run().strip()
+
+	if not hostname:
+		return preset
 
-		if hostname:
-			return hostname
+	return hostname
 
 
 def ask_for_a_timezone(preset: Optional[str] = None) -> Optional[str]:
diff --git a/archinstall/lib/interactions/manage_users_conf.py b/archinstall/lib/interactions/manage_users_conf.py
index 879578da..ca912283 100644
--- a/archinstall/lib/interactions/manage_users_conf.py
+++ b/archinstall/lib/interactions/manage_users_conf.py
@@ -75,7 +75,11 @@ class UserList(ListManager):
 		prompt = '\n\n' + str(_('Enter username (leave blank to skip): '))
 
 		while True:
-			username = input(prompt).strip(' ')
+			try:
+				username = input(prompt).strip(' ')
+			except (KeyboardInterrupt, EOFError):
+				return None
+
 			if not username:
 				return None
 			if not self._check_for_correct_username(username):
diff --git a/archinstall/lib/interactions/utils.py b/archinstall/lib/interactions/utils.py
index f6b5b2d3..fdbb4625 100644
--- a/archinstall/lib/interactions/utils.py
+++ b/archinstall/lib/interactions/utils.py
@@ -17,7 +17,12 @@ def get_password(prompt: str = '') -> Optional[str]:
 	if not prompt:
 		prompt = _("Enter a password: ")
 
-	while password := getpass.getpass(prompt):
+	while True:
+		try:
+			password = getpass.getpass(prompt)
+		except (KeyboardInterrupt, EOFError):
+			break
+
 		if len(password.strip()) <= 0:
 			break