server/profile/node.go

package mhayaProfile

import (
	"fmt"

	cerr "github.com/mhaya/error"
	cfacade "github.com/mhaya/facade"
)

// Node node info
type Node struct {
	nodeId     string
	nodeType   string
	address    string
	rpcAddress string
	settings   cfacade.ProfileJSON
	enabled    bool
}

func (n *Node) NodeId() string {
	return n.nodeId
}

func (n *Node) NodeType() string {
	return n.nodeType
}

func (n *Node) Address() string {
	return n.address
}

func (n *Node) RpcAddress() string {
	return n.rpcAddress
}

func (n *Node) Settings() cfacade.ProfileJSON {
	return n.settings
}

func (n *Node) Enabled() bool {
	return n.enabled
}

const stringFormat = "nodeId = %s, nodeType = %s, address = %s, rpcAddress = %s, enabled = %v"

func (n *Node) String() string {
	return fmt.Sprintf(stringFormat,
		n.nodeId,
		n.nodeType,
		n.address,
		n.rpcAddress,
		n.enabled,
	)
}

func GetNodeWithConfig(config *Config, nodeId string) (cfacade.INode, error) {
	nodeConfig := config.GetConfig("node")
	if nodeConfig.LastError() != nil {
		return nil, cerr.Error("`nodes` property not found in profile file.")
	}

	for _, nodeType := range nodeConfig.Keys() {
		typeJson := nodeConfig.GetConfig(nodeType)
		for i := 0; i < typeJson.Size(); i++ {
			item := typeJson.GetConfig(i)

			if !findNodeId(nodeId, item.GetConfig("node_id")) {
				continue
			}

			node := &Node{
				nodeId:     nodeId,
				nodeType:   nodeType,
				address:    item.GetString("address"),
				rpcAddress: item.GetString("rpc_address"),
				settings:   item.GetConfig("__settings__"),
				enabled:    item.GetBool("enabled"),
			}

			return node, nil
		}
	}

	return nil, cerr.Errorf("nodeId = %s not found.", nodeId)
}

func LoadNode(nodeId string) (cfacade.INode, error) {
	return GetNodeWithConfig(cfg.jsonConfig, nodeId)
}

func findNodeId(nodeId string, nodeIdJson cfacade.ProfileJSON) bool {
	if nodeIdJson.ToString() == nodeId {
		return true
	}

	for i := 0; i < nodeIdJson.Size(); i++ {
		if nodeIdJson.GetString(i) == nodeId {
			return true
		}
	}

	return false
}