tracer-cloud--opensre/surfaces/cli/wizard/integration_validators/client_validators.py

"""Client-backed onboarding integration validators."""

from __future__ import annotations

from pathlib import Path

from integrations.alertmanager.client import make_alertmanager_client
from integrations.betterstack import build_betterstack_config, validate_betterstack_config
from integrations.config_models import (
    AWSIntegrationConfig,
    CoralogixIntegrationConfig,
    GoogleDocsIntegrationConfig,
    GrafanaIntegrationConfig,
    HoneycombIntegrationConfig,
    IncidentIoIntegrationConfig,
    PagerDutyIntegrationConfig,
)
from integrations.coralogix.client import CoralogixClient
from integrations.dagster import build_dagster_config, validate_dagster_config
from integrations.datadog.client import DatadogClient, DatadogConfig
from integrations.elasticsearch.client import ElasticsearchClient, ElasticsearchConfig
from integrations.gitlab import build_gitlab_config, validate_gitlab_config
from integrations.grafana.client import get_grafana_client_from_credentials
from integrations.honeycomb.client import HoneycombClient
from integrations.incident_io.client import IncidentIoClient
from integrations.jenkins import build_jenkins_config, validate_jenkins_config
from integrations.opsgenie.client import OpsGenieClient, OpsGenieConfig
from integrations.pagerduty.client import PagerDutyClient
from integrations.sentry import build_sentry_config, validate_sentry_config
from integrations.splunk.client import SplunkClient, SplunkConfig
from integrations.tempo import build_tempo_config, validate_tempo_config
from integrations.vercel.client import VercelClient, VercelConfig

from .shared import IntegrationHealthResult


def validate_grafana_integration(*, endpoint: str, api_key: str) -> IntegrationHealthResult:
    """Validate Grafana credentials by discovering datasource UIDs."""
    try:
        grafana_config = GrafanaIntegrationConfig.model_validate(
            {"endpoint": endpoint, "api_key": api_key}
        )
        client = get_grafana_client_from_credentials(
            endpoint=grafana_config.endpoint,
            api_key=grafana_config.api_key,
            account_id="opensre_onboard_probe",
        )
        discovered = client.discover_datasource_uids()
        if not discovered:
            return IntegrationHealthResult(
                ok=False,
                detail="Grafana is reachable, but no datasources could be discovered with this token.",
            )

        available = ", ".join(sorted(discovered))
        return IntegrationHealthResult(
            ok=True,
            detail=f"Grafana validated with datasource discovery: {available}.",
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"Grafana validation failed: {err}")


def validate_datadog_integration(
    *, api_key: str, app_key: str, site: str
) -> IntegrationHealthResult:
    """Validate Datadog credentials with a monitor list request."""
    client = DatadogClient(DatadogConfig(api_key=api_key, app_key=app_key, site=site))
    result = client.list_monitors()
    if result.get("success"):
        return IntegrationHealthResult(
            ok=True,
            detail=f"Datadog validated against {site}; fetched {result.get('total', 0)} monitors.",
        )
    return IntegrationHealthResult(
        ok=False,
        detail=f"Datadog validation failed: {result.get('error', 'unknown error')}",
    )


def validate_honeycomb_integration(
    *,
    api_key: str,
    dataset: str,
    base_url: str,
) -> IntegrationHealthResult:
    """Validate Honeycomb credentials with auth and a lightweight query."""
    try:
        honeycomb_config = HoneycombIntegrationConfig.model_validate(
            {
                "api_key": api_key,
                "dataset": dataset,
                "base_url": base_url,
            }
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=str(err))

    client = HoneycombClient(honeycomb_config)
    auth_result = client.validate_access()
    if not auth_result.get("success"):
        return IntegrationHealthResult(
            ok=False,
            detail=f"Honeycomb auth failed: {auth_result.get('error', 'unknown error')}",
        )

    query_result = client.run_query(
        {"calculations": [{"op": "COUNT"}], "time_range": 900},
        limit=1,
    )
    if not query_result.get("success"):
        return IntegrationHealthResult(
            ok=False,
            detail=f"Honeycomb query failed: {query_result.get('error', 'unknown error')}",
        )

    return IntegrationHealthResult(
        ok=True,
        detail=(
            f"Honeycomb validated against dataset {honeycomb_config.dataset} "
            f"at {honeycomb_config.base_url}."
        ),
    )


