wehub-resource-sync
125 lines
2.9 KiB
Go
125 lines
2.9 KiB
Go
package luaplugin
|
|
|
|
import (
|
|
lua "github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
// luaVis wraps a Lua visualizer plugin, caching function references
|
|
// for render() and optional init()/destroy() callbacks.
|
|
type luaVis struct {
|
|
name string
|
|
plugin *Plugin // owns the LState and mutex
|
|
obj *lua.LTable
|
|
render *lua.LFunction
|
|
init *lua.LFunction
|
|
destroy *lua.LFunction
|
|
last string // previous frame output (reused on error)
|
|
}
|
|
|
|
// registerVisPlugin is called during plugin.register() for type="visualizer".
|
|
func (m *Manager) registerVisPlugin(L *lua.LState, obj *lua.LTable, p *Plugin) {
|
|
vis := &luaVis{
|
|
name: p.Name,
|
|
plugin: p,
|
|
obj: obj,
|
|
}
|
|
|
|
m.mu.Lock()
|
|
m.visPlugs = append(m.visPlugs, vis)
|
|
m.visMap[p.Name] = vis
|
|
m.mu.Unlock()
|
|
}
|
|
|
|
// finalizeVisualizers is called after all plugins are loaded to resolve
|
|
// render/init/destroy function references from the plugin objects.
|
|
func (m *Manager) finalizeVisualizers() {
|
|
for _, vis := range m.visPlugs {
|
|
if fn, ok := vis.obj.RawGetString("render").(*lua.LFunction); ok {
|
|
vis.render = fn
|
|
}
|
|
if fn, ok := vis.obj.RawGetString("init").(*lua.LFunction); ok {
|
|
vis.init = fn
|
|
}
|
|
if fn, ok := vis.obj.RawGetString("destroy").(*lua.LFunction); ok {
|
|
vis.destroy = fn
|
|
}
|
|
}
|
|
}
|
|
|
|
// Visualizers returns the names of all Lua visualizer plugins.
|
|
func (m *Manager) Visualizers() []string {
|
|
m.mu.RLock()
|
|
defer m.mu.RUnlock()
|
|
names := make([]string, len(m.visPlugs))
|
|
for i, v := range m.visPlugs {
|
|
names[i] = v.name
|
|
}
|
|
return names
|
|
}
|
|
|
|
// InitVis calls a Lua visualizer's init(rows, cols) if it exists.
|
|
func (m *Manager) InitVis(name string, rows, cols int) {
|
|
m.mu.RLock()
|
|
vis, ok := m.visMap[name]
|
|
m.mu.RUnlock()
|
|
if !ok || vis.init == nil {
|
|
return
|
|
}
|
|
|
|
vis.plugin.mu.Lock()
|
|
defer vis.plugin.mu.Unlock()
|
|
|
|
_ = vis.plugin.callBounded(0, vis.init, vis.obj, lua.LNumber(rows), lua.LNumber(cols))
|
|
}
|
|
|
|
// DestroyVis calls a Lua visualizer's destroy() if it exists.
|
|
func (m *Manager) DestroyVis(name string) {
|
|
m.mu.RLock()
|
|
vis, ok := m.visMap[name]
|
|
m.mu.RUnlock()
|
|
if !ok || vis.destroy == nil {
|
|
return
|
|
}
|
|
|
|
vis.plugin.mu.Lock()
|
|
defer vis.plugin.mu.Unlock()
|
|
|
|
_ = vis.plugin.callBounded(0, vis.destroy, vis.obj)
|
|
}
|
|
|
|
// RenderVis calls a Lua visualizer's render(bands, frame) and returns
|
|
// the terminal text. On error, the previous frame is reused.
|
|
func (m *Manager) RenderVis(name string, bands [10]float64, rows, cols int, frame uint64) string {
|
|
m.mu.RLock()
|
|
vis, ok := m.visMap[name]
|
|
m.mu.RUnlock()
|
|
if !ok || vis.render == nil {
|
|
return ""
|
|
}
|
|
|
|
vis.plugin.mu.Lock()
|
|
defer vis.plugin.mu.Unlock()
|
|
|
|
L := vis.plugin.L
|
|
|
|
// Build bands table (1-indexed).
|
|
tbl := L.NewTable()
|
|
for i, b := range bands {
|
|
tbl.RawSetInt(i+1, lua.LNumber(b))
|
|
}
|
|
|
|
err := vis.plugin.callBounded(1, vis.render, vis.obj, tbl, lua.LNumber(frame), lua.LNumber(rows), lua.LNumber(cols))
|
|
if err != nil {
|
|
return vis.last
|
|
}
|
|
|
|
result := L.Get(-1)
|
|
L.Pop(1)
|
|
|
|
if str, ok := result.(lua.LString); ok {
|
|
vis.last = string(str)
|
|
return vis.last
|
|
}
|
|
return vis.last
|
|
}
|