Added bbox filter support for collections search

Author: Guillermo Tesoro Calvo

eodag/rest/core.py

@@ -370,6 +370,7 @@ async def all_collections(
     instrument: Optional[str] = None,
     constellation: Optional[str] = None,
     datetime: Optional[str] = None,
+    bbox: Optional[str] = None,
 ) -> dict[str, Any]:
     """Build STAC collections
 
@@ -395,6 +396,7 @@ async def _fetch() -> dict[str, Any]:
             instrument=instrument,
             constellation=constellation,
             datetime=datetime,
+            bbox=bbox,
         )
 
         # # parse f-strings

eodag/rest/server.py

@@ -437,6 +437,7 @@ async def collections(
     request: Request,
     provider: Optional[str] = None,
     q: Optional[str] = None,
+    bbox: Optional[str] = None,
     platform: Optional[str] = None,
     instrument: Optional[str] = None,
     constellation: Optional[str] = None,
@@ -450,8 +451,9 @@ async def collections(
     logger.info(f"{request.method} {request.state.url}")
 
     collections = await all_collections(
-        request, provider, q, platform, instrument, constellation, datetime
+        request, provider, q, platform, instrument, constellation, datetime, bbox
     )
+
     return ORJSONResponse(collections)
 
 

eodag/rest/stac.py

@@ -34,13 +34,15 @@
 
 import geojson
 from jsonpath_ng.jsonpath import Child
+from shapely.geometry import Polygon
 
 from eodag.api.product.metadata_mapping import (
     DEFAULT_METADATA_MAPPING,
     format_metadata,
     get_metadata_path,
 )
 from eodag.rest.config import Settings
+from eodag.rest.types.stac_search import SearchPostRequest
 from eodag.rest.utils.rfc3339 import str_to_interval
 from eodag.utils import (
     deepcopy,
@@ -818,6 +820,7 @@ def get_collection_list(
         instrument: Optional[str] = None,
         constellation: Optional[str] = None,
         datetime: Optional[str] = None,
+        bbox: Optional[str] = None,
     ) -> list[dict[str, Any]]:
         """Build STAC collections list
 
@@ -852,13 +855,26 @@ def get_collection_list(
         else:
             product_types = all_pt
 
+        if bbox:
+            bbox = [float(x) for x in bbox.split(",")]
+            SearchPostRequest.validate_bbox(bbox)
+            bbox = Polygon.from_bounds(*bbox)
+
         # list product types with all metadata using guessed ids
         collection_list: list[dict[str, Any]] = []
         for product_type in product_types:
             stac_collection = self.__generate_stac_collection(
                 collection_model, product_type
             )
-            collection_list.append(stac_collection)
+            # Apply bbox filter
+            if bbox:
+                _bbox = Polygon.from_bounds(
+                    *stac_collection["extent"]["spatial"]["bbox"][0]
+                )
+                if bbox.intersects(_bbox):
+                    collection_list.append(stac_collection)
+            else:
+                collection_list.append(stac_collection)
 
         return collection_list