1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- package main
- import (
- "flag"
- "fmt"
- "github.com/bwmarrin/discordgo"
- )
- // Variables used for command line parameters
- var (
- Token string
- BotID string
- )
- func init() {
- flag.StringVar(&Token, "t", "", "Bot Token")
- flag.Parse()
- }
- func main() {
- // Create a new Discord session using the provided bot token.
- dg, err := discordgo.New("Bot " + 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.
- err = dg.Open()
- if err != nil {
- fmt.Println("error opening connection,", err)
- return
- }
- 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!")
- }
- }
|