|
|
|
@@ -1,91 +1,75 @@
|
|
|
|
package config
|
|
|
|
package config
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
|
|
|
|
"reflect"
|
|
|
|
|
|
|
|
"fmt"
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
type Config struct {
|
|
|
|
type Config struct {
|
|
|
|
Hostname string `mapstructure:"HOSTNAME"`
|
|
|
|
Hostname string `mapstructure:"HOSTNAME"`
|
|
|
|
Port string `mapstructure:"PORT"`
|
|
|
|
Port string `mapstructure:"PORT"`
|
|
|
|
|
|
|
|
DatabaseUrl string `mapstructure:"POSTGRES_URL"`
|
|
|
|
DatabaseUrl string `mapstructure:"DATABASE_URL"`
|
|
|
|
RedisUrl string `mapstructure:"REDIS_URL"`
|
|
|
|
RedisUrl string `mapstructure:"REDIS_URL"`
|
|
|
|
MinioUrl string `mapstructure:"MINIO_URL"`
|
|
|
|
MinioUrl string `mapstructure:"MINIO_URL"`
|
|
|
|
JwtAlgorithm string `mapstructure:"JWT_ALGORITHM"`
|
|
|
|
|
|
|
|
JwtSecret string `mapstructure:"JWT_SECRET"`
|
|
|
|
JwtAlgorithm string `mapstructure:"JWT_ALGORITHM"`
|
|
|
|
JwtIssuer string `mapstructure:"JWT_ISSUER"`
|
|
|
|
JwtSecret string `mapstructure:"JWT_SECRET"`
|
|
|
|
JwtAudience string `mapstructure:"JWT_AUDIENCE"`
|
|
|
|
JwtIssuer string `mapstructure:"JWT_ISSUER"`
|
|
|
|
JwtExpAccess string `mapstructure:"JWT_EXP_ACCESS"`
|
|
|
|
JwtAudience string `mapstructure:"JWT_AUDIENCE"`
|
|
|
|
JwtExpRefresh string `mapstructure:"JWT_EXP_REFRESH"`
|
|
|
|
JwtExpAccess string `mapstructure:"JWT_EXP_ACCESS"`
|
|
|
|
Environment string `mapstructure:"ENVIRONMENT"`
|
|
|
|
JwtExpRefresh string `mapstructure:"JWT_EXP_REFRESH"`
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Environment string `mapstructure:"ENVIRONMENT"`
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
var config *Config
|
|
|
|
func Load() (*Config, error) {
|
|
|
|
|
|
|
|
|
|
|
|
func LoadConfig() error {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Load .env file
|
|
|
|
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
|
|
|
|
|
|
log.Printf("Error reading config file, proceeding with environment variables. %s", err)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Set default parameters
|
|
|
|
|
|
|
|
viper.SetDefault("HOSTNAME", "localhost")
|
|
|
|
viper.SetDefault("HOSTNAME", "localhost")
|
|
|
|
viper.SetDefault("PORT", "8080")
|
|
|
|
viper.SetDefault("PORT", "8080")
|
|
|
|
viper.SetDefault("DATABASE_URL", "mydb")
|
|
|
|
|
|
|
|
viper.SetDefault("REDIS_URL", "myredis")
|
|
|
|
|
|
|
|
viper.SetDefault("MINIO_URL", "myminio")
|
|
|
|
|
|
|
|
viper.SetDefault("JWT_ALGORITHM", "HS256")
|
|
|
|
viper.SetDefault("JWT_ALGORITHM", "HS256")
|
|
|
|
viper.SetDefault("JWT_SECRET", "default_jwt_secret_please_change") // TODO: remove and randomly generate
|
|
|
|
viper.SetDefault("JWT_SECRET", "default_jwt_secret_please_change")
|
|
|
|
viper.SetDefault("JWT_EXP_ACCESS", "5m")
|
|
|
|
viper.SetDefault("JWT_EXP_ACCESS", "5m")
|
|
|
|
viper.SetDefault("JWT_EXP_REFRESH", "1w")
|
|
|
|
viper.SetDefault("JWT_EXP_REFRESH", "1w")
|
|
|
|
viper.SetDefault("JWT_AUDIENCE", "easywish")
|
|
|
|
viper.SetDefault("JWT_AUDIENCE", "easywish")
|
|
|
|
viper.SetDefault("JWT_ISSUER", "easywish")
|
|
|
|
viper.SetDefault("JWT_ISSUER", "easywish")
|
|
|
|
|
|
|
|
|
|
|
|
viper.SetDefault("ENVIRONMENT", "production")
|
|
|
|
viper.SetDefault("ENVIRONMENT", "production")
|
|
|
|
|
|
|
|
|
|
|
|
// Set the file name and type for Viper
|
|
|
|
viper.AutomaticEnv()
|
|
|
|
viper.SetConfigName(".env") // name of config file (without extension)
|
|
|
|
|
|
|
|
viper.SetConfigType("env") // REQUIRED if the config file does not have the extension
|
|
|
|
|
|
|
|
viper.AddConfigPath(".") // optionally look for config in the working directory
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Unmarshal the configuration into the Config struct
|
|
|
|
required := []string{
|
|
|
|
if err := viper.Unmarshal(&config); err != nil {
|
|
|
|
"POSTGRES_URL",
|
|
|
|
log.Fatalf("Unable to decode into struct, %v", err)
|
|
|
|
"REDIS_URL",
|
|
|
|
|
|
|
|
"MINIO_URL",
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var missing []string
|
|
|
|
// Perform validation
|
|
|
|
for _, key := range required {
|
|
|
|
if err := validateConfig(); err != nil {
|
|
|
|
if !viper.IsSet(key) {
|
|
|
|
return err
|
|
|
|
missing = append(missing, key)
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func validateConfig() error {
|
|
|
|
|
|
|
|
v := reflect.ValueOf(*config)
|
|
|
|
|
|
|
|
t := v.Type()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for i := range v.NumField() {
|
|
|
|
|
|
|
|
field := v.Field(i)
|
|
|
|
|
|
|
|
fieldType := t.Field(i)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Check if the field is a string and is empty
|
|
|
|
|
|
|
|
if field.Kind() == reflect.String && field.String() == "" {
|
|
|
|
|
|
|
|
return fmt.Errorf("Missing required configuration: %s", fieldType.Name)
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(missing) > 0 {
|
|
|
|
|
|
|
|
return nil, fmt.Errorf("missing required environment variables: %s", strings.Join(missing, ", "))
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
var cfg Config
|
|
|
|
|
|
|
|
if err := viper.Unmarshal(&cfg); err != nil {
|
|
|
|
|
|
|
|
return nil, fmt.Errorf("failed to unmarshal config: %w", err)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config = &cfg
|
|
|
|
|
|
|
|
return &cfg, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func GetConfig() *Config {
|
|
|
|
func GetConfig() *Config {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if config == nil {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if _, err := Load(); err != nil {
|
|
|
|
|
|
|
|
panic(err)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return config
|
|
|
|
return config
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var config *Config
|
|
|
|
|