Skip to content

Commit

Permalink
feat: load protos from JSON, grpc-fallback support (#142)
Browse files Browse the repository at this point in the history
  • Loading branch information
yoshi-automation authored and JustinBeckwith committed Sep 3, 2019
1 parent d785b7a commit a2c440b
Show file tree
Hide file tree
Showing 11 changed files with 1,133 additions and 314 deletions.
1,127 changes: 915 additions & 212 deletions packages/google-cloud-securitycenter/protos/protos.json

Large diffs are not rendered by default.

21 changes: 21 additions & 0 deletions packages/google-cloud-securitycenter/src/browser.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
// Copyright 2019 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

// Set a flag that we are running in a browser bundle.
global.isBrowser = true;

// Re-export all exports from ./index.js.
module.exports = require('./index');

This file was deleted.

115 changes: 66 additions & 49 deletions packages/google-cloud-securitycenter/src/v1/security_center_client.js
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
const gapicConfig = require('./security_center_client_config.json');
const gax = require('google-gax');
const path = require('path');
const protobuf = require('protobufjs');

const VERSION = require('../../package.json').version;

Expand Down Expand Up @@ -59,6 +58,16 @@ class SecurityCenterClient {
opts = opts || {};
this._descriptors = {};

if (global.isBrowser) {
// If we're in browser, we use gRPC fallback.
opts.fallback = true;
}

// If we are in browser, we are already using fallback because of the
// "browser" field in package.json.
// But if we were explicitly requested to use fallback, let's do it now.
const gaxModule = !global.isBrowser && opts.fallback ? gax.fallback : gax;

const servicePath =
opts.servicePath || opts.apiEndpoint || this.constructor.servicePath;

Expand All @@ -75,54 +84,69 @@ class SecurityCenterClient {
// Create a `gaxGrpc` object, with any grpc-specific options
// sent to the client.
opts.scopes = this.constructor.scopes;
const gaxGrpc = new gax.GrpcClient(opts);
const gaxGrpc = new gaxModule.GrpcClient(opts);

// Save the auth object to the client, for use by other methods.
this.auth = gaxGrpc.auth;

// Determine the client header string.
const clientHeader = [
`gl-node/${process.versions.node}`,
`grpc/${gaxGrpc.grpcVersion}`,
`gax/${gax.version}`,
`gapic/${VERSION}`,
];
const clientHeader = [];

if (typeof process !== 'undefined' && 'versions' in process) {
clientHeader.push(`gl-node/${process.versions.node}`);
}
clientHeader.push(`gax/${gaxModule.version}`);
if (opts.fallback) {
clientHeader.push(`gl-web/${gaxModule.version}`);
} else {
clientHeader.push(`grpc/${gaxGrpc.grpcVersion}`);
}
clientHeader.push(`gapic/${VERSION}`);
if (opts.libName && opts.libVersion) {
clientHeader.push(`${opts.libName}/${opts.libVersion}`);
}

// Load the applicable protos.
// For Node.js, pass the path to JSON proto file.
// For browsers, pass the JSON content.

const nodejsProtoPath = path.join(
__dirname,
'..',
'..',
'protos',
'protos.json'
);
const protos = gaxGrpc.loadProto(
path.join(__dirname, '..', '..', 'protos'),
['google/cloud/securitycenter/v1/securitycenter_service.proto']
opts.fallback ? require('../../protos/protos.json') : nodejsProtoPath
);

// This API contains "path templates"; forward-slash-separated
// identifiers to uniquely identify resources within the API.
// Create useful helper objects for these.
this._pathTemplates = {
assetPathTemplate: new gax.PathTemplate(
assetPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/assets/{asset}'
),
assetSecurityMarksPathTemplate: new gax.PathTemplate(
assetSecurityMarksPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/assets/{asset}/securityMarks'
),
findingPathTemplate: new gax.PathTemplate(
findingPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/sources/{source}/findings/{finding}'
),
findingSecurityMarksPathTemplate: new gax.PathTemplate(
findingSecurityMarksPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/sources/{source}/findings/{finding}/securityMarks'
),
organizationPathTemplate: new gax.PathTemplate(
organizationPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}'
),
organizationSettingsPathTemplate: new gax.PathTemplate(
organizationSettingsPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/organizationSettings'
),
organizationSourcesPathTemplate: new gax.PathTemplate(
organizationSourcesPathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/sources/-'
),
sourcePathTemplate: new gax.PathTemplate(
sourcePathTemplate: new gaxModule.PathTemplate(
'organizations/{organization}/sources/{source}'
),
};
Expand All @@ -131,48 +155,41 @@ class SecurityCenterClient {
// (e.g. 50 results at a time, with tokens to get subsequent
// pages). Denote the keys used for pagination and results.
this._descriptors.page = {
groupAssets: new gax.PageDescriptor(
groupAssets: new gaxModule.PageDescriptor(
'pageToken',
'nextPageToken',
'groupByResults'
),
groupFindings: new gax.PageDescriptor(
groupFindings: new gaxModule.PageDescriptor(
'pageToken',
'nextPageToken',
'groupByResults'
),
listAssets: new gax.PageDescriptor(
listAssets: new gaxModule.PageDescriptor(
'pageToken',
'nextPageToken',
'listAssetsResults'
),
listFindings: new gax.PageDescriptor(
listFindings: new gaxModule.PageDescriptor(
'pageToken',
'nextPageToken',
'listFindingsResults'
),
listSources: new gax.PageDescriptor(
listSources: new gaxModule.PageDescriptor(
'pageToken',
'nextPageToken',
'sources'
),
};
let protoFilesRoot = new gax.GoogleProtoFilesRoot();
protoFilesRoot = protobuf.loadSync(
path.join(
__dirname,
'..',
'..',
'protos',
'google/cloud/securitycenter/v1/securitycenter_service.proto'
),
protoFilesRoot
);

const protoFilesRoot = opts.fallback
? gaxModule.protobuf.Root.fromJSON(require('../../protos/protos.json'))
: gaxModule.protobuf.loadSync(nodejsProtoPath);

// This API contains "long-running operations", which return a
// an Operation object that allows for tracking of the operation,
// rather than holding a request open.
this.operationsClient = new gax.lro({
this.operationsClient = new gaxModule.lro({
auth: gaxGrpc.auth,
grpc: gaxGrpc.grpc,
}).operationsClient(opts);
Expand All @@ -185,7 +202,7 @@ class SecurityCenterClient {
);

this._descriptors.longrunning = {
runAssetDiscovery: new gax.LongrunningDescriptor(
runAssetDiscovery: new gaxModule.LongrunningDescriptor(
this.operationsClient,
runAssetDiscoveryResponse.decode.bind(runAssetDiscoveryResponse),
runAssetDiscoveryMetadata.decode.bind(runAssetDiscoveryMetadata)
Expand All @@ -208,7 +225,9 @@ class SecurityCenterClient {
// Put together the "service stub" for
// google.cloud.securitycenter.v1.SecurityCenter.
const securityCenterStub = gaxGrpc.createStub(
protos.google.cloud.securitycenter.v1.SecurityCenter,
opts.fallback
? protos.lookupService('google.cloud.securitycenter.v1.SecurityCenter')
: protos.google.cloud.securitycenter.v1.SecurityCenter,
opts
);

Expand All @@ -235,18 +254,16 @@ class SecurityCenterClient {
'updateSecurityMarks',
];
for (const methodName of securityCenterStubMethods) {
this._innerApiCalls[methodName] = gax.createApiCall(
securityCenterStub.then(
stub =>
function() {
const args = Array.prototype.slice.call(arguments, 0);
return stub[methodName].apply(stub, args);
},
err =>
function() {
throw err;
}
),
const innerCallPromise = securityCenterStub.then(
stub => (...args) => {
return stub[methodName].apply(stub, args);
},
err => () => {
throw err;
}
);
this._innerApiCalls[methodName] = gaxModule.createApiCall(
innerCallPromise,
defaults[methodName],
this._descriptors.page[methodName] ||
this._descriptors.longrunning[methodName]
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
[
"../../protos/google/cloud/securitycenter/v1/securitycenter_service.proto"
]
Loading

0 comments on commit a2c440b

Please sign in to comment.