hdkv/i2_bot
Archived
1
Fork 0
Code Releases Activity

Pokememes database saving. Now we're stable

Browse Source
This commit is contained in:
Vladimir Hodakov
2017-10-07 02:23:25 +04:00
parent fe496d696f
commit 2724970e3d
20 changed files with 516 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
lib/router/router.go
View File

@@ -8,11 +8,12 @@ import (
"fmt"
"log"
"regexp"
"strings"
"time"
// 3rd party
"gopkg.in/telegram-bot-api.v4"
// local
"../dbmappings"
"../talkers"
)
type Router struct {}
@@ -29,6 +30,19 @@ func (r *Router) RouteRequest(update tgbotapi.Update) string {
if err != nil {
log.Printf("Message user not found in database.")
log.Printf(err.Error())
// Create "nobody" user
player_raw.Telegram_id = user_id
player_raw.League_id = 0
player_raw.Squad_id = 0
player_raw.Status = "nobody"
player_raw.Created_at = time.Now().UTC()
player_raw.Updated_at = time.Now().UTC()
_, erradd := c.Db.NamedExec("INSERT INTO players VALUES(NULL, :telegram_id, :league_id, :squad_id, :status, :created_at, :updated_at)", &player_raw)
if erradd != nil {
log.Printf(erradd.Error())
return "fail"
}
} else {
log.Printf("Message user found in database.")
}
@@ -41,9 +55,13 @@ func (r *Router) RouteRequest(update tgbotapi.Update) string {
var blMsg = regexp.MustCompile("\\s(Б|б)(Л|л)(Я|я)(Т|т|Д|д)")
var ebMsg = regexp.MustCompile("(Е|е|Ё|ё)(Б|б)(\\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")
// Forwards
var pokememeMsg = regexp.MustCompile("(Уровень)(.+)(Опыт)(.+)\n(Элементы:)(.+)\n(.+)(💙MP)")
if update.Message.ForwardFrom != nil {
@@ -54,9 +72,17 @@ func (r *Router) RouteRequest(update tgbotapi.Update) string {
if pokememeMsg.MatchString(text) {
if player_raw.Id != 0 {
log.Printf("Pokememe posted!")
c.Parsers.ParsePokememe(text, player_raw)
status := c.Parsers.ParsePokememe(text, player_raw)
switch status {
case "ok":
c.Talkers.PokememeAddSuccessMessage(update)
case "dup":
c.Talkers.PokememeAddDuplicateMessage(update)
case "fail":
c.Talkers.PokememeAddFailureMessage(update)
}
} else {
talkers.AnyMessageUnauthorized(c.Bot, update)
c.Talkers.AnyMessageUnauthorized(update)
}
} else {
log.Printf(text)
@@ -67,24 +93,39 @@ func (r *Router) RouteRequest(update tgbotapi.Update) string {
switch {
case helloMsg.MatchString(text):
if player_raw.Id != 0 {
talkers.HelloMessageAuthorized(c.Bot, update, player_raw)
c.Talkers.HelloMessageAuthorized(update, player_raw)
} else {
talkers.HelloMessageUnauthorized(c.Bot, update)
c.Talkers.HelloMessageUnauthorized(update)
}
// Help
case helpMsg.MatchString(text):
talkers.HelpMessage(c.Bot, update)
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)
}
// Easter eggs
case huMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
c.Talkers.MatMessage(update)
case blMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
c.Talkers.MatMessage(update)
case ebMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
c.Talkers.MatMessage(update)
case piMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
c.Talkers.MatMessage(update)
case durakMsg.MatchString(text):
talkers.DurakMessage(c.Bot, update)
c.Talkers.DurakMessage(update)
default:
log.Printf("User posted unknown command.")
return "fail"