#region Import and Reload

# import core lib
import bpy
import typing, enum

# reload if needed
# TODO: finish reload feature if needed.
# (reload script raise too much exceptions so I usually restart blender to test my plugin.)
if "bpy" in locals():
    import importlib

# we must load icons manager first
# and register it
from . import UTIL_icons_manager
UTIL_icons_manager.register()

# then load other modules
from . import UTIL_translation
from . import PROP_preferences, PROP_ptrprop_resolver, PROP_virtools_material, PROP_virtools_texture, PROP_virtools_mesh, PROP_virtools_light, PROP_virtools_group
from . import PROP_ballance_element, PROP_bme_material, PROP_ballance_map_info
from . import OP_IMPORT_bmfile, OP_EXPORT_bmfile, OP_IMPORT_virtools, OP_EXPORT_virtools
from . import OP_UV_flatten_uv, OP_UV_rail_uv
from . import OP_MTL_fix_materials
from . import OP_ADDS_component, OP_ADDS_bme, OP_ADDS_rail
from . import OP_OBJECT_legacy_align, OP_OBJECT_virtools_group, OP_OBJECT_snoop_group_then_to_mesh, OP_OBJECT_naming_convention

#endregion

#region Menu and Sidebar Panel

#region Ballance Adder Menu and Panel

class DrawTarget(enum.IntEnum):
    BldMenu = enum.auto()
    BldPanel = enum.auto()

def reuse_create_layout(layout: bpy.types.UILayout, target: DrawTarget) -> bpy.types.UILayout:
    # If we are draw for Panel, we need use Grid to use space enough.
    match target:
        case DrawTarget.BldMenu:
            return layout
        case DrawTarget.BldPanel:
            return layout.grid_flow(even_columns=True, even_rows=True)

def reuse_draw_add_bme(layout: bpy.types.UILayout, target: DrawTarget):
    # Draw operators.
    OP_ADDS_bme.BBP_OT_add_bme_struct.draw_blc_menu(reuse_create_layout(layout, target))

def reuse_draw_add_rail(layout: bpy.types.UILayout, target: DrawTarget):
    layout.label(text="Sections", icon='MESH_CIRCLE', text_ctxt='BBP/__init__.reuse_draw_add_rail()')
    sublayout = reuse_create_layout(layout, target)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_rail_section.bl_idname)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_transition_section.bl_idname)

    layout.separator()
    layout.label(text="Straight Rails", icon='IPO_CONSTANT', text_ctxt='BBP/__init__.reuse_draw_add_rail()')
    sublayout = reuse_create_layout(layout, target)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_straight_rail.bl_idname)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_transition_rail.bl_idname)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_side_rail.bl_idname)

    layout.separator()
    layout.label(text="Curve Rails", icon='MOD_SCREW', text_ctxt='BBP/__init__.reuse_draw_add_rail()')
    sublayout = reuse_create_layout(layout, target)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_arc_rail.bl_idname)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_spiral_rail.bl_idname)
    sublayout.operator(OP_ADDS_rail.BBP_OT_add_side_spiral_rail.bl_idname)

def reuse_draw_add_component(layout: bpy.types.UILayout, target: DrawTarget):
    # We only use Grid for basic components
    layout.label(text="Basic Components", text_ctxt='BBP/__init__.reuse_draw_add_component()')
    OP_ADDS_component.BBP_OT_add_component.draw_blc_menu(reuse_create_layout(layout, target))
    
    layout.separator()
    layout.label(text="Nong Components", text_ctxt='BBP/__init__.reuse_draw_add_component()')
    sublayout = reuse_create_layout(layout, target)
    OP_ADDS_component.BBP_OT_add_nong_extra_point.draw_blc_menu(sublayout)
    OP_ADDS_component.BBP_OT_add_nong_ventilator.draw_blc_menu(sublayout)

    layout.separator()
    layout.label(text="Series Components", text_ctxt='BBP/__init__.reuse_draw_add_component()')
    sublayout = reuse_create_layout(layout, target)
    OP_ADDS_component.BBP_OT_add_tilting_block_series.draw_blc_menu(sublayout)
    OP_ADDS_component.BBP_OT_add_swing_series.draw_blc_menu(sublayout)
    OP_ADDS_component.BBP_OT_add_ventilator_series.draw_blc_menu(sublayout)

    layout.separator()
    layout.label(text="Components Pair", text_ctxt='BBP/__init__.reuse_draw_add_component()')
    sublayout = reuse_create_layout(layout, target)
    OP_ADDS_component.BBP_OT_add_sector_component_pair.draw_blc_menu(sublayout)

