/
handler.go
82 lines (66 loc) · 1.57 KB
/
handler.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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package commands
import (
"strings"
"github.com/bwmarrin/discordgo"
)
// [prefix][invoke/alias] [1st arg] [2nd arg] [3rd agrg]
// ;;kick zekro ist dumm
type CommandHandler struct {
prefix string
cmdInstances []Command
cmdMap map[string]Command
middlewares []Middleware
OnError func(err error, ctx *Context)
}
func NewCommandHandler(prefix string) *CommandHandler {
return &CommandHandler{
prefix: prefix,
cmdInstances: make([]Command, 0),
cmdMap: make(map[string]Command),
middlewares: make([]Middleware, 0),
OnError: func(error, *Context) {},
}
}
func (c *CommandHandler) RegisterCommand(cmd Command) {
c.cmdInstances = append(c.cmdInstances, cmd)
for _, invoke := range cmd.Invokes() {
c.cmdMap[invoke] = cmd
}
}
func (c *CommandHandler) RegisterMiddleware(mw Middleware) {
c.middlewares = append(c.middlewares, mw)
}
func (c *CommandHandler) HandleMessage(s *discordgo.Session, e *discordgo.MessageCreate) {
if e.Author.ID == s.State.User.ID || e.Author.Bot || !strings.HasPrefix(e.Content, c.prefix) {
return
}
split := strings.Split(e.Content[len(c.prefix):], " ")
if len(split) < 1 {
return
}
invoke := strings.ToLower(split[0])
args := split[1:]
cmd, ok := c.cmdMap[invoke]
if !ok || cmd == nil {
return
}
ctx := &Context{
Session: s,
Args: args,
Handler: c,
Message: e.Message,
}
for _, mw := range c.middlewares {
next, err := mw.Exec(ctx, cmd)
if err != nil {
c.OnError(err, ctx)
return
}
if !next {
return
}
}
if err := cmd.Exec(ctx); err != nil {
c.OnError(err, ctx)
}
}