12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- package main
- import (
- "flag"
- "fmt"
- "github.com/bwmarrin/discordgo"
- )
- var (
- Email string
- Password string
- Token string
- BotID string
- )
- func init() {
- flag.StringVar(&Email, "e", "", "Account Email")
- flag.StringVar(&Password, "p", "", "Account Password")
- flag.StringVar(&Token, "t", "", "Account Token")
- flag.Parse()
- }
- func main() {
- // Create a new Discord session using the provided login information.
- dg, err := discordgo.New(Email, Password, Token)
- if err != nil {
- fmt.Println("error creating Discord session,", err)
- return
- }
- // Get the account information.
- u, err := dg.User("@me")
- if err != nil {
- fmt.Println("error obtaining account details,", err)
- }
- // Store the account ID for later use.
- BotID = u.ID
- // Register messageCreate as a callback for the messageCreate events.
- dg.AddHandler(messageCreate)
- // Open the websocket and begin listening.
- dg.Open()
- fmt.Println("Bot is now running. Press CTRL-C to exit.")
- // Simple way to keep program running until CTRL-C is pressed.
- <-make(chan struct{})
- return
- }
- // This function will be called (due to AddHandler above) every time a new
- // message is created on any channel that the autenticated bot has access to.
- func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
- // Ignore all messages created by the bot itself
- if m.Author.ID == BotID {
- return
- }
- // If the message is "ping" reply with "Pong!"
- if m.Content == "ping" {
- s.ChannelMessageSend(m.ChannelID, "Pong!")
- }
- // If the message is "pong" reply with "Ping!"
- if m.Content == "pong" {
- s.ChannelMessageSend(m.ChannelID, "Ping!")
- }
- }
|