feat: only delete files after new files have been generated with clea…

…n option
elyukai · Nov 27, 2024 · cf0433e · cf0433e
1 parent 355a290
commit cf0433e
Showing 1 changed file with 76 additions and 47 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -1,5 +1,5 @@
-import { Dirent, existsSync } from "node:fs"
-import { mkdir, readdir, rm, writeFile } from "node:fs/promises"
+import { Dirent } from "node:fs"
+import { mkdir, readdir, rm, stat, writeFile } from "node:fs/promises"
 import {
   basename,
   dirname,
@@ -169,14 +169,20 @@ export const generate = async (options: GeneratorOptions): Promise<void> => {
   // KEEP ALWAYS, SIDE EFFECT: it fills the parent references of nodes
   const checker = program.getTypeChecker()
 
-  for (const { targetDir, clean: cleanSingle } of outputs) {
-    if ((cleanSingle ?? clean) && existsSync(targetDir)) {
-      await rm(targetDir, { recursive: true })
-    }
-
+  for (const { targetDir } of outputs) {
     await mkdir(targetDir, { recursive: true })
   }
 
+  const filesToCleanUp = await Promise.all(
+    outputs.map(({ targetDir }) => readdir(targetDir, { recursive: true }))
+  )
+
+  const removeFileFromCleanUp = (outputIndex: number, path: string) => {
+    filesToCleanUp[outputIndex] = filesToCleanUp[outputIndex]!.filter(
+      (existingPath) => existingPath !== path
+    )
+  }
+
   const rootFiles = program
     .getSourceFiles()
     .filter((file) => tsFiles.includes(file.fileName))
@@ -225,50 +231,54 @@ export const generate = async (options: GeneratorOptions): Promise<void> => {
           }
         }
 
-        for (const { targetDir, renderer } of outputs) {
-          const {
-            transformer,
-            fileExtension,
-            resolveTypeParameters = false,
-          } = renderer
-          const outputDir = join(targetDir, dir)
-
-          const outputAbsoluteFilePath = format({
-            dir: outputDir,
-            name,
-            ext: fileExtension,
-          })
-
-          const outputRelativeFilePath = relative(
-            targetDir,
-            outputAbsoluteFilePath
-          )
-
-          const meta: MetaInformation = {
-            absolutePath: outputAbsoluteFilePath,
-            relativePath: outputRelativeFilePath,
-          }
-
-          const output = resolveTypeParameters
-            ? resolvedAst !== undefined
-              ? transformer(resolvedAst, meta)
-              : undefined
-            : transformer(ast, meta)
+        await Promise.all(
+          outputs.map(async ({ targetDir, renderer }, outputIndex) => {
+            const {
+              transformer,
+              fileExtension,
+              resolveTypeParameters = false,
+            } = renderer
+            const outputDir = join(targetDir, dir)
+
+            const outputAbsoluteFilePath = format({
+              dir: outputDir,
+              name,
+              ext: fileExtension,
+            })
+
+            const outputRelativeFilePath = relative(
+              targetDir,
+              outputAbsoluteFilePath
+            )
 
-          if (output === undefined) {
-            if (verbose) {
-              console.log(`-> empty output`)
+            const meta: MetaInformation = {
+              absolutePath: outputAbsoluteFilePath,
+              relativePath: outputRelativeFilePath,
             }
-          } else {
-            await mkdir(outputDir, { recursive: true })
-            await writeFile(outputAbsoluteFilePath, output, "utf-8")
-            outputFilesCount++
 
-            if (verbose) {
-              console.log(`-> ${outputAbsoluteFilePath}`)
+            const output = resolveTypeParameters
+              ? resolvedAst !== undefined
+                ? transformer(resolvedAst, meta)
+                : undefined
+              : transformer(ast, meta)
+
+            if (output === undefined) {
+              if (verbose) {
+                console.log(`-> empty output`)
+              }
+            } else {
+              await mkdir(outputDir, { recursive: true })
+              await writeFile(outputAbsoluteFilePath, output, "utf-8")
+              outputFilesCount++
+
+              removeFileFromCleanUp(outputIndex, outputRelativeFilePath)
+
+              if (verbose) {
+                console.log(`-> ${outputAbsoluteFilePath}`)
+              }
             }
-          }
-        }
+          })
+        )
       } else {
         if (verbose) {
           console.log(`file does not contain renderable content`)
@@ -286,6 +296,25 @@ export const generate = async (options: GeneratorOptions): Promise<void> => {
     }
   }
 
+  await Promise.all(
+    outputs.map(async ({ targetDir, clean: cleanSingle }, outputIndex) => {
+      if (cleanSingle ?? clean) {
+        await Promise.all(
+          filesToCleanUp[outputIndex]!.map(async (filePath) => {
+            const absolutePath = join(targetDir, filePath)
+            const stats = await stat(absolutePath)
+            if (
+              !stats.isDirectory() ||
+              (await readdir(absolutePath)).length === 0
+            ) {
+              await rm(join(targetDir, filePath), { recursive: true })
+            }
+          })
+        )
+      }
+    })
+  )
+
   if (verbose) {
     console.log(
       `Generating ${outputFilesCount} output file(s) finished successfully.`