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