aboutsummaryrefslogtreecommitdiffstats
path: root/packages/main/src/infrastructure/electron/impl/ElectronViewFactory.ts
blob: f8b4a362819ca1ffa56715859416e9b74a6b8c96 (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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
 * 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 { readFile } from 'node:fs/promises';

import { ServiceToMainIpcMessage } from '@sophie/service-shared';
import { ipcMain, WebSource } from 'electron';

import type MainStore from '../../../stores/MainStore';
import type Profile from '../../../stores/Profile';
import type Service from '../../../stores/Service';
import { getLogger } from '../../../utils/log';
import type Resources from '../../resources/Resources';
import type UserAgents from '../UserAgents';
import type { MainWindow, Partition, ServiceView, ViewFactory } from '../types';

import ElectronMainWindow from './ElectronMainWindow';
import ElectronPartition from './ElectronPartition';
import ElectronServiceView from './ElectronServiceView';

const log = getLogger('ElectronViewFactory');

export default class ElectronViewFactory implements ViewFactory {
  private readonly webContentsIdToServiceView = new Map<
    number,
    ElectronServiceView
  >();

  private serviceInjectSource: WebSource | undefined;

  constructor(
    readonly userAgents: UserAgents,
    readonly resources: Resources,
    readonly devMode: boolean,
  ) {
    ipcMain.handle(ServiceToMainIpcMessage.ApiExposedInMainWorld, (event) => {
      if (!this.webContentsIdToServiceView.has(event.sender.id)) {
        log.error(
          'Unexpected',
          ServiceToMainIpcMessage.ApiExposedInMainWorld,
          'IPC message from webContents',
          event.sender.id,
        );
        throw new Error('Invalid IPC call');
      }
      if (this.serviceInjectSource === undefined) {
        log.error('Service inject source was not loaded');
      }
      return this.serviceInjectSource;
    });
  }

  async createMainWindow(store: MainStore): Promise<MainWindow> {
    const mainWindow = new ElectronMainWindow(store, this);
    await mainWindow.loadInterface();
    return mainWindow;
  }

  createPartition(profile: Profile): Partition {
    return new ElectronPartition(profile, this);
  }

  createServiceView(service: Service, partition: Partition): ServiceView {
    if (partition instanceof ElectronPartition) {
      const serviceView = new ElectronServiceView(
        service,
        this.resources,
        partition,
        this,
      );
      this.webContentsIdToServiceView.set(
        serviceView.webContentsId,
        serviceView,
      );
      return serviceView;
    }
    throw new TypeError('Unexpected ProfileSession is not a WrappedSession');
  }

  async loadServiceInject(): Promise<void> {
    const injectPackage = 'service-inject';
    const injectFile = 'index.js';
    const injectPath = this.resources.getPath(injectPackage, injectFile);
    this.serviceInjectSource = {
      code: await readFile(injectPath, 'utf-8'),
      url: this.resources.getFileURL(injectPackage, injectFile),
    };
  }

  dispose(): void {
    if (this.webContentsIdToServiceView.size > 0) {
      throw new Error(
        'Must dispose all ServiceView instances before disposing ViewFactory',
      );
    }
    ipcMain.removeHandler(ServiceToMainIpcMessage.ApiExposedInMainWorld);
  }

  unregisterServiceView(id: number): void {
    this.webContentsIdToServiceView.delete(id);
  }
}