Files
installer/iridium_installer/ui/window.py

151 lines
5.3 KiB
Python

import gi
gi.require_version("Gtk", "4.0")
gi.require_version("Adw", "1")
from gi.repository import Adw, Gtk
from .pages.install_mode import InstallModePage
from .pages.partitioning import PartitioningPage
from .pages.storage import StoragePage
from .pages.user import UserPage
from .pages.welcome import WelcomePage
class InstallerWindow(Adw.ApplicationWindow):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.set_default_size(900, 650)
self.set_title("Iridium Installer")
self.toolbar_view = Adw.ToolbarView()
self.set_content(self.toolbar_view)
# Header Bar (Top)
self.header_bar = Adw.HeaderBar()
self.toolbar_view.add_top_bar(self.header_bar)
# Content Stack
self.stack = Adw.ViewStack()
self.stack.set_vexpand(True)
self.toolbar_view.set_content(self.stack)
# Navigation Bar (Bottom)
self.bottom_bar = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL)
self.bottom_bar.add_css_class("toolbar")
self.bottom_bar.set_margin_top(12)
self.bottom_bar.set_margin_bottom(12)
self.bottom_bar.set_margin_start(12)
self.bottom_bar.set_margin_end(12)
self.bottom_bar.set_spacing(12)
self.toolbar_view.add_bottom_bar(self.bottom_bar)
# Back Button
self.back_button = Gtk.Button(label="Back")
self.back_button.connect("clicked", self.on_back_clicked)
self.back_button.set_sensitive(False)
self.bottom_bar.append(self.back_button)
# Spacer to push Next button to the right
spacer = Gtk.Label()
spacer.set_hexpand(True)
self.bottom_bar.append(spacer)
# Next Button
self.next_button = Gtk.Button(label="Next")
self.next_button.add_css_class("suggested-action")
self.next_button.connect("clicked", self.on_next_clicked)
self.bottom_bar.append(self.next_button)
# Page Management
self.page_ids = []
self.current_page_index = 0
# Initialize Pages
self.welcome_page = WelcomePage()
self.storage_page = StoragePage()
self.install_mode_page = InstallModePage()
self.partitioning_page = PartitioningPage()
self.user_page = UserPage()
# Add Pages
self.add_page(self.welcome_page, "welcome")
self.add_page(self.storage_page, "storage")
self.add_page(self.install_mode_page, "install_mode")
self.add_page(self.partitioning_page, "partitioning")
self.add_page(self.user_page, "user")
# Initialize view
if self.page_ids:
self.stack.set_visible_child_name(self.page_ids[0])
def add_page(self, widget, name):
self.stack.add_named(widget, name)
self.page_ids.append(name)
def update_buttons(self):
# Back button state
self.back_button.set_sensitive(self.current_page_index > 0)
# Next button label/state
if self.current_page_index == len(self.page_ids) - 1:
self.next_button.set_label("Install")
self.next_button.add_css_class("destructive-action")
self.next_button.remove_css_class("suggested-action")
else:
self.next_button.set_label("Next")
self.next_button.remove_css_class("destructive-action")
self.next_button.add_css_class("suggested-action")
def on_back_clicked(self, button):
current_page = self.page_ids[self.current_page_index]
prev_page = self.page_ids[self.current_page_index - 1]
if current_page == "user":
mode = self.install_mode_page.get_mode()
if mode == "automatic":
# Skip partitioning
self.current_page_index = self.page_ids.index("install_mode")
else:
self.current_page_index -= 1
else:
if self.current_page_index > 0:
self.current_page_index -= 1
self.stack.set_visible_child_name(self.page_ids[self.current_page_index])
self.update_buttons()
def on_next_clicked(self, button):
# Logic before transition
current_page_name = self.page_ids[self.current_page_index]
if current_page_name == "storage":
selected_disk = self.storage_page.get_selected_disk()
if selected_disk:
self.partitioning_page.load_partitions(selected_disk)
next_index = self.current_page_index + 1
if current_page_name == "install_mode":
mode = self.install_mode_page.get_mode()
if mode == "automatic":
# Skip partitioning page
next_index = self.page_ids.index("user")
else:
# Go to partitioning page
next_index = self.page_ids.index("partitioning")
if current_page_name == "user":
print("Install process triggered!")
print(f"Disk: {self.storage_page.get_selected_disk()}")
mode = self.install_mode_page.get_mode()
print(f"Mode: {mode}")
if mode == "manual":
print(f"Partitioning: {self.partitioning_page.get_config()}")
return
if next_index < len(self.page_ids):
self.current_page_index = next_index
self.stack.set_visible_child_name(self.page_ids[self.current_page_index])
self.update_buttons()