# === SpyCloudEnterpriseProtectionFeed Integration ===

from datetime import datetime, timedelta, UTC
from json import dumps
from typing import Any
from requests import Response
from urllib3 import disable_warnings
from CommonServerPython import *

disable_warnings()  # Disable SSL warnings

INTEGRATION_CONTEXT_NAME = "SpyCloud"
INVALID_CREDENTIALS_ERROR_MSG = (
    "Authorization Error: The provided API Key for SpyCloud is invalid. Please provide a valid API Key."
)
MAX_RETRIES = 5
BACK_OFF_TIME = 0.1
DEFAULT_FETCH_LIMIT = 200

# Error and endpoint constants
PAGE_NUMBER_ERROR_MSG = "Invalid Input Error: page number should be greater than zero."
PAGE_SIZE_ERROR_MSG = "Invalid Input Error: page size should be greater than zero."
LIMIT_EXCEED = "LimitExceededException"
TOO_MANY_REQUESTS = "TooManyRequestsException"
INVALID_IP = "Invalid IP"
INVALID_API_KEY = "Invalid API key"
X_AMAZON_ERROR_TYPE = "x-amzn-ErrorType"
SPYCLOUD_ERROR = "SpyCloud-Error"
INVALID_IP_MSG = "Kindly contact SpyCloud support to whitelist your IP Address."
WRONG_API_URL = "Verify that the API URL parameter is correct and that you have access to the server from your host"
MONTHLY_QUOTA_EXCEED_MSG = "You have exceeded your monthly quota. Kindly contact SpyCloud support."

# Relative endpoints — base_url should include /enterprise-v2/
WATCHLIST_ENDPOINT = "breach/data/watchlist"
DOMAIN_ENDPOINT = "breach/data/domains/"
DATE_TIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ"
DEFAULT_DATE = "-1days"

INCIDENT_TYPE = {
    2: "SpyCloud Informative Data",
    5: "SpyCloud Informative Data",
    20: "SpyCloud Breach Data",
    25: "SpyCloud Malware Data",
    30: "SpyCloud Access Data",
}
INCIDENT_NAME = {
    2: "SpyCloud Informative Alert on",
    5: "SpyCloud Informative Alert on",
    20: "SpyCloud Breach Alert on",
    25: "SpyCloud Malware Alert on",
    30: "SpyCloud Access Alert on",
}
SEVERITY_VALUE = {
    2: IncidentSeverity.INFO,
    5: IncidentSeverity.INFO,
    20: IncidentSeverity.HIGH,
    25: IncidentSeverity.CRITICAL,
    30: IncidentSeverity.CRITICAL,
}


# =========================
# Helper: Safe logging
# =========================
def safe_log(msg: str) -> str:
    """Convert any string to ASCII, removing characters that break Latin-1 logs."""
    return str(msg).encode("ascii", errors="ignore").decode()


class Client(BaseClient):
    def __init__(self, base_url: str, apikey: str, verify=None, proxy=None):
        headers = {"Accept": "application/json", "X-API-Key": apikey, "User-Agent": "XSOAR-ENT/1.0.11"}
        super().__init__(
            base_url=base_url,
            verify=verify,
            proxy=proxy,
            headers=headers,
        )

    def query_spy_cloud_api(self, end_point: str, params: dict[Any, Any] = None, is_retry: bool = False) -> dict:
        if params is None:
            params = {}

        url_path = urljoin(self._base_url, end_point) if not is_retry else end_point

        # sanitize logs
        log_params = {k: v for k, v in params.items() if k.lower() != "apikey"}
        demisto.info(safe_log(f"[SpyCloud] Querying endpoint: {url_path} with params: {log_params}"))

        retries = None
        status_list_to_retry = None
        backoff_factor = None

        if is_retry:
            retries = MAX_RETRIES
            status_list_to_retry = {429, 504}
            backoff_factor = BACK_OFF_TIME

        response = self._http_request(
            method="GET",
            full_url=url_path,
            params=params,
            headers=self._headers,
            retries=retries,
            status_list_to_retry=status_list_to_retry,
            backoff_factor=backoff_factor,
            error_handler=self.spy_cloud_error_handler,
        )
        return response

    def spy_cloud_error_handler(self, response: Response):
        try:
            err_msg = response.json().get("message") or response.json().get("errorMessage")
        except Exception:
            err_msg = response.text

        safe_msg = safe_log(err_msg)
        response_headers = response.headers or {}  # type: ignore

        # Retry on throttling via Amazon error type
        if TOO_MANY_REQUESTS in response_headers.get(X_AMAZON_ERROR_TYPE, ""):
            self.query_spy_cloud_api(response.url, is_retry=True)
            return
        elif LIMIT_EXCEED in response_headers.get(X_AMAZON_ERROR_TYPE, ""):
            raise DemistoException(MONTHLY_QUOTA_EXCEED_MSG, res=response)

        # Allow _http_request retry mechanism to handle 429
        if response.status_code == 429:
            return

        elif response.status_code == 403:
            raise DemistoException(f"Authorization or IP error. {safe_msg}", res=response)

        else:
            raise DemistoException(f"SpyCloud API error: {safe_msg}", res=response)

    @staticmethod
    def set_last_run():
        current_date = datetime.utcnow()
        # Add 1 second to avoid duplicate fetches
        next_run = current_date + timedelta(seconds=1)
        demisto.setLastRun({"lastRun": next_run.strftime(DATE_TIME_FORMAT)})

    @staticmethod
    def get_last_run() -> str:
        return demisto.getLastRun().get("lastRun")


