feat(kubernetesEngine): add tke service

Author: Christopher Brandt

README.md

@@ -70,3 +70,4 @@ CloudGraph Tencent Provider will ask you what regions you would like to crawl an
 | vpnConnection | vpc, vpnGateway, customerGateway |
 | vpnGateway | vpc, vpnGatewayRoute, vpnConnection |
 | vpnGatewayRoute | vpnGateway |
+| kubernetesEngine | subnet, vpc |

src/enums/schemasMap.ts

@@ -13,6 +13,7 @@ export default {
   [services.routeTable]: 'tencentRouteTable',
   [services.securityGroup]: 'tencentSecurityGroup',
   [services.securityGroupRule]: 'tencentSecurityGroupRule',
+  [services.kubernetesEngine]: 'tencentKubernetesCluster',
   [services.subnet]: 'tencentSubnet',
   [services.vpc]: 'tencentVpc',
   [services.vpnConnection]: 'tencentVpnConnection',

src/enums/serviceAliases.ts

@@ -5,6 +5,7 @@ export default {
   customerGateway: 'customerGateways',
   routeTable: 'routeTables',
   securityGroup: 'securityGroups',
+  kubernetesEngine: 'kubernetesClusters',
   subnet: 'subnets',
   vpc: 'vpcInstances',
   vpnConnection: 'vpnConnections',

src/enums/serviceMap.ts

@@ -4,6 +4,7 @@ import TencentSecurityGroup from '../services/securityGroup'
 import TencentSecurityGroupRule from '../services/securityGroupRule'
 import TencentCcn from '../services/ccn'
 import TencentCcnAttachment from '../services/ccnAttachment'
+import TencentKubernetesEngine from '../services/kubernetesEngine'
 import TencentSubnet from '../services/subnet'
 import TencentVpc from '../services/vpc'
 import TencentTag from '../services/tag'
@@ -29,6 +30,7 @@ export default {
   [services.routeTable]: TencentRouteTable,
   [services.securityGroup]: TencentSecurityGroup,
   [services.securityGroupRule]: TencentSecurityGroupRule,
+  [services.kubernetesEngine]: TencentKubernetesEngine,
   [services.subnet]: TencentSubnet,
   [services.vpc]: TencentVpc,
   [services.vpnConnection]: TencentVpnConnection,

src/enums/services.ts

@@ -8,9 +8,11 @@ export default {
   routeTable: 'routeTable',
   securityGroup: 'securityGroup',
   securityGroupRule: 'securityGroupRule',
+  kubernetesEngine: 'kubernetesEngine',
   subnet: 'subnet',
   vpc: 'vpc',
   vpnConnection: 'vpnConnection',
   vpnGateway: 'vpnGateway',
   vpnGatewayRoute: 'vpnGatewayRoute',
+  tag: 'tag'
 }

src/services/kubernetesEngine/data.ts

@@ -0,0 +1,59 @@
+import * as tencentcloud from 'tencentcloud-sdk-nodejs'
+import { Cluster } from 'tencentcloud-sdk-nodejs/tencentcloud/services/tke/v20180525/tke_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
+const serviceName = 'KubernetesEngine'
+const apiEndpoint = initTestEndpoint(serviceName)
+
+export interface RawTencentKubernetesEngine extends Cluster {
+  id: string
+  region: string
+  subnets: string[]
+}
+
+export default async ({
+  regions,
+  config,
+}: TencentServiceInput): Promise<{
+  [region: string]: RawTencentKubernetesEngine[]
+}> =>
+  new Promise(async resolve => {
+    const clusterList: RawTencentKubernetesEngine[] = []
+
+    for (const region of regions.split(',')) {
+      /**
+       * Get all clusters
+       */
+      try {
+        const TkeClient = tencentcloud.tke.v20180525.Client
+        const clientConfig: ClientConfig  = { credential: config, region, profile: { httpProfile: { endpoint: apiEndpoint } } }
+        const tke = new TkeClient(clientConfig)
+        const response = await tke.DescribeClusters(null)
+
+        if (response && !isEmpty(response.Clusters)) {
+          for (const instance of response.Clusters) {
+            clusterList.push({
+              id: instance.ClusterId,
+              ...instance,
+              subnets: instance.ClusterNetworkSettings?.Subnets?.map(subnet => subnet),
+              region,
+            })
+          }
+        }
+        
+      } catch (error) {
+        generateTencentErrorLog(serviceName, 'tke:DescribeClusters', error)
+      }
+    }
+
+    logger.debug(lt.foundResources(serviceName, clusterList.length))
+    resolve(groupBy(clusterList, 'region'))
+  })

src/services/kubernetesEngine/format.ts

@@ -0,0 +1,82 @@
+import cuid from 'cuid'
+import { TencentKubernetesCluster } from '../../types/generated'
+import { RawTencentKubernetesEngine } from './data'
+
+export default ({
+  service,
+  account,
+  region,
+}: {
+  service: RawTencentKubernetesEngine
+  account: string
+  region: string
+}): TencentKubernetesCluster => {
+  const {
+    id,
+    ClusterName: name,
+    ClusterDescription: clusterDescription,
+    ClusterVersion: clusterVersion,
+    ClusterOs: clusterOs,
+    ClusterType: clusterType,
+    ClusterNetworkSettings,
+    ClusterNodeNum: clusterNodeNum,
+    ProjectId: projectId,
+    TagSpecification,
+    ClusterStatus: clusterStatus,
+    Property: property,
+    ClusterMaterNodeNum: clusterMaterNodeNum,
+    ImageId: imageId,
+    OsCustomizeType: osCustomizeType,
+    ContainerRuntime: containerRuntime,
+    CreatedTime: createdTime,
+    DeletionProtection: deletionProtection,
+    EnableExternalNode: enableExternalNode,
+    ClusterLevel: clusterLevel,
+    AutoUpgradeClusterLevel: autoUpgradeClusterLevel,
+  } = service
+
+  const Tags = []
+  TagSpecification?.map(spec => {
+    Tags.push(spec.Tags)
+  })
+
+  return {
+    id,
+    region,
+    name,
+    clusterDescription,
+    clusterVersion,
+    clusterOs,
+    clusterType,
+    clusterNetworkSettings: {
+      clusterCIDR: ClusterNetworkSettings?.ClusterCIDR,
+      ignoreClusterCIDRConflict: ClusterNetworkSettings?.IgnoreClusterCIDRConflict,
+      maxNodePodNum: ClusterNetworkSettings?.MaxNodePodNum,
+      maxClusterServiceNum: ClusterNetworkSettings?.MaxClusterServiceNum,
+      ipvs: ClusterNetworkSettings?.Ipvs,
+      vpcId: ClusterNetworkSettings?.VpcId,
+      cni: ClusterNetworkSettings?.Cni,
+      kubeProxyMode: ClusterNetworkSettings?.KubeProxyMode,
+      serviceCIDR: ClusterNetworkSettings?.ServiceCIDR,
+      subnets: ClusterNetworkSettings?.Subnets,
+    },
+    clusterNodeNum,
+    projectId,
+    tags: Tags.map(tag => ({
+      id: cuid(),
+      key: tag.Key,
+      value: tag.Value,
+    })),
+    clusterStatus,
+    property,
+    clusterMaterNodeNum,
+    imageId,
+    osCustomizeType,
+    containerRuntime,
+    createdTime,
+    deletionProtection,
+    enableExternalNode,
+    clusterLevel,
+    autoUpgradeClusterLevel,
+  }
+}

src/services/kubernetesEngine/index.ts

@@ -0,0 +1,16 @@
+import { Service } from '@cloudgraph/sdk'
+import BaseService from '../base'
+import format from './format'
+import getData from './data'
+import { getMutation } from '../../utils'
+import services from '../../enums/services'
+import schemasMap from '../../enums/schemasMap'
+
+
+export default class TencentKubernetesEngine extends BaseService implements Service {
+  format = format.bind(this)
+
+  getData = getData.bind(this)
+
+  mutation = getMutation(schemasMap[services.kubernetesEngine])
+}

src/services/kubernetesEngine/schema.graphql

@@ -0,0 +1,42 @@
+type tencentKubernetesCluster implements tencentBaseService @key(fields: "id") {
+  name: String @search(by: [hash, regexp])
+  clusterDescription: String @search(by: [hash, regexp])
+  clusterVersion: String @search(by: [hash, regexp])
+  clusterOs: String @search(by: [hash, regexp])
+  clusterType: String @search(by: [hash, regexp])
+  clusterNetworkSettings: tencentKubernetesClusterNetworkSettings
+  clusterNodeNum: Int @search
+  projectId: Int @search
+  tags: [tencentRawTag]
+  clusterStatus: String @search(by: [hash, regexp])
+  property: String @search(by: [hash, regexp])
+  clusterMaterNodeNum: Int @search
+  imageId: String @search(by: [hash, regexp])
+  osCustomizeType: String @search(by: [hash, regexp])
+  containerRuntime: String @search(by: [hash, regexp])
+  createdTime: String @search(by: [hash, regexp])
+  deletionProtection: Boolean @search
+  enableExternalNode: Boolean @search
+  clusterLevel: String @search(by: [hash, regexp])
+  autoUpgradeClusterLevel: Boolean @search
+  subnets: [tencentSubnet] @hasInverse(field: kubernetesClusters)
+  vpcInstances: [tencentVpc] @hasInverse(field: kubernetesClusters)
+}
+
+type tencentKubernetesClusterNetworkSettings
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  ) {
+  clusterCIDR: String @search(by: [hash, regexp])
+  ignoreClusterCIDRConflict: Boolean @search
+  maxNodePodNum: Int @search
+  maxClusterServiceNum: Int @search
+  ipvs: Boolean @search
+  vpcId: String @search(by: [hash, regexp])
+  cni: Boolean @search
+  kubeProxyMode: String @search(by: [hash, regexp])
+  serviceCIDR: String @search(by: [hash, regexp])
+  subnets: [String] @search(by: [hash, regexp])
+}

src/services/subnet/data.ts

@@ -10,7 +10,7 @@ import { initTestEndpoint, generateTencentErrorLog } from '../../utils'
 
 const lt = { ...loggerText }
 const { logger } = CloudGraph
-export const serviceName = 'Subnet'
+const serviceName = 'Subnet'
 const apiEndpoint = initTestEndpoint(serviceName)
 
 export interface RawTencentSubnet extends Subnet {