package main import ( "log" "sync" "github.com/spf13/viper" ) type LdapConfig struct { Url string AdminUser string UserAttr string UserOu string MineUserOu string LdapDc string LdapPass string } type MailConfig struct { Username string Password string SmtpServer string SmtpPort int } type Config struct { Ldap *LdapConfig Mail *MailConfig Secret string TplPath string UserTplPath string Tls bool Key string Cert string Port string MaxID int lock sync.Mutex } func validateConfigEntry(entry string, name string) bool { if entry == "" { log.Printf("Error: %v unset", name) return true } return false } func LoadConfig() (*Config, error) { viper.SetConfigName("guildgate") viper.AddConfigPath("/etc/guildgate") viper.AddConfigPath(".") err := viper.ReadInConfig() if err != nil { log.Fatalf("FATAL: Fatal error reading config file: %v \n", err) } viper.SetConfigType("yaml") c := &Config{} l := &LdapConfig{} m := &MailConfig{} viper.SetDefault("port", "8080") viper.SetDefault("secret", "") viper.SetDefault("Tls", false) //Load configs l.Url = viper.GetString("ldapUrl") l.AdminUser = viper.GetString("adminUser") l.UserAttr = viper.GetString("userAttr") l.UserOu = viper.GetString("userOu") l.MineUserOu = viper.GetString("mineUserOu") l.LdapDc = viper.GetString("ldapDc") l.LdapPass = viper.GetString("ldapPass") c.Secret = viper.GetString("secret") c.Tls = viper.GetBool("tls") c.Port = viper.GetString("port") c.Key = viper.GetString("tls_key") c.Cert = viper.GetString("tls_cert") c.TplPath = viper.GetString("templates_path") c.UserTplPath = viper.GetString("user_templates_path") m.SmtpServer = viper.GetString("SmtpServer") m.Username = viper.GetString("SmtpUsername") m.Password = viper.GetString("SmtpPassword") m.SmtpPort = viper.GetInt("SmtpPort") //Validate configs if validateConfigEntry(l.Url, "ldapUrl") || validateConfigEntry(l.AdminUser, "adminUser") || validateConfigEntry(l.UserOu, "userOu") || validateConfigEntry(l.LdapDc, "ldapDc") || validateConfigEntry(l.UserAttr, "userAttr") { log.Fatalf("FATAL: Error in config file, bailing") } c.Ldap = l c.Mail = m return c, nil }