feat: Handle TODOs for EC2

Author: m-pizarro

src/services/ec2/connections.ts

@@ -16,6 +16,9 @@ import { RawAwsSubnet } from '../subnet/data'
 import { RawAwsEcsContainer } from '../ecsContainer/data'
 import { RawAwsSystemsManagerInstance } from '../systemsManagerInstance/data'
 import { ssmManagedInstanceArn } from '../../utils/generateArns'
+import { RawAwsElasticBeanstalkEnv } from '../elasticBeanstalkEnvironment/data'
+import { RawAwsEksCluster } from '../eksCluster/data'
+import { getEksClusterName, getElasticBeanstalkEnvId } from './utils'
 
 /**
  * EC2
@@ -44,6 +47,7 @@ export default ({
     SecurityGroups: instanceSecurityGroups = [],
     NetworkInterfaces: instanceNetworkInterfaces = [],
     SubnetId: subnetId,
+    Tags: tags,
   } = instance
 
   /**
@@ -189,17 +193,37 @@ export default ({
           id: subnet.SubnetId,
           resourceType: services.subnet,
           relation: 'child',
-          field: 'subnet',
+          field: 'subnets',
         })
       }
     }
   }
 
   /**
-   * Find EKS
+   * Find EKS cluster
    * related to this EC2
    */
-  // TODO: Implement when eks service is ready
+  const eksClusterName = getEksClusterName(tags)
+  const eksClusters: {
+    name: string
+    data: { [property: string]: any[] }
+  } = data.find(({ name }) => name === services.eksCluster)
+  if (eksClusters?.data?.[region]) {
+    const eksClustersInRegion: RawAwsEksCluster[] = eksClusters.data[
+      region
+    ].filter(({ name }: RawAwsEksCluster) => name === eksClusterName)
+
+    if (!isEmpty(eksClustersInRegion)) {
+      for (const eksCluster of eksClustersInRegion) {
+        connections.push({
+          id: eksCluster.arn,
+          resourceType: services.eksCluster,
+          relation: 'child',
+          field: 'eksCluster',
+        })
+      }
+    }
+  }
 
   /**
    * Find ECS Container
@@ -210,9 +234,9 @@ export default ({
     data: { [property: string]: any[] }
   } = data.find(({ name }) => name === services.ecsContainer)
   if (ecsContainers?.data?.[region]) {
-    const containersInRegion: RawAwsEcsContainer[] = ecsContainers.data[region].filter(
-      ({ ec2InstanceId }) => ec2InstanceId === id
-    )
+    const containersInRegion: RawAwsEcsContainer[] = ecsContainers.data[
+      region
+    ].filter(({ ec2InstanceId }) => ec2InstanceId === id)
 
     if (!isEmpty(containersInRegion)) {
       for (const container of containersInRegion) {
@@ -230,18 +254,24 @@ export default ({
    * Find SSM managed instances
    * related to this EC2 instance
    */
