wip

Author: ryukzak

src/api/writer-generator/mustache.ts

@@ -25,6 +25,7 @@ import type {
 } from "@mustache/types";
 import type { TypeSchemaIndex } from "@root/typeschema/utils";
 import type { CodegenLogger } from "@root/utils/codegen-logger";
+import { type Identifier, isComplexTypeIdentifier, isResourceIdentifier, type TypeSchema } from "@typeschema/types";
 import { default as Mustache } from "mustache";
 import { FileSystemWriter, type FileSystemWriterOptions } from "./writer";
 
@@ -125,6 +126,27 @@ function runCommand(cmd: string, args: string[] = [], options = {}) {
     });
 }
 
+const checkFilter = (type: Identifier, filters: MustacheFilter): boolean => {
+    if (type.kind !== "complex-type" && type.kind !== "resource") {
+        return true;
+    }
+    if (!filters[type.kind as "resource" | "complexType"]) {
+        return true;
+    }
+    const whitelist = filters[type.kind as "resource" | "complexType"]?.whitelist ?? [];
+    const blacklist = filters[type.kind as "resource" | "complexType"]?.blacklist ?? [];
+    if (!whitelist.length && !blacklist.length) {
+        return true;
+    }
+    if (blacklist.find((pattern: any) => type.name.match(pattern as any))) {
+        return false;
+    }
+    if (whitelist.find((pattern: any) => type.name.match(pattern as any))) {
+        return true;
+    }
+    return whitelist.length === 0;
+};
+
 export class MustacheGenerator extends FileSystemWriter<MustacheGeneratorOptions> {
     private readonly templateFileCache: TemplateFileCache;
     private readonly nameGenerator: NameGenerator;
@@ -151,13 +173,19 @@ export class MustacheGenerator extends FileSystemWriter<MustacheGeneratorOptions
             resourcesByUri: {},
             complexTypesByUri: {},
         };
-        schemaLoaderFacade
-            .getComplexTypes()
+        tsIndex
+            .collectComplexTypes()
+            .map((i) => i.identifier)
+            .filter((i) => checkFilter(i, this.opts.filters))
+            .sort((a, b) => a.url.localeCompare(b.url))
             .map((typeRef) => modelFactory.createComplexType(typeRef, cache))
             .forEach(this._renderComplexType.bind(this));
 
-        schemaLoaderFacade
-            .getResources()
+        tsIndex
+            .collectResources()
+            .map((i) => i.identifier)
+            .filter((i) => checkFilter(i, this.opts.filters))
+            .sort((a, b) => a.url.localeCompare(b.url))
             .map((typeRef) => modelFactory.createResource(typeRef, cache))
             .forEach(this._renderResource.bind(this));