233 lines
7.6 KiB
Go
233 lines
7.6 KiB
Go
// i2_bot – Instinct PokememBro Bot
|
||
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov
|
||
|
||
package talkers
|
||
|
||
import (
|
||
// stdlib
|
||
"log"
|
||
"strings"
|
||
"strconv"
|
||
// 3rd party
|
||
"github.com/go-telegram-bot-api/telegram-bot-api"
|
||
// local
|
||
"../dbmappings"
|
||
)
|
||
|
||
type PokememeFull struct {
|
||
Pokememe dbmappings.Pokememes
|
||
Elements []dbmappings.Elements
|
||
Locations []dbmappings.Locations
|
||
}
|
||
|
||
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"
|
||
message += "Отображаем покемемов с " + strconv.Itoa(((page - 1)*50)+1) + " по " + strconv.Itoa(page*50) + "\n"
|
||
if len(pokememes) > page*50 {
|
||
message += "Переход на следующую страницу: /pokedeks" + strconv.Itoa(page + 1)
|
||
}
|
||
if page > 1 {
|
||
message += "\nПереход на предыдущую страницу: /pokedeks" + strconv.Itoa(page - 1)
|
||
}
|
||
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
|
||
}
|
||
message += "] " + c.Parsers.ReturnPoints(pk.Price) + "$ /pk" + strconv.Itoa(pk.Id)
|
||
message += "\n"
|
||
}
|
||
|
||
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
|
||
msg.ParseMode = "Markdown"
|
||
|
||
c.Bot.Send(msg)
|
||
}
|