import bpy

from pathlib import Path


class ExportForFange(bpy.types.Operator):
    """Export a building for fange"""
    bl_idname = 'graou.building_export'
    bl_label = 'Easily export a building asset'

    def __init__(self):
        self.coll_layout = bpy.data.collections.get('Placeholder')
        self.output = Path("W:\\", "Graou Studio", "Game Projects", "Fange Prototype", "Assets", "Graph", "Buildings")
        self.fbx_preset = 'sm-unity-building'

    @classmethod
    def poll(cls, context):
        col = bpy.data.collections.get('Placeholder')

        if col is not None:
            return True

    def execute(self, context):
        bpy.ops.object.select_all(action='DESELECT')
        print(f'[Pipeline] Start to export the building props.')

        if len(self.coll_layout.collection_children):
            for coll in self.coll_layout.children:
                print(f'[Pipeline] Update "{coll.name}" mesh')
                child = bpy.data.collections.get(coll.name)

                print(f'[Pipeline] Select all mesh inside the collection "{child.name}"')
                for ob in child.all_objects:
                    print(f'\tLook "{ob.name}", his type are "{type(ob.data)}"')
                    if isinstance(ob.data, bpy.types.Mesh) or isinstance(ob.data, bpy.types.Empty):
                        ob.select_set(True)

                # TODO The plugin export only the garden building
                abs_export = self.output.joinpath("Garden", "Meshes")
                if not abs_export.exists():
                    abs_export.mkdir()

                # TODO Use a preset system
                bpy.ops.export_scene.fbx(filepath=abs_export.joinpath(f"SM_{coll.name}.fbx").as_posix(),
                                         use_selection=True,
                                         use_visible=False,
                                         use_active_collection=False,
                                         global_scale=1.0,
                                         apply_unit_scale=True,
                                         apply_scale_options='FBX_SCALE_NONE',
                                         use_space_transform=True,
                                         bake_space_transform=True,
                                         object_types={'MESH', 'EMPTY'},
                                         use_mesh_modifiers=True,
                                         use_mesh_modifiers_render=True,
                                         mesh_smooth_type='OFF',
                                         colors_type='SRGB',
                                         prioritize_active_color=False,
                                         use_subsurf=False,
                                         use_mesh_edges=False,
                                         use_tspace=False,
                                         use_triangles=False,
                                         use_custom_props=False,
                                         add_leaf_bones=True,
                                         primary_bone_axis='Y',
                                         secondary_bone_axis='X',
                                         use_armature_deform_only=False,
                                         armature_nodetype='NULL',
                                         bake_anim=False,
                                         bake_anim_use_all_bones=True,
                                         bake_anim_use_nla_strips=True,
                                         bake_anim_use_all_actions=True,
                                         bake_anim_force_startend_keying=True,
                                         bake_anim_step=1.0,
                                         bake_anim_simplify_factor=1.0,
                                         path_mode='AUTO',
                                         embed_textures=False,
                                         batch_mode='OFF',
                                         use_batch_own_dir=True,
                                         axis_forward='X',
                                         axis_up='Y'
                                         )

                print(f'[Pipeline] Export here "{abs_export}"')
                bpy.ops.object.select_all(action='DESELECT')

        # for coll in coll_layout.children:
        #    print(f'[Pipeline] Check {coll}. Item type {type(coll)}')
        return {'FINISHED'}