"""Distributor registry (T03) — flavor -> distributor, the one place that knows
about all flavor edges. Adding a flavor = one entry here + one adapter module.
"""

from __future__ import annotations

from typing import Optional

from .base import BaseDistributor
from .claude import ClaudeDistributor
from .codex import CodexDistributor
from .grok import GrokDistributor

_REGISTRY: dict[str, BaseDistributor] = {
    "claude": ClaudeDistributor(),
    "codex": CodexDistributor(),
    "grok": GrokDistributor(),
}


def get_distributor(flavor: str) -> Optional[BaseDistributor]:
    return _REGISTRY.get(flavor)


def all_flavors() -> list[str]:
    return list(_REGISTRY)