Differential D7154 Diff 24367 web/database/worker/db-worker.js

Changeset View

Standalone View

web/database/worker/db-worker.js

// @flow		// @flow

import localforage from 'localforage';		import localforage from 'localforage';
import initSqlJs, { type SqliteDatabase } from 'sql.js';		import initSqlJs, { type SqliteDatabase } from 'sql.js';

import type {		import type {
ClientDBDraftStoreOperation,		ClientDBDraftStoreOperation,
DraftStoreOperation,		DraftStoreOperation,
} from 'lib/types/draft-types.js';		} from 'lib/types/draft-types.js';
		import type { ClientDBStore } from 'lib/types/store-ops-types.js';

import {		import {
type SharedWorkerMessageEvent,		type SharedWorkerMessageEvent,
type WorkerRequestMessage,		type WorkerRequestMessage,
type WorkerResponseMessage,		type WorkerResponseMessage,
workerRequestMessageTypes,		workerRequestMessageTypes,
workerResponseMessageTypes,		workerResponseMessageTypes,
type WorkerRequestProxyMessage,		type WorkerRequestProxyMessage,
} from '../../types/worker-types.js';		} from '../../types/worker-types.js';
import { getSQLiteDBVersion, setupSQLiteDB } from '../queries/db-queries.js';		import { getSQLiteDBVersion, setupSQLiteDB } from '../queries/db-queries.js';
import {		import {
		getAllDrafts,
moveDraft,		moveDraft,
removeAllDrafts,		removeAllDrafts,
updateDraft,		updateDraft,
} from '../queries/draft-queries.js';		} from '../queries/draft-queries.js';
import { SQLITE_CONTENT, SQLITE_ENCRYPTION_KEY } from '../utils/constants.js';		import { SQLITE_CONTENT, SQLITE_ENCRYPTION_KEY } from '../utils/constants.js';
import { generateDatabaseCryptoKey } from '../utils/worker-crypto-utils.js';		import { generateDatabaseCryptoKey } from '../utils/worker-crypto-utils.js';

const localforageConfig: PartialConfig = {		const localforageConfig: PartialConfig = {
▲ Show 20 Lines • Show All 47 Lines • ▼ Show 20 Lines	if (operation.type === 'remove_all') {
const { oldKey, newKey } = operation.payload;		const { oldKey, newKey } = operation.payload;
moveDraft(sqliteDb, oldKey, newKey);		moveDraft(sqliteDb, oldKey, newKey);
} else {		} else {
throw new Error('Unsupported draft operation');		throw new Error('Unsupported draft operation');
}		}
}		}
}		}

		function getClientStore(): ClientDBStore {
		if (!sqliteDb) {
		throw new Error('Database not initialized');
		}
		return {
		drafts: getAllDrafts(sqliteDb),
		messages: [],
		threads: [],
		};
		}

async function processAppRequest(		async function processAppRequest(
message: WorkerRequestMessage,		message: WorkerRequestMessage,
): Promise<?WorkerResponseMessage> {		): Promise<?WorkerResponseMessage> {
if (message.type === workerRequestMessageTypes.PING) {		if (message.type === workerRequestMessageTypes.PING) {
return {		return {
type: workerResponseMessageTypes.PONG,		type: workerResponseMessageTypes.PONG,
text: 'PONG',		text: 'PONG',
};		};
Show All 9 Lines	): Promise<?WorkerResponseMessage> {
} else if (		} else if (
message.type === workerRequestMessageTypes.PROCESS_STORE_OPERATIONS		message.type === workerRequestMessageTypes.PROCESS_STORE_OPERATIONS
) {		) {
const { draftStoreOperations } = message.storeOperations;		const { draftStoreOperations } = message.storeOperations;
if (draftStoreOperations) {		if (draftStoreOperations) {
processDraftStoreOperations(draftStoreOperations);		processDraftStoreOperations(draftStoreOperations);
}		}
return;		return;
		} else if (message.type === workerRequestMessageTypes.GET_CLIENT_STORE) {
		return {
		type: workerResponseMessageTypes.CLIENT_STORE,
		store: getClientStore(),
		};
}		}

throw new Error('Request type not supported');		throw new Error('Request type not supported');
}		}

function connectHandler(event: SharedWorkerMessageEvent) {		function connectHandler(event: SharedWorkerMessageEvent) {
if (!event.ports.length) {		if (!event.ports.length) {
return;		return;
Show All 30 Lines