111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
"""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 Provider._base import FileProvider, SearchProvider, SearchResult
|
|
from Provider.bandcamp import Bandcamp
|
|
from Provider.libgen import Libgen
|
|
from Provider.matrix import Matrix
|
|
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,
|
|
"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",
|
|
]
|