From 4464c878e930e059b99dde85957cbfd6152c5749 Mon Sep 17 00:00:00 2001 From: Tyler Date: Thu, 17 Dec 2020 19:04:56 -0500 Subject: [PATCH] Fancy usage of arrays in config files --- main.go | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 546f788..4497cc5 100644 --- a/main.go +++ b/main.go @@ -42,7 +42,14 @@ func main() { log.Fatalln("Invalid token/token is not set") } - channelStrs := strings.Split(viper.GetString("channels"), ",") + var channelStrs []string + + if viper.ConfigFileUsed() != "" { + channelStrs = viper.GetStringSlice("channels") + } else { + channelStrs = strings.Split(viper.GetString("channels"), ",") + } + channels = make([]discord.ChannelID, len(channelStrs)) for i, channelStr := range channelStrs { @@ -55,7 +62,11 @@ func main() { channels[i] = discord.ChannelID(sf) } - prohibitedPhrases = strings.Split(viper.GetString("prohibitedPhrases"), ",") + if viper.ConfigFileUsed() != "" { + prohibitedPhrases = viper.GetStringSlice("phohibitedPhrases") + } else { + prohibitedPhrases = strings.Split(viper.GetString("prohibitedPhrases"), ",") + } var err error