feat: respect common resources

templates/typescript_gapic/src/$version/$service_client.ts.njk

@@ -56,7 +56,7 @@ export class {{ service.name }}Client {
   */
   private _descriptors: Descriptors = {page: {}, stream: {}, longrunning: {}};
   private _innerApiCalls: {[name: string]: Function};
-  {%- if (service.pathTemplate.length > 0) %}
+  {%- if (service.pathTemplates.length > 0) %}
   private _pathTemplates: {[name: string]: gax.PathTemplate};
   {%- endif %}
   auth: gax.GoogleAuth;
@@ -145,9 +145,9 @@ export class {{ service.name }}Client {
         require("../../protos/protos.json") :
         nodejsProtoPath
     );
-{%- if (service.pathTemplate.length > 0) %}
+{%- if (service.pathTemplates.length > 0) %}
     this._pathTemplates = {
-{%- for template in service.pathTemplate %}
+{%- for template in service.pathTemplates %}
       {{ template.type.toCamelCase() }}PathTemplate: new gaxModule.PathTemplate(
         '{{ template.pattern }}'
       ),
@@ -496,11 +496,11 @@ export class {{ service.name }}Client {
     return this._innerApiCalls.{{ method.name.toCamelCase() }}(request, options, callback);
   } 
 {%- endfor %}
-{%- if (service.pathTemplate.length > 0) %}
+{%- if (service.pathTemplates.length > 0) %}
   // --------------------
   // -- Path templates --
   // --------------------
-{%- for template in service.pathTemplate %}
+{%- for template in service.pathTemplates %}
   {{ template.type.toLowerCase() }}Path(
 {%- set paramJoiner = joiner() %}
 {%- for param in template.params %}

typescript/src/schema/api.ts

@@ -1,7 +1,7 @@
 import * as plugin from '../../../pbjs-genfiles/plugin';
 
 import { Naming } from './naming';
-import { Proto } from './proto';
+import { Proto, MessagesMap, ResourceDescriptor, ResourceMap } from './proto';
 import { fstat } from 'fs-extra';
 
 export interface ProtosMap {
@@ -26,11 +26,19 @@ export class API {
         fd => fd.package && fd.package.startsWith(packageName)
       )
     );
+    // construct resource map
+    const resourceMap = getResourceMap(fileDescriptors);
+    // parse resource map to Proto constructor
     this.protos = fileDescriptors
       .filter(fd => fd.name)
       .reduce(
         (map, fd) => {
-          map[fd.name!] = new Proto(fd, packageName, grpcServiceConfig);
+          map[fd.name!] = new Proto(
+            fd,
+            packageName,
+            grpcServiceConfig,
+            resourceMap
+          );
           return map;
         },
         {} as ProtosMap
@@ -83,3 +91,48 @@ export class API {
     );
   }
 }
+
+function getResourceMap(
+  fd: plugin.google.protobuf.IFileDescriptorProto[]
+): ResourceMap {
+  const resourceMap: ResourceMap = {};
+  for (const fileDescriptor of fd) {
+    const messages = fileDescriptor
+      .messageType!.filter(message => message.name)
+      .reduce(
+        (map, message) => {
+          map['.' + fileDescriptor.package! + '.' + message.name!] = message;
+          return map;
+        },
+        {} as MessagesMap
+      );
+    for (const property of Object.keys(messages)) {
+      const m = messages[property];
+      if (m && m.options) {
+        const option = m.options;
+        if (option && option['.google.api.resource']) {
+          const opt = option['.google.api.resource'];
+          const oneResource = option[
+            '.google.api.resource'
+          ] as ResourceDescriptor;
+          if (opt.type) {
+            const arr = opt.type.match(/\/([^.]+)$/);
+            if (arr) {
+              oneResource.name = arr[arr.length - 1];
+            }
+          }
+          const pattern = opt.pattern;
+          if (pattern && pattern[0]) {
+            const params = pattern[0].match(/{[a-zA-Z]+}/g) || [];
+            for (let i = 0; i < params.length; i++) {
+              params[i] = params[i].replace('{', '').replace('}', '');
+            }
+            oneResource.params = params;
+          }
+          resourceMap[opt.type!] = oneResource;
+        }
+      }
+    }
+  }
+  return resourceMap;
+}

typescript/src/schema/proto.ts

@@ -158,17 +158,22 @@ interface ServiceDescriptorProto
   port: number;
   oauthScopes: string[];
   comments: string;
-  pathTemplate: ResourceDescriptor[];
+  pathTemplates: ResourceDescriptor[];
   commentsMap: CommentsMap;
   retryableCodeMap: RetryableCodeMap;
   grpcServiceConfig: plugin.grpc.service_config.ServiceConfig;
 }
 
 export interface ResourceDescriptor
   extends plugin.google.api.IResourceDescriptor {
+  name: string;
   params: string[];
 }
 
+export interface ResourceMap {
+  [name: string]: ResourceDescriptor;
+}
+
 export interface ServicesMap {
   [name: string]: ServiceDescriptorProto;
 }
@@ -385,7 +390,8 @@ function augmentService(
   packageName: string,
   service: plugin.google.protobuf.IServiceDescriptorProto,
   commentsMap: CommentsMap,
-  grpcServiceConfig: plugin.grpc.service_config.ServiceConfig
+  grpcServiceConfig: plugin.grpc.service_config.ServiceConfig,
+  resourceMap: ResourceMap
 ) {
   const augmentedService = service as ServiceDescriptorProto;
   augmentedService.packageName = packageName;
@@ -442,32 +448,28 @@ function augmentService(
       '.google.api.oauthScopes'
     ].split(',');
   }
-  augmentedService.pathTemplate = [];
+  augmentedService.pathTemplates = [];
   for (const property of Object.keys(messages)) {
     const m = messages[property];
-    if (m && m.options) {
-      const option = m.options;
-      if (option && option['.google.api.resource']) {
-        const opt = option['.google.api.resource'];
-        const onePathTemplate = option[
-          '.google.api.resource'
-        ] as ResourceDescriptor;
-        if (opt.type) {
-          const arr = opt.type.match(/\/([^.]+)$/);
-          if (arr) {
-            opt.type = arr[arr.length - 1];
-          }
-        }
-        const pattern = opt.pattern;
-        //TODO: SUPPORT MULTIPLE PATTERNS
-        if (pattern && pattern[0]) {
-          const params = pattern[0].match(/{[a-zA-Z]+}/g) || [];
-          for (let i = 0; i < params.length; i++) {
-            params[i] = params[i].replace('{', '').replace('}', '');
+    if (m && m.field) {
+      const fields = m.field;
+      for (const fieldDescriptor of fields) {
+        if (fieldDescriptor && fieldDescriptor.options) {
+          const option = fieldDescriptor.options;
+          if (option && option['.google.api.resourceReference']) {
+            const resourceReference = option['.google.api.resourceReference'];
+            const type = resourceReference.type;
+            if (!type || !resourceMap[type.toString()]) {
+              console.warn(
+                'resource does not have a type or resource is not in the map'
+              );
+              continue;
+            }
+            const resource = resourceMap[resourceReference.type!.toString()];
+            if (augmentedService.pathTemplates.includes(resource)) continue;
+            augmentedService.pathTemplates.push(resource);
           }
-          onePathTemplate.params = params;
         }
-        augmentedService.pathTemplate.push(onePathTemplate);
       }
     }
   }
@@ -485,7 +487,8 @@ export class Proto {
   constructor(
     fd: plugin.google.protobuf.IFileDescriptorProto,
     packageName: string,
-    grpcServiceConfig: plugin.grpc.service_config.ServiceConfig
+    grpcServiceConfig: plugin.grpc.service_config.ServiceConfig,
+    resourceMap: ResourceMap
   ) {
     fd.enumType = fd.enumType || [];
     fd.messageType = fd.messageType || [];
@@ -525,7 +528,8 @@ export class Proto {
           packageName,
           service,
           commentsMap,
-          grpcServiceConfig
+          grpcServiceConfig,
+          resourceMap
         )
       )
       .reduce(