feat(update): automatic function update config (optional) (#1)

* [ADD] auto update feature

* Update action.yml

* [ADD] find functions with all layer versions

* [ADD] find functions with all layer versions

* [FIX] indentation

* [FIX] indentation

* [FIX] update README.md

Author: ktx-mega

README.md

@@ -56,6 +56,7 @@ Replace the placeholders with the appropriate values:
 - `runtimes`: The list of valid runtimes that this layer can support. The valid values are `["nodejs" || "nodejs4.3" || "nodejs6.10" || "nodejs8.10" || "nodejs10.x" || "nodejs12.x" || "nodejs14.x" || "nodejs16.x" || "nodejs18.x"|| "nodejs20.x" || "java8" || "java8.al2" || "java11" || "python2.7" || "python3.6" || "python3.7" || "python3.8" || "python3.9" || "dotnetcore1.0" || "dotnetcore2.0" || "dotnetcore2.1" || "dotnetcore3.1" || "dotnet6" || "nodejs4.3-edge" || "go1.x" || "ruby2.5" || "ruby2.7" || "provided" || "provided.al2" || "nodejs18.x" || "python3.10" || "java17" || "ruby3.2" || "python3.11"]`<br/>
 - `s3_bucket`: An optional parameter for S3 Bucket Name In case if layer exceeds the allocated threshold size. We recommend to always provide with this value since if this value is present then action will try to upload layer via S3 automatically as failsafe mechanism for larger layers. Overcoming the limitation of direct AWS Lambda Layer size constraints over SDK.
 - `functions`: An Optional array of functions to refresh the layer to the latest uploaded version. It is recommended to pass down all the functions in this argument that uses this layer. Action will try to refresh all the dependent functions to use the latest layer version automatically.
+- `auto_update`: An optional boolean to automatically find and update all lambdas using the layer. With this option you don't have to manually specify the `functions` parameter. Use with caution as it might also facilitate the propagation of faulty layers. Disabled by default.
 
 This GitHub Action is distributed under the MIT License. See LICENSE for more information.
 

action.yml

@@ -32,9 +32,12 @@ inputs:
     description: "Optional Lambda function name(s) to update the layer to latest. String[]"
     required: false
     default: '[]'
+  auto_update:
+    description: "Optional Automatically update all Lambda functions using the layer to the latest layer version. Boolean"
+    required: false
 runs:
   using: 'node16'
   main: 'index.js'
 branding:
   icon: 'archive'
-  color: 'gray-dark'
+  color: 'gray-dark'

helper.js

@@ -1,5 +1,5 @@
 const { readFileSync, statSync } = require('fs');
-const { LambdaClient, PublishLayerVersionCommand, UpdateFunctionConfigurationCommand } = require('@aws-sdk/client-lambda');
+const { LambdaClient, PublishLayerVersionCommand, UpdateFunctionConfigurationCommand, ListFunctionsCommand } = require('@aws-sdk/client-lambda');
 const { S3Client, PutObjectCommand, DeleteObjectCommand } = require('@aws-sdk/client-s3');
 
 const non_error_response_codes = [200, 201, 204];
@@ -142,7 +142,38 @@ exports.refreshLambdaLayerVersion = async ({
 			FunctionName: functionName,
 			Layers: [layerARN]
 		})));
-
 	const response = await Promise.all(commands);
 	return response;
+}
+
+/**
+ * List all the lambda functions that use the specified layer
+ */
+exports.listLambdaFunctionsWithLayer = async ({
+	region,
+	accessKeyId,
+	secretAccessKey,
+	layerARN
+}) => {
+	try {
+		const client = lambdaClient({ region, accessKeyId, secretAccessKey });
+		
+		const allFunctions = [];
+		let nextMarker = null;
+		do {
+			const listFunctionsCommand = new ListFunctionsCommand({ Marker: nextMarker });
+			const { Functions: functions, NextMarker: nextPageMarker } = await client.send(listFunctionsCommand);
+			
+			allFunctions.push(...functions);
+			nextMarker = nextPageMarker;
+		} while (nextMarker);
+		
+		const layerARNWithoutVersion = layerARN.split(':').slice(0, -1).join(':')
+		const matchinFunctions = allFunctions.filter((func) => func.Layers && func.Layers.some((layer) => layer.Arn.includes(layerARNWithoutVersion)))
+		const functionNames = matchinFunctions.map((func) => func.FunctionName);
+		return functionNames;
+	} catch (error) {
+		console.error("Error:", error);
+		return [];
+	}
 }

index.js

@@ -5,7 +5,8 @@ const {
 	publishS3LayerArchive,
 	getArchiveSize,
 	deleteTemporaryArchiveFromS3,
-	refreshLambdaLayerVersion
+	refreshLambdaLayerVersion,
+	listLambdaFunctionsWithLayer
 } = require('./helper');
 
 
@@ -20,10 +21,11 @@ const {
 		const architectures = getInput('architectures') ? JSON.parse(getInput('architectures')) : [];
 		const s3Bucket = getInput('s3_bucket');
 		const functionNames = JSON.parse(getInput('functions'));
-
+		const autoUpdate = getInput('auto_update') || false;
+		
 		console.log(runtimes, architectures);
 		const creds = { region, accessKeyId, secretAccessKey };
-
+		
 		const size = getArchiveSize(archive);
 		console.log(`Archive size is ${size}`);
 		let layerResponse;
@@ -75,6 +77,20 @@ const {
 				layerARN: layerResponse.LayerVersionArn
 			});
 		}
+		if (autoUpdate) {
+			// find automatically all matching functions
+			const foundFunctionNames = await listLambdaFunctionsWithLayer({
+				...creds,
+				layerARN: layerResponse.LayerVersionArn,
+			})
+			console.log('The following functions will be updated :', foundFunctionNames)
+			// trigger functions update
+			await refreshLambdaLayerVersion({
+				...creds,
+				functionNames: foundFunctionNames,
+				layerARN: layerResponse.LayerVersionArn
+			});
+		}
 	} catch (err) {
 		setFailed(err.message);
 	}