refactor: Simplify script injection

Inject CSS and main world scripts synchronously to avoid race conditions
with page loading.

Don't try to miming userAgentData for now, since it won't bypass
google's checks. However, simply omitting chrome from the user agent
does bypass them, at least for now.

1 files changed, 18 insertions, 3 deletions

diff --git a/packages/service-preload/src/index.ts b/packages/service-preload/src/index.ts
index 3f54c0b..e42c406 100644
--- a/packages/service-preload/src/index.ts
+++ b/packages/service-preload/src/index.ts
@@ -18,7 +18,22 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
-import { ipcRenderer } from 'electron';
-import { ServiceToMainIpcMessage } from '@sophie/service-shared';
+import { ipcRenderer, webFrame } from 'electron';
+import { ServiceToMainIpcMessage, webSource } from '@sophie/service-shared';
 
-ipcRenderer.send(ServiceToMainIpcMessage.ApiExposedInMainWorld);
+if (webFrame.parent === null) {
+  // Inject CSS to simulate `browserView.setBackgroundColor`.
+  // This is injected before the page loads, so the styles from the website will overwrite it.
+  webFrame.insertCSS('html { background-color: #fff; }');
+}
+
+const injectSource = webSource.safeParse(ipcRenderer.sendSync(ServiceToMainIpcMessage.ApiExposedInMainWorld));
+if (injectSource.success) {
+  webFrame.executeJavaScriptInIsolatedWorld(0, [
+    injectSource.data,
+  ]).catch((err) => {
+    console.log('Failed to inject source:', err);
+  });
+} else {
+  console.log('Invalid source to inject:', injectSource.error);
+}