import json import urllib.error import urllib.parse import urllib.request from envfile import env_value DEFAULT_API_URL = "http://localhost:5000" IMAGE_PATH_PREFIX = "/api/library/images/" REQUEST_TIMEOUT_SECONDS = 60 class LibraryError(ValueError): pass def api_url() -> str: return (env_value("LIBRARY_API_URL") or DEFAULT_API_URL).rstrip("/") def api_key() -> str: key = env_value("LIBRARY_API_KEY") if not key: raise LibraryError("Set LIBRARY_API_KEY in .env to change the library") return key def error_message(error: urllib.error.HTTPError) -> str: try: return json.loads(error.read().decode("utf-8")).get("message") or error.reason except ValueError: return str(error.reason) def send(method: str, path: str, body: dict | None = None, signed: bool = True): headers = {"Accept": "application/json"} data = None if signed: headers["Authorization"] = f"Bearer {api_key()}" if body is not None: headers["Content-Type"] = "application/json" data = json.dumps(body).encode("utf-8") request = urllib.request.Request(f"{api_url()}{path}", data=data, headers=headers, method=method) try: with urllib.request.urlopen(request, timeout=REQUEST_TIMEOUT_SECONDS) as response: text = response.read().decode("utf-8") except urllib.error.HTTPError as error: raise LibraryError(f"{api_url()} said {error.code}: {error_message(error)}") from error except urllib.error.URLError as error: raise LibraryError(f"Could not reach {api_url()}: {error.reason}") from error return json.loads(text) if text else None def quoted(identifier: str) -> str: return urllib.parse.quote(identifier, safe="") def fetch_library() -> dict: return send("GET", "/api/library", signed=False) def save_media(imdb_id: str, fields: dict) -> dict: return send("PUT", f"/api/library/media/{quoted(imdb_id)}", fields) def delete_media(imdb_id: str) -> None: send("DELETE", f"/api/library/media/{quoted(imdb_id)}") def save_book(goodreads_id: str, fields: dict) -> dict: return send("PUT", f"/api/library/books/{quoted(goodreads_id)}", fields) def delete_book(goodreads_id: str) -> None: send("DELETE", f"/api/library/books/{quoted(goodreads_id)}") def save_game(igdb_id: str, fields: dict) -> dict: return send("PUT", f"/api/library/games/{quoted(igdb_id)}", fields) def delete_game(igdb_id: str) -> None: send("DELETE", f"/api/library/games/{quoted(igdb_id)}") def is_stored_image(path: str | None) -> bool: return bool(path) and path.startswith(IMAGE_PATH_PREFIX)