112 lines
2.3 KiB
Go
112 lines
2.3 KiB
Go
package laniakea
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
)
|
|
|
|
func (b *Bot) handle(u *Update) {
|
|
ctx := &MsgContext{Bot: b, Update: u, Api: b.api}
|
|
for _, middleware := range b.middlewares {
|
|
middleware.Execute(ctx, b.dbContext)
|
|
}
|
|
|
|
if u.CallbackQuery != nil {
|
|
b.handleCallback(u, ctx)
|
|
} else {
|
|
b.handleMessage(u, ctx)
|
|
}
|
|
}
|
|
|
|
func (b *Bot) handleMessage(update *Update, ctx *MsgContext) {
|
|
if update.Message == nil {
|
|
return
|
|
}
|
|
|
|
var text string
|
|
if len(update.Message.Text) > 0 {
|
|
text = update.Message.Text
|
|
} else {
|
|
text = update.Message.Caption
|
|
}
|
|
|
|
text = strings.TrimSpace(text)
|
|
prefix, hasPrefix := b.checkPrefixes(text)
|
|
if !hasPrefix {
|
|
return
|
|
}
|
|
ctx.Prefix = prefix
|
|
ctx.FromID = update.Message.From.ID
|
|
ctx.From = update.Message.From
|
|
ctx.Msg = update.Message
|
|
|
|
text = strings.TrimSpace(text[len(prefix):])
|
|
|
|
for _, plugin := range b.plugins {
|
|
for cmd := range plugin.Commands {
|
|
if !strings.HasPrefix(text, cmd) {
|
|
continue
|
|
}
|
|
requestParts := strings.Split(text, " ")
|
|
cmdParts := strings.Split(cmd, " ")
|
|
isValid := true
|
|
for i, part := range cmdParts {
|
|
if part != requestParts[i] {
|
|
isValid = false
|
|
break
|
|
}
|
|
}
|
|
if !isValid {
|
|
continue
|
|
}
|
|
|
|
ctx.Text = strings.TrimSpace(text[len(cmd):])
|
|
ctx.Args = strings.Split(ctx.Text, " ")
|
|
|
|
if !plugin.executeMiddlewares(ctx, b.dbContext) {
|
|
return
|
|
}
|
|
go plugin.Execute(cmd, ctx, b.dbContext)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (b *Bot) handleCallback(update *Update, ctx *MsgContext) {
|
|
data := new(CallbackData)
|
|
err := json.Unmarshal([]byte(update.CallbackQuery.Data), data)
|
|
if err != nil {
|
|
b.logger.Errorln(err)
|
|
return
|
|
}
|
|
|
|
ctx.FromID = update.CallbackQuery.From.ID
|
|
ctx.From = update.CallbackQuery.From
|
|
ctx.Msg = update.CallbackQuery.Message
|
|
ctx.CallbackMsgId = update.CallbackQuery.Message.MessageID
|
|
ctx.CallbackQueryId = update.CallbackQuery.ID
|
|
ctx.Args = data.Args
|
|
|
|
for _, plugin := range b.plugins {
|
|
_, ok := plugin.Payloads[data.Command]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if !plugin.executeMiddlewares(ctx, b.dbContext) {
|
|
return
|
|
}
|
|
go plugin.ExecutePayload(data.Command, ctx, b.dbContext)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (b *Bot) checkPrefixes(text string) (string, bool) {
|
|
for _, prefix := range b.prefixes {
|
|
if strings.HasPrefix(text, prefix) {
|
|
return prefix, true
|
|
}
|
|
}
|
|
return "", false
|
|
}
|