1 files changed, 9 insertions, 7 deletions

diff --git a/api/internal/database/database.go b/api/internal/database/database.go
index 7af9780..1866655 100644
--- a/api/internal/database/database.go
+++ b/api/internal/database/database.go
@@ -4,17 +4,19 @@ import (
         "database/sql"
         _ "github.com/mattn/go-sqlite3"
         "log"
+        "path/filepath"
+        "water/api/internal/config"
 )
 
-func SetupDatabase() {
-        _, err := sql.Open("sqlite3", "water.sqlite3")
+func EstablishDBConnection() *sql.DB {
+        c, err := config.Load()
+
+        driver := c.GetString("DB_DRIVER")
+        path, err := filepath.Abs(c.GetString("DB_PATH"))
         if err != nil {
-                log.Fatal(err)
+                log.Fatal("There was and error getting the absolute path of the database.")
         }
-}
-
-func EstablishDBConnection() *sql.DB {
-        db, err := sql.Open("sqlite3", "../db/water.sqlite3")
+        db, err := sql.Open(driver, path)
         if err != nil {
                 panic(err)
         }