mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-20 05:43:21 +00:00
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
// Package main demonstrates the PreHandler API of the State.
|
|
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/diamondburned/arikawa/v2/gateway"
|
|
"github.com/diamondburned/arikawa/v2/state"
|
|
"github.com/diamondburned/arikawa/v2/utils/handler"
|
|
)
|
|
|
|
// To run, do `BOT_TOKEN="TOKEN HERE" go run .`
|
|
|
|
func main() {
|
|
var token = os.Getenv("BOT_TOKEN")
|
|
if token == "" {
|
|
log.Fatalln("No $BOT_TOKEN given.")
|
|
}
|
|
|
|
s, err := state.New("Bot " + token)
|
|
if err != nil {
|
|
log.Fatalln("Session failed:", err)
|
|
}
|
|
|
|
// Make a pre-handler
|
|
s.PreHandler = handler.New()
|
|
s.PreHandler.Synchronous = true
|
|
s.PreHandler.AddHandler(func(c *gateway.MessageDeleteEvent) {
|
|
// Grab from the state
|
|
m, err := s.Message(c.ChannelID, c.ID)
|
|
if err != nil {
|
|
log.Println("Not found:", c.ID)
|
|
} else {
|
|
log.Println(m.Author.Username, "deleted", m.Content)
|
|
}
|
|
})
|
|
|
|
// Add the needed Gateway intents.
|
|
s.Gateway.AddIntents(gateway.IntentGuildMessages)
|
|
s.Gateway.AddIntents(gateway.IntentDirectMessages)
|
|
|
|
if err := s.Open(); err != nil {
|
|
log.Fatalln("Failed to connect:", err)
|
|
}
|
|
defer s.Close()
|
|
|
|
u, err := s.Me()
|
|
if err != nil {
|
|
log.Fatalln("Failed to get myself:", err)
|
|
}
|
|
|
|
log.Println("Started as", u.Username)
|
|
|
|
// Block forever.
|
|
select {}
|
|
}
|