feat: create schema automatically (#96)

Author: kevcodez

src/server.ts

@@ -3,6 +3,10 @@ import { Server, IncomingMessage, ServerResponse } from 'http'
 import { runMigrations } from './utils/migrate'
 import { createServer } from './app'
 import pino from 'pino'
+import { getConfig } from './utils/config'
+import { Client } from 'pg'
+
+const config = getConfig()
 
 const logger = pino({
   formatters: {
@@ -23,8 +27,24 @@ const main = async () => {
   // Init config
   const port = process.env.PORT || 8080
 
+  // Init DB
+  const dbConfig = {
+    connectionString: config.DATABASE_URL,
+    connectionTimeoutMillis: 10_000,
+  }
+  const client = new Client(dbConfig)
+
   // Run migrations
-  await runMigrations()
+  try {
+    await client.connect()
+
+    // Ensure schema exists, not doing it via migration to not break current migration checksums
+    await client.query(`CREATE SCHEMA IF NOT EXISTS ${config.SCHEMA};`)
+
+    await runMigrations(client)
+  } finally {
+    await client.end()
+  }
 
   // Start the server
   app.listen({ port: Number(port), host: '0.0.0.0' }, (err, address) => {

src/utils/migrate.ts

@@ -5,44 +5,32 @@ import fs from 'node:fs'
 
 const config = getConfig()
 
-async function connectAndMigrate(
-  databaseUrl: string | undefined,
-  migrationsDirectory: string,
-  logOnError = false
-) {
+async function connectAndMigrate(client: Client, migrationsDirectory: string, logOnError = false) {
   if (!fs.existsSync(migrationsDirectory)) {
     console.log(`Migrations directory ${migrationsDirectory} not found, skipping`)
     return
   }
 
-  const dbConfig = {
-    connectionString: databaseUrl,
-    connectionTimeoutMillis: 10_000,
-  }
   const optionalConfig = {
     schemaName: config.SCHEMA,
     tableName: 'migrations',
   }
 
-  const client = new Client(dbConfig)
   try {
-    await client.connect()
     await migrate({ client }, migrationsDirectory, optionalConfig)
   } catch (error) {
     if (logOnError && error instanceof Error) {
       console.error('Migration error:', error.message)
     } else {
       throw error
     }
-  } finally {
-    await client.end()
   }
 }
 
-export async function runMigrations(): Promise<void> {
+export async function runMigrations(client: Client): Promise<void> {
   try {
     console.log('Running migrations')
-    await connectAndMigrate(config.DATABASE_URL, './db/migrations')
+    await connectAndMigrate(client, './db/migrations')
   } catch (error) {
     throw error
   } finally {