Create Thunder database if not exists

example.env

@@ -2,7 +2,7 @@ POSTGRES_USER=user
 POSTGRES_PASSWORD=password
 POSTGRES_HOSTNAME=postgres
 POSTGRES_PORT=5432
-POSTGRES_DATABASE=thunder-database
+POSTGRES_DATABASE=thunder_database
 
 APNS_KEY_ID=key_id
 APNS_TEAM_ID=team_id

src/database/database.ts

@@ -4,9 +4,28 @@ import { Sequelize } from "sequelize";
 import dotenv from "dotenv";
 dotenv.config();
 
-// Configure database
-const sequelize = new Sequelize(
-  `postgres://${process.env.POSTGRES_USER ?? "user"}:${process.env.POSTGRES_PASSWORD ?? "password"}@${process.env.POSTGRES_HOSTNAME ?? "postgres"}:${process.env.POSTGRES_PORT ?? "5432"}/${process.env.POSTGRES_DATABASE ?? "thunder-database"}`
+// Connect to the default Postgres db
+var sequelize = new Sequelize(
+  `postgres://${process.env.POSTGRES_USER ?? "user"}:${process.env.POSTGRES_PASSWORD ?? "password"}@${process.env.POSTGRES_HOSTNAME ?? "postgres"}:${process.env.POSTGRES_PORT ?? "5432"}/postgres`
 );
 
-export default sequelize;
+// Connect to a default database
+async function ensureThunderDatabaseExists() {
+  const thunderDatabase = process.env.POSTGRES_DATABASE ?? "thunder_database";
+
+  // Check if the Thunder database exists
+  const result = await sequelize.query(`SELECT FROM pg_database WHERE datname = '${thunderDatabase}'`);
+  const exists = result[0].length > 0;
+
+  // Create if not
+  if (!exists) {
+    await sequelize.query(`CREATE DATABASE ${thunderDatabase}`);
+  }
+
+  // Connect to the Thunder database
+  sequelize = new Sequelize(
+    `postgres://${process.env.POSTGRES_USER ?? "user"}:${process.env.POSTGRES_PASSWORD ?? "password"}@${process.env.POSTGRES_HOSTNAME ?? "postgres"}:${process.env.POSTGRES_PORT ?? "5432"}/${thunderDatabase}`
+  );
+}
+
+export { ensureThunderDatabaseExists, sequelize };

src/database/models/account_notification.ts

@@ -1,6 +1,6 @@
 import { DataTypes, Model } from "sequelize";
 
-import sequelize from "../database";
+import { sequelize } from "../database";
 
 class AccountNotification extends Model {}
 

src/index.ts

@@ -8,7 +8,7 @@ import {
 } from "./types/notification_request";
 
 // Database
-import sequelize from "./database/database";
+import { ensureThunderDatabaseExists, sequelize } from "./database/database";
 
 // Helper functions
 import { addAccountNotification } from "./notifications/notifications";
@@ -70,6 +70,7 @@ app.delete("/notifications", (req, res) => {
 app.listen(port, async () => {
   console.log(`Server is running on http://localhost:${port}`);
 
+  await ensureThunderDatabaseExists();
   await sequelize.sync({ force: DEBUG_MODE });
   console.log("All models were synchronized successfully.");
 });