"""Enricher registry - the plugin system. New OSINT sources register here and are

immediately available via the API without touching core code."""

from __future__ import annotations

from app.services.enrichment.base import BaseEnricher, EnrichResult

from app.services.enrichment.plugins import (

    CrtShEnricher, DnsEnricher, GithubEnricher, IpAsnEnricher,

    RdapEnricher, RobotsEnricher, SecurityTxtEnricher, TlsCertEnricher,

)

_REGISTRY: dict[str, BaseEnricher] = {

    e.name: e for e in [

        DnsEnricher(), RdapEnricher(), CrtShEnricher(), IpAsnEnricher(),

        SecurityTxtEnricher(), RobotsEnricher(), GithubEnricher(), TlsCertEnricher(),

    ]

}

def list_enrichers() -> list[dict]:

    return [{"name": e.name, "kind": e.kind, "requires_network": e.requires_network}

            for e in _REGISTRY.values()]

def get_enricher(name: str) -> BaseEnricher | None:

    return _REGISTRY.get(name)

def run_enricher(name: str, target: str) -> EnrichResult:

    enricher = _REGISTRY.get(name)

    if not enricher:

        return EnrichResult(enricher=name, target=target, error=f"unknown enricher '{name}'", confidence=0.0)

    return enricher.run(target)

__all__ = ["list_enrichers", "get_enricher", "run_enricher", "EnrichResult"]