chore(init): initial commit

Author: ops-gaurav

.gitignore

@@ -0,0 +1,2 @@
+## AWS Lambda Layer Release
+Your one stop AWS Lambda Layer Release Function.

README.md

@@ -0,0 +1,36 @@
+name: 'AWS Lambda Layer Release'
+description: 'Action lets you release your latest lambda layer version'
+inputs:
+  access_key_id:  # id of input
+    description: 'AWS_ACCESS_KEY_ID. String'
+    required: true
+  secret_access_key:
+    description: 'AWS_SECRET_ACCESS_KEY. String'
+    required: true
+  layer_name:
+    description: "AWS lambda layer name. String"
+    required: true
+  archive:
+    description: "Path to lambda layer zip archive. String"
+    required: true
+  region:
+    description: "The AWS region for this lambda and layer. Defaults to ap-south-1 (Mumbai). String"
+    required: true
+    default: "ap-south-1"
+  runtimes:
+    description: 'List of runtimes. "nodejs" || "nodejs4.3" || "nodejs6.10" || "nodejs8.10" || "nodejs10.x" || "nodejs12.x" || "nodejs14.x" || "nodejs16.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". String[]'
+    required: false
+  architectures:
+    description: 'List of architectures. "x86_64" || "arm64". String[]'
+    required: false
+    default: '["x86_64", "arm64"]'
+  s3_bucket:
+    description: 's3 bucket is required if layer exceeds 10MB size. String'
+    required: false
+  functions:
+    description: "Optional Lambda function name(s) to update the layer to latest. String[]"
+    required: false
+    default: '[]'
+runs:
+  using: 'node16'
+  main: 'index.js'

action.yml

@@ -0,0 +1,153 @@
+const { readFileSync, statSync } = require('fs');
+const { LambdaClient, PublishLayerVersionCommand, UpdateFunctionConfigurationCommand } = require('@aws-sdk/client-lambda');
+const { S3Client, PutObjectCommand, DeleteObjectCommand } = require('@aws-sdk/client-s3');
+
+const non_error_response_codes = [200, 201, 204];
+
+/**
+ * Reusable function to initiate the lambda client
+ * @param {*} param0
+ * @returns
+ */
+const lambdaClient = ({ region, accessKeyId, secretAccessKey }) => new LambdaClient({
+	region,
+	credentials: { accessKeyId, secretAccessKey }
+});
+/**
+ * Reusable function to initiate the s3 client
+ * @param {*} param0
+ * @returns
+ */
+const s3Client = ({ region, accessKeyId, secretAccessKey }) => new S3Client({
+	region,
+	credentials: { accessKeyId, secretAccessKey }
+})
+
+const errored = response => non_error_response_codes.indexOf(response.$metadata.httpStatusCode) == -1 ? true : false;
+
+exports.getArchiveSize = archive => statSync(archive).size
+/**
+ * Resuable function to publish lambda layer. This function dynamically identifies the config
+ * based on whether to pick archive from file or S3
+ * @param {*} param0
+ */
+exports.publishLambdaLayer = async ({
+	region,
+	accessKeyId,
+	secretAccessKey,
+	layerName,
+	archive,
+	architectures,
+	runtimes,
+	s3Bucket = null
+}) => {
+	/**
+	 * Initiate the lambda client
+	 */
+	const client = lambdaClient({ region, accessKeyId, secretAccessKey });
+	const payload = {
+		LayerName: layerName,
+		Description: "",
+		/**
+		 * If s3Bucket is defined in the input Parameters then value S3 uploaded layer.
+		 */
+		Content: s3Bucket ? {
+			S3Bucket: s3Bucket,
+			S3Key: `${layerName}.zip`
+		} : {
+			/**
+			 * Direct parse layer only if s3Bucket is null in params
+			 */
+			ZipFile: readFileSync(archive)
+		},
+		CompatibleArchitectures: architectures,
+		CompatibleRuntimes: runtimes
+	}
+	const command = new PublishLayerVersionCommand(payload);
+	const response = await client.send(command);
+	if (errored(response)) {
+		console.log(JSON.stringify(response));
+		throw new Error("Error While publishing layer. If you feel this is a bug, raise a ticket on the repo.");
+	}
+	console.log("Success Uploading Layer!");
+	return response;
+};
+/**
+ * Reusable function that will publish layer to S3 just in case of layer size is higher than
+ * the expected size. The code will identify whether this archive can be directly uploaded and if not,
+ * It should be uploaded to S3 first using this function and then layer parsing is done by dynamoc Content config
+ * in @function publishLambdaLayer
+ * @param {*} param0
+ */
+exports.publishS3LayerArchive = async ({
+	region,
+	accessKeyId,
+	secretAccessKey,
+	s3Bucket,
+	layerName,
+	archive
+}) => {
+	const client = s3Client({ region, accessKeyId, secretAccessKey });
+	const command = new PutObjectCommand({
+		Bucket: s3Bucket,
+		Key: `${layerName}.zip`,
+		Body: readFileSync(archive)
+	});
+	const response = await client.send(command);
+	if (errored(response)) {
+		console.log(JSON.stringify(response));
+		throw new Error("Error While publishing layer to S3. If you feel this is a bug, raise a ticket on the repo.");
+	}
+	console.log("Success Uoloading Layer to S3!");
+	return response;
+}
+/**
+ * Reusable function to cleanup the temporary created layer archive on S3 to reduce the accumulating charges over time.
+ * @param {*} param0
+ */
+exports.deleteTemporaryArchiveFromS3 = async ({
+	region,
+	accessKeyId,
+	secretAccessKey,
+	s3Bucket,
+	s3Key
+}) => {
+	const client = s3Client({ region, accessKeyId, secretAccessKey });
+	const command = new DeleteObjectCommand({
+		Bucket: s3Bucket,
+		Key: s3Key
+	});
+	const response = await client.send(command);
+	if (errored(response)) {
+		console.log(JSON.stringify(response));
+		throw new Error("Error While Deleting Layer From S3. If you feel this is a bug, raise a ticket on the repo.");
+	}
+	console.log("Success Deleting Layer From S3!");
+	return response;
+}
+/**
+ * Refresh lambda function to use the latest version of layer
+ */
+exports.refreshLambdaLayerVersion = async ({
+	region,
+	accessKeyId,
+	secretAccessKey,
+	functionNames,
+	layerARN,
+}) => {
+	const client = lambdaClient({ region, accessKeyId, secretAccessKey });
+	const commands = []
+	for (const functionName of functionNames)
+		commands.push(client.send(new UpdateFunctionConfigurationCommand({
+			FunctionName: functionName,
+			Layers: [layerARN]
+		})));
+
+	const response = await Promise.all(commands);
+	if (errored(response)) {
+		console.log(JSON.stringify(response));
+		throw new Error("Error While Refreshing Lambda For Latest Layer. If you feel this is a bug, raise a ticket on the repo.");
+	}
+	console.log("Refreshing Lambda For Latest Layer.");
+	return response;
+}

