Merge pull request #9 from WodenWang820118/develop

refactor: refactor Electron main process

.gitignore

@@ -39,6 +39,6 @@ testem.log
 .DS_Store
 Thumbs.db
 
-.nx/cache
 .angular
 .database.sqlite3
+*nx

hooks.cjs

@@ -0,0 +1,19 @@
+const { join } = require('path');
+const { runCommand } = require('./main-process/command-utils.cjs');
+
+(async () => {
+  try {
+    // Run npm install in the specified directory
+    await runCommand(
+      'npm install --prefer-offline --no-audit --progress=false --omit=dev && npm install sqlite3 --prefer-offline --no-audit --progress=false --omit=dev',
+      {
+        cwd: join(__dirname, 'dist/nest-backend'),
+        stdio: 'inherit',
+      }
+    );
+
+    console.log('npm install completed successfully');
+  } catch (err) {
+    console.error(`Error: ${err.message}`);
+  }
+})();

main-process/backend.cjs

@@ -0,0 +1,101 @@
+'use strict';
+// const { utilityProcess } = require('electron');
+const { fork } = require('child_process');
+const { join } = require('path');
+const constants = require('./constants.cjs');
+const environmentUtils = require('./environment-utils.cjs');
+const fileUtils = require('./file-utils.cjs');
+const pathUtils = require('./path-utils.cjs');
+
+function startBackend(resourcesPath) {
+  let env;
+  const rootBackendFolderPath = pathUtils.getRootBackendFolderPath(
+    environmentUtils.getEnvironment(),
+    resourcesPath
+  );
+  const serverPath = join(rootBackendFolderPath, 'main.js');
+  const databasePath = join(
+    rootBackendFolderPath,
+    constants.ROOT_DATABASE_NAME
+  );
+  switch (environmentUtils.getEnvironment()) {
+    case 'dev':
+      env = {
+        NODE_ENV: 'dev',
+        DATABASE_PATH: databasePath,
+        PORT: 3000,
+      };
+      break;
+    case 'staging':
+      env = {
+        NODE_ENV: 'staging',
+        DATABASE_PATH: databasePath,
+        PORT: 3000,
+      };
+      break;
+    case 'prod':
+      env = {
+        NODE_ENV: 'prod',
+        DATABASE_PATH: databasePath,
+        PORT: 5000,
+      };
+      break;
+    default:
+      env = {
+        NODE_ENV: 'prod',
+        DATABASE_PATH: databasePath,
+        PORT: 5000,
+      };
+      break;
+  }
+
+  fileUtils.writePath(
+    join(
+      pathUtils.getRootBackendFolderPath(
+        environmentUtils.getEnvironment(),
+        resourcesPath
+      ),
+      'env.txt'
+    ),
+    JSON.stringify(env, null, 2)
+  );
+
+  fileUtils.writePath(
+    join(
+      pathUtils.getRootBackendFolderPath(
+        environmentUtils.getEnvironment(),
+        resourcesPath
+      ),
+      'serverPath.txt'
+    ),
+    serverPath
+  );
+
+  // return utilityProcess.fork(serverPath, { env });
+  return fork(serverPath, { env });
+}
+
+async function checkIfPortIsOpen(urls, maxAttempts = 20, timeout = 2000) {
+  for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+    for (const url of urls) {
+      try {
+        const response = await fetch(url);
+        if (response) {
+          console.log('Server is ready');
+          return true; // Port is open
+        }
+      } catch (error) {
+        console.log(`Attempt ${attempt}: Waiting for server to start...`);
+      }
+    }
+    await new Promise((resolve) => setTimeout(resolve, timeout)); // Wait for 2 seconds before retrying
+  }
+  throw new Error(
+    `Failed to connect to the server after ${maxAttempts} attempts`
+  );
+}
+
+module.exports = {
+  startBackend,
+  checkIfPortIsOpen,
+};

main-process/command-utils.cjs

@@ -0,0 +1,25 @@
+'use strict';
+const { spawn } = require('child_process');
+
+// Helper function to run a command and return a promise
+const runCommand = (command, options) => {
+  return new Promise((resolve, reject) => {
+    const process = spawn(command, { ...options, shell: true });
+
+    process.on('close', (code) => {
+      if (code !== 0) {
+        reject(new Error(`${command} failed with code ${code}`));
+      } else {
+        resolve();
+      }
+    });
+
+    process.on('error', (err) => {
+      reject(err);
+    });
+  });
+};
+
+module.exports = {
+  runCommand,
+};

main-process/constants.cjs

