feat: support multi-pattern resources

.circleci/config.yml

@@ -35,6 +35,7 @@ jobs:
             cp -r typescript/test/protos .test-out-keymanager
             cp -r typescript/test/protos .test-out-translate
             cp -r typescript/test/protos .test-out-texttospeech
+            cp -r typescript/test/protos .test-out-dlp
       - persist_to_workspace:
           root: /tmp
           paths:
@@ -121,6 +122,23 @@ jobs:
             npm run system-test
             npm run docs
             npm run docs-test
+  dlpLibTest:
+    docker:
+      - image: circleci/node:10-browsers
+    steps: 
+      - checkout
+      - attach_workspace:
+          at: workspace
+      - run:
+          name: Run unit tests, system tests, jsdoc generation, and gts fix of the generated dlp library
+          command: |
+            cd workspace/workspace/gapic-generator-typescript/.test-out-dlp
+            npm install
+            npm test
+            npm run fix
+            npm run compile
+            npm run system-test
+            npm run docs
   ttsLibTest:
     docker:
       - image: circleci/node:10-browsers
@@ -156,6 +174,9 @@ workflows:
       - kmsLibTest:
           requires:
             - testGenerator
+      - dlpLibTest:
+          requires:
+            - testGenerator
       - translateLibTest:
           requires:
             - testGenerator

typescript/src/schema/proto.ts

@@ -543,12 +543,19 @@ function augmentService(
       );
 
       // 2. If this resource reference has .type, we should have a known resource with this type.
-      const resource = resourceDatabase.getResourceByType(
+      const resourceByType = resourceDatabase.getResourceByType(
         resourceReference?.type,
         errorLocation
       );
-      if (resource) {
-        uniqueResources[resource.name] = resource;
+      if (!resourceByType || !resourceByType.pattern) continue;
+      // For multi pattern resources, we look up the type first, and get the [pattern] from resource,
+      // look up pattern map for all resources.
+      for (const pattern of resourceByType!.pattern!) {
+        const resourceByPattern = resourceDatabase.getResourceByPattern(
+          pattern
+        );
+        if (!resourceByPattern) continue;
+        uniqueResources[resourceByPattern.name] = resourceByPattern;
       }
     }
   }

typescript/src/schema/resourceDatabase.ts

@@ -55,32 +55,47 @@ export class ResourceDatabase {
       }
       return;
     }
-    const name = arr[1];
-
-    const pattern = resource.pattern;
-    if (!pattern?.[0]) {
+    const patterns = resource.pattern;
+    if (!patterns?.[0]) {
       if (errorLocation) {
         console.warn(
           `Warning: ${errorLocation} refers to a resource which does not have a proper pattern: ${resource}`
         );
       }
       return;
     }
-    const params = pattern[0].match(/{[a-zA-Z_]+(?:=.*?)?}/g) || [];
-    for (let i = 0; i < params.length; i++) {
-      params[i] = params[i].replace(/{([a-zA-Z_]+).*/, '$1');
-    }
-
-    const resourceDescriptor: ResourceDescriptor = Object.assign(
-      {
+    const multiPattern = patterns!.length > 1;
+    // only one pattern exists for the resource.
+    if (!multiPattern) {
+      const name = arr![1];
+      const params = this.getParams(patterns![0]);
+      const resourceDescriptor = this.getResourceDescriptor(
         name,
         params,
-      },
-      resource
-    );
-
-    this.patterns[pattern?.[0]] = resourceDescriptor;
-    this.types[resourceDescriptor.type!] = resourceDescriptor;
+        resource
+      );
+      this.patterns[patterns?.[0]] = resourceDescriptor;
+      this.types[resourceDescriptor.type!] = resourceDescriptor;
+    }
+    // resource: {name, type, pattern: [p1, p2]}
+    // register resource does: in type map {type: { name, type, pattern: [p1, p2]} }
+    //                         in pattern map {p1: { name1, type, p1} , p2: { name2, type, p2}}
+    else {
+      for (const pattern of patterns!) {
+        const params = this.getParams(pattern);
+        const name = params.map(r => r.toPascalCase()).join('');
+        let resourceDescriptor: ResourceDescriptor = {
+          name,
+          params,
+          pattern: [pattern],
+          type: resource.type,
+        };
+        this.patterns[pattern] = resourceDescriptor;
+        resourceDescriptor = this.getResourceDescriptor(name, params, resource);
+        if (this.types[resource.type]) continue;
+        this.types[resource.type] = resourceDescriptor;
+      }
+    }
   }
 
   getResourceByType(
@@ -155,4 +170,26 @@ export class ResourceDatabase {
 
     return result;
   }
+
+  private getParams(pattern: string): string[] {
+    const params = pattern.match(/{[a-zA-Z_]+(?:=.*?)?}/g) || [];
+    for (let i = 0; i < params.length; i++) {
+      params[i] = params[i].replace(/{([a-zA-Z_]+).*/, '$1');
+    }
+    return params;
+  }
+  private getResourceDescriptor(
+    name: string,
+    params: string[],
+    resource: plugin.google.api.IResourceDescriptor
+  ): ResourceDescriptor {
+    const resourceDescriptor = Object.assign(
+      {
+        name,
+        params,
+      },
+      resource
+    );
+    return resourceDescriptor;
+  }
 }