Medios-Macina/Store/registry.py

"""Store registry.

Concrete store implementations live in the `Store/` package.
This module is the single source of truth for store discovery.

Config schema (canonical):

{
  "store": {
    "folder": {
      "default": {"path": "C:/Media"},
      "test": {"path": "C:/Temp"}
    },
    "hydrusnetwork": {
      "home": {"Hydrus-Client-API-Access-Key": "...", "url": "http://..."}
    }
  }
}
"""

from __future__ import annotations

from pathlib import Path
from typing import Any, Dict, Optional

from SYS.logger import debug

from Store._base import Store as BaseStore
from Store.Folder import Folder
from Store.HydrusNetwork import HydrusNetwork


class Store:
    def __init__(self, config: Optional[Dict[str, Any]] = None, suppress_debug: bool = False) -> None:
        self._config = config or {}
        self._suppress_debug = suppress_debug
        self._backends: Dict[str, BaseStore] = {}
        self._load_backends()

    def _load_backends(self) -> None:
        store_cfg = self._config.get("store")
        if not isinstance(store_cfg, dict):
            store_cfg = {}

        folder_cfg = store_cfg.get("folder")
        if isinstance(folder_cfg, dict):
            for name, value in folder_cfg.items():
                path_val: Optional[str]
                if isinstance(value, dict):
                    path_val = value.get("path")
                elif isinstance(value, (str, bytes)):
                    path_val = str(value)
                else:
                    path_val = None

                if not path_val:
                    continue

                location = str(Path(str(path_val)).expanduser())
                self._backends[str(name)] = Folder(location=location, name=str(name))

        hydrus_cfg = store_cfg.get("hydrusnetwork")
        if isinstance(hydrus_cfg, dict):
            for instance_name, instance_config in hydrus_cfg.items():
                if not isinstance(instance_config, dict):
                    continue

                api_key = instance_config.get("Hydrus-Client-API-Access-Key")
                url = instance_config.get("url")
                if not api_key or not url:
                    continue

                try:
                    self._backends[str(instance_name)] = HydrusNetwork(
                        instance_name=str(instance_name),
                        api_key=str(api_key),
                        url=str(url),
                    )
                except Exception as exc:
                    if not self._suppress_debug:
                        debug(f"[Store] Failed to register Hydrus instance '{instance_name}': {exc}")

    def list_backends(self) -> list[str]:
        return sorted(self._backends.keys())

    def list_searchable_backends(self) -> list[str]:
        searchable: list[str] = []
        for name, backend in self._backends.items():
            if type(backend).search is not BaseStore.search:
                searchable.append(name)
        return sorted(searchable)

    def __getitem__(self, backend_name: str) -> BaseStore:
        if backend_name not in self._backends:
            raise KeyError(f"Unknown store backend: {backend_name}. Available: {list(self._backends.keys())}")
        return self._backends[backend_name]

    def is_available(self, backend_name: str) -> bool:
        return backend_name in self._backends