85 lines
2.1 KiB
Go
Raw Normal View History

2018-03-06 16:34:02 +00:00
package main
import (
"math/rand"
2018-03-06 16:34:02 +00:00
"strings"
"github.com/animenotifier/arn"
2018-04-19 20:15:05 +00:00
"github.com/animenotifier/arn/search"
2018-03-06 16:34:02 +00:00
"github.com/bwmarrin/discordgo"
)
// OnMessageCreate is called every time a new message is created on any channel.
func OnMessageCreate(s *discordgo.Session, msg *discordgo.MessageCreate) {
// Ignore all messages created by the bot itself
if msg.Author.ID == s.State.User.ID {
return
}
if msg.Content == "!help" || msg.Content == "!commands" {
2018-03-06 16:34:02 +00:00
s.ChannelMessageSend(msg.ChannelID, `
**!a** [anime search term]
2018-03-06 16:34:02 +00:00
**!animelist** [username]
**!play** [status text]
**!randomquote**
2018-07-09 10:38:44 +00:00
**!source**
**!region** [region]`)
2018-03-06 16:34:02 +00:00
}
// Has the bot been mentioned?
for _, user := range msg.Mentions {
if user.ID == discord.State.User.ID {
s.ChannelMessageSend(msg.ChannelID, msg.Author.Mention()+" :heart:")
return
}
}
// Anime search
if strings.HasPrefix(msg.Content, "!a ") {
term := msg.Content[len("!a "):]
animes := search.Anime(term, 3)
message := ""
for _, anime := range animes {
message += "https://notify.moe" + anime.Link() + "\n"
}
if len(animes) == 0 {
message = "Sorry, I couldn't find anything using that term."
}
s.ChannelMessageSend(msg.ChannelID, message)
2018-03-06 16:34:02 +00:00
return
}
// Anime list of user
2018-03-06 16:34:02 +00:00
if strings.HasPrefix(msg.Content, "!animelist ") {
2018-07-02 07:24:04 +00:00
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/+"+strings.Split(msg.Content, " ")[1]+"/animelist/watching")
2018-03-06 16:34:02 +00:00
return
}
// Play status
2018-03-06 16:34:02 +00:00
if strings.HasPrefix(msg.Content, "!play ") {
s.UpdateStatus(0, msg.Content[len("!play "):])
return
}
// Random quote
if msg.Content == "!randomquote" {
2018-07-02 07:22:37 +00:00
allQuotes := arn.FilterQuotes(func(quote *arn.Quote) bool {
return !quote.IsDraft && quote.IsValid()
})
quote := allQuotes[rand.Intn(len(allQuotes))]
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe"+quote.Link())
2018-03-15 17:01:17 +00:00
return
}
// GitHub source of the bot
if msg.Content == "!source" {
s.ChannelMessageSend(msg.ChannelID, msg.Author.Mention()+" B-baaaaaaaka! Y..you...you want to...TOUCH MY CODE?!\n\nhttps://github.com/animenotifier/notify.moe/tree/go/bots/discord")
2018-04-15 10:13:19 +00:00
return
}
2018-03-06 16:34:02 +00:00
}