i2_bot/lib/router/private_request.go

// i2_bot – Instinct PokememBro Bot
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov

package router

import (
	"github.com/go-telegram-bot-api/telegram-bot-api"
	"lab.pztrn.name/fat0troll/i2_bot/lib/dbmapping"
	"regexp"
)

func (r *Router) routePrivateRequest(update *tgbotapi.Update, playerRaw *dbmapping.Player, chatRaw *dbmapping.Chat) string {
	text := update.Message.Text

	// Commands with regexps
	var pokedexMsg = regexp.MustCompile("/pokede(x|ks)\\d?\\z")
	var pokememeInfoMsg = regexp.MustCompile("/pk(\\d+)")
	var usersMsg = regexp.MustCompile("/users\\d?\\z")
	var squadInfoMsg = regexp.MustCompile("/show_squad(\\d+)\\z")
	var orderSendMsg = regexp.MustCompile("/send_order(\\d+)\\z")

	if update.Message.ForwardFrom != nil {
		if update.Message.ForwardFrom.ID != 360402625 {
			c.Log.Info("Forward from another user or bot. Ignoring")
		} else {
			c.Log.Info("Forward from PokememBro bot! Processing...")
			if playerRaw.ID != 0 {
				c.Forwarder.ProcessForward(update, playerRaw)
			} else {
				return c.Talkers.AnyMessageUnauthorized(update)
			}
		}
	} else {
		if update.Message.IsCommand() {
			switch {
			case update.Message.Command() == "start":
				if playerRaw.LeagueID != 0 {
					if playerRaw.Status == "special" {
						c.Welcomer.PrivateWelcomeMessageSpecial(update, playerRaw)
						return "ok"
					}

					c.Welcomer.PrivateWelcomeMessageAuthorized(update, playerRaw)
					return "ok"
				}

				c.Welcomer.PrivateWelcomeMessageUnauthorized(update)
				return "ok"
			case update.Message.Command() == "help":
				c.Talkers.HelpMessage(update, playerRaw)
				return "ok"
			case pokedexMsg.MatchString(text):
				c.Pokedexer.PokememesList(update)
				return "ok"
			case pokememeInfoMsg.MatchString(text):
				c.Pokedexer.PokememeInfo(update, playerRaw)
				return "ok"
			case update.Message.Command() == "me":
				if playerRaw.ID != 0 {
					c.Users.ProfileMessage(update, playerRaw)
					return "ok"
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "best":
				c.Pokedexer.BestPokememesList(update, playerRaw)
				return "ok"
			case update.Message.Command() == "send_all":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					c.Broadcaster.AdminBroadcastMessageCompose(update, playerRaw)
					return "ok"
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "send_league":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					c.Broadcaster.AdminBroadcastMessageCompose(update, playerRaw)
					return "ok"
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "send_confirm":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					c.Broadcaster.AdminBroadcastMessageSend(update, playerRaw)
					return "ok"
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "chats":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					c.Chatter.GroupsList(update)
					return "ok"
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "squads":
				return c.Squader.SquadsList(update, playerRaw)
			case update.Message.Command() == "make_squad":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Squader.CreateSquad(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "pin":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Pinner.PinMessageToSomeChats(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case update.Message.Command() == "pin_all":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Pinner.PinMessageToAllChats(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)

			case update.Message.Command() == "orders":
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Orders.ListAllOrders(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)
			case orderSendMsg.MatchString(text):
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Orders.SendOrder(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)

			case usersMsg.MatchString(text):
				if c.Users.PlayerBetterThan(playerRaw, "admin") {
					return c.Users.UsersList(update)
				}

				return c.Talkers.AnyMessageUnauthorized(update)

			case update.Message.Command() == "squad_add_user":
				return c.Squader.AddUserToSquad(update, playerRaw)
			case update.Message.Command() == "squad_add_commander":
				return c.Squader.AddUserToSquad(update, playerRaw)

			case squadInfoMsg.MatchString(text):
				return c.Squader.SquadInfo(update, playerRaw)
			}
		}
	}

	return "fail"
}