feat: allow inclusion of image copy to pipeline

Support both codepipeline and pipelines

.projenrc.ts

@@ -36,6 +36,7 @@ const project = new CdklabsConstructLibrary({
   repositoryUrl: 'https://github.com/cdklabs/cdk-ecr-deployment', /* The repository is the location where the actual code for your package lives. */
   gitignore: [
     'cdk.out/',
+    'lambda/out',
   ], /* Additional entries to .gitignore. */
   npmignore: [
     '/cdk.out',

src/ecr-deployment-step.ts

@@ -0,0 +1,31 @@
+import { aws_lambda as lambda } from 'aws-cdk-lib';
+import { Construct } from 'constructs';
+import { addFunctionPermissions, getFunctionProps, LambdaInvokeStep } from './lambda';
+import { ECRDeploymentStepProps } from './types';
+
+export class ECRDeploymentStep extends Construct {
+
+  private handler: lambda.Function;
+
+  constructor(scope: Construct, id: string, props: ECRDeploymentStepProps) {
+    super(scope, id);
+
+    this.handler = new lambda.Function(this, 'ImageCopyHandler', {
+      ...getFunctionProps(props),
+      environment: {
+        ...props.environment,
+        INVOKER: 'CODEPIPELINE',
+      },
+    });
+
+    const handlerRole = this.handler.role;
+    if (!handlerRole) { throw new Error('lambda.Function should have created a Role'); }
+
+    addFunctionPermissions(handlerRole);
+
+    const factory = new LambdaInvokeStep(this.handler, props, 'ImageCopy');
+
+    props.stage?.addAction(factory.getAction());
+    props.wave?.addPost(factory);
+  }
+}