From c5e59d6d32d585b73f8da92adfc0d591bb50d24f Mon Sep 17 00:00:00 2001
From: Pionxzh <pionxzh@csie.io>
Date: Fri, 15 Dec 2023 01:59:46 +0800
Subject: [PATCH] feat(cli): generate `perf.json` when perf is enabled

---
 packages/cli/src/cli.ts | 33 ++++++++++++++++++++++++---------
 1 file changed, 24 insertions(+), 9 deletions(-)

diff --git a/packages/cli/src/cli.ts b/packages/cli/src/cli.ts
index d6f21506..4006f3c6 100644
--- a/packages/cli/src/cli.ts
+++ b/packages/cli/src/cli.ts
@@ -406,7 +406,9 @@ async function interactive({
         outro(`Output directory: ${c.green(getRelativePath(cwd, outputDir))}`)
 
         if (perf) {
-            handlePerfMeasurements(measurements)
+            printPerfStats(measurements)
+
+            writePerfStats(measurements, path.join(outputBase, 'perf.json'))
         }
     }
 
@@ -456,19 +458,19 @@ async function nonInteractive(features: Feature[], {
         return process.exit(1)
     }
 
-    const output = _output ?? defaultOutputBase
+    const outputBase = _output ?? defaultOutputBase
     const singleFeature = features.length === 1
-    const unpackerOutput = _unpackerOutput ?? (singleFeature ? output : path.join(output, defaultUnpackerOutputFolder))
-    const unminifyOutput = _unminifyOutput ?? (singleFeature ? output : path.join(output, defaultUnminifyOutputFolder))
+    const unpackerOutput = _unpackerOutput ?? (singleFeature ? outputBase : path.join(outputBase, defaultUnpackerOutputFolder))
+    const unminifyOutput = _unminifyOutput ?? (singleFeature ? outputBase : path.join(outputBase, defaultUnminifyOutputFolder))
 
-    if (!isPathInside(cwd, output)) {
+    if (!isPathInside(cwd, outputBase)) {
         log.error('Output directory must be inside the current working directory')
         return process.exit(1)
     }
 
     if (!force) {
-        if (fsa.existsSync(output)) {
-            log.error(`Output directory already exists at ${c.green(output)}. Pass ${c.green('--force')} to overwrite`)
+        if (fsa.existsSync(outputBase)) {
+            log.error(`Output directory already exists at ${c.green(outputBase)}. Pass ${c.green('--force')} to overwrite`)
             return process.exit(1)
         }
 
@@ -566,7 +568,9 @@ async function nonInteractive(features: Feature[], {
         outro(`Output directory: ${c.green(relativeOutputPath)}`)
 
         if (perf) {
-            handlePerfMeasurements(measurements)
+            printPerfStats(measurements)
+
+            writePerfStats(measurements, path.join(outputBase, 'perf.json'))
         }
     }
 }
@@ -578,7 +582,7 @@ function formatElapsed(elapsed: number) {
     return `${~~(elapsed / 1000 / 60 / 60)}h${~~((elapsed / 1000 / 60) % 60)}m${~~((elapsed / 1000) % 60)}s`
 }
 
-function handlePerfMeasurements(measurements: Measurement[]) {
+function printPerfStats(measurements: Measurement[]) {
     const groupedByRules = measurements
         .flat()
         .reduce<Record<string, number>>((acc, { key, time }) => {
@@ -592,6 +596,17 @@ function handlePerfMeasurements(measurements: Measurement[]) {
     console.table(table, ['key', 'time'])
 }
 
+function writePerfStats(measurements: Measurement[], outputPath: string) {
+    fsa.writeJSONSync(outputPath, measurements.flat(), {
+        encoding: 'utf-8',
+        spaces: 2,
+    })
+
+    console.log()
+    console.log(`Performance statistics generated at ${c.green(getRelativePath(process.cwd(), outputPath))}`)
+    console.log()
+}
+
 function generateModuleMeta(modules: Module[]) {
     return modules.reduce<ModuleMeta>((acc, mod) => {
         acc[mod.id] = {