From c86a7ae0e40a2b27facbb7e623705687a5d6ac3d Mon Sep 17 00:00:00 2001
From: Leons Aleksandrovs <58330666+Skrazzo@users.noreply.github.com>
Date: Sun, 13 Jul 2025 23:05:55 +0300
Subject: [PATCH] fix(domain): added env domain for cookies

---
 .env-example                     | 3 ++-
 backend/config/config.go         | 2 ++
 backend/controllers/user/user.go | 2 +-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/.env-example b/.env-example
index 2b4b511..b3a959b 100644
--- a/.env-example
+++ b/.env-example
@@ -6,4 +6,5 @@ CHATGPT_MODEL=gpt-4o
 JWT_SECRET=just a random string here
 # Set to false to disable registration
 ALLOW_REGISTER=true
-
+# Working domain
+DOMAIN=localhost
diff --git a/backend/config/config.go b/backend/config/config.go
index 3864ea8..323fbdc 100644
--- a/backend/config/config.go
+++ b/backend/config/config.go
@@ -27,4 +27,6 @@ func LoadEnv() {
 	log.Printf("[INFO] Register set to '%s'\n", Env["REGISTER"])
 	Env["CHATGPT_MODEL"] = defaultValue(os.Getenv("CHATGPT_MODEL"), "gpt-4o")
 	log.Printf("[INFO] ChatGPT model set to '%s'\n", Env["CHATGPT_MODEL"])
+	Env["DOMAIN"] = defaultValue(os.Getenv("DOMAIN"), "localhost")
+	log.Printf("[INFO] Domain set to '%s'\n", Env["DOMAIN"])
 }
diff --git a/backend/controllers/user/user.go b/backend/controllers/user/user.go
index bec5300..9bc7be1 100644
--- a/backend/controllers/user/user.go
+++ b/backend/controllers/user/user.go
@@ -134,7 +134,7 @@ func Login(c *gin.Context) {
 	// Return token as cookie
 	secureCookie := config.Env["Environment"] != "dev" // In dev environment cookie wont be secure
 	// 3600S -> 1H * 24H -> 1D * 7 -> 1W
-	c.SetCookie("jwt-token", signedToken, 3600*24*7, "/", "localhost", secureCookie, true)
+	c.SetCookie("jwt-token", signedToken, 3600*24*7, "/", config.Env["DOMAIN"], secureCookie, true)
 
 	// Return successful login
 	res.Success(c, gin.H{"message": "Successfully logged in"})