Return signed url as JSON object

Author: alukach

src/stac_auth_proxy/handlers/s3_asset_signer.py

@@ -3,6 +3,7 @@
 import logging
 import re
 from dataclasses import dataclass
+from typing import Literal
 
 import boto3
 from botocore.exceptions import ClientError
@@ -16,17 +17,20 @@ class S3AssetSigner:
     bucket_pattern: str = r".*"
     max_expiration: int = 3600
 
-    def endpoint(self, payload: "S3AssetSignerPayload", expiration: int = 3600) -> str:
+    def endpoint(
+        self, payload: "S3AssetSignerPayload", expiration: int = 3600
+    ) -> dict[Literal["signed_url"], str]:
         """Generate a presigned URL to share an S3 object."""
         if not re.match(self.bucket_pattern, payload.bucket_name):
             return HTTPException(status_code=404, detail="Item not found")
 
         try:
-            return boto3.client("s3").generate_presigned_url(
+            url = boto3.client("s3").generate_presigned_url(
                 "get_object",
                 Params={"Bucket": payload.bucket_name, "Key": payload.object_name},
                 ExpiresIn=min(expiration, self.max_expiration),
             )
+            return {"signed_url": url}
         except ClientError as e:
             logging.error(e)
             return HTTPException(status_code=500, detail="Internal server error")