pkts.go

   1 package main
   2
   3 import (
   4         "github.com/anon55555/mt"
   5         "github.com/dragonfireclient/hydra-dragonfire/convert"
   6         "github.com/yuin/gopher-lua"
   7         "sync"
   8 )
   9
  10 type Pkts struct {
  11         client     *Client
  12         mu         sync.Mutex
  13         wildcard   bool
  14         subscribed map[string]struct{}
  15         userdata   *lua.LUserData
  16 }
  17
  18 var pktsFuncs = map[string]lua.LGFunction{
  19         "subscribe":   l_pkts_subscribe,
  20         "unsubscribe": l_pkts_unsubscribe,
  21         "wildcard":    l_pkts_wildcard,
  22 }
  23
  24 type EventPkt struct {
  25         pktType string
  26         pktData *mt.Pkt
  27 }
  28
  29 func (evt EventPkt) handle(l *lua.LState, val lua.LValue) {
  30         l.SetField(val, "type", lua.LString("pkt"))
  31         l.SetField(val, "pkt_type", lua.LString(evt.pktType))
  32         l.SetField(val, "pkt_data", convert.PushPkt(l, evt.pktData))
  33 }
  34
  35 func getPkts(l *lua.LState) *Pkts {
  36         return l.CheckUserData(1).Value.(*Pkts)
  37 }
  38
  39 func (pkts *Pkts) create(client *Client, l *lua.LState) {
  40         pkts.client = client
  41         pkts.wildcard = false
  42         pkts.subscribed = map[string]struct{}{}
  43         pkts.userdata = l.NewUserData()
  44         pkts.userdata.Value = pkts
  45         l.SetMetatable(pkts.userdata, l.GetTypeMetatable("hydra.pkts"))
  46 }
  47
  48 func (pkts *Pkts) push() lua.LValue {
  49         return pkts.userdata
  50 }
  51
  52 func (pkts *Pkts) connect() {
  53 }
  54
  55 func (pkts *Pkts) process(pkt *mt.Pkt) {
  56         pktType := string(convert.PushPktType(pkt))
  57
  58         pkts.mu.Lock()
  59         _, subscribed := pkts.subscribed[pktType]
  60         pkts.mu.Unlock()
  61
  62         if subscribed || pkts.wildcard {
  63                 pkts.client.queue <- EventPkt{pktType: pktType, pktData: pkt}
  64         }
  65 }
  66
  67 func l_pkts_subscribe(l *lua.LState) int {
  68         pkts := getPkts(l)
  69         n := l.GetTop()
  70
  71         pkts.mu.Lock()
  72         defer pkts.mu.Unlock()
  73
  74         for i := 2; i <= n; i++ {
  75                 pkts.subscribed[l.CheckString(i)] = struct{}{}
  76         }
  77
  78         return 0
  79 }
  80
  81 func l_pkts_unsubscribe(l *lua.LState) int {
  82         pkts := getPkts(l)
  83         n := l.GetTop()
  84
  85         pkts.mu.Lock()
  86         defer pkts.mu.Unlock()
  87
  88         for i := 2; i <= n; i++ {
  89                 delete(pkts.subscribed, l.CheckString(i))
  90         }
  91
  92         return 0
  93 }
  94
  95 func l_pkts_wildcard(l *lua.LState) int {
  96         pkts := getPkts(l)
  97         pkts.wildcard = l.ToBool(2)
  98         return 0
  99 }