helper.js

@@ -0,0 +1,81 @@
+const { getInput, setFailed } = require('@actions/core');
+
+const {
+	publishLambdaLayer,
+	publishS3LayerArchive,
+	getArchiveSize,
+	deleteTemporaryArchiveFromS3,
+	refreshLambdaLayerVersion
+} = require('./helper');
+
+
+(async () => {
+	try {
+		const accessKeyId = getInput('access_key_id');
+		const secretAccessKey = getInput('secret_access_key');
+		const layerName = getInput('layer_name');
+		const archive = getInput('archive');
+		const region = getInput('region');
+		const runtimes = getInput('runtimes') ? JSON.parse(getInput('runtimes')) : [];
+		const architectures = getInput('architectures') ? JSON.parse(getInput('architectures')) : [];
+		const s3Bucket = getInput('s3_bucket');
+		const functionNames = JSON.parse(getInput('functions'));
+
+		console.log(runtimes, architectures);
+		const creds = { region, accessKeyId, secretAccessKey };
+
+		const size = getArchiveSize(archive);
+		console.log(`Archive size is ${size}`);
+		let layerResponse;
+		if (size > 10000000) {
+			if (!s3Bucket) {
+				setFailed('Param s3_bucket is required if layer size exceeds 10MB.');
+			}
+			// upload s3 archive
+			await publishS3LayerArchive({
+				...creds,
+				s3Bucket,
+				archive,
+				layerName
+			});
+			layerResponse = await publishLambdaLayer({
+				...creds,
+				archive,
+				s3Bucket,
+				layerName,
+				architectures,
+				runtimes,
+			});
+			/**
+			 * try to remove the archive from s3 bucket
+			 *
+			 */
+			await deleteTemporaryArchiveFromS3({
+				...creds,
+				s3Bucket,
+				s3Key: `${layerName}.zip`
+			});
+			/**
+			 * refresh the lambda functions to use the latest layer
+			 */
+		} else {
+			layerResponse = await publishLambdaLayer({
+				...creds,
+				archive,
+				layerName,
+				architectures,
+				runtimes,
+			});
+		}
+		if (functionNames.length) {
+			// trigger functions update
+			await refreshLambdaLayerVersion({
+				...creds,
+				functionNames,
+				layerARN: layerResponse.LayerVersionArn
+			});
+		}
+	} catch (err) {
+		setFailed(err.message);
+	}
+})();