fix: 修复主进程编译函数

ch1ny · Oct 26, 2023 · fbc02c2 · fbc02c2
1 parent c3a974e
commit fbc02c2
Showing 1 changed file with 21 additions and 7 deletions.
diff --git a/src/creta/src/cli/utils/build.ts b/src/creta/src/cli/utils/build.ts
@@ -28,10 +28,24 @@ export const buildPreload = async () =>
 		path.resolve(scriptsCwd, 'src', 'preload', 'tsconfig.json')
 	);
 
-export const buildMain = async () =>
-	tscBuild(
-		(await fs.promises.readdir(path.resolve(scriptsCwd, 'src', 'main')))
-			.filter((file) => file.endsWith('.js') || file.endsWith('.ts'))
-			.map((file) => path.resolve(scriptsCwd, 'src', 'main', file)),
-		path.resolve(scriptsCwd, 'src', 'main', 'tsconfig.json')
-	);
+export const buildMain = async () => {
+	const mainRootDir = path.resolve(scriptsCwd, 'src', 'main');
+	const nextDirList = [mainRootDir];
+	const filesToBuild: string[] = [];
+
+	let nextDir: string | undefined;
+	while ((nextDir = nextDirList.shift())) {
+		const files = await fs.promises.readdir(path.resolve(nextDir));
+		files.map(async (fileName) => {
+			const filePath = path.resolve(nextDir!, fileName);
+			const fsStatus = await fs.promises.stat(filePath);
+			if (fsStatus.isDirectory()) {
+				nextDirList.push(filePath);
+			} else if (filePath.endsWith('.js') || filePath.endsWith('.ts')) {
+				filesToBuild.push(filePath);
+			}
+		});
+	}
+
+	await tscBuild(filesToBuild, path.resolve(scriptsCwd, 'src', 'main', 'tsconfig.json'));
+};