From 911240877f437cc599f22744401afa2a5c5cde63 Mon Sep 17 00:00:00 2001
From: Shreerang Kale <shreerangkale@gmail.com>
Date: Fri, 7 Feb 2025 10:04:58 +0530
Subject: [PATCH] Take database path from env

---
 .env.example       | 2 ++
 src/data-source.ts | 6 +++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/.env.example b/.env.example
index ea255af..770e14f 100644
--- a/.env.example
+++ b/.env.example
@@ -26,3 +26,5 @@ WSOL_MINT_ADDRESS=So11111111111111111111111111111111111111112
 WSOL_LOCK_DURATION_IN_SECONDS=172800 # 48 hours
 REWARD_MULTIPLIER=4
 MAX_RETRIES_FOR_LOCK_TX=5
+
+DB_PATH='./database.sqlite'
diff --git a/src/data-source.ts b/src/data-source.ts
index df8bfdb..74e669f 100644
--- a/src/data-source.ts
+++ b/src/data-source.ts
@@ -1,12 +1,16 @@
 import { DataSource } from 'typeorm';
+import assert from 'assert';
 
 export async function initializeDataSource() {
   try {
+    const DB_PATH = process.env.DB_PATH;
+    assert(DB_PATH, 'DB_PATH not set');
+
     console.log('Initializing Data Source');
 
     const appDataSource = new DataSource({
       type: 'sqlite',
-      database: './database.sqlite',
+      database: DB_PATH,
       synchronize: true,
       logging: false,
       entities: [global.entities.Payment, global.entities.Tweet],