def create_spycloud_args(args: dict, client: Client) -> dict:
    now = datetime.now(UTC)
    last_run = client.get_last_run()

    if last_run:
        # Use last run timestamp for since
        since = last_run
        # until defaults to now timestamp
        until = now.strftime(DATE_TIME_FORMAT)
        since_modification_date = (now - timedelta(days=1)).strftime("%Y-%m-%d")
        until_modification_date = (now - timedelta(days=1)).strftime("%Y-%m-%d")
    else:
        # first run
        since_dt = arg_to_datetime(args.get("first_fetch") or DEFAULT_DATE)
        since = since_dt.strftime("%Y-%m-%d") if since_dt else now.strftime("%Y-%m-%d")

        if args.get("until"):
            # customer-provided until: only date
            until_dt = arg_to_datetime(args.get("until")) or now
            until = until_dt.strftime("%Y-%m-%d")
        else:
            # default until: use full timestamp
            until = now.strftime(DATE_TIME_FORMAT)

        since_mod_dt = arg_to_datetime(args.get("since_modification_date")) if args.get("since_modification_date") else since_dt
        since_modification_date = since_mod_dt.strftime("%Y-%m-%d") if since_mod_dt else since
        until_modification_date = until

    severity_list = argToList(args.get("severity", []))
    supported_severities = {"2", "5", "25", "20", "30"}
    if any(sev not in supported_severities for sev in severity_list):
        raise DemistoException("Invalid input error: supported values for severity are: 2, 5, 20, 25, 30")

    return {
        "type": args.get("type", ""),
        "severity": ",".join(severity_list),
        "source_id": args.get("source_id", ""),
        "query": args.get("query", ""),
        "domain_search": args.get("domain_search", ""),
        "watchlist_type": args.get("watchlist_type", ""),
        "salt": args.get("salt"),
        "since": since,
        "until": until,
        "since_modification_date": since_modification_date,
        "until_modification_date": until_modification_date,
        "fetch_limit": args.get("fetch_limit", DEFAULT_FETCH_LIMIT),
    }


def fetch_domain_or_watchlist_data(client: Client, args: dict, base_args: dict) -> list:
    domain_search = (args.get("domain_search") or "").strip()
    results = []

    if domain_search:
        domains = [d.strip() for d in domain_search.split(",") if d.strip()]
        for domain in domains:
            endpoint = f"{DOMAIN_ENDPOINT}{domain}"
            domain_params = base_args.copy()

            # Do NOT override type
            if "type" in domain_params and not domain_params["type"]:
                del domain_params["type"]

            cursor = " "
            while cursor:
                try:
                    response = client.query_spy_cloud_api(endpoint, {**domain_params, "cursor": cursor})
                    domain_results = response.get("results", [])
                    results.extend(domain_results)
                    cursor = response.get("cursor", "")
                except Exception as e:
                    demisto.error(safe_log(f"[SpyCloud] Failed fetching domain {domain}: {str(e)}"))
                    break

    else:
        endpoint = WATCHLIST_ENDPOINT
        cursor = " "
        while cursor:
            try:
                response = client.query_spy_cloud_api(endpoint, {**base_args, "cursor": cursor})
                watchlist_results = response.get("results", [])
                results.extend(watchlist_results)
                cursor = response.get("cursor", "")
            except Exception as e:
                demisto.error(safe_log(f"[SpyCloud] Failed fetching watchlist: {str(e)}"))
                break

    return results


