feat: use obstore (#15)

Author: gadomski

pyproject.toml

@@ -8,6 +8,7 @@ dependencies = [
     "attr>=0.3.2",
     "fastapi>=0.115.8",
     "geojson-pydantic>=1.2.0",
+    "obstore>=0.6.0",
     "pydantic>=2.10.4",
     "pystac>=1.13.0",
     "rustac==0.7.0b3",

src/stac_fastapi/geoparquet/api.py

@@ -1,8 +1,11 @@
 import json
+import urllib.parse
 from collections import defaultdict
 from contextlib import asynccontextmanager
+from pathlib import Path
 from typing import Any, AsyncIterator, TypedDict
 
+import obstore.store
 import pystac.utils
 from fastapi import FastAPI, HTTPException
 from rustac import Collection, DuckdbClient
@@ -91,8 +94,16 @@ def create(
         )
 
     if settings.stac_fastapi_collections_href:
-        with open(settings.stac_fastapi_collections_href, "rb") as f:
-            collections = json.load(f)
+        if urllib.parse.urlparse(settings.stac_fastapi_collections_href).scheme:
+            href = settings.stac_fastapi_collections_href
+        else:
+            href = "file://" + str(
+                Path(settings.stac_fastapi_collections_href).absolute()
+            )
+        prefix, file_name = href.rsplit("/", 1)
+        store = obstore.store.from_url(prefix)
+        result = store.get(file_name)
+        collections = json.loads(bytes(result.bytes()))
     else:
         collections = []