hdkv/i2_bot
Archived
1
Fork 0
Code Releases Activity

Database connections, refactoring, pokememe parsing in progress

Browse Source
This commit is contained in:
Vladimir Hodakov
2017-10-06 02:56:06 +04:00
parent 4fec8f0fe7
commit fe496d696f
29 changed files with 842 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
lib/router/router.go
View File

@@ -5,11 +5,13 @@ package router
import (
// stdlib
"fmt"
"log"
"regexp"
// 3rd party
"gopkg.in/telegram-bot-api.v4"
// local
"../dbmappings"
"../talkers"
)
@@ -20,6 +22,18 @@ type Router struct {}
// If command doesn't exist, it's "fail"
func (r *Router) RouteRequest(update tgbotapi.Update) string {
text := update.Message.Text
user_id := update.Message.From.ID
player_raw := dbmappings.Players{}
err := c.Db.Get(&player_raw, c.Db.Rebind("SELECT * FROM players WHERE telegram_id=?"), user_id)
if err != nil {
log.Printf("Message user not found in database.")
log.Printf(err.Error())
} else {
log.Printf("Message user found in database.")
}
fmt.Println(player_raw)
// Regular expressions
var durakMsg = regexp.MustCompile("(Д|д)(У|у)(Р|р)(А|а|Е|е|О|о)")
@@ -28,23 +42,53 @@ func (r *Router) RouteRequest(update tgbotapi.Update) string {
var ebMsg = regexp.MustCompile("(Е|е|Ё|ё)(Б|б)(\\s|А|а|Т|т|У|у|Е|е|Ё|ё|И|и)")
var piMsg = regexp.MustCompile("(П|п)(И|и)(З|з)(Д|д)")
var helpMsg = regexp.MustCompile("/help\\z")
var helloMsg = regexp.MustCompile("/start\\z")
switch {
case helpMsg.MatchString(text):
talkers.HelpMessage(c.Bot, update)
case huMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case blMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case ebMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case piMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case durakMsg.MatchString(text):
talkers.DurakMessage(c.Bot, update)
default:
log.Printf("User posted unknown command.")
return "fail"
var pokememeMsg = regexp.MustCompile("(Уровень)(.+)(Опыт)(.+)\n(Элементы:)(.+)\n(.+)(💙MP)")
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 pokememeMsg.MatchString(text) {
if player_raw.Id != 0 {
log.Printf("Pokememe posted!")
c.Parsers.ParsePokememe(text, player_raw)
} else {
talkers.AnyMessageUnauthorized(c.Bot, update)
}
} else {
log.Printf(text)
}
}
} else {
// Direct messages from user
switch {
case helloMsg.MatchString(text):
if player_raw.Id != 0 {
talkers.HelloMessageAuthorized(c.Bot, update, player_raw)
} else {
talkers.HelloMessageUnauthorized(c.Bot, update)
}
// Help
case helpMsg.MatchString(text):
talkers.HelpMessage(c.Bot, update)
// Easter eggs
case huMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case blMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case ebMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case piMsg.MatchString(text):
talkers.MatMessage(c.Bot, update)
case durakMsg.MatchString(text):
talkers.DurakMessage(c.Bot, update)
default:
log.Printf("User posted unknown command.")
return "fail"
}
}
return "ok"