initial commit

index.js

@@ -0,0 +1,118 @@
+const AWS = require("aws-sdk");
+
+AWS.config.update({
+  region: "us-east-1"
+});
+
+const EC2 = new AWS.EC2({ apiVersion: "2016-11-15" });
+
+const BaseNameAMI = "Base Name Instance Backup"; // <-- Required
+// ^^^ Please keep this field unique because this function will
+// try to delete images once total number of image crossed
+// using basename base name
+const TotalNumberOfImage = 32; // <-- Required Total number of backup should stored
+const InstanceId = "i-xxxxxxxxxxxxxx"; // <-- required EC2 Instance ID
+
+// Create a new AMI
+
+async function createImage() {
+  try {
+    return await EC2.createImage({
+      InstanceId, // Id of instance to create an image
+      Name:
+        BaseNameAMI +
+        " " +
+        new Date().toLocaleDateString("en-us", {
+          month: "short",
+          day: "numeric",
+          timeZone: "UTC"
+        }) +
+        new Date().getTime(), // Name of image
+      NoReboot: true // No-reboot before creating an image
+    }).promise();
+  } catch (err) {
+    throw err;
+  }
+}
+
+/**
+ * This function will delete snapshot once
+ * @param {any} {ImageId}
+ */
+async function deleteImageWithSnapshots({ ImageId }) {
+  try {
+    await EC2.deregisterImage({
+      ImageId
+    }).promise();
+
+    // Fetch list of snapshots
+    let data = await EC2.describeSnapshots({
+      OwnerIds: ["self"]
+    }).promise();
+
+    // Delete snapshots of particular AMI
+    data = data.Snapshots.filter(d => d.Description.indexOf(ImageId) !== -1);
+    console.log(data);
+    for (let i = 0; i < data.length; i++) {
+      try {
+        await EC2.deleteSnapshot({
+          SnapshotId: data[i].SnapshotId
+        }).promise();
+      } catch (err) {
+        console.log("err deleting snaps", err);
+      }
+    }
+
+    console.log(
+      `Deleted image id ${ImageId} at ${new Date().toLocaleDateString("en-us", {
+        month: "short",
+        day: "numeric",
+        year: "numeric",
+        hour: "numeric",
+        minute: "numeric",
+        timeZone: "UTC"
+      })}`
+    );
+  } catch (err) {
+    console.log("error deleteing image", err);
+    throw err;
+  }
+}
+
+// const main = async () => {
+exports.handler = async () => {
+  console.log("triggered");
+  try {
+    // If need image id of created image use this function
+    const { ImageId } = await createImage();
+
+    console.log(
+      `created image id ${ImageId} at ${new Date().toLocaleDateString("en-us", {
+        month: "short",
+        day: "numeric",
+        year: "numeric",
+        hour: "numeric",
+        minute: "numeric",
+        timeZone: "UTC"
+      })}`
+    );
+    // Fetch list of AMI
+    let { Images } = await EC2.describeImages({
+      Owners: ["self"]
+    }).promise();
+
+    Images = Images.filter(d => d.Name.indexOf(BaseNameAMI) !== -1);
+    Images = Images.sort(
+      (first, second) =>
+        new Date(first.CreationDate).getTime() -
+        new Date(second.CreationDate).getTime()
+    );
+    // console.log(Images.map((image) => image.Name))
+    if (Images.length >= TotalNumberOfImage) {
+      console.log(Images[0]);
+      await deleteImageWithSnapshots(Images[0]);
+    }
+  } catch (err) {
+    console.log(err);
+  }
+};

readme.md

@@ -0,0 +1,28 @@
+# AWS AMI Generator
+
+```javascript
+const BaseNameAMI = "Base Name Instance Backup"; // <-- Required
+// ^^^ Please keep this field unique because this function will
+// try to delete images once total number of image crossed
+// using basename base name
+const TotalNumberOfImage = 32; // <-- Required Total number of backup should stored
+const InstanceId = "i-xxxxxxxxxxxxxx"; // <-- required EC2 Instance ID
+```
+
+## Configurations
+
+Copy `index.js` file and paste it in your created lambda function.
+
+Add lambda role with EC2 instance.
+
+This function will automatically delete previously generated images with the same base name.
+
+## Contributing
+
+Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
+
+Please make sure to update tests as appropriate.
+
+## License
+
+[MIT](https://choosealicense.com/licenses/mit/)