Hilbish/moonlight/table_clua.go

//go:build midnight

package moonlight

import (
	"github.com/aarzilli/golua/lua"
)

type Table struct {
	refIdx       int
	mlr          *Runtime
	nativeFields map[Value]Value
}

func NewTable() *Table {
	return &Table{
		refIdx:       -1,
		nativeFields: make(map[Value]Value),
	}
}

func (t *Table) SetRuntime(mlr *Runtime) {
	t.mlr = mlr

	if t.refIdx == -1 {
		mlr.state.NewTable()
		t.refIdx = mlr.state.Ref(lua.LUA_REGISTRYINDEX)
		t.Push() // because Ref pops off the stack
		t.syncToLua()
		mlr.state.Pop(1)
	}
}

func (t *Table) Get(val Value) Value {
	return NilValue
}

func (t *Table) Push() {
	t.mlr.state.RawGeti(lua.LUA_REGISTRYINDEX, t.refIdx)
}

func (t *Table) SetField(key string, value Value) {
	if t.refIdx != -1 {
		t.setInLua(StringValue(key), value)
		return
	}

	t.setInGo(key, value)
}

func (t *Table) setInLua(key Value, value Value) {
	t.Push()
	defer t.mlr.state.Pop(1)

	t.mlr.pushToState(value)
	t.mlr.pushToState(key)
	t.mlr.state.Insert(-2)
	t.mlr.state.SetTable(-3)
}

func (t *Table) setInGo(key string, value Value) {
	t.nativeFields[StringValue(key)] = value
}

func (t *Table) Set(key Value, value Value) {
	t.nativeFields[key] = value
}

func (t *Table) syncToLua() {
	for k, v := range t.nativeFields {
		t.setInLua(k, v)
	}
}

func ForEach(tbl *Table, cb func(key Value, val Value)) {
}

func (mlr *Runtime) GlobalTable() *Table {
	mlr.state.GetGlobal("_G")
	return &Table{
		refIdx: mlr.state.Ref(lua.LUA_REGISTRYINDEX),
	}
}

func ToTable(v Value) *Table {
	return &Table{
		refIdx: -1,
	}
}

func TryTable(v Value) (*Table, bool) {
	return nil, false
}

func (t *Table) setRefIdx(mlr *Runtime, i int) {
	t.refIdx = mlr.state.Ref(i)
}