@nestia/sdk 2.4.7 → 2.5.0-dev.20240128

package/src/generates/internal/SdkFileProgrammer.ts

@@ -1,11 +1,14 @@
 import fs from "fs";
+import { format } from "prettier";
+import ts from "typescript";
 
 import { INestiaConfig } from "../../INestiaConfig";
 import { IRoute } from "../../structures/IRoute";
 import { ImportDictionary } from "../../utils/ImportDictionary";
 import { MapUtil } from "../../utils/MapUtil";
-import { SdkFunctionProgrammer } from "./SdkFunctionProgrammer";
+import { NodeUtil } from "../../utils/NodeUtil";
 import { SdkRouteDirectory } from "./SdkRouteDirectory";
+import { SdkRouteProgrammer } from "./SdkRouteProgrammer";
 
 export namespace SdkFileProgrammer {
   /* ---------------------------------------------------------
@@ -51,12 +54,21 @@ export namespace SdkFileProgrammer {
       } catch {}
 
       // ITERATE CHILDREN
-      const content: string[] = [];
+      const statements: ts.Statement[] = [];
       for (const [key, value] of directory.children) {
         await iterate(config)(value)(`${outDir}/${key}`);
-        content.push(`export * as ${key} from "./${key}";`);
+        statements.push(
+          ts.factory.createExportDeclaration(
+            undefined,
+            false,
+            ts.factory.createNamespaceExport(ts.factory.createIdentifier(key)),
+            ts.factory.createStringLiteral(`./${key}`),
+            undefined,
+          ),
+        );
       }
-      if (content.length && directory.routes.length) content.push("");
+      if (statements.length && directory.routes.length)
+        statements.push(NodeUtil.enter());
 
       // ITERATE ROUTES
       const importer: ImportDictionary = new ImportDictionary(
@@ -80,17 +92,19 @@ export namespace SdkFileProgrammer {
                 instance,
                 type: true,
               });
-
-        content.push(SdkFunctionProgrammer.generate(config)(importer)(route));
-        if (i !== directory.routes.length - 1) content.push("");
+        statements.push(
+          ...SdkRouteProgrammer.generate(config)(importer)(route),
+        );
+        if (i !== directory.routes.length - 1)
+          statements.push(NodeUtil.enter());
       });
 
       // FINALIZE THE CONTENT
       if (directory.routes.length !== 0)
-        content.push(
-          importer.toScript(outDir),
-          "",
-          ...content.splice(0, content.length),
+        statements.push(
+          ...importer.toStatements(outDir),
+          ...(!importer.empty() && statements.length ? [NodeUtil.enter()] : []),
+          ...statements.splice(0, statements.length),
         );
 
       const script: string =
@@ -100,7 +114,22 @@ export namespace SdkFileProgrammer {
         " * @nestia Generated by Nestia - https://github.com/samchon/nestia \n" +
         " */\n" +
         "//================================================================\n" +
-        content.join("\n");
-      await fs.promises.writeFile(importer.file, script, "utf8");
+        ts
+          .createPrinter()
+          .printFile(
+            ts.factory.createSourceFile(
+              statements,
+              ts.factory.createToken(ts.SyntaxKind.EndOfFileToken),
+              ts.NodeFlags.None,
+            ),
+          );
+      const beautify = async () => {
+        try {
+          return await format(script, { parser: "typescript" });
+        } catch {
+          return script;
+        }
+      };
+      await fs.promises.writeFile(importer.file, await beautify(), "utf8");
     };
 }