(feature) add aws traffic mirroring to cli (#55)

Author: akshay288

cli/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metlo/cli",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "license": "MIT",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -16,10 +16,13 @@
   "devDependencies": {
     "@types/axios": "^0.14.0",
     "@types/node": "^18.7.16",
+    "@types/uuid": "^8.3.4",
     "prettier": "^2.7.1",
     "typescript": "^4.7.4"
   },
   "dependencies": {
+    "@aws-sdk/client-ec2": "^3.198.0",
+    "@aws-sdk/client-sts": "^3.198.0",
     "@metlo/testing": "^0.0.3",
     "@types/valid-url": "^1.0.3",
     "axios": "^0.27.2",

cli/src/aws/constants.ts

@@ -0,0 +1,123 @@
+export const AWS_REGIONS = [
+  "us-west-1",
+  "us-west-2",
+  "us-east-1",
+  "us-east-2",
+  "eu-central-1",
+  "eu-west-1",
+  "eu-west-2",
+  "eu-west-3",
+  "eu-north-1",
+]
+
+export const SUPPORTED_AWS_INSTANCES = [
+  "a1",
+  "c5",
+  "c5a",
+  "c5ad",
+  "c5d",
+  "c5n",
+  "c6a",
+  "c6g",
+  "c6gd",
+  "c6gn",
+  "c6i",
+  "c6id",
+  "d3",
+  "d3en",
+  "dl1",
+  "g4",
+  "g4ad",
+  "g5",
+  "g5g",
+  "hpc6a",
+  "i3en",
+  "i4i",
+  "im4gn",
+  "inf1",
+  "is4gen",
+  "m5",
+  "m5a",
+  "m5ad",
+  "m5d",
+  "m5dn",
+  "m5n",
+  "m5zn",
+  "m6a",
+  "m6g",
+  "m6gd",
+  "m6i",
+  "m6id",
+  "p3dn.24xlarge",
+  "p4",
+  "r5",
+  "r5a",
+  "r5ad",
+  "r5b",
+  "r5d",
+  "r5dn",
+  "r5n",
+  "r6a",
+  "r6g",
+  "r6gd",
+  "r6i",
+  "r6id",
+  "t3",
+  "t3a",
+  "t4g",
+  "u-6tb1.56xlarge",
+  "u-6tb1.112xlarge",
+  "u-9tb1.112xlarge",
+  "u-12tb1.112xlarge",
+  "vt1",
+  "x2gd",
+  "x2idn",
+  "x2iedn",
+  "x2iezn",
+  "z1d",
+  "a1.metal",
+  "c5.metal",
+  "c5d.metal",
+  "c5n.metal",
+  "c6a.metal",
+  "c6g.metal",
+  "c6gd.metal",
+  "c6i.metal",
+  "c6id.metal",
+  "g4dn.metal",
+  "g5g.metal",
+  "i3.metal",
+  "i3en.metal",
+  "i4i.metal",
+  "m5.metal",
+  "m5d.metal",
+  "m5dn.metal",
+  "m5n.metal",
+  "m5zn.metal",
+  "m6a.metal",
+  "m6g.metal",
+  "m6gd.metal",
+  "m6i.metal",
+  "m6id.metal",
+  "mac1.metal",
+  "r5.metal",
+  "r5b.metal",
+  "r5d.metal",
+  "r5dn.metal",
+  "r5n.metal",
+  "r6a.metal",
+  "r6g.metal",
+  "r6gd.metal",
+  "r6i.metal",
+  "r6id.metal",
+  "u-6tb1.metal",
+  "u-9tb1.metal",
+  "u-12tb1.metal",
+  "u-18tb1.metal",
+  "u-24tb1.metal",
+  "x2gd.metal",
+  "x2idn.metal",
+  "x2iedn.metal",
+  "x2iezn.metal",
+  "z1d.metal",
+]

cli/src/aws/ec2Utils.ts

@@ -0,0 +1,120 @@
+import {
+  EC2Client,
+  DescribeInstanceTypesCommand,
+  DescribeInstanceTypesCommandInput,
+  DescribeInstancesCommand,
+  VirtualizationType,
+  DescribeNetworkInterfacesCommand,
+  InstanceTypeInfoFromInstanceRequirements,
+  GetInstanceTypesFromInstanceRequirementsCommand,
+  GetInstanceTypesFromInstanceRequirementsCommandInput,
+} from "@aws-sdk/client-ec2"
+import { SUPPORTED_AWS_INSTANCES } from "./constants"
+import { MachineSpecifications } from "./types"
+
+export class EC2_CONN {
+  private access_id: string
+  private secret_key: string
+  private region?: string
+  private conn: EC2Client
+  constructor(access_id: string, secret_key: string, region?: string) {
+    this.access_id = access_id
+    this.secret_key = secret_key
+    this.region = region
+  }
+
+  public get_conn() {
+    if (this.conn) {
+      return this.conn
+    }
+    this.conn = new EC2Client({
+      credentials: {
+        accessKeyId: this.access_id,
+        secretAccessKey: this.secret_key,
+      },
+      region: this.region,
+    })
+    return this.conn
+  }
+
+  public disconnect() {
+    if (this.conn) this.conn.destroy()
+  }
+
+  public async describe_instance(ec2_instance_id) {
+    let resp = await this.get_conn().send(
+      new DescribeInstancesCommand({ InstanceIds: [ec2_instance_id] }),
+    )
+    return resp
+  }
+
+  public async describe_interface(ec2_interface_id) {
+    let resp = await this.get_conn().send(
+      new DescribeNetworkInterfacesCommand({
+        NetworkInterfaceIds: [ec2_interface_id],
+      }),
+    )
+    return resp
+  }
+
+  public async describe_type(Instance_type: string) {
+    let command = new DescribeInstanceTypesCommand({
+      InstanceTypes: [Instance_type],
+    } as DescribeInstanceTypesCommandInput)
+    let resp = await this.get_conn().send(command)
+    return resp.InstanceTypes[0]
+  }
+
+  public async get_valid_types(
+    specs?: MachineSpecifications,
+    vtx_type?: VirtualizationType,
+  ): Promise<Array<InstanceTypeInfoFromInstanceRequirements>> {
+    let command = new GetInstanceTypesFromInstanceRequirementsCommand({
+      ArchitectureTypes: ["x86_64"],
+      VirtualizationTypes: vtx_type ? [vtx_type] : ["hvm", "paravirtual"],
+      InstanceRequirements: {
+        VCpuCount: specs
+          ? { Min: specs.minCpu, Max: specs.maxCpu }
+          : { Min: 0 },
+        MemoryMiB: specs
+          ? {
+              Min: specs.minMem * 1024,
+              Max: specs.maxMem ? specs.maxMem * 1024 : null,
+            }
+          : { Min: 0 },
+        InstanceGenerations: ["current"],
+        BurstablePerformance: "included",
+        BareMetal: "included",
+        ExcludedInstanceTypes: [
+          "t2.nano",
+          "t2.micro",
+          "t2.small",
+          "t2.medium",
+          "t2.large",
+          "t2.xlarge",
+          "t2.2xlarge",
+          "c3.large",
+          "c3.xlarge",
+          "c3.2xlarge",
+          "c3.4xlarge",
+          "c3.8xlarge",
+          "r3.large",
+          "r3.xlarge",
+          "r3.2xlarge",
+          "r3.4xlarge",
+          "r3.8xlarge",
+          "i3.xlarge",
+          "i3.2xlarge",
+          "i3.4xlarge",
+          "i3.8xlarge",
+        ],
+      },
+    } as GetInstanceTypesFromInstanceRequirementsCommandInput)
+    let conn = this.get_conn()
+    let resp = await conn.send(command)
+    return resp.InstanceTypes.filter(x => {
+      let a = SUPPORTED_AWS_INSTANCES.filter(y => x.InstanceType.includes(y))
+      return a.length > 0
+    })
+  }
+}