import gi

gi.require_version("Gtk", "4.0")
gi.require_version("Adw", "1")
from gi.repository import Adw, Gtk

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")

        # Main Layout
        main_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
        self.set_content(main_box)

        # Header Bar
        header = Adw.HeaderBar()
        main_box.append(header)

        # Content Stack
        self.stack = Adw.ViewStack()
        self.stack.set_vexpand(True)
        main_box.append(self.stack)

        # Navigation Bar (Bottom)
        nav_box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL)
        nav_box.add_css_class("toolbar")
        nav_box.set_margin_top(12)
        nav_box.set_margin_bottom(12)
        nav_box.set_margin_start(12)
        nav_box.set_margin_end(12)
        nav_box.set_spacing(12)
        main_box.append(nav_box)

        # Back Button
        self.back_button = Gtk.Button(label="Back")
        self.back_button.connect("clicked", self.on_back_clicked)
        self.back_button.set_sensitive(False)
        nav_box.append(self.back_button)

        # Spacer to push Next button to the right
        spacer = Gtk.Label()
        spacer.set_hexpand(True)
        nav_box.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)
        nav_box.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.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.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):
        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":
            # Pass selected disk to partitioning page
            selected_disk = self.storage_page.get_selected_disk()
            if selected_disk:
                self.partitioning_page.load_partitions(selected_disk)
            else:
                # Optionally handle no disk selected (alert user)
                pass

        if self.current_page_index < len(self.page_ids) - 1:
            self.current_page_index += 1
            self.stack.set_visible_child_name(self.page_ids[self.current_page_index])
            self.update_buttons()
        else:
            print("Install process triggered!")