#region Reload and Import

# import core lib
import bpy
import typing, collections

# 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

#endregion

# 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_material
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

#region Menu

# ===== Menu Defines =====

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_material.BBP_OT_fix_all_material.bl_idname)

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):
        layout = self.layout
        OP_ADDS_bme.BBP_OT_add_bme_struct.draw_blc_menu(layout)
        
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):
        layout = self.layout

        layout.label(text="Sections", icon='MESH_CIRCLE', text_ctxt='BBP_MT_AddRailMenu/draw')
        layout.operator(OP_ADDS_rail.BBP_OT_add_rail_section.bl_idname)
        layout.operator(OP_ADDS_rail.BBP_OT_add_transition_section.bl_idname)

        layout.separator()
        layout.label(text="Straight Rails", icon='IPO_CONSTANT', text_ctxt='BBP_MT_AddRailMenu/draw')
        layout.operator(OP_ADDS_rail.BBP_OT_add_straight_rail.bl_idname)
        layout.operator(OP_ADDS_rail.BBP_OT_add_transition_rail.bl_idname)
        layout.operator(OP_ADDS_rail.BBP_OT_add_side_rail.bl_idname)

        layout.separator()
        layout.label(text="Curve Rails", icon='MOD_SCREW', text_ctxt='BBP_MT_AddRailMenu/draw')
        layout.operator(OP_ADDS_rail.BBP_OT_add_arc_rail.bl_idname)
        layout.operator(OP_ADDS_rail.BBP_OT_add_spiral_rail.bl_idname)
        layout.operator(OP_ADDS_rail.BBP_OT_add_side_spiral_rail.bl_idname)
        
class BBP_MT_AddComponentsMenu(bpy.types.Menu):
    """Add Ballance Component"""
    bl_idname = "BBP_MT_AddComponentsMenu"
    bl_label = "Components"
    bl_translation_context = 'BBP_MT_AddComponentsMenu'

    def draw(self, context):
        layout = self.layout

        layout.label(text="Basic Components", text_ctxt='BBP_MT_AddComponentsMenu/draw')
        OP_ADDS_component.BBP_OT_add_component.draw_blc_menu(layout)
        
        layout.separator()
        layout.label(text="Nong Components", text_ctxt='BBP_MT_AddComponentsMenu/draw')
        OP_ADDS_component.BBP_OT_add_nong_extra_point.draw_blc_menu(layout)
        OP_ADDS_component.BBP_OT_add_nong_ventilator.draw_blc_menu(layout)

        layout.separator()
        layout.label(text="Series Components", text_ctxt='BBP_MT_AddComponentsMenu/draw')
        OP_ADDS_component.BBP_OT_add_tilting_block_series.draw_blc_menu(layout)
        OP_ADDS_component.BBP_OT_add_swing_series.draw_blc_menu(layout)
        OP_ADDS_component.BBP_OT_add_ventilator_series.draw_blc_menu(layout)

        layout.separator()
        layout.label(text="Components Pair", text_ctxt='BBP_MT_AddComponentsMenu/draw')
        OP_ADDS_component.BBP_OT_add_sector_component_pair.draw_blc_menu(layout)

# ===== Menu Drawer =====

MenuDrawer_t = 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_AddComponentsMenu.bl_idname, icon='MESH_ICOSPHERE')

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

    # NOTE: because outline context may change operator context
    # 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'
    # thus, 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()

    # same reason 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

#region Register and Unregister.

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

class MenuEntry():
    mContainerMenu: bpy.types.Menu
    mIsAppend: bool
    mMenuDrawer: MenuDrawer_t
    def __init__(self, cont: bpy.types.Menu, is_append: bool, menu_func: MenuDrawer_t):
        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 double (for 2 menus)
     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_material.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_material.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