Added welcome messages to discord bot

bots/discord/OnGuildMemberAdd.go

@@ -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
+	}
+}

bots/discord/OnGuildMemberRemove.go

@@ -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()+"!")
+}

bots/discord/OnMessageCreate.go

@@ -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
+	}
+}

bots/discord/discord.go

@@ -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
+	// Receive events
-	discord.AddHandler(onMessage)
+	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
-	}
-}