Allow multiple clients to share a map

[hydra-dragonfire.git] / comp_pkts.go

diff --git a/comp_pkts.go b/comp_pkts.go
new file mode 100644 (file)
index 0000000..37e2284
--- /dev/null
+++ b/comp_pkts.go
@@ -0,0 +1,99 @@
+package main
+
+import (
+       "github.com/anon55555/mt"
+       "github.com/dragonfireclient/hydra-dragonfire/convert"
+       "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
+}