hdkv/i2_bot
Archived
1
Fork 0
Code Releases Activity

Pokememes possibilities, some refactoring

Browse Source
This commit is contained in:
Vladimir Hodakov
2017-10-13 03:05:26 +04:00
parent 2849d16587
commit ef924d26c7
29 changed files with 438 additions and 293 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/talkers/errors.go
View File

@@ -17,3 +17,13 @@ func (t *Talkers) AnyMessageUnauthorized(update tgbotapi.Update) {
c.Bot.Send(msg)
}
func (t *Talkers) GetterError(update tgbotapi.Update) {
error_message := "Ой, внутренняя ошибка в боте :(\n\n"
error_message += "Напиши @fat0troll, приложив форвардом последние сообщения до этого.\n"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, error_message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
}

4
lib/talkers/hello.go
View File

@@ -7,7 +7,7 @@ import (
// 3rd party
"github.com/go-telegram-bot-api/telegram-bot-api"
// local
"../dbmappings"
"../dbmapping"
)
func (t *Talkers) HelloMessageUnauthorized(update tgbotapi.Update) {
@@ -21,7 +21,7 @@ func (t *Talkers) HelloMessageUnauthorized(update tgbotapi.Update) {
c.Bot.Send(msg)
}
func (t *Talkers) HelloMessageAuthorized(update tgbotapi.Update, player_raw dbmappings.Players) {
func (t *Talkers) HelloMessageAuthorized(update tgbotapi.Update, player_raw dbmapping.Player) {
hello_message := "*Бот Инстинкта приветствует тебя. Снова.*\n\n"
hello_message += "Привет, " + update.Message.From.FirstName + " " + update.Message.From.LastName + "!\n"
hello_message += "Последнее обновление информации о тебе: " + player_raw.Updated_at.Format("02.01.2006 15:04:05 -0700")

2
lib/talkers/help.go
View File

@@ -15,7 +15,7 @@ func (t *Talkers) HelpMessage(update tgbotapi.Update) {
help_message += "Текущая версия: *" + config.VERSION + "*\n\n"
help_message += "Список команд:\n\n"
help_message += "+ /me – посмотреть свой сохраненный профиль в боте\n"
help_message += "+ /pokedex – получить список известных боту покемемов\n"
help_message += "+ /pokedeks – получить список известных боту покемемов\n"
help_message += "+ /help – выводит данное сообщение\n"
help_message += "\n\n"
help_message += "Связаться с автором: @fat0troll\n"

312
lib/talkers/pokedex.go
View File

@@ -5,205 +5,22 @@ package talkers
import (
// stdlib
"log"
"strings"
"strconv"
// 3rd party
"github.com/go-telegram-bot-api/telegram-bot-api"
// local
"../dbmappings"
"../dbmapping"
)
type PokememeFull struct {
Pokememe dbmappings.Pokememes
Elements []dbmappings.Elements
Locations []dbmappings.Locations
}
// Internal functions
func (t *Talkers) PokememeInfo(update tgbotapi.Update, player_raw dbmappings.Players) string {
pokememe_number := strings.Replace(update.Message.Text, "/pk", "", 1)
var calculate_possibilites bool = false
profile_raw := dbmappings.Profiles{}
err := c.Db.Get(&profile_raw, c.Db.Rebind("SELECT * FROM profiles WHERE player_id=? ORDER BY created_at DESC LIMIT 1"), player_raw.Id)
if err != nil {
log.Println(err)
} else {
calculate_possibilites = true
}
// Building pokememe
pk := dbmappings.Pokememes{}
// Checking if pokememe exists in database
err = c.Db.Get(&pk, c.Db.Rebind("SELECT * FROM pokememes WHERE id='" + pokememe_number + "'"))
if err != nil {
log.Println(err)
return "fail"
}
elements := []dbmappings.Elements{}
err = c.Db.Select(&elements, "SELECT * FROM elements");
if err != nil {
log.Println(err)
}
locations := []dbmappings.Locations{}
err = c.Db.Select(&locations, "SELECT * FROM locations");
if err != nil {
log.Println(err)
}
pokememes_elements := []dbmappings.PokememesElements{}
err = c.Db.Select(&pokememes_elements, "SELECT * FROM pokememes_elements WHERE pokememe_id='" + pokememe_number + "'");
if err != nil {
log.Println(err)
}
pokememes_locations := []dbmappings.PokememesLocations{}
err = c.Db.Select(&pokememes_locations, "SELECT * FROM pokememes_locations WHERE pokememe_id='" + pokememe_number + "'");
if err != nil {
log.Println(err)
}
message := strconv.Itoa(pk.Grade) + "⃣ *" + pk.Name + "*\n"
message += pk.Description + "\n\n"
message += "Элементы:"
for i := range(pokememes_elements) {
for j := range(elements) {
if pokememes_elements[i].Element_id == elements[j].Id {
message += " " + elements[j].Symbol
}
}
}
message += "\n⚔ Атака: *" + c.Parsers.ReturnPoints(pk.Attack)
message += "*\n❤️ HP: *" + c.Parsers.ReturnPoints(pk.HP)
message += "*\n💙 MP: *" + c.Parsers.ReturnPoints(pk.MP)
if (pk.Defence != pk.Attack) {
message += "*\n🛡Защита: *" + c.Parsers.ReturnPoints(pk.Defence) + "* _(сопротивляемость покемема к поимке)_"
} else {
message += "*"
}
message += "\nСтоимость: *" + c.Parsers.ReturnPoints(pk.Price)
message += "*\nКупить: *"
if pk.Purchaseable {
message += "Можно"
} else {
message += "Нельзя"
}
message += "*\nОбитает:"
for i := range(pokememes_locations) {
for j := range(locations) {
if pokememes_locations[i].Location_id == locations[j].Id {
message += " *" + locations[j].Name + "*"
if (i + 1) < len(pokememes_locations) {
message += ","
}
}
}
}
if calculate_possibilites {
message += "\nВероятность поимки:"
for i := range(pokememes_locations) {
for j := range(locations) {
if pokememes_locations[i].Location_id == locations[j].Id {
message += "\n" + locations[j].Name + " – " + strconv.Itoa(c.Getters.PossibilityRequiredPokeballs(locations[j].Id, pk.Grade, profile_raw.Level_id)) + "⭕"
}
}
}
}
message += "\n" + pk.Image_url
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
keyboard := tgbotapi.InlineKeyboardMarkup{}
for i := range(pokememes_locations) {
for j := range(locations) {
if pokememes_locations[i].Location_id == locations[j].Id {
var row []tgbotapi.InlineKeyboardButton
btn := tgbotapi.NewInlineKeyboardButtonSwitch(locations[j].Symbol + locations[j].Name, locations[j].Symbol + locations[j].Name)
row = append(row, btn)
keyboard.InlineKeyboard = append(keyboard.InlineKeyboard, row)
}
}
}
msg.ReplyMarkup = keyboard
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
return "ok"
}
func (t *Talkers) PokememesList(update tgbotapi.Update, page int) {
pokememes := []dbmappings.Pokememes{}
err := c.Db.Select(&pokememes, "SELECT * FROM pokememes");
if err != nil {
log.Println(err)
}
pokememes_limited := []dbmappings.Pokememes{}
err = c.Db.Select(&pokememes_limited, "SELECT * FROM pokememes ORDER BY grade asc, name asc LIMIT 50 OFFSET " + strconv.Itoa(50*(page-1)));
if err != nil {
log.Println(err)
}
elements := []dbmappings.Elements{}
err = c.Db.Select(&elements, "SELECT * FROM elements");
if err != nil {
log.Println(err)
}
locations := []dbmappings.Locations{}
err = c.Db.Select(&locations, "SELECT * FROM locations");
if err != nil {
log.Println(err)
}
pokememes_elements := []dbmappings.PokememesElements{}
err = c.Db.Select(&pokememes_elements, "SELECT * FROM pokememes_elements");
if err != nil {
log.Println(err)
}
pokememes_locations := []dbmappings.PokememesLocations{}
err = c.Db.Select(&pokememes_locations, "SELECT * FROM pokememes_locations");
if err != nil {
log.Println(err)
}
pokememes_full := []PokememeFull{}
for i := range(pokememes_limited) {
full_pokememe := PokememeFull{}
elements_listed := []dbmappings.Elements{}
locations_listed := []dbmappings.Locations{}
for j := range(pokememes_locations) {
if pokememes_locations[j].Pokememe_id == pokememes_limited[i].Id {
for l := range(locations) {
if pokememes_locations[j].Location_id == locations[l].Id {
locations_listed = append(locations_listed, locations[l])
}
}
}
}
for k := range(pokememes_elements) {
if pokememes_elements[k].Pokememe_id == pokememes_limited[i].Id {
for e := range(elements) {
if pokememes_elements[k].Element_id == elements[e].Id {
elements_listed = append(elements_listed, elements[e])
}
}
}
}
full_pokememe.Pokememe = pokememes_limited[i]
full_pokememe.Elements = elements_listed
full_pokememe.Locations = locations_listed
pokememes_full = append(pokememes_full, full_pokememe)
}
message := "*Известные боту покемемы*\n"
message += "Список отсортирован по грейду и алфавиту.\n"
message += "Покедекс: " + strconv.Itoa(len(pokememes)) + " / 206\n"
func (t *Talkers) pokememesListing(update tgbotapi.Update, page int, pokememes_full []dbmapping.PokememeFull) {
message := "*Известные боту покемемы*\n"
message += "Список отсортирован по грейду и алфавиту.\n"
message += "Покедекс: " + strconv.Itoa(len(pokememes_full)) + " / 206\n"
message += "Отображаем покемемов с " + strconv.Itoa(((page - 1)*50)+1) + " по " + strconv.Itoa(page*50) + "\n"
if len(pokememes) > page*50 {
if len(pokememes_full) > page*50 {
message += "Переход на следующую страницу: /pokedeks" + strconv.Itoa(page + 1)
}
if page > 1 {
@@ -212,21 +29,110 @@ func (t *Talkers) PokememesList(update tgbotapi.Update, page int) {
message += "\n\n"
for i := range(pokememes_full) {
pk := pokememes_full[i].Pokememe
pk_e := pokememes_full[i].Elements
message += strconv.Itoa(i + 1 + (50*(page-1))) + ". " + strconv.Itoa(pk.Grade)
message += "⃣ *" + pk.Name
message += "* (" + c.Parsers.ReturnPoints(pk.HP) + "-" + c.Parsers.ReturnPoints(pk.MP) + ") ⚔️ *"
message += c.Parsers.ReturnPoints(pk.Attack) + "* \\["
for j := range(pk_e) {
message += pk_e[j].Symbol
if (i + 1 > 50*(page - 1)) && (i + 1 < (50*page) + 1) {
pk := pokememes_full[i].Pokememe
pk_e := pokememes_full[i].Elements
message += strconv.Itoa(i + 1) + ". " + strconv.Itoa(pk.Grade)
message += "⃣ *" + pk.Name
message += "* (" + c.Parsers.ReturnPoints(pk.HP) + "-" + c.Parsers.ReturnPoints(pk.MP) + ") ⚔️ *"
message += c.Parsers.ReturnPoints(pk.Attack) + "* \\["
for j := range(pk_e) {
message += pk_e[j].Symbol
}
message += "] " + c.Parsers.ReturnPoints(pk.Price) + "$ /pk" + strconv.Itoa(pk.Id)
message += "\n"
}
message += "] " + c.Parsers.ReturnPoints(pk.Price) + "$ /pk" + strconv.Itoa(pk.Id)
message += "\n"
}
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
c.Bot.Send(msg)
}
// External functions
func (t *Talkers) PokememesList(update tgbotapi.Update, page int) {
pokememes_full, ok := c.Getters.GetPokememes()
if !ok {
t.GetterError(update)
} else {
t.pokememesListing(update, page, pokememes_full)
}
}
func (t *Talkers) PokememeInfo(update tgbotapi.Update, player_raw dbmapping.Player) string {
pokememe_number := strings.Replace(update.Message.Text, "/pk", "", 1)
var calculate_possibilites bool = true
profile_raw, ok := c.Getters.GetProfile(player_raw.Id)
if !ok {
calculate_possibilites = false
}
pokememe, ok := c.Getters.GetPokememeByID(pokememe_number)
if !ok {
return "fail"
}
pk := pokememe.Pokememe
message := strconv.Itoa(pk.Grade) + "⃣ *" + pk.Name + "*\n"
message += pk.Description + "\n\n"
message += "Элементы:"
for i := range(pokememe.Elements) {
message += " " + pokememe.Elements[i].Symbol
}
message += "\n⚔ Атака: *" + c.Parsers.ReturnPoints(pk.Attack)
message += "*\n❤️ HP: *" + c.Parsers.ReturnPoints(pk.HP)
message += "*\n💙 MP: *" + c.Parsers.ReturnPoints(pk.MP)
if (pk.Defence != pk.Attack) {
message += "*\n🛡Защита: *" + c.Parsers.ReturnPoints(pk.Defence) + "* _(сопротивляемость покемема к поимке)_"
} else {
message += "*"
}
message += "\nСтоимость: *" + c.Parsers.ReturnPoints(pk.Price)
message += "*\nКупить: *"
if pk.Purchaseable {
message += "Можно"
} else {
message += "Нельзя"
}
message += "*\nОбитает:"
for i := range(pokememe.Locations) {
message += " *" + pokememe.Locations[i].Name + "*"
if (i + 1) < len(pokememe.Locations) {
message += ","
}
}
if calculate_possibilites {
if (pk.Grade < profile_raw.Level_id + 2) || (pk.Grade > profile_raw.Level_id - 3) {
message += "\nВероятность поимки:"
for i := range(pokememe.Locations) {
percentile, pokeballs := c.Getters.PossibilityRequiredPokeballs(pokememe.Locations[i].Id, pk.Grade, profile_raw.Level_id)
message += "\n" + pokememe.Locations[i].Name + " – "
message += strconv.FormatFloat(percentile, 'f', 2, 64) + "% или "
message += strconv.Itoa(pokeballs) + "⭕"
}
}
}
message += "\n" + pk.Image_url
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
keyboard := tgbotapi.InlineKeyboardMarkup{}
for i := range(pokememe.Locations) {
var row []tgbotapi.InlineKeyboardButton
btn := tgbotapi.NewInlineKeyboardButtonSwitch(pokememe.Locations[i].Symbol + pokememe.Locations[i].Name, pokememe.Locations[i].Symbol + pokememe.Locations[i].Name)
row = append(row, btn)
keyboard.InlineKeyboard = append(keyboard.InlineKeyboard, row)
}
msg.ReplyMarkup = keyboard
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
return "ok"
}

22
lib/talkers/profile.go
View File

@@ -10,40 +10,38 @@ import (
// 3rd party
"github.com/go-telegram-bot-api/telegram-bot-api"
// local
"../dbmappings"
"../dbmapping"
)
func (t *Talkers) ProfileMessage(update tgbotapi.Update, player_raw dbmappings.Players) string {
profile_raw := dbmappings.Profiles{}
err := c.Db.Get(&profile_raw, c.Db.Rebind("SELECT * FROM profiles WHERE player_id=? ORDER BY created_at DESC LIMIT 1"), player_raw.Id)
if err != nil {
log.Println(err)
func (t *Talkers) ProfileMessage(update tgbotapi.Update, player_raw dbmapping.Player) string {
profile_raw, ok := c.Getters.GetProfile(player_raw.Id)
if !ok {
c.Talkers.AnyMessageUnauthorized(update)
return "fail"
}
league := dbmappings.Leagues{}
err = c.Db.Get(&league, c.Db.Rebind("SELECT * FROM leagues WHERE id=?"), player_raw.League_id)
league := dbmapping.League{}
err := c.Db.Get(&league, c.Db.Rebind("SELECT * FROM leagues WHERE id=?"), player_raw.League_id)
if err != nil {
log.Println(err)
}
level := dbmappings.Levels{}
level := dbmapping.Level{}
err = c.Db.Get(&level, c.Db.Rebind("SELECT * FROM levels WHERE id=?"), profile_raw.Level_id)
if err != nil {
log.Println(err)
}
weapon := dbmappings.Weapons{}
weapon := dbmapping.Weapon{}
if profile_raw.Weapon_id != 0 {
err = c.Db.Get(&weapon, c.Db.Rebind("SELECT * FROM weapons WHERE id=?"), profile_raw.Weapon_id)
if err != nil {
log.Println(err)
}
}
p_pk := []dbmappings.ProfilesPokememes{}
p_pk := []dbmapping.ProfilePokememe{}
err = c.Db.Select(&p_pk, c.Db.Rebind("SELECT * FROM profiles_pokememes WHERE profile_id=?"), profile_raw.Id)
if err != nil {
log.Println(err)
}
pokememes := []dbmappings.Pokememes{}
pokememes := []dbmapping.Pokememe{}
err = c.Db.Select(&pokememes, c.Db.Rebind("SELECT * FROM pokememes"))
if err != nil {
log.Println(err)

9
lib/talkers/talkersinterface/talkersinterface.go
View File

@@ -7,17 +7,17 @@ import (
// 3rd party
"github.com/go-telegram-bot-api/telegram-bot-api"
// local
"../../dbmappings"
"../../dbmapping"
)
type TalkersInterface interface {
Init()
// Commands
HelloMessageUnauthorized(update tgbotapi.Update)
HelloMessageAuthorized(update tgbotapi.Update, player_raw dbmappings.Players)
HelloMessageAuthorized(update tgbotapi.Update, player_raw dbmapping.Player)
HelpMessage(update tgbotapi.Update)
PokememesList(update tgbotapi.Update, page int)
PokememeInfo(update tgbotapi.Update, player_raw dbmappings.Players) string
PokememeInfo(update tgbotapi.Update, player_raw dbmapping.Player) string
// Returns
PokememeAddSuccessMessage(update tgbotapi.Update)
@@ -25,10 +25,11 @@ type TalkersInterface interface {
PokememeAddFailureMessage(update tgbotapi.Update)
ProfileAddSuccessMessage(update tgbotapi.Update)
ProfileAddFailureMessage(update tgbotapi.Update)
ProfileMessage(update tgbotapi.Update, player_raw dbmappings.Players) string
ProfileMessage(update tgbotapi.Update, player_raw dbmapping.Player) string
// Errors
AnyMessageUnauthorized(update tgbotapi.Update)
GetterError(update tgbotapi.Update)
// Easter eggs
DurakMessage(update tgbotapi.Update)