Merge pull request #4 from cloudgraphdev/feature/CG-1117-tencent-route-table

Feature/cg 1117 tencent route table

Author: tyler-dunkel

README.md

@@ -56,10 +56,11 @@ CloudGraph Tencent Provider will ask you what regions you would like to crawl an
 
 | Service | Relations |
 | ------------------- | ------------------- |
+| routeTable | vpc, subnet |
 | securityGroup | | 
 | securityGroupRule | |
 | ccn | ccnAttachment |
 | ccnAttachment | ccn |
-| subnet | vpc |
-| vpc | subnet, vpnGateway |
-| vpnGateway | vpc |
+| subnet | vpc, routeTable |
+| vpc | subnet, vpnGateway, routeTable |
+| vpnGateway | vpc |

src/enums/schemasMap.ts

@@ -4,6 +4,7 @@ import services from './services'
  * schemasMap is an object that contains schemas name by resource
  */
 export default {
+  [services.routeTable]: 'tencentRouteTable',
   [services.securityGroup]: 'tencentSecurityGroup',
   [services.securityGroupRule]: 'tencentSecurityGroupRule',
   [services.ccn]: 'tencentCcn',

src/enums/serviceAliases.ts

@@ -1,4 +1,5 @@
 export default {
+  routeTable: 'routeTables',
   securityGroup: 'securityGroups',
   ccn: 'ccns',
   ccnAttachment: 'ccnAttachments',

src/enums/serviceMap.ts

@@ -6,13 +6,15 @@ import TencentCcnAttachment from '../services/ccnAttachment'
 import TencentSubnet from '../services/subnet'
 import TencentVpc from '../services/vpc'
 import TencentTag from '../services/tag'
+import TencentRouteTable from '../services/routeTable'
 import TencentVpnGateway from '../services/vpnGateway'
 
 /**
  * serviceMap is an object that contains all currently supported services
  * serviceMap is used by the serviceFactory to produce instances of service classes
  */
 export default {
+  [services.routeTable]: TencentRouteTable,
   [services.securityGroup]: TencentSecurityGroup,
   [services.securityGroupRule]: TencentSecurityGroupRule,
   [services.ccn]: TencentCcn,

src/enums/services.ts

@@ -1,4 +1,5 @@
 export default {
+  routeTable: 'routeTable',
   securityGroup: 'securityGroup',
   securityGroupRule: 'securityGroupRule',
   ccn: 'ccn',

src/services/routeTable/connections.ts

@@ -0,0 +1,44 @@
+import { ServiceConnection } from '@cloudgraph/sdk'
+import { RawTencentRouteTable } from './data'
+import services from '../../enums/services'
+import aliases from '../../enums/serviceAliases'
+
+export default ({
+  service,
+  data,
+  region,
+}: {
+  service: RawTencentRouteTable
+  data: { name: string; data: { [property: string]: any[] } }[]
+  region: string
+}): {
+  [property: string]: ServiceConnection[]
+} => {
+  const { id } = service
+  const connections: ServiceConnection[] = []
+
+  const subnetSets = service.AssociationSet.map(({SubnetId}) => SubnetId)
+
+  const instances: {
+    name: string
+    data: { [property: string]: any[] }
+  } = data.find(({ name }) => name === services.subnet)
+
+  if (instances?.data?.[region]) {
+    for (const service of instances.data[region]) {
+      if (subnetSets.includes(service.id)) {
+        connections.push({
+          id: service.id,
+          resourceType: services.subnet,
+          relation: 'child',
+          field: aliases[services.subnet]
+        })
+      }
+    }
+  }
+
+  const result = {
+    [id]: connections,
+  }
+  return result
+}

src/services/routeTable/data.ts

@@ -0,0 +1,56 @@
+import * as tencentcloud from 'tencentcloud-sdk-nodejs'
+import CloudGraph from '@cloudgraph/sdk'
+import groupBy from 'lodash/groupBy'
+import isEmpty from 'lodash/isEmpty'
+
+import { ClientConfig } from 'tencentcloud-sdk-nodejs/tencentcloud/common/interface'
+import { RouteTable } from 'tencentcloud-sdk-nodejs/tencentcloud/services/vpc/v20170312/vpc_models'
+import loggerText from '../../properties/logger'
+import { TencentServiceInput } from '../../types'
+import { initTestEndpoint, generateTencentErrorLog } from '../../utils'
+
+export const serviceName = 'RouteTable'
+
+const lt = { ...loggerText }
+const apiEndpoint = initTestEndpoint(serviceName)
+const { logger } = CloudGraph
+
+export interface RawTencentRouteTable extends RouteTable {
+  id: string
+  region: string
+}
+
+export default async ({
+  regions,
+  config,
+}: TencentServiceInput): Promise<{
+  [region: string]: RawTencentRouteTable[]
+}> =>
+  new Promise(async resolve => {
+    const routeTableList: RawTencentRouteTable[] = []
+
+    for (const region of regions.split(',')) {
+      try {
+        const VpcClient = tencentcloud.vpc.v20170312.Client
+        const clientConfig: ClientConfig  = { credential: config, region, profile: { httpProfile: { endpoint: apiEndpoint } } }
+        const vpc = new VpcClient(clientConfig)
+        const response = await vpc.DescribeRouteTables({})
+        
+        if (response && !isEmpty(response.RouteTableSet)) {
+          for (const instance of response.RouteTableSet) {
+            routeTableList.push({
+              id: instance.RouteTableId,
+              ...instance,
+              region,
+            })
+          }
+        }
+      } catch (error) {
+        generateTencentErrorLog(serviceName, 'vpc:DescribeRouteTables', error)
+      }
+    }
+
+    logger.debug(lt.foundResources(serviceName, routeTableList.length))
+
+    resolve(groupBy(routeTableList, 'region'))
+  })

src/services/routeTable/format.ts

@@ -0,0 +1,84 @@
+import cuid from 'cuid'
+import { Route } from 'tencentcloud-sdk-nodejs/tencentcloud/services/vpc/v20170312/vpc_models'
+import { TencentRouteTable, TencentRouteTableRoute } from '../../types/generated'
+import { formatTagSet } from '../../utils/format'
+import { RawTencentRouteTable } from './data'
+
+const formatRouteTableRoute = (route: Route): TencentRouteTableRoute => {
+  const {
+    DestinationCidrBlock: destinationCidrBlock,
+    GatewayType: gatewayType,
+    GatewayId: gatewayId,
+    RouteId: routeId = 0,
+    RouteDescription: routeDescription = '',
+    Enabled: enabled = false,
+    RouteType: routeType = '',
+    RouteTableId: routeTableId = '',
+    DestinationIpv6CidrBlock: destinationIpv6CidrBlock = '',
+    RouteItemId: routeItemId = '',
+    PublishedToVbc: publishedToVbc = false,
+    CreatedTime: createdTime = '',
+  } = route
+
+  return {
+    id: cuid(),
+    destinationCidrBlock,
+    gatewayType,
+    gatewayId,
+    routeId,
+    routeDescription,
+    enabled,
+    routeType,
+    routeTableId,
+    destinationIpv6CidrBlock,
+    routeItemId,
+    publishedToVbc,
+    createdTime,
+  }
+}
+
+
+export default ({
+  service,
+  region,
+}: {
+  service: RawTencentRouteTable
+  region: string
+}): TencentRouteTable=> {
+  const {
+    id,
+    RouteTableId: routeTableId,
+    RouteTableName: routeTableName,
+    AssociationSet = [],
+    RouteSet = [],
+    Main: main,
+    CreatedTime: createdTime = '',
+    TagSet,
+    LocalCidrForCcn = [],
+  } = service
+
+  return {
+    id,
+    region,
+    routeTableId,
+    routeTableName,
+    associationSet: AssociationSet.map(({SubnetId: subnetId, RouteTableId: associationRouteTableId}) => {
+      return {
+        id: cuid(),
+        subnetId,
+        routeTableId: associationRouteTableId,
+      }
+    }),
+    routeSet: RouteSet.map(formatRouteTableRoute),
+    main,
+    createdTime,
+    tags: formatTagSet(TagSet),
+    localCidrForCcn: LocalCidrForCcn.map(({Cidr: cidr, PublishedToVbc: publishedToVbc}) => {
+      return {
+        id: cuid(),
+        cidr,
+        publishedToVbc,
+      }
+    })
+  }
+}

src/services/routeTable/index.ts

@@ -0,0 +1,16 @@
+import { Service } from '@cloudgraph/sdk'
+import BaseService from '../base'
+import format from './format'
+import getData, { serviceName } from './data'
+import getConnections from './connections'
+import { getMutation } from '../../utils'
+
+export default class TencentRouteTable extends BaseService implements Service {
+  format = format.bind(this)
+
+  getData = getData.bind(this)
+  
+  getConnections = getConnections.bind(this)
+
+  mutation = getMutation(serviceName)
+}

src/services/routeTable/schema.graphql

@@ -0,0 +1,58 @@
+type tencentRouteTableAssociation
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  )
+  @key(fields: "id") {
+  id: String! @id
+  subnetId: String @search(by: [hash, regexp])
+  routeTableId: String @search(by: [hash, regexp])
+}
+
+type tencentRouteTableRoute
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  )
+  @key(fields: "id") {
+  id: String! @id
+  destinationCidrBlock: String @search(by: [hash, regexp])
+  gatewayType: String @search(by: [hash, regexp])
+  gatewayId: String @search(by: [hash, regexp])
+  routeId: Int @search
+  routeDescription: String @search(by: [hash, regexp])
+  enabled: Boolean @search
+  routeType: String @search(by: [hash, regexp])
+  routeTableId: String @search(by: [hash, regexp])
+  destinationIpv6CidrBlock: String @search(by: [hash, regexp])
+  routeItemId: String @search(by: [hash, regexp])
+  publishedToVbc: Boolean @search
+  createdTime: String @search(by: [hash, regexp])
+}
+
+type tencentRouteTableLocalCidrForCcnn
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  )
+  @key(fields: "id") {
+  id: String! @id
+  cidr: String @search(by: [hash, regexp])
+  publishedToVbc: Boolean @search
+}
+
+type tencentRouteTable implements tencentBaseService @key(fields: "id") {
+  routeTableId: String @search(by: [hash, regexp])
+  routeTableName: String @search(by: [hash, regexp])
+  associationSet: [tencentRouteTableAssociation]
+  routeSet: [tencentRouteTableRoute]
+  main: Boolean @search
+  createdTime: String @search(by: [hash, regexp])
+  tags: [tencentRawTag]
+  localCidrForCcn: [tencentRouteTableLocalCidrForCcnn]
+  vpcInstances: [tencentVpc] @hasInverse(field: routeTables)
+  subnets: [tencentSubnet] @hasInverse(field: routeTables)
+}