4e021d9140
* feat: auto moderation * feat(examples/automod): add message content intent * style(examples/automod): newline between sections * feat(AutoModerationActionExecution): add user id Add user_id field to AutoModerationActionExecution event. * refactor(events): remove todos Remove TODO comments for AutoModerationRuleUpdate and AutoModerationRuleDelete. * feat(AutoModerationEventMessageSend): doc comment Add documentation comment to AutoModerationEventMessageSend constant.
117 lines
3.9 KiB
Go
117 lines
3.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"sync"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
// Command line flags
|
|
var (
|
|
BotToken = flag.String("token", "", "Bot authorization token")
|
|
GuildID = flag.String("guild", "", "ID of the testing guild")
|
|
ChannelID = flag.String("channel", "", "ID of the testing channel")
|
|
)
|
|
|
|
func init() { flag.Parse() }
|
|
|
|
func main() {
|
|
session, _ := discordgo.New("Bot " + *BotToken)
|
|
session.Identify.Intents |= discordgo.IntentAutoModerationExecution
|
|
session.Identify.Intents |= discordgo.IntentMessageContent
|
|
|
|
enabled := true
|
|
rule, err := session.AutoModerationRuleCreate(*GuildID, &discordgo.AutoModerationRule{
|
|
Name: "Auto Moderation example",
|
|
EventType: discordgo.AutoModerationEventMessageSend,
|
|
TriggerType: discordgo.AutoModerationEventTriggerKeyword,
|
|
TriggerMetadata: &discordgo.AutoModerationTriggerMetadata{
|
|
KeywordFilter: []string{"*cat*"},
|
|
},
|
|
|
|
Enabled: &enabled,
|
|
Actions: []discordgo.AutoModerationAction{
|
|
{Type: discordgo.AutoModerationRuleActionBlockMessage},
|
|
},
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
fmt.Println("Successfully created the rule")
|
|
defer session.AutoModerationRuleDelete(*GuildID, rule.ID)
|
|
|
|
session.AddHandlerOnce(func(s *discordgo.Session, e *discordgo.AutoModerationActionExecution) {
|
|
_, err = session.AutoModerationRuleEdit(*GuildID, rule.ID, &discordgo.AutoModerationRule{
|
|
TriggerMetadata: &discordgo.AutoModerationTriggerMetadata{
|
|
KeywordFilter: []string{"cat"},
|
|
},
|
|
Actions: []discordgo.AutoModerationAction{
|
|
{Type: discordgo.AutoModerationRuleActionTimeout, Metadata: &discordgo.AutoModerationActionMetadata{Duration: 60}},
|
|
{Type: discordgo.AutoModerationRuleActionSendAlertMessage, Metadata: &discordgo.AutoModerationActionMetadata{
|
|
ChannelID: e.ChannelID,
|
|
}},
|
|
},
|
|
})
|
|
if err != nil {
|
|
session.AutoModerationRuleDelete(*GuildID, rule.ID)
|
|
panic(err)
|
|
}
|
|
|
|
s.ChannelMessageSend(e.ChannelID, "Congratulations! You have just triggered an auto moderation rule.\n"+
|
|
"The current trigger can match anywhere in the word, so even if you write the trigger word as a part of another word, it will still match.\n"+
|
|
"The rule has now been changed, now the trigger matches only in the full words.\n"+
|
|
"Additionally, when you send a message, an alert will be sent to this channel and you will be **timed out** for a minute.\n")
|
|
|
|
var counter int
|
|
var counterMutex sync.Mutex
|
|
session.AddHandler(func(s *discordgo.Session, e *discordgo.AutoModerationActionExecution) {
|
|
action := "unknown"
|
|
switch e.Action.Type {
|
|
case discordgo.AutoModerationRuleActionBlockMessage:
|
|
action = "block message"
|
|
case discordgo.AutoModerationRuleActionSendAlertMessage:
|
|
action = "send alert message into <#" + e.Action.Metadata.ChannelID + ">"
|
|
case discordgo.AutoModerationRuleActionTimeout:
|
|
action = "timeout"
|
|
}
|
|
|
|
counterMutex.Lock()
|
|
counter++
|
|
if counter == 1 {
|
|
counterMutex.Unlock()
|
|
s.ChannelMessageSend(e.ChannelID, "Nothing has changed, right? "+
|
|
"Well, since separate gateway events are fired per each action (current is "+action+"), "+
|
|
"you'll see a second message about an action pop up soon")
|
|
} else if counter == 2 {
|
|
counterMutex.Unlock()
|
|
s.ChannelMessageSend(e.ChannelID, "Now the second ("+action+") action got executed.")
|
|
s.ChannelMessageSend(e.ChannelID, "And... you've made it! That's the end of the example.\n"+
|
|
"For more information about the automod and how to use it, "+
|
|
"you can visit the official Discord docs: https://discord.dev/resources/auto-moderation or ask in our server: https://discord.gg/6dzbuDpSWY",
|
|
)
|
|
|
|
session.Close()
|
|
session.AutoModerationRuleDelete(*GuildID, rule.ID)
|
|
os.Exit(0)
|
|
}
|
|
})
|
|
})
|
|
|
|
err = session.Open()
|
|
if err != nil {
|
|
log.Fatalf("Cannot open the session: %v", err)
|
|
}
|
|
defer session.Close()
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt)
|
|
<-stop
|
|
log.Println("Graceful shutdown")
|
|
|
|
}
|