fix: prune unreachable definitions when --type "*" is used with multiple exports (vega#2284)

Co-authored-by: Arthur Fiorette <47537704+arthurfiorette@users.noreply.github.com>

Author: alexchexes

src/SchemaGenerator.ts

@@ -31,8 +31,8 @@ export class SchemaGenerator {
             rootType: this.nodeParser.createType(rootNode, new Context()),
         }));
 
-        const rootTypeDefinition =
-            roots.length === 1 ? this.getRootTypeDefinition(roots[0].rootType, roots[0].rootNode) : undefined;
+        const rootTypeDefinitions = roots.map((root) => this.getRootTypeDefinition(root.rootType, root.rootNode));
+        const rootTypeDefinition = rootTypeDefinitions.length === 1 ? rootTypeDefinitions[0] : undefined;
         const definitions: StringMap<Definition> = {};
 
         for (const root of roots) {
@@ -47,7 +47,10 @@ export class SchemaGenerator {
             }
         }
 
-        const reachableDefinitions = removeUnreachable(rootTypeDefinition, definitions);
+        const reachableDefinitions = rootTypeDefinitions.reduce<StringMap<Definition>>(
+            (acc, def) => Object.assign(acc, removeUnreachable(def, definitions)),
+            {},
+        );
 
         return {
             ...(this.config?.schemaId ? { $id: this.config.schemaId } : {}),
@@ -229,11 +232,18 @@ export class SchemaGenerator {
                 return;
             }
 
-            // export { variable } clauses
+            if (node.exportClause) {
+                // export { Foo } from './lib' or export { Foo };
+                // export * as Foo from './lib' should not import all exports
+                ts.forEachChild(node.exportClause, (subnode) => this.inspectNode(subnode, typeChecker, allTypes));
+                return;
+            }
+
             if (!node.moduleSpecifier) {
                 return;
             }
 
+            // export * from './lib'
             const symbol = typeChecker.getSymbolAtLocation(node.moduleSpecifier);
 
             // should never hit this (maybe type error in user's code)

test/valid-data-type.test.ts

@@ -151,4 +151,8 @@ describe("valid-data-type", () => {
     it("promise-extensions", assertValidSchema("promise-extensions", "*"));
 
     it("export-star", assertValidSchema("export-star", "*", undefined, { mainTsOnly: true }));
+    it(
+        "export-star-prune-unreachable",
+        assertValidSchema("export-star-prune-unreachable", "*", undefined, { mainTsOnly: true }),
+    );
 });

test/valid-data/export-star-prune-unreachable/dep.ts

@@ -0,0 +1,5 @@
+export interface SomeInterface {
+    foo?: string;
+}
+
+export type DepType = string;

test/valid-data/export-star-prune-unreachable/dep2.ts

@@ -0,0 +1 @@
+export type DepType2 = string;

test/valid-data/export-star-prune-unreachable/main.ts

@@ -0,0 +1,14 @@
+import type { SomeInterface } from "./dep";
+export { DepType } from "./dep";
+export * from "./dep2";
+
+export type MyType = string;
+
+export interface MyObject extends SomeInterface {
+    bar?: number;
+    baz?: Internal;
+}
+
+interface Internal {
+    nested?: boolean;
+}

test/valid-data/export-star-prune-unreachable/schema.json

@@ -0,0 +1,35 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "definitions": {
+    "DepType": {
+      "type": "string"
+    },
+    "DepType2": {
+      "type": "string"
+    },
+    "MyObject": {
+      "additionalProperties": false,
+      "properties": {
+        "bar": {
+          "type": "number"
+        },
+        "baz": {
+          "additionalProperties": false,
+          "properties": {
+            "nested": {
+              "type": "boolean"
+            }
+          },
+          "type": "object"
+        },
+        "foo": {
+          "type": "string"
+        }
+      },
+      "type": "object"
+    },
+    "MyType": {
+      "type": "string"
+    }
+  }
+}