84 lines
2.1 KiB
TypeScript
84 lines
2.1 KiB
TypeScript
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import { ipcRenderer } from 'electron';
|
|
import * as log from '../logging/log';
|
|
import createTaskWithTimeout from '../textsecure/TaskWithTimeout';
|
|
import { explodePromise } from '../util/explodePromise';
|
|
import { missingCaseError } from '../util/missingCaseError';
|
|
|
|
const SQL_READ_KEY = 'sql-channel:read';
|
|
const SQL_WRITE_KEY = 'sql-channel:write';
|
|
const SQL_REMOVE_DB_KEY = 'sql-channel:remove-db';
|
|
let activeJobCount = 0;
|
|
let resolveShutdown: (() => void) | undefined;
|
|
let shutdownPromise: Promise<void> | null = null;
|
|
|
|
export enum AccessType {
|
|
Read = 'Read',
|
|
Write = 'Write',
|
|
}
|
|
|
|
export async function ipcInvoke<T>(
|
|
access: AccessType,
|
|
name: string,
|
|
args: ReadonlyArray<unknown>
|
|
): Promise<T> {
|
|
const fnName = String(name);
|
|
|
|
if (shutdownPromise && name !== 'close') {
|
|
throw new Error(
|
|
`Rejecting SQL channel job (${access}, ${fnName}); ` +
|
|
'application is shutting down'
|
|
);
|
|
}
|
|
|
|
let channel: string;
|
|
if (access === AccessType.Read) {
|
|
channel = SQL_READ_KEY;
|
|
} else if (access === AccessType.Write) {
|
|
channel = SQL_WRITE_KEY;
|
|
} else {
|
|
throw missingCaseError(access);
|
|
}
|
|
|
|
activeJobCount += 1;
|
|
return createTaskWithTimeout(async () => {
|
|
try {
|
|
return await ipcRenderer.invoke(channel, name, ...args);
|
|
} finally {
|
|
activeJobCount -= 1;
|
|
if (activeJobCount === 0) {
|
|
resolveShutdown?.();
|
|
}
|
|
}
|
|
}, `SQL channel call (${access}, ${fnName})`)();
|
|
}
|
|
|
|
export async function doShutdown(): Promise<void> {
|
|
log.info(
|
|
`data.shutdown: shutdown requested. ${activeJobCount} jobs outstanding`
|
|
);
|
|
|
|
if (shutdownPromise) {
|
|
return shutdownPromise;
|
|
}
|
|
|
|
// No outstanding jobs, return immediately
|
|
if (activeJobCount === 0) {
|
|
return;
|
|
}
|
|
|
|
({ promise: shutdownPromise, resolve: resolveShutdown } =
|
|
explodePromise<void>());
|
|
|
|
try {
|
|
await shutdownPromise;
|
|
} finally {
|
|
log.info('data.shutdown: process complete');
|
|
}
|
|
}
|
|
|
|
export async function removeDB(): Promise<void> {
|
|
return ipcRenderer.invoke(SQL_REMOVE_DB_KEY);
|
|
}
|