package config import ( "errors" "fmt" "log" "os" "strconv" "github.com/joho/godotenv" ) const ( ServerAddress = "SERVER_ADDRESS" Sql_Connection_String = "SQL_CONNECTION_STRING" FEATURE_ENABLE_REDDIT_BACKEND = "FEATURE_ENABLE_REDDIT_BACKEND" REDDIT_PULL_TOP = "REDDIT_PULL_TOP" REDDIT_PULL_HOT = "REDDIT_PULL_HOT" REDDIT_PULL_NSFW = "REDDIT_PULL_NSFW" FEATURE_ENABLE_YOUTUBE_BACKEND = "FEATURE_ENABLE_YOUTUBE_BACKEND" YOUTUBE_DEBUG = "YOUTUBE_DEBUG" FEATURE_ENABLE_TWITCH_BACKEND = "FEATURE_ENABLE_TWITCH_BACKEND" TWITCH_CLIENT_ID = "TWITCH_CLIENT_ID" TWITCH_CLIENT_SECRET = "TWITCH_CLIENT_SECRET" TWITCH_MONITOR_CLIPS = "TWITCH_MONITOR_CLIPS" TWITCH_MONITOR_VOD = "TWITCH_MONITOR_VOD" FEATURE_ENABLE_FFXIV_BACKEND = "FEATURE_ENABLE_FFXIV_BACKEND" ) type ConfigClient struct{} func New() ConfigClient { c := ConfigClient{} c.RefreshEnv() return c } func (cc *ConfigClient) GetConfig(key string) string { res, filled := os.LookupEnv(key) if !filled { log.Printf("Missing the a value for '%v'. Could generate errors.", key) } return res } // Looks for a value in the env and will panic if it does not exist. func (c ConfigClient) MustGetString(key string) string { res, filled := os.LookupEnv(key) if !filled { msg := fmt.Sprintf("No value was found for '%v'", key) panic(msg) } return res } func (cc *ConfigClient) GetFeature(flag string) (bool, error) { cc.RefreshEnv() res, filled := os.LookupEnv(flag) if !filled { errorMessage := fmt.Sprintf("'%v' was not found", flag) return false, errors.New(errorMessage) } b, err := strconv.ParseBool(res) if err != nil { return false, err } return b, nil } // Use this when your ConfigClient has been opened for awhile and you want to ensure you have the most recent env changes. func (cc *ConfigClient) RefreshEnv() { // Check to see if we have the env file on the system _, err := os.Stat(".env") // We have the file, load it. if err == nil { _, err := os.Open(".env") if err == nil { loadEnvFile() } } } func loadEnvFile() { err := godotenv.Load() if err != nil { log.Fatalln(err) } }