GetCampaignIndicatorsByIncidentId

Source

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


def get_incidents_ids_from_context() -> list:
    """
    Gets the campaign incident ids.

    Returns:
        List of incident ids.
    """
    relevant_incidents: list | None = demisto.get(demisto.context(), "EmailCampaign.incidents")
    ids = [item.get("id") for item in relevant_incidents] if relevant_incidents else []

    return ids


def get_indicators_from_incidents(incident_ids: list):
    """
    Gets the campaign indicators by the incident ids.

    Args:
        incident_ids: List of the campaign incident ids.

    Returns:
        List of the campaign indicators.
    """
    indicators_query = f"""investigationIDs:({' '.join(f'"{id_}"' for id_ in incident_ids)})"""
    fields = ["id", "indicator_type", "investigationIDs", "investigationsCount", "score", "value"]
    search_indicators = IndicatorsSearcher(query=indicators_query, limit=150, size=500, filter_fields=",".join(fields))
    indicators: list[dict] = []
    for ioc_res in search_indicators:
        indicators.extend(ioc_res.get("iocs") or [])
    return indicators


def format_results(indicators: list, incident_ids: list) -> str:
    """
    Format the indicators result to a readable markdown table.

    Args:
        indicators: The campaign indicators.
        incident_ids: The campaign incident ids.

    Returns:
        A readable markdown table of the campaign indicators.
    """
    indicators_df = pd.DataFrame(data=indicators)

    if len(indicators_df) == 0:
        return "No mutual indicators were found."

    indicators_df = indicators_df[indicators_df["relatedIncCount"] < 150]
    indicators_df["Involved Incidents Count"] = indicators_df["investigationIDs"].apply(
        lambda x: sum(id_ in x for id_ in incident_ids)
    )
    indicators_df = indicators_df[indicators_df["Involved Incidents Count"] > 1]

    if len(indicators_df) == 0:
        return "No mutual indicators were found."

    associate_to_current_incident(indicators)
    indicators_df["Id"] = indicators_df["id"].apply(lambda x: f"[{x}](#/indicator/{x})")
    indicators_df = indicators_df.sort_values(["score", "Involved Incidents Count"], ascending=False)
    indicators_df["Reputation"] = indicators_df["score"].apply(scoreToReputation)
    indicators_df = indicators_df.rename({"value": "Value", "indicator_type": "Type"}, axis=1)
    indicators_headers = ["Id", "Value", "Type", "Reputation", "Involved Incidents Count"]

    return tableToMarkdown("", indicators_df.to_dict(orient="records"), headers=indicators_headers)


def associate_to_current_incident(indicators: list[dict[str, str]]):
    incident_id = demisto.incident()["id"]
    execute_command(
        "associateIndicatorsToIncident", {"incidentId": incident_id, "indicatorsValues": [x["value"] for x in indicators]}
    )


def main():  # pragma: no cover
    """This script should run from a campaign incident, and expect to have incidents
    to get indicators from."""
    try:
        if incident_ids := get_incidents_ids_from_context():
            indicators = get_indicators_from_incidents(incident_ids)
            formatted_results = format_results(indicators, incident_ids)
        else:
            formatted_results = "No mutual indicators were found."

        execute_command("setIncident", {"campaignmutualindicators": formatted_results})

    except Exception as ex:
        return_error(f"Failed to execute BaseScript. Error: {ex!s}")


if __name__ in ("__main__", "__builtin__", "builtins"):
    main()

README