feat: Add transit gateway routes

Author: m-pizarro

src/services/routeTable/schema.graphql

@@ -7,8 +7,6 @@ type awsRouteTable implements awsBaseService @key(fields: "arn") {
   tags: [awsRawTag]
   subnet: [awsSubnet] @hasInverse(field: routeTable) #change to plural
   vpc: [awsVpc] @hasInverse(field: routeTables)
-  transitGateway: [awsTransitGateway] @hasInverse(field: routeTable)
-  transitGatewayAttachment: [awsTransitGatewayAttachment] @hasInverse(field: routeTable)
   vpcEndpoints: [awsVpcEndpoint] @hasInverse(field: routeTables)
 }
 

src/services/transitGatewayRouteTable/data.ts

@@ -5,6 +5,9 @@ import isEmpty from 'lodash/isEmpty'
 import EC2, {
   DescribeTransitGatewayRouteTablesRequest,
   DescribeTransitGatewayRouteTablesResult,
+  SearchTransitGatewayRoutesRequest,
+  SearchTransitGatewayRoutesResult,
+  TransitGatewayRoute,
   TransitGatewayRouteTable,
 } from 'aws-sdk/clients/ec2'
 
@@ -83,13 +86,51 @@ const listTransitGatewayRouteTablesData = async ({
     listTransitGatewayRouteTables()
   })
 
+const searchTransitGatewayRoutes = async ({
+  ec2,
+  transitGatewayRouteTableId,
+}: {
+  ec2: EC2
+  transitGatewayRouteTableId: string
+}): Promise<TransitGatewayRoute[]> =>
+  new Promise<TransitGatewayRoute[]>(resolve => {
+    const args: SearchTransitGatewayRoutesRequest = {
+      TransitGatewayRouteTableId: transitGatewayRouteTableId,
+      Filters: [
+        {
+          Name: 'type',
+          Values: ['propagated', 'static'],
+        },
+      ],
+    }
+    ec2.searchTransitGatewayRoutes(
+      args,
+      (err: AWSError, data: SearchTransitGatewayRoutesResult) => {
+        if (err) {
+          errorLog.generateAwsErrorLog({
+            functionName: 'ec2:searchTransitGatewayRoutes',
+            err,
+          })
+        }
+
+        if (!isEmpty(data)) {
+          const { Routes: routes = [] } = data
+          resolve(routes)
+        }
+
+        resolve([])
+      }
+    )
+  })
+
 /**
  * Transit Gateway Route Table
  */
 export interface RawAwsTransitGatewayRouteTable
   extends Omit<TransitGatewayRouteTable, 'Tags'> {
   region: string
   Tags?: TagMap
+  Routes?: TransitGatewayRoute[]
 }
 
 export default async ({
@@ -103,6 +144,7 @@ export default async ({
 }> =>
   new Promise(async resolve => {
     const transitGatewayRouteTableResult: RawAwsTransitGatewayRouteTable[] = []
+    const routesPromises = []
 
     const regionPromises = regions.split(',').map(region => {
       const ec2 = new EC2({ ...config, region, endpoint })
@@ -130,7 +172,30 @@ export default async ({
     })
 
     await Promise.all(regionPromises)
-    errorLog.reset()
 
+    transitGatewayRouteTableResult.map(
+      (
+        { TransitGatewayRouteTableId: transitGatewayRouteTableId, region },
+        idx
+      ) => {
+        const ec2 = new EC2({ ...config, region, endpoint })
+        const routePromise = new Promise<void>(async resolveRoute => {
+          const routes: TransitGatewayRoute[] =
+            await searchTransitGatewayRoutes({
+              ec2,
+              transitGatewayRouteTableId,
+            })
+          transitGatewayRouteTableResult[idx] = {
+            ...transitGatewayRouteTableResult[idx],
+            Routes: routes || [],
+          }
+          resolveRoute()
+        })
+        routesPromises.push(routePromise)
+      }
+    )
+    await Promise.all(routesPromises)
+
+    errorLog.reset()
     resolve(groupBy(transitGatewayRouteTableResult, 'region'))
   })

src/services/transitGatewayRouteTable/format.ts

@@ -1,3 +1,4 @@
+import cuid from 'cuid'
 import { formatTagsFromMap } from '../../utils/format'
 import { RawAwsTransitGatewayRouteTable } from './data'
 import { AwsTransitGatewayRouteTable } from '../../types/generated'
@@ -23,6 +24,7 @@ export default ({
     DefaultPropagationRouteTable: defaultPropagationRouteTable,
     CreationTime: creationTime,
     Tags: tags,
+    Routes: routes = [],
   } = rawData
 
   const transitGatewayRouteTable = {
@@ -35,6 +37,14 @@ export default ({
     defaultPropagationRouteTable,
     creationTime: creationTime?.toISOString(),
     tags: formatTagsFromMap(tags),
+    routes:
+      routes?.map(r => ({
+        id: cuid(),
+        destinationCidrBlock: r.DestinationCidrBlock,
+        type: r.Type,
+        state: r.State,
+        prefixListId: r.PrefixListId,
+      })) || [],
   }
 
   return transitGatewayRouteTable

src/services/transitGatewayRouteTable/schema.graphql

@@ -1,8 +1,22 @@
+type awsTransitGatewayRoute
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  ) {
+  id: String! @id
+  destinationCidrBlock: String @search(by: [hash, regexp])
+  type: String @search(by: [hash, regexp])
+  state: String @search(by: [hash, regexp])
+  prefixListId: String @search(by: [hash])
+}
+
 type awsTransitGatewayRouteTable implements awsBaseService @key(fields: "arn") {
   state: String @search(by: [hash, regexp])
   defaultAssociationRouteTable: Boolean @search
   defaultPropagationRouteTable: Boolean @search
   creationTime: DateTime @search(by: [day])
+  routes: [awsTransitGatewayRoute]
   tags: [awsRawTag]
   transitGateway: [awsTransitGateway] @hasInverse(field: routeTables)
   transitGatewayAttachment: [awsTransitGatewayAttachment] @hasInverse(field: routeTable)

src/types/generated.ts

@@ -4338,10 +4338,19 @@ export type AwsTransitGatewayAttachment = {
   vpnConnection?: Maybe<Array<Maybe<AwsVpnConnection>>>;
 };
 
+export type AwsTransitGatewayRoute = {
+  destinationCidrBlock?: Maybe<Scalars['String']>;
+  id: Scalars['String'];
+  prefixListId?: Maybe<Scalars['String']>;
+  state?: Maybe<Scalars['String']>;
+  type?: Maybe<Scalars['String']>;
+};
+
 export type AwsTransitGatewayRouteTable = AwsBaseService & {
   creationTime?: Maybe<Scalars['DateTime']>;
   defaultAssociationRouteTable?: Maybe<Scalars['Boolean']>;
   defaultPropagationRouteTable?: Maybe<Scalars['Boolean']>;
+  routes?: Maybe<Array<Maybe<AwsTransitGatewayRoute>>>;
   state?: Maybe<Scalars['String']>;
   tags?: Maybe<Array<Maybe<AwsRawTag>>>;
   transitGateway?: Maybe<Array<Maybe<AwsTransitGateway>>>;