import logging

import maya.cmds as cmds
import maya.mel as mel

import scriptsmenu
from .vendor.Qt import QtCore, QtWidgets

log = logging.getLogger(__name__)


def register_repeat_last(action):
    """Register the action in repeatLast to ensure the RepeatLast hotkey works

    Args:
        action (action.Action): Action wigdet instance

    Returns:
        int: 0

    """
    command = action.process_command()
    command = command.replace("\n", "; ")
    # Register command to Maya (mel)
    cmds.repeatLast(addCommand='python("{}")'.format(command),
                    addCommandLabel=action.label)

    return 0


def to_shelf(action):
    """Copy clicked menu item to the currently active Maya shelf
    Args:
        action (action.Action): the action instance which is clicked

    Returns:
        int: 1

    """

    shelftoplevel = mel.eval("$gShelfTopLevel = $gShelfTopLevel;")
    current_active_shelf = cmds.tabLayout(shelftoplevel,
                                          query=True,
                                          selectTab=True)

    cmds.shelfButton(command=action.process_command(),
                     sourceType="python",
                     parent=current_active_shelf,
                     image=action.iconfile or "pythonFamily.png",
                     annotation=action.statusTip(),
                     imageOverlayLabel=action.label or "")

    return 1


def _maya_main_window():
    """Return Maya's main window"""
    for obj in QtWidgets.QApplication.topLevelWidgets():
        if obj.objectName() == 'MayaWindow':
            return obj
    raise RuntimeError('Could not find MayaWindow instance')


def _maya_main_menubar():
    """Retrieve the main menubar of the Maya window"""
    mayawindow = _maya_main_window()
    menubar = [i for i in mayawindow.children()
               if isinstance(i, QtWidgets.QMenuBar)]

    assert len(menubar) == 1, "Error, could not find menu bar!"

    return menubar[0]


def find_scripts_menu(title, parent):
    """
    Check if the menu exists with the given title in the parent

    Args:
        title (str): the title name of the scripts menu

        parent (QtWidgets.QMenuBar): the menubar to check

    Returns:
        QtWidgets.QMenu or None

    """

    menu = None
    search = [i for i in parent.children() if
              isinstance(i, scriptsmenu.ScriptsMenu)
              and i.title() == title]

    if search:
        assert len(search) < 2, ("Multiple instances of menu '{}' "
                                 "in menu bar".format(title))
        menu = search[0]

    return menu


def main(title="Scripts", parent=None, objectName=None):
    """Build the main scripts menu in Maya

    Args:
        title (str): name of the menu in the application

        parent (QtWidgets.QtMenuBar): the parent object for the menu

        objectName (str): custom objectName for scripts menu

    Returns:
        scriptsmenu.ScriptsMenu instance

    """

    mayamainbar = parent or _maya_main_menubar()
    try:
        # check menu already exists
        menu = find_scripts_menu(title, mayamainbar)
        if not menu:
            log.info("Attempting to build menu ...")
            object_name = objectName or title.lower()
            menu = scriptsmenu.ScriptsMenu(title=title,
                                           parent=mayamainbar,
                                           objectName=object_name)
    except Exception as e:
        log.error(e)
        return

    # Register control + shift callback to add to shelf (maya behavior)
    modifiers = QtCore.Qt.ControlModifier | QtCore.Qt.ShiftModifier
    menu.register_callback(int(modifiers), to_shelf)

    menu.register_callback(0, register_repeat_last)

    return menu