feat: Add support for a regex parameter \n\n BREAKING CHANGE: Update repo_mapping variable to support the regex attribute

Author: fabidick22

src/call_receiver/app.py

@@ -2,6 +2,7 @@
 import boto3
 import requests
 import os
+import re
 
 secretsmanager = boto3.client('secretsmanager')
 TOKEN_SECRET_NAME = os.environ['FLUX2_WEBHOOK_TOKEN_SECRET_NAME']
@@ -54,7 +55,8 @@ def make_requests(webhook_url, repository, headers):
             'repository': repository
         }))
 
-def call_flux_webhook(repository):
+
+def call_flux_webhook(repository, image_tag):
     # Retrieve the map of values from Secrets Manager
     webhook_map = get_webhook_map()
 
@@ -63,13 +65,14 @@ def call_flux_webhook(repository):
     token = None
     if repository in webhook_map:
         repo_data = webhook_map[repository]
-        webhook_urls = repo_data.get('webhook')
-        token = repo_data.get('token', get_global_token())
-        for webhook in webhook_urls:
-            headers = {'Authorization': f'Bearer {token}'}
-            make_requests(webhook, repository, headers)
-
-
+        for key, data in repo_data.items():
+            webhook_urls = data.get('webhook')
+            token = data.get('token', get_global_token())
+            regex = data.get('regex', '.*')
+            for webhook in webhook_urls:
+                headers = {'Authorization': f'Bearer {token}'}
+                if regex and re.match(regex, image_tag):
+                    make_requests(webhook, repository, headers)
 
 
 def lambda_handler(event, context):
@@ -82,7 +85,7 @@ def lambda_handler(event, context):
     process_ecr_push_event(detail)
 
     # Call the Flux webhook with the event repository
-    call_flux_webhook(detail['repository-name'])
+    call_flux_webhook(detail['repository-name'], detail['image-tag'])
 
     return {
         'statusCode': 200,