yanic/output/meshviewer/output.go

package meshviewer

import (
	"fmt"

	"github.com/bdlm/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) {
	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)
	}
}