Update electron forge after completion logic

Author: Shellishack

desktop/main.mjs

@@ -1,7 +1,6 @@
 import { app, BrowserWindow, dialog, ipcMain } from "electron";
 import serve from "electron-serve";
 import { join } from "path";
-
 import { fileURLToPath } from "url";
 import path from "path";
 import { nativeTheme } from "electron/main";

forge.config.ts

@@ -2,6 +2,16 @@ import type { ForgeConfig } from "@electron-forge/shared-types";
 import path from "path";
 import fs from "fs-extra";
 
+function removeNextjsModules(exceptionList: string[], extractPath: string) {
+  const nodeModulesPath = path.join(extractPath, "resources/app/node_modules");
+
+  fs.readdirSync(nodeModulesPath).forEach((module) => {
+    if (!exceptionList.includes(module)) {
+      fs.removeSync(path.join(nodeModulesPath, module));
+    }
+  });
+}
+
 const config: ForgeConfig = {
   outDir: "out-desktop",
   packagerConfig: {
@@ -16,24 +26,20 @@ const config: ForgeConfig = {
         return false;
       } else if (path.match(/desktop/)) {
         return false;
+      } else if (path.match(/node_modules/)) {
+        return false;
       }
       return true;
     },
-    extraResource: "node_modules/electron-serve",
     afterComplete: [
       (extractPath, electronVersion, platform, arch, done) => {
-        // move node_modules/electron-serve to the app directory
-        const from = path.join(extractPath, "node_modules/electron-serve");
-        const to = path.join(
-          extractPath,
-          "resources/app/node_modules/electron-serve",
-        );
-        if (fs.existsSync(from)) {
-          fs.moveSync(
-            path.join(extractPath, "resources/electron-serve"),
-            path.join(extractPath, "resources/app/node_modules/electron-serve"),
-          );
-        }
+        // We need electron-serve to exist inside the electron build's node_modules.
+        // All other modules from nextjs are not needed and can be removed.
+
+        const electronModules = ["electron-serve"];
+
+        removeNextjsModules(electronModules, extractPath);
+
         done();
       },
     ],