"""Path → pytest target mapping for branch-scoped test runs (CI.md §2).

This module is the single source of truth for ``make test-scope``. Edit rules
here only — do not duplicate the mapping table in CI.md.
"""

from __future__ import annotations

from dataclasses import dataclass
from pathlib import Path

# Distinct app areas in one diff that trigger escalation to ``make test-cov``.
ESCALATION_AREA_THRESHOLD = 3


@dataclass(frozen=True, slots=True)
class PathRule:
    """Map changed paths under ``path_prefix`` to pytest targets."""

    path_prefix: str
    test_targets: tuple[str, ...]
    always_escalate: bool = False


# Matched in list order — more specific prefixes must appear before parents.
RULES: tuple[PathRule, ...] = (
    # Shared core (always escalate)
    PathRule("core/domain/", (), always_escalate=True),
    PathRule("core/", ("tests/core/",)),
    PathRule("tools/investigation/reporting/", ("tests/delivery/",)),
    PathRule("tools/investigation/", (), always_escalate=True),
    PathRule("utils/", (), always_escalate=True),
    # Specific sub-packages before their parent
    PathRule("integrations/llm_cli/", ("tests/integrations/llm_cli/",)),
    PathRule("integrations/opensre/", ("tests/integrations/opensre/",)),
    PathRule("integrations/hermes/", ("tests/hermes/",)),
    PathRule(
        "integrations/alertmanager/",
        ("tests/integrations/alertmanager/", "tests/e2e/alertmanager/"),
    ),
    PathRule(
        "integrations/dagster/",
        ("tests/integrations/dagster/", "tests/synthetic/test_dagster_scenario.py"),
    ),
    PathRule(
        "integrations/eks/",
        (
            "tests/integrations/eks/",
            "tests/tools/test_eks_deployment_status_tool.py",
            "tests/tools/test_eks_describe_addon_tool.py",
            "tests/tools/test_eks_describe_cluster_tool.py",
            "tests/tools/test_eks_events_tool.py",
            "tests/tools/test_eks_list_clusters_tool.py",
            "tests/tools/test_eks_list_deployments_tool.py",
            "tests/tools/test_eks_list_namespaces_tool.py",
            "tests/tools/test_eks_list_pods_tool.py",
            "tests/tools/test_eks_node_health_tool.py",
            "tests/tools/test_eks_nodegroup_health_tool.py",
            "tests/tools/test_eks_pod_logs_tool.py",
            "tests/tools/test_telemetry.py",
            "tests/benchmarks/cloudopsbench/tests/test_bench_agent.py",
        ),
    ),
    PathRule(
        "integrations/elasticsearch/",
        (
            "tests/integrations/elasticsearch/",
            "tests/tools/test_elasticsearch_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/google_docs/",
        (
            "tests/integrations/google_docs/",
            "tests/tools/test_google_docs_create_report_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/groundcover/",
        ("tests/integrations/groundcover/", "tests/tools/test_groundcover_tools.py"),
    ),
    PathRule(
        "integrations/helm/",
        ("tests/integrations/helm/", "tests/tools/test_helm_tools.py"),
    ),
    PathRule(
        "integrations/incident_io/",
        ("tests/integrations/incident_io/", "tests/tools/test_incident_io_tool.py"),
    ),
    PathRule(
        "integrations/jira/",
        (
            "tests/integrations/jira/",
            "tests/tools/test_jira_add_comment_tool.py",
            "tests/tools/test_jira_create_issue_tool.py",
            "tests/tools/test_jira_issue_detail_tool.py",
            "tests/tools/test_jira_search_issues_tool.py",
        ),
    ),
    PathRule(
        "integrations/clickhouse/",
        (
            "tests/integrations/clickhouse/",
            "tests/tools/test_clickhouse_query_activity_tool.py",
            "tests/tools/test_clickhouse_system_health_tool.py",
        ),
    ),
    PathRule(
        "integrations/mariadb/",
        (
            "tests/integrations/mariadb/",
            "tests/tools/test_mariadb_innodb_status_tool.py",
            "tests/tools/test_mariadb_process_list_tool.py",
            "tests/tools/test_mariadb_replication_tool.py",
            "tests/tools/test_mariadb_slow_queries_tool.py",
            "tests/tools/test_mariadb_status_tool.py",
            "tests/e2e/mariadb/",
        ),
    ),
    PathRule(
        "integrations/mongodb_atlas/",
        (
            "tests/integrations/mongodb_atlas/",
            "tests/tools/test_mongodb_atlas_alerts_tool.py",
            "tests/tools/test_mongodb_atlas_clusters_tool.py",
            "tests/tools/test_mongodb_atlas_events_tool.py",
            "tests/tools/test_mongodb_atlas_metrics_tool.py",
            "tests/tools/test_mongodb_atlas_performance_advisor_tool.py",
        ),
    ),
    PathRule(
        "integrations/mongodb/",
        (
            "tests/integrations/mongodb/",
            "tests/tools/test_mongodb_collection_stats_tool.py",
            "tests/tools/test_mongodb_current_ops_tool.py",
            "tests/tools/test_mongodb_profiler_tool.py",
            "tests/tools/test_mongodb_replica_status_tool.py",
            "tests/tools/test_mongodb_server_status_tool.py",
            "tests/e2e/mongodb/",
        ),
    ),
    PathRule(
        "integrations/mysql/",
        (
            "tests/integrations/mysql/",
            "tests/tools/test_mysql_current_processes_tool.py",
            "tests/tools/test_mysql_replication_status_tool.py",
            "tests/tools/test_mysql_server_status_tool.py",
            "tests/tools/test_mysql_slow_queries_tool.py",
            "tests/tools/test_mysql_table_stats_tool.py",
            "tests/e2e/mysql/",
        ),
    ),
    PathRule(
        "integrations/postgresql/",
        (
            "tests/integrations/postgresql/",
            "tests/tools/test_postgresql_current_queries_tool.py",
            "tests/tools/test_postgresql_locks_tool.py",
            "tests/tools/test_postgresql_replication_status_tool.py",
            "tests/tools/test_postgresql_server_status_tool.py",
            "tests/tools/test_postgresql_slow_queries_tool.py",
            "tests/tools/test_postgresql_table_stats_tool.py",
            "tests/e2e/postgresql/",
        ),
    ),
    PathRule(
        "integrations/redis/",
        (
            "tests/integrations/redis/",
            "tests/tools/test_redis_client_list_tool.py",
            "tests/tools/test_redis_key_scan_tool.py",
            "tests/tools/test_redis_latency_doctor_tool.py",
            "tests/tools/test_redis_list_depth_tool.py",
            "tests/tools/test_redis_replication_tool.py",
            "tests/tools/test_redis_server_info_tool.py",
            "tests/tools/test_redis_slowlog_tool.py",
            "tests/e2e/redis/",
        ),
    ),
    PathRule(
        "integrations/snowflake/",
        (
            "tests/integrations/snowflake/",
            "tests/tools/test_snowflake_query_history_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/azure/",
        (
            "tests/tools/test_azure_monitor_logs_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/azure_sql/",
        (
            "tests/integrations/test_azure_sql.py",
            "tests/tools/test_azure_sql_current_queries_tool.py",
            "tests/tools/test_azure_sql_resource_stats_tool.py",
            "tests/tools/test_azure_sql_server_status_tool.py",
            "tests/tools/test_azure_sql_slow_queries_tool.py",
            "tests/tools/test_azure_sql_wait_stats_tool.py",
        ),
    ),
    PathRule(
        "integrations/betterstack/",
        (
            "tests/integrations/test_betterstack.py",
            "tests/tools/test_betterstack_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/hermes/tools/",
        (
            "tests/tools/test_hermes_logs_tool.py",
            "tests/tools/test_hermes_session_evidence_tool.py",
        ),
    ),
    PathRule(
        "integrations/kafka/",
        (
            "tests/integrations/test_kafka.py",
            "tests/tools/test_kafka_consumer_group_tool.py",
            "tests/tools/test_kafka_topic_health_tool.py",
        ),
    ),
    PathRule(
        "integrations/openclaw/",
        (
            "tests/tools/test_openclaw_mcp_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/openobserve/",
        (
            "tests/tools/test_openobserve_logs_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/opensearch/",
        (
            "tests/integrations/test_opensearch_catalog.py",
            "tests/tools/test_opensearch_analytics_tool.py",
        ),
    ),
    PathRule(
        "integrations/posthog_mcp/",
        (
            "tests/integrations/test_posthog_mcp.py",
            "tests/tools/test_posthog_mcp_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/rabbitmq/",
        (
            "tests/integrations/test_rabbitmq.py",
            "tests/tools/test_rabbitmq_broker_overview_tool.py",
            "tests/tools/test_rabbitmq_connection_stats_tool.py",
            "tests/tools/test_rabbitmq_consumer_health_tool.py",
            "tests/tools/test_rabbitmq_node_health_tool.py",
            "tests/tools/test_rabbitmq_queue_backlog_tool.py",
        ),
    ),
    PathRule(
        "integrations/sentry_mcp/",
        (
            "tests/integrations/test_sentry_mcp.py",
            "tests/tools/test_sentry_mcp_tool.py",
            "tests/tools/test_telemetry.py",
        ),
    ),
    PathRule(
        "integrations/sentry/",
        (
            "tests/tools/test_sentry_issue_details_tool.py",
            "tests/tools/test_sentry_issue_events_tool.py",
            "tests/tools/test_sentry_search_issues_tool.py",
        ),
    ),
    PathRule(
        "integrations/supabase/",
        (
            "tests/integrations/test_supabase.py",
            "tests/tools/test_supabase_health_tool.py",
            "tests/tools/test_supabase_storage_tool.py",
        ),
    ),
    PathRule(
        "integrations/bitbucket/",
        (
            "tests/integrations/test_bitbucket.py",
            "tests/tools/test_bitbucket_commits_tool.py",
            "tests/tools/test_bitbucket_file_contents_tool.py",
            "tests/tools/test_bitbucket_search_code_tool.py",
        ),
    ),
    PathRule(
        "integrations/telegram/tools/",
        ("tests/tools/test_telegram_send_message_tool.py",),
    ),
    PathRule(
        "integrations/tracer/tools/",
        (
            "tests/tools/test_tracer_airflow_metrics_tool.py",
            "tests/tools/test_tracer_batch_statistics_tool.py",
            "tests/tools/test_tracer_error_logs_tool.py",
            "tests/tools/test_tracer_failed_jobs_tool.py",
            "tests/tools/test_tracer_failed_run_tool.py",
            "tests/tools/test_tracer_failed_tools_tool.py",
            "tests/tools/test_tracer_host_metrics_tool.py",
            "tests/tools/test_tracer_run_tool.py",
            "tests/tools/test_tracer_tasks_tool.py",
        ),
    ),
    PathRule(
        "integrations/twilio/",
        (
            "tests/integrations/test_twilio.py",
            "tests/tools/test_twilio_notify_tool.py",
        ),
    ),
    PathRule(
        "integrations/github/tools/",
        (
            "tests/tools/test_github_actions_tool.py",
            "tests/tools/test_github_commits_tool.py",
            "tests/tools/test_github_file_contents_tool.py",
            "tests/tools/test_github_helpers.py",
            "tests/tools/test_github_issues_tool.py",
            "tests/tools/test_github_repo_scope.py",
            "tests/tools/test_github_repository_tool.py",
            "tests/tools/test_github_repository_tree_tool.py",
            "tests/tools/test_github_search_code_tool.py",
            "tests/tools/test_github_workflow_tools.py",
        ),
    ),
    PathRule(
        "integrations/gitlab/",
        (
            "tests/integrations/test_gitlab.py",
            "tests/tools/test_gitlab_commits_tool.py",
            "tests/tools/test_gitlab_file_tool.py",
            "tests/tools/test_gitlab_mrs_tool.py",
            "tests/tools/test_gitlab_pipelines_tool.py",
            "tests/e2e/gitlab/",
        ),
    ),
    PathRule(
        "integrations/aws/tools/",
        ("tests/tools/test_aws_operation_tool.py",),
    ),
    PathRule(
        "integrations/aws_lambda/",
        (
            "tests/integrations/aws/test_lambda_client.py",
            "tests/tools/test_lambda_config_tool.py",
            "tests/tools/test_lambda_errors_tool.py",
            "tests/tools/test_lambda_inspect_tool.py",
            "tests/tools/test_lambda_invocation_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/cloudtrail/",
        ("tests/tools/test_cloudtrail_events.py",),
    ),
    PathRule(
        "integrations/cloudwatch/",
        (
            "tests/integrations/aws/test_cloudwatch_client.py",
            "tests/tools/test_cloudwatch_batch_metrics_tool.py",
            "tests/tools/test_cloudwatch_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/ec2/",
        ("tests/tools/test_ec2_instances_by_tag_tool.py",),
    ),
    PathRule(
        "integrations/elb/",
        ("tests/tools/test_elb_target_health_tool.py",),
    ),
    PathRule(
        "integrations/rds/",
        (
            "tests/integrations/test_rds.py",
            "tests/tools/test_rds_tools.py",
        ),
    ),
    PathRule(
        "integrations/s3/",
        (
            "tests/integrations/aws/test_s3_client.py",
            "tests/tools/test_s3_get_object_tool.py",
            "tests/tools/test_s3_inspect_tool.py",
            "tests/tools/test_s3_list_tool.py",
            "tests/tools/test_s3_marker_tool.py",
        ),
    ),
    PathRule(
        "integrations/opsgenie/",
        (
            "tests/integrations/opsgenie/",
            "tests/tools/test_opsgenie_alert_detail_tool.py",
            "tests/tools/test_opsgenie_alerts_tool.py",
        ),
    ),
    PathRule(
        "integrations/pagerduty/",
        (
            "tests/integrations/pagerduty/",
            "tests/tools/test_pagerduty_incident_detail_tool.py",
            "tests/tools/test_pagerduty_incidents_tool.py",
            "tests/tools/test_pagerduty_oncall_tool.py",
            "tests/tools/test_pagerduty_services_tool.py",
        ),
    ),
    PathRule(
        "integrations/prefect/",
        (
            "tests/integrations/prefect/",
            "tests/tools/test_prefect_flow_runs_tool.py",
            "tests/tools/test_prefect_worker_health_tool.py",
        ),
    ),
    PathRule(
        "integrations/signoz/",
        (
            "tests/integrations/signoz/",
            "tests/tools/test_signoz_tools.py",
            "tests/synthetic/test_signoz_scenario.py",
        ),
    ),
    PathRule(
        "integrations/splunk/",
        ("tests/integrations/splunk/", "tests/tools/test_splunk_search_tool.py"),
    ),
    PathRule(
        "integrations/tempo/",
        (
            "tests/integrations/tempo/",
            "tests/tools/test_tempo_tools.py",
            "tests/synthetic/test_tempo_scenario.py",
        ),
    ),
    PathRule(
        "integrations/temporal/",
        (
            "tests/integrations/temporal/",
            "tests/integrations/test_temporal_catalog.py",
            "tests/synthetic/test_temporal_scenario.py",
            "tests/tools/test_temporal_namespace_info_tool.py",
            "tests/tools/test_temporal_task_queue_tool.py",
            "tests/tools/test_temporal_workflow_history_tool.py",
            "tests/tools/test_temporal_workflows_tool.py",
        ),
    ),
    PathRule(
        "integrations/vercel/",
        (
            "tests/integrations/vercel/",
            "tests/tools/test_vercel_deployment_status_tool.py",
            "tests/tools/test_vercel_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/victoria_logs/",
        (
            "tests/integrations/victoria_logs/",
            "tests/tools/test_victoria_logs_tool.py",
            "tests/e2e/victoria_logs/",
        ),
    ),
    PathRule(
        "integrations/x_mcp/",
        (
            "tests/integrations/test_x_mcp.py",
            "tests/tools/test_x_mcp_tool.py",
        ),
    ),
    PathRule(
        "integrations/argocd/",
        (
            "tests/integrations/argocd/",
            "tests/tools/test_argocd_tools.py",
        ),
    ),
    PathRule(
        "integrations/coralogix/",
        (
            "tests/integrations/coralogix/",
            "tests/tools/test_coralogix_logs_tool.py",
        ),
    ),
    PathRule(
        "integrations/honeycomb/",
        (
            "tests/integrations/honeycomb/",
            "tests/tools/test_honeycomb_traces_tool.py",
        ),
    ),
    PathRule(
        "integrations/jenkins/",
        ("tests/integrations/test_jenkins.py", "tests/synthetic/test_jenkins_scenario.py"),
    ),
    PathRule(
        "integrations/datadog/",
        (
            "tests/integrations/datadog/",
            "tests/tools/test_datadog_context_tool.py",
            "tests/tools/test_datadog_events_tool.py",
            "tests/tools/test_datadog_logs_tool.py",
            "tests/tools/test_datadog_metrics_tool.py",
            "tests/tools/test_datadog_monitors_tool.py",
            "tests/tools/test_datadog_node_pods_tool.py",
        ),
    ),
    PathRule(
        "integrations/grafana/",
        (
            "tests/integrations/grafana/",
            "tests/tools/test_grafana_alert_rules_tool.py",
            "tests/tools/test_grafana_annotations_tool.py",
            "tests/tools/test_grafana_logs_tool.py",
            "tests/tools/test_grafana_metrics_tool.py",
            "tests/tools/test_grafana_service_names_tool.py",
            "tests/tools/test_grafana_traces_tool.py",
            "tests/e2e/grafana_validation/",
        ),
    ),
    PathRule("integrations/", ("tests/integrations/",)),
    PathRule("tools/system/fleet_monitoring/", ("tests/agent/", "tests/fleet_monitoring/")),
    PathRule("surfaces/cli/", ("tests/cli/",)),
    PathRule("surfaces/interactive_shell/", ("tests/interactive_shell/",)),
    PathRule("gateway/", ("gateway/tests/",)),
    PathRule("tools/system/watch_dog/", ("tests/watch_dog/",)),
    PathRule("tools/", ("tests/tools/",)),
    PathRule("platform/analytics/", ("tests/analytics/",)),
    PathRule("platform/guardrails/", ("tests/platform/guardrails/",)),
    PathRule("platform/masking/", ("tests/masking/",)),
    PathRule("platform/packaging/", ("tests/packaging/",)),
    PathRule("platform/sandbox/", ("tests/sandbox/",)),
    PathRule(
        "platform/deployment/",
        ("tests/deployment/", "tests/platform/deployment/test_deployment_health.py"),
    ),
    PathRule("platform/auth/", ("tests/platform/auth/",)),
    PathRule("gateway/webapp.py", ("gateway/tests/test_webapp.py",)),
    # Repo-wide config
    PathRule("pyproject.toml", (), always_escalate=True),
    PathRule("uv.lock", (), always_escalate=True),
    PathRule("pytest.ini", (), always_escalate=True),
    PathRule("Makefile", (), always_escalate=True),
    PathRule(".github/ci/", ("tests/github_ci/",)),
)


def _matches(path: str, prefix: str) -> bool:
    return path.startswith(prefix) or path == prefix.rstrip("/")


def _area_key(prefix: str) -> str:
    parts = prefix.split("/")
    if parts[0] == "deployment" or parts[:2] == ["platform", "deployment"]:
        return "deployment"
    return prefix


def classify(changed: list[str]) -> tuple[bool, list[str], list[str]]:
    """Return ``(should_escalate, test_targets, matched_areas)``."""
    escalate = False
    targets: list[str] = []
    areas: list[str] = []

    for path in changed:
        matched = False
        for rule in RULES:
            if not _matches(path, rule.path_prefix):
                continue
            matched = True
            if rule.always_escalate:
                escalate = True
            else:
                area = _area_key(rule.path_prefix)
                if area not in areas:
                    areas.append(area)
                for target in rule.test_targets:
                    if target not in targets:
                        targets.append(target)
            break

        if not matched and path.startswith("tests/") and path not in targets:
            targets.append(path)

    if len(areas) >= ESCALATION_AREA_THRESHOLD:
        escalate = True

    existing = [t for t in targets if Path(t).exists()]
    dropped = [t for t in targets if t not in existing]
    if dropped:
        print(f"  (skipping non-existent targets: {', '.join(dropped)})", flush=True)
    return escalate, existing, areas