more refactor

main.go

@@ -1,55 +1,13 @@
 package main
 
 import (
-	"auth_service/api/auth"
-	"auth_service/api/users"
-	"auth_service/database"
-	"log"
-	"net/http"
-	"os"
-
-	"github.com/joho/godotenv"
+	"auth_service/server"
+	"auth_service/utils"
 )
 
 func main() {
-	err := godotenv.Load("auth_service.config")
-	if err != nil {
-		log.Fatal("Error loading .env file")
-	}
-	database.Init()
-	database.Migrate()
-
-	if os.Getenv("USERS_FILE") != "" {
-		allusers := database.GetUsersFromFile(os.Getenv("USERS_FILE"))
-		for i := range allusers {
-			database.CreateUser(&allusers[i])
-		}
-	}
-
-	http.HandleFunc("/api/token", auth.Token)
-	http.HandleFunc("/api/token/verify", auth.VerifyToken)
-	http.HandleFunc("/api/access", auth.VerifyToken)
-	if os.Getenv("ENABLE_SELF_REGISTER") == "true" {
-		http.HandleFunc("/api/register", users.RegisterUser)
-	}
-
-	http.HandleFunc("/api/users", users.GetAllUsers)
-	http.HandleFunc("/api/users/user", users.GetUser)
-	http.HandleFunc("/api/users/password", users.ChangePassword)
-	http.HandleFunc("/api/users/change", users.ChangeUser)
-	http.HandleFunc("/api/users/del", users.DelUser)
-	http.HandleFunc("/api/users/add", users.AddUser)
-	if os.Getenv("ENABLE_TLS") == "true" {
-		http.ListenAndServeTLS(
-			os.Getenv("SERVER_ADDRESS")+":"+os.Getenv("SERVER_PORT"),
-			os.Getenv("CERT_FILE"),
-			os.Getenv("KEY_FILE"),
-			nil,
-		)
-	} else {
-		http.ListenAndServe(
-			os.Getenv("SERVER_ADDRESS")+":"+os.Getenv("SERVER_PORT"),
-			nil,
-		)
-	}
+	utils.LoadEnv()
+	utils.InitDatabase()
+	server.Configure()
+	server.Run()
 }

server/Configure.go

@@ -0,0 +1,23 @@
+package server
+
+import (
+	"auth_service/api/auth"
+	"auth_service/api/users"
+	"net/http"
+	"os"
+)
+
+func Configure() {
+	http.HandleFunc("/api/token", auth.Token)
+	http.HandleFunc("/api/token/verify", auth.VerifyToken)
+	http.HandleFunc("/api/access", auth.VerifyToken)
+	if os.Getenv("ENABLE_SELF_REGISTER") == "true" {
+		http.HandleFunc("/api/register", users.RegisterUser)
+	}
+	http.HandleFunc("/api/users", users.GetAllUsers)
+	http.HandleFunc("/api/users/user", users.GetUser)
+	http.HandleFunc("/api/users/password", users.ChangePassword)
+	http.HandleFunc("/api/users/change", users.ChangeUser)
+	http.HandleFunc("/api/users/del", users.DelUser)
+	http.HandleFunc("/api/users/add", users.AddUser)
+}

server/Run.go

@@ -0,0 +1,22 @@
+package server
+
+import (
+	"net/http"
+	"os"
+)
+
+func Run() {
+	if os.Getenv("ENABLE_TLS") == "true" {
+		http.ListenAndServeTLS(
+			os.Getenv("SERVER_ADDRESS")+":"+os.Getenv("SERVER_PORT"),
+			os.Getenv("CERT_FILE"),
+			os.Getenv("KEY_FILE"),
+			nil,
+		)
+	} else {
+		http.ListenAndServe(
+			os.Getenv("SERVER_ADDRESS")+":"+os.Getenv("SERVER_PORT"),
+			nil,
+		)
+	}
+}

utils/InitDatabase.go

@@ -0,0 +1,18 @@
+package utils
+
+import (
+	"auth_service/database"
+	"os"
+)
+
+func InitDatabase() {
+	database.Init()
+	database.Migrate()
+
+	if os.Getenv("USERS_FILE") != "" {
+		allusers := database.GetUsersFromFile(os.Getenv("USERS_FILE"))
+		for i := range allusers {
+			database.CreateUser(&allusers[i])
+		}
+	}
+}

utils/LoadEnv.go

@@ -0,0 +1,14 @@
+package utils
+
+import (
+	"log"
+
+	"github.com/joho/godotenv"
+)
+
+func LoadEnv() {
+	err := godotenv.Load("auth_service.config")
+	if err != nil {
+		log.Fatal("Error loading .env file")
+	}
+}