src/features/serviceProxy/index.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

import { autorun, observable } from 'mobx';
import { remote } from 'electron';

import { DEFAULT_FEATURES_CONFIG, HEALTHCHECK_URL } from '../../config';

const { session } = remote;

const debug = require('debug')('Franz:feature:serviceProxy');

export const config = observable({
  isEnabled: DEFAULT_FEATURES_CONFIG.isServiceProxyEnabled,
  isPremium: DEFAULT_FEATURES_CONFIG.isServiceProxyPremiumFeature,
});

export default function init(stores) {
  debug('Initializing `serviceProxy` feature');

  const { defaultSession } = session;
  defaultSession.resolveProxy(HEALTHCHECK_URL, (proxy) => {
    debug('Resolve proxy', proxy);

    let systemProxy = 'direct://';
    const proxyMatch = proxy.match(/PROXY (.*)/i);
    if (proxyMatch.length) {
      systemProxy = proxyMatch[1].trim();
    }

    autorun(() => {
      const { isServiceProxyEnabled, isServiceProxyPremiumFeature } = stores.features.features;

      config.isEnabled = isServiceProxyEnabled !== undefined ? isServiceProxyEnabled : DEFAULT_FEATURES_CONFIG.isServiceProxyEnabled;
      config.isPremium = isServiceProxyPremiumFeature !== undefined ? isServiceProxyPremiumFeature : DEFAULT_FEATURES_CONFIG.isServiceProxyPremiumFeature;

      const services = stores.services.enabled;
      const isPremiumUser = stores.user.data.isPremium;

      services.forEach((service) => {
        let proxyHost = systemProxy;
        const s = session.fromPartition(`persist:service-${service.id}`);

        if (config.isEnabled && (isPremiumUser || !config.isPremium)) {
          const serviceProxyConfig = stores.settings.proxy[service.id];

          if (serviceProxyConfig && serviceProxyConfig.isEnabled && serviceProxyConfig.host) {
            proxyHost = serviceProxyConfig.host;
          }
        }

        s.setProxy({ proxyRules: proxyHost }, () => {
          debug(`Using proxy "${proxyHost}" for "${service.name}" (${service.id})`);
        });
      });
    });
  });
}