customizable database path

1 files changed, 13 insertions, 12 deletions

diff --git a/src/Main.hs b/src/Main.hs
index d69ced6..095cdbe 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -150,20 +150,21 @@ sendPendingBangs cfg pool = do
         "Please do something about it.\n"
 
 main :: IO ()
-main = runStdoutLoggingT $ withSqlitePool "banger.db" 2 $ \pool -> do
-    runResourceT $ flip runSqlPool pool $ runMigration migrateAll
+main = do
+    cfg <- getConfig
+    runStdoutLoggingT $ withSqlitePool (confDbPath cfg) 2 $ \pool -> do
+        runResourceT $ flip runSqlPool pool $ runMigration migrateAll
 
-    bs <- liftIO initBangState
-    cfg <- liftIO getConfig
-    $(logInfo) $ "Using config: " <> T.pack (show cfg)
+        bs <- liftIO initBangState
+        $(logInfo) $ "Using config: " <> T.pack (show cfg)
 
-    sendPbIO <- toIO $ sendPendingBangs cfg pool
-    notifyPb <- liftIO $ makeRunner sendPbIO 600000000 {- 10 min -}
+        sendPbIO <- toIO $ sendPendingBangs cfg pool
+        notifyPb <- liftIO $ makeRunner sendPbIO 600000000 {- 10 min -}
 
-    sApp <- liftIO $ toWaiApp $ Search bs pool cfg notifyPb -- includes middlewares
+        sApp <- liftIO $ toWaiApp $ Search bs pool cfg notifyPb -- includes middlewares
 
-    let settings = defaultSettings
-            & setHost (confHost cfg)
-            & setPort (confPort cfg)
+        let settings = defaultSettings
+                & setHost (confHost cfg)
+                & setPort (confPort cfg)
 
-    liftIO $ runSettings settings sApp
+        liftIO $ runSettings settings sApp