140 lines
3.5 KiB
Go
140 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"math/rand"
|
|
"strings"
|
|
|
|
"github.com/animenotifier/arn"
|
|
|
|
"github.com/animenotifier/arn/search"
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
var regions = map[string]string{
|
|
"africa": "465876853236826112",
|
|
"america": "465876808311635979",
|
|
"asia": "465876834031108096",
|
|
"australia": "465876893036707840",
|
|
"europe": "465876773029019659",
|
|
}
|
|
|
|
// 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" {
|
|
s.ChannelMessageSend(msg.ChannelID, `
|
|
**!a** [anime search term]
|
|
**!animelist** [username]
|
|
**!play** [status text]
|
|
**!randomquote**
|
|
**!source**
|
|
**!region** [region]`)
|
|
}
|
|
|
|
// 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)
|
|
return
|
|
}
|
|
|
|
// Anime list of user
|
|
if strings.HasPrefix(msg.Content, "!animelist ") {
|
|
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/+"+strings.Split(msg.Content, " ")[1]+"/animelist/watching")
|
|
return
|
|
}
|
|
|
|
// Play status
|
|
if strings.HasPrefix(msg.Content, "!play ") {
|
|
s.UpdateStatus(0, msg.Content[len("!play "):])
|
|
return
|
|
}
|
|
|
|
// Random quote
|
|
if msg.Content == "!randomquote" {
|
|
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())
|
|
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")
|
|
return
|
|
}
|
|
|
|
// Set the specific region role for the user
|
|
if strings.HasPrefix(msg.Content, "!region ") {
|
|
region := strings.ToLower(msg.Content[len("!region "):])
|
|
|
|
// check to make sure the region is in the region map
|
|
if _, ok := regions[region]; !ok {
|
|
s.ChannelMessageSend(msg.ChannelID, "This is not a region!")
|
|
return
|
|
}
|
|
|
|
// Get the channel, this is used to get the guild ID
|
|
c, _ := s.Channel(msg.ChannelID)
|
|
|
|
// Check to see if user already has a region role
|
|
user, _ := s.GuildMember(c.GuildID, msg.Author.ID)
|
|
|
|
for _, role := range user.Roles {
|
|
match := false
|
|
|
|
// We also need to loop through our map because discord doesn't return roles as names
|
|
// but rather IDs.
|
|
for _, id := range regions {
|
|
if role == id {
|
|
// Remove the role and set match to true.
|
|
s.GuildMemberRoleRemove(c.GuildID, msg.Author.ID, id)
|
|
match = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if match {
|
|
break
|
|
}
|
|
}
|
|
|
|
// Try to set the role.
|
|
err := s.GuildMemberRoleAdd(c.GuildID, msg.Author.ID, regions[region])
|
|
|
|
if err != nil {
|
|
s.ChannelMessageSend(msg.ChannelID, "The region role could not be set!")
|
|
return
|
|
}
|
|
|
|
s.ChannelMessageSend(msg.ChannelID, "The "+region+" role has been set on your user!")
|
|
}
|
|
return
|
|
}
|