func main() {
configPath := flag.String("c", "", "path to config file")
flag.Parse()
if *configPath == "" {
flag.Usage()
return
}
// read and decode configuration
var config Config
cfg, err := ioutil.ReadFile(*configPath)
if err != nil {
log.Panic(err)
}
if _, err := toml.Decode(string(cfg), &config); err != nil {
log.Panic(err)
}
// initialize yahoo finance
var f *yahoo.YahooFinanceService
f, err = yahoo.New(config.Yahoo)
if err != nil {
log.Panic(err)
}
// initialize telegram framework
app, err := telegram.New(config.Telegram)
if err != nil {
log.Panic("could not init app : ", err)
}
// helper handlers
app.Use(
&slugger.Slugger{},
&canceler.Canceler{config.Canceler.Duration},
telegram.HandlerFunc(func(ctrl *telegram.Control, bot *tgbotapi.BotAPI, update tgbotapi.Update) {
ctrl.Log().Println(fmt.Sprintf(
"incoming message: text:%q; query:%q",
update.Message.Text, update.InlineQuery.Query,
))
ctrl.NextWithValue("start", time.Now())
ctrl.Next()
}),
)
// logic
app.UseOn("/help", helpHandler.New())
fh := financeHandler.New(f)
app.Use(condition.Condition{
matcher.RegExp{
Source: matcher.SourceText,
Pattern: regexp.MustCompile(`^\/([a-z]{3}$|rate.*$)`),
},
fh,
})
app.Use(condition.Condition{
matcher.RegExp{
Source: matcher.SourceQuery,
Pattern: regexp.MustCompile(`^([a-z]{3}|[a-z]{3} [a-z]{3})`),
},
fh,
})
// error handler
app.UseErrFunc(func(ctrl *telegram.Control, bot *tgbotapi.BotAPI, update tgbotapi.Update, err error) {
if update.Message.Text != "" {
bot.Send(tgbotapi.NewMessage(update.Message.Chat.ID, emoji.Sprintf(":space_invader:error: %s", err)))
}
ctrl.Log().Println("got error", err)
ctrl.Stop()
})
app.Use(telegram.HandlerFunc(func(ctrl *telegram.Control, bot *tgbotapi.BotAPI, update tgbotapi.Update) {
var dur float64
if start, ok := ctrl.Context().Value("start").(time.Time); ok {
dur = time.Since(start).Seconds() * 1000
} else {
dur = -1
}
ctrl.Log().Println(fmt.Sprintf(
"message processing complete in %.2fmsec", dur,
))
ctrl.Next()
}))
http.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("KURSOBOT_OK"))
})
log.Println("about to start app...")
app.Bot().Send(tgbotapi.NewMessage(config.NoticeChatID, emoji.Sprintf("Hey man, I have been restarted! :fireworks::sunglasses:")))
// start listen
log.Fatal(app.Listen())
}
func TestRoute(t *testing.T) {
g := Goblin(t)
g.Describe("Route", func() {
g.It("Should call valid pattern", func() {
var called int
route := Condition{matcher.Equal{Pattern: "/pattern"}, telegram.HandlerFunc(func(ctrl *telegram.Control, bot *tgbotapi.BotAPI, update tgbotapi.Update) {
called++
})}
route.Serve(&telegram.Control{}, &tgbotapi.BotAPI{}, tgbotapi.Update{Message: tgbotapi.Message{Text: "/pattern"}})
g.Assert(called).Eql(1)
})
})
}
