newsbot-api/internal/services/config.go

package services

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 Configs struct {
	ServerAddress 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),

		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)
	}
}