telegram-bot-api/configs.go

456 lines
9.3 KiB
Go
Raw Normal View History

2015-11-20 11:42:26 +01:00
package tgbotapi
import (
2015-11-20 12:19:37 +01:00
"encoding/json"
2015-11-20 11:42:26 +01:00
"io"
"net/url"
2015-11-20 12:06:51 +01:00
"strconv"
2015-11-20 11:42:26 +01:00
)
// Telegram constants
const (
// APIEndpoint is the endpoint for all API methods, with formatting for Sprintf
APIEndpoint = "https://api.telegram.org/bot%s/%s"
// FileEndpoint is the endpoint for downloading a file from Telegram
FileEndpoint = "https://api.telegram.org/file/bot%s/%s"
)
// Constant values for ChatActions
const (
ChatTyping = "typing"
ChatUploadPhoto = "upload_photo"
ChatRecordVideo = "record_video"
ChatUploadVideo = "upload_video"
ChatRecordAudio = "record_audio"
ChatUploadAudio = "upload_audio"
ChatUploadDocument = "upload_document"
ChatFindLocation = "find_location"
)
// API errors
const (
// APIForbidden happens when a token is bad
APIForbidden = "forbidden"
)
// Constant values for ParseMode in MessageConfig
const (
ModeMarkdown = "Markdown"
)
2015-11-20 15:08:53 +01:00
type Chattable interface {
Values() (url.Values, error)
2015-11-20 15:55:32 +01:00
Method() string
2015-11-20 15:08:53 +01:00
}
type Fileable interface {
Chattable
Params() (map[string]string, error)
2015-11-20 15:31:01 +01:00
Name() string
2015-11-20 15:08:53 +01:00
GetFile() interface{}
2015-11-20 15:31:01 +01:00
UseExistingFile() bool
2015-11-20 15:08:53 +01:00
}
2015-11-20 11:42:26 +01:00
// Base struct for all chat event(Message, Photo and so on)
2015-11-20 15:08:53 +01:00
type BaseChat struct {
2015-11-20 16:30:50 +01:00
ChatID int
ChannelUsername string
ReplyToMessageID int
ReplyMarkup interface{}
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (chat *BaseChat) Values() (url.Values, error) {
2015-11-20 12:06:51 +01:00
v := url.Values{}
2015-11-20 15:08:53 +01:00
if chat.ChannelUsername != "" {
v.Add("chat_id", chat.ChannelUsername)
2015-11-20 12:06:51 +01:00
} else {
2015-11-20 15:08:53 +01:00
v.Add("chat_id", strconv.Itoa(chat.ChatID))
2015-11-20 12:06:51 +01:00
}
2015-11-20 16:30:50 +01:00
if chat.ReplyToMessageID != 0 {
v.Add("reply_to_message_id", strconv.Itoa(chat.ReplyToMessageID))
}
if chat.ReplyMarkup != nil {
data, err := json.Marshal(chat.ReplyMarkup)
if err != nil {
return v, err
}
v.Add("reply_markup", string(data))
}
2015-11-20 12:06:51 +01:00
return v, nil
}
2015-11-20 15:08:53 +01:00
type BaseFile struct {
BaseChat
FilePath string
File interface{}
FileID string
UseExisting bool
}
func (file BaseFile) Params() (map[string]string, error) {
params := make(map[string]string)
if file.ChannelUsername != "" {
params["chat_id"] = file.ChannelUsername
} else {
params["chat_id"] = strconv.Itoa(file.ChatID)
}
2015-11-20 16:30:50 +01:00
if file.ReplyToMessageID != 0 {
params["reply_to_message_id"] = strconv.Itoa(file.ReplyToMessageID)
}
if file.ReplyMarkup != nil {
data, err := json.Marshal(file.ReplyMarkup)
if err != nil {
return params, err
}
params["reply_markup"] = string(data)
}
2015-11-20 15:08:53 +01:00
return params, nil
}
func (file BaseFile) GetFile() interface{} {
var result interface{}
if file.FilePath == "" {
result = file.File
} else {
result = file.FilePath
}
return result
}
2015-11-20 15:31:01 +01:00
func (file BaseFile) UseExistingFile() bool {
return file.UseExisting
}
2015-11-20 11:42:26 +01:00
// MessageConfig contains information about a SendMessage request.
type MessageConfig struct {
2015-11-20 15:08:53 +01:00
BaseChat
2015-11-20 11:42:26 +01:00
Text string
ParseMode string
DisableWebPagePreview bool
ReplyMarkup interface{}
}
2015-11-20 15:08:53 +01:00
func (config MessageConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
v.Add("text", config.Text)
v.Add("disable_web_page_preview", strconv.FormatBool(config.DisableWebPagePreview))
if config.ParseMode != "" {
v.Add("parse_mode", config.ParseMode)
}
return v, nil
}
2015-11-20 15:55:32 +01:00
func (config MessageConfig) Method() string {
return "SendMessage"
}
2015-11-20 11:42:26 +01:00
// ForwardConfig contains information about a ForwardMessage request.
type ForwardConfig struct {
2015-11-20 15:08:53 +01:00
BaseChat
2015-11-20 11:42:26 +01:00
FromChatID int
FromChannelUsername string
MessageID int
}
2015-11-20 15:08:53 +01:00
func (config ForwardConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 18:26:12 +01:00
v.Add("from_chat_id", strconv.Itoa(config.FromChatID))
2015-11-20 12:06:51 +01:00
v.Add("message_id", strconv.Itoa(config.MessageID))
return v, nil
}
2015-11-20 15:55:32 +01:00
func (config ForwardConfig) Method() string {
return "forwardMessage"
}
2015-11-20 11:42:26 +01:00
// PhotoConfig contains information about a SendPhoto request.
type PhotoConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 16:30:50 +01:00
Caption string
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config PhotoConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
if config.Caption != "" {
params["caption"] = config.Caption
}
return params, nil
}
func (config PhotoConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
if config.Caption != "" {
v.Add("caption", config.Caption)
}
return v, nil
}
2015-11-20 15:31:01 +01:00
func (config PhotoConfig) Name() string {
return "photo"
}
2015-11-20 15:55:32 +01:00
func (config PhotoConfig) Method() string {
return "SendPhoto"
}
2015-11-20 11:42:26 +01:00
// AudioConfig contains information about a SendAudio request.
type AudioConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 16:30:50 +01:00
Duration int
Performer string
Title string
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config AudioConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
if config.Duration != 0 {
v.Add("duration", strconv.Itoa(config.Duration))
}
if config.Performer != "" {
v.Add("performer", config.Performer)
}
if config.Title != "" {
v.Add("title", config.Title)
}
return v, nil
}
2015-11-20 15:08:53 +01:00
func (config AudioConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
if config.Duration != 0 {
params["duration"] = strconv.Itoa(config.Duration)
}
if config.Performer != "" {
params["performer"] = config.Performer
}
if config.Title != "" {
params["title"] = config.Title
}
return params, nil
}
2015-11-20 15:31:01 +01:00
func (config AudioConfig) Name() string {
return "audio"
}
2015-11-20 15:55:32 +01:00
func (config AudioConfig) Method() string {
return "SendAudio"
}
2015-11-20 11:42:26 +01:00
// DocumentConfig contains information about a SendDocument request.
type DocumentConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config DocumentConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
return v, nil
}
2015-11-20 15:08:53 +01:00
func (config DocumentConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
return params, nil
}
2015-11-20 15:31:01 +01:00
func (config DocumentConfig) Name() string {
return "document"
}
2015-11-20 15:55:32 +01:00
func (config DocumentConfig) Method() string {
return "sendDocument"
}
2015-11-20 11:42:26 +01:00
// StickerConfig contains information about a SendSticker request.
type StickerConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config StickerConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
return v, nil
}
2015-11-20 15:08:53 +01:00
func (config StickerConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
return params, nil
}
2015-11-20 15:31:01 +01:00
func (config StickerConfig) Name() string {
return "sticker"
}
2015-11-20 15:55:32 +01:00
func (config StickerConfig) Method() string {
return "sendSticker"
}
2015-11-20 11:42:26 +01:00
// VideoConfig contains information about a SendVideo request.
type VideoConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 16:30:50 +01:00
Duration int
Caption string
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config VideoConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
if config.Duration != 0 {
v.Add("duration", strconv.Itoa(config.Duration))
}
if config.Caption != "" {
v.Add("caption", config.Caption)
}
return v, nil
}
2015-11-20 15:08:53 +01:00
func (config VideoConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
return params, nil
}
2015-11-20 15:31:01 +01:00
func (config VideoConfig) Name() string {
2015-11-20 16:30:50 +01:00
return "video"
2015-11-20 15:31:01 +01:00
}
2015-11-20 15:55:32 +01:00
func (config VideoConfig) Method() string {
return "sendVideo"
}
2015-11-20 11:42:26 +01:00
// VoiceConfig contains information about a SendVoice request.
type VoiceConfig struct {
2015-11-20 15:08:53 +01:00
BaseFile
2015-11-20 16:30:50 +01:00
Duration int
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config VoiceConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
2015-11-21 12:22:08 +01:00
v.Add(config.Name(), config.FileID)
2015-11-20 12:06:51 +01:00
if config.Duration != 0 {
v.Add("duration", strconv.Itoa(config.Duration))
}
return v, nil
}
2015-11-20 15:08:53 +01:00
func (config VoiceConfig) Params() (map[string]string, error) {
params, _ := config.BaseFile.Params()
if config.Duration != 0 {
params["duration"] = strconv.Itoa(config.Duration)
}
return params, nil
}
2015-11-20 15:31:01 +01:00
func (config VoiceConfig) Name() string {
return "voice"
}
2015-11-20 15:55:32 +01:00
func (config VoiceConfig) Method() string {
return "sendVoice"
}
2015-11-20 11:42:26 +01:00
// LocationConfig contains information about a SendLocation request.
type LocationConfig struct {
2015-11-20 15:08:53 +01:00
BaseChat
2015-11-20 16:30:50 +01:00
Latitude float64
Longitude float64
2015-11-20 11:42:26 +01:00
}
2015-11-20 15:08:53 +01:00
func (config LocationConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
v.Add("latitude", strconv.FormatFloat(config.Latitude, 'f', 6, 64))
v.Add("longitude", strconv.FormatFloat(config.Longitude, 'f', 6, 64))
2015-11-20 12:19:37 +01:00
2015-11-20 12:06:51 +01:00
return v, nil
}
2015-11-20 15:55:32 +01:00
func (config LocationConfig) Method() string {
return "sendLocation"
}
2015-11-20 11:42:26 +01:00
// ChatActionConfig contains information about a SendChatAction request.
type ChatActionConfig struct {
2015-11-20 15:08:53 +01:00
BaseChat
2015-11-20 11:42:26 +01:00
Action string
}
2015-11-20 15:08:53 +01:00
func (config ChatActionConfig) Values() (url.Values, error) {
v, _ := config.BaseChat.Values()
2015-11-20 12:06:51 +01:00
v.Add("action", config.Action)
return v, nil
}
2015-11-20 11:42:26 +01:00
// UserProfilePhotosConfig contains information about a GetUserProfilePhotos request.
type UserProfilePhotosConfig struct {
UserID int
Offset int
Limit int
}
// FileConfig has information about a file hosted on Telegram
type FileConfig struct {
FileID string
}
// UpdateConfig contains information about a GetUpdates request.
type UpdateConfig struct {
Offset int
Limit int
Timeout int
}
// WebhookConfig contains information about a SetWebhook request.
type WebhookConfig struct {
URL *url.URL
Certificate interface{}
}
// FileBytes contains information about a set of bytes to upload as a File.
type FileBytes struct {
Name string
Bytes []byte
}
// FileReader contains information about a reader to upload as a File.
// If Size is -1, it will read the entire Reader into memory to calculate a Size.
type FileReader struct {
Name string
Reader io.Reader
Size int64
}