149 lines
4.4 KiB
Go
149 lines
4.4 KiB
Go
// i2_bot – Instinct PokememBro Bot
|
||
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov
|
||
|
||
package router
|
||
|
||
import (
|
||
// stdlib
|
||
"log"
|
||
"regexp"
|
||
"strings"
|
||
// 3rd party
|
||
"github.com/go-telegram-bot-api/telegram-bot-api"
|
||
)
|
||
|
||
// Router is a function-handling struct for router
|
||
type Router struct{}
|
||
|
||
// RouteRequest decides, what to do with user input
|
||
func (r *Router) RouteRequest(update tgbotapi.Update) string {
|
||
text := update.Message.Text
|
||
|
||
playerRaw, ok := c.Getters.GetOrCreatePlayer(update.Message.From.ID)
|
||
if !ok {
|
||
// Silently fail
|
||
return "fail"
|
||
}
|
||
|
||
chatID := update.Message.Chat.ID
|
||
fromID := int64(update.Message.From.ID)
|
||
log.Println(chatID)
|
||
log.Println(fromID)
|
||
if chatID == fromID {
|
||
log.Println("Private chat")
|
||
} else {
|
||
log.Println("Group")
|
||
}
|
||
|
||
// Regular expressions
|
||
var durakMsg = regexp.MustCompile("(Д|д)(У|у)(Р|р)(А|а|Е|е|О|о)")
|
||
var huMsg = regexp.MustCompile("(Х|х)(У|у)(Й|й|Я|я|Ю|ю|Е|е)")
|
||
var blMsg = regexp.MustCompile("(\\s|^)(Б|б)(Л|л)(Я|я)(Т|т|Д|д)")
|
||
var ebMsg = regexp.MustCompile("(\\s|^|ЗА|За|зА|за)(Е|е|Ё|ё)(Б|б)(\\s|Л|л|А|а|Т|т|У|у|Е|е|Ё|ё|И|и)")
|
||
var piMsg = regexp.MustCompile("(П|п)(И|и)(З|з)(Д|д)")
|
||
|
||
// Commands
|
||
var helpMsg = regexp.MustCompile("/help\\z")
|
||
var helloMsg = regexp.MustCompile("/start\\z")
|
||
var pokedexMsg = regexp.MustCompile("/pokede(x|ks)\\d?\\z")
|
||
var pokememeInfoMsg = regexp.MustCompile("/pk(\\d+)")
|
||
var meMsg = regexp.MustCompile("/me\\z")
|
||
var bestMsg = regexp.MustCompile("/best\\z")
|
||
|
||
// Forwards
|
||
var pokememeMsg = regexp.MustCompile("(Уровень)(.+)(Опыт)(.+)\n(Элементы:)(.+)\n(.+)(💙MP)")
|
||
var profileMsg = regexp.MustCompile(`(Онлайн: )(\d+)\n(Турнир через)(.+)\n\n(.*)\n(Элементы)(.+)\n(.*)\n\n(.+)(Уровень)(.+)\n`)
|
||
|
||
if update.Message.ForwardFrom != nil {
|
||
if update.Message.ForwardFrom.ID != 360402625 {
|
||
log.Printf("Forward from another user or bot. Ignoring")
|
||
} else {
|
||
log.Printf("Forward from PokememBro bot! Processing...")
|
||
if playerRaw.ID != 0 {
|
||
switch {
|
||
case pokememeMsg.MatchString(text):
|
||
log.Printf("Pokememe posted!")
|
||
status := c.Parsers.ParsePokememe(text, playerRaw)
|
||
switch status {
|
||
case "ok":
|
||
c.Talkers.PokememeAddSuccessMessage(update)
|
||
case "dup":
|
||
c.Talkers.PokememeAddDuplicateMessage(update)
|
||
case "fail":
|
||
c.Talkers.PokememeAddFailureMessage(update)
|
||
}
|
||
case profileMsg.MatchString(text):
|
||
log.Printf("Profile posted!")
|
||
status := c.Parsers.ParseProfile(update, playerRaw)
|
||
switch status {
|
||
case "ok":
|
||
c.Talkers.ProfileAddSuccessMessage(update)
|
||
case "fail":
|
||
c.Talkers.ProfileAddFailureMessage(update)
|
||
}
|
||
default:
|
||
log.Printf(text)
|
||
}
|
||
} else {
|
||
c.Talkers.AnyMessageUnauthorized(update)
|
||
}
|
||
}
|
||
} else {
|
||
// Direct messages from user
|
||
switch {
|
||
case helloMsg.MatchString(text):
|
||
if playerRaw.ID != 0 {
|
||
c.Talkers.HelloMessageAuthorized(update, playerRaw)
|
||
} else {
|
||
c.Talkers.HelloMessageUnauthorized(update)
|
||
}
|
||
// Help
|
||
case helpMsg.MatchString(text):
|
||
c.Talkers.HelpMessage(update)
|
||
// Pokememes info
|
||
case pokedexMsg.MatchString(text):
|
||
if strings.HasSuffix(text, "1") {
|
||
c.Talkers.PokememesList(update, 1)
|
||
} else if strings.HasSuffix(text, "2") {
|
||
c.Talkers.PokememesList(update, 2)
|
||
} else if strings.HasSuffix(text, "3") {
|
||
c.Talkers.PokememesList(update, 3)
|
||
} else if strings.HasSuffix(text, "4") {
|
||
c.Talkers.PokememesList(update, 4)
|
||
} else if strings.HasSuffix(text, "5") {
|
||
c.Talkers.PokememesList(update, 5)
|
||
} else {
|
||
c.Talkers.PokememesList(update, 1)
|
||
}
|
||
case pokememeInfoMsg.MatchString(text):
|
||
c.Talkers.PokememeInfo(update, playerRaw)
|
||
// Profile info
|
||
case meMsg.MatchString(text):
|
||
if playerRaw.ID != 0 {
|
||
c.Talkers.ProfileMessage(update, playerRaw)
|
||
} else {
|
||
c.Talkers.AnyMessageUnauthorized(update)
|
||
}
|
||
// Suggestions
|
||
case bestMsg.MatchString(text):
|
||
c.Talkers.BestPokememesList(update, playerRaw)
|
||
// Easter eggs
|
||
case huMsg.MatchString(text):
|
||
c.Talkers.MatMessage(update)
|
||
case blMsg.MatchString(text):
|
||
c.Talkers.MatMessage(update)
|
||
case ebMsg.MatchString(text):
|
||
c.Talkers.MatMessage(update)
|
||
case piMsg.MatchString(text):
|
||
c.Talkers.MatMessage(update)
|
||
case durakMsg.MatchString(text):
|
||
c.Talkers.DurakMessage(update)
|
||
default:
|
||
log.Printf("User posted unknown command.")
|
||
return "fail"
|
||
}
|
||
}
|
||
|
||
return "ok"
|
||
}
|