hdkv/i2_bot
Archived
1
Fork 0
Code Releases Activity

Refactoring: now we respect gofmt and have some comments

Browse Source
This commit is contained in:
Vladimir Hodakov
2017-10-18 09:39:50 +04:00
parent 6f374e560e
commit 8dab6c0699
47 changed files with 607 additions and 479 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/talkers/easter.go
View File

@@ -12,6 +12,7 @@ import (
"github.com/go-telegram-bot-api/telegram-bot-api"
)
// DurakMessage is an easter egg
func (t *Talkers) DurakMessage(update tgbotapi.Update) {
log.Printf("[%s] %s", update.Message.From.UserName, update.Message.Text)
@@ -30,6 +31,7 @@ func (t *Talkers) DurakMessage(update tgbotapi.Update) {
c.Bot.Send(msg)
}
// MatMessage is an easter rgg
func (t *Talkers) MatMessage(update tgbotapi.Update) {
log.Printf("[%s] %s", update.Message.From.UserName, update.Message.Text)

14
lib/talkers/errors.go
View File

@@ -8,21 +8,23 @@ import (
"github.com/go-telegram-bot-api/telegram-bot-api"
)
// AnyMessageUnauthorized throws when user can't do something
func (t *Talkers) AnyMessageUnauthorized(update tgbotapi.Update) {
error_message := "Извини, действие для тебя недоступно. Возможно, у меня нет твоего профиля или же твои права недостаточны для совершения данного действия\n\n"
error_message += "Если тебе кажется, что это ошибка, пиши @fat0troll.\n"
message := "Извини, действие для тебя недоступно. Возможно, у меня нет твоего профиля или же твои права недостаточны для совершения данного действия\n\n"
message += "Если тебе кажется, что это ошибка, пиши @fat0troll.\n"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, error_message)
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
}
// GetterError throws when bot can't get something
func (t *Talkers) GetterError(update tgbotapi.Update) {
error_message := "Ой, внутренняя ошибка в боте :(\n\n"
error_message += "Напиши @fat0troll, приложив форвардом последние сообщения до этого.\n"
message := "Ой, внутренняя ошибка в боте :(\n\n"
message += "Напиши @fat0troll, приложив форвардом последние сообщения до этого.\n"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, error_message)
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)

3
lib/talkers/exported.go
View File

@@ -15,14 +15,17 @@ var (
c *appcontext.Context
)
// Talkers is a function-handling struct for talkers
type Talkers struct{}
// New is a appcontext initialization function
func New(ac *appcontext.Context) {
c = ac
m := &Talkers{}
c.RegisterTalkersInterface(talkersinterface.TalkersInterface(m))
}
// Init is an initialization function for talkers
func (t *Talkers) Init() {
log.Printf("Initializing responders...")
}

22
lib/talkers/hello.go
View File

@@ -10,23 +10,25 @@ import (
"../dbmapping"
)
// HelloMessageUnauthorized tell new user what to do.
func (t *Talkers) HelloMessageUnauthorized(update tgbotapi.Update) {
hello_message := "*Бот Инстинкта приветствует тебя!*\n\n"
hello_message += "Для начала работы с ботом, пожалуйста, перешли от бота игры @PokememBroBot профиль героя.\n"
hello_message += "Все дальнейшие действия с ботом возможны лишь при наличии профиля игрока."
message := "*Бот Инстинкта приветствует тебя!*\n\n"
message += "Для начала работы с ботом, пожалуйста, перешли от бота игры @PokememBroBot профиль героя.\n"
message += "Все дальнейшие действия с ботом возможны лишь при наличии профиля игрока."
msg := tgbotapi.NewMessage(update.Message.Chat.ID, hello_message)
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)
}
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")
hello_message += "\nПосмотреть информацию о себе: /me"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, hello_message)
// HelloMessageAuthorized greets existing user
func (t *Talkers) HelloMessageAuthorized(update tgbotapi.Update, playerRaw dbmapping.Player) {
message := "*Бот Инстинкта приветствует тебя. Снова.*\n\n"
message += "Привет, " + update.Message.From.FirstName + " " + update.Message.From.LastName + "!\n"
message += "Последнее обновление информации о тебе: " + playerRaw.UpdatedAt.Format("02.01.2006 15:04:05 -0700")
message += "\nПосмотреть информацию о себе: /me"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)

21
lib/talkers/help.go
View File

