feat: Handle content to inject in worker

Author: MagnunAVFAzion

lib/build/bundlers/esbuild/index.js

@@ -34,6 +34,17 @@ class Esbuild {
       config.plugins = [NodeModulesPolyfillPlugin(), ...config.plugins];
     }
 
+    // inject content in worker initial code.
+    if (this.builderConfig.contentToInject) {
+      const workerInitContent = this.builderConfig.contentToInject;
+
+      if (config.banner?.js) {
+        config.banner.js = `${config.banner.js} ${workerInitContent}`;
+      } else {
+        config.banner = { js: workerInitContent };
+      }
+    }
+
     try {
       await esbuild.build(config);
     } catch (error) {

lib/build/bundlers/webpack/index.js

@@ -30,6 +30,36 @@ class Webpack {
       config = merge(this.customConfig, config);
     }
 
+    // inject content in worker initial code.
+    if (this.builderConfig.contentToInject) {
+      const workerInitContent = this.builderConfig.contentToInject;
+
+      const bannerPluginIndex = config.plugins.findIndex(
+        (plugin) => plugin instanceof webpack.BannerPlugin,
+      );
+
+      if (bannerPluginIndex !== -1) {
+        const oldContent = config.plugins[bannerPluginIndex].options.banner;
+        const pluginToRemove = config.plugins[bannerPluginIndex];
+        config.plugins = config.plugins.filter(
+          (plugin) => plugin !== pluginToRemove,
+        );
+        config.plugins.push(
+          new webpack.BannerPlugin({
+            banner: `${oldContent} ${workerInitContent}`,
+            raw: true,
+          }),
+        );
+      } else {
+        config.plugins.push(
+          new webpack.BannerPlugin({
+            banner: workerInitContent,
+            raw: true,
+          }),
+        );
+      }
+    }
+
     try {
       const stats = await runWebpack(config);