i2_bot/lib/router/router.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"
	}

	chatRaw, ok := c.Getters.GetOrCreateChat(&update)
	if !ok {
		return "fail"
	}

	log.Printf("Received message from chat ")
	log.Println(chatRaw.TelegramID)

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