def validate_coralogix_integration(
    *,
    api_key: str,
    base_url: str,
    application_name: str = "",
    subsystem_name: str = "",
) -> IntegrationHealthResult:
    """Validate Coralogix access with a lightweight DataPrime query."""
    try:
        coralogix_config = CoralogixIntegrationConfig.model_validate(
            {
                "api_key": api_key,
                "base_url": base_url,
                "application_name": application_name,
                "subsystem_name": subsystem_name,
            }
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=str(err))

    client = CoralogixClient(coralogix_config)
    result = client.validate_access()
    if not result.get("success"):
        return IntegrationHealthResult(
            ok=False,
            detail=f"Coralogix validation failed: {result.get('error', 'unknown error')}",
        )

    scope: list[str] = []
    if coralogix_config.application_name:
        scope.append(f"application {coralogix_config.application_name}")
    if coralogix_config.subsystem_name:
        scope.append(f"subsystem {coralogix_config.subsystem_name}")
    scope_suffix = f" ({', '.join(scope)})" if scope else ""
    return IntegrationHealthResult(
        ok=True,
        detail=(
            f"Coralogix validated against {coralogix_config.base_url}{scope_suffix}; "
            f"DataPrime returned {result.get('total', 0)} row(s)."
        ),
    )


def validate_google_docs_integration(
    *,
    credentials_file: str,
    folder_id: str,
) -> IntegrationHealthResult:
    """Validate Google Docs credentials and folder access."""
    from integrations.google_docs.client import GoogleDocsClient

    try:
        config = GoogleDocsIntegrationConfig.model_validate(
            {
                "credentials_file": credentials_file,
                "folder_id": folder_id,
            }
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=str(err))

    if not config.credentials_file or not config.folder_id:
        return IntegrationHealthResult(ok=False, detail="Missing credentials_file or folder_id.")

    if not Path(config.credentials_file).exists():
        return IntegrationHealthResult(
            ok=False, detail=f"Credentials file not found: {config.credentials_file}"
        )

    try:
        client = GoogleDocsClient(config)
        result = client.validate_access()
    except Exception as exc:
        return IntegrationHealthResult(ok=False, detail=f"Google API validation failed: {exc}")

    if not result.get("success"):
        return IntegrationHealthResult(
            ok=False, detail=f"Folder access check failed: {result.get('error', 'unknown error')}"
        )

    return IntegrationHealthResult(
        ok=True,
        detail=f"Connected to Drive folder {config.folder_id} ({result.get('file_count', 0)} items).",
    )


def validate_aws_integration(
    *,
    region: str,
    role_arn: str = "",
    external_id: str = "",
    access_key_id: str = "",
    secret_access_key: str = "",
    session_token: str = "",
) -> IntegrationHealthResult:
    """Validate AWS credentials with STS GetCallerIdentity."""
    try:
        import boto3
    except ImportError:
        return IntegrationHealthResult(
            ok=False, detail="AWS validation failed: boto3 is not installed."
        )

    try:
        aws_config = AWSIntegrationConfig.model_validate(
            {
                "region": region,
                "role_arn": role_arn,
                "external_id": external_id,
                "credentials": (
                    {
                        "access_key_id": access_key_id,
                        "secret_access_key": secret_access_key,
                        "session_token": session_token,
                    }
                    if access_key_id or secret_access_key or session_token
                    else None
                ),
            }
        )
        if role_arn:
            sts = boto3.client("sts", region_name=aws_config.region)
            assume_kwargs: dict[str, str] = {
                "RoleArn": aws_config.role_arn,
                "RoleSessionName": "opensre-onboard-check",
            }
            if aws_config.external_id:
                assume_kwargs["ExternalId"] = aws_config.external_id
            creds = sts.assume_role(**assume_kwargs)["Credentials"]
            assumed = boto3.client(
                "sts",
                region_name=aws_config.region,
                aws_access_key_id=creds["AccessKeyId"],
                aws_secret_access_key=creds["SecretAccessKey"],
                aws_session_token=creds["SessionToken"],
            )
            identity = assumed.get_caller_identity()
            return IntegrationHealthResult(
                ok=True,
                detail=f"AWS role validated for account {identity.get('Account')} as {identity.get('Arn')}.",
            )

        sts = boto3.client(
            "sts",
            region_name=aws_config.region,
            aws_access_key_id=aws_config.credentials.access_key_id
            if aws_config.credentials
            else "",
            aws_secret_access_key=aws_config.credentials.secret_access_key
            if aws_config.credentials
            else "",
            aws_session_token=(
                aws_config.credentials.session_token if aws_config.credentials else ""
            )
            or None,
        )
        identity = sts.get_caller_identity()
        return IntegrationHealthResult(
            ok=True,
            detail=f"AWS credentials validated for account {identity.get('Account')} as {identity.get('Arn')}.",
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"AWS validation failed: {err}")


