remove files from dist once deleted from src

Author: xettri

src/core/postbuild.ts

@@ -48,6 +48,73 @@ export async function copyRequiredFiles() {
   await Promise.all(task);
 }
 
+// function to unlink files from build path, this expect files paths exist in src
+// this used when watch mode is enabled and we need to unlink files from build path
+// when file is deleted in src path
+export async function unlinkFilesFormBuild(
+  files: string | string[],
+  isDir: boolean = false,
+) {
+  const { srcPath, lib } = getConfig();
+
+  const cjsPath = lib?.cjs?.output?.path;
+  const esmPath = lib?.esm?.output?.path;
+
+  const hasCJS = fse.pathExistsSync(cjsPath);
+  const hasESM = fse.pathExistsSync(esmPath);
+
+  const tasks: Promise<void>[] = [];
+  const isTS = (ext: string) => ['.ts', '.tsx', '.mts'].includes(ext);
+
+  function removeFile(file: string) {
+    const relativePath = path.relative(srcPath, file);
+
+    // helper function to remove from both CJS and ESM paths
+    const removeFromBuildPaths = (targetPath: string) => {
+      if (hasCJS) {
+        const cjsTarget = path.resolve(cjsPath, targetPath);
+        if (fse.existsSync(cjsTarget)) {
+          tasks.push(fse.remove(cjsTarget));
+        }
+      }
+
+      if (hasESM) {
+        const esmTarget = path.resolve(esmPath, targetPath);
+        if (fse.existsSync(esmTarget)) {
+          tasks.push(fse.remove(esmTarget));
+        }
+      }
+    };
+
+    if (isDir) {
+      // for directories, remove directly from both CJS and ESM paths
+      removeFromBuildPaths(relativePath);
+    } else {
+      // for files, handle extension conversion
+      const dirName = path.dirname(relativePath);
+      const fileName = path.basename(relativePath);
+      const ext = path.extname(fileName);
+
+      const buildFileName =
+        // in case file is ts group then we need to unlink the js file
+        // but for dts files we do not convert to js
+        isTS(ext) && !fileName.endsWith('.d.ts')
+          ? fileName.replace(ext, '.js')
+          : fileName;
+
+      removeFromBuildPaths(path.join(dirName, buildFileName));
+    }
+  }
+
+  if (Array.isArray(files)) {
+    files.forEach(removeFile);
+  } else {
+    removeFile(files);
+  }
+
+  await Promise.all(tasks);
+}
+
 async function postbuild(getBuildTime: ReturnType<typeof createTimer>) {
   const { root, srcPath, buildPath, lib } = getConfig();
 

src/core/watch/worker.ts

@@ -1,9 +1,8 @@
 import { parentPort, isMainThread } from 'node:worker_threads';
-import { globSync } from 'fast-glob';
 import { log, createTimer, createFileHash } from '@/utils';
-import { getConfig, initConfig } from '@/config';
+import { initConfig } from '@/config';
 import build from '@/core/build';
-import runPostbuild from '@/core/postbuild';
+import runPostbuild, { unlinkFilesFormBuild } from '@/core/postbuild';
 import type { CliProps } from '@/types';
 
 const fileHashes = new Map<string, string>();
@@ -18,17 +17,19 @@ const actionOnWatch = async (
 ) => {
   // init config on first run to prevent empty config
   await initConfig(cliProps);
-  const config = getConfig();
 
   // handle file hash management
   if (event === 'unlink') {
     fileHashes.delete(path);
+    await unlinkFilesFormBuild([path]);
   } else if (event === 'unlinkDir') {
-    globSync(`${path}/**/*`, {
-      ignore: config.ignore,
-    }).forEach(file => {
-      fileHashes.delete(file);
-    });
+    // iterate through fileHashes to find files that were in the deleted directory
+    for (const [filePath] of Array.from(fileHashes.entries())) {
+      if (filePath.startsWith(path + '/')) {
+        fileHashes.delete(filePath);
+      }
+    }
+    await unlinkFilesFormBuild(path, true);
   }
 
   async function runBuildProcess() {