Error handling. Vendoring
Signed-off-by: Ettore Dreucci <ettore.dreucci@gmail.com>
This commit is contained in:
187
vendor/gopkg.in/tucnak/telebot.v2/api.go
generated
vendored
Normal file
187
vendor/gopkg.in/tucnak/telebot.v2/api.go
generated
vendored
Normal file
@@ -0,0 +1,187 @@
|
||||
package telebot
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// Raw lets you call any method of Bot API manually.
|
||||
func (b *Bot) Raw(method string, payload interface{}) ([]byte, error) {
|
||||
url := fmt.Sprintf("https://api.telegram.org/bot%s/%s", b.Token, method)
|
||||
|
||||
var buf bytes.Buffer
|
||||
if err := json.NewEncoder(&buf).Encode(payload); err != nil {
|
||||
return []byte{}, wrapSystem(err)
|
||||
}
|
||||
|
||||
resp, err := b.client.Post(url, "application/json", &buf)
|
||||
if err != nil {
|
||||
return []byte{}, errors.Wrap(err, "http.Post failed")
|
||||
}
|
||||
resp.Close = true
|
||||
defer resp.Body.Close()
|
||||
json, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return []byte{}, wrapSystem(err)
|
||||
}
|
||||
|
||||
return json, nil
|
||||
}
|
||||
|
||||
func (b *Bot) sendFiles(
|
||||
method string,
|
||||
files map[string]string,
|
||||
params map[string]string) ([]byte, error) {
|
||||
// ---
|
||||
body := &bytes.Buffer{}
|
||||
writer := multipart.NewWriter(body)
|
||||
|
||||
for name, path := range files {
|
||||
if err := func() error {
|
||||
file, err := os.Open(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
part, err := writer.CreateFormFile(name, filepath.Base(path))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = io.Copy(part, file)
|
||||
return err
|
||||
} (); err != nil {
|
||||
return nil, wrapSystem(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
for field, value := range params {
|
||||
writer.WriteField(field, value)
|
||||
}
|
||||
|
||||
if err := writer.Close(); err != nil {
|
||||
return nil, wrapSystem(err)
|
||||
}
|
||||
|
||||
url := fmt.Sprintf("https://api.telegram.org/bot%s/%s", b.Token, method)
|
||||
req, err := http.NewRequest("POST", url, body)
|
||||
if err != nil {
|
||||
return nil, wrapSystem(err)
|
||||
}
|
||||
|
||||
req.Header.Add("Content-Type", writer.FormDataContentType())
|
||||
|
||||
resp, err := b.client.Do(req)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "http.Post failed")
|
||||
}
|
||||
|
||||
if resp.StatusCode == http.StatusInternalServerError {
|
||||
return nil, errors.New("api error: internal server error")
|
||||
}
|
||||
|
||||
json, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, wrapSystem(err)
|
||||
}
|
||||
|
||||
return json, nil
|
||||
}
|
||||
|
||||
func (b *Bot) sendObject(f *File, what string, params map[string]string) (*Message, error) {
|
||||
sendWhat := "send" + strings.Title(what)
|
||||
|
||||
if what == "videoNote" {
|
||||
what = "video_note"
|
||||
}
|
||||
|
||||
var respJSON []byte
|
||||
var err error
|
||||
|
||||
if f.InCloud() {
|
||||
params[what] = f.FileID
|
||||
respJSON, err = b.Raw(sendWhat, params)
|
||||
} else if f.FileURL != "" {
|
||||
params[what] = f.FileURL
|
||||
respJSON, err = b.Raw(sendWhat, params)
|
||||
} else {
|
||||
respJSON, err = b.sendFiles(sendWhat,
|
||||
map[string]string{what: f.FileLocal}, params)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return extractMsgResponse(respJSON)
|
||||
}
|
||||
|
||||
func (b *Bot) getMe() (*User, error) {
|
||||
meJSON, err := b.Raw("getMe", nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var botInfo struct {
|
||||
Ok bool
|
||||
Result *User
|
||||
Description string
|
||||
}
|
||||
|
||||
err = json.Unmarshal(meJSON, &botInfo)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "bad response json")
|
||||
}
|
||||
|
||||
if !botInfo.Ok {
|
||||
return nil, errors.Errorf("api error: %s", botInfo.Description)
|
||||
}
|
||||
|
||||
return botInfo.Result, nil
|
||||
|
||||
}
|
||||
|
||||
func (b *Bot) getUpdates(offset int, timeout time.Duration) (upd []Update, err error) {
|
||||
params := map[string]string{
|
||||
"offset": strconv.Itoa(offset),
|
||||
"timeout": strconv.Itoa(int(timeout / time.Second)),
|
||||
}
|
||||
updatesJSON, errCommand := b.Raw("getUpdates", params)
|
||||
if errCommand != nil {
|
||||
err = errCommand
|
||||
return
|
||||
|
||||
}
|
||||
var updatesReceived struct {
|
||||
Ok bool
|
||||
Result []Update
|
||||
Description string
|
||||
}
|
||||
|
||||
err = json.Unmarshal(updatesJSON, &updatesReceived)
|
||||
if err != nil {
|
||||
err = errors.Wrap(err, "bad response json")
|
||||
return
|
||||
}
|
||||
|
||||
if !updatesReceived.Ok {
|
||||
err = errors.Errorf("api error: %s", updatesReceived.Description)
|
||||
return
|
||||
}
|
||||
|
||||
return updatesReceived.Result, nil
|
||||
}
|
Reference in New Issue
Block a user