class BBP_MT_AddBmeMenu(bpy.types.Menu):
    """Add Ballance Floor"""
    bl_idname = "BBP_MT_AddBmeMenu"
    bl_label = "Floors"
    bl_translation_context = 'BBP_MT_AddBmeMenu'

    def draw(self, context):
        reuse_draw_add_bme(self.layout, DrawTarget.BldMenu)
        
class BBP_MT_AddRailMenu(bpy.types.Menu):
    """Add Ballance Rail"""
    bl_idname = "BBP_MT_AddRailMenu"
    bl_label = "Rails"
    bl_translation_context = 'BBP_MT_AddRailMenu'

    def draw(self, context):
        reuse_draw_add_rail(self.layout, DrawTarget.BldMenu)

class BBP_MT_AddComponentMenu(bpy.types.Menu):
    """Add Ballance Component"""
    bl_idname = "BBP_MT_AddComponentsMenu"
    bl_label = "Components"
    bl_translation_context = 'BBP_MT_AddComponentsMenu'

    def draw(self, context):
        reuse_draw_add_component(self.layout, DrawTarget.BldMenu)

class BBP_PT_SidebarAddBmePanel(bpy.types.Panel):
    """Add Ballance Floor"""
    bl_label = "Floors"
    bl_idname = "BBP_PT_SidebarAddBmePanel"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_context = "objectmode"
    bl_category = 'Ballance'
    bl_options = {'DEFAULT_CLOSED'}
    bl_translation_context = 'BBP_PT_SidebarAddBmePanel'

    def draw(self, context):
        reuse_draw_add_bme(self.layout, DrawTarget.BldPanel)

class BBP_PT_SidebarAddRailPanel(bpy.types.Panel):
    """Add Ballance Rail"""
    bl_label = "Rails"
    bl_idname = "BBP_PT_SidebarAddRailPanel"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_context = "objectmode"
    bl_category = 'Ballance'
    bl_options = {'DEFAULT_CLOSED'}
    bl_translation_context = 'BBP_PT_SidebarAddRailPanel'

    def draw(self, context):
        reuse_draw_add_rail(self.layout, DrawTarget.BldPanel)

class BBP_PT_SidebarAddComponentPanel(bpy.types.Panel):
    """Add Ballance Component"""
    bl_label = "Components"
    bl_idname = "BBP_PT_SidebarAddComponentPanel"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_context = "objectmode"
    bl_category = 'Ballance'
    bl_options = {'DEFAULT_CLOSED'}
    bl_translation_context = 'BBP_PT_SidebarAddComponentPanel'

    def draw(self, context):
        reuse_draw_add_component(self.layout, DrawTarget.BldPanel)

#endregion

#region Other Menu

class BBP_MT_View3DMenu(bpy.types.Menu):
    """Ballance 3D related operators"""
    bl_idname = "BBP_MT_View3DMenu"
    bl_label = "Ballance"
    bl_translation_context = 'BBP_MT_View3DMenu'

    def draw(self, context):
        layout = self.layout
        layout.label(text='UV', icon='UV', text_ctxt='BBP_MT_View3DMenu/draw')
        layout.operator(OP_UV_flatten_uv.BBP_OT_flatten_uv.bl_idname)
        layout.operator(OP_UV_rail_uv.BBP_OT_rail_uv.bl_idname)
        layout.separator()
        layout.label(text='Align', icon='SNAP_ON', text_ctxt='BBP_MT_View3DMenu/draw')
        layout.operator(OP_OBJECT_legacy_align.BBP_OT_legacy_align.bl_idname)
        layout.separator()
        layout.label(text='Select', icon='SELECT_SET', text_ctxt='BBP_MT_View3DMenu/draw')
        layout.operator(OP_OBJECT_virtools_group.BBP_OT_select_object_by_virtools_group.bl_idname)
        layout.separator()
        layout.label(text='Material', icon='MATERIAL', text_ctxt='BBP_MT_View3DMenu/draw')
        layout.operator(OP_MTL_fix_materials.BBP_OT_fix_all_materials.bl_idname)

#endregion

#region Menu Drawer

TFctMenuDrawer = typing.Callable[[typing.Any, typing.Any], None]

