Merge pull request #79 from cloudgraphdev/feature/EP-3226-add-route-table-services

feat(azure): add route table services

Author: m-pizarro

README.md

@@ -104,6 +104,8 @@ CloudGraph needs read permissions in order to ingest your data. To keep things e
 | replicationNetworks                         | resourceGroup                                                                 |
 | replicationPolicies                         | resourceGroup                                                                 |
 | resourceGroup                               | **all services**                                                              |
+| routeFilter                                 |                                                                               |
+| routeTable                                  |                                                                               |
 | securityAssessments                         |                                                                               |
 | securityContacts                            |                                                                               |
 | securityGroup                               | networkInterface, resourceGroup                                               |

src/enums/schemasMap.ts

@@ -77,6 +77,8 @@ export default {
   [services.replicationNetworks]: 'azureReplicationNetwork',
   [services.replicationPolicies]: 'azureReplicationPolicy',
   [services.resourceGroup]: 'azureResourceGroup',
+  [services.routeFilter]: 'azureRouteFilter',
+  [services.routeTable]: 'azureRouteTable',
   [services.securityAssesments]: 'azureSecurityAssesment',
   [services.securityContacts]: 'azureSecurityContact',
   [services.securityGroup]: 'azureNetworkSecurityGroup',

src/enums/serviceAliases.ts

@@ -65,6 +65,8 @@ export default {
   [services.replicationNetworks]: 'replicationNetworks',
   [services.replicationPolicies]: 'replicationPolicies',
   [services.resourceGroup]: 'resourceGroups',
+  [services.routeFilter]: 'routeFilters',
+  [services.routeTable]: 'routeTables',
   [services.securityAssesments]: 'securityAssesments',
   [services.securityContacts]: 'securityContacts',
   [services.securityGroup]: 'securityGroups',

src/enums/serviceMap.ts

@@ -89,6 +89,8 @@ import AzureBilling from '../services/billing'
 import AzureLogProfiles from '../services/logProfiles'
 import Subscription from '../services/subscription'
 import AzureNetworkWatcher from '../services/networkWatcher'
+import AzureRouteTable from '../services/routeTable'
+import AzureRouteFilter from '../services/routeFilter'
 
 /**
  * serviceMap is an object that contains all currently supported services for AZURE
@@ -169,6 +171,8 @@ export default {
   [services.replicationNetworks]: AzureReplicationNetwork,
   [services.replicationPolicies]: AzureReplicationPolicy,
   [services.resourceGroup]: AzureResourceGroup,
+  [services.routeFilter]: AzureRouteFilter,
+  [services.routeTable]: AzureRouteTable,
   [services.securityAssesments]: AzureSecurityAssesments,
   [services.securityContacts]: AzureSecurityContacts,
   [services.securityGroup]: AzureNetworkSecurityGroup,

src/enums/services.ts

@@ -71,6 +71,8 @@ export default {
   replicationNetworks: 'replicationNetworks',
   replicationPolicies: 'replicationPolicies',
   resourceGroup: 'resourceGroup',
+  routeFilter: 'routeFilter',
+  routeTable: 'routeTable',
   securityAssesments: 'securityAssesments',
   securityContacts: 'securityContacts',
   securityGroup: 'securityGroup',

src/properties/logger.ts

@@ -173,6 +173,10 @@ export default {
     `Found ${num} replication policies`,
   // Resource Groups
   foundResourceGroups: (num: number): string => `Found ${num} resource groups`,
+  // Route Filters
+  foundRouteFilters: (num: number): string => `Found ${num} route filters`,
+  // Route Tables
+  foundRouteTables: (num: number): string => `Found ${num} route tables`,
   // Security Assesments
   foundSecurityAssesments: (num: number): string =>
     `Found ${num} security assesments`,

src/services/routeFilter/data.ts

@@ -0,0 +1,79 @@
+import { NetworkManagementClient, RouteFilter } from '@azure/arm-network'
+import { PagedAsyncIterableIterator } from '@azure/core-paging'
+import CloudGraph from '@cloudgraph/sdk'
+
+import azureLoggerText from '../../properties/logger'
+import { AzureServiceInput, TagMap } from '../../types'
+import { tryCatchWrapper } from '../../utils'
+import { lowerCaseLocation } from '../../utils/format'
+import { getResourceGroupFromEntity } from '../../utils/idParserUtils'
+
+const { logger } = CloudGraph
+const lt = { ...azureLoggerText }
+const serviceName = 'RouteFilter'
+
+export interface RawAzureRouteFilter
+  extends Omit<RouteFilter, 'tags' | 'location'> {
+  region: string
+  resourceGroupId: string
+  Tags: TagMap
+}
+
+export default async ({
+  regions,
+  config,
+}: AzureServiceInput): Promise<{
+  [property: string]: RawAzureRouteFilter[]
+}> => {
+  try {
+    const { tokenCredentials, subscriptionId } = config
+    const client = new NetworkManagementClient(tokenCredentials, subscriptionId)
+
+    const routeFilterData: RawAzureRouteFilter[] = []
+    await tryCatchWrapper(
+      async () => {
+        const routeFilterIterable: PagedAsyncIterableIterator<RouteFilter> =
+          client.routeFilters.list()
+        for await (const routeFilter of routeFilterIterable) {
+          if (routeFilter) {
+            const { location, tags, ...rest } = routeFilter
+            const resourceGroupId = getResourceGroupFromEntity(rest)
+            const region = lowerCaseLocation(location)
+            routeFilterData.push({
+              ...rest,
+              region,
+              resourceGroupId,
+              Tags: tags || {},
+            })
+          }
+        }
+      },
+      {
+        service: serviceName,
+        client,
+        scope: 'routeFilters',
+        operation: 'list',
+      }
+    )
+    const result: { [property: string]: RawAzureRouteFilter[] } = {}
+    let numOfGroups = 0
+    routeFilterData.forEach(({ region, ...rest }) => {
+      if (regions.includes(region)) {
+        if (!result[region]) {
+          result[region] = []
+        }
+        result[region].push({
+          region,
+          ...rest,
+        })
+        numOfGroups += 1
+      }
+    })
+    logger.debug(lt.foundRouteFilters(numOfGroups))
+
+    return result
+  } catch (e) {
+    logger.error(e)
+    return {}
+  }
+}

src/services/routeFilter/format.ts

@@ -0,0 +1,22 @@
+import { RawAzureRouteFilter } from './data'
+import { AzureRouteFilter } from '../../types/generated'
+import { formatTagsFromMap } from '../../utils/format'
+
+export default ({
+  service,
+  account: subscriptionId,
+}: {
+  service: RawAzureRouteFilter
+  account: string
+}): AzureRouteFilter => {
+  const { id, name, type, region, Tags } = service
+
+  return {
+    id,
+    name,
+    type,
+    region,
+    subscriptionId,
+    tags: formatTagsFromMap(Tags),
+  }
+}

src/services/routeFilter/index.ts

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

src/services/routeFilter/mutation.ts

@@ -0,0 +1,5 @@
+export default `mutation($input: [AddazureRouteFilterInput!]!) {
+  addazureRouteFilter(input: $input, upsert: true) {
+    numUids
+  }
+}`;