vite-plugin-blocklet 0.13.2 → 0.14.0

package/dist/index.cjs

@@ -8,9 +8,11 @@ var YAML = require('yaml');
 var Mcrypto = require('@ocap/mcrypto');
 var util = require('@ocap/util');
 var did = require('@arcblock/did');
+var fs = require('fs');
+var path = require('path');
 var ufo = require('ufo');
 var isMobile = require('ismobilejs');
-var path = require('node:path');
+var path$1 = require('node:path');
 var mlly = require('mlly');
 var externalGlobals = require('rollup-plugin-external-globals');
 var pMap = require('p-map');
@@ -148,7 +150,17 @@ runInit();`,
   };
 }
 
-function createConfigPlugin$1() {
+const SIZE = 1000;
+const MAX_CHUNK_SIZE = 2000;
+
+/**
+ * @param {object} options
+ * @param {number} [options.chunkSizeLimit=2048] - The chunk size limit in KB.
+ * @return {import('vite').Plugin} The Vite config plugin.
+ */
+function createConfigPlugin$1({ chunkSizeLimit = MAX_CHUNK_SIZE }) {
+  let resolvedConfig;
+
   return {
     name: 'blocklet:config',
     configureServer(server) {
@@ -182,6 +194,11 @@ function createConfigPlugin$1() {
       }
 
       if (command === 'build') {
+        let targetConfig = {
+          build: {
+            chunkSizeWarningLimit: chunkSizeLimit,
+          },
+        };
         if (!config.base) {
           try {
             let { name, did } = await getBlockletYAML();
@@ -190,19 +207,75 @@ function createConfigPlugin$1() {
             }
             if (did) {
               const base = `/.blocklet/proxy/${did}/`;
-              return {
-                base,
-              };
+              targetConfig.base = base;
+              return targetConfig;
             }
           } catch (err) {
             console.error(err);
-            return {};
+            return targetConfig;
           }
         }
       }
 
       return {};
     },
+    configResolved(config) {
+      resolvedConfig = config;
+    },
+    closeBundle() {
+      const limitInKB = chunkSizeLimit;
+      const outDir = resolvedConfig.build.outDir || 'dist';
+      const distPath = path.resolve(resolvedConfig.root, outDir);
+
+      if (!fs.existsSync(distPath)) return;
+
+      // 递归获取所有文件（包含子目录，如 assets/）
+      const getAllFiles = (dirPath, arrayOfFiles = []) => {
+        const files = fs.readdirSync(dirPath);
+        files.forEach((file) => {
+          const fullPath = path.join(dirPath, file);
+          if (fs.statSync(fullPath).isDirectory()) {
+            getAllFiles(fullPath, arrayOfFiles);
+          } else {
+            arrayOfFiles.push(fullPath);
+          }
+        });
+        return arrayOfFiles;
+      };
+
+      const allFiles = getAllFiles(distPath);
+      const overSizedFiles = [];
+
+      allFiles.forEach((filePath) => {
+        // 只检查 JS 和 CSS
+        if (filePath.endsWith('.js') || filePath.endsWith('.css')) {
+          const stats = fs.statSync(filePath);
+          const sizeKB = Number((stats.size / SIZE).toFixed(2));
+
+          if (sizeKB > limitInKB) {
+            overSizedFiles.push({
+              // 显示相对路径，方便查看是哪个 chunk
+              File: path.relative(distPath, filePath),
+              'Size(KB)': sizeKB,
+              'Limit(KB)': limitInKB,
+            });
+          }
+        }
+      });
+
+      if (overSizedFiles.length > 0) {
+        overSizedFiles.sort((a, b) => b['Size(KB)'] - a['Size(KB)']);
+        console.log('\n\x1b[41m\x1b[37m ERROR \x1b[0m \x1b[31mChunk Size Limit Exceeded:\x1b[0m');
+        console.table(overSizedFiles); // 以表格形式漂亮地打印
+
+        console.error(`\x1b[31mTotal ${overSizedFiles.length} files exceeded the ${limitInKB}KB limit.\x1b[0m\n`);
+
+        // 强制退出，确保 CI/CD 流程中断
+        process.exit(1);
+      } else {
+        console.log('\n\x1b[42m\x1b[30m DONE \x1b[0m All chunks are within the size limit.');
+      }
+    },
   };
 }
 
@@ -735,7 +808,7 @@ function createEmbedPlugin(options) {
                 lib: {
                   entry: entryItem,
                   formats: ['es'],
-                  fileName: path.parse(ENTRY_FILE).name,
+                  fileName: path$1.parse(ENTRY_FILE).name,
                 },
                 rollupOptions: {
                   external,
@@ -928,6 +1001,7 @@ async function setupClient(app, options = {}) {
  * @property {number} [embedBuildConcurrency=0] - The plugins to be used in the embeds.
  *
  * @property {'middleware'|'client'|'server'|'wsUpgrade'} [hmrMode='middleware'] - 当未传入任何 option 参数时，会自动变为 middleware 模式
+ * @property {number} [chunkSizeLimit=2000] - The chunk size limit in KB.
  */
 
 /**
@@ -959,7 +1033,7 @@ function createBlockletPlugin(options = {}) {
     plugins.push(createMetaPlugin());
   }
   if (!disableConfig) {
-    plugins.push(createConfigPlugin$1());
+    plugins.push(createConfigPlugin$1(restOptions));
   }
   if (!disableHmr) {
     plugins.push(createHmrPlugin(restOptions));

package/index.js

@@ -41,6 +41,7 @@ import setupClient from './libs/client.js';
  * @property {number} [embedBuildConcurrency=0] - The plugins to be used in the embeds.
  *
  * @property {'middleware'|'client'|'server'|'wsUpgrade'} [hmrMode='middleware'] - 当未传入任何 option 参数时，会自动变为 middleware 模式
+ * @property {number} [chunkSizeLimit=2000] - The chunk size limit in KB.
  */
 
 /**

package/libs/config.js

@@ -1,7 +1,19 @@
+import fs from 'fs';
+import path from 'path';
 import { isEqual, joinURL, withTrailingSlash } from 'ufo';
 import { toBlockletDid, isInBlocklet, blockletPort, blockletPrefix, getBlockletYAML } from './utils.js';
 
-export default function createConfigPlugin() {
+const SIZE = 1000;
+const MAX_CHUNK_SIZE = 2000;
+
+/**
+ * @param {object} options
+ * @param {number} [options.chunkSizeLimit=2048] - The chunk size limit in KB.
+ * @return {import('vite').Plugin} The Vite config plugin.
+ */
+export default function createConfigPlugin({ chunkSizeLimit = MAX_CHUNK_SIZE }) {
+  let resolvedConfig;
+
   return {
     name: 'blocklet:config',
     configureServer(server) {
@@ -35,6 +47,11 @@ export default function createConfigPlugin() {
       }
 
       if (command === 'build') {
+        let targetConfig = {
+          build: {
+            chunkSizeWarningLimit: chunkSizeLimit,
+          },
+        };
         if (!config.base) {
           try {
             let { name, did } = await getBlockletYAML();
@@ -43,18 +60,74 @@ export default function createConfigPlugin() {
             }
             if (did) {
               const base = `/.blocklet/proxy/${did}/`;
-              return {
-                base,
-              };
+              targetConfig.base = base;
+              return targetConfig;
             }
           } catch (err) {
             console.error(err);
-            return {};
+            return targetConfig;
           }
         }
       }
 
       return {};
     },
+    configResolved(config) {
+      resolvedConfig = config;
+    },
+    closeBundle() {
+      const limitInKB = chunkSizeLimit;
+      const outDir = resolvedConfig.build.outDir || 'dist';
+      const distPath = path.resolve(resolvedConfig.root, outDir);
+
+      if (!fs.existsSync(distPath)) return;
+
+      // 递归获取所有文件（包含子目录，如 assets/）
+      const getAllFiles = (dirPath, arrayOfFiles = []) => {
+        const files = fs.readdirSync(dirPath);
+        files.forEach((file) => {
+          const fullPath = path.join(dirPath, file);
+          if (fs.statSync(fullPath).isDirectory()) {
+            getAllFiles(fullPath, arrayOfFiles);
+          } else {
+            arrayOfFiles.push(fullPath);
+          }
+        });
+        return arrayOfFiles;
+      };
+
+      const allFiles = getAllFiles(distPath);
+      const overSizedFiles = [];
+
+      allFiles.forEach((filePath) => {
+        // 只检查 JS 和 CSS
+        if (filePath.endsWith('.js') || filePath.endsWith('.css')) {
+          const stats = fs.statSync(filePath);
+          const sizeKB = Number((stats.size / SIZE).toFixed(2));
+
+          if (sizeKB > limitInKB) {
+            overSizedFiles.push({
+              // 显示相对路径，方便查看是哪个 chunk
+              File: path.relative(distPath, filePath),
+              'Size(KB)': sizeKB,
+              'Limit(KB)': limitInKB,
+            });
+          }
+        }
+      });
+
+      if (overSizedFiles.length > 0) {
+        overSizedFiles.sort((a, b) => b['Size(KB)'] - a['Size(KB)']);
+        console.log('\n\x1b[41m\x1b[37m ERROR \x1b[0m \x1b[31mChunk Size Limit Exceeded:\x1b[0m');
+        console.table(overSizedFiles); // 以表格形式漂亮地打印
+
+        console.error(`\x1b[31mTotal ${overSizedFiles.length} files exceeded the ${limitInKB}KB limit.\x1b[0m\n`);
+
+        // 强制退出，确保 CI/CD 流程中断
+        process.exit(1);
+      } else {
+        console.log('\n\x1b[42m\x1b[30m DONE \x1b[0m All chunks are within the size limit.');
+      }
+    },
   };
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "vite-plugin-blocklet",
   "type": "module",
-  "version": "0.13.2",
+  "version": "0.14.0",
   "description": "",
   "main": "index.js",
   "files": [