feat: User agent data simulator

1 files changed, 51 insertions, 5 deletions

diff --git a/packages/main/src/index.ts b/packages/main/src/index.ts
index b6df67b..70acb15 100644
--- a/packages/main/src/index.ts
+++ b/packages/main/src/index.ts
@@ -19,6 +19,7 @@
  */
 
 import { app, BrowserView, BrowserWindow } from 'electron';
+import { readFile, readFileSync } from 'fs';
 import { autorun } from 'mobx';
 import { getSnapshot, onPatch } from 'mobx-state-tree';
 import { join } from 'path';
@@ -39,7 +40,6 @@ import {
   openDevToolsWhenReady,
 } from './devTools';
 import { createRootStore } from './stores/RootStore';
-import { reduceUserAgent } from './userAgent';
 
 const isDevelopment = import.meta.env.MODE === 'development';
 
@@ -67,12 +67,33 @@ app.commandLine.appendSwitch(
 
 // Remove sophie and electron from the user-agent string to avoid detection.
 const originalUserAgent = app.userAgentFallback;
-const userAgent = reduceUserAgent(originalUserAgent);
+const userAgent = originalUserAgent.replaceAll(/\s(sophie|Electron)\/\S+/g, '');
+const platformInUa = userAgent.match(/\((Win|Mac|X11; L)/);
+let platform = 'Unknown';
+if (platformInUa !== null) {
+  switch (platformInUa[1]) {
+    case 'Win':
+      platform = 'Windows';
+      break;
+    case 'Mac':
+      platform = 'macOS';
+      break;
+    case 'X11; L':
+      platform = 'Linux';
+      break;
+  }
+}
+const chromiumVersion = process.versions.chrome.split('.')[0];
 // Removing the electron version breaks redux devtools, so we only do this in production.
 if (!isDevelopment) {
   app.userAgentFallback = userAgent;
 }
 
+let serviceInjectRelativePath = '../../service-inject/dist/index.cjs';
+let serviceInjectPath = join(__dirname, serviceInjectRelativePath);
+let serviceInjectUrl = new URL(serviceInjectRelativePath, `file://${__dirname}`).toString();
+let serviceInject: string = readFileSync(serviceInjectPath, 'utf8');
+
 if (isDevelopment) {
   installDevToolsExtensions(app);
 }
@@ -92,7 +113,7 @@ function createWindow(): Promise<unknown> {
   });
 
   if (isDevelopment) {
-    openDevToolsWhenReady(mainWindow);
+    //openDevToolsWhenReady(mainWindow);
   }
 
   mainWindow.on('ready-to-show', () => {
@@ -131,7 +152,14 @@ function createWindow(): Promise<unknown> {
           store.setPaletteMode(paletteMode.parse(args[0]))
           break;
         case RendererToMainIpcMessage.ReloadAllServices:
-          browserView.webContents.reload();
+          readFile(serviceInjectPath, 'utf8', (err, data) => {
+            if (err === null) {
+              serviceInject = data;
+            } else {
+              console.error('Error while reloading', serviceInjectPath, err);
+            }
+            browserView.webContents.reload();
+          });
           break;
         default:
           console.error('Unknown IPC message:', channel, args);
@@ -149,6 +177,16 @@ function createWindow(): Promise<unknown> {
   browserView.webContents.on('ipc-message', (_event, channel, ...args) => {
     try {
       switch (channel) {
+        case ServiceToMainIpcMessage.ApiExposedInMainWorld:
+          browserView.webContents.executeJavaScriptInIsolatedWorld(0, [
+            {
+              code: serviceInject,
+              url: serviceInjectUrl,
+            }
+          ]).catch((err) => {
+            console.error('Cannot inject script:', err);
+          });
+          break;
         case ServiceToMainIpcMessage.SetUnreadCount:
           console.log('Unread count:', unreadCount.parse(args[0]));
           break;
@@ -172,13 +210,21 @@ function createWindow(): Promise<unknown> {
     );
   });
 
+  browserView.webContents.session.webRequest.onBeforeSendHeaders(({ requestHeaders }, callback) => {
+    requestHeaders['User-Agent'] = userAgent;
+    requestHeaders['Sec-CH-UA'] = `" Not A;Brand";v="99", "Chromium";v="${chromiumVersion}"`;
+    requestHeaders['Sec-CH-UA-Mobile'] = '?0';
+    requestHeaders['Sec-CH-UA-Platform'] = platform;
+    callback({ requestHeaders });
+  });
+
   const pageUrl = (isDevelopment && import.meta.env.VITE_DEV_SERVER_URL !== undefined)
     ? import.meta.env.VITE_DEV_SERVER_URL
     : new URL('../renderer/dist/index.html', `file://${__dirname}`).toString();
 
   return Promise.all([
     mainWindow.loadURL(pageUrl),
-    browserView.webContents.loadURL('https://git.marussy.com/sophie/about'),
+    browserView.webContents.loadURL('https://gmail.com').then(() => browserView.webContents.openDevTools()),
   ]);
 }