aboutsummaryrefslogtreecommitdiffstats
path: root/packages/main/src/infrastructure/electron/impl/ElectronPartition.ts
blob: e60ce2167663f27b08bcdc2c9fc8427954dd14ea (plain) (blame)
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
57
58
59
/*
 * Copyright (C)  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
 */

import { Session, session } from 'electron';

import type Profile from '../../../stores/Profile';
import type { Partition } from '../types';

import type ElectronViewFactory from './ElectronViewFactory';

export default class ElectronPartition implements Partition {
  readonly id: string;

  readonly session: Session;

  constructor(profile: Profile, parent: ElectronViewFactory) {
    this.id = profile.id;
    this.session = session.fromPartition(`persist:${profile.id}`);
    this.session.setPermissionRequestHandler(
      (_webContents, permission, callback) => {
        // TODO Handle screen sharing.
        callback(permission === 'notifications');
      },
    );
    this.session.setUserAgent(parent.userAgents.serviceUserAgent());
    this.session.webRequest.onBeforeSendHeaders(
      ({ url, requestHeaders }, callback) => {
        callback({
          requestHeaders: {
            ...requestHeaders,
            'User-Agent': parent.userAgents.serviceUserAgent(url),
          },
        });
      },
    );
  }

  // eslint-disable-next-line class-methods-use-this -- Implementing interface method.
  dispose(): void {
    // No reactions to dispose yet.
  }
}