Archived
1
This repository has been archived on 2022-11-04. You can view files and clone it, but cannot push or open issues or pull requests.
i2_bot/lib/router/router.go

149 lines
4.4 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

// 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"
}