Expose LambdaDeployment.alias (#28)

Author: 72636c

packages/infra/README.md

@@ -34,15 +34,16 @@ In a CDK stack:
 + import { LambdaDeployment } from '@seek/aws-codedeploy-infra';
 import { aws_lambda_nodejs } from 'aws-cdk-lib';
 
-const lambdaFunction = new aws_lambda_nodejs.NodejsFunction(
-  this,
-  'MyLambdaFunction',
-  {
-    // ...
-  },
-);
-
-+ new LambdaDeployment(this, 'MyLambdaDeployment', { lambdaFunction });
+const lambdaFunction = new aws_lambda_nodejs.NodejsFunction(this, 'Function', {
+  // ...
+});
+
++ const lambdaDeployment = new LambdaDeployment(this, 'Deployment', {
++   lambdaFunction,
++ });
+
+- lambdaFunction.addEventSource(source);
++ lambdaDeployment.alias.addEventSource(source);
 ```
 
 The `LambdaDeployment` construct creates a CodeDeploy application and deployment group for your Lambda function,

packages/infra/src/constructs/__snapshots__/lambdaDeployment.test.ts.snap

@@ -401,8 +401,36 @@ exports[`returns expected CloudFormation stack 1`] = `
         },
       },
     },
+    "LambdaFunctionAliasLiveSqsEventSourceQueue453DF2EA": {
+      "Properties": {
+        "EventSourceArn": "arn:aws:sqs:us-east-2:123456789012:queue",
+        "FunctionName": {
+          "Fn::Join": [
+            "",
+            [
+              {
+                "Fn::Select": [
+                  6,
+                  {
+                    "Fn::Split": [
+                      ":",
+                      {
+                        "Ref": "LambdaFunctionAliasLive400E15E9",
+                      },
+                    ],
+                  },
+                ],
+              },
+              ":Live",
+            ],
+          ],
+        },
+      },
+      "Type": "AWS::Lambda::EventSourceMapping",
+    },
     "LambdaFunctionBF21E41F": {
       "DependsOn": [
+        "LambdaFunctionServiceRoleDefaultPolicy32EEEE35",
         "LambdaFunctionServiceRoleC555A460",
       ],
       "Properties": {
@@ -471,6 +499,33 @@ exports[`returns expected CloudFormation stack 1`] = `
       },
       "Type": "AWS::IAM::Role",
     },
+    "LambdaFunctionServiceRoleDefaultPolicy32EEEE35": {
+      "Properties": {
+        "PolicyDocument": {
+          "Statement": [
+            {
+              "Action": [
+                "sqs:ReceiveMessage",
+                "sqs:ChangeMessageVisibility",
+                "sqs:GetQueueUrl",
+                "sqs:DeleteMessage",
+                "sqs:GetQueueAttributes",
+              ],
+              "Effect": "Allow",
+              "Resource": "arn:aws:sqs:us-east-2:123456789012:queue",
+            },
+          ],
+          "Version": "2012-10-17",
+        },
+        "PolicyName": "LambdaFunctionServiceRoleDefaultPolicy32EEEE35",
+        "Roles": [
+          {
+            "Ref": "LambdaFunctionServiceRoleC555A460",
+          },
+        ],
+      },
+      "Type": "AWS::IAM::Policy",
+    },
   },
   "Rules": {
     "CheckBootstrapVersion": {

packages/infra/src/constructs/lambdaDeployment.test.ts

@@ -1,4 +1,11 @@
-import { App, Stack, assertions, aws_lambda } from 'aws-cdk-lib';
+import {
+  App,
+  Stack,
+  assertions,
+  aws_lambda,
+  aws_lambda_event_sources,
+  aws_sqs,
+} from 'aws-cdk-lib';
 
 import { LambdaDeployment } from './lambdaDeployment';
 
@@ -20,8 +27,17 @@ it('returns expected CloudFormation stack', () => {
     runtime: aws_lambda.Runtime.NODEJS_20_X,
   });
 
-  // eslint-disable-next-line no-new
-  new LambdaDeployment(stack, null, { lambdaFunction });
+  const deployment = new LambdaDeployment(stack, null, { lambdaFunction });
+
+  const queue = aws_sqs.Queue.fromQueueArn(
+    stack,
+    'Queue',
+    'arn:aws:sqs:us-east-2:123456789012:queue',
+  );
+
+  const eventSource = new aws_lambda_event_sources.SqsEventSource(queue);
+
+  deployment.alias.addEventSource(eventSource);
 
   const template = assertions.Template.fromStack(stack);
 

packages/infra/src/constructs/lambdaDeployment.ts

@@ -10,6 +10,8 @@ export type LambdaDeploymentProps = {
 };
 
 export class LambdaDeployment extends Construct {
+  alias: Readonly<aws_lambda.Alias>;
+
   constructor(
     scope: Construct,
     id: string | null,
@@ -23,6 +25,8 @@ export class LambdaDeployment extends Construct {
       description: 'The Lambda version currently receiving traffic',
     });
 
+    this.alias = alias;
+
     const application = new aws_codedeploy.LambdaApplication(
       this,
       'CodeDeployLambdaApplication',