chore: servicesStore + models into typescript (#344)

1 files changed, 116 insertions, 0 deletions

diff --git a/src/models/UserAgent.ts b/src/models/UserAgent.ts
new file mode 100644
index 000000000..1d06d72b0
--- /dev/null
+++ b/src/models/UserAgent.ts
@@ -0,0 +1,116 @@
+import { action, computed, observe, observable } from 'mobx';
+
+import ElectronWebView from 'react-electron-web-view';
+import defaultUserAgent from '../helpers/userAgent-helpers';
+
+const debug = require('../preload-safe-debug')('Ferdium:UserAgent');
+
+export default class UserAgent {
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  _willNavigateListener = (_event: any): void => {};
+
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  _didNavigateListener = (_event: any): void => {};
+
+  @observable.ref webview: ElectronWebView = null;
+
+  @observable chromelessUserAgent: boolean = false;
+
+  @observable userAgentPref: string | null = null;
+
+  @observable overrideUserAgent = (): string => '';
+
+  constructor(overrideUserAgent: any = null) {
+    if (typeof overrideUserAgent === 'function') {
+      this.overrideUserAgent = overrideUserAgent;
+    }
+
+    observe(this, 'webview', change => {
+      const { oldValue, newValue } = change;
+      if (oldValue !== null) {
+        this._removeWebviewEvents(oldValue);
+      }
+      if (newValue !== null) {
+        this._addWebviewEvents(newValue);
+      }
+    });
+  }
+
+  @computed get defaultUserAgent(): string {
+    const replacedUserAgent = this.overrideUserAgent();
+    if (replacedUserAgent.length > 0) {
+      return replacedUserAgent;
+    }
+
+    const globalPref = window['ferdium'].stores.settings.all.app.userAgentPref;
+    if (typeof globalPref === 'string') {
+      const trimmed = globalPref.trim();
+      if (trimmed !== '') {
+        return trimmed;
+      }
+    }
+    return defaultUserAgent();
+  }
+
+  @computed get serviceUserAgentPref(): string | null {
+    if (typeof this.userAgentPref === 'string') {
+      const trimmed = this.userAgentPref.trim();
+      if (trimmed !== '') {
+        return trimmed;
+      }
+    }
+    return null;
+  }
+
+  @computed get userAgentWithoutChromeVersion(): string {
+    const withChrome = this.defaultUserAgent;
+    return withChrome.replace(/Chrome\/[\d.]+/, 'Chrome');
+  }
+
+  @computed get userAgent(): string {
+    return (
+      this.serviceUserAgentPref ||
+      (this.chromelessUserAgent
+        ? this.userAgentWithoutChromeVersion
+        : this.defaultUserAgent)
+    );
+  }
+
+  @action setWebviewReference(webview: ElectronWebView): void {
+    this.webview = webview;
+  }
+
+  @action _handleNavigate(url: string, forwardingHack: boolean = false): void {
+    if (url.startsWith('https://accounts.google.com')) {
+      if (!this.chromelessUserAgent) {
+        debug('Setting user agent to chromeless for url', url);
+        this.chromelessUserAgent = true;
+        this.webview.userAgent = this.userAgent;
+        if (forwardingHack) {
+          this.webview.loadURL(url);
+        }
+      }
+    } else if (this.chromelessUserAgent) {
+      debug('Setting user agent to contain chrome for url', url);
+      this.chromelessUserAgent = false;
+      this.webview.userAgent = this.userAgent;
+    }
+  }
+
+  _addWebviewEvents(webview: ElectronWebView): void {
+    debug('Adding event handlers');
+
+    this._willNavigateListener = event => this._handleNavigate(event.url, true);
+    webview.addEventListener('will-navigate', this._willNavigateListener);
+
+    this._didNavigateListener = event => this._handleNavigate(event.url);
+    webview.addEventListener('did-navigate', this._didNavigateListener);
+  }
+
+  _removeWebviewEvents(webview: ElectronWebView): void {
+    debug('Removing event handlers');
+
+    webview.removeEventListener('will-navigate', this._willNavigateListener);
+    webview.removeEventListener('did-navigate', this._didNavigateListener);
+  }
+}