fix: add api resource option to resourceDatabase (#211)

* add api resource option to resourceDatabase * lint Co-authored-by: Alexander Fenster <[email protected]>
googleapis · Jan 31, 2020 · 7ea7034 · 7ea7034
1 parent 90177c6
commit 7ea7034
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/typescript/src/schema/proto.ts b/typescript/src/schema/proto.ts
@@ -528,6 +528,21 @@ function augmentService(
   const uniqueResources: { [name: string]: ResourceDescriptor } = {};
   for (const property of Object.keys(messages)) {
     const errorLocation = `service ${service.name} message ${property}`;
+    // take the option['.google.api.resource'] of the message as resource, add it to resourceDatabase id it's not there.
+    const descriptorProto = messages[property];
+    if (
+      descriptorProto.options &&
+      descriptorProto.options['.google.api.resource']
+    ) {
+      const resource = descriptorProto.options['.google.api.resource'];
+      if (!resourceDatabase.getResourceByType(resource.type)) {
+        resourceDatabase.registerResource(resource);
+        const registeredResource = resourceDatabase.getResourceByType(
+          resource.type
+        )!;
+        uniqueResources[registeredResource.name] = registeredResource;
+      }
+    }
     for (const fieldDescriptor of messages[property].field ?? []) {
       // note: ResourceDatabase can accept `undefined` values, so we happily use optional chaining here.
       const resourceReference =