feat: add architecture option to generate layers for ARM64

Author: rubenfonseca

.projen/deps.json

@@ -4,7 +4,7 @@ const project = new awscdk.AwsCdkConstructLibrary({
   authorUrl: 'https://aws.amazon.com',
   authorOrganization: true,
   keywords: ['aws', 'cdk', 'powertools', 'python', 'layer', 'lambda', 'devax', 'typescript', 'nodejs'],
-  cdkVersion: '2.24.1',
+  cdkVersion: '2.44.0',
   defaultReleaseBranch: 'main',
   majorVersion: 2,
   name: 'cdk-aws-lambda-powertools-layer',

.projenrc.js

@@ -1,5 +1,6 @@
 import * as path from 'path';
 import { aws_lambda as lambda } from 'aws-cdk-lib';
+import { Architecture } from 'aws-cdk-lib/aws-lambda';
 import { Construct } from 'constructs';
 
 /**
@@ -26,6 +27,11 @@ export interface PowertoolsLayerProps {
    * the runtime of the layer
    */
   readonly runtimeFamily?: lambda.RuntimeFamily;
+
+  /**
+   * The compatible architectures for the layer
+   */
+  readonly compatibleArchitectures?: lambda.Architecture[];
 }
 
 /**
@@ -70,6 +76,8 @@ export class LambdaPowertoolsLayer extends lambda.LayerVersion {
     const runtimeFamily = props?.runtimeFamily ?? lambda.RuntimeFamily.PYTHON;
     const languageName = getLanguageNameFromRuntimeFamily(runtimeFamily);
     const dockerFilePath = path.join(__dirname, `../layer/${languageName}`);
+    const compatibleArchitectures = props?.compatibleArchitectures ?? [lambda.Architecture.X86_64];
+
     console.log(`path ${dockerFilePath}`);
     super(scope, id, {
       code: lambda.Code.fromDockerBuild(dockerFilePath, {
@@ -80,10 +88,12 @@ export class LambdaPowertoolsLayer extends lambda.LayerVersion {
             props?.version,
           ),
         },
+        platform: getPlatformNameFromArchitectures(compatibleArchitectures),
       }),
       layerVersionName: props?.layerVersionName ? props?.layerVersionName : undefined,
       license: 'MIT-0',
       compatibleRuntimes: getRuntimesFromRuntimeFamily(runtimeFamily),
+      compatibleArchitectures,
       description: `Lambda Powertools for ${languageName}${
         props?.includeExtras ? ' with Extras' : ''
       } ${props?.version ? `version ${props?.version}` : 'latest version'}`.trim(),
@@ -120,3 +130,14 @@ function getLanguageNameFromRuntimeFamily(runtimeFamily: lambda.RuntimeFamily):
       return 'Unknown';
   }
 }
+
+function getPlatformNameFromArchitectures(architectures: lambda.Architecture[]): string {
+  if (architectures.length == 1) {
+    return architectures[0].dockerPlatform;
+  } else {
+    // if we have multiple architectures, we default to x86_64, hoping for the
+    // layer not to have any architecture specific code or at least contain
+    // binary code for all architectures
+    return Architecture.X86_64.dockerPlatform;
+  }
+}

package.json

@@ -1,6 +1,6 @@
 import { Stack } from 'aws-cdk-lib';
 import { Template } from 'aws-cdk-lib/assertions';
-import { RuntimeFamily } from 'aws-cdk-lib/aws-lambda';
+import { Architecture, RuntimeFamily } from 'aws-cdk-lib/aws-lambda';
 import { LambdaPowertoolsLayer } from '../src';
 
 
@@ -31,6 +31,21 @@ describe('with no configuration the construct', () => {
   });
 });
 
+describe('with arm64 architecture', () => {
+  const stack = new Stack();
+  new LambdaPowertoolsLayer(stack, 'PowertoolsLayer', {
+    runtimeFamily: RuntimeFamily.PYTHON,
+    compatibleArchitectures: [Architecture.ARM_64],
+  });
+  const template = Template.fromStack(stack);
+  test('synthesizes successfully', () => {
+    template.hasResourceProperties('AWS::Lambda::LayerVersion', {
+      Description: 'Lambda Powertools for Python latest version',
+      CompatibleArchitectures: ['arm64'],
+    });
+  });
+});
+
 describe('for layerVersionName configuraiton the construct', () => {
   test('synthisizes to a layer with provided name', () => {
     const stack = new Stack();