check table exists

Author: phartenfeller

src/lib/sqlite/initStorages.ts

@@ -0,0 +1,29 @@
+import type { TableStructure } from '../../routes/api/data/types';
+import { sendMsgToWorker } from './messageBus';
+import { WorkerMessageTypes, type WorkerMessage, type TableExistsResponseData } from './types';
+
+const storages = ['customers_v1'];
+
+async function getStructure(storage: string): Promise<TableStructure> {
+	const res = await fetch(`/api/data/${storage}/structure`);
+	const data = (await res.json()) as TableStructure;
+	return data;
+}
+
+export default async function initStorages() {
+	for (const storageId of storages) {
+		const res = (await sendMsgToWorker({
+			storageId,
+			type: WorkerMessageTypes.TABLE_EXISTS,
+			expectedType: WorkerMessageTypes.TABLE_EXISTS_RESPONSE,
+			data: undefined
+		})) as WorkerMessage<TableExistsResponseData>;
+
+		if (res.data.tableExists) {
+			console.log(`Table ${storageId} exists. Has data: ${res.data.hasData}`);
+		} else {
+			const structure = await getStructure(storageId);
+			console.log(`Table ${storageId} does not exist. Creating...`, structure);
+		}
+	}
+}

src/lib/sqlite/loadWorker.ts

@@ -1,3 +1,4 @@
+import initStorages from './initStorages';
 import { initMsgBus, sendMsgToWorker } from './messageBus';
 import { WorkerMessageTypes } from './types';
 
@@ -15,4 +16,6 @@ export default async function initWorker() {
 	});
 
 	console.log('Init worker response:', res);
+
+	initStorages();
 }

src/lib/sqlite/types.ts

@@ -1,6 +1,8 @@
 export enum WorkerMessageTypes {
 	INIT_DB,
-	INIT_DB_RESPONSE
+	INIT_DB_RESPONSE,
+	TABLE_EXISTS,
+	TABLE_EXISTS_RESPONSE
 }
 
 export type WorkerMessageBase = {
@@ -10,3 +12,9 @@ export type WorkerMessageBase = {
 };
 
 export type WorkerMessage<T> = { data: T } & WorkerMessageBase;
+
+export type TableExistsResponseData = {
+	tableExists: boolean;
+	hasData: boolean;
+	errorMsg?: string;
+};

src/lib/sqlite/worker/storageHandlers.ts

@@ -0,0 +1,28 @@
+import type { TableExistsResponseData, WorkerMessage } from '../types';
+import { tableExists, tableHasData } from './util/tableInfo';
+
+export function handleTableExists(data: WorkerMessage<unknown>): TableExistsResponseData {
+	try {
+		const { storageId } = data;
+
+		const resData = <TableExistsResponseData>{};
+		resData.tableExists = tableExists(storageId);
+
+		if (resData.tableExists) {
+			resData.hasData = tableHasData(storageId);
+		} else {
+			resData.hasData = false;
+		}
+
+		return resData;
+	} catch (err) {
+		const msg = `Error checking if table exists: ${err}`;
+		console.error(msg);
+
+		return {
+			tableExists: false,
+			hasData: false,
+			errorMsg: msg
+		};
+	}
+}

src/lib/sqlite/worker/util/tableInfo.ts

@@ -0,0 +1,21 @@
+import { db } from '../initDb';
+
+export function tableExists(table: string) {
+	const sql = `SELECT count(*) as "cnt" FROM sqlite_master WHERE type='table' AND name= $table`;
+
+	const res = db.selectObject(sql, {
+		$table: table
+	}) as { cnt: number };
+
+	return res.cnt > 0;
+}
+
+export function tableHasData(table: string) {
+	const sql = `SELECT count(*) as "cnt" FROM ${table}`;
+
+	const res = db.selectObject(sql, {
+		$table: table
+	}) as { cnt: number };
+
+	return res.cnt > 0;
+}

src/lib/sqlite/worker/worker.ts

@@ -1,6 +1,7 @@
 /* eslint-disable no-case-declarations */
-import { WorkerMessageTypes, type WorkerMessage } from '../types';
+import { WorkerMessageTypes, type WorkerMessage, type TableExistsResponseData } from '../types';
 import { initDb } from './initDb';
+import { handleTableExists } from './storageHandlers';
 
 console.log('worker loaded');
 
@@ -29,6 +30,18 @@ function sendMsgToMain(obj: WorkerMessage<unknown>) {
 
 				break;
 
+			case WorkerMessageTypes.TABLE_EXISTS:
+				const tableExistData = handleTableExists(data);
+
+				const tableExistsResult: WorkerMessage<TableExistsResponseData> = {
+					type: WorkerMessageTypes.TABLE_EXISTS_RESPONSE,
+					messageId: data.messageId,
+					storageId: data.storageId,
+					data: tableExistData
+				};
+				sendMsgToMain(tableExistsResult);
+				break;
+
 			default:
 				throw new Error(`Unknown message type: ${data.type}`);
 		}