Error handling. Vendoring
Signed-off-by: Ettore Dreucci <ettore.dreucci@gmail.com>
This commit is contained in:
209
vendor/gopkg.in/tucnak/telebot.v2/util.go
generated
vendored
Normal file
209
vendor/gopkg.in/tucnak/telebot.v2/util.go
generated
vendored
Normal file
@@ -0,0 +1,209 @@
|
||||
package telebot
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
func (b *Bot) debug(err error) {
|
||||
if b.reporter != nil {
|
||||
b.reporter(errors.WithStack(err))
|
||||
}
|
||||
}
|
||||
|
||||
func (b *Bot) deferDebug() {
|
||||
if r := recover(); r != nil {
|
||||
if err, ok := r.(error); ok {
|
||||
b.debug(err)
|
||||
} else if str, ok := r.(string); ok {
|
||||
b.debug(errors.Errorf("%s", str))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (b *Bot) sendText(to Recipient, text string, opt *SendOptions) (*Message, error) {
|
||||
params := map[string]string{
|
||||
"chat_id": to.Recipient(),
|
||||
"text": text,
|
||||
}
|
||||
embedSendOptions(params, opt)
|
||||
|
||||
respJSON, err := b.Raw("sendMessage", params)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return extractMsgResponse(respJSON)
|
||||
}
|
||||
|
||||
func wrapSystem(err error) error {
|
||||
return errors.Wrap(err, "system error")
|
||||
}
|
||||
|
||||
func isUserInList(user *User, list []User) bool {
|
||||
for _, user2 := range list {
|
||||
if user.ID == user2.ID {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func extractMsgResponse(respJSON []byte) (*Message, error) {
|
||||
var resp struct {
|
||||
Ok bool
|
||||
Result *Message
|
||||
Description string
|
||||
}
|
||||
|
||||
err := json.Unmarshal(respJSON, &resp)
|
||||
if err != nil {
|
||||
var resp struct {
|
||||
Ok bool
|
||||
Result bool
|
||||
Description string
|
||||
}
|
||||
|
||||
err := json.Unmarshal(respJSON, &resp)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "bad response json")
|
||||
}
|
||||
|
||||
if !resp.Ok {
|
||||
return nil, errors.Errorf("api error: %s", resp.Description)
|
||||
}
|
||||
}
|
||||
|
||||
if !resp.Ok {
|
||||
return nil, errors.Errorf("api error: %s", resp.Description)
|
||||
}
|
||||
|
||||
return resp.Result, nil
|
||||
}
|
||||
|
||||
func extractOkResponse(respJSON []byte) error {
|
||||
var resp struct {
|
||||
Ok bool
|
||||
Description string
|
||||
}
|
||||
|
||||
err := json.Unmarshal(respJSON, &resp)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "bad response json")
|
||||
}
|
||||
|
||||
if !resp.Ok {
|
||||
return errors.Errorf("api error: %s", resp.Description)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func extractOptions(how []interface{}) *SendOptions {
|
||||
var opts *SendOptions
|
||||
|
||||
for _, prop := range how {
|
||||
switch opt := prop.(type) {
|
||||
case *SendOptions:
|
||||
opts = opt.copy()
|
||||
|
||||
case *ReplyMarkup:
|
||||
if opts == nil {
|
||||
opts = &SendOptions{}
|
||||
}
|
||||
opts.ReplyMarkup = opt.copy()
|
||||
|
||||
case Option:
|
||||
if opts == nil {
|
||||
opts = &SendOptions{}
|
||||
}
|
||||
|
||||
switch opt {
|
||||
case NoPreview:
|
||||
opts.DisableWebPagePreview = true
|
||||
case Silent:
|
||||
opts.DisableNotification = true
|
||||
case ForceReply:
|
||||
if opts.ReplyMarkup == nil {
|
||||
opts.ReplyMarkup = &ReplyMarkup{}
|
||||
}
|
||||
opts.ReplyMarkup.ForceReply = true
|
||||
case OneTimeKeyboard:
|
||||
if opts.ReplyMarkup == nil {
|
||||
opts.ReplyMarkup = &ReplyMarkup{}
|
||||
}
|
||||
opts.ReplyMarkup.OneTimeKeyboard = true
|
||||
default:
|
||||
panic("telebot: unsupported flag-option")
|
||||
}
|
||||
|
||||
case ParseMode:
|
||||
if opts == nil {
|
||||
opts = &SendOptions{}
|
||||
}
|
||||
opts.ParseMode = opt
|
||||
|
||||
default:
|
||||
panic("telebot: unsupported send-option")
|
||||
}
|
||||
}
|
||||
|
||||
return opts
|
||||
}
|
||||
|
||||
func embedSendOptions(params map[string]string, opt *SendOptions) {
|
||||
if opt == nil {
|
||||
return
|
||||
}
|
||||
|
||||
if opt.ReplyTo != nil && opt.ReplyTo.ID != 0 {
|
||||
params["reply_to_message_id"] = strconv.Itoa(opt.ReplyTo.ID)
|
||||
}
|
||||
|
||||
if opt.DisableWebPagePreview {
|
||||
params["disable_web_page_preview"] = "true"
|
||||
}
|
||||
|
||||
if opt.DisableNotification {
|
||||
params["disable_notification"] = "true"
|
||||
}
|
||||
|
||||
if opt.ParseMode != ModeDefault {
|
||||
params["parse_mode"] = string(opt.ParseMode)
|
||||
}
|
||||
|
||||
if opt.ReplyMarkup != nil {
|
||||
processButtons(opt.ReplyMarkup.InlineKeyboard)
|
||||
replyMarkup, _ := json.Marshal(opt.ReplyMarkup)
|
||||
params["reply_markup"] = string(replyMarkup)
|
||||
}
|
||||
}
|
||||
|
||||
func processButtons(keys [][]InlineButton) {
|
||||
if keys == nil || len(keys) < 1 || len(keys[0]) < 1 {
|
||||
return
|
||||
}
|
||||
|
||||
for i, _ := range keys {
|
||||
for j, _ := range keys[i] {
|
||||
key := &keys[i][j]
|
||||
if key.Unique != "" {
|
||||
// Format: "\f<callback_name>|<data>"
|
||||
data := key.Data
|
||||
if data == "" {
|
||||
key.Data = "\f" + key.Unique
|
||||
} else {
|
||||
key.Data = "\f" + key.Unique + "|" + data
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func embedRights(p map[string]string, prv Rights) {
|
||||
jsonRepr, _ := json.Marshal(prv)
|
||||
json.Unmarshal(jsonRepr, p)
|
||||
}
|
Reference in New Issue
Block a user