From 44ada32d601e781ddcb784d06a72abf7d68e1b3d Mon Sep 17 00:00:00 2001
From: ACoolName <acoolname@acooldomain.co>
Date: Thu, 23 May 2024 23:27:06 +0300
Subject: [PATCH] added domain to config

---
 auth/auth.go       | 4 +++-
 models/config.go   | 1 +
 servers/servers.go | 4 +++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/auth/auth.go b/auth/auth.go
index b50a98a..b2aa52b 100644
--- a/auth/auth.go
+++ b/auth/auth.go
@@ -19,6 +19,7 @@ import (
 
 var secret []byte
 var method string
+var DOMAIN string
 
 type Connection struct {
 	DatabaseConnection *mongo.Client
@@ -222,7 +223,7 @@ func (con Connection) verify(c *gin.Context) {
 		}
 	}
 
-	c.Redirect(303, "http://%s/login")
+	c.Redirect(303, fmt.Sprintf("http://%s/login", DOMAIN))
 }
 
 func (con Connection) ServerAuthorized(permissions models.Permission) func(*gin.Context) bool {
@@ -260,6 +261,7 @@ func LoadGroup(group *gin.RouterGroup, client *mongo.Client, config models.Globa
 
 	secret = []byte(config.Key)
 	method = config.Algorithm
+	DOMAIN = config.Domain
 
 	group.POST("/signin", connection.signIn)
 	group.POST("/signup", AuthorizedTo(models.Admin), connection.signUp)
diff --git a/models/config.go b/models/config.go
index 1feb802..dc8842c 100644
--- a/models/config.go
+++ b/models/config.go
@@ -11,4 +11,5 @@ type GlobalConfig struct {
 	Email     EmailConfig
 	Key       string
 	Algorithm string
+	Domain    string
 }
diff --git a/servers/servers.go b/servers/servers.go
index 2e08a90..8fe1fd9 100644
--- a/servers/servers.go
+++ b/servers/servers.go
@@ -31,7 +31,7 @@ import (
 
 const FILE_BROWSER_IMAGE = "filebrowser/filebrowser:latest"
 
-var DOMAIN = os.Getenv("SERVER_DOMAIN")
+var DOMAIN string = ""
 
 type Connection struct {
 	databaseConnection *mongo.Client
@@ -797,6 +797,8 @@ func LoadGroup(group *gin.RouterGroup, mongo_client *mongo.Client, config models
 	}
 	defer apiClient.Close()
 
+	DOMAIN = config.Domain
+
 	connection := Connection{databaseConnection: mongo_client, dockerClient: apiClient}
 	authConnection := auth.Connection{DatabaseConnection: mongo_client}