fix search self link mediatype (#217)

Author: vincentsarago

CHANGES.md

@@ -2,6 +2,12 @@
 
 ## [Unreleased]
 
+## [5.0.1] - 2025-03-27
+
+### Fixed
+
+- fix media type for `self` links in `/search` responses
+
 ## [5.0.0] - 2025-03-10
 
 ### Changed

VERSION

@@ -1 +1 @@
-5.0.0
+5.0.1

pyproject.toml

@@ -29,7 +29,7 @@ ignore = [
 
 
 [tool.bumpversion]
-current_version = "5.0.0"
+current_version = "5.0.1"
 parse = """(?x)
     (?P<major>\\d+)\\.
     (?P<minor>\\d+)\\.

stac_fastapi/pgstac/core.py

@@ -28,6 +28,7 @@
     ItemCollectionLinks,
     ItemLinks,
     PagingLinks,
+    SearchLinks,
 )
 from stac_fastapi.pgstac.types.search import PgstacSearch
 from stac_fastapi.pgstac.utils import filter_fields
@@ -466,6 +467,11 @@ async def post_search(
             if fields.include or fields.exclude:
                 return JSONResponse(item_collection)  # type: ignore
 
+        links = await SearchLinks(request=request).get_links(
+            extra_links=item_collection["links"]
+        )
+        item_collection["links"] = links
+
         return ItemCollection(**item_collection)
 
     async def get_search(
@@ -522,6 +528,11 @@ async def get_search(
 
         item_collection = await self._search_base(search_request, request=request)
 
+        links = await SearchLinks(request=request).get_links(
+            extra_links=item_collection["links"]
+        )
+        item_collection["links"] = links
+
         # If we have the `fields` extension enabled
         # we need to avoid Pydantic validation because the
         # Items might not be a valid STAC Item objects

stac_fastapi/pgstac/models/links.py

@@ -266,6 +266,19 @@ def link_items(self) -> Dict:
         }
 
 
+@attr.s
+class SearchLinks(BaseLinks):
+    """Create inferred links specific to collections."""
+
+    def link_self(self) -> Dict:
+        """Return the self link."""
+        return {
+            "rel": Relations.self.value,
+            "type": MimeTypes.geojson.value,
+            "href": self.resolve("search"),
+        }
+
+
 @attr.s
 class ItemCollectionLinks(CollectionLinksBase):
     """Create inferred links specific to collections."""

stac_fastapi/pgstac/version.py

@@ -1,3 +1,3 @@
 """library version."""
 
-__version__ = "5.0.0"
+__version__ = "5.0.1"

tests/resources/test_item.py

@@ -1662,3 +1662,22 @@ async def test_get_filter_extension(app_client, load_test_data, load_test_collec
     fc = resp.json()
     assert len(fc["features"]) == 1
     assert fc["features"][0]["id"] == search_id
+
+
+async def test_get_search_link_media(app_client):
+    """Test Search request returned links"""
+    # GET
+    resp = await app_client.get("/search")
+    assert resp.status_code == 200
+    links = resp.json()["links"]
+    assert len(links) == 2
+    get_self_link = next((link for link in links if link["rel"] == "self"), None)
+    assert get_self_link["type"] == "application/geo+json"
+
+    # POST
+    resp = await app_client.post("/search", json={})
+    assert resp.status_code == 200
+    links = resp.json()["links"]
+    assert len(links) == 2
+    get_self_link = next((link for link in links if link["rel"] == "self"), None)
+    assert get_self_link["type"] == "application/geo+json"