import bpy

from .common.image import is_image_empty
from .operators.upscale import upscale_output_size


class AnyImageImageMenu(bpy.types.Menu):
    bl_idname = "ANYIMAGE_MT_image_context"
    bl_label = "AnyImage"

    def draw(self, context):
        from .properties import ai_setup_label, ai_status

        layout = self.layout
        layout.operator_context = "INVOKE_DEFAULT"
        layout.operator(
            "anyimage.convert_to_plane",
            text="Convert to Plane",
            icon="MESH_PLANE",
        )
        layout.operator(
            "anyimage.convert_to_depth_plane",
            text="Convert to Depth Plane",
            icon="MOD_DISPLACE",
        )
        layout.separator()
        layout.operator(
            "anyimage.remove_image_background",
            text="Remove Background",
            icon="IMAGE_ALPHA",
        )
        layout.operator(
            "anyimage.upscale_image",
            text=upscale_menu_label(context),
            icon="FULLSCREEN_ENTER",
        )
        layout.separator()
        status = ai_status()
        if status["ready"]:
            layout.operator(
                "anyimage.open_ai_environment_settings",
                text="AI Environment Settings",
                icon="PREFERENCES",
            )
        else:
            layout.operator(
                "anyimage.setup_ai_environment",
                text=ai_setup_label(status),
                icon="IMPORT",
            )


def upscale_menu_label(context):
    source_object = getattr(context, "object", None)
    if not is_image_empty(source_object):
        return "Upscale"
    output_size = upscale_output_size(source_object.data)
    if output_size is None:
        return "Upscale"
    width, height = output_size
    return f"Upscale ({width} × {height})"


def draw_image_context_menu(self, context):
    if not is_image_empty(context.object):
        return
    self.layout.menu(AnyImageImageMenu.bl_idname, text="AnyImage", icon="PLUGIN")
    self.layout.separator()