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
83
84
85
|
package main
import (
"github.com/Shopify/go-lua"
"github.com/anon55555/mt"
)
type Callbacks struct {
wildcard bool
subscribed map[string]struct{}
}
func getCallbacks(l *lua.State) *Callbacks {
return lua.CheckUserData(l, 1, "hydra.callbacks").(*Callbacks)
}
func (handler *Callbacks) create(client *Client) {
handler.subscribed = map[string]struct{}{}
}
func (handler *Callbacks) push(l *lua.State) {
l.PushUserData(handler)
if lua.NewMetaTable(l, "hydra.callbacks") {
lua.NewLibrary(l, []lua.RegistryFunction{
{Name: "wildcard", Function: l_callbacks_wildcard},
{Name: "subscribe", Function: l_callbacks_subscribe},
{Name: "unsubscribe", Function: l_callbacks_unsubscribe},
})
l.SetField(-2, "__index")
}
l.SetMetaTable(-2)
}
func (handler *Callbacks) canConnect() (bool, string) {
return true, ""
}
func (handler *Callbacks) connect() {
}
func (handler *Callbacks) handle(pkt *mt.Pkt, l *lua.State, idx int) {
if !handler.wildcard && pkt != nil {
if _, exists := handler.subscribed[pktToString(pkt)]; !exists {
return
}
}
if !l.IsFunction(2) {
return
}
l.PushValue(2) // callback
l.RawGetInt(1, idx) // arg 1: client
pktToLua(l, pkt) // arg 2: pkt
l.Call(2, 0)
}
func l_callbacks_wildcard(l *lua.State) int {
handler := getCallbacks(l)
handler.wildcard = l.ToBoolean(2)
return 0
}
func l_callbacks_subscribe(l *lua.State) int {
handler := getCallbacks(l)
n := l.Top()
for i := 2; i <= n; i++ {
handler.subscribed[lua.CheckString(l, i)] = struct{}{}
}
return 0
}
func l_callbacks_unsubscribe(l *lua.State) int {
handler := getCallbacks(l)
n := l.Top()
for i := 2; i <= n; i++ {
delete(handler.subscribed, lua.CheckString(l, i))
}
return 0
}
|
