bbgo_origin/examples/telebot/main.go
2022-01-13 22:15:05 +08:00

155 lines
3.8 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package main
import (
"os"
"time"
"github.com/google/uuid"
log "github.com/sirupsen/logrus"
tb "gopkg.in/tucnak/telebot.v2"
)
func main() {
var (
// Universal markup builders.
menu = &tb.ReplyMarkup{ResizeReplyKeyboard: true}
// Reply buttons.
btnHelp = menu.Text(" Help")
btnSettings = menu.Text("⚙ Settings")
)
menu.Reply(
menu.Row(btnHelp),
menu.Row(btnSettings),
)
b, err := tb.NewBot(tb.Settings{
// You can also set custom API URL.
// If field is empty it equals to "https://api.telegram.org".
// URL: "http://195.129.111.17:8012",
Token: os.Getenv("TELEGRAM_BOT_TOKEN"),
Poller: &tb.LongPoller{Timeout: 10 * time.Second},
// Synchronous: false,
Verbose: false,
// ParseMode: "",
// Reporter: nil,
// Client: nil,
// Offline: false,
})
if err != nil {
log.Fatal(err)
return
}
// Command: /start <PAYLOAD>
b.Handle("/start", func(m *tb.Message) {
if !m.Private() {
return
}
b.Send(m.Sender, "Hello!", menu)
})
// On reply button pressed (message)
b.Handle(&btnHelp, func(m *tb.Message) {
log.Infof("btnHelp: %#v", m)
var (
// Inline buttons.
//
// Pressing it will cause the client to
// send the bot a callback.
//
// Make sure Unique stays unique as per button kind,
// as it has to be for callback routing to work.
//
selector = &tb.ReplyMarkup{}
btnPrev = selector.Data("⬅", "prev", "data1", "data2")
btnNext = selector.Data("➡", "next", "data1", "data2")
)
selector.Inline(
selector.Row(btnPrev, btnNext),
)
// On inline button pressed (callback)
b.Handle(&btnPrev, func(c *tb.Callback) {
// Always respond!
b.Respond(c, &tb.CallbackResponse{
Text: "callback response",
ShowAlert: false,
// URL: "",
})
})
b.Send(m.Sender, "help button clicked", selector)
})
b.Handle("/hello", func(m *tb.Message) {
log.Infof("/hello %#v", m)
// b.Send(m.Sender, "Hello World!")
})
b.Handle(tb.OnText, func(m *tb.Message) {
log.Infof("[onText] %#v", m)
// all the text messages that weren't
// captured by existing handlers
})
b.Handle(tb.OnQuery, func(q *tb.Query) {
log.Infof("[onQuery] %#v", q)
// r := &tb.ReplyMarkup{}
// r.URL("test", "https://media.tenor.com/images/f176705ae1bb3c457e19d8cd71718ac0/tenor.gif")
urls := []string{
// "https://media.tenor.com/images/aae0cdf3c5a291cd7b96432180f6eee3/tenor.png",
// "https://media.tenor.com/images/905c1a9b1f56ae3c458b1ef58fd46357/tenor.png",
"https://media.tenor.com/images/2e69768f9537957ed3015a80ebc3f0f1/tenor.gif",
"https://media.tenor.com/images/6fcd72b29127a55e5c35db86d06d665c/tenor.gif",
"https://media.tenor.com/images/05dbf5bf3a3b88275bb045691541dc53/tenor.gif",
"https://media.tenor.com/images/0e1a52cfe5616c1509090d6ec2312db0/tenor.gif",
"https://media.tenor.com/images/1ca04a449b26e1f7d45682a79d2c8697/tenor.gif",
"https://media.tenor.com/images/a2844b186fb71c376226b56c4ea7730a/tenor.gif",
"https://media.tenor.com/images/ec636a1ebce1a3fc1c795b851c125b31/tenor.gif",
"https://media.tenor.com/images/ae103819cb05a0cf7497900b77b87d80/tenor.gif",
}
results := make(tb.Results, len(urls)) // []tb.Result
for i, url := range urls {
// result := &tb.PhotoResult{
result := &tb.GifResult{
ResultBase: tb.ResultBase{
// Type: "photo",
// Content: nil,
// ReplyMarkup: nil,
ID: uuid.New().String(),
},
URL: url,
// required for photos
ThumbURL: url,
}
results[i] = result
// needed to set a unique string ID for each result
// results[i].SetResultID(strconv.Itoa(i))
}
err := b.Answer(q, &tb.QueryResponse{
QueryID: q.ID,
Results: results,
// CacheTime: 60, // a minute
IsPersonal: true,
})
if err != nil {
log.Println(err)
}
})
b.Start()
}