import pyblish.api


from openpype.hosts.fusion.api.lib import get_current_comp
from openpype.pipeline.publish import get_errored_instances_from_context


class SelectInvalidAction(pyblish.api.Action):
    """Select invalid nodes in Maya when plug-in failed.

    To retrieve the invalid nodes this assumes a static `get_invalid()`
    method is available on the plugin.

    """
    label = "Select invalid"
    on = "failed"  # This action is only available on a failed plug-in
    icon = "search"  # Icon from Awesome Icon

    def process(self, context, plugin):
        errored_instances = get_errored_instances_from_context(context)

        # Apply pyblish.logic to get the instances for the plug-in
        instances = pyblish.api.instances_by_plugin(errored_instances, plugin)

        # Get the invalid nodes for the plug-ins
        self.log.info("Finding invalid nodes..")
        invalid = list()
        for instance in instances:
            invalid_nodes = plugin.get_invalid(instance)
            if invalid_nodes:
                if isinstance(invalid_nodes, (list, tuple)):
                    invalid.extend(invalid_nodes)
                else:
                    self.log.warning("Plug-in returned to be invalid, "
                                     "but has no selectable nodes.")

        if not invalid:
            # Assume relevant comp is current comp and clear selection
            self.log.info("No invalid tools found.")
            comp = get_current_comp()
            flow = comp.CurrentFrame.FlowView
            flow.Select()  # No args equals clearing selection
            return

        # Assume a single comp
        first_tool = invalid[0]
        comp = first_tool.Comp()
        flow = comp.CurrentFrame.FlowView
        flow.Select()  # No args equals clearing selection
        names = set()
        for tool in invalid:
            flow.Select(tool, True)
            names.add(tool.Name)
        self.log.info("Selecting invalid tools: %s" % ", ".join(sorted(names)))