166 lines
3.0 KiB
Go
166 lines
3.0 KiB
Go
package telegram
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/kirillDanshin/dlog"
|
|
json "github.com/pquerna/ffjson/ffjson"
|
|
http "github.com/valyala/fasthttp"
|
|
)
|
|
|
|
type UpdatesChannel <-chan Update
|
|
|
|
func (bot *Bot) NewLongPollingChannel(params *GetUpdatesParameters) UpdatesChannel {
|
|
if params == nil {
|
|
params = &GetUpdatesParameters{
|
|
Offset: 0,
|
|
Limit: 100,
|
|
Timeout: 60,
|
|
}
|
|
}
|
|
|
|
channel := make(chan Update, params.Limit)
|
|
go func() {
|
|
for {
|
|
updates, err := bot.GetUpdates(params)
|
|
if err != nil {
|
|
dlog.Ln(err.Error())
|
|
dlog.Ln("Failed to get updates, retrying in 3 seconds...")
|
|
time.Sleep(time.Second * 3)
|
|
continue
|
|
}
|
|
|
|
for _, update := range updates {
|
|
if update.ID >= params.Offset {
|
|
params.Offset = update.ID + 1
|
|
channel <- update
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
return channel
|
|
}
|
|
|
|
func (bot *Bot) NewWebhookChannel(params *SetWebhookParameters, certFile, keyFile, set, listen, serve string) (updates UpdatesChannel) {
|
|
if params == nil {
|
|
params = &SetWebhookParameters{
|
|
URL: set,
|
|
MaxConnections: 40,
|
|
}
|
|
}
|
|
|
|
var err error
|
|
channel := make(chan Update, 100)
|
|
requiredPath := []byte(listen)
|
|
dlog.Ln("requiredPath:", string(requiredPath))
|
|
handleFunc := func(ctx *http.RequestCtx) {
|
|
dlog.Ln("Request path:", string(ctx.Path()))
|
|
if !bytes.HasPrefix(ctx.Path(), requiredPath) {
|
|
dlog.Ln("Unsupported request path:", string(ctx.Path()))
|
|
return
|
|
}
|
|
dlog.Ln("Catched supported request path:", string(ctx.Path()))
|
|
|
|
var update Update
|
|
if err = json.Unmarshal(ctx.Request.Body(), &update); err != nil {
|
|
return
|
|
}
|
|
|
|
channel <- update
|
|
}
|
|
|
|
go func() {
|
|
if certFile != "" && keyFile != "" {
|
|
dlog.Ln("Creating TLS router...")
|
|
err = http.ListenAndServeTLS(serve, certFile, keyFile, handleFunc)
|
|
} else {
|
|
dlog.Ln("Creating simple router...")
|
|
err = http.ListenAndServe(serve, handleFunc)
|
|
}
|
|
if err != nil {
|
|
log.Fatalln(err.Error())
|
|
}
|
|
}()
|
|
|
|
if _, err = bot.SetWebhook(params); err != nil {
|
|
log.Fatalln(err.Error())
|
|
}
|
|
|
|
return channel
|
|
}
|
|
|
|
func (upd *Update) IsMessage() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.Message != nil
|
|
}
|
|
|
|
func (upd *Update) IsEditedMessage() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.EditedMessage != nil
|
|
}
|
|
|
|
func (upd *Update) IsChannelPost() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.ChannelPost != nil
|
|
}
|
|
|
|
func (upd *Update) IsEditedChannelPost() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.EditedChannelPost != nil
|
|
}
|
|
|
|
func (upd *Update) IsInlineQuery() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.InlineQuery != nil
|
|
}
|
|
|
|
func (upd *Update) IsChosenInlineResult() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.ChosenInlineResult != nil
|
|
}
|
|
|
|
func (upd *Update) IsCallbackQuery() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.CallbackQuery != nil
|
|
}
|
|
|
|
func (upd *Update) IsShippingQuery() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.ShippingQuery != nil
|
|
}
|
|
|
|
func (upd *Update) IsPreCheckoutQuery() bool {
|
|
if upd == nil {
|
|
return false
|
|
}
|
|
|
|
return upd.PreCheckoutQuery != nil
|
|
}
|