From 41f04470d0f51b899c8036a65dbd0c00845300b7 Mon Sep 17 00:00:00 2001
From: Xuan Son Nguyen <son@huggingface.co>
Date: Fri, 13 Dec 2024 12:22:07 +0100
Subject: [PATCH] add bundle size check

---
 examples/server/webui/vite.config.js | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/examples/server/webui/vite.config.js b/examples/server/webui/vite.config.js
index f2449af42..a6a42f148 100644
--- a/examples/server/webui/vite.config.js
+++ b/examples/server/webui/vite.config.js
@@ -3,6 +3,8 @@ import { viteSingleFile } from 'vite-plugin-singlefile';
 import path from 'path';
 import fs from 'fs';
 
+const MAX_BUNDLE_SIZE = 1024 * 1024; // only increase when absolutely necessary
+
 const GUIDE_FOR_FRONTEND = `
 <!--
   This is a single file build of the frontend.
@@ -26,6 +28,13 @@ const BUILD_PLUGINS = [
         const outputIndexHtml = path.join(config.build.outDir, 'index.html');
         const content = fs.readFileSync(outputIndexHtml, 'utf-8');
 
+        if (content.length > MAX_BUNDLE_SIZE) {
+          throw new Error(
+            `Bundle size is too large (${Math.ceil(content.length / 1024)} KB).\n` +
+            `Please reduce the size of the frontend or increase MAX_BUNDLE_SIZE in vite.config.js.\n`,
+          );
+        }
+
         const targetOutputFile = path.join(config.build.outDir, '../../public/index.html');
         fs.writeFileSync(targetOutputFile, GUIDE_FOR_FRONTEND + '\n' + content);
       }