feat: Documentation.toJson() can handle nested submodules

cdklabs · Nov 2, 2023 · fb07563 · fb07563
1 parent e278bc5
commit fb07563
Show file tree

Hide file tree

Showing 5 changed files with 7,936 additions and 979 deletions.
diff --git a/src/docgen/view/documentation.ts b/src/docgen/view/documentation.ts
@@ -315,19 +315,25 @@ export class Documentation {
    * Lookup a submodule by a submodule name.
    */
   private findSubmodule(assembly: reflect.Assembly, submodule: string): reflect.Submodule {
-    const submodules = assembly.submodules.filter(
-      (s) => s.name === submodule,
-    );
+    type ReflectSubmodules = typeof assembly.submodules;
+    return recurse(submodule.split('.'), assembly.submodules);
 
-    if (submodules.length === 0) {
-      throw new Error(`Submodule ${submodule} not found in assembly ${assembly.name}@${assembly.version}`);
-    }
+    function recurse(names: string[], submodules: ReflectSubmodules): reflect.Submodule {
+      const [head, ...tail] = names;
+      const found = submodules.filter(
+        (s) => s.name === head,
+      );
 
-    if (submodules.length > 1) {
-      throw new Error(`Found multiple submodules with name: ${submodule} in assembly ${assembly.name}@${assembly.version}`);
-    }
+      if (found.length === 0) {
+        throw new Error(`Submodule ${submodule} not found in assembly ${assembly.name}@${assembly.version}`);
+      }
 
-    return submodules[0];
+      if (found.length > 1) {
+        throw new Error(`Found multiple submodules with name: ${submodule} in assembly ${assembly.name}@${assembly.version}`);
+      }
+
+      return tail.length === 0 ? found[0] : recurse(tail, found[0].submodules);
+    }
   }
 
   private async createAssembly(