def validate_sentry_integration(
    *,
    base_url: str,
    organization_slug: str,
    auth_token: str,
    project_slug: str = "",
) -> IntegrationHealthResult:
    """Validate Sentry connectivity with an organization issues query."""
    config = build_sentry_config(
        {
            "base_url": base_url,
            "organization_slug": organization_slug,
            "auth_token": auth_token,
            "project_slug": project_slug,
        }
    )
    result = validate_sentry_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_gitlab_integration(
    *,
    base_url: str,
    auth_token: str,
) -> IntegrationHealthResult:
    """Validate Gitlab connectivity with an users api."""
    config = build_gitlab_config({"base_url": base_url, "auth_token": auth_token})
    result = validate_gitlab_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_dagster_integration(
    *,
    endpoint: str,
    api_token: str = "",
) -> IntegrationHealthResult:
    """Validate Dagster connectivity via a GraphQL version probe."""
    config = build_dagster_config({"endpoint": endpoint, "api_token": api_token})
    result = validate_dagster_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_jenkins_integration(
    *,
    base_url: str,
    username: str,
    api_token: str,
) -> IntegrationHealthResult:
    """Validate Jenkins connectivity with a server-info query."""
    config = build_jenkins_config(
        {"base_url": base_url, "username": username, "api_token": api_token}
    )
    result = validate_jenkins_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_betterstack_integration(
    *,
    query_endpoint: str,
    username: str,
    password: str,
    sources: list[str] | None = None,
) -> IntegrationHealthResult:
    """Validate Better Stack Telemetry credentials via a ``SELECT 1`` probe."""
    try:
        config = build_betterstack_config(
            {
                "query_endpoint": query_endpoint,
                "username": username,
                "password": password,
                "sources": list(sources or []),
            }
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"Better Stack config invalid: {err}")
    result = validate_betterstack_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_vercel_integration(*, api_token: str, team_id: str = "") -> IntegrationHealthResult:
    """Validate Vercel credentials by listing accessible projects."""
    if not api_token:
        return IntegrationHealthResult(ok=False, detail="Vercel API token is required.")
    try:
        with VercelClient(VercelConfig(api_token=api_token, team_id=team_id)) as client:
            result = client.list_projects()
        if result.get("success"):
            return IntegrationHealthResult(
                ok=True,
                detail=f"Vercel validated; listed {result.get('total', 0)} project(s).",
            )
        return IntegrationHealthResult(
            ok=False,
            detail=f"Vercel validation failed: {result.get('error', 'unknown error')}",
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"Vercel validation failed: {err}")


def validate_alertmanager_integration(
    *,
    base_url: str,
    bearer_token: str = "",
    username: str = "",
    password: str = "",
) -> IntegrationHealthResult:
    """Validate Alertmanager connectivity via the /api/v2/status endpoint."""
    if not base_url:
        return IntegrationHealthResult(ok=False, detail="Alertmanager URL is required.")
    client = make_alertmanager_client(
        base_url=base_url,
        bearer_token=bearer_token or None,
        username=username or None,
        password=password or None,
    )
    if client is None:
        return IntegrationHealthResult(ok=False, detail="Invalid Alertmanager URL.")
    try:
        with client:
            result = client.get_status()
        if result.get("success"):
            status_data = result.get("status", {})
            cluster_status = (
                status_data.get("cluster", {}).get("status", "unknown")
                if isinstance(status_data, dict)
                else "ok"
            )
            return IntegrationHealthResult(
                ok=True,
                detail=f"Connected to Alertmanager at {base_url}; cluster status: {cluster_status}.",
            )
        return IntegrationHealthResult(
            ok=False,
            detail=f"Alertmanager validation failed: {result.get('error', 'unknown error')}",
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"Alertmanager validation failed: {err}")


def validate_opsgenie_integration(
    *,
    api_key: str,
    region: str = "us",
) -> IntegrationHealthResult:
    """Validate OpsGenie connectivity by listing alerts."""
    if not api_key:
        return IntegrationHealthResult(ok=False, detail="OpsGenie API key is required.")
    try:
        config = OpsGenieConfig(api_key=api_key, region=region)
        with OpsGenieClient(config) as client:
            result = client.list_alerts(limit=1)
        if result.get("success"):
            return IntegrationHealthResult(
                ok=True,
                detail=f"OpsGenie validated ({config.region.upper()} region); API key accepted.",
            )
        return IntegrationHealthResult(
            ok=False,
            detail=f"OpsGenie validation failed: {result.get('error', 'unknown error')}",
        )
    except Exception as err:
        return IntegrationHealthResult(
            ok=False,
            detail=f"OpsGenie validation failed: {err}",
        )


