@nestia/sdk 3.0.0-dev.20231209 → 3.0.0-dev.20240412

package/src/generates/internal/SdkFileProgrammer.ts

@@ -1,106 +1,115 @@
-import fs from "fs";
-
-import { INestiaConfig } from "../../INestiaConfig";
-import { IRoute } from "../../structures/IRoute";
-import { ImportDictionary } from "../../utils/ImportDictionary";
-import { MapUtil } from "../../utils/MapUtil";
-import { SdkFunctionProgrammer } from "./SdkFunctionProgrammer";
-import { SdkRouteDirectory } from "./SdkRouteDirectory";
-
-export namespace SdkFileProgrammer {
-  /* ---------------------------------------------------------
-        CONSTRUCTOR
-    --------------------------------------------------------- */
-  export const generate =
-    (config: INestiaConfig) =>
-    async (routeList: IRoute[]): Promise<void> => {
-      // CONSTRUCT FOLDER TREE
-      const root: SdkRouteDirectory = new SdkRouteDirectory(null, "functional");
-      for (const route of routeList) emplace(root)(route);
-
-      // ITERATE FILES
-      await iterate(config)(root)(config.output + "/functional");
-    };
-
-  const emplace =
-    (directory: SdkRouteDirectory) =>
-    (route: IRoute): void => {
-      // OPEN DIRECTORIES
-      for (const key of route.accessors.slice(0, -1)) {
-        directory = MapUtil.take(
-          directory.children,
-          key,
-          () => new SdkRouteDirectory(directory, key),
-        );
-      }
-
-      // ADD ROUTE
-      directory.routes.push(route);
-    };
-
-  /* ---------------------------------------------------------
-        FILE ITERATOR
-    --------------------------------------------------------- */
-  const iterate =
-    (config: INestiaConfig) =>
-    (directory: SdkRouteDirectory) =>
-    async (outDir: string): Promise<void> => {
-      // CREATE A NEW DIRECTORY
-      try {
-        await fs.promises.mkdir(outDir);
-      } catch {}
-
-      // ITERATE CHILDREN
-      const content: string[] = [];
-      for (const [key, value] of directory.children) {
-        await iterate(config)(value)(`${outDir}/${key}`);
-        content.push(`export * as ${key} from "./${key}";`);
-      }
-      if (content.length && directory.routes.length) content.push("");
-
-      // ITERATE ROUTES
-      const importer: ImportDictionary = new ImportDictionary(
-        `${outDir}/index.ts`,
-      );
-      if (
-        config.simulate === true &&
-        directory.routes.some((r) => !!r.parameters.length)
-      )
-        importer.internal({
-          file: `${config.output}/utils/NestiaSimulator.ts`,
-          instance: "NestiaSimulator",
-          type: false,
-        });
-      directory.routes.forEach((route, i) => {
-        if (config.clone !== true)
-          for (const tuple of route.imports)
-            for (const instance of tuple[1])
-              importer.internal({
-                file: tuple[0],
-                instance,
-                type: true,
-              });
-
-        content.push(SdkFunctionProgrammer.generate(config)(importer)(route));
-        if (i !== directory.routes.length - 1) content.push("");
-      });
-
-      // FINALIZE THE CONTENT
-      if (directory.routes.length !== 0)
-        content.push(
-          importer.toScript(outDir),
-          "",
-          ...content.splice(0, content.length),
-        );
-
-      const script: string =
-        "/**\n" +
-        " * @packageDocumentation\n" +
-        ` * @module ${directory.module}\n` +
-        " * @nestia Generated by Nestia - https://github.com/samchon/nestia \n" +
-        " */\n" +
-        "//================================================================\n" +
-        content.join("\n");
-      await fs.promises.writeFile(importer.file, script, "utf8");
-    };
-}
+import fs from "fs";
+import ts from "typescript";
+
+import { INestiaConfig } from "../../INestiaConfig";
+import { IRoute } from "../../structures/IRoute";
+import { MapUtil } from "../../utils/MapUtil";
+import { FilePrinter } from "./FilePrinter";
+import { ImportDictionary } from "./ImportDictionary";
+import { SdkRouteDirectory } from "./SdkRouteDirectory";
+import { SdkRouteProgrammer } from "./SdkRouteProgrammer";
+
+export namespace SdkFileProgrammer {
+  /* ---------------------------------------------------------
+        CONSTRUCTOR
+    --------------------------------------------------------- */
+  export const generate =
+    (checker: ts.TypeChecker) =>
+    (config: INestiaConfig) =>
+    async (routeList: IRoute[]): Promise<void> => {
+      // CONSTRUCT FOLDER TREE
+      const root: SdkRouteDirectory = new SdkRouteDirectory(null, "functional");
+      for (const route of routeList) emplace(root)(route);
+
+      // ITERATE FILES
+      await iterate(checker)(config)(root)(config.output + "/functional");
+    };
+
+  const emplace =
+    (directory: SdkRouteDirectory) =>
+    (route: IRoute): void => {
+      // OPEN DIRECTORIES
+      for (const key of route.accessors.slice(0, -1)) {
+        directory = MapUtil.take(
+          directory.children,
+          key,
+          () => new SdkRouteDirectory(directory, key),
+        );
+      }
+
+      // ADD ROUTE
+      directory.routes.push(route);
+    };
+
+  /* ---------------------------------------------------------
+        FILE ITERATOR
+    --------------------------------------------------------- */
+  const iterate =
+    (checker: ts.TypeChecker) =>
+    (config: INestiaConfig) =>
+    (directory: SdkRouteDirectory) =>
+    async (outDir: string): Promise<void> => {
+      // CREATE A NEW DIRECTORY
+      try {
+        await fs.promises.mkdir(outDir);
+      } catch {}
+
+      // ITERATE CHILDREN
+      const statements: ts.Statement[] = [];
+      for (const [key, value] of directory.children) {
+        await iterate(checker)(config)(value)(`${outDir}/${key}`);
+        statements.push(
+          ts.factory.createExportDeclaration(
+            undefined,
+            false,
+            ts.factory.createNamespaceExport(ts.factory.createIdentifier(key)),
+            ts.factory.createStringLiteral(`./${key}`),
+            undefined,
+          ),
+        );
+      }
+      if (statements.length && directory.routes.length)
+        statements.push(FilePrinter.enter());
+
+      // ITERATE ROUTES
+      const importer: ImportDictionary = new ImportDictionary(
+        `${outDir}/index.ts`,
+      );
+      directory.routes.forEach((route, i) => {
+        if (config.clone !== true)
+          for (const tuple of route.imports)
+            for (const instance of tuple[1])
+              importer.internal({
+                file: tuple[0],
+                instance,
+                type: true,
+              });
+        statements.push(
+          ...SdkRouteProgrammer.generate(checker)(config)(importer)(route),
+        );
+        if (i !== directory.routes.length - 1)
+          statements.push(FilePrinter.enter());
+      });
+
+      // FINALIZE THE CONTENT
+      if (directory.routes.length !== 0)
+        statements.push(
+          ...importer.toStatements(outDir),
+          ...(!importer.empty() && statements.length
+            ? [FilePrinter.enter()]
+            : []),
+          ...statements.splice(0, statements.length),
+        );
+      await FilePrinter.write({
+        location: importer.file,
+        statements,
+        top:
+          "/**\n" +
+          " * @packageDocumentation\n" +
+          ` * @module ${directory.module}\n` +
+          " * @nestia Generated by Nestia - https://github.com/samchon/nestia \n" +
+          " */\n" +
+          "//================================================================\n",
+      });
+    };
+}