generated from petuhovskiy/go-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
64 lines (53 loc) · 1.49 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package main
import (
"encoding/json"
"github.com/petuhovskiy/telegram"
"github.com/petuhovskiy/telegram/updates"
"github.com/rwlist/bubot/pkg/base"
"github.com/rwlist/bubot/pkg/handlers"
"net/http"
"github.com/rwlist/bubot/pkg/conf"
"github.com/prometheus/client_golang/prometheus/promhttp"
log "github.com/sirupsen/logrus"
)
func main() {
log.SetFormatter(&log.JSONFormatter{})
log.SetReportCaller(true)
log.SetLevel(log.DebugLevel)
cfg, err := conf.ParseEnv()
if err != nil {
log.WithError(err).Fatal("failed to parse config from env")
}
go func() {
mux := http.NewServeMux()
mux.Handle("/metrics", promhttp.Handler())
err := http.ListenAndServe(cfg.PrometheusBind, mux)
if err != nil && err != http.ErrServerClosed {
log.WithError(err).Fatal("prometheus server error")
}
}()
bot := telegram.NewBotWithOpts(cfg.Bot.Token, &telegram.Opts{
Middleware: func(handler telegram.RequestHandler) telegram.RequestHandler {
return func(methodName string, req interface{}) (message json.RawMessage, err error) {
res, err := handler(methodName, req)
if err != nil {
log.WithError(err).Error("telegram response error")
}
return res, err
}
},
})
ch, err := updates.StartPolling(bot, telegram.GetUpdatesRequest{
Offset: 0,
Limit: 50,
Timeout: 10,
})
if err != nil {
log.WithError(err).Fatal("in updates.StartPolling()")
}
rateHandler := &handlers.Rate{}
for upd := range ch {
req := base.NewRequest(bot, upd)
rateHandler.HandleCommand(req)
}
}