def validate_pagerduty_integration(
    *,
    api_key: str,
    base_url: str,
) -> IntegrationHealthResult:
    """Validate Pagerduty connectivity by listing alerts."""
    if not api_key:
        return IntegrationHealthResult(ok=False, detail="PagerDuty API key is required.")
    try:
        config = PagerDutyIntegrationConfig(api_key=api_key, base_url=base_url)
        with PagerDutyClient(config) as client:
            result = client.list_incidents(limit=1)
        if result.get("success"):
            return IntegrationHealthResult(
                ok=True,
                detail="Connected to PagerDuty; API key accepted.",
            )
        return IntegrationHealthResult(
            ok=False,
            detail=f"PagerDuty validation failed: {result.get('error', 'unknown error')}",
        )
    except Exception as err:
        return IntegrationHealthResult(
            ok=False,
            detail=f"PagerDuty validation failed: {str(err).replace(api_key, '[REDACTED]')}",
        )


def validate_incident_io_integration(
    *,
    api_key: str,
    base_url: str = "",
) -> IntegrationHealthResult:
    """Validate incident.io connectivity by listing one incident."""
    if not api_key:
        return IntegrationHealthResult(ok=False, detail="incident.io API key is required.")
    try:
        config = IncidentIoIntegrationConfig(api_key=api_key, base_url=base_url)
        with IncidentIoClient(config) as client:
            result = client.list_incidents(status_category="", page_size=1)
        if result.get("success"):
            return IntegrationHealthResult(
                ok=True,
                detail="incident.io validated; API key accepted.",
            )
        return IntegrationHealthResult(
            ok=False,
            detail=f"incident.io validation failed: {result.get('error', 'unknown error')}",
        )
    except Exception as err:
        return IntegrationHealthResult(
            ok=False,
            detail=f"incident.io validation failed: {str(err).replace(api_key, '[REDACTED]')}",
        )


def validate_splunk_integration(
    *,
    base_url: str,
    token: str,
    index: str = "main",
    verify_ssl: bool = True,
    ca_bundle: str = "",
) -> IntegrationHealthResult:
    """Validate Splunk credentials by calling the server info endpoint."""
    client = SplunkClient(
        SplunkConfig(
            base_url=base_url,
            token=token,
            index=index,
            verify_ssl=verify_ssl,
            ca_bundle=ca_bundle,
        )
    )
    result = client.validate_access()
    if result.get("success"):
        return IntegrationHealthResult(ok=True, detail=result.get("detail", "Splunk connected."))
    return IntegrationHealthResult(
        ok=False,
        detail=f"Splunk validation failed: {result.get('error', 'unknown error')}",
    )


def validate_tempo_integration(
    *,
    url: str,
    api_key: str = "",
    username: str = "",
    password: str = "",
    org_id: str = "",
) -> IntegrationHealthResult:
    """Validate Tempo connectivity via the tag-search endpoint."""
    try:
        config = build_tempo_config(
            {
                "url": url,
                "api_key": api_key,
                "username": username,
                "password": password,
                "org_id": org_id,
            }
        )
    except Exception as err:
        return IntegrationHealthResult(ok=False, detail=f"Tempo config invalid: {err}")
    result = validate_tempo_config(config)
    return IntegrationHealthResult(ok=result.ok, detail=result.detail)


def validate_opensearch_integration(
    *,
    url: str,
    api_key: str = "",
    username: str = "",
    password: str = "",
) -> IntegrationHealthResult:
    """Validate OpenSearch / Elasticsearch connectivity via GET /_cluster/health.

    Supports three authentication modes:
    - No authentication (security disabled clusters)
    - API key (native to Elasticsearch and some OpenSearch deployments)
    - HTTP Basic Auth (default for most self-hosted OpenSearch clusters)
    """
    if not url:
        return IntegrationHealthResult(ok=False, detail="OpenSearch URL is required.")
    config = ElasticsearchConfig(
        url=url,
        api_key=api_key or None,
        username=username or None,
        password=password or None,
    )
    client = ElasticsearchClient(config)
    result = client.get_cluster_health()
    if result.get("success"):
        cluster_name = result.get("cluster_name") or "unknown"
        cluster_status = result.get("status") or "unknown"
        node_count = result.get("number_of_nodes", 0)
        return IntegrationHealthResult(
            ok=True,
            detail=(
                f"Connected to OpenSearch cluster '{cluster_name}' "
                f"({cluster_status}, {node_count} node(s))."
            ),
        )
    return IntegrationHealthResult(
        ok=False,
        detail=f"OpenSearch validation failed: {result.get('error', 'unknown error')}",
    )