import dearpygui.dearpygui as dpg
import logging
import json
import os

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from ..negstation import EditorManager


class LayoutManager:
    INI_PATH = "negstation_layout.ini"
    WIDGET_DATA_PATH = "negstation_widgets.json"

    def __init__(self, manager: "EditorManager", logger: logging.Logger):
        self.manager = manager
        self.logger = logger

    def save_layout(self):
        self.logger.info("Saving layout...")
        dpg.save_init_file(self.INI_PATH)
        widget_data = [
            {"widget_type": type(w).__name__, "config": w.get_config()}
            for w in self.manager.widgets
        ]
        with open(self.WIDGET_DATA_PATH, "w") as f:
            json.dump(widget_data, f, indent=4)
        self.logger.info("Layout saved successfully.")

    def load_layout(self):
        self.logger.info("Loading layout...")
        if not os.path.exists(self.WIDGET_DATA_PATH):
            return
        with open(self.WIDGET_DATA_PATH, "r") as f:
            widget_data = json.load(f)
        for data in widget_data:
            if data.get("widget_type") in self.manager.widget_classes:
                self.manager._add_widget(widget_type=data.get("widget_type"))

        if os.path.exists(self.INI_PATH):
            dpg.configure_app(init_file=self.INI_PATH)
            self.logger.info(f"Applied UI layout from {self.INI_PATH}")