chore: additional testing, refactor

fedifetcher/api/__init__.py

@@ -1,13 +1,14 @@
 """__init__.py for api."""
 
-from . import api
+from .api import API, FederationInterface
 from .firefish import api_firefish, api_firefish_types
 from .lemmy import api_lemmy
 from .mastodon import api_mastodon, api_mastodon_types
 from .postgresql import postgresql
 
 __all__ = [
-    "api",
+    "API",
+    "FederationInterface",
     "api_firefish",
     "api_firefish_types",
     "api_lemmy",

fedifetcher/api/api.py

@@ -29,3 +29,29 @@ def get_ids_from_list(self, uris: list[str]) -> Coroutine[Any, Any, dict[str, st
     def get_context(self, uri: str) -> Coroutine[Any, Any, list[str]]:
         """Get the context of an object by URI."""
         raise NotImplementedError
+
+class FederationInterface:
+    """Interface for dependency injection of different federation APIs."""
+
+    _equipped_api: API
+
+    def __init__(self, equippable_api: API) -> None:
+        """Initialize the API."""
+        self._equipped_api: API = equippable_api
+
+    def get(self, uri: str) -> Coroutine[Any, Any, dict | bool]:
+        """Get an object by URI."""
+        return self._equipped_api.get(uri)
+
+    def get_id(self, uri: str) -> Coroutine[Any, Any, str | None]:
+        """Get the ID of an object by URI."""
+        return self._equipped_api.get_id(uri)
+
+    def get_ids_from_list(self, uris: list[str]) -> Coroutine[Any, Any, dict[str, str]]:
+        """Get the IDs of objects by URIs."""
+        return self._equipped_api.get_ids_from_list(uris)
+
+    def get_context(self, uri: str) -> Coroutine[Any, Any, list[str]]:
+        """Get the context of an object by URI."""
+        return self._equipped_api.get_context(uri)
+