package services import ( "errors" "fmt" "log" "os" "strconv" "github.com/joho/godotenv" ) const ( ServerAddress = "ServerAddress" //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 Configs struct { ServerAddress string JwtSecret string AdminSecret string RedditEnabled bool RedditPullTop bool RedditPullHot bool RedditPullNsfw bool YoutubeEnabled bool YoutubeDebug bool TwitchEnabled bool TwitchClientId string TwitchClientSecret string TwitchMonitorClips bool TwitchMonitorVOD bool FfxivEnabled bool } type ConfigClient struct{} func NewConfig() ConfigClient { c := ConfigClient{} c.RefreshEnv() return c } func GetEnvConfig() Configs { return Configs{ ServerAddress: os.Getenv(ServerAddress), JwtSecret: os.Getenv("JwtSecret"), AdminSecret: os.Getenv("AdminSecret"), RedditEnabled: processBoolConfig(os.Getenv(FEATURE_ENABLE_REDDIT_BACKEND)), RedditPullTop: processBoolConfig(os.Getenv(REDDIT_PULL_TOP)), RedditPullHot: processBoolConfig(os.Getenv(REDDIT_PULL_HOT)), RedditPullNsfw: processBoolConfig(os.Getenv(REDDIT_PULL_NSFW)), YoutubeEnabled: processBoolConfig(os.Getenv(FEATURE_ENABLE_YOUTUBE_BACKEND)), YoutubeDebug: processBoolConfig(os.Getenv(YOUTUBE_DEBUG)), TwitchEnabled: processBoolConfig(os.Getenv(FEATURE_ENABLE_TWITCH_BACKEND)), TwitchClientId: os.Getenv(TWITCH_CLIENT_ID), TwitchClientSecret: os.Getenv(TWITCH_CLIENT_SECRET), TwitchMonitorClips: processBoolConfig(TWITCH_MONITOR_CLIPS), TwitchMonitorVOD: processBoolConfig(os.Getenv(TWITCH_MONITOR_VOD)), FfxivEnabled: processBoolConfig(os.Getenv(FEATURE_ENABLE_FFXIV_BACKEND)), } } // This will parse a string and convert it to a bool. // If it runs into any errors, it will default to false func processBoolConfig(value string) bool { b, err := strconv.ParseBool(value) if err != nil { return false } return b } 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) } }