Merge pull request #7 from cloudgraphdev/feature/CG-1127-tencent-ccn

Feature/cg 1127 tencent ccn

Author: tyler-dunkel

README.md

@@ -56,5 +56,7 @@ CloudGraph Tencent Provider will ask you what regions you would like to crawl an
 
 | Service | Relations |
 | ------------------- | ------------------- |
+| ccn | ccnAttachment |
+| ccnAttachment | ccn |
 | subnet | vpc |
 | vpc | subnet |

src/enums/schemasMap.ts

@@ -4,6 +4,8 @@ import services from './services'
  * schemasMap is an object that contains schemas name by resource
  */
 export default {
+  [services.ccn]: 'tencentCcn',
+  [services.ccnAttachment]: 'tencentCcnAttachment',
   [services.subnet]: 'tencentSubnet',
   [services.vpc]: 'tencentVpc',
   tag: 'tencentTag',

src/enums/serviceAliases.ts

@@ -1,4 +1,6 @@
 export default {
+  ccn: 'ccns',
+  ccnAttachment: 'ccnAttachments',
   subnet: 'subnets',
   vpc: 'vpcInstances',
 }

src/enums/serviceMap.ts

@@ -1,4 +1,6 @@
 import services from './services'
+import TencentCcn from '../services/ccn'
+import TencentCcnAttachment from '../services/ccnAttachment'
 import TencentSubnet from '../services/subnet'
 import TencentVpc from '../services/vpc'
 import TencentTag from '../services/tag'
@@ -8,6 +10,8 @@ import TencentTag from '../services/tag'
  * serviceMap is used by the serviceFactory to produce instances of service classes
  */
 export default {
+  [services.ccn]: TencentCcn,
+  [services.ccnAttachment]: TencentCcnAttachment,
   [services.subnet]: TencentSubnet,
   [services.vpc]: TencentVpc,
   tag: TencentTag,

src/enums/services.ts

@@ -1,4 +1,6 @@
 export default {
+  ccn: 'ccn',
+  ccnAttachment: 'ccnAttachment',
   subnet: 'subnet',
   vpc: 'vpc',
 }

src/services/ccn/connections.ts

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

src/services/ccn/data.ts

@@ -0,0 +1,54 @@
+import * as tencentcloud from 'tencentcloud-sdk-nodejs'
+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 { CCN } from 'tencentcloud-sdk-nodejs/tencentcloud/services/vpc/v20170312/vpc_models'
+import loggerText from '../../properties/logger'
+import { TencentServiceInput } from '../../types'
+import { initTestEndpoint, generateTencentErrorLog } from '../../utils'
+
+const lt = { ...loggerText }
+const { logger } = CloudGraph
+export const serviceName = 'Ccn'
+const apiEndpoint = initTestEndpoint(serviceName)
+
+export interface RawTencentCcn extends CCN {
+  id: string
+  region: string
+}
+
+export default async ({
+  regions,
+  config,
+}: TencentServiceInput): Promise<{
+  [region: string]: RawTencentCcn[]
+}> =>
+  new Promise(async resolve => {
+    const ccnList: RawTencentCcn[] = []
+
+    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.DescribeCcns(null)
+  
+        if (response && !isEmpty(response.CcnSet)) {
+          for (const instance of response.CcnSet) {
+            ccnList.push({
+              id: instance.CcnId,
+              ...instance,
+              region,
+            })
+          }
+        }
+
+      } catch (error) {
+        generateTencentErrorLog(serviceName, 'vpc:DescribeCcns', error)
+      }
+    }
+
+    logger.debug(lt.foundResources(serviceName, ccnList.length))
+    resolve(groupBy(ccnList, 'region'))
+  })

src/services/ccn/format.ts

@@ -0,0 +1,44 @@
+import { TencentCcn } from '../../types/generated'
+import { formatTagSet } from '../../utils/format'
+import { RawTencentCcn } from './data'
+
+export default ({
+  service,
+  region,
+}: {
+  service: RawTencentCcn
+  region: string
+}): TencentCcn => {
+  const {
+    id,
+    CcnName: ccnName,
+    CcnDescription: ccnDescription,
+    InstanceCount: instanceCount,
+    CreateTime: createTime,
+    State: state,
+    QosLevel: qosLevel,
+    InstanceChargeType: instanceChargeType,
+    BandwidthLimitType: bandwidthLimitType,
+    TagSet,
+    RoutePriorityFlag: routePriorityFlag,
+    RouteTableCount: routeTableCount,
+    RouteTableFlag: routeTableFlag,
+  } = service
+
+  return {
+    id,
+    region,
+    ccnName,
+    ccnDescription,
+    instanceCount,
+    createTime,
+    state,
+    qosLevel,
+    instanceChargeType,
+    bandwidthLimitType,
+    tags: formatTagSet(TagSet),
+    routePriorityFlag,
+    routeTableCount,
+    routeTableFlag,
+  }
+}

src/services/ccn/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 TencentCcn extends BaseService implements Service {
+  format = format.bind(this)
+
+  getData = getData.bind(this)
+
+  getConnections = getConnections.bind(this)
+
+  mutation = getMutation(serviceName)
+}

src/services/ccn/schema.graphql

@@ -0,0 +1,15 @@
+type tencentCcn implements tencentBaseService @key(fields: "id") {
+  ccnName: String @search(by: [hash, regexp])
+  ccnDescription: String @search(by: [hash, regexp])
+  instanceCount: Int @search
+  createTime: String @search(by: [hash, regexp])
+  state: String @search(by: [hash, regexp])
+  qosLevel: String @search(by: [hash, regexp])
+  instanceChargeType: String @search(by: [hash, regexp])
+  bandwidthLimitType: String @search(by: [hash, regexp])
+  tags: [tencentRawTag]
+  routePriorityFlag: Boolean @search
+  routeTableCount: Int @search
+  routeTableFlag: Boolean @search
+  ccnAttachments: [tencentCcnAttachment] @hasInverse(field: ccns)
+}