Start refactoring

config/config.go

@@ -1,84 +1,69 @@
 package config
 
 import (
-	"flag"
+	"fmt"
 	"golang.org/x/crypto/bcrypt"
 	"gopkg.in/yaml.v3"
-	"log"
 	"os"
 )
 
-type Configuration struct {
-	Server   ServerConfiguration   `yaml:"server"`
-	Database DatabaseConfiguration `yaml:"database"`
-	Features FeaturesConfiguration `yaml:"features"`
-	Path     PathConfiguration     `yaml:"path"`
-}
+type (
+	Configuration struct {
+		Server   ServerConfiguration   `yaml:"server"`
+		Database DatabaseConfiguration `yaml:"database"`
+		Features FeaturesConfiguration `yaml:"features"`
+		Path     PathConfiguration     `yaml:"path"`
+	}
 
-type PathConfiguration struct {
-	Cache   string `yaml:"cache"`
-	Storage string `yaml:"storage"`
-}
+	PathConfiguration struct {
+		Cache   string `yaml:"cache"`
+		Storage string `yaml:"storage"`
+		RSAKey  string `yaml:"rsa_key"`
+	}
 
-type ServerConfiguration struct {
-	Port int `yaml:"port"`
-}
+	ServerConfiguration struct {
+		Port         int  `yaml:"port"`
+		PingEndPoint bool `yaml:"ping"`
+		Compress     bool `yaml:"compress"`
+	}
 
-type DatabaseConfiguration struct {
-	Host     string  `yaml:"host"`
-	Port     int     `yaml:"port"`
-	Username string  `yaml:"username"`
-	Password *string `yaml:"password"`
-}
+	DatabaseConfiguration struct {
+		URL string `yaml:"url"`
+	}
 
-type FeaturesConfiguration struct {
-	AllowRegister    bool `yaml:"allow_register"`
-	PasswordHashCost *int `yaml:"password_hash_cost"`
-}
+	FeaturesConfiguration struct {
+		AllowRegister    bool `yaml:"allow_register"`
+		PasswordHashCost *int `yaml:"password_hash_cost"`
+	}
+)
 
-var currentConfig *Configuration
-
-func Init() {
-	path := flag.String("config", "./config.yml", "Set the configuration file path")
-	flag.Parse()
-	configYamlContent, err := os.ReadFile(*path)
+func Load(path string) (Configuration, error) {
+	configYamlContent, err := os.ReadFile(path)
 	if err != nil {
-		log.Fatal(err)
+		return Configuration{}, err
 	}
-	err = yaml.Unmarshal(configYamlContent, &currentConfig)
-	if err != nil {
-		log.Fatalf("error: %s", err)
+	var config Configuration
+	if err := yaml.Unmarshal(configYamlContent, &config); err != nil {
+		return Configuration{}, err
 	}
-	checkConfig()
+	if err := checkConfig(config); err != nil {
+		return Configuration{}, err
+	}
+	return config, nil
 }
 
-func checkConfig() {
-	if currentConfig.Features.PasswordHashCost == nil {
-		currentConfig.Features.PasswordHashCost = new(int)
-		*currentConfig.Features.PasswordHashCost = bcrypt.DefaultCost
-	} else if *currentConfig.Features.PasswordHashCost < bcrypt.MinCost && *currentConfig.Features.PasswordHashCost > bcrypt.MaxCost {
-		log.Fatalf("password_hash_cost is not on the supported range (%d < x < %d)", bcrypt.MinCost, bcrypt.MaxCost)
+func checkConfig(c Configuration) error {
+	if c.Features.PasswordHashCost == nil {
+		c.Features.PasswordHashCost = new(int)
+		*c.Features.PasswordHashCost = bcrypt.DefaultCost
+	} else if *c.Features.PasswordHashCost < bcrypt.MinCost && *c.Features.PasswordHashCost > bcrypt.MaxCost {
+		return fmt.Errorf("password_hash_cost is not on the supported range (%d < x < %d)", bcrypt.MinCost, bcrypt.MaxCost)
 	}
-	if _, err := os.Stat(currentConfig.Path.Storage); err != nil {
-		log.Fatal(err)
+	if _, err := os.Stat(c.Path.Storage); err != nil {
+		return nil
 	}
-	if _, err := os.Stat(currentConfig.Path.Cache); err != nil {
-		log.Fatal(err)
+	if _, err := os.Stat(c.Path.Cache); err != nil {
+		return err
 	}
-}
-
-func Database() *DatabaseConfiguration {
-	return &currentConfig.Database
-}
-
-func Features() *FeaturesConfiguration {
-	return &currentConfig.Features
-}
-
-func Path() *PathConfiguration {
-	return &currentConfig.Path
-}
-
-func Server() *ServerConfiguration {
-	return &currentConfig.Server
+	return nil
 }