155 lines
3.7 KiB
Go
155 lines
3.7 KiB
Go
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)
|
|
}
|
|
}
|