feat(service): GCP add new connections sslPolicy <-> targetSslProxy AND...

m-pizarro · tyler-dunkel · commit d52e7031c067 · 2022-03-11T22:26:04.000Z
diff --git a/README.md b/README.md
@@ -105,10 +105,10 @@ CloudGraph GCP Provider will ask you what regions you would like to crawl and wi
 | secretManager                         | project                                                                                                                             |
 | serviceAccounts                       | project                                                                                                                             |
 | sqlInstances                          | project, network                                                                                                                    |
-| sslPolicies                           | project                                                                                                                             |
+| sslPolicies                           | project, targetHttpsProxies, targetSslProxies                                                                                                                    |
 | storageBucket                         | project                                                                                                                             |
 | subnet                                | project, network, vmInstance, vpcConnectors, aiPlatformNotebooks                                                                    |
-| targetSslProxies                      | project                                                                                                                             |
-| targetHttpsProxies                    | project                                                                                                                             |
+| targetSslProxies                      | project, sslPolicies                                                                                                                         |
+| targetHttpsProxies                    | project, sslPolicies                                                                                                                         |
 | vmInstance                            | project, network, subnet                                                                                                            |
 | vpcConnectors                         | cloudFunction, project, network, subnet                                                                                             |
diff --git a/src/services/sslPolicy/schema.graphql b/src/services/sslPolicy/schema.graphql
@@ -10,6 +10,8 @@ type gcpSslPolicy implements gcpBaseResource @key(fields: "id") {
   selfLink: String @search(by: [hash, regexp])
   warnings: [gcpComputeWarnings]
   project: [gcpProject] @hasInverse(field: sslPolicies)
+  targetSslProxies: [gcpTargetSslProxy] @hasInverse(field: sslPolicy)
+  targetHttpsProxies: [gcpTargetHttpsProxy] @hasInverse(field: sslPolicy)
 }
 
 type gcpComputeWarnings
