Added welcome messages to discord bot

This commit is contained in:
Eduard Urbach 2018-03-06 17:34:02 +01:00
parent e37fe686aa
commit bc8527b896
4 changed files with 159 additions and 89 deletions

View File

@ -0,0 +1,47 @@
package main
import (
"fmt"
"github.com/bwmarrin/discordgo"
)
// Channels
const welcomeChannel = "420600996268474368"
const enChannel = "134910939140063232"
const jpChannel = "334016470474424322"
// Emoji
const ramFive = ":ramFive:418520880511975426"
const remFive = ":remFive:403671154553782275"
// OnGuildMemberAdd is called every time a user joins the server.
func OnGuildMemberAdd(session *discordgo.Session, event *discordgo.GuildMemberAdd) {
fmt.Println(event.Member.User.Username + " just joined!")
msg, err := session.ChannelMessageSend(welcomeChannel, fmt.Sprintf(
"**Welcome** %s!\n\nPlease introduce yourself in <#%s>.\n\n日本人は <#%s> で自己紹介して下さい!",
event.Member.User.Mention(),
enChannel,
jpChannel,
))
if err != nil {
fmt.Println(err)
return
}
err = session.MessageReactionAdd(welcomeChannel, msg.ID, ramFive)
if err != nil {
fmt.Println(err)
return
}
err = session.MessageReactionAdd(welcomeChannel, msg.ID, remFive)
if err != nil {
fmt.Println(err)
return
}
}

View File

@ -0,0 +1,14 @@
package main
import (
"fmt"
"github.com/bwmarrin/discordgo"
)
// OnGuildMemberRemove is called every time a user leaves the server.
func OnGuildMemberRemove(session *discordgo.Session, event *discordgo.GuildMemberRemove) {
fmt.Println(event.Member.User.Username + " just left!")
// session.ChannelMessageSend(welcomeChannel, "Good bye "+event.Member.User.Mention()+"!")
}

View File

@ -0,0 +1,94 @@
package main
import (
"strings"
"github.com/animenotifier/arn"
"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 == "!commands" {
s.ChannelMessageSend(msg.ChannelID, `
**!user** [username]
**!anime** [id]
**!animelist** [username]
**!tag** [forum tag]`)
}
// 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
}
}
if strings.HasPrefix(msg.Content, "!user ") {
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/+"+strings.Split(msg.Content, " ")[1])
return
}
if strings.HasPrefix(msg.Content, "!anime ") {
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/anime/"+strings.Split(msg.Content, " ")[1])
return
}
if strings.HasPrefix(msg.Content, "!animelist ") {
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/+"+strings.Split(msg.Content, " ")[1]+"/animelist")
return
}
if strings.HasPrefix(msg.Content, "!tag ") {
s.ChannelMessageSend(msg.ChannelID, "https://notify.moe/forum/"+strings.ToLower(strings.Split(msg.Content, " ")[1]))
return
}
if strings.HasPrefix(msg.Content, "!play ") {
s.UpdateStatus(0, msg.Content[len("!play "):])
return
}
if strings.HasPrefix(msg.Content, "!s ") {
term := msg.Content[len("!s "):]
users, animes, posts, threads, tracks, characters := arn.Search(term, 3, 3, 3, 3, 3, 3)
message := ""
for _, user := range users {
message += "https://notify.moe" + user.Link() + "\n"
}
for _, anime := range animes {
message += "https://notify.moe" + anime.Link() + "\n"
}
for _, post := range posts {
message += "https://notify.moe" + post.Link() + "\n"
}
for _, thread := range threads {
message += "https://notify.moe" + thread.Link() + "\n"
}
for _, track := range tracks {
message += "https://notify.moe" + track.Link() + "\n"
}
for _, character := range characters {
message += "https://notify.moe" + character.Link() + "\n"
}
if len(users) == 0 && len(animes) == 0 && len(posts) == 0 && len(threads) == 0 && len(tracks) == 0 {
message = "Sorry, I couldn't find anything using that term."
}
s.ChannelMessageSend(msg.ChannelID, message)
return
}
}

View File

@ -4,7 +4,6 @@ import (
"log"
"os"
"os/signal"
"strings"
"syscall"
"github.com/animenotifier/arn"
@ -42,8 +41,10 @@ func main() {
defer discord.Close()
// Receive messages
discord.AddHandler(onMessage)
// Receive events
discord.AddHandler(OnMessageCreate)
discord.AddHandler(OnGuildMemberAdd)
discord.AddHandler(OnGuildMemberRemove)
// Wait for a CTRL-C
log.Printf("Tsundere is ready. Press CTRL-C to exit.")
@ -51,89 +52,3 @@ func main() {
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
<-sc
}
// This function will be called every time a new message is created on any channel.
func onMessage(s *discordgo.Session, m *discordgo.MessageCreate) {
// Ignore all messages created by the bot itself
if m.Author.ID == s.State.User.ID {
return
}
if m.Content == "!commands" {
s.ChannelMessageSend(m.ChannelID, `
**!user** [username]
**!anime** [id]
**!animelist** [username]
**!tag** [forum tag]`)
}
// Has the bot been mentioned?
for _, user := range m.Mentions {
if user.ID == discord.State.User.ID {
s.ChannelMessageSend(m.ChannelID, m.Author.Mention()+" :heart:")
return
}
}
if strings.HasPrefix(m.Content, "!user ") {
s.ChannelMessageSend(m.ChannelID, "https://notify.moe/+"+strings.Split(m.Content, " ")[1])
return
}
if strings.HasPrefix(m.Content, "!anime ") {
s.ChannelMessageSend(m.ChannelID, "https://notify.moe/anime/"+strings.Split(m.Content, " ")[1])
return
}
if strings.HasPrefix(m.Content, "!animelist ") {
s.ChannelMessageSend(m.ChannelID, "https://notify.moe/+"+strings.Split(m.Content, " ")[1]+"/animelist")
return
}
if strings.HasPrefix(m.Content, "!tag ") {
s.ChannelMessageSend(m.ChannelID, "https://notify.moe/forum/"+strings.ToLower(strings.Split(m.Content, " ")[1]))
return
}
if strings.HasPrefix(m.Content, "!play ") {
s.UpdateStatus(0, strings.Split(m.Content, " ")[1])
return
}
if strings.HasPrefix(m.Content, "!s ") {
term := m.Content[len("!s "):]
users, animes, posts, threads, tracks, characters := arn.Search(term, 3, 3, 3, 3, 3, 3)
message := ""
for _, user := range users {
message += "https://notify.moe" + user.Link() + "\n"
}
for _, anime := range animes {
message += "https://notify.moe" + anime.Link() + "\n"
}
for _, post := range posts {
message += "https://notify.moe" + post.Link() + "\n"
}
for _, thread := range threads {
message += "https://notify.moe" + thread.Link() + "\n"
}
for _, track := range tracks {
message += "https://notify.moe" + track.Link() + "\n"
}
for _, character := range characters {
message += "https://notify.moe" + character.Link() + "\n"
}
if len(users) == 0 && len(animes) == 0 && len(posts) == 0 && len(threads) == 0 && len(tracks) == 0 {
message = "Sorry, I couldn't find anything using that term."
}
s.ChannelMessageSend(m.ChannelID, message)
return
}
}