github.com/diamondburned/arikawa/v2@v2.1.0/_example/undeleter/main.go (about)

     1  // Package main demonstrates the PreHandler API of the State.
     2  package main
     3  
     4  import (
     5  	"log"
     6  	"os"
     7  
     8  	"github.com/diamondburned/arikawa/v2/gateway"
     9  	"github.com/diamondburned/arikawa/v2/state"
    10  	"github.com/diamondburned/arikawa/v2/utils/handler"
    11  )
    12  
    13  // To run, do `BOT_TOKEN="TOKEN HERE" go run .`
    14  
    15  func main() {
    16  	var token = os.Getenv("BOT_TOKEN")
    17  	if token == "" {
    18  		log.Fatalln("No $BOT_TOKEN given.")
    19  	}
    20  
    21  	s, err := state.New("Bot " + token)
    22  	if err != nil {
    23  		log.Fatalln("Session failed:", err)
    24  	}
    25  
    26  	// Make a pre-handler
    27  	s.PreHandler = handler.New()
    28  	s.PreHandler.Synchronous = true
    29  	s.PreHandler.AddHandler(func(c *gateway.MessageDeleteEvent) {
    30  		// Grab from the state
    31  		m, err := s.Message(c.ChannelID, c.ID)
    32  		if err != nil {
    33  			log.Println("Not found:", c.ID)
    34  		} else {
    35  			log.Println(m.Author.Username, "deleted", m.Content)
    36  		}
    37  	})
    38  
    39  	// Add the needed Gateway intents.
    40  	s.Gateway.AddIntents(gateway.IntentGuildMessages)
    41  	s.Gateway.AddIntents(gateway.IntentDirectMessages)
    42  
    43  	if err := s.Open(); err != nil {
    44  		log.Fatalln("Failed to connect:", err)
    45  	}
    46  	defer s.Close()
    47  
    48  	u, err := s.Me()
    49  	if err != nil {
    50  		log.Fatalln("Failed to get myself:", err)
    51  	}
    52  
    53  	log.Println("Started as", u.Username)
    54  
    55  	// Block forever.
    56  	select {}
    57  }