tracer-cloud--opensre/tools/system/fleet_monitoring/provider_ids.py

"""Canonical provider IDs and discovery-name to provider-ID mapping.

The provider-ID set is the shared contract every layer of the agents
stack agrees on: discovery classifies commands into these names,
the token source and meter registries key off the same names, and
``provider_for`` resolves an ``AgentRecord`` to one of them. Keeping
the constants and the small name-resolution helper here lets all of
those consumers depend on a single thin module instead of pulling in
the heavier discovery (``ps``, process dedupe, Cursor terminal
metadata) just to read a frozenset.
"""

from __future__ import annotations

KNOWN_PROVIDERS: frozenset[str] = frozenset(
    {
        "claude-code",
        "codex",
        "cursor",
        "aider",
        "gemini-cli",
        "antigravity-cli",
        "opencode",
        "kimi",
        "copilot",
    }
)

# ``cursor-claude-code`` is the Anthropic extension wrapping the real
# ``claude`` binary with ``--output-format stream-json``; same NDJSON,
# same meter. The other cursor flavors emit plain text.
_CURSOR_FAMILY_TO_PROVIDER: dict[str, str] = {
    "cursor-claude-code": "claude-code",
    "cursor-agent-exec": "cursor",
    "cursor-agent": "cursor",
}


def provider_from_classified_name(name: str) -> str | None:
    """Derive a canonical provider id from a discovery-style name."""
    base = _strip_pid_suffix(name)
    if base in _CURSOR_FAMILY_TO_PROVIDER:
        return _CURSOR_FAMILY_TO_PROVIDER[base]
    if base in KNOWN_PROVIDERS:
        return base
    return None


def _strip_pid_suffix(name: str) -> str:
    if "-" not in name:
        return name
    base, _, tail = name.rpartition("-")
    if tail.isdigit():
        return base
    return name


__all__ = ["KNOWN_PROVIDERS", "provider_from_classified_name"]