-   const instances: {
+  const instances: {
     name: string
     data: { [property: string]: any[] }
   } = data.find(({ name }) => name === services.systemsManagerInstance)
   if (instances?.data?.[region]) {
-    const dataInRegion: RawAwsSystemsManagerInstance[] = instances.data[region].filter(
+    const dataInRegion: RawAwsSystemsManagerInstance[] = instances.data[
+      region
+    ].filter(
       ({ InstanceId }: RawAwsSystemsManagerInstance) => InstanceId === id
     )
 
     if (!isEmpty(dataInRegion)) {
       for (const ssmInstance of dataInRegion) {
-        const arn = ssmManagedInstanceArn({ region, account, name: ssmInstance.InstanceId })
+        const arn = ssmManagedInstanceArn({
+          region,
+          account,
+          name: ssmInstance.InstanceId,
+        })
         connections.push({
           id: arn,
           resourceType: services.systemsManagerInstance,
@@ -254,9 +284,31 @@ export default ({
 
   /**
    * Find Elastic Beanstalk
-   * related to this EC2 loadbalancer
+   * related to this EC2 instance
    */
-  // TODO: Implement when eb service is ready
+  const elasticBeanstalkEnvId = getElasticBeanstalkEnvId(tags)
+  const elasticBeanstalkEnvs: {
+    name: string
+    data: { [property: string]: any[] }
+  } = data.find(({ name }) => name === services.elasticBeanstalkEnv)
+  if (elasticBeanstalkEnvs?.data?.[region]) {
+    const elasticBeanstalkEnvsInRegion: RawAwsElasticBeanstalkEnv[] =
+    elasticBeanstalkEnvs.data[region].filter(
+        ({ EnvironmentId }: RawAwsElasticBeanstalkEnv) =>
+          elasticBeanstalkEnvId === EnvironmentId
+      )
+
+    if (!isEmpty(elasticBeanstalkEnvsInRegion)) {
+      for (const elasticBeanstalkEnv of elasticBeanstalkEnvsInRegion) {
+        connections.push({
+          id: elasticBeanstalkEnv.EnvironmentId,
+          resourceType: services.elasticBeanstalkEnv,
+          relation: 'child',
+          field: 'elasticBeanstalkEnv',
+        })
+      }
+    }
+  }
 
   const ec2Result = {
     [id]: connections,

src/services/ec2/schema.graphql

@@ -116,8 +116,10 @@ type awsEc2 @key(fields: "arn") {
   eip: [awsEip] @hasInverse(field: ec2Instance)
   networkInterfaces: [awsNetworkInterface] @hasInverse(field: ec2Instance)
   securityGroups: [awsSecurityGroup] @hasInverse(field: ec2Instance)
-  subnet: [awsSubnet] @hasInverse(field: ec2Instance) #change to plural
-  ecsContainer: [awsEcsContainer] @hasInverse(field: ec2Instance) #change to plural
+  subnets: [awsSubnet] @hasInverse(field: ec2Instances)
+  ecsContainer: [awsEcsContainer] @hasInverse(field: ec2Instances)
   emrInstance: [awsEmrInstance] @hasInverse(field: ec2Instance)
   systemsManagerInstance: [awsSystemsManagerInstance] @hasInverse(field: ec2Instance)
+  eksCluster: [awsEksCluster] @hasInverse(field: ec2Instances)
+  elasticBeanstalkEnv: [awsElasticBeanstalkEnv] @hasInverse(field: ec2Instances)
 }

src/services/ec2/utils.ts

@@ -0,0 +1,35 @@
+import { Tag } from 'aws-sdk/clients/ec2'
+import { isEmpty, last } from 'lodash'
+
+const clusterTag = 'kubernetes.io/cluster/'
+const environmentIdTag = 'elasticbeanstalk:environment-id'
+
+export const getEksClusterName = (tags?: Tag[]): string => {
+  if (isEmpty(tags)) {
+    return ''
+  }
+  let eksClusterName = ''
+  Object.keys(tags)?.some(key => {
+    const isMatch = key.includes(clusterTag)
+    if (isMatch) {
+      eksClusterName = last(key.split('/'))
+    }
+    return isMatch
+  })
+  return eksClusterName
+}
+
+export const getElasticBeanstalkEnvId = (tags?: Tag[]): string => {
+  if (isEmpty(tags)) {
+    return ''
+  }
+  let elasticBeanstalkId = ''
+  Object.keys(tags)?.some(key => {
+    const isMatch = key.includes(environmentIdTag)
+    if (isMatch) {
+      elasticBeanstalkId = tags[key]
+    }
+    return isMatch
+  })
+  return elasticBeanstalkId
+}

src/services/ecsContainer/schema.graphql

@@ -18,7 +18,7 @@ type awsEcsContainer @key(fields: "arn") {
   registeredAt: String @search(by: [hash, regexp])
   attachments: [awsEcsAttachment]
   tags: [awsRawTag]
-  ec2Instance: [awsEc2] @hasInverse(field: ecsContainer)
+  ec2Instances: [awsEc2] @hasInverse(field: ecsContainer)
   ecsTask: [awsEcsTask] @hasInverse(field: ecsContainer)
 }
 

src/services/eksCluster/connections.ts

@@ -120,7 +120,7 @@ export default ({
         id: subnet.SubnetId,
         resourceType: services.subnet,
         relation: 'child',
-        field: 'subnet',
+        field: 'subnets',
       })
     }
   }

src/services/eksCluster/schema.graphql

@@ -20,8 +20,9 @@ type awsEksCluster @key(fields: "arn") {
   iamRoles: [awsIamRole] @hasInverse(field: eksCluster)
   kms: [awsKms] @hasInverse(field: eksCluster)
   securityGroups: [awsSecurityGroup] @hasInverse(field: eksCluster)
-  subnet: [awsSubnet] @hasInverse(field: eksCluster) #change to plural
+  subnets: [awsSubnet] @hasInverse(field: eksCluster)
   vpc: [awsVpc] @hasInverse(field: eksCluster)
+  ec2Instances: [awsEc2] @hasInverse(field: eksCluster)
 }
 
 type awsEksVpcConfigResponse {

src/services/elasticBeanstalkEnvironment/schema.graphql

@@ -14,8 +14,9 @@ type awsElasticBeanstalkEnv @key(fields: "arn") {
   solutionStackName: String @search(by: [hash, regexp])
   tier: String @search(by: [hash])
   versionLabel: String @search(by: [hash, regexp])
-  elasticBeanstalkApp: [awsElasticBeanstalkApp] @hasInverse(field: elasticBeanstalkEnv)
   tags: [awsRawTag]
+  elasticBeanstalkApp: [awsElasticBeanstalkApp] @hasInverse(field: elasticBeanstalkEnv)
+  ec2Instances: [awsEc2] @hasInverse(field: elasticBeanstalkEnv)
 }
 
 type awsElasticBeanstalkEnvSetting {

src/services/subnet/schema.graphql

@@ -14,15 +14,15 @@ type awsSubnet @key(fields: "id") {
   tags: [awsRawTag]
   alb: [awsAlb] @hasInverse(field: subnet) #change to plural
   asg: [awsAsg] @hasInverse(field: subnet) #change to plural
-  ec2Instance: [awsEc2] @hasInverse(field: subnet) #change to plural
+  ec2Instances: [awsEc2] @hasInverse(field: subnets) #change to plural
   elb: [awsElb] @hasInverse(field: subnet) #change to plural
   lambda: [awsLambda] @hasInverse(field: subnet) #change to plural
   natGateway: [awsNatGateway] @hasInverse(field: subnet) #change to plural
   networkInterface: [awsNetworkInterface] @hasInverse(field: subnet) #change to plural
   routeTable: [awsRouteTable] @hasInverse(field: subnet)
   vpc: [awsVpc] @hasInverse(field: subnet)
   rdsDbInstance: [awsRdsDbInstance] @hasInverse(field: subnet) #change to plural
-  eksCluster: [awsEksCluster] @hasInverse(field: subnet) #change to plural
+  eksCluster: [awsEksCluster] @hasInverse(field: subnets)
   ecsService: [awsEcsService] @hasInverse(field: subnet) #change to plural
   efsMountTarget: [awsEfsMountTarget] @hasInverse(field: subnet) #change to plural
   flowLogs: [awsFlowLog] @hasInverse(field: subnet) #change to singular

src/types/generated.ts

@@ -1512,6 +1512,8 @@ export type AwsEc2 = {
   ebsOptimized?: Maybe<Scalars['String']>;
   ecsContainer?: Maybe<Array<Maybe<AwsEcsContainer>>>;
   eip?: Maybe<Array<Maybe<AwsEip>>>;
+  eksCluster?: Maybe<Array<Maybe<AwsEksCluster>>>;
+  elasticBeanstalkEnv?: Maybe<Array<Maybe<AwsElasticBeanstalkEnv>>>;
   elasticIps?: Maybe<Scalars['String']>;
   emrInstance?: Maybe<Array<Maybe<AwsEmrInstance>>>;
   ephemeralBlockDevices?: Maybe<Array<Maybe<AwsEc2Blockdevice>>>;
@@ -1540,7 +1542,7 @@ export type AwsEc2 = {
   securityGroupIds?: Maybe<Array<Maybe<Scalars['String']>>>;
   securityGroups?: Maybe<Array<Maybe<AwsSecurityGroup>>>;
   sourceDestCheck?: Maybe<Scalars['String']>;
-  subnet?: Maybe<Array<Maybe<AwsSubnet>>>;
+  subnets?: Maybe<Array<Maybe<AwsSubnet>>>;
   systemsManagerInstance?: Maybe<Array<Maybe<AwsSystemsManagerInstance>>>;
   tags?: Maybe<Array<Maybe<AwsRawTag>>>;
   tenancy?: Maybe<Scalars['String']>;
@@ -1685,8 +1687,8 @@ export type AwsEcsContainer = {
   attachments?: Maybe<Array<Maybe<AwsEcsAttachment>>>;
   attributes?: Maybe<Array<Maybe<AwsEcsAttribute>>>;
   capacityProviderName?: Maybe<Scalars['String']>;
-  ec2Instance?: Maybe<Array<Maybe<AwsEc2>>>;
   ec2InstanceId?: Maybe<Scalars['String']>;
+  ec2Instances?: Maybe<Array<Maybe<AwsEc2>>>;
   ecsTask?: Maybe<Array<Maybe<AwsEcsTask>>>;
   id: Scalars['String'];
   pendingTasksCount?: Maybe<Scalars['Int']>;
@@ -2321,6 +2323,7 @@ export type AwsEksCluster = {
   certificateAuthority?: Maybe<AwsEksCertificate>;
   clientRequestToken?: Maybe<Scalars['String']>;
   createdAt?: Maybe<Scalars['String']>;
+  ec2Instances?: Maybe<Array<Maybe<AwsEc2>>>;
   encryptionConfig?: Maybe<Array<Maybe<AwsEksEncryptionConfig>>>;
   endpoint?: Maybe<Scalars['String']>;
   iamRoles?: Maybe<Array<Maybe<AwsIamRole>>>;
@@ -2335,7 +2338,7 @@ export type AwsEksCluster = {
   resourcesVpcConfig?: Maybe<AwsEksVpcConfigResponse>;
   securityGroups?: Maybe<Array<Maybe<AwsSecurityGroup>>>;
   status?: Maybe<Scalars['String']>;
-  subnet?: Maybe<Array<Maybe<AwsSubnet>>>;
+  subnets?: Maybe<Array<Maybe<AwsSubnet>>>;
   tags?: Maybe<Array<Maybe<AwsRawTag>>>;
   version?: Maybe<Scalars['String']>;
   vpc?: Maybe<Array<Maybe<AwsVpc>>>;
@@ -2594,6 +2597,7 @@ export type AwsElasticBeanstalkEnv = {
   arn: Scalars['String'];
   cname?: Maybe<Scalars['String']>;
   description?: Maybe<Scalars['String']>;
+  ec2Instances?: Maybe<Array<Maybe<AwsEc2>>>;
   elasticBeanstalkApp?: Maybe<Array<Maybe<AwsElasticBeanstalkApp>>>;
   endpointUrl?: Maybe<Scalars['String']>;
   id: Scalars['String'];
@@ -4141,7 +4145,7 @@ export type AwsSubnet = {
   codebuilds?: Maybe<Array<Maybe<AwsCodebuild>>>;
   defaultForAz?: Maybe<Scalars['Boolean']>;
   dmsReplicationInstances?: Maybe<Array<Maybe<AwsDmsReplicationInstance>>>;
-  ec2Instance?: Maybe<Array<Maybe<AwsEc2>>>;
+  ec2Instances?: Maybe<Array<Maybe<AwsEc2>>>;
   ecsService?: Maybe<Array<Maybe<AwsEcsService>>>;
   efsMountTarget?: Maybe<Array<Maybe<AwsEfsMountTarget>>>;
   eksCluster?: Maybe<Array<Maybe<AwsEksCluster>>>;