feat: User agent data simulator

2 files changed, 51 insertions, 66 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()),
   ]);
 }
 
diff --git a/packages/main/src/userAgent.ts b/packages/main/src/userAgent.ts
deleted file mode 100644
index 298e565..0000000
--- a/packages/main/src/userAgent.ts
+++ /dev/null
@@ -1,61 +0,0 @@
-/*
- * Copyright (C)  2021-2022 Kristóf Marussy <kristof@marussy.com>
- *
- * This file is part of Sophie.
- *
- * Sophie is free software: you can redistribute it and/or modify
- * it under the terms of the GNU Affero General Public License as
- * published by the Free Software Foundation, version 3.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU Affero General Public License for more details.
- *
- * You should have received a copy of the GNU Affero General Public License
- * along with this program.  If not, see <https://www.gnu.org/licenses/>.
- *
- * SPDX-License-Identifier: AGPL-3.0-only
- */
-
-/**
- * @file Based on the javascript code snippet available at
- * https://github.com/GoogleChrome/developer.chrome.com/blob/c33f7f4f2964e7deb28aee44de745c3725b70063/site/en/docs/privacy-sandbox/user-agent/snippets/index.md
- *
- * Used under the Apache 2.0 license according to
- * https://github.com/GoogleChrome/developer.chrome.com/blob/c33f7f4f2964e7deb28aee44de745c3725b70063/LICENSE
- */
-
-const electronUAParts = / (sophie|Electron)\/[^\s]+/g;
-const chromeUAs = /^Mozilla\/5\.0 \(((?<platform>Lin|Win|Mac|X11; C|X11; L)+[^\)]+)\) AppleWebKit\/537.36 \(KHTML, like Gecko\) Chrome\/(?<major>\d+)[\d\.]+(?<mobile>[ Mobile]*) Safari\/537\.36$/;
-const unifiedPlatform = {
-  'Lin': 'Linux; Android 10; K',
-  'Win': 'Windows NT 10.0; Win64; x64',
-  'Mac': 'Macintosh; Intel Mac OS X 10_15_7',
-  'X11; C': 'X11; CrOS x86_64',
-  'X11; L': 'X11; Linux x86_64',
-};
-
-/**
- * Reduces the information exposed in the user-agent string.
- *
- * @param userAgent The original user-agent string.
- * @returns The reduces user-agent string.
- * @see https://developer.chrome.com/docs/privacy-sandbox/user-agent/
- */
-export function reduceUserAgent(userAgent: string): string {
-  const userAgentWithoutElectron = userAgent.replaceAll(electronUAParts, '');
-  const matched = chromeUAs.exec(userAgentWithoutElectron) as unknown as {
-    groups: {
-      platform: 'Lin' | 'Win' | 'Mac' | 'X11; C' | 'X11; L',
-      major: string,
-      mobile: string,
-    }
-  };
-  if (matched) {
-    return `Mozilla/5.0 (${unifiedPlatform[matched.groups.platform]}) ` +
-      `AppleWebKit/537.36 (KHTML, like Gecko) ` +
-      `Chrome/${matched.groups.major}.0.0.0${matched.groups.mobile} Safari/537.36`
-  }
-  return userAgentWithoutElectron;
-}