legalgo-BE-go/config/configs.go

package config

import (
	"fmt"
	"os"
	"sync"

	"github.com/spf13/viper"
	_ "gopkg.in/yaml.v3"
)

const (
	YAML_PATH        = "env/%s"
	ENV_MODE         = "ENV_MODE"
	DEFAULT_ENV_MODE = "local"
)

var (
	validEnvMode = map[string]struct{}{
		"local":      {},
		"staging":    {},
		"production": {},
		"debug":      {},
	}
)

type Config struct {
	Server    Server    `mapstructure:"server"`
	Database  Database  `mapstructure:"postgresql"`
	Jwt       Jwt       `mapstructure:"jwt"`
	OSSConfig OSSConfig `mapstructure:"oss"`
	Redis     Redis     `mapstructure:"redis"`
}

var (
	config     *Config
	configOnce sync.Once
)

func LoadConfig() *Config {
	envMode := os.Getenv(ENV_MODE)
	if _, ok := validEnvMode[envMode]; !ok {
		envMode = DEFAULT_ENV_MODE // default env mode
	}
	cfgFilePath := fmt.Sprintf(YAML_PATH, envMode)

	configOnce.Do(func() {
		v := viper.New()
		v.SetConfigType("yaml")
		v.AddConfigPath(".")
		v.SetConfigName(cfgFilePath)
		if err := v.ReadInConfig(); err != nil {
			panic(fmt.Errorf("failed to read config file: %s", err))
		}

		config = &Config{}
		if err := v.Unmarshal(config); err != nil {
			panic(fmt.Errorf("failed to unmarshal config: %s", err))
		}
	})

	return config
}

func (c *Config) Auth() *AuthConfig {
	return &AuthConfig{
		jwtTokenSecret:     c.Jwt.Token.Secret,
		jwtTokenExpiresTTL: c.Jwt.Token.ExpiresTTL,
	}
}