Medios-Macina/ProviderCore/registry.py

"""Provider registry.

Concrete provider implementations live in the `Provider/` package.
This module is the single source of truth for provider discovery.
"""

from __future__ import annotations

from typing import Any, Dict, Optional, Type
import sys

from SYS.logger import log

from ProviderCore.base import FileProvider, SearchProvider, SearchResult
from Provider.bandcamp import Bandcamp
from Provider.libgen import Libgen
from Provider.matrix import Matrix
from Provider.openlibrary import OpenLibrary
from Provider.soulseek import Soulseek, download_soulseek_file
from Provider.youtube import YouTube
from Provider.zeroxzero import ZeroXZero


_SEARCH_PROVIDERS: Dict[str, Type[SearchProvider]] = {
    "libgen": Libgen,
    "openlibrary": OpenLibrary,
    "soulseek": Soulseek,
    "bandcamp": Bandcamp,
    "youtube": YouTube,
}


def get_search_provider(name: str, config: Optional[Dict[str, Any]] = None) -> Optional[SearchProvider]:
    """Get a search provider by name."""

    provider_class = _SEARCH_PROVIDERS.get((name or "").lower())
    if provider_class is None:
        log(f"[provider] Unknown search provider: {name}", file=sys.stderr)
        return None

    try:
        provider = provider_class(config)
        if not provider.validate():
            log(f"[provider] Provider '{name}' is not available", file=sys.stderr)
            return None
        return provider
    except Exception as exc:
        log(f"[provider] Error initializing '{name}': {exc}", file=sys.stderr)
        return None


def list_search_providers(config: Optional[Dict[str, Any]] = None) -> Dict[str, bool]:
    """List all search providers and their availability."""

    availability: Dict[str, bool] = {}
    for name, provider_class in _SEARCH_PROVIDERS.items():
        try:
            provider = provider_class(config)
            availability[name] = provider.validate()
        except Exception:
            availability[name] = False
    return availability


_FILE_PROVIDERS: Dict[str, Type[FileProvider]] = {
    "0x0": ZeroXZero,
    "matrix": Matrix,
}


def get_file_provider(name: str, config: Optional[Dict[str, Any]] = None) -> Optional[FileProvider]:
    """Get a file provider by name."""

    provider_class = _FILE_PROVIDERS.get((name or "").lower())
    if provider_class is None:
        log(f"[provider] Unknown file provider: {name}", file=sys.stderr)
        return None

    try:
        provider = provider_class(config)
        if not provider.validate():
            log(f"[provider] File provider '{name}' is not available", file=sys.stderr)
            return None
        return provider
    except Exception as exc:
        log(f"[provider] Error initializing file provider '{name}': {exc}", file=sys.stderr)
        return None


def list_file_providers(config: Optional[Dict[str, Any]] = None) -> Dict[str, bool]:
    """List all file providers and their availability."""

    availability: Dict[str, bool] = {}
    for name, provider_class in _FILE_PROVIDERS.items():
        try:
            provider = provider_class(config)
            availability[name] = provider.validate()
        except Exception:
            availability[name] = False
    return availability


__all__ = [
    "SearchResult",
    "SearchProvider",
    "FileProvider",
    "get_search_provider",
    "list_search_providers",
    "get_file_provider",
    "list_file_providers",
    "download_soulseek_file",
]