1 files changed, 47 insertions, 0 deletions
diff --git a/archinstall/lib/models/bootloader.py b/archinstall/lib/models/bootloader.py
new file mode 100644
index 00000000..aa1a8e27
--- /dev/null
+++ b/archinstall/lib/models/bootloader.py
@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+import sys
+from enum import Enum
+from typing import List
+
+from ..hardware import SysInfo
+from ..output import warn
+
+
+class Bootloader(Enum):
+	Systemd = 'Systemd-boot'
+	Grub = 'Grub'
+	Efistub = 'Efistub'
+	Limine = 'Limine'
+
+	def has_uki_support(self) -> bool:
+		match self:
+			case Bootloader.Efistub | Bootloader.Systemd:
+				return True
+			case _:
+				return False
+
+	def json(self) -> str:
+		return self.value
+
+	@staticmethod
+	def values() -> List[str]:
+		return [e.value for e in Bootloader]
+
+	@classmethod
+	def get_default(cls) -> Bootloader:
+		if SysInfo.has_uefi():
+			return Bootloader.Systemd
+		else:
+			return Bootloader.Grub
+
+	@classmethod
+	def from_arg(cls, bootloader: str) -> Bootloader:
+		# to support old configuration files
+		bootloader = bootloader.capitalize()
+
+		if bootloader not in cls.values():
+			values = ', '.join(cls.values())
+			warn(f'Invalid bootloader value "{bootloader}". Allowed values: {values}')
+			sys.exit(1)
+		return Bootloader(bootloader)