@@ -0,0 +1,9 @@
+'use strict';
+
+const URLs = ['http://localhost:5000', 'http://localhost:3000'];
+const ROOT_DATABASE_NAME = 'database.sqlite3';
+
+module.exports = {
+  URLs,
+  ROOT_DATABASE_NAME,
+};

main-process/environment-utils.cjs

@@ -0,0 +1,6 @@
+'use strict';
+function getEnvironment() {
+  return process.env.NODE_ENV || 'prod';
+}
+
+module.exports = { getEnvironment };

main-process/file-utils.cjs

@@ -0,0 +1,10 @@
+'use strict';
+const { writeFileSync } = require('fs');
+
+function writePath(filePath, content) {
+  writeFileSync(filePath, content, 'utf8');
+}
+
+module.exports = {
+  writePath,
+};

main-process/frontend.cjs

@@ -0,0 +1,77 @@
+'use strict';
+const { BrowserWindow } = require('electron');
+const { join } = require('path');
+const pathUtils = require('./path-utils.cjs');
+const fileUtils = require('./file-utils.cjs');
+
+let loadingWindow = null;
+
+function createLoadingWindow() {
+  console.log('Creating loading window');
+  if (loadingWindow) {
+    console.log('Loading window already exists');
+    return loadingWindow;
+  }
+
+  try {
+    loadingWindow = new BrowserWindow({
+      width: 400,
+      height: 200,
+      frame: false,
+      transparent: true,
+      alwaysOnTop: true,
+      webPreferences: {
+        nodeIntegration: true,
+      },
+    });
+
+    loadingWindow.loadFile(join(__dirname, 'loading.html'));
+    loadingWindow.center();
+
+    loadingWindow.on('closed', () => {
+      loadingWindow = null;
+    });
+
+    return loadingWindow;
+  } catch (error) {
+    console.error('Error creating loading window:', error);
+    return null;
+  }
+}
+
+function createWindow(env, resourcesPath) {
+  const mainWindow = new BrowserWindow({
+    width: 1400,
+    height: 900,
+    webPreferences: {
+      nodeIntegration: true,
+    },
+  });
+
+  try {
+    const entryPath = pathUtils.getFrontendPath(env, resourcesPath);
+    fileUtils.writePath(
+      join(
+        pathUtils.getRootBackendFolderPath(env, resourcesPath),
+        'entryPath.txt'
+      ),
+      entryPath
+    );
+    mainWindow.loadFile(entryPath);
+    mainWindow.webContents.openDevTools(); // Open DevTools in development
+  } catch (e) {
+    console.error(e);
+    fileUtils.writePath(
+      join(
+        pathUtils.getRootBackendFolderPath(env, resourcesPath),
+        'entryPathError.txt'
+      ),
+      e.message
+    );
+  }
+}
+
+module.exports = {
+  createLoadingWindow,
+  createWindow,
+};

main-process/loading.html

@@ -0,0 +1,36 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>Loading...</title>
+    <style>
+      body {
+        background: transparent;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        height: 100vh;
+        margin: 0;
+        font-family: Arial, sans-serif;
+      }
+      .loader {
+        border: 5px solid #f3f3f3;
+        border-top: 5px solid #3498db;
+        border-radius: 50%;
+        width: 50px;
+        height: 50px;
+        animation: spin 1s linear infinite;
+      }
+      @keyframes spin {
+        0% {
+          transform: rotate(0deg);
+        }
+        100% {
+          transform: rotate(360deg);
+        }
+      }
+    </style>
+  </head>
+  <body>
+    <div class="loader"></div>
+  </body>
+</html>

main-process/path-utils.cjs

@@ -0,0 +1,41 @@
+'use strict';
+const { join } = require('path');
+
+function getRootBackendFolderPath(env, resourcesPath) {
+  switch (env) {
+    case 'dev':
+    case 'staging':
+      return join('dist', 'nest-backend');
+    case 'prod':
+      return resourcesPath;
+    default:
+      // Default to production path
+      return resourcesPath;
+  }
+}
+
+function getFrontendPath(env, resourcesPath) {
+  const devFrontendPath = join('dist', 'ng-tracker', 'browser', 'index.html');
+
+  const prodFrontendPath = join(
+    resourcesPath,
+    'ng-tracker',
+    'browser',
+    'index.html'
+  );
+
+  switch (env) {
+    case 'dev':
+    case 'staging':
+      return devFrontendPath;
+    case 'prod':
+      return prodFrontendPath;
+    default:
+      return prodFrontendPath;
+  }
+}
+
+module.exports = {
+  getRootBackendFolderPath,
+  getFrontendPath,
+};