diff --git a/src/services/targetHttpsProxy/connections.ts b/src/services/targetHttpsProxy/connections.ts
@@ -0,0 +1,54 @@
+import { ServiceConnection } from '@cloudgraph/sdk'
+import { isEmpty } from 'lodash'
+import { RawGcpTargetHttpsProxy } from './data'
+import { RawGcpSslPolicy } from '../sslPolicy/data'
+import { GLOBAL_REGION } from '../../config/constants'
+import services from '../../enums/services'
+
+export default ({
+  service,
+  data,
+  region,
+}: {
+  account: string
+  service: RawGcpTargetHttpsProxy
+  data: { name: string; data: { [property: string]: any[] } }[]
+  region: string
+}): {
+  [property: string]: ServiceConnection[]
+} => {
+  const { id, sslPolicy } = service
+  const connections: ServiceConnection[] = []
+
+  /**
+   * Find related ssl policies
+   */
+  const sslPolicies: {
+    name: string
+    data: { [property: string]: any[] }
+  } = data.find(({ name }) => name === services.sslPolicies)
+
+  const regions = [region, GLOBAL_REGION]
+  for (const reg of regions) {
+    if (sslPolicies?.data?.[reg]) {
+      const policy = sslPolicies.data[reg].find(
+        ({ selfLink }: RawGcpSslPolicy) => selfLink === sslPolicy
+      )
+
+      if (!isEmpty(policy)) {
+        connections.push({
+          id: policy.id,
+          resourceType: services.sslPolicies,
+          relation: 'child',
+          field: 'sslPolicy',
+        })
+      }
+    }
+  }
+
+  const result = {
+    [id]: connections,
+  }
+
+  return result
+}
diff --git a/src/services/targetHttpsProxy/format.ts b/src/services/targetHttpsProxy/format.ts
@@ -23,7 +23,6 @@ export default ({
     selfLink,
     serverTlsPolicy,
     sslCertificates,
-    sslPolicy,
     urlMap,
   } = service
 
@@ -42,7 +41,6 @@ export default ({
     selfLink,
     serverTlsPolicy,
     sslCertificates,
-    sslPolicy,
     urlMap,
   }
 }
diff --git a/src/services/targetHttpsProxy/index.ts b/src/services/targetHttpsProxy/index.ts
@@ -2,12 +2,15 @@ import { Service } from '@cloudgraph/sdk'
 import BaseService from '../base'
 import format from './format'
 import getData from './data'
+import getConnections from './connections'
 import mutation from './mutation'
 
 export default class GcpTargetHttpsProxies extends BaseService implements Service {
   format = format.bind(this)
 
   getData = getData.bind(this)
 
+  getConnections = getConnections.bind(this)
+  
   mutation = mutation;
 }
diff --git a/src/services/targetHttpsProxy/schema.graphql b/src/services/targetHttpsProxy/schema.graphql
@@ -9,7 +9,7 @@ type gcpTargetHttpsProxy implements gcpBaseResource @key(fields: "id") {
   selfLink: String @search(by: [hash, regexp])
   serverTlsPolicy: String @search(by: [hash, regexp])
   sslCertificates: [String] @search(by: [hash])
-  sslPolicy: String @search(by: [hash, regexp])
   urlMap: String @search(by: [hash, regexp])
   project: [gcpProject] @hasInverse(field: targetHttpsProxies)
+  sslPolicy: [gcpSslPolicy] @hasInverse(field: targetHttpsProxies)
 }
diff --git a/src/services/targetSslProxy/connections.ts b/src/services/targetSslProxy/connections.ts
@@ -0,0 +1,54 @@
+import { ServiceConnection } from '@cloudgraph/sdk'
+import { isEmpty } from 'lodash'
+import { RawGcpTargetSslProxy } from './data'
+import { RawGcpSslPolicy } from '../sslPolicy/data'
+import { GLOBAL_REGION } from '../../config/constants'
+import services from '../../enums/services'
+
+export default ({
+  service,
+  data,
+  region,
+}: {
+  account: string
+  service: RawGcpTargetSslProxy
+  data: { name: string; data: { [property: string]: any[] } }[]
+  region: string
+}): {
+  [property: string]: ServiceConnection[]
+} => {
+  const { id, sslPolicy } = service
+  const connections: ServiceConnection[] = []
+
+  /**
+   * Find related ssl policies
+   */
+  const sslPolicies: {
+    name: string
+    data: { [property: string]: any[] }
+  } = data.find(({ name }) => name === services.sslPolicies)
+
+  const regions = [region, GLOBAL_REGION]
+  for (const reg of regions) {
+    if (sslPolicies?.data?.[reg]) {
+      const policy = sslPolicies.data[reg].find(
+        ({ selfLink }: RawGcpSslPolicy) => selfLink === sslPolicy
+      )
+
+      if (!isEmpty(policy)) {
+        connections.push({
+          id: policy.id,
+          resourceType: services.sslPolicies,
+          relation: 'child',
+          field: 'sslPolicy',
+        })
+      }
+    }
+  }
+
+  const result = {
+    [id]: connections,
+  }
+
+  return result
+}
diff --git a/src/services/targetSslProxy/format.ts b/src/services/targetSslProxy/format.ts
@@ -19,7 +19,6 @@ export default ({
     proxyHeader,
     selfLink,
     sslCertificates,
-    sslPolicy,
   } = service
 
   return {
@@ -34,6 +33,5 @@ export default ({
     selfLink,
     service: service?.service,
     sslCertificates,
-    sslPolicy,
   }
 }
diff --git a/src/services/targetSslProxy/index.ts b/src/services/targetSslProxy/index.ts
@@ -2,12 +2,15 @@ import { Service } from '@cloudgraph/sdk'
 import BaseService from '../base'
 import format from './format'
 import getData from './data'
+import getConnections from './connections'
 import mutation from './mutation'
 
 export default class GcpTargetSslProxies extends BaseService implements Service {
   format = format.bind(this)
 
   getData = getData.bind(this)
 
+  getConnections = getConnections.bind(this)
+
   mutation = mutation;
 }
diff --git a/src/services/targetSslProxy/schema.graphql b/src/services/targetSslProxy/schema.graphql
@@ -6,6 +6,6 @@ type gcpTargetSslProxy implements gcpBaseResource @key(fields: "id") {
   selfLink: String @search(by: [hash, regexp])
   service: String @search(by: [hash, regexp])
   sslCertificates: [String] @search(by: [hash])
-  sslPolicy: String @search(by: [hash, regexp])
   project: [gcpProject] @hasInverse(field: targetSslProxies)
+  sslPolicy: [gcpSslPolicy] @hasInverse(field: targetSslProxies)
 }
diff --git a/src/types/generated.ts b/src/types/generated.ts
@@ -2581,6 +2581,8 @@ export type GcpSslPolicy = GcpBaseResource & {
   profile?: Maybe<Scalars['String']>;
   project?: Maybe<Array<Maybe<GcpProject>>>;
   selfLink?: Maybe<Scalars['String']>;
+  targetHttpsProxies?: Maybe<Array<Maybe<GcpTargetHttpsProxy>>>;
+  targetSslProxies?: Maybe<Array<Maybe<GcpTargetSslProxy>>>;
   warnings?: Maybe<Array<Maybe<GcpComputeWarnings>>>;
 };
 
@@ -2687,7 +2689,7 @@ export type GcpTargetHttpsProxy = GcpBaseResource & {
   selfLink?: Maybe<Scalars['String']>;
   serverTlsPolicy?: Maybe<Scalars['String']>;
   sslCertificates?: Maybe<Array<Maybe<Scalars['String']>>>;
-  sslPolicy?: Maybe<Scalars['String']>;
+  sslPolicy?: Maybe<Array<Maybe<GcpSslPolicy>>>;
   urlMap?: Maybe<Scalars['String']>;
 };
 
@@ -2700,7 +2702,7 @@ export type GcpTargetSslProxy = GcpBaseResource & {
   selfLink?: Maybe<Scalars['String']>;
   service?: Maybe<Scalars['String']>;
   sslCertificates?: Maybe<Array<Maybe<Scalars['String']>>>;
-  sslPolicy?: Maybe<Scalars['String']>;
+  sslPolicy?: Maybe<Array<Maybe<GcpSslPolicy>>>;
 };
 
 export type GcpValue = {

Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,8 @@ type gcpSslPolicy implements gcpBaseResource @key(fields: "id") {`
`10`	`10`	`selfLink: String @search(by: [hash, regexp])`
`11`	`11`	`warnings: [gcpComputeWarnings]`
`12`	`12`	`project: [gcpProject] @hasInverse(field: sslPolicies)`
	`13`	`+ targetSslProxies: [gcpTargetSslProxy] @hasInverse(field: sslPolicy)`
	`14`	`+ targetHttpsProxies: [gcpTargetHttpsProxy] @hasInverse(field: sslPolicy)`
`13`	`15`	`}`
`14`	`16`
`15`	`17`	`type gcpComputeWarnings`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,6 @@ export default ({`
`23`	`23`	`selfLink,`
`24`	`24`	`serverTlsPolicy,`
`25`	`25`	`sslCertificates,`
`26`		`- sslPolicy,`
`27`	`26`	`urlMap,`
`28`	`27`	`} = service`
`29`	`28`
`@@ -42,7 +41,6 @@ export default ({`
`42`	`41`	`selfLink,`
`43`	`42`	`serverTlsPolicy,`
`44`	`43`	`sslCertificates,`
`45`		`- sslPolicy,`
`46`	`44`	`urlMap,`
`47`	`45`	`}`
`48`	`46`	`}`
Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ type gcpTargetHttpsProxy implements gcpBaseResource @key(fields: "id") {`
`9`	`9`	`selfLink: String @search(by: [hash, regexp])`
`10`	`10`	`serverTlsPolicy: String @search(by: [hash, regexp])`
`11`	`11`	`sslCertificates: [String] @search(by: [hash])`
`12`		`- sslPolicy: String @search(by: [hash, regexp])`
`13`	`12`	`urlMap: String @search(by: [hash, regexp])`
`14`	`13`	`project: [gcpProject] @hasInverse(field: targetHttpsProxies)`
	`14`	`+ sslPolicy: [gcpSslPolicy] @hasInverse(field: targetHttpsProxies)`
`15`	`15`	`}`
Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,6 @@ export default ({`
`19`	`19`	`proxyHeader,`
`20`	`20`	`selfLink,`
`21`	`21`	`sslCertificates,`
`22`		`- sslPolicy,`
`23`	`22`	`} = service`
`24`	`23`
`25`	`24`	`return {`
`@@ -34,6 +33,5 @@ export default ({`
`34`	`33`	`selfLink,`
`35`	`34`	`service: service?.service,`
`36`	`35`	`sslCertificates,`
`37`		`- sslPolicy,`
`38`	`36`	`}`
`39`	`37`	`}`
Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,6 @@ type gcpTargetSslProxy implements gcpBaseResource @key(fields: "id") {`
`6`	`6`	`selfLink: String @search(by: [hash, regexp])`
`7`	`7`	`service: String @search(by: [hash, regexp])`
`8`	`8`	`sslCertificates: [String] @search(by: [hash])`
`9`		`- sslPolicy: String @search(by: [hash, regexp])`
`10`	`9`	`project: [gcpProject] @hasInverse(field: targetSslProxies)`
	`10`	`+ sslPolicy: [gcpSslPolicy] @hasInverse(field: targetSslProxies)`
`11`	`11`	`}`