1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-12-02 20:02:53 +00:00
arikawa/_example/undeleter/main.go

53 lines
1,020 B
Go

package main
import (
"log"
"os"
"github.com/diamondburned/arikawa/gateway"
"github.com/diamondburned/arikawa/handler"
"github.com/diamondburned/arikawa/state"
)
// 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)
}
})
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 {}
}