feat: support for tracking Rspack binary size (#104)

Author: chenjiahan

.github/workflows/manual.yaml

@@ -8,6 +8,7 @@ on:
         type: choice
         required: true
         options:
+          - RSPACK
           - RSBUILD
           - RSPRESS
           - RSLIB
@@ -17,6 +18,7 @@ on:
         type: choice
         required: true
         options:
+          - rspack-binary-size
           - rsbuild-react
           - rsbuild-vue3
           - rsbuild-lit
@@ -76,7 +78,7 @@ jobs:
       - name: Setup Node.js
         uses: actions/setup-node@v4
         with:
-          node-version: 18
+          node-version: 22
           cache-dependency-path: pnpm-lock.yaml
 
       - name: Activate corepack

.gitignore

@@ -49,6 +49,7 @@ temp/**/*
 rspress
 rsbuild
 rslib
+rspack
 !/cases/rspress/
 !/scripts/**/*/rspress
 !/cases/rsbuild/

scripts/package.json

@@ -10,7 +10,6 @@
   },
   "exports": {
     "./plugins/modernjs-framework": "./dist/plugins/modernjs-framework/index.js",
-    "./plugins/modernjs-module": "./dist/plugins/modernjs-module/index.js",
     "./plugins/rspress": "./dist/plugins/rspress/index.js",
     "./plugins/rsbuild": "./dist/plugins/rsbuild/index.js",
     "./plugins/rspack": "./dist/plugins/rspack/index.js"
@@ -20,9 +19,6 @@
       "plugins/modernjs-framework": [
         "./dist/plugins/modernjs-framework/index.d.ts"
       ],
-      "plugins/modernjs-module": [
-        "./dist/plugins/modernjs-module/index.d.ts"
-      ],
       "plugins/rspress": [
         "./dist/plugins/rspress/index.d.ts"
       ],

scripts/src/main.ts

@@ -4,6 +4,7 @@ import { build } from './runners/build';
 import { cloneRepo, getDataPath, mergeMetrics } from './shared';
 import { remove } from 'fs-extra';
 import { yarnInstall } from './runners/yarn-install';
+import { getBinarySize } from './runners/binary-size';
 
 const productName = process.argv[2];
 const caseName = process.argv[3];
@@ -17,28 +18,36 @@ async function main() {
     logger.error(`product not found: ${productName}`);
   }
 
-  if (caseName) {
-    process.env.PRODUCT_NAME = productName;
-    process.env.CASE_NAME = caseName;
-    await remove(dataPath);
+  process.env.PRODUCT_NAME = productName;
+  process.env.CASE_NAME = caseName;
 
-    if (process.env.ONLY_INSTALL_SIZE !== 'true') {
-      if (productName !== 'RSLIB') {
-        await dev(productName, caseName);
-      }
-      await build(productName, caseName);
-    }
+  await remove(dataPath);
 
-    try {
-      await yarnInstall(productName, caseName);
-    } catch (err) {
-      console.log('failed to collect install size metrics:', err);
-    }
+  if (!caseName) {
+    throw new Error('Missing case name!');
+  }
 
+  // Rspack only need to get binary size
+  if (productName === 'RSPACK') {
+    await getBinarySize();
     await mergeMetrics(productName, caseName);
-  } else {
-    logger.error(`Case not found: ${productName} ${caseName}`);
+    return;
   }
+
+  if (process.env.ONLY_INSTALL_SIZE !== 'true') {
+    if (productName !== 'RSLIB') {
+      await dev(productName, caseName);
+    }
+    await build(productName, caseName);
+  }
+
+  try {
+    await yarnInstall(productName, caseName);
+  } catch (err) {
+    console.log('failed to collect install size metrics:', err);
+  }
+
+  await mergeMetrics(productName, caseName);
 }
 
 main();

scripts/src/runners/binary-size.ts

@@ -0,0 +1,30 @@
+import { join, parse } from 'path';
+import { getRepoName, getRepoPath, runCommand, saveMetrics } from '../shared';
+import glob from 'fast-glob';
+import { statSync } from 'fs';
+
+export const getBinarySize = async () => {
+  const repoPath = getRepoPath(getRepoName('RSPACK'));
+
+  await runCommand(repoPath, 'pnpm run build:binding:release-prod');
+
+  const binaryPath = glob.sync(
+    join(repoPath, 'crates/node_binding/rspack.*.node'),
+    {
+      absolute: true,
+    },
+  );
+
+  if (!binaryPath.length) {
+    throw new Error('Binary file not found!');
+  }
+
+  const data: Record<string, number> = {};
+
+  for (const path of binaryPath) {
+    const size = statSync(path).size;
+    data[parse(path).name] = size;
+  }
+
+  return saveMetrics(data);
+};

scripts/src/shared/git.ts

@@ -68,6 +68,12 @@ export async function cloneRepo(productName: string, caseName: string) {
     return;
   }
 
+  if (productName === 'RSPACK') {
+    await runCommand(localRepoPath, 'corepack enable && pnpm -v');
+    await runCommand(localRepoPath, 'pnpm i');
+    return;
+  }
+
   await copy(
     join(getCaseSrcPath(productName), caseName),
     join(getCaseDistPath(productName), caseName),