From b68f45ef597e35e228677e2c8409405122a1796c Mon Sep 17 00:00:00 2001
From: Sidharth Vinod <sidharthv96@gmail.com>
Date: Mon, 14 Aug 2023 08:27:14 +0530
Subject: [PATCH] chore: Split chunks into folders

---
 .esbuild/build.ts | 51 +++++++++++++++++++++++++++++++++++------------
 .esbuild/util.ts  | 43 ++++++++++++++++++++++++++-------------
 2 files changed, 67 insertions(+), 27 deletions(-)

diff --git a/.esbuild/build.ts b/.esbuild/build.ts
index bee13af512..c5d8217065 100644
--- a/.esbuild/build.ts
+++ b/.esbuild/build.ts
@@ -1,21 +1,48 @@
 import { build } from 'esbuild';
 import { mkdir, writeFile } from 'node:fs/promises';
-import { getBuildConfig } from './util.js';
+import { MermaidBuildOptions, defaultOptions, getBuildConfig } from './util.js';
 import { packageOptions } from '../.build/common.js';
 
 const shouldVisualize = process.argv.includes('--visualize');
 
 const buildPackage = async (entryName: keyof typeof packageOptions) => {
-  await build(getBuildConfig({ entryName, minify: false }));
-  const { metafile } = await build(
-    getBuildConfig({ entryName, minify: true, metafile: shouldVisualize })
-  );
-  if (metafile) {
-    // Upload metafile into https://esbuild.github.io/analyze/
-    await writeFile(`stats/meta-${entryName}.json`, JSON.stringify(metafile));
+  const commonOptions = { ...defaultOptions, entryName } as const;
+  const buildConfigs = [
+    // package.mjs
+    { ...commonOptions },
+    // package.min.mjs
+    {
+      ...commonOptions,
+      minify: true,
+      metafile: shouldVisualize,
+    },
+    // package.core.mjs
+    { ...commonOptions, core: true },
+  ];
+
+  if (entryName === 'mermaid') {
+    const iifeOptions: MermaidBuildOptions = { ...commonOptions, format: 'iife' };
+    buildConfigs.push(
+      // mermaid.js
+      { ...iifeOptions },
+      // mermaid.min.js
+      { ...iifeOptions, minify: true, metafile: shouldVisualize }
+    );
+  }
+
+  const results = await Promise.all(buildConfigs.map((option) => build(getBuildConfig(option))));
+
+  if (shouldVisualize) {
+    for (const { metafile } of results) {
+      if (!metafile) {
+        continue;
+      }
+      const fileName = Object.keys(metafile.outputs)
+        .filter((key) => key.includes('.min') && key.endsWith('js'))[0]
+        .replace('dist/', '');
+      await writeFile(`stats/${fileName}.meta.json`, JSON.stringify(metafile));
+    }
   }
-  await build(getBuildConfig({ entryName, minify: false, core: true }));
-  await build(getBuildConfig({ entryName, minify: true, format: 'iife' }));
 };
 
 const handler = (e) => {
@@ -26,9 +53,7 @@ const handler = (e) => {
 const main = async () => {
   await mkdir('stats').catch(() => {});
   const packageNames = Object.keys(packageOptions) as (keyof typeof packageOptions)[];
-  for (const pkg of packageNames) {
-    await buildPackage(pkg).catch(handler);
-  }
+  await Promise.allSettled(packageNames.map((pkg) => buildPackage(pkg).catch(handler)));
 };
 
 void main();
diff --git a/.esbuild/util.ts b/.esbuild/util.ts
index 249045652d..973582ae0a 100644
--- a/.esbuild/util.ts
+++ b/.esbuild/util.ts
@@ -8,14 +8,21 @@ import { jisonPlugin } from './jisonPlugin.js';
 
 const __dirname = fileURLToPath(new URL('.', import.meta.url));
 
-interface MermaidBuildOptions {
+export interface MermaidBuildOptions {
   minify: boolean;
-  core?: boolean;
-  metafile?: boolean;
-  format?: 'esm' | 'iife';
+  core: boolean;
+  metafile: boolean;
+  format: 'esm' | 'iife';
   entryName: keyof typeof packageOptions;
 }
 
+export const defaultOptions: Omit<MermaidBuildOptions, 'entryName'> = {
+  minify: false,
+  metafile: false,
+  core: false,
+  format: 'esm',
+} as const;
+
 const buildOptions = (override: BuildOptions): BuildOptions => {
   return {
     bundle: true,
@@ -32,24 +39,32 @@ const buildOptions = (override: BuildOptions): BuildOptions => {
   };
 };
 
-export const getBuildConfig = ({
-  minify,
-  core,
-  entryName,
-  metafile,
-  format,
-}: MermaidBuildOptions): BuildOptions => {
+const getFileName = (fileName: string, { core, format, minify }: MermaidBuildOptions) => {
+  if (core) {
+    fileName += '.core';
+  }
+  if (format === 'esm') {
+    fileName += '.esm';
+  }
+  if (minify) {
+    fileName += '.min';
+  }
+  return fileName;
+};
+
+export const getBuildConfig = (options: MermaidBuildOptions): BuildOptions => {
+  const { core, entryName, metafile, format } = options;
   const external: string[] = ['require', 'fs', 'path'];
   const { name, file, packageName } = packageOptions[entryName];
+  const outFileName = getFileName(name, options);
   let output: BuildOptions = buildOptions({
     absWorkingDir: resolve(__dirname, `../packages/${packageName}`),
     entryPoints: {
-      [`${name}${core ? '.core' : format === 'iife' ? '' : '.esm'}${
-        minify ? '.min' : ''
-      }`]: `src/${file}`,
+      [outFileName]: `src/${file}`,
     },
     metafile,
     logLevel: 'info',
+    chunkNames: `chunks/${outFileName}/[name]-[hash]`,
   });
 
   if (core) {