7a749a3d0c
Fix int64 ids
152 lines
4.1 KiB
Go
152 lines
4.1 KiB
Go
package tamtam
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/neonxp/tamtam/schemes"
|
|
)
|
|
|
|
type messages struct {
|
|
client *client
|
|
}
|
|
|
|
func newMessages(client *client) *messages {
|
|
return &messages{client: client}
|
|
}
|
|
|
|
//GetMessages returns messages in chat: result page and marker referencing to the next page. Messages traversed in reverse direction so the latest message in chat will be first in result array. Therefore if you use from and to parameters, to must be less than from
|
|
func (a *messages) GetMessages(chatID int64, messageIDs []string, from int, to int, count int) (*schemes.MessageList, error) {
|
|
result := new(schemes.MessageList)
|
|
values := url.Values{}
|
|
if chatID != 0 {
|
|
values.Set("chat_id", strconv.Itoa(int(chatID)))
|
|
}
|
|
if len(messageIDs) > 0 {
|
|
for _, mid := range messageIDs {
|
|
values.Add("message_ids", mid)
|
|
}
|
|
}
|
|
if from != 0 {
|
|
values.Set("from", strconv.Itoa(from))
|
|
}
|
|
if to != 0 {
|
|
values.Set("to", strconv.Itoa(to))
|
|
}
|
|
if count > 0 {
|
|
values.Set("count", strconv.Itoa(count))
|
|
}
|
|
body, err := a.client.request(http.MethodGet, "messages", values, nil)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer func() {
|
|
if err := body.Close(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
return result, json.NewDecoder(body).Decode(result)
|
|
}
|
|
|
|
//EditMessage updates message by id
|
|
func (a *messages) EditMessage(messageID int64, message *Message) error {
|
|
s, err := a.editMessage(messageID, message.message)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !s.Success {
|
|
return errors.New(s.Message)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//DeleteMessage deletes message by id
|
|
func (a *messages) DeleteMessage(messageID int64) (*schemes.SimpleQueryResult, error) {
|
|
result := new(schemes.SimpleQueryResult)
|
|
values := url.Values{}
|
|
values.Set("message_id", strconv.Itoa(int(messageID)))
|
|
body, err := a.client.request(http.MethodDelete, "messages", values, nil)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer func() {
|
|
if err := body.Close(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
return result, json.NewDecoder(body).Decode(result)
|
|
}
|
|
|
|
//AnswerOnCallback should be called to send an answer after a user has clicked the button. The answer may be an updated message or/and a one-time user notification.
|
|
func (a *messages) AnswerOnCallback(callbackID string, callback *schemes.CallbackAnswer) (*schemes.SimpleQueryResult, error) {
|
|
result := new(schemes.SimpleQueryResult)
|
|
values := url.Values{}
|
|
values.Set("callback_id", callbackID)
|
|
body, err := a.client.request(http.MethodPost, "answers", values, callback)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer func() {
|
|
if err := body.Close(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
return result, json.NewDecoder(body).Decode(result)
|
|
}
|
|
|
|
//NewKeyboardBuilder returns new keyboard builder helper
|
|
func (a *messages) NewKeyboardBuilder() *Keyboard {
|
|
return &Keyboard{
|
|
rows: make([]*KeyboardRow, 0),
|
|
}
|
|
}
|
|
|
|
//Send sends a message to a chat. As a result for this method new message identifier returns.
|
|
func (a *messages) Send(m *Message) error {
|
|
return a.sendMessage(m.chatID, m.userID, m.message)
|
|
}
|
|
|
|
func (a *messages) sendMessage(chatID int64, userID int64, message *schemes.NewMessageBody) error {
|
|
result := new(schemes.Error)
|
|
values := url.Values{}
|
|
if chatID != 0 {
|
|
values.Set("chat_id", strconv.Itoa(int(chatID)))
|
|
}
|
|
if userID != 0 {
|
|
values.Set("user_id", strconv.Itoa(int(userID)))
|
|
}
|
|
body, err := a.client.request(http.MethodPost, "messages", values, message)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer body.Close()
|
|
if err := json.NewDecoder(body).Decode(result); err != nil {
|
|
// Message sent without errors
|
|
return nil
|
|
}
|
|
if result.Code == "" {
|
|
return nil
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (a *messages) editMessage(messageID int64, message *schemes.NewMessageBody) (*schemes.SimpleQueryResult, error) {
|
|
result := new(schemes.SimpleQueryResult)
|
|
values := url.Values{}
|
|
values.Set("message_id", strconv.Itoa(int(messageID)))
|
|
body, err := a.client.request(http.MethodPut, "messages", values, message)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer func() {
|
|
if err := body.Close(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
return result, json.NewDecoder(body).Decode(result)
|
|
}
|