mirror of
https://github.com/dragonfireclient/hydra-dragonfire.git
synced 2024-11-21 10:24:01 -05:00
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
_ "embed"
|
|
"github.com/yuin/gopher-lua"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
var lastTime = time.Now()
|
|
var signalChannel chan os.Signal
|
|
|
|
var serializeVer uint8 = 28
|
|
var protoVer uint16 = 39
|
|
|
|
//go:embed builtin/luax/init.lua
|
|
var builtinLuaX string
|
|
|
|
//go:embed builtin/vector.lua
|
|
var builtinVector string
|
|
|
|
//go:embed builtin/escapes.lua
|
|
var builtinEscapes string
|
|
|
|
//go:embed builtin/client.lua
|
|
var builtinClient string
|
|
|
|
//go:embed builtin/base64.lua
|
|
var builtinBase64 string
|
|
|
|
var builtinFiles = []string{
|
|
builtinLuaX,
|
|
builtinVector,
|
|
builtinEscapes,
|
|
builtinClient,
|
|
builtinBase64,
|
|
}
|
|
|
|
var hydraFuncs = map[string]lua.LGFunction{
|
|
"client": l_client,
|
|
"map": l_map,
|
|
"dtime": l_dtime,
|
|
"poll": l_poll,
|
|
"close": l_close,
|
|
}
|
|
|
|
func l_dtime(l *lua.LState) int {
|
|
l.Push(lua.LNumber(time.Since(lastTime).Seconds()))
|
|
lastTime = time.Now()
|
|
return 1
|
|
}
|
|
|
|
func l_poll(l *lua.LState) int {
|
|
return doPoll(l, getClients(l))
|
|
}
|
|
|
|
func l_close(l *lua.LState) int {
|
|
for _, client := range getClients(l) {
|
|
client.closeConn()
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
panic("missing filename")
|
|
}
|
|
|
|
signalChannel = make(chan os.Signal, 1)
|
|
signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM, syscall.SIGHUP)
|
|
|
|
l := lua.NewState()
|
|
defer l.Close()
|
|
|
|
arg := l.NewTable()
|
|
for i, a := range os.Args {
|
|
l.RawSetInt(arg, i-1, lua.LString(a))
|
|
}
|
|
l.SetGlobal("arg", arg)
|
|
|
|
hydra := l.SetFuncs(l.NewTable(), hydraFuncs)
|
|
l.SetField(hydra, "BS", lua.LNumber(10.0))
|
|
l.SetField(hydra, "serialize_ver", lua.LNumber(serializeVer))
|
|
l.SetField(hydra, "proto_ver", lua.LNumber(protoVer))
|
|
l.SetGlobal("hydra", hydra)
|
|
|
|
l.SetField(l.NewTypeMetatable("hydra.client"), "__index", l.NewFunction(l_client_index))
|
|
l.SetField(l.NewTypeMetatable("hydra.map"), "__index", l.SetFuncs(l.NewTable(), mapFuncs))
|
|
|
|
l.SetField(l.NewTypeMetatable("hydra.comp.auth"), "__index", l.SetFuncs(l.NewTable(), compAuthFuncs))
|
|
l.SetField(l.NewTypeMetatable("hydra.comp.map"), "__index", l.SetFuncs(l.NewTable(), compMapFuncs))
|
|
l.SetField(l.NewTypeMetatable("hydra.comp.pkts"), "__index", l.SetFuncs(l.NewTable(), compPktsFuncs))
|
|
|
|
for _, str := range builtinFiles {
|
|
if err := l.DoString(str); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
if err := l.DoFile(os.Args[1]); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|