feat(vpnGateway): add tencent vpnGateway

Author: james-zhou-inspire11

README.md

@@ -57,4 +57,5 @@ CloudGraph Tencent Provider will ask you what regions you would like to crawl an
 | Service | Relations |
 | ------------------- | ------------------- |
 | subnet | vpc |
-| vpc | subnet |
+| vpc | subnet, vpnGateway |
+| vpnGateway | vpc |

src/enums/schemasMap.ts

@@ -6,5 +6,6 @@ import services from './services'
 export default {
   [services.subnet]: 'tencentSubnet',
   [services.vpc]: 'tencentVpc',
+  [services.vpnGateway]: 'tencentVpnGateway',
   tag: 'tencentTag',
 }

src/enums/serviceAliases.ts

@@ -1,4 +1,5 @@
 export default {
   subnet: 'subnets',
   vpc: 'vpcInstances',
+  vpnGateway: 'vpnGateways',
 }

src/enums/serviceMap.ts

@@ -2,6 +2,7 @@ import services from './services'
 import TencentSubnet from '../services/subnet'
 import TencentVpc from '../services/vpc'
 import TencentTag from '../services/tag'
+import TencentVpnGateway from '../services/vpnGateway'
 
 /**
  * serviceMap is an object that contains all currently supported services
@@ -10,5 +11,6 @@ import TencentTag from '../services/tag'
 export default {
   [services.subnet]: TencentSubnet,
   [services.vpc]: TencentVpc,
+  [services.vpnGateway]: TencentVpnGateway,
   tag: TencentTag,
 }

src/enums/services.ts

@@ -1,4 +1,5 @@
 export default {
   subnet: 'subnet',
   vpc: 'vpc',
+  vpnGateway: 'vpnGateway',
 }

src/services/vpc/schema.graphql

@@ -12,6 +12,7 @@ type tencentVpc implements tencentBaseService @key(fields: "id") {
   tags: [tencentRawTag]
   assistantCidrSet: [tencentVpcAssistantCidr]
   subnets: [tencentSubnet] @hasInverse(field: vpcInstances)
+  vpnGateways: [tencentVpnGateway] @hasInverse(field: vpcInstances)
 }
 
 type tencentVpcAssistantCidr

src/services/vpnGateway/data.ts

@@ -0,0 +1,57 @@
+import * as tencentcloud from 'tencentcloud-sdk-nodejs'
+import { VpnGateway } from 'tencentcloud-sdk-nodejs/tencentcloud/services/vpc/v20170312/vpc_models'
+import { ClientConfig } from 'tencentcloud-sdk-nodejs/tencentcloud/common/interface'
+import CloudGraph from '@cloudgraph/sdk'
+import groupBy from 'lodash/groupBy'
+import isEmpty from 'lodash/isEmpty'
+import loggerText from '../../properties/logger'
+import { TencentServiceInput } from '../../types'
+import { initTestEndpoint, generateTencentErrorLog } from '../../utils'
+
+const lt = { ...loggerText }
+const { logger } = CloudGraph
+export const serviceName = 'VpnGateway'
+const apiEndpoint = initTestEndpoint(serviceName)
+
+export interface RawTencentVpnGateway extends VpnGateway {
+  id: string
+  region: string
+}
+
+export default async ({
+  regions,
+  config,
+}: TencentServiceInput): Promise<{
+  [region: string]: RawTencentVpnGateway[]
+}> =>
+  new Promise(async resolve => {
+    const vpnGatewayList: RawTencentVpnGateway[] = []
+
+    for (const region of regions.split(',')) {
+      /**
+       * Get all the vpn gateways
+       */
+      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.DescribeVpnGateways(null)
+
+        if (response && !isEmpty(response.VpnGatewaySet)) {
+          for (const instance of response.VpnGatewaySet) {
+            vpnGatewayList.push({
+              id: instance.VpnGatewayId,
+              ...instance,
+              region,
+            })
+          }
+        }
+        
+      } catch (error) {
+        generateTencentErrorLog(serviceName, 'vpc:DescribeVpnGateways', error)
+      }
+    }
+
+    logger.debug(lt.foundResources(serviceName, vpnGatewayList.length))
+    resolve(groupBy(vpnGatewayList, 'region'))
+  })

