feat: add sources

src/source.ts

@@ -1,6 +1,7 @@
 //import * as fs from 'fs';
 //import * as path from 'path';
 import { Fn } from 'aws-cdk-lib';
+import * as ecr from 'aws-cdk-lib/aws-ecr';
 import * as ecr_assets from 'aws-cdk-lib/aws-ecr-assets';
 import * as iam from 'aws-cdk-lib/aws-iam';
 import { Construct } from 'constructs';
@@ -52,6 +53,14 @@ export abstract class Source {
     return new DirectorySource(path);
   }
 
+  public static ecr(repository: ecr.IRepository, tag: string): Source {
+    return new EcrSource(repository, tag);
+  }
+
+  public static asset(asset: ecr_assets.DockerImageAsset): Source { // | ecr_assets.TarballImageAsset): Source {
+    return new AssetSource(asset);
+  }
+
   /**
    * bind grants the CodeBuild role permissions to pull from a repository if necessary
    * bind should be invoked by the caller to get the SourceConfig
@@ -85,3 +94,45 @@ class DirectorySource extends Source {
     };
   }
 }
+
+class EcrSource extends Source {
+  private repository: ecr.IRepository;
+  private tag: string;
+
+  constructor(repository: ecr.IRepository, tag: string) {
+    super();
+    this.repository = repository;
+    this.tag = tag;
+  }
+
+  public bind(_scope: Construct, context: SourceContext): SourceConfig {
+
+    this.repository.grantPull(context.handlerRole);
+
+    return {
+      imageUri: this.repository.repositoryUriForTag(this.tag),
+      imageTag: this.tag,
+    };
+  }
+}
+
+class AssetSource extends Source {
+  private repository: ecr.IRepository;
+  private tag: string;
+
+  constructor(asset: ecr_assets.DockerImageAsset) {
+    super();
+    this.repository = asset.repository;
+    this.tag = Fn.select(1, Fn.split(':', asset.imageUri));
+  }
+
+  public bind(_scope: Construct, context: SourceContext): SourceConfig {
+
+    this.repository.grantPull(context.handlerRole);
+
+    return {
+      imageUri: this.repository.repositoryUriForTag(this.tag),
+      imageTag: this.tag,
+    };
+  }
+}

test/docker-image-deploy.test.ts

@@ -2,6 +2,7 @@
 import * as path from 'path';
 import { Match, Template } from 'aws-cdk-lib/assertions';
 import * as ecr from 'aws-cdk-lib/aws-ecr';
+import * as ecr_assets from 'aws-cdk-lib/aws-ecr-assets';
 import * as cdk from 'aws-cdk-lib/core';
 import * as imagedeploy from '../src';
 
@@ -308,6 +309,49 @@ describe('DockerImageDeploy', () => {
     });
   });
 
+  describe('Source: ecr', () => {
+    // GIVEN
+    const sourceRepo = ecr.Repository.fromRepositoryName(stack, 'TestRepo', 'testrepo');
+    const ecrSource = imagedeploy.Source.ecr(sourceRepo, 'testtag');
+
+    describe('Destination: ecr', () => {
+      // GIVEN
+      const destinationRepo = new ecr.Repository(stack, 'TestDestinationRepository1');
+      const testDesination = imagedeploy.Destination.ecr(destinationRepo, { tag: 'testtag' });
+
+      // WHEN
+      test('dockerimagedeploy construct', () => {
+        new imagedeploy.DockerImageDeployment(stack, 'AssetSourceTestDeployment', {
+          source: ecrSource,
+          destination: testDesination,
+        });
+      });
+    });
+
+  });
+
+  describe('Source: local asset', () => {
+    // GIVEN
+    const asset = new ecr_assets.DockerImageAsset(stack, 'TestAsset', {
+      directory: path.join(__dirname, 'assets/test1'),
+    });
+    const assetSource = imagedeploy.Source.asset(asset);
+
+    describe('Destination: ecr', () => {
+      // GIVEN
+      const destinationRepo = new ecr.Repository(stack, 'TestDestinationRepository2');
+      const testDesination = imagedeploy.Destination.ecr(destinationRepo, { tag: 'testtag' });
+
+      // WHEN
+      test('dockerimagedeploy construct', () => {
+        new imagedeploy.DockerImageDeployment(stack, 'EcrSourceTestDeployment', {
+          source: assetSource,
+          destination: testDesination,
+        });
+      });
+    });
+  });
+
   describe('Custom Resrouces', () => {
     test('onEventHandler has correct permissions', () => {
       Template.fromStack(stack).hasResourceProperties('AWS::IAM::Policy', {