119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
from archinstall.lib.translationhandler import tr
|
|
from archinstall.tui.curses_menu import SelectMenu
|
|
from archinstall.tui.menu_item import MenuItem, MenuItemGroup
|
|
from archinstall.tui.result import ResultType
|
|
from archinstall.tui.types import Alignment, FrameProperties, FrameStyle, Orientation, PreviewStyle
|
|
|
|
from ..hardware import GfxDriver, SysInfo
|
|
|
|
|
|
def select_kernel(preset: list[str] = []) -> list[str]:
|
|
"""
|
|
Asks the user to select a kernel for system.
|
|
|
|
:return: The string as a selected kernel
|
|
:rtype: string
|
|
"""
|
|
kernels = ['linux', 'linux-lts', 'linux-zen', 'linux-hardened']
|
|
default_kernel = 'linux'
|
|
|
|
items = [MenuItem(k, value=k) for k in kernels]
|
|
|
|
group = MenuItemGroup(items, sort_items=True)
|
|
group.set_default_by_value(default_kernel)
|
|
group.set_focus_by_value(default_kernel)
|
|
group.set_selected_by_value(preset)
|
|
|
|
result = SelectMenu[str](
|
|
group,
|
|
allow_skip=True,
|
|
allow_reset=True,
|
|
alignment=Alignment.CENTER,
|
|
frame=FrameProperties.min(tr('Kernel')),
|
|
multi=True,
|
|
).run()
|
|
|
|
match result.type_:
|
|
case ResultType.Skip:
|
|
return preset
|
|
case ResultType.Reset:
|
|
return []
|
|
case ResultType.Selection:
|
|
return result.get_values()
|
|
|
|
|
|
def select_driver(options: list[GfxDriver] = [], preset: GfxDriver | None = None) -> GfxDriver | None:
|
|
"""
|
|
Some what convoluted function, whose job is simple.
|
|
Select a graphics driver from a pre-defined set of popular options.
|
|
|
|
(The template xorg is for beginner users, not advanced, and should
|
|
there for appeal to the general public first and edge cases later)
|
|
"""
|
|
if not options:
|
|
options = [driver for driver in GfxDriver]
|
|
|
|
items = [MenuItem(o.value, value=o, preview_action=lambda x: x.value.packages_text()) for o in options]
|
|
group = MenuItemGroup(items, sort_items=True)
|
|
group.set_default_by_value(GfxDriver.AllOpenSource)
|
|
|
|
if preset is not None:
|
|
group.set_focus_by_value(preset)
|
|
|
|
header = ''
|
|
if SysInfo.has_amd_graphics():
|
|
header += tr('For the best compatibility with your AMD hardware, you may want to use either the all open-source or AMD / ATI options.') + '\n'
|
|
if SysInfo.has_intel_graphics():
|
|
header += tr('For the best compatibility with your Intel hardware, you may want to use either the all open-source or Intel options.\n')
|
|
if SysInfo.has_nvidia_graphics():
|
|
header += tr('For the best compatibility with your Nvidia hardware, you may want to use the Nvidia proprietary driver.\n')
|
|
|
|
result = SelectMenu[GfxDriver](
|
|
group,
|
|
header=header,
|
|
allow_skip=True,
|
|
allow_reset=True,
|
|
preview_size='auto',
|
|
preview_style=PreviewStyle.BOTTOM,
|
|
preview_frame=FrameProperties(tr('Info'), h_frame_style=FrameStyle.MIN),
|
|
).run()
|
|
|
|
match result.type_:
|
|
case ResultType.Skip:
|
|
return preset
|
|
case ResultType.Reset:
|
|
return None
|
|
case ResultType.Selection:
|
|
return result.get_value()
|
|
|
|
|
|
def ask_for_swap(preset: bool = True) -> bool:
|
|
if preset:
|
|
default_item = MenuItem.yes()
|
|
else:
|
|
default_item = MenuItem.no()
|
|
|
|
prompt = tr('Would you like to use swap on zram?') + '\n'
|
|
|
|
group = MenuItemGroup.yes_no()
|
|
group.set_focus_by_value(default_item)
|
|
|
|
result = SelectMenu[bool](
|
|
group,
|
|
header=prompt,
|
|
columns=2,
|
|
orientation=Orientation.HORIZONTAL,
|
|
alignment=Alignment.CENTER,
|
|
allow_skip=True,
|
|
).run()
|
|
|
|
match result.type_:
|
|
case ResultType.Skip:
|
|
return preset
|
|
case ResultType.Selection:
|
|
return result.item() == MenuItem.yes()
|
|
case ResultType.Reset:
|
|
raise ValueError('Unhandled result type')
|