trim domain

This commit is contained in:
2022-08-09 13:26:53 -04:00
parent d68f027f42
commit efbe429824
2 changed files with 13 additions and 8 deletions

View File

@ -93,11 +93,11 @@ func setupCommands() {
}
commands = append(commands, activityReport)
urlWhitelist := Command{
Name: "Whitelist URL",
Name: "Whitelist URL",
RequiresAdmin: true,
Keywords: []string{"whitelist", "wl"},
Exec: WhitelistURL,
Help: "Add a domain to the HTTP whitelist domains are in the format `thisvid.com` without the subdomain.",
Keywords: []string{"whitelist", "wl"},
Exec: WhitelistURL,
Help: "Add a domain to the HTTP whitelist domains are in the format `thisvid.com` without the subdomain.",
}
commands = append(commands, urlWhitelist)
}
@ -292,9 +292,14 @@ func Status(b BotCommand) bool {
func WhitelistURL(b BotCommand) bool {
defer log.PanicSafe()
config.WhitelistURLs = append(config.WhitelistURLs, strings.ReplaceAll(strings.ReplaceAll(b.Message.Content, b.Command, ""), "<@688025671968096341>", ""))
config.WhitelistURLs = append(config.WhitelistURLs, strings.Trim(
strings.ReplaceAll(
strings.ReplaceAll(b.Message.Content, b.Command, ""),
"<@688025671968096341>", ""),
),
)
domains := strings.Join(config.WhitelistURLs, "\n")
b.Session.ChannelMessageSend(config.AdminChannel, fmt.Sprintf("Current whitelisted domains: %+v", domains))
log.LogDebug(fmt.Sprintf("Current whitelisted domains: %+v", domains))
return true;
}
return true
}

View File

@ -57,7 +57,7 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
if strings.Contains(m.Content, "http") {
safe := false
for _, testURL := range config.WhitelistURLs {
if strings.Contains(m.Content, testURL) {
if strings.Contains(m.Content, testURL) {
safe = true
}
}