Added support for streaming URL

1 files changed, 8 insertions, 4 deletions

diff --git a/config/config.go b/config/config.go
index e9f08b5..e3fc5e4 100644
--- a/config/config.go
+++ b/config/config.go
@@ -32,6 +32,7 @@ func init() {
 		YoutubeAPIKey:       getEnv("YOUTUBE_API_KEY"),
 		Activity:            types.ActivityType(getIntEnv("ACTIVITY")),
 		ActivityMessage:     getEnv("ACTIVITY_MESSAGE"),
+		ActivityURL:         getEnv("ACTIVITY_URL"),
 	}
 
 	if Config.DiscordToken == "" {
@@ -50,20 +51,23 @@ func init() {
 		logger.Log("Unable to read YouTube API key. environment variable YOUTUBE_API_KEY is required", logOptions)
 	}
 
+	logOptions.Level = types.Warn
+	logOptions.Fatal = false
 	if Config.Activity == 0 {
-		logOptions.Level = types.Warn
-		logOptions.Fatal = false
 		logger.Log("Activity message is empty or not set. Defaulting to PLAYING", logOptions)
 		Config.Activity = types.PLAYING
 	}
 
 	if Config.ActivityMessage == "" {
-		logOptions.Level = types.Warn
-		logOptions.Fatal = false
 		logger.Log("Activity message is empty or not set. Defaulting to empty string", logOptions)
 		Config.ActivityMessage = ""
 	}
 
+	if Config.Activity == types.STREAMING && Config.ActivityURL == "" {
+		logger.Log("Activity URL is empty or not set. Defaulting to empty string", logOptions)
+		Config.ActivityURL = ""
+	}
+
 	logOptions.Level = types.Success
 	logOptions.Fatal = false
 	logger.Log("Config loaded successfully", logOptions)