Merge pull request #4 from isovalent/pr/bkrasko/update-webhook-lambda-to-set-region-from-sqs-url

feat: Update webhook lambda to set the region from sqs url

Author: bogdankrasko

lambdas/functions/webhook/src/sqs/index.ts

@@ -4,6 +4,8 @@ import { createChildLogger, getTracedAWSV3Client } from '@aws-github-runner/aws-
 
 const logger = createChildLogger('sqs');
 
+const sqsClientsByRegion = new Map<string, SQS>();
+
 export interface ActionRequestMessage {
   id: number;
   eventType: string;
@@ -32,7 +34,8 @@ export interface GithubWorkflowEvent {
 }
 
 export const sendActionRequest = async (message: ActionRequestMessage): Promise<void> => {
-  const sqs = getTracedAWSV3Client(new SQS({ region: process.env.AWS_REGION }));
+  const region = getRegionFromQueueUrl(message.queueId) ?? process.env.AWS_REGION;
+  const sqs = getSqsClient(region);
 
   const sqsMessage: SendMessageCommandInput = {
     QueueUrl: message.queueId,
@@ -43,3 +46,30 @@ export const sendActionRequest = async (message: ActionRequestMessage): Promise<
 
   await sqs.sendMessage(sqsMessage);
 };
+
+function getSqsClient(region: string | undefined): SQS {
+  if (!region) {
+    return getTracedAWSV3Client(new SQS({}));
+  }
+
+  const cached = sqsClientsByRegion.get(region);
+  if (cached) {
+    return cached;
+  }
+
+  const client = getTracedAWSV3Client(new SQS({ region }));
+  sqsClientsByRegion.set(region, client);
+  return client;
+}
+
+function getRegionFromQueueUrl(queueUrl: string): string | undefined {
+  try {
+    const url = new URL(queueUrl);
+    const parts = url.hostname.split('.');
+    if (parts.length >= 3 && parts[0] === 'sqs') {
+      return parts[1];
+    }
+  } catch {}
+
+  return undefined;
+}