add support for a types array

src/Config.ts

@@ -13,6 +13,11 @@ export interface Config {
      */
     type?: string;
 
+    /**
+     * Array of type names to generate schemas for. Cannot be used with --type.
+     */
+    types?: string[];
+
     /**
      * Minify the output JSON schema (no whitespace).
      * When false, the schema is pretty-printed with 2-space indentation.
@@ -143,7 +148,10 @@ export type CompletedConfig = Config & typeof DEFAULT_CONFIG;
 
 export type FunctionOptions = "fail" | "comment" | "hide";
 
-export const DEFAULT_CONFIG: Omit<Required<Config>, "path" | "type" | "schemaId" | "tsconfig" | "tsProgram"> = {
+export const DEFAULT_CONFIG: Omit<
+    Required<Config>,
+    "path" | "type" | "types" | "schemaId" | "tsconfig" | "tsProgram"
+> = {
     expose: "export",
     topRef: true,
     jsDoc: "extended",

src/SchemaGenerator.ts

@@ -20,7 +20,7 @@ export class SchemaGenerator {
         protected readonly config?: Config,
     ) {}
 
-    public createSchema(fullName?: string): Schema {
+    public createSchema(fullName?: string | string[]): Schema {
         const rootNodes = this.getRootNodes(fullName);
         return this.createSchemaFromNodes(rootNodes);
     }
@@ -60,9 +60,10 @@ export class SchemaGenerator {
         };
     }
 
-    protected getRootNodes(fullName: string | undefined): ts.Node[] {
+    protected getRootNodes(fullName: string | string[] | undefined): ts.Node[] {
         if (fullName && fullName !== "*") {
-            return [this.findNamedNode(fullName)];
+            const fullNameArr = Array.isArray(fullName) ? fullName : [fullName];
+            return fullNameArr.map((name) => this.findNamedNode(name));
         }
 
         const rootFileNames = this.program.getRootFileNames();

test/config.test.ts

@@ -26,7 +26,7 @@ const basePath = "test/config";
 
 function assertSchema(
     name: string,
-    userConfig: Config & { type: string },
+    userConfig: (Config & { type: string }) | (Config & { types: string[] }),
     tsconfig?: boolean,
     formatterAugmentor?: FormatterAugmentor,
     parserAugmentor?: ParserAugmentor,
@@ -51,7 +51,7 @@ function assertSchema(
             config,
         );
 
-        const schema = generator.createSchema(config.type);
+        const schema = generator.createSchema(config.type ?? config.types);
         const schemaFile = resolve(`${basePath}/${name}/schema.json`);
 
         if (process.env.UPDATE_SCHEMA) {
@@ -390,6 +390,20 @@ describe("config", () => {
         }),
     );
 
+    it(
+        "multiple-types",
+        assertSchema("multiple-types", {
+            types: ["MyObject1", "MyObject2"],
+        }),
+    );
+
+    it(
+        "multiple-types-all",
+        assertSchema("multiple-types-all", {
+            types: ["MyObject1", "MyObject2", "Object1Prop", "Object2Prop"],
+        }),
+    );
+
     it(
         "mapped-intersection",
         assertSchema("mapped-intersection", {

test/config/multiple-types-all/main.ts

@@ -0,0 +1,30 @@
+type NonExportedType = {
+    misc: number;
+};
+
+export type ExportedType = {
+    val: string;
+    val2: NonExportedType;
+};
+
+export interface ExportedInterface {
+    val: string;
+}
+
+export type Object1Prop = {
+    name: string;
+};
+
+export type Object2Prop = {
+    description: string;
+};
+
+export type MyObject1 = {
+    id: number;
+    bar: Object1Prop;
+};
+
+export type MyObject2 = {
+    idStr: string;
+    baz: Object2Prop;
+};

test/config/multiple-types-all/schema.json

@@ -0,0 +1,61 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "definitions": {
+    "MyObject1": {
+      "properties": {
+        "id": {
+          "type": "number"
+        },
+        "bar": {
+          "$ref": "#/definitions/Object1Prop"
+        }
+      },
+      "required": [
+        "id",
+        "bar"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    },
+    "MyObject2": {
+      "properties": {
+        "idStr": {
+          "type": "string"
+        },
+        "baz": {
+          "$ref": "#/definitions/Object2Prop"
+        }
+      },
+      "required": [
+        "idStr",
+        "baz"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    },
+    "Object1Prop": {
+      "properties": {
+        "name": {
+          "type": "string"
+        }
+      },
+      "required": [
+        "name"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    },
+    "Object2Prop": {
+      "properties": {
+        "description": {
+          "type": "string"
+        }
+      },
+      "required": [
+        "description"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    }
+  }
+}

test/config/multiple-types/main.ts

@@ -0,0 +1,31 @@
+type NonExportedType = {
+    misc: number;
+};
+
+export type ExportedType = {
+    val: string;
+    val2: NonExportedType;
+};
+
+export interface ExportedInterface {
+    val: string;
+}
+
+// Exported, so we include it as a root node
+export type Object1Prop = {
+    name: string;
+};
+
+type Object2Prop = {
+    description: string;
+};
+
+export type MyObject1 = {
+    id: number;
+    bar: Object1Prop;
+};
+
+export type MyObject2 = {
+    idStr: string;
+    baz: Object2Prop;
+};

test/config/multiple-types/schema.json

@@ -0,0 +1,58 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "definitions": {
+    "MyObject1": {
+      "properties": {
+        "id": {
+          "type": "number"
+        },
+        "bar": {
+          "$ref": "#/definitions/Object1Prop"
+        }
+      },
+      "required": [
+        "id",
+        "bar"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    },
+    "MyObject2": {
+      "properties": {
+        "idStr": {
+          "type": "string"
+        },
+        "baz": {
+          "properties": {
+            "description": {
+              "type": "string"
+            }
+          },
+          "required": [
+            "description"
+          ],
+          "type": "object",
+          "additionalProperties": false
+        }
+      },
+      "required": [
+        "idStr",
+        "baz"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    },
+    "Object1Prop": {
+      "properties": {
+        "name": {
+          "type": "string"
+        }
+      },
+      "required": [
+        "name"
+      ],
+      "type": "object",
+      "additionalProperties": false
+    }
+  }
+}