import bpy

from .models import MODELS, selected_model_key
from .operators import active_job
from .runtime import model_ready, runtime_ready


class RuntimeModelsPanel(bpy.types.Panel):
    bl_label = "Setup"
    bl_idname = "SNAPFORM_PT_runtime_models"
    bl_space_type = "VIEW_3D"
    bl_region_type = "UI"
    bl_category = "SnapForm"

    def draw(self, _context):
        layout = self.layout
        running = active_job() is not None

        if not runtime_ready():
            layout.label(text="Runtime is not installed", icon="ERROR")
            if not running:
                operator = layout.operator(
                    "snapform.run_job",
                    text="Install Runtime",
                    icon="IMPORT",
                )
                operator.job_type = "INSTALL_RUNTIME"

        for model in MODELS.values():
            self._draw_model(layout, model, running)

    def _draw_model(self, layout, model, running):
        row = layout.row(align=True)
        row.label(text=model.label)
        if model_ready(model.key):
            row.label(text="Ready", icon="CHECKMARK")
        elif runtime_ready() and not running:
            operator = row.operator(
                "snapform.run_job",
                text="Download",
                icon="IMPORT",
            )
            operator.job_type = "DOWNLOAD_MODEL"
            operator.model_key = model.key
        else:
            row.label(text="Not downloaded", icon="ERROR")


class DebugPanel(bpy.types.Panel):
    bl_label = "Debug"
    bl_idname = "SNAPFORM_PT_debug"
    bl_space_type = "VIEW_3D"
    bl_region_type = "UI"
    bl_category = "SnapForm"
    bl_options = {"DEFAULT_CLOSED"}

    def draw(self, context):
        layout = self.layout
        settings = context.scene.snapform_settings

        layout.prop(settings, "export_format")
        model_key = selected_model_key(settings)
        model_property = (
            "gs_model" if settings.export_format == "gs_ply" else "model"
        )
        model_row = layout.row(align=True)
        model_row.prop(settings, model_property)
        if model_ready(model_key):
            model_row.label(text="", icon="CHECKMARK")
        else:
            model_row.label(text="", icon="ERROR")

        layout.prop(settings, "input_path")
        input_path = bpy.path.abspath(settings.input_path)
        if input_path.lower().endswith(
            (
                ".mp4",
                ".mov",
                ".avi",
                ".mkv",
                ".webm",
                ".m4v",
                ".mts",
                ".mpg",
                ".mpeg",
            )
        ):
            layout.prop(settings, "video_frames")
        layout.prop(settings, "device")
        layout.prop(settings, "process_res")
        layout.prop(settings, "process_res_method")
        layout.prop(settings, "use_ray_pose")
        layout.prop(settings, "ref_view_strategy")

        if settings.export_format == "glb":
            glb_box = layout.box()
            glb_box.label(text="GLB Settings")
            glb_box.prop(settings, "conf_thresh_percentile")
            glb_box.prop(settings, "num_max_points")
            glb_box.prop(settings, "show_cameras")
        elif settings.export_format == "gs_ply":
            ply_box = layout.box()
            ply_box.label(text="PLY (3DGS) Settings")
            ply_box.prop(settings, "gs_views_interval")

        if active_job() is None:
            operator = layout.operator(
                "snapform.run_job",
                text="Export",
                icon="EXPORT",
            )
            operator.job_type = "RECONSTRUCT"