89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
|
package meshviewer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/yanic/output"
|
||
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Output struct {
|
||
|
|
output.Output
|
||
|
|
config Config
|
||
|
|
nodes *runtime.Nodes
|
||
|
|
builder nodeBuilder
|
||
|
|
filter filter
|
||
|
|
}
|
||
|
|
|
||
|
|
type Config map[string]interface{}
|
||
|
|
|
||
|
|
func (c Config) Enable() bool {
|
||
|
|
return c["enable"].(bool)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c Config) Version() int64 {
|
||
|
|
return c["version"].(int64)
|
||
|
|
}
|
||
|
|
func (c Config) NodesPath() string {
|
||
|
|
if c["nodes_path"] == nil {
|
||
|
|
log.Panic("in configuration of [[nodes.output.meshviewer]] was no nodes_path defined", c)
|
||
|
|
}
|
||
|
|
return c["nodes_path"].(string)
|
||
|
|
}
|
||
|
|
func (c Config) GraphPath() string {
|
||
|
|
return c["graph_path"].(string)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c Config) FilterOption() filterConfig {
|
||
|
|
if v, ok := c["filter"]; ok {
|
||
|
|
var filterMap filterConfig
|
||
|
|
filterMap = v.(map[string]interface{})
|
||
|
|
return filterMap
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type nodeBuilder func(filter, *runtime.Nodes) interface{}
|
||
|
|
|
||
|
|
var nodeFormats = map[int64]nodeBuilder{
|
||
|
|
1: BuildNodesV1,
|
||
|
|
2: BuildNodesV2,
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
output.RegisterAdapter("meshviewer", Register)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Register(nodes *runtime.Nodes, configuration interface{}) (output.Output, error) {
|
||
|
|
var config Config
|
||
|
|
config = configuration.(map[string]interface{})
|
||
|
|
if !config.Enable() {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
builder := nodeFormats[config.Version()]
|
||
|
|
if builder == nil {
|
||
|
|
log.Panicf("invalid nodes version: %d", config.Version())
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Output{
|
||
|
|
nodes: nodes,
|
||
|
|
config: config,
|
||
|
|
builder: builder,
|
||
|
|
filter: createFilter(config.FilterOption()),
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *Output) Save() {
|
||
|
|
o.nodes.RLock()
|
||
|
|
defer o.nodes.RUnlock()
|
||
|
|
|
||
|
|
if path := o.config.NodesPath(); path != "" {
|
||
|
|
runtime.SaveJSON(o.builder(o.filter, o.nodes), path)
|
||
|
|
}
|
||
|
|
|
||
|
|
if path := o.config.GraphPath(); path != "" {
|
||
|
|
runtime.SaveJSON(BuildGraph(o.nodes), path)
|
||
|
|
}
|
||
|
|
}
|