def build_iterators(client: Client, results: list) -> list:
    incident_record = []
    for item in results:
        source_id = item.get("source_id")
        catalog_resp = client.query_spy_cloud_api(f"breach/catalog/{source_id}", {}).get("results", [])
        item["breach_title"] = catalog_resp[0].get("title") if catalog_resp else ""
        severity = item["severity"]
        name_ext = item.get("email") or (item.get("ip_addresses") or [""])[0] or item.get("username") or item["document_id"]
        incident_record.append(
            {
                "type": INCIDENT_TYPE[severity],
                "name": f"{INCIDENT_NAME[severity]} {name_ext}",
                "rawJSON": dumps(item),
                "severity": SEVERITY_VALUE[severity],
                "dbotMirrorId": item["document_id"],
            }
        )
    return incident_record


def fetch_incident(client: Client, args: dict):
    context = demisto.getIntegrationContext() or {}
    left_results = context.get("results", [])
    last_mod_check = context.get("last_mod_check")

    try:
        max_fetch = int(args.get("fetch_limit", DEFAULT_FETCH_LIMIT))
    except (ValueError, TypeError):
        max_fetch = DEFAULT_FETCH_LIMIT

    # process leftover results
    if left_results:
        incidents = build_iterators(client, left_results[:max_fetch])
        context["results"] = left_results[max_fetch:] if len(left_results) > max_fetch else []
        demisto.setIntegrationContext(context)
        return incidents

    # new fetch
    param = create_spycloud_args(args, client)
    since = param.pop("since")
    until = param.pop("until")
    param.pop("since_modification_date", None)
    param.pop("until_modification_date", None)

    now = datetime.now(UTC)
    yesterday_str = (now - timedelta(days=1)).strftime("%Y-%m-%d")

    demisto.info(safe_log(f"[SpyCloud] Last modification check: {last_mod_check}, yesterday_str: {yesterday_str}"))

    # primary fetch
    since_results = fetch_domain_or_watchlist_data(client, args, {**param, "since": since, "until": until})

    # daily modification fetch
    run_mod_check = last_mod_check != yesterday_str
    demisto.info(safe_log(f"[SpyCloud] Will run once-per-day modification fetch: {run_mod_check}"))

    modified_results = []
    if run_mod_check:
        context["last_mod_check"] = yesterday_str
        demisto.setIntegrationContext(context)
        modified_results = fetch_domain_or_watchlist_data(
            client,
            args,
            {**param, "since_modification_date": yesterday_str, "until_modification_date": yesterday_str},
        )

    client.set_last_run()

    # dedupe
    document_ids_seen = {r["document_id"] for r in modified_results}
    deduped_results = modified_results + [r for r in since_results if r["document_id"] not in document_ids_seen]

    if len(deduped_results) > max_fetch:
        context["results"] = deduped_results[max_fetch:]
        demisto.setIntegrationContext(context)
        return build_iterators(client, deduped_results[:max_fetch])

    context["results"] = []
    demisto.setIntegrationContext(context)
    return build_iterators(client, deduped_results)


def test_module(client: Client, params: dict) -> str:
    args = create_spycloud_args(params, client)
    client.query_spy_cloud_api(WATCHLIST_ENDPOINT, args)
    return "ok"


def main():
    params = demisto.params()
    apikey = params.get("apikey")
    verify_certificate = not params.get("insecure", False)
    proxy = params.get("proxy", False)
    handle_proxy()
    command = demisto.command()

    try:
        base_url = params.get("url")
        client = Client(base_url, apikey, verify=verify_certificate, proxy=proxy)

        if command == "test-module":
            return_results(test_module(client, params))

        elif command == "fetch-incidents":
            demisto.incidents(fetch_incident(client, params))

        else:
            raise NotImplementedError(f"command {command} is not supported")

    except Exception as e:
        return_error(safe_log(f"Failed to execute {command} command. Error: {str(e)}"))


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