fix(url-shortener): incorrect bucket name in redirect edge function

src/url-shortener/redirect.edge-lambda.ts

@@ -10,12 +10,16 @@ export async function handler(event: AWSLambda.CloudFrontRequestEvent): Promise<
       throw new Error('No S3 origin');
     }
 
-    const s3Client = new S3Client({ region: s3Origin.region });
-    const bucket = s3Origin.domainName.replace(new RegExp(`.s3.${s3Origin.region}.amazonaws.com$`), '');
+    const s3details = extractS3Details(s3Origin.domainName);
+    if (!s3details) {
+      throw new Error('No S3 details');
+    }
+
+    const s3Client = new S3Client({ region: s3details.region });
     const key = request.uri.substring(1); // remove first slash
 
     const data = await s3Client.send(new GetObjectCommand({
-      Bucket: bucket,
+      Bucket: s3details.bucket,
       Key: key,
     }));
 
@@ -44,3 +48,18 @@ export async function handler(event: AWSLambda.CloudFrontRequestEvent): Promise<
     };
   }
 }
+
+function extractS3Details(hostname: string): { bucket: string; region: string } | undefined {
+  const regex = /^(.*)\.s3\.(.*)\.amazonaws\.com$/;
+
+  const match = hostname.match(regex);
+
+  if (match && match.length === 3) {
+    return {
+      bucket: match[1],
+      region: match[2],
+    };
+  }
+
+  return undefined;
+}

test/url-shortener/url-shortener.integ.snapshot/edge-lambda-stack-c8e731c8ad0787291628b399d525c90dc78319b7d7.assets.json

@@ -1,31 +1,31 @@
 {
   "version": "48.0.0",
   "files": {
-    "ef50eada816754194800ebd82b6d5459ee1af84a5fbdfcb142b289e893b6fa07": {
+    "415cf1c4e7a6cbb293e3e195a88b9531b51ee2565f7943b3287e812512f382db": {
       "displayName": "Redirect/Code",
       "source": {
-        "path": "asset.ef50eada816754194800ebd82b6d5459ee1af84a5fbdfcb142b289e893b6fa07.edge-lambda",
+        "path": "asset.415cf1c4e7a6cbb293e3e195a88b9531b51ee2565f7943b3287e812512f382db.edge-lambda",
         "packaging": "zip"
       },
       "destinations": {
-        "current_account-us-east-1-18e1a286": {
+        "current_account-us-east-1-0f2eb817": {
           "bucketName": "cdk-hnb659fds-assets-${AWS::AccountId}-us-east-1",
-          "objectKey": "ef50eada816754194800ebd82b6d5459ee1af84a5fbdfcb142b289e893b6fa07.zip",
+          "objectKey": "415cf1c4e7a6cbb293e3e195a88b9531b51ee2565f7943b3287e812512f382db.zip",
           "region": "us-east-1",
           "assumeRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-file-publishing-role-${AWS::AccountId}-us-east-1"
         }
       }
     },
-    "d2ab57caeb7f8091f5291b4d42cc8154899de3d50abfa7174654a1b11d1bbf5c": {
+    "7b4c6d501f56517e1d218aac356518731867e5a91f375b925ca4dacff77d47cc": {
       "displayName": "edge-lambda-stack-c8e731c8ad0787291628b399d525c90dc78319b7d7 Template",
       "source": {
         "path": "edge-lambda-stack-c8e731c8ad0787291628b399d525c90dc78319b7d7.template.json",
         "packaging": "file"
       },
       "destinations": {
-        "current_account-us-east-1-53ca69e4": {
+        "current_account-us-east-1-ff77a5bf": {
           "bucketName": "cdk-hnb659fds-assets-${AWS::AccountId}-us-east-1",
-          "objectKey": "d2ab57caeb7f8091f5291b4d42cc8154899de3d50abfa7174654a1b11d1bbf5c.json",
+          "objectKey": "7b4c6d501f56517e1d218aac356518731867e5a91f375b925ca4dacff77d47cc.json",
           "region": "us-east-1",
           "assumeRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-file-publishing-role-${AWS::AccountId}-us-east-1"
         }

test/url-shortener/url-shortener.integ.snapshot/edge-lambda-stack-c8e731c8ad0787291628b399d525c90dc78319b7d7.template.json

@@ -95,7 +95,7 @@
      "S3Bucket": {
       "Fn::Sub": "cdk-hnb659fds-assets-${AWS::AccountId}-us-east-1"
      },
-     "S3Key": "ef50eada816754194800ebd82b6d5459ee1af84a5fbdfcb142b289e893b6fa07.zip"
+     "S3Key": "415cf1c4e7a6cbb293e3e195a88b9531b51ee2565f7943b3287e812512f382db.zip"
     },
     "Description": "src/url-shortener/redirect.edge-lambda.ts",
     "Handler": "index.handler",
@@ -112,7 +112,7 @@
     "RedirectServiceRole39B6C513"
    ]
   },
-  "RedirectCurrentVersion479E25ECda849a4c7ccf6efe08941f8fce999820": {
+  "RedirectCurrentVersion479E25EC8eb57aa85fd449a985b41f5b6e8c9d03": {
    "Type": "AWS::Lambda::Version",
    "Properties": {
     "FunctionName": {
@@ -129,7 +129,7 @@
     "Name": "/cdk/EdgeFunctionArn/eu-west-1/url-shortener-integ/UrlShortener/Redirect",
     "Type": "String",
     "Value": {
-     "Ref": "RedirectCurrentVersion479E25ECda849a4c7ccf6efe08941f8fce999820"
+     "Ref": "RedirectCurrentVersion479E25EC8eb57aa85fd449a985b41f5b6e8c9d03"
     }
    }
   }

test/url-shortener/url-shortener.integ.snapshot/url-shortener-integ.assets.json

@@ -46,16 +46,16 @@
         }
       }
     },
-    "d67dc00290eadaf204923d9d7ca97a66ebddf4724436fb483da8f7261978db7c": {
+    "daeaa425429d57768650b4c16a36a8ff56720490e6bf5aaf484ea16a84edefa1": {
       "displayName": "url-shortener-integ Template",
       "source": {
         "path": "url-shortener-integ.template.json",
         "packaging": "file"
       },
       "destinations": {
-        "current_account-eu-west-1-fd32efc5": {
+        "current_account-eu-west-1-ab46e594": {
           "bucketName": "cdk-hnb659fds-assets-${AWS::AccountId}-eu-west-1",
-          "objectKey": "d67dc00290eadaf204923d9d7ca97a66ebddf4724436fb483da8f7261978db7c.json",
+          "objectKey": "daeaa425429d57768650b4c16a36a8ff56720490e6bf5aaf484ea16a84edefa1.json",
           "region": "eu-west-1",
           "assumeRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-file-publishing-role-${AWS::AccountId}-eu-west-1"
         }

test/url-shortener/url-shortener.integ.snapshot/url-shortener-integ.template.json

@@ -160,7 +160,7 @@
     },
     "Region": "us-east-1",
     "ParameterName": "/cdk/EdgeFunctionArn/eu-west-1/url-shortener-integ/UrlShortener/Redirect",
-    "RefreshToken": "RedirectCurrentVersion479E25ECda849a4c7ccf6efe08941f8fce999820"
+    "RefreshToken": "RedirectCurrentVersion479E25EC8eb57aa85fd449a985b41f5b6e8c9d03"
    },
    "UpdateReplacePolicy": "Delete",
    "DeletionPolicy": "Delete"