lib.go

package luadoc

import (
	"fmt"
	"strings"

	"github.com/samber/lo"
	lua "github.com/yuin/gopher-lua"
)

type Lib struct {
	Name        string
	Description string
	Vars        []*Var
	Funcs       []*Func
	Classes     []*Class
	Libs        []*Lib
}

func (l *Lib) Value(state *lua.LState) *lua.LTable {
	var (
		vars  = make(map[string]lua.LValue)
		funcs = make(map[string]lua.LGFunction)
	)

	for _, f := range l.Funcs {
		if f.Value == nil {
			panic(fmt.Sprintf("function %s.%s has no value", l.Name, f.Name))
		}

		for _, param := range f.Params {
			if param.Type == "" {
				panic(fmt.Sprintf("function %s.%s has a parameter %s with no type", l.Name, f.Name, param.Name))
			}
		}

		for _, ret := range f.Returns {
			if ret.Type == "" {
				panic(fmt.Sprintf("function %s.%s has a return %s with no type", l.Name, f.Name, ret.Name))
			}
		}

		funcs[f.Name] = f.Value
	}

	for _, v := range l.Vars {
		if v.Value == nil {
			panic(fmt.Sprintf("variable %s.%s has no value", l.Name, v.Name))
		}

		vars[v.Name] = v.Value
	}

	for _, l := range l.Libs {
		vars[l.Name] = l.Value(state)
	}

	for _, c := range l.Classes {
		methods := make(map[string]lua.LGFunction)
		for _, m := range c.Methods {
			if m.Value == nil {
				panic(fmt.Sprintf("method %s of class %s.%s has no value", m.Name, l.Name, c.Name))
			}

			for _, param := range m.Params {
				if param.Type == "" {
					panic(fmt.Sprintf("method %s of class %s.%s has a parameter %s with no type", m.Name, l.Name, c.Name, param.Name))
				}
			}

			for _, ret := range m.Returns {
				if ret.Type == "" {
					panic(fmt.Sprintf("method %s of class %s.%s has a return %s with no type", m.Name, l.Name, c.Name, ret.Name))
				}
			}

			methods[m.Name] = m.Value
		}

		mt := state.NewTypeMetatable(c.Name)
		state.SetField(mt, "__index", state.SetFuncs(state.NewTable(), methods))
	}

	return newTable(state, vars, funcs)
}

func (l *Lib) LuaDoc() string {
	var b strings.Builder
	lo.Must0(templateLuaDocLib.Execute(&b, l))

	return b.String()
}

func (l *Lib) Loader() lua.LGFunction {
	return func(state *lua.LState) int {
		libs := state.NewTable()

		for _, lib := range l.Libs {
			state.SetField(libs, lib.Name, lib.Value(state))
		}

		state.Push(libs)
		return 1
	}
}