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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
package main
import (
"github.com/dragonfireclient/hydra-dragonfire/convert"
"github.com/dragonfireclient/mt"
"github.com/yuin/gopher-lua"
"sync"
)
type CompPkts struct {
client *Client
mu sync.Mutex
wildcard bool
subscribed map[string]struct{}
userdata *lua.LUserData
}
var compPktsFuncs = map[string]lua.LGFunction{
"subscribe": l_comp_pkts_subscribe,
"unsubscribe": l_comp_pkts_unsubscribe,
"wildcard": l_comp_pkts_wildcard,
}
type EventPkt struct {
pktType string
pktData *mt.Pkt
}
func (evt EventPkt) handle(l *lua.LState, val lua.LValue) {
l.SetField(val, "type", lua.LString("pkt"))
l.SetField(val, "pkt_type", lua.LString(evt.pktType))
l.SetField(val, "pkt_data", convert.PushPkt(l, evt.pktData))
}
func getCompPkts(l *lua.LState) *CompPkts {
return l.CheckUserData(1).Value.(*CompPkts)
}
func (comp *CompPkts) create(client *Client, l *lua.LState) {
comp.client = client
comp.wildcard = false
comp.subscribed = map[string]struct{}{}
comp.userdata = l.NewUserData()
comp.userdata.Value = comp
l.SetMetatable(comp.userdata, l.GetTypeMetatable("hydra.comp.pkts"))
}
func (comp *CompPkts) push() lua.LValue {
return comp.userdata
}
func (comp *CompPkts) connect() {
}
func (comp *CompPkts) process(pkt *mt.Pkt) {
pktType := string(convert.PushPktType(pkt))
comp.mu.Lock()
_, subscribed := comp.subscribed[pktType]
comp.mu.Unlock()
if subscribed || comp.wildcard {
comp.client.queue <- EventPkt{pktType: pktType, pktData: pkt}
}
}
func l_comp_pkts_subscribe(l *lua.LState) int {
comp := getCompPkts(l)
n := l.GetTop()
comp.mu.Lock()
defer comp.mu.Unlock()
for i := 2; i <= n; i++ {
comp.subscribed[l.CheckString(i)] = struct{}{}
}
return 0
}
func l_comp_pkts_unsubscribe(l *lua.LState) int {
comp := getCompPkts(l)
n := l.GetTop()
comp.mu.Lock()
defer comp.mu.Unlock()
for i := 2; i <= n; i++ {
delete(comp.subscribed, l.CheckString(i))
}
return 0
}
func l_comp_pkts_wildcard(l *lua.LState) int {
comp := getCompPkts(l)
comp.wildcard = l.ToBool(2)
return 0
}
|
