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);
}
}
|