87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
var (
|
|
auth Auth
|
|
redditbooru_sources []string
|
|
)
|
|
|
|
func init() {
|
|
// build the authentication struct
|
|
build_auth()
|
|
|
|
// build the redditbooru sources slice
|
|
build_redditbooru_sources()
|
|
}
|
|
|
|
func main() {
|
|
// create Discord session
|
|
dg, err := discordgo.New("Bot " + auth.discord)
|
|
if err != nil {
|
|
fmt.Println("Error creating Discord session, ", err)
|
|
return
|
|
}
|
|
|
|
// register the message_create func as a callback for MessageCreate events
|
|
dg.AddHandler(message_create)
|
|
|
|
// open a websocket connection to Discord and begin listening
|
|
err = dg.Open()
|
|
if err != nil {
|
|
fmt.Println("Error opening connection, ", err)
|
|
return
|
|
}
|
|
|
|
// wait here until term signal
|
|
fmt.Println("Bot is now running. Press Ctrl+C to exit.")
|
|
sc := make(chan os.Signal, 1)
|
|
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
|
|
<-sc
|
|
|
|
// cleanly close
|
|
dg.Close()
|
|
|
|
}
|
|
|
|
func message_create (s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
// ignore messages from the bot
|
|
if m.Author.ID == s.State.User.ID {
|
|
return
|
|
}
|
|
|
|
// make sure we match our bot's message token
|
|
if len(m.Content) > 2 {
|
|
input_token := m.Content[0:2]
|
|
if strings.ToLower(string(input_token)) != "t." {
|
|
return
|
|
}
|
|
}
|
|
|
|
// get our message without our bot's token
|
|
message := strings.ToLower(m.Content[2:])
|
|
|
|
// determine our actions
|
|
if message == "coinflip" || message == "coin" { // flip a coin
|
|
s.ChannelMessageSend(m.ChannelID, "<@" + m.Author.ID + "> flipped a coin, it landed on **heads!**") // totally fair
|
|
} else if len(message) > 0 { // as long as there is a message, try to find a picture
|
|
// get url
|
|
url := <-get_image(message)
|
|
|
|
// make sure we have a url returned
|
|
if len(url) > 0 {
|
|
s.ChannelMessageSend(m.ChannelID, url)
|
|
} else {
|
|
s.ChannelMessageSend(m.ChannelID, "I couldn't find that, sauce?")
|
|
}
|
|
}
|
|
}
|