def menu_drawer_import(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    # layout.operator(
    #     OP_IMPORT_bmfile.BBP_OT_import_bmfile.bl_idname,
    #     text="Ballance Map (.bmx)",
    #     text_ctxt='BBP/__init__.menu_drawer_import()'
    # )
    layout.operator(
        OP_IMPORT_virtools.BBP_OT_import_virtools.bl_idname,
        text="Virtools File (.nmo/.cmo/.vmo) (experimental)",
        text_ctxt='BBP/__init__.menu_drawer_import()'
    )

def menu_drawer_export(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    # layout.operator(
    #     OP_EXPORT_bmfile.BBP_OT_export_bmfile.bl_idname,
    #     text="Ballance Map (.bmx)",
    #     text_ctxt='BBP/__init__.menu_drawer_export()'
    # )
    layout.operator(
        OP_EXPORT_virtools.BBP_OT_export_virtools.bl_idname,
        text="Virtools File (.nmo/.cmo/.vmo) (experimental)",
        text_ctxt='BBP/__init__.menu_drawer_export()'
    )

def menu_drawer_view3d(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    layout.menu(BBP_MT_View3DMenu.bl_idname)

def menu_drawer_add(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    layout.separator()
    layout.label(text="Ballance", text_ctxt='BBP/__init__.menu_drawer_add()')
    layout.menu(BBP_MT_AddBmeMenu.bl_idname, icon='MESH_CUBE')
    layout.menu(BBP_MT_AddRailMenu.bl_idname, icon='MESH_CIRCLE')
    layout.menu(BBP_MT_AddComponentMenu.bl_idname, icon='MESH_ICOSPHERE')

def menu_drawer_grouping(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    layout.separator()

    # YYC MARK:
    # Because outline context change operator context into EXEC_*,
    # so it will cause no popup window when click operator in outline.
    # Thus we create a sub layout and set its operator context as 'INVOKE_DEFAULT',
    # so that all operators can pop up normally.
    col = layout.column()
    col.operator_context = 'INVOKE_DEFAULT'

    col.label(text="Virtools Group", text_ctxt='BBP/__init__.menu_drawer_grouping()')
    col.operator(
        OP_OBJECT_virtools_group.BBP_OT_add_objects_virtools_group.bl_idname, icon='ADD', text="Group into...",
        text_ctxt='BBP/__init__.menu_drawer_grouping()')
    col.operator(
        OP_OBJECT_virtools_group.BBP_OT_rm_objects_virtools_group.bl_idname, icon='REMOVE', text="Ungroup from...",
        text_ctxt='BBP/__init__.menu_drawer_grouping()')
    col.operator(
        OP_OBJECT_virtools_group.BBP_OT_clear_objects_virtools_group.bl_idname, icon='TRASH', text="Clear All Groups",
        text_ctxt='BBP/__init__.menu_drawer_grouping()')

def menu_drawer_snoop_then_conv(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    layout.separator()
    layout.label(text="Ballance", text_ctxt='BBP/__init__.menu_drawer_snoop_then_conv()')
    layout.operator(OP_OBJECT_snoop_group_then_to_mesh.BBP_OT_snoop_group_then_to_mesh.bl_idname, icon='OUTLINER_OB_MESH')

def menu_drawer_naming_convention(self, context) -> None:
    layout: bpy.types.UILayout = self.layout
    layout.separator()

    # YYC MARK:
    # Same reason for changing operator context introduced in `menu_drawer_grouping()`
    col = layout.column()
    col.operator_context = 'INVOKE_DEFAULT'

    col.label(text="Ballance", text_ctxt='BBP/__init__.menu_drawer_naming_convention()')
    col.operator(OP_OBJECT_naming_convention.BBP_OT_regulate_objects_name.bl_idname, icon='GREASEPENCIL')
    col.operator(OP_OBJECT_naming_convention.BBP_OT_auto_grouping.bl_idname, icon='GROUP')
    col.operator(OP_OBJECT_naming_convention.BBP_OT_convert_to_imengyu.bl_idname, icon='ARROW_LEFTRIGHT')

#endregion

#endregion

#region Register and Unregister.

g_BldClasses: tuple[typing.Any, ...] = (
    BBP_MT_View3DMenu,
    BBP_MT_AddBmeMenu,
    BBP_MT_AddRailMenu,
    BBP_MT_AddComponentMenu,

    BBP_PT_SidebarAddBmePanel,
    BBP_PT_SidebarAddRailPanel,
    BBP_PT_SidebarAddComponentPanel,
)

class MenuEntry():
    mContainerMenu: bpy.types.Menu
    mIsAppend: bool
    mMenuDrawer: TFctMenuDrawer
    def __init__(self, cont: bpy.types.Menu, is_append: bool, menu_func: TFctMenuDrawer):
        self.mContainerMenu = cont
        self.mIsAppend = is_append
        self.mMenuDrawer = menu_func

g_BldMenus: tuple[MenuEntry, ...] = (
    MenuEntry(bpy.types.VIEW3D_MT_editor_menus, False, menu_drawer_view3d),
    MenuEntry(bpy.types.TOPBAR_MT_file_import, True, menu_drawer_import),
    MenuEntry(bpy.types.TOPBAR_MT_file_export, True, menu_drawer_export),
    MenuEntry(bpy.types.VIEW3D_MT_add, True, menu_drawer_add),

    MenuEntry(bpy.types.VIEW3D_MT_object_context_menu, True, menu_drawer_snoop_then_conv),

    # Register this twice (for 2 menus respectively)
    MenuEntry(bpy.types.VIEW3D_MT_object_context_menu, True, menu_drawer_grouping),
    MenuEntry(bpy.types.OUTLINER_MT_object, True, menu_drawer_grouping),

    MenuEntry(bpy.types.OUTLINER_MT_collection, True, menu_drawer_naming_convention),
)

def register() -> None:
    # register module
    UTIL_translation.register()
    PROP_preferences.register()

    PROP_ptrprop_resolver.register()

    PROP_virtools_material.register()
    PROP_virtools_texture.register()
    PROP_virtools_mesh.register()
    PROP_virtools_light.register()
    PROP_virtools_group.register()
    PROP_ballance_element.register()
    PROP_bme_material.register()
    PROP_ballance_map_info.register()

    OP_IMPORT_bmfile.register()
    OP_EXPORT_bmfile.register()
    OP_IMPORT_virtools.register()
    OP_EXPORT_virtools.register()

    OP_UV_rail_uv.register()
    OP_UV_flatten_uv.register()

    OP_MTL_fix_materials.register()

    OP_ADDS_component.register()
    OP_ADDS_bme.register()
    OP_ADDS_rail.register()

    OP_OBJECT_legacy_align.register()
    OP_OBJECT_virtools_group.register()
    OP_OBJECT_snoop_group_then_to_mesh.register()
    OP_OBJECT_naming_convention.register()

    # register other classes
    for cls in g_BldClasses:
        bpy.utils.register_class(cls)

    # add menu drawer
    for entry in g_BldMenus:
        if entry.mIsAppend:
            entry.mContainerMenu.append(entry.mMenuDrawer)
        else:
            entry.mContainerMenu.prepend(entry.mMenuDrawer)

def unregister() -> None:
    # remove menu drawer
    for entry in g_BldMenus:
        entry.mContainerMenu.remove(entry.mMenuDrawer)

    # unregister other classes
    for cls in g_BldClasses:
        bpy.utils.unregister_class(cls)

    # unregister modules
    OP_OBJECT_naming_convention.unregister()
    OP_OBJECT_snoop_group_then_to_mesh.unregister()
    OP_OBJECT_virtools_group.unregister()
    OP_OBJECT_legacy_align.unregister()

    OP_ADDS_rail.unregister()
    OP_ADDS_bme.unregister()
    OP_ADDS_component.unregister()

    OP_MTL_fix_materials.unregister()

    OP_UV_flatten_uv.unregister()
    OP_UV_rail_uv.unregister()

    OP_EXPORT_virtools.unregister()
    OP_IMPORT_virtools.unregister()
    OP_EXPORT_bmfile.unregister()
    OP_IMPORT_bmfile.unregister()

    PROP_ballance_map_info.unregister()
    PROP_bme_material.unregister()
    PROP_ballance_element.unregister()
    PROP_virtools_group.unregister()
    PROP_virtools_light.unregister()
    PROP_virtools_mesh.unregister()
    PROP_virtools_texture.unregister()
    PROP_virtools_material.unregister()
    
    PROP_ptrprop_resolver.unregister()

    PROP_preferences.unregister()
    UTIL_translation.unregister()

if __name__ == "__main__":
    register()

#endregion