From 146fc311509471981726180e20d12e73e43e595d Mon Sep 17 00:00:00 2001
From: Tim Fish <tim@timfish.uk>
Date: Fri, 23 Jul 2021 16:36:34 +0100
Subject: [PATCH] fix(webpack-plugin): Ensure asset relocator injected code
 works with nodeIntegration disabled (#2396)

* Adjust injected code when nodeIntegration = false

Co-authored-by: Mark Lee <electronjs@lazymalevolence.com>
---
 packages/plugin/webpack/src/WebpackConfig.ts  |  5 ++-
 .../webpack/src/util/AssetRelocatorPatch.ts   | 33 ++++++++++++++-----
 .../webpack/test/AssetRelocatorPatch_spec.ts  | 18 +++++++++-
 3 files changed, 45 insertions(+), 11 deletions(-)

diff --git a/packages/plugin/webpack/src/WebpackConfig.ts b/packages/plugin/webpack/src/WebpackConfig.ts
index d795708dbe..c512e9cb14 100644
--- a/packages/plugin/webpack/src/WebpackConfig.ts
+++ b/packages/plugin/webpack/src/WebpackConfig.ts
@@ -201,7 +201,10 @@ export default class WebpackConfigGenerator {
         filename: `${entryPoint.name}/index.html`,
         chunks: [entryPoint.name].concat(entryPoint.additionalChunks || []),
       }) as WebpackPluginInstance).concat(
-        [new webpack.DefinePlugin(defines), new AssetRelocatorPatch(this.isProd)],
+        [
+          new webpack.DefinePlugin(defines),
+          new AssetRelocatorPatch(this.isProd, !!this.pluginConfig.renderer.nodeIntegration),
+        ],
       );
     return webpackMerge({
       entry,
diff --git a/packages/plugin/webpack/src/util/AssetRelocatorPatch.ts b/packages/plugin/webpack/src/util/AssetRelocatorPatch.ts
index 6d6b9093e0..6661073a05 100644
--- a/packages/plugin/webpack/src/util/AssetRelocatorPatch.ts
+++ b/packages/plugin/webpack/src/util/AssetRelocatorPatch.ts
@@ -3,8 +3,26 @@ import { Chunk, Compiler } from 'webpack';
 export default class AssetRelocatorPatch {
   private readonly isProd: boolean;
 
-  constructor(isProd: boolean) {
+  private readonly nodeIntegration: boolean;
+
+  constructor(isProd: boolean, nodeIntegration: boolean) {
     this.isProd = isProd;
+    this.nodeIntegration = nodeIntegration;
+  }
+
+  private injectedProductionDirnameCode(): string {
+    if (this.nodeIntegration) {
+      // In production the assets are found one directory up from
+      // __dirname
+      //
+      // __dirname cannot be used directly until this PR lands
+      // https://github.com/jantimon/html-webpack-plugin/pull/1650
+      return 'require("path").resolve(require("path").dirname(__filename), "..")';
+    }
+
+    // If nodeIntegration is disabled, we replace __dirname
+    // with an empty string so no error is thrown at runtime
+    return '""';
   }
 
   public apply(compiler: Compiler) {
@@ -32,20 +50,17 @@ export default class AssetRelocatorPatch {
                   throw new Error('The installed version of @vercel/webpack-asset-relocator-loader does not appear to be compatible with Forge');
                 }
 
+                if (this.isProd) {
+                  return originalInjectCode.replace('__dirname', this.injectedProductionDirnameCode());
+                }
+
                 return originalInjectCode.replace(
                   '__dirname',
-                  this.isProd
-                  // In production the assets are found one directory up from
-                  // __dirname
-                  //
-                  // __dirname cannot be used directly until this PR lands
-                  // https://github.com/jantimon/html-webpack-plugin/pull/1650
-                    ? 'require("path").resolve(require("path").dirname(__filename), "..")'
                   // In development, the app is loaded via webpack-dev-server
                   // so __dirname is useless because it points to Electron
                   // internal code. Instead we hard-code the absolute path to
                   // the webpack output.
-                    : JSON.stringify(compiler.options.output.path),
+                  JSON.stringify(compiler.options.output.path),
                 );
               };
             }
diff --git a/packages/plugin/webpack/test/AssetRelocatorPatch_spec.ts b/packages/plugin/webpack/test/AssetRelocatorPatch_spec.ts
index 88f9c2b8d7..0d136e3d05 100644
--- a/packages/plugin/webpack/test/AssetRelocatorPatch_spec.ts
+++ b/packages/plugin/webpack/test/AssetRelocatorPatch_spec.ts
@@ -109,6 +109,7 @@ describe('AssetRelocatorPatch', () => {
   const config = {
     mainConfig: './webpack.main.config.js',
     renderer: {
+      nodeIntegration: true,
       config: './webpack.renderer.config.js',
       entryPoints: [
         {
@@ -176,7 +177,7 @@ describe('AssetRelocatorPatch', () => {
   });
 
   describe('Production', () => {
-    const generator = new WebpackConfigGenerator(config, appPath, true, 3000);
+    let generator = new WebpackConfigGenerator(config, appPath, true, 3000);
 
     it('builds main', async () => {
       const mainConfig = generator.getMainConfig();
@@ -224,5 +225,20 @@ describe('AssetRelocatorPatch', () => {
       expect(output).to.contain('Hello, world! from the preload');
       expect(output).to.contain('Hello, world! from the renderer');
     });
+
+    it('builds renderer with nodeIntegration = false', async () => {
+      config.renderer.nodeIntegration = false;
+      generator = new WebpackConfigGenerator(config, appPath, true, 3000);
+
+      const rendererConfig = await generator.getRendererConfig(config.renderer.entryPoints);
+      await asyncWebpack(rendererConfig);
+
+      await expectOutputFileToHaveTheCorrectNativeModulePath({
+        outDir: rendererOut,
+        jsPath: path.join(rendererOut, 'main_window/index.js'),
+        nativeModulesString: '.ab="/native_modules/"',
+        nativePathString: `.ab+"${nativePathSuffix}"`,
+      });
+    });
   });
 });