feat(Service): Add modifyRequestHeaders for services (🙌 @kamyweb)

Author: adlk

appveyor.yml

@@ -30,4 +30,4 @@ artifacts:
 
 branches:
   except:
-    - i18n
+    - i18n

src/index.js

@@ -5,6 +5,7 @@ import {
   BrowserWindow,
   shell,
   ipcMain,
+  session,
 } from 'electron';
 
 // import isDevMode from 'electron-is-dev';
@@ -394,6 +395,29 @@ ipcMain.on('feature-basic-auth-credentials', (e, { user, password }) => {
   authCallback = noop;
 });
 
+ipcMain.on('modifyRequestHeaders', (e, { modifiedRequestHeaders, serviceId }) => {
+  debug('Received modifyRequestHeaders', modifiedRequestHeaders, serviceId);
+  modifiedRequestHeaders.forEach((headerFilterSet) => {
+    const { headers, requestFilters } = headerFilterSet;
+    session.fromPartition(`persist:service-${serviceId}`).webRequest.onBeforeSendHeaders(requestFilters, (details, callback) => {
+      for (const key in headers) {
+        if (Object.prototype.hasOwnProperty.call(headers, key)) {
+          const value = headers[key];
+          if (value === 'RefererHost') {
+            if (Object.prototype.hasOwnProperty.call(details.requestHeaders, 'Referer')) {
+              const { hostname } = new URL(details.requestHeaders.Referer);
+              details.requestHeaders[key] = `https://${hostname}`;
+            }
+          } else {
+            details.requestHeaders[key] = value;
+          }
+        }
+      }
+      callback({ requestHeaders: details.requestHeaders });
+    });
+  });
+});
+
 ipcMain.on('feature-basic-auth-cancel', () => {
   debug('Cancel basic auth');
 

src/models/Service.js

@@ -2,6 +2,7 @@ import { webContents } from '@electron/remote';
 import {
   computed, observable, autorun,
 } from 'mobx';
+import { ipcRenderer } from 'electron';
 import path from 'path';
 import normalizeUrl from 'normalize-url';
 
@@ -231,6 +232,20 @@ export default class Service {
   initializeWebViewEvents({ handleIPCMessage, openWindow, stores }) {
     const webviewWebContents = webContents.fromId(this.webview.getWebContentsId());
 
+    // If the recipe has implemented modifyRequestHeaders,
+    // Send those headers to ipcMain so that it can be set in session
+    if (typeof this.recipe.modifyRequestHeaders === 'function') {
+      const modifiedRequestHeaders = this.recipe.modifyRequestHeaders();
+      console.warn(modifiedRequestHeaders);
+      debug(this.name, 'modifiedRequestHeaders', modifiedRequestHeaders);
+      ipcRenderer.send('modifyRequestHeaders', {
+        modifiedRequestHeaders,
+        serviceId: this.id,
+      });
+    } else {
+      debug(this.name, 'modifyRequestHeaders is not defined in the recipe');
+    }
+
     const handleUserAgent = (url, forwardingHack = false) => {
       if (url.startsWith('https://accounts.google.com')) {
         if (!this.chromelessUserAgent) {