Merge pull request #43 from cloudgraphdev/feature/CG-1270-use-nodeRecorceGroup-to-get-resources

feat(aksManagedCluster): CG-1270 add VMSS connection

Author: hjaraujof

README.md

@@ -42,7 +42,7 @@ CloudGraph needs read permissions in order to ingest your data. To keep things e
 | adIdentitySecurityDefaultsEnforcementPolicy |                                                                               |
 | adServicePrincipal                          | adApplication, authRoleAssignment                                             |
 | adUser                                      | adApplication, authRoleAssignment                                             |
-| aksManagedCluster                           | resourceGroup                                                                 |
+| aksManagedCluster                           | resourceGroup, virtualMachineScaleSet                                         |
 | appInsights                                 | resourceGroup                                                                 |
 | appServiceEnvironment                       | resourceGroup, virtualNetwork                                                 |
 | appServicePlan                              | resourceGroup, appServiceWebApp                                               |
@@ -117,7 +117,7 @@ CloudGraph needs read permissions in order to ingest your data. To keep things e
 | synapseWorkspaces                           | resourceGroup, synapseBigDataPools, synapseSqlPools                           |
 | trafficManager                              | resourceGroup                                                                 |
 | virtualMachine                              | disk, networkInterface, resourceGroup, virtualNetwork, virtualMachineScaleSet |
-| virtualMachineScaleSet                      | resourceGroup, virtualMachine                                                 |
+| virtualMachineScaleSet                      | resourceGroup, virtualMachine, aksManagedCluster                              |
 | virtualNetwork                              | appServiceEnvironment,networkInterface, resourceGroup virtualMachine          |
 
 ## Development

src/services/aksManagedCluster/connections.ts

@@ -4,6 +4,7 @@ import { isEmpty } from 'lodash'
 import services from '../../enums/services'
 import { caseInsensitiveEqual } from '../../utils'
 import { RawAzureResourceGroup } from '../resourceGroup/data'
+import { RawAzureVirtualMachineScaleSet } from '../virtualMachineScaleSet/data'
 import { RawAzureAksManagedCluster } from './data'
 
 export default ({
@@ -18,7 +19,7 @@ export default ({
   [property: string]: ServiceConnection[]
 } => {
   const connections: ServiceConnection[] = []
-  const { id, resourceGroupId: rgName } = service
+  const { id, resourceGroupId: rgName, nodeResourceGroup: nodeRgName } = service
 
   /**
    * Find resource group related to this kubernetes environment
@@ -47,6 +48,33 @@ export default ({
     }
   }
 
+  /**
+   * Find VMSS related to this kubernetes environment
+   */
+  const vmssList: {
+    name: string
+    data: { [property: string]: RawAzureVirtualMachineScaleSet[] }
+  } = data.find(({ name }) => name === services.virtualMachineScaleSet)
+
+  if (vmssList?.data?.[region]) {
+    const vmssInRegion: RawAzureVirtualMachineScaleSet[] = vmssList.data[
+      region
+    ].filter(({resourceGroupId: vmssRgName}: RawAzureVirtualMachineScaleSet) => 
+      caseInsensitiveEqual(vmssRgName, nodeRgName)
+    )
+
+    if (!isEmpty(vmssInRegion)) {
+      for (const vmss of vmssInRegion) {
+        connections.push({
+          id: vmss.id,
+          resourceType: services.virtualMachineScaleSet,
+          relation: 'child',
+          field: 'virtualMachineScaleSets',
+        })
+      }
+    }
+  }
+
   const kResult = {
     [id]: connections,
   }

src/services/aksManagedCluster/schema.graphql

@@ -193,4 +193,5 @@ type azureAksManagedCluster implements azureResource
   skuName: String @search(by: [hash, regexp])
   skuTier: String @search(by: [hash, regexp])
   resourceGroup: [azureResourceGroup] @hasInverse(field: aksManagedClusters)
+  virtualMachineScaleSets: [azureVirtualMachineScaleSet] @hasInverse(field: aksManagedClusters)
 }

src/services/virtualMachineScaleSet/schema.graphql

@@ -313,4 +313,6 @@ type azureVirtualMachineScaleSet implements azureResource
     @hasInverse(field: virtualMachineScaleSets)
   virtualMachines: [azureVirtualMachine]
     @hasInverse(field: virtualMachineScaleSets)
+  aksManagedClusters: [azureAksManagedCluster]
+    @hasInverse(field: virtualMachineScaleSets)
 }

src/types/generated.ts

@@ -342,6 +342,7 @@ export type AzureAksManagedCluster = AzureResource & {
   servicePrincipalProfileClientId?: Maybe<Scalars['String']>;
   skuName?: Maybe<Scalars['String']>;
   skuTier?: Maybe<Scalars['String']>;
+  virtualMachineScaleSets?: Maybe<Array<Maybe<AzureVirtualMachineScaleSet>>>;
 };
 
 export type AzureAksManagedClusterAadProfile = {
@@ -5294,6 +5295,7 @@ export type AzureVirtualMachinePlan = {
 };
 
 export type AzureVirtualMachineScaleSet = AzureResource & {
+  aksManagedClusters?: Maybe<Array<Maybe<AzureAksManagedCluster>>>;
   doNotRunExtensionsOnOverprovisionedVMs?: Maybe<Scalars['Boolean']>;
   overprovision?: Maybe<Scalars['Boolean']>;
   platformFaultDomainCount?: Maybe<Scalars['Int']>;