Added auto-db migration back.

SkewedAspect · May 20, 2024 · 8e61f93 · 8e61f93
1 parent 1ac07f4
commit 8e61f93
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/src/server/server.ts b/src/server/server.ts
@@ -44,7 +44,7 @@ import { errorLogger, requestLogger, serveIndex } from './routes/utils';
 import { setSIOInstance } from './utils/sio';
 import program from './utils/args';
 import { getVersion } from './utils/version';
-import { getDB } from './utils/database';
+import { getDB, runMigrations } from './utils/database';
 
 // ---------------------------------------------------------------------------------------------------------------------
 // Server Configuration
@@ -86,6 +86,16 @@ async function main() : Promise<void>
 
     await permsMan.init();
 
+    //------------------------------------------------------------------------------------------------------------------
+    // Database
+    //------------------------------------------------------------------------------------------------------------------
+
+    if(!devMode)
+    {
+        logger.info('Running database migrations...');
+        await runMigrations();
+    }
+
     //------------------------------------------------------------------------------------------------------------------
 
     const store = new KnexSessionStore({

diff --git a/src/server/utils/database.ts b/src/server/utils/database.ts
@@ -141,4 +141,25 @@ export async function getDB() : Promise<Knex>
     }
 }
 
+export async function runMigrations(runSeeds = true, convertToJs = true) : Promise<void>
+{
+    const db = await getDB();
+
+    if(convertToJs)
+    {
+        // Ensure that the migrations are in .js format
+        await db.update({ name: db.raw('replace(name, \'.ts\', \'.js\')') })
+            .from('knex_migrations');
+    }
+
+    // Run the migrations
+    await db.migrate.latest({ directory: './dist/server/knex/migrations' });
+
+    // Run the seeds
+    if(runSeeds)
+    {
+        await db.seed.run({ directory: './dist/server/knex/seeds' });
+    }
+}
+
 // ---------------------------------------------------------------------------------------------------------------------