ProvidesCommand

Source

import sys

import demistomock as demisto  # noqa: F401
from CommonServerPython import *  # noqa: F401


def main():
    DEMISTO_ARGS = demisto.args()

    COMMAND = DEMISTO_ARGS["command"]
    ENABLED = None  # If 'enabled' param isn't present, return both enabled and disabled commands
    if "enabled" in DEMISTO_ARGS:
        enabled = DEMISTO_ARGS["enabled"]
        if enabled.lower() == "true":
            ENABLED = True
        elif enabled.lower() == "false":
            ENABLED = False

    integration_commands_args = {"uri": "/settings/integration-commands"}
    integration_commands_res = demisto.executeCommand("core-api-get", integration_commands_args)

    integration_search = None
    if ENABLED is not None:
        # if we only want to search enabled integrations, we must fetch that list from another API
        integration_search_args = {"uri": "/settings/integration/search", "body": {"size": 1000}}
        integration_search_res = demisto.executeCommand("core-api-post", integration_search_args)
        try:
            integration_search = integration_search_res[0]["Contents"]["response"]
        except KeyError:
            return_error("Did not receive expected response from Demisto API")
            sys.exit()

    if integration_search:
        integration_instances = integration_search["instances"]
        integration_instances_enabled = {}
        for integration in integration_instances:
            name = integration["brand"]
            if integration["enabled"] == "true":
                integration_instances_enabled[name] = True

    try:
        integration_commands = integration_commands_res[0]["Contents"]["response"]
    except KeyError:
        return_error("Did not receive expected response from Demisto API")
        sys.exit()

    integrations_that_implement = []

    for integration in integration_commands:
        integration_name = integration["name"]

        if "commands" in integration:
            for command in integration["commands"]:
                command_name = command["name"]
                if command_name == COMMAND:
                    if ENABLED is None:
                        integrations_that_implement.append(integration_name)
                    elif ENABLED is True and integration_name in integration_instances_enabled:
                        integrations_that_implement.append(integration_name)
                    elif ENABLED is False and integration_name not in integration_instances_enabled:
                        integrations_that_implement.append(integration_name)

    if len(integrations_that_implement) == 0:
        demisto.results("No matching commands found")
    else:
        demisto.results(",".join(integrations_that_implement))


if __name__ == "__builtin__" or __name__ == "builtins":
    main()

README