1 files changed, 38 insertions, 2 deletions
diff --git a/server.go b/server.go
index 6720b41..b03b9b0 100644
--- a/server.go
+++ b/server.go
@@ -1,14 +1,50 @@
 package main
 
 import (
+	"github.com/joho/godotenv"
 	"github.com/justsaumit/go-fis-api/handlers"
 	"github.com/labstack/echo/v4"
+	"log"
+	"fmt"
+	"os"
 )
 
 func main() {
+	err := godotenv.Load() // Load .env file
+	if err != nil {
+		log.Fatal("Error loading .env file")
+	}
+
+	port := os.Getenv("PORT")
+	if port == "" {
+		port = "3000" // default port if not set
+	}
+
+	domain := os.Getenv("DOMAIN")
+	api_endpoint_url := os.Getenv("API_ENDPOINT_URL")
+	if domain == "" || api_endpoint_url == "" {
+		log.Println("Warning: DOMAIN and API_ENDPOINT_URL environment variable not set. Using localhost as default.")
+		domain = "localhost"
+		api_endpoint_url = "http://localhost:3000"
+	}
+	fmt.Println("Domain: " + domain)
+	fmt.Println("API Endpoint URL: " + api_endpoint_url)
+
+	certPath := "/etc/letsencrypt/live/" + domain + "/fullchain.pem"
+	keyPath := "/etc/letsencrypt/live/" + domain + "/privkey.pem"
+
 	e := echo.New()
 	e.POST("/upload", handlers.AddHash)
 	e.POST("/verify", handlers.VerifyHash)
-        //e.Logger.Fatal(e.Start(":3000"))
-	e.Logger.Fatal(e.StartTLS(":3000", "/etc/letsencrypt/live/draconyan.xyz/fullchain.pem", "/etc/letsencrypt/live/draconyan.xyz/privkey.pem"))
+
+	environment := os.Getenv("ENVIRONMENT")
+	switch environment {
+	case "development":
+		e.Logger.Fatal(e.Start(":" + port))
+	case "production":
+		e.Logger.Fatal(e.StartTLS(":"+port, certPath, keyPath))
+	default:
+		log.Printf("Unknown environment '%s', starting on default port %s\n", environment, port)
+		e.Logger.Fatal(e.Start(":" + port))
+	}
 }