src/services/vpnGateway/format.ts

@@ -0,0 +1,67 @@
+import cuid from 'cuid'
+import { TencentVpnGateway } from '../../types/generated'
+import { RawTencentVpnGateway } from './data'
+
+export default ({
+  service,
+  region,
+}: {
+  service: RawTencentVpnGateway
+  region: string
+}): TencentVpnGateway=> {
+  const {
+    id,
+    VpnGatewayName: name,
+    Type: type,
+    State: state,
+    PublicIpAddress: publicIpAddress,
+    RenewFlag: renewFlag,
+    InstanceChargeType: instanceChargeType,
+    InternetMaxBandwidthOut: internetMaxBandwidthOut,
+    CreatedTime: createdTime,
+    ExpiredTime: expiredTime,
+    IsAddressBlocked: isAddressBlocked,
+    NewPurchasePlan: newPurchasePlan,
+    RestrictState: restrictState,
+    Zone: zone,
+    VpnGatewayQuotaSet,
+    Version: version,
+    NetworkInstanceId: networkInstanceId,
+    CdcId: cdcId,
+    MaxConnection: maxConnection,
+  } = service
+
+  return {
+    id,
+    region,
+    name,
+    type,
+    state,
+    publicIpAddress,
+    renewFlag,
+    instanceChargeType,
+    internetMaxBandwidthOut,
+    createdTime,
+    expiredTime,
+    isAddressBlocked,
+    newPurchasePlan,
+    restrictState,
+    zone,
+    vpnGatewayQuotaSet: VpnGatewayQuotaSet.map(({
+      Bandwidth: bandwidth,
+      Cname: chineseName,
+      Name: englishName,
+    }) => {
+      return {
+        id: cuid(),
+        bandwidth,
+        cname: chineseName,
+        name: englishName,
+      }
+    }),
+    version,
+    networkInstanceId,
+    cdcId,
+    maxConnection,
+  }
+}

src/services/vpnGateway/index.ts

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

src/services/vpnGateway/schema.graphql

@@ -0,0 +1,34 @@
+type tencentVpnGatewayQuota
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  )
+  @key(fields: "id") {
+    id: String! @id @search(by: [hash, regexp])
+    bandwidth: Int @search
+    cname: String @search(by: [hash, regexp])
+    name: String @search(by: [hash, regexp])
+  }
+
+type tencentVpnGateway implements tencentBaseService @key(fields: "id") {
+  name: String @search(by: [hash, regexp])
+  type: String @search(by: [hash, regexp])
+  state: String @search(by: [hash, regexp])
+  publicIpAddress: String @search(by: [hash, regexp])
+  renewFlag: String @search(by: [hash, regexp])
+  instanceChargeType: String @search(by: [hash, regexp])
+  internetMaxBandwidthOut: Int @search
+  createdTime: String @search(by: [hash, regexp])
+  expiredTime: String @search(by: [hash, regexp])
+  isAddressBlocked: Boolean @search
+  newPurchasePlan: String @search(by: [hash, regexp])
+  restrictState: String @search(by: [hash, regexp])
+  zone: String @search(by: [hash, regexp])
+  vpnGatewayQuotaSet: [tencentVpnGatewayQuota],
+  version: String @search(by: [hash, regexp])
+  networkInstanceId: String @search(by: [hash, regexp])
+  cdcId: String @search(by: [hash, regexp])
+  maxConnection: Int @search
+  vpcInstances: [tencentVpc] @hasInverse(field: vpnGateways)
+}