PyParted and a large rewrite of the underlying partitioning (#1604)

* Invert mypy files

* Add optional pre-commit hooks

* New profile structure

* Serialize profiles

* Use profile instead of classmethod

* Custom profile setup

* Separator between back

* Support profile import via url

* Move profiles module

* Refactor files

* Remove symlink

* Add user to docker group

* Update schema description

* Handle list services

* mypy fixes

* mypy fixes

* Rename profilesv2 to profiles

* flake8

* mypy again

* Support selecting DM

* Fix mypy

* Cleanup

* Update greeter setting

* Update schema

* Revert toml changes

* Poc external dependencies

* Dependency support

* New encryption menu

* flake8

* Mypy and flake8

* Unify lsblk command

* Update bootloader configuration

* Git hooks

* Fix import

* Pyparted

* Remove custom font setting

* flake8

* Remove default preview

* Manual partitioning menu

* Update structure

* Disk configuration

* Update filesystem

* luks2 encryption

* Everything works until installation

* Btrfsutil

* Btrfs handling

* Update btrfs

* Save encryption config

* Fix pipewire issue

* Update mypy version

* Update all pre-commit

* Update package versions

* Revert audio/pipewire

* Merge master PRs

* Add master changes

* Merge master changes

* Small renaming

* Pull master changes

* Reset disk enc after disk config change

* Generate locals

* Update naming

* Fix imports

* Fix broken sync

* Fix pre selection on table menu

* Profile menu

* Update profile

* Fix post_install

* Added python-pyparted to PKGBUILD, this requires [testing] to be enabled in order to run makepkg. Package still works via python -m build etc.

* Swaped around some setuptools logic in pyproject

Since we define `package-data` and `packages` there should be no need for:
```
[tool.setuptools.packages.find]
where = ["archinstall", "archinstall.*"]
```

* Removed pyproject collisions. Duplicate definitions.

* Made sure pyproject.toml includes languages

* Add example and update README

* Fix pyproject issues

* Generate locale

* Refactor imports

* Simplify imports

* Add profile description and package examples

* Align code

* Fix mypy

* Simplify imports

* Fix saving config

* Fix wrong luks merge

* Refactor installation

* Fix cdrom device loading

* Fix wrongly merged code

* Fix imports and greeter

* Don't terminate on partprobe error

* Use specific path on partprobe from luks

* Update archinstall/lib/disk/device_model.py

Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com>

* Update archinstall/lib/disk/device_model.py

Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com>

* Update github workflow to test archinstall installation

* Update sway merge

* Generate locales

* Update workflow

---------

Co-authored-by: Daniel Girtler <girtler.daniel@gmail.com>
Co-authored-by: Anton Hvornum <anton@hvornum.se>
Co-authored-by: Anton Hvornum <anton.feeds+github@gmail.com>
Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com>

1 files changed, 218 insertions, 0 deletions

diff --git a/archinstall/default_profiles/custom.py b/archinstall/default_profiles/custom.py
new file mode 100644
index 00000000..f7e100ed
--- /dev/null
+++ b/archinstall/default_profiles/custom.py
@@ -0,0 +1,218 @@
+from typing import List, Dict, Optional, TYPE_CHECKING, Any
+
+from ..lib import menu
+from archinstall.lib.output import log, FormattedOutput
+from archinstall.lib.profile.profiles_handler import profile_handler
+from archinstall.default_profiles.profile import Profile, ProfileType, SelectResult, ProfileInfo, TProfile
+
+if TYPE_CHECKING:
+	from archinstall.lib.installer import Installer
+	_: Any
+
+
+class CustomProfileList(menu.ListManager):
+	def __init__(self, prompt: str, profiles: List[TProfile]):
+		self._actions = [
+			str(_('Add profile')),
+			str(_('Edit profile')),
+			str(_('Delete profile'))
+		]
+		super().__init__(prompt, profiles, [self._actions[0]], self._actions[1:])
+
+	def reformat(self, data: List[TProfile]) -> Dict[str, Optional[TProfile]]:
+		table = FormattedOutput.as_table(data)
+		rows = table.split('\n')
+
+		# these are the header rows of the table and do not map to any profile obviously
+		# we're adding 2 spaces as prefix because the menu selector '> ' will be put before
+		# the selectable rows so the header has to be aligned
+		display_data: Dict[str, Optional[TProfile]] = {f'  {rows[0]}': None, f'  {rows[1]}': None}
+
+		for row, profile in zip(rows[2:], data):
+			row = row.replace('|', '\\|')
+			display_data[row] = profile
+
+		return display_data
+
+	def selected_action_display(self, profile: TProfile) -> str:
+		return profile.name
+
+	def handle_action(
+		self,
+		action: str,
+		entry: Optional['CustomTypeProfile'],
+		data: List['CustomTypeProfile']
+	) -> List['CustomTypeProfile']:
+		if action == self._actions[0]:  # add
+			new_profile = self._add_profile()
+			if new_profile is not None:
+				# in case a profile with the same name as an existing profile
+				# was created we'll replace the existing one
+				data = [d for d in data if d.name != new_profile.name]
+				data += [new_profile]
+		elif entry is not None:
+			if action == self._actions[1]:  # edit
+				new_profile = self._add_profile(entry)
+				if new_profile is not None:
+					# we'll remove the original profile and add the modified version
+					data = [d for d in data if d.name != entry.name and d.name != new_profile.name]
+					data += [new_profile]
+			elif action == self._actions[2]:  # delete
+				data = [d for d in data if d != entry]
+
+		return data
+
+	def _is_new_profile_name(self, name: str) -> bool:
+		existing_profile = profile_handler.get_profile_by_name(name)
+		if existing_profile is not None and existing_profile.profile_type != ProfileType.CustomType:
+			return False
+		return True
+
+	def _add_profile(self, editing: Optional['CustomTypeProfile'] = None) -> Optional['CustomTypeProfile']:
+		name_prompt = '\n\n' + str(_('Profile name: '))
+
+		while True:
+			profile_name = menu.TextInput(name_prompt, editing.name if editing else '').run().strip()
+
+			if not profile_name:
+				return None
+
+			if not self._is_new_profile_name(profile_name):
+				error_prompt = str(_("The profile name you entered is already in use. Try again"))
+				print(error_prompt)
+			else:
+				break
+
+		packages_prompt = str(_('Packages to be install with this profile (space separated, leave blank to skip): '))
+		edit_packages = ' '.join(editing.packages) if editing else ''
+		packages = menu.TextInput(packages_prompt, edit_packages).run().strip()
+
+		services_prompt = str(_('Services to be enabled with this profile (space separated, leave blank to skip): '))
+		edit_services = ' '.join(editing.services) if editing else ''
+		services = menu.TextInput(services_prompt, edit_services).run().strip()
+
+		choice = menu.Menu(
+			str(_('Should this profile be enabled for installation?')),
+			menu.Menu.yes_no(),
+			skip=False,
+			default_option=menu.Menu.no(),
+			clear_screen=False,
+			show_search_hint=False
+		).run()
+
+		enable_profile = True if choice.value == menu.Menu.yes() else False
+
+		profile = CustomTypeProfile(
+			profile_name,
+			enabled=enable_profile,
+			packages=packages.split(' '),
+			services=services.split(' ')
+		)
+
+		return profile
+
+
+# TODO
+# Still needs some ironing out
+class CustomProfile():
+	def __init__(self):
+		super().__init__(
+			'Custom',
+			ProfileType.Custom,
+			description=str(_('Create your own'))
+		)
+
+	def json(self) -> Dict[str, Any]:
+		data: Dict[str, Any] = {'main': self.name, 'gfx_driver': self.gfx_driver, 'custom': []}
+
+		for profile in self._current_selection:
+			data['custom'].append({
+				'name': profile.name,
+				'packages': profile.packages,
+				'services': profile.services,
+				'enabled': profile.custom_enabled
+			})
+
+		return data
+
+	def do_on_select(self) -> SelectResult:
+		custom_profile_list = CustomProfileList('', profile_handler.get_custom_profiles())
+		custom_profiles = custom_profile_list.run()
+
+		# we'll first remove existing custom default_profiles with
+		# the same name and then add the new ones this
+		# will avoid errors of default_profiles with duplicate naming
+		profile_handler.remove_custom_profiles(custom_profiles)
+		profile_handler.add_custom_profiles(custom_profiles)
+
+		self.set_current_selection(custom_profiles)
+
+		if custom_profile_list.is_last_choice_cancel():
+			return SelectResult.SameSelection
+
+		enabled_profiles = [p for p in self._current_selection if p.custom_enabled]
+		# in  case we only created inactive default_profiles we wanna store them but
+		# we want to reset the original setting
+		if not enabled_profiles:
+			return SelectResult.ResetCurrent
+
+		return SelectResult.NewSelection
+
+	def post_install(self, install_session: 'Installer'):
+		for profile in self._current_selection:
+			profile.post_install(install_session)
+
+	def install(self, install_session: 'Installer'):
+		driver_packages = self.gfx_driver_packages()
+		install_session.add_additional_packages(driver_packages)
+
+		for profile in self._current_selection:
+			if profile.custom_enabled:
+				log(f'Installing custom profile {profile.name}...')
+
+				install_session.add_additional_packages(profile.packages)
+				install_session.enable_service(profile.services)
+
+				profile.install(install_session)
+
+	def info(self) -> Optional[ProfileInfo]:
+		enabled_profiles = [p for p in self._current_selection if p.custom_enabled]
+		if enabled_profiles:
+			details = ', '.join([p.name for p in enabled_profiles])
+			gfx_driver = self.gfx_driver
+			return ProfileInfo(self.name, details, gfx_driver)
+
+		return None
+
+	def reset(self):
+		for profile in self._current_selection:
+			profile.set_enabled(False)
+
+		self.gfx_driver = None
+
+
+class CustomTypeProfile(Profile):
+	def __init__(
+		self,
+		name: str,
+		enabled: bool = False,
+		packages: List[str] = [],
+		services: List[str] = []
+	):
+		super().__init__(
+			name,
+			ProfileType.CustomType,
+			packages=packages,
+			services=services,
+			support_gfx_driver=True
+		)
+
+		self.custom_enabled = enabled
+
+	def json(self) -> Dict[str, Any]:
+		return {
+			'name': self.name,
+			'packages': self.packages,
+			'services': self.services,
+			'enabled': self.custom_enabled
+		}