@@ -10,18 +10,19 @@ import (
"../config"
)
// HelpMessage gives user all available commands
func (t *Talkers) HelpMessage(update tgbotapi.Update) {
help_message := "*Бот Инстинкта Enchanched.*\n\n"
help_message += "Текущая версия: *" + config.VERSION + "*\n\n"
help_message += "Список команд:\n\n"
help_message += "+ /me – посмотреть свой сохраненный профиль в боте\n"
help_message += "+ /best – посмотреть лучших покемонов для поимки\n"
help_message += "+ /pokedeks – получить список известных боту покемемов\n"
help_message += "+ /help – выводит данное сообщение\n"
help_message += "\n\n"
help_message += "Связаться с автором: @fat0troll\n"
message := "*Бот Инстинкта Enchanched.*\n\n"
message += "Текущая версия: *" + config.VERSION + "*\n\n"
message += "Список команд:\n\n"
message += "+ /me – посмотреть свой сохраненный профиль в боте\n"
message += "+ /best – посмотреть лучших покемонов для поимки\n"
message += "+ /pokedeks – получить список известных боту покемемов\n"
message += "+ /help – выводит данное сообщение\n"
message += "\n\n"
message += "Связаться с автором: @fat0troll\n"
msg := tgbotapi.NewMessage(update.Message.Chat.ID, help_message)
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
msg.ParseMode = "Markdown"
c.Bot.Send(msg)

44
lib/talkers/pokedex.go
View File

@@ -15,12 +15,12 @@ import (
// Internal functions
func (t *Talkers) pokememesListing(update tgbotapi.Update, page int, pokememes_full []dbmapping.PokememeFull) {
func (t *Talkers) pokememesListing(update tgbotapi.Update, page int, pokememesArray []dbmapping.PokememeFull) {
message := "*Известные боту покемемы*\n"
message += "Список отсортирован по грейду и алфавиту.\n"
message += "Покедекс: " + strconv.Itoa(len(pokememes_full)) + " / 206\n"
message += "Покедекс: " + strconv.Itoa(len(pokememesArray)) + " / 206\n"
message += "Отображаем покемемов с " + strconv.Itoa(((page-1)*50)+1) + " по " + strconv.Itoa(page*50) + "\n"
if len(pokememes_full) > page*50 {
if len(pokememesArray) > page*50 {
message += "Переход на следующую страницу: /pokedeks" + strconv.Itoa(page+1)
}
if page > 1 {
@@ -28,18 +28,18 @@ func (t *Talkers) pokememesListing(update tgbotapi.Update, page int, pokememes_f
}
message += "\n\n"
for i := range pokememes_full {
for i := range pokememesArray {
if (i+1 > 50*(page-1)) && (i+1 < (50*page)+1) {
pk := pokememes_full[i].Pokememe
pk_e := pokememes_full[i].Elements
pk := pokememesArray[i].Pokememe
pkE := pokememesArray[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
for j := range pkE {
message += pkE[j].Symbol
}
message += "] " + c.Parsers.ReturnPoints(pk.Price) + "$ /pk" + strconv.Itoa(pk.Id)
message += "] " + c.Parsers.ReturnPoints(pk.Price) + "$ /pk" + strconv.Itoa(pk.ID)
message += "\n"
}
}
@@ -53,24 +53,26 @@ func (t *Talkers) pokememesListing(update tgbotapi.Update, page int, pokememes_f
// External functions
// PokememesList lists all known pokememes
func (t *Talkers) PokememesList(update tgbotapi.Update, page int) {
pokememes_full, ok := c.Getters.GetPokememes()
pokememesArray, ok := c.Getters.GetPokememes()
if !ok {
t.GetterError(update)
} else {
t.pokememesListing(update, page, pokememes_full)
t.pokememesListing(update, page, pokememesArray)
}
}
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)
// PokememeInfo shows information about single pokememe based on internal ID
func (t *Talkers) PokememeInfo(update tgbotapi.Update, playerRaw dbmapping.Player) string {
pokememeNumber := strings.Replace(update.Message.Text, "/pk", "", 1)
var calculatePossibilites = true
profileRaw, ok := c.Getters.GetProfile(playerRaw.ID)
if !ok {
calculate_possibilites = false
calculatePossibilites = false
}
pokememe, ok := c.Getters.GetPokememeByID(pokememe_number)
pokememe, ok := c.Getters.GetPokememeByID(pokememeNumber)
if !ok {
return "fail"
}
@@ -106,11 +108,11 @@ func (t *Talkers) PokememeInfo(update tgbotapi.Update, player_raw dbmapping.Play
}
}
if calculate_possibilites {
if (pk.Grade < profile_raw.Level_id+2) && (pk.Grade > profile_raw.Level_id-3) {
if calculatePossibilites {
if (pk.Grade < profileRaw.LevelID+2) && (pk.Grade > profileRaw.LevelID-3) {
message += "\nВероятность поимки:"
for i := range pokememe.Locations {
percentile, pokeballs := c.Getters.PossibilityRequiredPokeballs(pokememe.Locations[i].Id, pk.Grade, profile_raw.Level_id)
percentile, pokeballs := c.Getters.PossibilityRequiredPokeballs(pokememe.Locations[i].ID, pk.Grade, profileRaw.LevelID)
message += "\n" + pokememe.Locations[i].Name + " – "
message += strconv.FormatFloat(percentile, 'f', 2, 64) + "% или "
message += strconv.Itoa(pokeballs) + "⭕"
@@ -118,7 +120,7 @@ func (t *Talkers) PokememeInfo(update tgbotapi.Update, player_raw dbmapping.Play
}
}
message += "\n" + pk.Image_url
message += "\n" + pk.ImageURL
msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
keyboard := tgbotapi.InlineKeyboardMarkup{}

3
lib/talkers/pokememe_add.go
View File

@@ -8,6 +8,7 @@ import (
"github.com/go-telegram-bot-api/telegram-bot-api"
)
// PokememeAddSuccessMessage shows pokememe adding success message
func (t *Talkers) PokememeAddSuccessMessage(update tgbotapi.Update) {
message := "*Покемем успешно добавлен.*\n\n"
message += "Посмотреть всех известных боту покемемов можно командой /pokedeks"
@@ -18,6 +19,7 @@ func (t *Talkers) PokememeAddSuccessMessage(update tgbotapi.Update) {
c.Bot.Send(msg)
}
// PokememeAddDuplicateMessage shows pokememe add duplication message
func (t *Talkers) PokememeAddDuplicateMessage(update tgbotapi.Update) {
message := "*Мы уже знаем об этом покемеме*\n\n"
message += "Посмотреть всех известных боту покемемов можно командой /pokedeks\n\n"
@@ -29,6 +31,7 @@ func (t *Talkers) PokememeAddDuplicateMessage(update tgbotapi.Update) {
c.Bot.Send(msg)
}
// PokememeAddFailureMessage shows pokememe add error message
func (t *Talkers) PokememeAddFailureMessage(update tgbotapi.Update) {
message := "*Неудачно получилось :(*\n\n"
message += "Случилась жуткая ошибка, и мы не смогли записать покемема в базу. Напиши @fat0troll, он разберется.\n\n"

69
lib/talkers/profile.go
View File

@@ -13,31 +13,32 @@ import (
"../dbmapping"
)
func (t *Talkers) ProfileMessage(update tgbotapi.Update, player_raw dbmapping.Player) string {
profile_raw, ok := c.Getters.GetProfile(player_raw.Id)
// ProfileMessage shows current player's profile
func (t *Talkers) ProfileMessage(update tgbotapi.Update, playerRaw dbmapping.Player) string {
profileRaw, ok := c.Getters.GetProfile(playerRaw.ID)
if !ok {
c.Talkers.AnyMessageUnauthorized(update)
return "fail"
}
league := dbmapping.League{}
err := c.Db.Get(&league, c.Db.Rebind("SELECT * FROM leagues WHERE id=?"), player_raw.League_id)
err := c.Db.Get(&league, c.Db.Rebind("SELECT * FROM leagues WHERE id=?"), playerRaw.LeagueID)
if err != nil {
log.Println(err)
}
level := dbmapping.Level{}
err = c.Db.Get(&level, c.Db.Rebind("SELECT * FROM levels WHERE id=?"), profile_raw.Level_id)
err = c.Db.Get(&level, c.Db.Rebind("SELECT * FROM levels WHERE id=?"), profileRaw.LevelID)
if err != nil {
log.Println(err)
}
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 profileRaw.WeaponID != 0 {
err = c.Db.Get(&weapon, c.Db.Rebind("SELECT * FROM weapons WHERE id=?"), profileRaw.WeaponID)
if err != nil {
log.Println(err)
}
}
p_pk := []dbmapping.ProfilePokememe{}
err = c.Db.Select(&p_pk, c.Db.Rebind("SELECT * FROM profiles_pokememes WHERE profile_id=?"), profile_raw.Id)
profilePokememes := []dbmapping.ProfilePokememe{}
err = c.Db.Select(&profilePokememes, c.Db.Rebind("SELECT * FROM profiles_pokememes WHERE profile_id=?"), profileRaw.ID)
if err != nil {
log.Println(err)
}
@@ -47,53 +48,53 @@ func (t *Talkers) ProfileMessage(update tgbotapi.Update, player_raw dbmapping.Pl
log.Println(err)
}
attack_pm := 0
for i := range p_pk {
attackPokememes := 0
for i := range profilePokememes {
for j := range pokememes {
if p_pk[i].Pokememe_id == pokememes[j].Id {
single_attack := float64(pokememes[j].Attack)
single_attack = single_attack * float64(p_pk[i].Pokememe_lvl)
if p_pk[i].Pokememe_rarity == "rare" {
single_attack = single_attack * 1.15
if profilePokememes[i].PokememeID == pokememes[j].ID {
singleAttack := float64(pokememes[j].Attack)
singleAttack = singleAttack * float64(profilePokememes[i].PokememeLevel)
if profilePokememes[i].PokememeRarity == "rare" {
singleAttack = singleAttack * 1.15
}
attack_pm += int(single_attack)
attackPokememes += int(singleAttack)
}
}
}
message := "*Профиль игрока "
message += profile_raw.Nickname + "* (@" + profile_raw.TelegramNickname + ")\n"
message += profileRaw.Nickname + "* (@" + profileRaw.TelegramNickname + ")\n"
message += "\nЛига: " + league.Symbol + league.Name
message += "\n👤 " + strconv.Itoa(profile_raw.Level_id)
message += " | 🎓 " + strconv.Itoa(profile_raw.Exp) + "/" + strconv.Itoa(level.Max_exp)
message += " | 🥚 " + strconv.Itoa(profile_raw.Egg_exp) + "/" + strconv.Itoa(level.Max_egg)
message += "\n💲" + c.Parsers.ReturnPoints(profile_raw.Wealth)
message += " |💎" + strconv.Itoa(profile_raw.Crystalls)
message += " |⭕" + strconv.Itoa(profile_raw.Pokeballs)
message += "\n⚔Атака: 1 + " + c.Parsers.ReturnPoints(weapon.Power) + " + " + c.Parsers.ReturnPoints(attack_pm) + "\n"
message += "\n👤 " + strconv.Itoa(profileRaw.LevelID)
message += " | 🎓 " + strconv.Itoa(profileRaw.Exp) + "/" + strconv.Itoa(level.MaxExp)
message += " | 🥚 " + strconv.Itoa(profileRaw.EggExp) + "/" + strconv.Itoa(level.MaxEgg)
message += "\n💲" + c.Parsers.ReturnPoints(profileRaw.Wealth)
message += " |💎" + strconv.Itoa(profileRaw.Crystalls)
message += " |⭕" + strconv.Itoa(profileRaw.Pokeballs)
message += "\n⚔Атака: 1 + " + c.Parsers.ReturnPoints(weapon.Power) + " + " + c.Parsers.ReturnPoints(attackPokememes) + "\n"
if profile_raw.Weapon_id != 0 {
if profileRaw.WeaponID != 0 {
message += "\n🔫Оружие: " + weapon.Name + " " + c.Parsers.ReturnPoints(weapon.Power) + "⚔"
}
message += "\n🐱Покемемы:"
for i := range p_pk {
for i := range profilePokememes {
for j := range pokememes {
if p_pk[i].Pokememe_id == pokememes[j].Id {
single_attack := float64(pokememes[j].Attack)
single_attack = single_attack * float64(p_pk[i].Pokememe_lvl)
if p_pk[i].Pokememe_rarity == "rare" {
single_attack = single_attack * 1.15
if profilePokememes[i].PokememeID == pokememes[j].ID {
singleAttack := float64(pokememes[j].Attack)
singleAttack = singleAttack * float64(profilePokememes[i].PokememeLevel)
if profilePokememes[i].PokememeRarity == "rare" {
singleAttack = singleAttack * 1.15
}
message += "\n" + strconv.Itoa(pokememes[j].Grade)
message += "⃣ " + pokememes[j].Name
message += " +" + c.Parsers.ReturnPoints(int(single_attack)) + "⚔"
message += " +" + c.Parsers.ReturnPoints(int(singleAttack)) + "⚔"
}
}
}
message += "\n\n💳" + strconv.Itoa(player_raw.Telegram_id)
message += "\n⏰Последнее обновление профиля: " + profile_raw.Created_at.Format("02.01.2006 15:04:05")
message += "\n\n💳" + strconv.Itoa(playerRaw.TelegramID)
message += "\n⏰Последнее обновление профиля: " + profileRaw.CreatedAt.Format("02.01.2006 15:04:05")
message += "\n\nНе забывай обновляться, это важно для получения актуальной информации.\n\n"
message += "/best – посмотреть лучших покемемов для поимки"

2
lib/talkers/profile_add.go
View File

@@ -8,6 +8,7 @@ import (
"github.com/go-telegram-bot-api/telegram-bot-api"
)
// ProfileAddSuccessMessage shows profile addition success message
func (t *Talkers) ProfileAddSuccessMessage(update tgbotapi.Update) {
message := "*Профиль успешно обновлен.*\n\n"
message += "Функциональность бота держится на актуальности профилей. Обновляйся почаще, и да пребудет с тобой Рандом!\n"
@@ -20,6 +21,7 @@ func (t *Talkers) ProfileAddSuccessMessage(update tgbotapi.Update) {
c.Bot.Send(msg)
}
// ProfileAddFailureMessage shows profile addition failure message
func (t *Talkers) ProfileAddFailureMessage(update tgbotapi.Update) {
message := "*Неудачно получилось :(*\n\n"
message += "Случилась жуткая ошибка, и мы не смогли записать профиль в базу. Напиши @fat0troll, он разберется."

21
lib/talkers/suggestions.go
View File

@@ -13,8 +13,9 @@ import (
"../dbmapping"
)
func (t *Talkers) BestPokememesList(update tgbotapi.Update, player_raw dbmapping.Player) string {
pokememes, ok := c.Getters.GetBestPokememes(player_raw.Id)
// BestPokememesList shows list for catching based on player league and grade
func (t *Talkers) BestPokememesList(update tgbotapi.Update, playerRaw dbmapping.Player) string {
pokememes, ok := c.Getters.GetBestPokememes(playerRaw.ID)
if !ok {
log.Printf("Cannot get pokememes from getter!")
return "fail"
@@ -23,20 +24,20 @@ func (t *Talkers) BestPokememesList(update tgbotapi.Update, player_raw dbmapping
message := "*Лучшие покемемы для ловли*\n\n"
for i := range pokememes {
pk := pokememes[i].Pokememe
pk_l := pokememes[i].Locations
pk_e := pokememes[i].Elements
pkL := pokememes[i].Locations
pkE := pokememes[i].Elements
message += strconv.Itoa(pk.Grade) + "⃣ "
message += pk.Name + " (⚔"
message += c.Parsers.ReturnPoints(pk.Attack)
message += ", 🛡" + c.Parsers.ReturnPoints(pk.Defence) + ")"
for i := range pk_e {
message += pk_e[i].Symbol
for i := range pkE {
message += pkE[i].Symbol
}
message += " /pk" + strconv.Itoa(pk.Id) + "\n"
message += " /pk" + strconv.Itoa(pk.ID) + "\n"
message += "Локации: "
for i := range pk_l {
message += pk_l[i].Symbol + pk_l[i].Name
if i+1 < len(pk_l) {
for i := range pkL {
message += pkL[i].Symbol + pkL[i].Name
if i+1 < len(pkL) {
message += ", "
}
}

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

@@ -10,29 +10,26 @@ import (
"../../dbmapping"
)
// TalkersInterface implements Talkers for importing via appcontex
type TalkersInterface interface {
Init()
// Commands
HelloMessageUnauthorized(update tgbotapi.Update)
HelloMessageAuthorized(update tgbotapi.Update, player_raw dbmapping.Player)
HelloMessageAuthorized(update tgbotapi.Update, playerRaw dbmapping.Player)
HelpMessage(update tgbotapi.Update)
PokememesList(update tgbotapi.Update, page int)
PokememeInfo(update tgbotapi.Update, player_raw dbmapping.Player) string
BestPokememesList(update tgbotapi.Update, player_raw dbmapping.Player) string
PokememeInfo(update tgbotapi.Update, playerRaw dbmapping.Player) string
BestPokememesList(update tgbotapi.Update, playerRaw dbmapping.Player) string
// Returns
PokememeAddSuccessMessage(update tgbotapi.Update)
PokememeAddDuplicateMessage(update tgbotapi.Update)
PokememeAddFailureMessage(update tgbotapi.Update)
ProfileAddSuccessMessage(update tgbotapi.Update)
ProfileAddFailureMessage(update tgbotapi.Update)
ProfileMessage(update tgbotapi.Update, player_raw dbmapping.Player) string
ProfileMessage(update tgbotapi.Update, playerRaw dbmapping.Player) string
// Errors
AnyMessageUnauthorized(update tgbotapi.Update)
GetterError(update tgbotapi.Update)
// Easter eggs
DurakMessage(update tgbotapi.Update)
MatMessage(update tgbotapi.Update)
}