load .env once on main

1 files changed, 9 insertions, 2 deletions

diff --git a/main.go b/main.go
index 21786e8..59ccfa5 100644
--- a/main.go
+++ b/main.go
@@ -1,17 +1,24 @@
 package main
 
 import (
+        "log"
+        "os"
+
         "github.com/gin-gonic/contrib/static"
         "github.com/gin-gonic/gin"
-        "github.com/pektezol/leastportals/backend/controllers"
+        "github.com/joho/godotenv"
         "github.com/pektezol/leastportals/backend/database"
         "github.com/pektezol/leastportals/backend/routes"
 )
 
 func main() {
-        if controllers.GetEnvKey("ENV") == "PROD" {
+        if os.Getenv("ENV") == "PROD" {
                 gin.SetMode(gin.ReleaseMode)
         }
+        err := godotenv.Load()
+        if err != nil {
+                log.Fatal("Error loading .env file")
+        }
         router := gin.Default()
         database.ConnectDB()
         router.Use(static.Serve("/", static.LocalFile("./frontend/dist", true)))