121 lines
3 KiB
TypeScript
121 lines
3 KiB
TypeScript
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import type { ElectronApplication, Page } from 'playwright';
|
|
import { _electron as electron } from 'playwright';
|
|
|
|
import type {
|
|
IPCRequest as ChallengeRequestType,
|
|
IPCResponse as ChallengeResponseType,
|
|
} from '../challenge';
|
|
|
|
export type AppLoadedInfoType = Readonly<{
|
|
loadTime: number;
|
|
messagesPerSec: number;
|
|
}>;
|
|
|
|
export type MessageSendInfoType = Readonly<{
|
|
timestamp: number;
|
|
delta: number;
|
|
}>;
|
|
|
|
export type ConversationOpenInfoType = Readonly<{
|
|
delta: number;
|
|
}>;
|
|
|
|
export type StorageServiceInfoType = Readonly<{
|
|
manifestVersion: number;
|
|
}>;
|
|
|
|
export type AppOptionsType = Readonly<{
|
|
main: string;
|
|
args: ReadonlyArray<string>;
|
|
config: string;
|
|
}>;
|
|
|
|
export class App {
|
|
private privApp: ElectronApplication | undefined;
|
|
|
|
constructor(private readonly options: AppOptionsType) {}
|
|
|
|
public async start(): Promise<void> {
|
|
this.privApp = await electron.launch({
|
|
executablePath: this.options.main,
|
|
args: this.options.args.slice(),
|
|
env: {
|
|
...process.env,
|
|
SIGNAL_CI_CONFIG: this.options.config,
|
|
},
|
|
locale: 'en',
|
|
});
|
|
}
|
|
|
|
public async waitForProvisionURL(): Promise<string> {
|
|
return this.waitForEvent('provisioning-url');
|
|
}
|
|
|
|
public async waitUntilLoaded(): Promise<AppLoadedInfoType> {
|
|
return this.waitForEvent('app-loaded');
|
|
}
|
|
|
|
public async waitForMessageSend(): Promise<MessageSendInfoType> {
|
|
return this.waitForEvent('message:send-complete');
|
|
}
|
|
|
|
public async waitForConversationOpen(): Promise<ConversationOpenInfoType> {
|
|
return this.waitForEvent('conversation:open');
|
|
}
|
|
|
|
public async waitForChallenge(): Promise<ChallengeRequestType> {
|
|
return this.waitForEvent('challenge');
|
|
}
|
|
|
|
public async waitForStorageService(): Promise<StorageServiceInfoType> {
|
|
return this.waitForEvent('storageServiceComplete');
|
|
}
|
|
|
|
public async waitForManifestVersion(version: number): Promise<void> {
|
|
// eslint-disable-next-line no-constant-condition
|
|
while (true) {
|
|
// eslint-disable-next-line no-await-in-loop
|
|
const { manifestVersion } = await this.waitForStorageService();
|
|
if (manifestVersion >= version) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
public async solveChallenge(response: ChallengeResponseType): Promise<void> {
|
|
const window = await this.getWindow();
|
|
|
|
await window.evaluate(
|
|
`window.Signal.CI.solveChallenge(${JSON.stringify(response)})`
|
|
);
|
|
}
|
|
|
|
public async close(): Promise<void> {
|
|
await this.app.close();
|
|
}
|
|
|
|
public async getWindow(): Promise<Page> {
|
|
return this.app.firstWindow();
|
|
}
|
|
|
|
private async waitForEvent<T>(event: string): Promise<T> {
|
|
const window = await this.getWindow();
|
|
|
|
const result = await window.evaluate(
|
|
`window.Signal.CI.waitForEvent(${JSON.stringify(event)})`
|
|
);
|
|
|
|
return result as T;
|
|
}
|
|
|
|
private get app(): ElectronApplication {
|
|
if (!this.privApp) {
|
|
throw new Error('Call ElectronWrap.start() first');
|
|
}
|
|
|
|
return this.privApp;
|
|
}
|
|
}
|