@cloudsnorkel/cdk-github-runners 0.3.0 → 0.4.0

package/lib/lambdas/build-image/index.js

@@ -17,7 +17,10 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target, mod));
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/lambdas/build-image/index.ts
@@ -26,17 +29,63 @@ __export(build_image_exports, {
   handler: () => handler
 });
 module.exports = __toCommonJS(build_image_exports);
+var AWS2 = __toESM(require("aws-sdk"));
+
+// src/lambdas/helpers.ts
 var AWS = __toESM(require("aws-sdk"));
-var codebuild = new AWS.CodeBuild();
-var ecr = new AWS.ECR();
+var sm = new AWS.SecretsManager();
+async function customResourceRespond(event, responseStatus, reason, physicalResourceId, data) {
+  const responseBody = JSON.stringify({
+    Status: responseStatus,
+    Reason: reason,
+    PhysicalResourceId: physicalResourceId,
+    StackId: event.StackId,
+    RequestId: event.RequestId,
+    LogicalResourceId: event.LogicalResourceId,
+    NoEcho: false,
+    Data: data
+  });
+  console.log("Responding", responseBody);
+  const parsedUrl = require("url").parse(event.ResponseURL);
+  const requestOptions = {
+    hostname: parsedUrl.hostname,
+    path: parsedUrl.path,
+    method: "PUT",
+    headers: {
+      "content-type": "",
+      "content-length": responseBody.length
+    }
+  };
+  return new Promise((resolve, reject) => {
+    try {
+      const request = require("https").request(requestOptions, resolve);
+      request.on("error", reject);
+      request.write(responseBody);
+      request.end();
+    } catch (e) {
+      reject(e);
+    }
+  });
+}
+
+// src/lambdas/build-image/index.ts
+var codebuild = new AWS2.CodeBuild();
+var ecr = new AWS2.ECR();
+var ib = new AWS2.Imagebuilder();
 async function handler(event, context) {
   try {
-    console.log(JSON.stringify(event));
+    console.log(JSON.stringify({ ...event, ResponseURL: "..." }));
+    const deleteOnly = event.ResourceProperties.DeleteOnly;
     const repoName = event.ResourceProperties.RepoName;
     const projectName = event.ResourceProperties.ProjectName;
+    const ibName = event.ResourceProperties.ImageBuilderName;
     switch (event.RequestType) {
       case "Create":
       case "Update":
+        if (deleteOnly) {
+          await customResourceRespond(event, "SUCCESS", "OK", "Deleter", {});
+          break;
+        }
         console.log(`Starting CodeBuild project ${projectName}`);
         await codebuild.startBuild({
           projectName,
@@ -65,54 +114,39 @@ async function handler(event, context) {
         }).promise();
         break;
       case "Delete":
-        const images = await ecr.listImages({ repositoryName: repoName, maxResults: 100 }).promise();
-        if (images.imageIds && images.imageIds.length > 0) {
+        const ecrImages = await ecr.listImages({ repositoryName: repoName, maxResults: 100 }).promise();
+        if (ecrImages.imageIds && ecrImages.imageIds.length > 0) {
           await ecr.batchDeleteImage({
-            imageIds: images.imageIds.map((i) => {
+            imageIds: ecrImages.imageIds.map((i) => {
               return { imageDigest: i.imageDigest };
             }),
             repositoryName: repoName
           }).promise();
         }
-        await respond("SUCCESS", "OK", event.PhysicalResourceId, {});
+        if (ibName) {
+          const ibImages = await ib.listImages({ filters: [{ name: "name", values: [ibName] }] }).promise();
+          if (ibImages.imageVersionList) {
+            for (const v of ibImages.imageVersionList) {
+              if (v.arn) {
+                const ibImageVersions = await ib.listImageBuildVersions({ imageVersionArn: v.arn }).promise();
+                if (ibImageVersions.imageSummaryList) {
+                  for (const vs of ibImageVersions.imageSummaryList) {
+                    if (vs.arn) {
+                      console.log(`Deleting ${vs.arn}`);
+                      await ib.deleteImage({ imageBuildVersionArn: vs.arn }).promise();
+                    }
+                  }
+                }
+              }
+            }
+          }
+        }
+        await customResourceRespond(event, "SUCCESS", "OK", event.PhysicalResourceId, {});
         break;
     }
   } catch (e) {
-    console.log(e);
-    await respond("FAILED", e.message || "Internal Error", context.logStreamName, {});
-  }
-  function respond(responseStatus, reason, physicalResourceId, data) {
-    const responseBody = JSON.stringify({
-      Status: responseStatus,
-      Reason: reason,
-      PhysicalResourceId: physicalResourceId,
-      StackId: event.StackId,
-      RequestId: event.RequestId,
-      LogicalResourceId: event.LogicalResourceId,
-      NoEcho: false,
-      Data: data
-    });
-    console.log("Responding", responseBody);
-    const parsedUrl = require("url").parse(event.ResponseURL);
-    const requestOptions = {
-      hostname: parsedUrl.hostname,
-      path: parsedUrl.path,
-      method: "PUT",
-      headers: {
-        "content-type": "",
-        "content-length": responseBody.length
-      }
-    };
-    return new Promise((resolve, reject) => {
-      try {
-        const request = require("https").request(requestOptions, resolve);
-        request.on("error", reject);
-        request.write(responseBody);
-        request.end();
-      } catch (e) {
-        reject(e);
-      }
-    });
+    console.error(e);
+    await customResourceRespond(event, "FAILED", e.message || "Internal Error", context.logStreamName, {});
   }
 }
 // Annotate the CommonJS export names for ESM import in node: