Checking for tables at startup

src/services/sql_init.js

@@ -15,9 +15,24 @@ const dbReady = utils.deferred();
 
 cls.init(initDbConnection);
 
-function schemaExists() {
+function checkTableExistsInDb(TableName) {
     return !!sql.getValue(`SELECT name FROM sqlite_master
-                                 WHERE type = 'table' AND name = 'options'`);
+                                 WHERE type = 'table' AND name = '${TableName}'`);
+}
+
+function schemaExists() {
+    return checkTableExistsInDb('options');
+}
+
+function readTableNamesFromSchema(schema) {
+    return schema.split(";").reduce((acc, str) => str.includes("CREATE TABLE IF NOT EXISTS ") ?
+        [...acc, str.replaceAll("\n", '').replace(/^.*CREATE TABLE IF NOT EXISTS[ \t]+"([^"]*)".*/, "$1")] : acc, []);
+}
+
+function getTableInitializingQuery(schema, tblName) {
+    return schema.split(";").filter((str) => {
+        return str.includes(tblName);
+    });
 }
 
 function isDbInitialized() {
@@ -38,6 +53,18 @@ async function initDbConnection() {
         return;
     }
 
+    const schema = fs.readFileSync(`${resourceDir.DB_INIT_DIR}/schema.sql`, 'UTF-8');
+
+    readTableNamesFromSchema(schema).forEach((tblName) => {
+        if (!checkTableExistsInDb(tblName)) {
+            log.info(`Table ${tblName} not found in DB, creating ...`);
+            getTableInitializingQuery(schema, tblName).forEach((sQuery) => {
+                sql.execute(sQuery);
+                log.info(sQuery);
+            });
+        }
+    })
+
     await migrationService.migrateIfNecessary();
 
     sql.execute('CREATE TEMP TABLE "param_list" (`paramId` TEXT NOT NULL PRIMARY KEY)');