Skip to content

Commit 08dfde1

Browse files
committed
bugfix: handle feature collection view
1 parent 0c1c9c8 commit 08dfde1

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

src/stac_auth_proxy/middleware/AuthenticationExtensionMiddleware.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,17 @@ def transform_json(self, doc: dict[str, Any]) -> dict[str, Any]:
125125
# ---
126126
# Annotate assets with "auth:refs": [signing_scheme]
127127
if self.signing_endpoint:
128-
for asset in doc.get("assets", {}).values():
128+
assets = chain(
129+
# Item
130+
doc.get("assets", {}).values(),
131+
# Items/Search
132+
(
133+
asset
134+
for item in doc.get("features", [])
135+
for asset in item.get("assets", {}).values()
136+
),
137+
)
138+
for asset in assets:
129139
if "href" not in asset:
130140
logger.warning("Asset %s has no href", asset)
131141
continue
@@ -134,7 +144,9 @@ def transform_json(self, doc: dict[str, Any]) -> dict[str, Any]:
134144

135145
# Annotate links with "auth:refs": [auth_scheme]
136146
links = chain(
147+
# Item/Collection
137148
doc.get("links", []),
149+
# Collections/Items/Search
138150
(
139151
link
140152
for prop in ["features", "collections"]

0 commit comments

Comments
 (0)