added poketwo handler

2 files changed, 89 insertions, 5 deletions

diff --git a/handlers/messageGatewayHandler.go b/handlers/messageGatewayHandler.go
index a6ec4d7..e688539 100644
--- a/handlers/messageGatewayHandler.go
+++ b/handlers/messageGatewayHandler.go
@@ -1,18 +1,16 @@
 package handlers
 
 import (
-	"strings"
+	messagehandlers "yuzaki/handlers/messageHandlers"
 
 	"github.com/bwmarrin/discordgo"
 )
 
 func MessageGatewayHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
+	messagehandlers.PoketwoHandler(s, m)
+
 	if m.Author.ID == s.State.User.ID || m.Author.Bot {
 		return
 	}
 
-	// if says "hello @bot" <- case insensitive <- bot responds with "Hello @user!"
-	if strings.EqualFold(strings.TrimSpace(m.Content), "hello "+s.State.User.Mention()) {
-		s.ChannelMessageSendReply(m.ChannelID, "Hello "+m.Author.Mention()+"!", m.Reference())
-	}
 }
diff --git a/handlers/messageHandlers/poketwoHandler.go b/handlers/messageHandlers/poketwoHandler.go
new file mode 100644
index 0000000..f124e02
--- /dev/null
+++ b/handlers/messageHandlers/poketwoHandler.go
@@ -0,0 +1,86 @@
+package messagehandlers
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+var (
+	poketwoID       = "716390085896962058"
+	poketwoChannels = []string{"1307335103508254844", "1307335132046168074"}
+)
+
+func PoketwoHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
+	isAllowed := isAllowedChannel(m.ChannelID)
+
+	if m.Author.ID == poketwoID {
+		if !isAllowed {
+			s.ChannelMessageDelete(m.ChannelID, m.ID)
+		}
+		return
+	}
+
+	isCommand := strings.HasPrefix(strings.ToLower(m.Content), "p!") ||
+		(len(m.Mentions) > 0 && m.Mentions[0].ID == poketwoID)
+
+	if isCommand && !isAllowed {
+		channel, err := s.Channel(m.ChannelID)
+		if err != nil {
+			return
+		}
+
+		fmt.Printf("%s#%s sent a Poketwo command in #%s - deleting\n",
+			m.Author.Username, m.Author.Discriminator, channel.Name)
+
+		go func() {
+			time.Sleep(1 * time.Second)
+			s.ChannelMessageDelete(m.ChannelID, m.ID)
+		}()
+
+		var channelMentions string
+		for i, channelID := range poketwoChannels {
+			if i == len(poketwoChannels)-1 && i > 0 {
+				channelMentions += "or "
+			}
+			channelMentions += fmt.Sprintf("<#%s>", channelID)
+			if i < len(poketwoChannels)-2 {
+				channelMentions += ", "
+			}
+		}
+
+		_, err = s.ChannelMessageSend(m.ChannelID,
+			fmt.Sprintf("<@%s>, please use %s for Pokétwo commands. If you need roles, visit <id:customize> to get them.",
+				m.Author.ID, channelMentions))
+
+		if err != nil {
+			return
+		}
+
+		go func() {
+			time.Sleep(1 * time.Second)
+			messages, err := s.ChannelMessages(m.ChannelID, 5, "", "", "")
+			if err != nil {
+				return
+			}
+
+			for _, msg := range messages {
+				if msg.Author.ID == poketwoID {
+					s.ChannelMessageDelete(m.ChannelID, msg.ID)
+					break
+				}
+			}
+		}()
+	}
+}
+
+func isAllowedChannel(channelID string) bool {
+	for _, channel := range poketwoChannels {
+		if channelID == channel {
+			return true
+		}
+	}
+	return false
+}