"""MEL/Python-visible commands: ``openposeCreateCharacter`` (undoable node
creation) and ``openposeShowMappingEditor`` (launches the mapping UI).
"""

import maya.api.OpenMaya as om

from . import mapping_node


class CreateCharacterCommand(om.MPxCommand):

    kCmdName = "openposeCreateCharacter"

    def __init__(self):
        om.MPxCommand.__init__(self)
        self._modifiers = []
        self._created_node_name = None

    @staticmethod
    def creator():
        return CreateCharacterCommand()

    @staticmethod
    def createSyntax():
        syntax = om.MSyntax()
        syntax.addFlag("-n", "-name", om.MSyntax.kString)
        return syntax

    def doIt(self, args):
        parser = om.MArgParser(self.syntax(), args)
        want_name = parser.flagArgumentString("-n", 0) if parser.isFlagSet("-n") else None

        self._modifiers = []

        # For a shape type, MDagModifier.createNode() returns the
        # auto-created *parent transform*, not the shape -- the shape itself
        # is its only child, and only exists once doIt() has run.
        create_modifier = om.MDagModifier()
        transform_obj = create_modifier.createNode(mapping_node.NODE_NAME)
        create_modifier.doIt()
        self._modifiers.append(create_modifier)

        shape_obj = om.MFnDagNode(transform_obj).child(0)

        if want_name:
            rename_modifier = om.MDagModifier()
            rename_modifier.renameNode(shape_obj, want_name)
            rename_modifier.doIt()
            self._modifiers.append(rename_modifier)

        self._created_node_name = om.MFnDagNode(shape_obj).name()
        self.setResult(self._created_node_name)

    def redoIt(self):
        for modifier in self._modifiers:
            modifier.doIt()

    def undoIt(self):
        for modifier in reversed(self._modifiers):
            modifier.undoIt()

    def isUndoable(self):
        return True


class ShowMappingEditorCommand(om.MPxCommand):

    kCmdName = "openposeShowMappingEditor"

    @staticmethod
    def creator():
        return ShowMappingEditorCommand()

    def doIt(self, args):
        from .ui import mapping_editor
        mapping_editor.show()

    def isUndoable(self):
        return False


def register(fn_plugin):
    fn_plugin.registerCommand(
        CreateCharacterCommand.kCmdName, CreateCharacterCommand.creator, CreateCharacterCommand.createSyntax
    )
    fn_plugin.registerCommand(ShowMappingEditorCommand.kCmdName, ShowMappingEditorCommand.creator)


def deregister(fn_plugin):
    fn_plugin.deregisterCommand(ShowMappingEditorCommand.kCmdName)
    fn_plugin.deregisterCommand(CreateCharacterCommand.kCmdName)