70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
package behx
|
||
|
|
|
||
|
|
import (
|
||
|
|
apix "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
// The wrapper around Telegram API.
|
||
|
|
type Bot struct {
|
||
|
|
*apix.BotAPI
|
||
|
|
*Behaviour
|
||
|
|
sessions SessionMap
|
||
|
|
}
|
||
|
|
|
||
|
|
// Return the new bot for running the Behaviour.
|
||
|
|
func NewBot(token string, beh *Behaviour, sessions SessionMap) (*Bot, error) {
|
||
|
|
bot, err := tgbotapi.NewBotAPI(token)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make new sessions if no current are provided.
|
||
|
|
if sessions == nil {
|
||
|
|
sessions = make(SessionMap)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Bot{
|
||
|
|
BotAPI: bot,
|
||
|
|
Behaviour: beh,
|
||
|
|
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run the bot with the Behaviour.
|
||
|
|
func (bot *Bot) Run() error {
|
||
|
|
bot.Debug = true
|
||
|
|
|
||
|
|
uc := tgbotapi.NewUpdate(0)
|
||
|
|
uc.Timeout = 60
|
||
|
|
|
||
|
|
updates := bot.GetUpdatesChan(uc)
|
||
|
|
|
||
|
|
for u := range updates {
|
||
|
|
// Create new session if the one does not exist
|
||
|
|
// for this user.
|
||
|
|
sid := SessionId(u.Message.Chat.ID)
|
||
|
|
if !bot.sessions.Exist(sid) {
|
||
|
|
bot.sessions.Add(sid)
|
||
|
|
}
|
||
|
|
|
||
|
|
session := bot.sessions.Get(sid)
|
||
|
|
ctx := &beh.Context{
|
||
|
|
B: bot,
|
||
|
|
S: session,
|
||
|
|
}
|
||
|
|
|
||
|
|
// The "start" command resets the bot
|
||
|
|
// by executing the Start Action.
|
||
|
|
if u.MessageIsCommand() {
|
||
|
|
cmd := u.Message.Command()
|
||
|
|
if cmd == "start" {
|
||
|
|
bot.Start.Act(ctx)
|
||
|
|
}
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|