hdkv
/
i2_bot
Archived
1
Fork 0

Pass game admins to our chats. Easter egg chance generation.

master
Vladimir Hodakov 2017-11-28 20:37:14 +04:00
parent 200a7d14ea
commit 9d21ebd44b
2 changed files with 76 additions and 26 deletions

View File

@ -6,6 +6,7 @@ package router
import (
"github.com/go-telegram-bot-api/telegram-bot-api"
"lab.pztrn.name/fat0troll/i2_bot/lib/dbmapping"
"math/rand"
"regexp"
)
@ -14,7 +15,7 @@ func (r *Router) routeGroupRequest(update *tgbotapi.Update, playerRaw *dbmapping
// Regular expressions
var durakMsg = regexp.MustCompile("(Д|д)(У|у)(Р|р)(А|а|Е|е|О|о)")
var huMsg = regexp.MustCompile("(Х|х)(У|у)(Й|й|Я|я|Ю|ю|Е|е)")
var blMsg = regexp.MustCompile("(\\s|^)(Б|б)(Л|л)(Я|я)(Т|т|Д|д)")
var blMsg = regexp.MustCompile("(\\s|^)(Б|б)(Л|л)((Я|я)(Т|т|Д|д)|)")
var ebMsg = regexp.MustCompile("(\\s|^|ЗА|За|зА|за)(Е|е|Ё|ё)(Б|б)(\\s|Л|л|А|а|Т|т|У|у|Е|е|Ё|ё|И|и)")
var piMsg = regexp.MustCompile("(П|п)(И|и)(З|з)(Д|д)")
@ -50,17 +51,21 @@ func (r *Router) routeGroupRequest(update *tgbotapi.Update, playerRaw *dbmapping
return "fail"
}
switch {
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)
// easter eggs
trigger := rand.Intn(5)
if trigger == 4 {
switch {
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)
}
}
return "ok"

View File

@ -465,8 +465,17 @@ func (s *Squader) ProcessMessage(update *tgbotapi.Update, chatRaw *dbmapping.Cha
}
if !isChatValid {
s.kickUserFromSquadChat(&newUsers[i], chatRaw)
messageProcessed = true
switch newUsers[i].UserName {
case "gantz_yaka":
messageProcessed = true
case "@agentpb":
messageProcessed = true
case "@pbhelp":
messageProcessed = true
default:
s.kickUserFromSquadChat(&newUsers[i], chatRaw)
messageProcessed = true
}
}
}
}
@ -514,30 +523,66 @@ func (s *Squader) ProtectBastion(update *tgbotapi.Update, newUser *tgbotapi.User
playerRaw, ok := c.Users.GetOrCreatePlayer(newUser.ID)
if !ok {
s.kickUserFromSquadChat(newUser, &chatRaw)
return "fail"
switch newUser.UserName {
case "gantz_yaka":
// do nothing
case "@agentpb":
// do nothing
case "@pbhelp":
// do nothing
default:
s.kickUserFromSquadChat(newUser, &chatRaw)
return "fail"
}
}
if playerRaw.LeagueID != 1 {
// Check for profile
_, profileOK := c.Users.GetProfile(playerRaw.ID)
if !profileOK {
message := "Привет, " + c.Users.FormatUsername(userName) + "! Напиши мне и скинь профиль для доступа в чаты Лиги!"
switch newUser.UserName {
case "gantz_yaka":
message := "Здравствуй, " + newUser.UserName + "!\n"
message += "Инстинкт рад приветствовать Бога мира ПокемемБро! Проходите, располагайтесь, чувствуйте себя, как дома.\n"
msg := tgbotapi.NewMessage(defaultChatID, message)
msg := tgbotapi.NewMessage(chatRaw.TelegramID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
} else {
message := "Привет, " + c.Users.FormatUsername(userName) + "! Там переход между лигами не завезли случайно? Переходи в нашу Лигу, будем рады тебя видеть... а пока — вход в наши чаты закрыт!"
case "@agentpb":
message := "Здравствуй, " + newUser.UserName + "!\n"
message += "Инстинкт рад приветствовать одного из богов мира ПокемемБро! Проходите, располагайтесь, чувствуйте себя, как дома.\n"
msg := tgbotapi.NewMessage(defaultChatID, message)
msg := tgbotapi.NewMessage(chatRaw.TelegramID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
case "@pbhelp":
message := "Здравствуй, " + newUser.UserName + "!\n"
message += "Инстинкт рад приветствовать одного из богов мира ПокемемБро! Проходите, располагайтесь, чувствуйте себя, как дома.\n"
msg := tgbotapi.NewMessage(chatRaw.TelegramID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
default:
// Check for profile
_, profileOK := c.Users.GetProfile(playerRaw.ID)
if !profileOK {
message := "Привет, " + c.Users.FormatUsername(userName) + "! Напиши мне и скинь профиль для доступа в чаты Лиги!"
msg := tgbotapi.NewMessage(defaultChatID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
} else {
message := "Привет, " + c.Users.FormatUsername(userName) + "! Там переход между лигами не завезли случайно? Переходи в нашу Лигу, будем рады тебя видеть... а пока — вход в наши чаты закрыт!"
msg := tgbotapi.NewMessage(defaultChatID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
}
s.kickUserFromSquadChat(newUser, &chatRaw)
return "fail"
}
s.kickUserFromSquadChat(newUser, &chatRaw)
return "fail"
}
return "ok"