""" Interface Controller Module. This module provides the main navigation interface and button controls for the Install Station GTK application wizard. """ import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk from install_station.install import InstallProgress, InstallWindow from install_station.partition import DiskPartition from install_station.window import Window from install_station.data import InstallationData, get_text from install_station.system_calls import localize_system, set_keyboard class Button: """ Button management class for navigation controls. Manages the Back, Cancel, and Next buttons used throughout the installation wizard interface. """ back_button: Gtk.Button = Gtk.Button(label=get_text('Back')) """This button is used to go back to the previous page.""" cancel_button: Gtk.Button = Gtk.Button(label=get_text('Cancel')) """This button is used to quit and clean up.""" next_button: Gtk.Button = Gtk.Button(label=get_text('Next')) """This button is used to go to the next page.""" _box: Gtk.Box | None = None @classmethod def update_button_labels(cls) -> None: """Update button labels with current language translations.""" cls.back_button.set_label(get_text('Back')) cls.cancel_button.set_label(get_text('Cancel')) cls.next_button.set_label(get_text('Next')) @classmethod def hide_all(cls) -> None: """ This method hides all buttons. """ cls.back_button.hide() cls.cancel_button.hide() cls.next_button.hide() @classmethod def show_initial(cls) -> None: """ This method shows the initial buttons. Cancel and Next. """ cls.back_button.hide() cls.cancel_button.show() cls.next_button.show() @classmethod def show_back(cls) -> None: """ This method shows the back button. """ cls.back_button.show() @classmethod def hide_back(cls) -> None: """ This method hides the back button. """ cls.back_button.hide() @classmethod def box(cls) -> Gtk.Box: """ This method creates a box container of buttons aligned to the right. Returns: Box container with buttons aligned to the right for navigation. """ if cls._box is None: # Use Box instead of Grid for better right-alignment control cls._box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, homogeneous=False, spacing=5) cls._box.set_halign(Gtk.Align.END) # Align the entire box to the right cls.back_button.connect("clicked", Interface.back_page) cls._box.pack_start(cls.back_button, False, False, 0) cls.cancel_button.connect("clicked", Interface.delete) cls._box.pack_start(cls.cancel_button, False, False, 0) cls.next_button.connect("clicked", Interface.next_page) cls._box.pack_start(cls.next_button, False, False, 0) cls._box.show() return cls._box class Interface: """ Main interface controller for the installation wizard. Manages the GTK Notebook pages and navigation between different screens in the installation process including language, keyboard, network setup, installation type, and configuration screens. """ welcome = None keyboard = None network_setup = None try_install = None installation_type = None custom_partition = None full_zfs = None boot_manager = None page: Gtk.Notebook = Gtk.Notebook() nbButton: Gtk.Notebook | None = None @classmethod def get_interface(cls) -> Gtk.Box: interface_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) interface_box.show() interface_box.pack_start(cls.page, True, True, 0) cls.page.show() cls.page.set_show_tabs(False) cls.page.set_show_border(False) welcome_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) welcome_box.show() cls.welcome.initialize() get_types = cls.welcome.get_model() welcome_box.pack_start(get_types, True, True, 0) Window.set_title(get_text("Welcome to GhostBSD")) label = Gtk.Label(label=get_text("Welcome to GhostBSD")) cls.page.insert_page(welcome_box, label, 0) # Set what page to start at type of installation cls.page.set_current_page(0) cls.nbButton = Gtk.Notebook() interface_box.pack_end(cls.nbButton, False, False, 5) cls.nbButton.show() cls.nbButton.set_show_tabs(False) cls.nbButton.set_show_border(False) label = Gtk.Label(label=get_text("Button")) cls.nbButton.insert_page(Button.box(), label, 0) return interface_box @classmethod def delete(cls, _widget: Gtk.Widget, _event=None) -> None: """Close the main window.""" InstallationData.reset() Gtk.main_quit() @classmethod def next_page(cls, _widget: Gtk.Button) -> None: """Go to the next window.""" page = cls.page.get_current_page() if page == 0: # Check if the keyboard page already exists if cls.page.get_n_pages() <= 1: keyboard_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) keyboard_box.show() get_keyboard = cls.keyboard.get_model() keyboard_box.pack_start(get_keyboard, True, True, 0) label = Gtk.Label(label=get_text("Keyboard Setup")) cls.page.insert_page(keyboard_box, label, 1) cls.page.next_page() cls.page.show_all() Button.show_back() elif page == 1: # Check if the network setup page already exists if cls.page.get_n_pages() <= 2: network_setup_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) network_setup_box.show() get_network_setup = cls.network_setup.get_model() network_setup_box.pack_start(get_network_setup, True, True, 0) label = Gtk.Label(label=get_text("Network Setup")) cls.page.insert_page(network_setup_box, label, 2) cls.page.next_page() cls.page.show_all() elif page == 2: # Check if the try_install page already exists if cls.page.get_n_pages() <= 3: try_install_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) try_install_box.show() get_try_install = cls.try_install.get_model() try_install_box.pack_start(get_try_install, True, True, 0) label = Gtk.Label(label=get_text("Try Or Install GhostBSD")) cls.page.insert_page(try_install_box, label, 3) cls.page.next_page() cls.page.show_all() elif page == 3: if cls.try_install.get_what() == 'install': if cls.page.get_n_pages() <= 4: type_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) type_box.show() get_types = cls.installation_type.get_model() type_box.pack_start(get_types, True, True, 0) label = Gtk.Label(label=get_text("Installation Types")) cls.page.insert_page(type_box, label, 4) cls.page.next_page() cls.page.show_all() else: # Apply localization and keyboard layout for live session # Apply system localization if language was selected if InstallationData.language_code: localize_system(InstallationData.language_code) # Apply keyboard layout if selected if InstallationData.keyboard_layout_code: set_keyboard( InstallationData.keyboard_layout_code, InstallationData.keyboard_variant, InstallationData.keyboard_model_code ) # Continue to network setup for live session cls.next_setup_page() elif page == 4: Button.show_back() if InstallationData.filesystem_type == "custom": custom_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) custom_box.show() get_part = cls.custom_partition.get_model() custom_box.pack_start(get_part, True, True, 0) label = Gtk.Label(label=get_text("Custom Configuration")) cls.page.insert_page(custom_box, label, 5) cls.page.next_page() cls.page.show_all() Button.next_button.set_sensitive(False) elif InstallationData.filesystem_type == "zfs": zfs_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) zfs_box.show() get_zfs = cls.full_zfs.get_model() zfs_box.pack_start(get_zfs, True, True, 0) label = Gtk.Label(label=get_text("ZFS Configuration")) cls.page.insert_page(zfs_box, label, 5) cls.page.next_page() cls.page.show_all() Button.next_button.set_sensitive(False) elif page == 5: boot_manager_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) boot_manager_box.show() get_root = cls.boot_manager.get_model() boot_manager_box.pack_start(get_root, True, True, 0) label = Gtk.Label(label=get_text("Boot Option")) cls.page.insert_page(boot_manager_box, label, 6) Button.next_button.set_label(get_text("Install")) cls.page.next_page() cls.page.show_all() Button.next_button.set_sensitive(True) elif page == 6: installation_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) installation_box.show() install_window = InstallWindow() get_install = install_window.get_model() installation_box.pack_start(get_install, True, True, 0) label = Gtk.Label(label=get_text("Installation Progress")) cls.page.insert_page(installation_box, label, 7) cls.page.next_page() installation_progressbar = InstallProgress() progressbar = installation_progressbar.get_progressbar() box1 = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) box1.show() label = Gtk.Label(label=get_text("Progress Bar")) box1.pack_end(progressbar, False, False, 0) cls.nbButton.insert_page(box1, label, 1) cls.nbButton.next_page() current_page_widget = cls.page.get_nth_page(cls.page.get_current_page()) title_text = cls.page.get_tab_label_text(current_page_widget) Window.set_title(title_text) @classmethod def next_setup_page(cls) -> None: page = cls.page.get_current_page() if page == 0: Button.next_button.show() Button.next_button.set_sensitive(False) Window.set_title(get_text("Network Setup")) net_setup_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, homogeneous=False, spacing=0) net_setup_box.show() model = cls.network_setup.get_model() net_setup_box.pack_start(model, True, True, 0) label = Gtk.Label(label=get_text("Network Setup")) cls.page.insert_page(net_setup_box, label, 1) cls.page.next_page() cls.page.show_all() if page == 1: with open('/usr/home/ghostbsd/.xinitrc', 'w') as xinitrc: xinitrc.writelines('gsettings set org.mate.SettingsDaemon.plugins.housekeeping active true &\n') xinitrc.writelines('gsettings set org.mate.screensaver lock-enabled false &\n') xinitrc.writelines('exec ck-launch-session mate-session\n') Gtk.main_quit() @classmethod def back_page(cls, _widget: Gtk.Button) -> None: """Go back to the previous window.""" current_page = cls.page.get_current_page() if current_page == 1: Button.hide_back() elif current_page == 3: Button.next_button.set_label(get_text("Next")) cls.page.prev_page() new_page = cls.page.get_current_page() if current_page == 1 and new_page == 0: # Reset partition configuration data when going back InstallationData.destroy = {} InstallationData.delete = [] InstallationData.create = [] InstallationData.new_partition = [] InstallationData.scheme = "" InstallationData.disk = "" InstallationData.slice = "" InstallationData.zfs_config_data = [] InstallationData.ufs_config_data = [] # Clean up temporary directory if it exists DiskPartition.create_partition_database() current_page_widget = cls.page.get_nth_page(cls.page.get_current_page()) title_text = cls.page.get_tab_label_text(current_page_widget) Window.set_title(title_text) # Button.next_button.set_sensitive(True)