Merge pull request #3 from cloudgraphdev/feature/CG-1122

Feature/CG-1122: add security group service

Author: Christopher Brandt

README.md

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

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.securityGroup]: 'tencentSecurityGroup',
   [services.subnet]: 'tencentSubnet',
   [services.vpc]: 'tencentVpc',
   tag: 'tencentTag',

src/enums/serviceAliases.ts

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

src/enums/serviceMap.ts

@@ -1,4 +1,5 @@
 import services from './services'
+import TencentSecurityGroup from '../services/securityGroup'
 import TencentSubnet from '../services/subnet'
 import TencentVpc from '../services/vpc'
 import TencentTag from '../services/tag'
@@ -8,6 +9,7 @@ import TencentTag from '../services/tag'
  * serviceMap is used by the serviceFactory to produce instances of service classes
  */
 export default {
+  [services.securityGroup]: TencentSecurityGroup,
   [services.subnet]: TencentSubnet,
   [services.vpc]: TencentVpc,
   tag: TencentTag,

src/enums/services.ts

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

src/services/securityGroup/data.ts

@@ -0,0 +1,57 @@
+import * as tencentcloud from 'tencentcloud-sdk-nodejs'
+import { SecurityGroup } 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 = 'SecurityGroup'
+const apiEndpoint = initTestEndpoint(serviceName)
+
+export interface RawTencentSecurityGroup extends SecurityGroup {
+  id: string
+  region: string
+}
+
+export default async ({
+  regions,
+  config,
+}: TencentServiceInput): Promise<{
+  [region: string]: RawTencentSecurityGroup[]
+}> =>
+  new Promise(async resolve => {
+    const vpcList: RawTencentSecurityGroup[] = []
+
+    for (const region of regions.split(',')) {
+      /**
+       * Get all security groups
+       */
+      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.DescribeSecurityGroups(null)
+
+        if (response && !isEmpty(response.SecurityGroupSet)) {
+          for (const instance of response.SecurityGroupSet) {
+            vpcList.push({
+              id: instance.SecurityGroupId,
+              ...instance,
+              region,
+            })
+          }
+        }
+        
+      } catch (error) {
+        generateTencentErrorLog(serviceName, 'vpc:DescribeSecurityGroups', error)
+      }
+    }
+
+    logger.debug(lt.foundResources(serviceName, vpcList.length))
+    resolve(groupBy(vpcList, 'region'))
+  })

src/services/securityGroup/format.ts

@@ -0,0 +1,40 @@
+import cuid from 'cuid'
+import { TencentSecurityGroup } from '../../types/generated'
+import { RawTencentSecurityGroup } from './data'
+
+export default ({
+  service,
+  account,
+  region,
+}: {
+  service: RawTencentSecurityGroup
+  account: string
+  region: string
+}): TencentSecurityGroup => {
+  const {
+    id,
+    SecurityGroupName: name,
+    SecurityGroupDesc: securityGroupDesc,
+    ProjectId: projectId,
+    IsDefault: isDefault,
+    CreatedTime: createdTime,
+    TagSet,
+    UpdateTime: updateTime,
+  } = service
+
+  return {
+    id,
+    region,
+    name,
+    securityGroupDesc,
+    projectId,
+    isDefault,
+    createdTime,
+    tags: TagSet?.map(tagSet => ({
+      id: cuid(),
+      key: tagSet.Key,
+      value: tagSet.Value,
+    })),
+    updateTime,
+  }
+}

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

src/services/securityGroup/schema.graphql

@@ -0,0 +1,9 @@
+type tencentSecurityGroup implements tencentBaseService @key(fields: "id") {
+  name: String @search(by: [hash, regexp])
+  securityGroupDesc: String @search(by: [hash, regexp])
+  projectId: String @search(by: [hash, regexp])
+  isDefault: Boolean @search
+  createdTime: String @search(by: [hash, regexp])
+  tags: [tencentRawTag]
+  updateTime: String @search(by: [hash, regexp])
+}

src/services/tag/schema.graphql

@@ -3,6 +3,7 @@ type tencentTag implements tencentBaseService @key(fields: "id") {
   value: String! @search(by: [hash, regexp])
   subnets: [tencentSubnet]
   vpcInstances: [tencentVpc]
+  securityGroups: [tencentSecurityGroup]
 }
 
 type tencentRawTag