configyaml.go

package main

import (
	"io/ioutil"
	"os"

	log "github.com/sirupsen/logrus"
	yaml "gopkg.in/yaml.v2"
)

var (
	qubeTemplateYaml = `
    genesis:
      # supported: (raft | istanbul | qbft)
      consensus: istanbul
      Test_QBFT_Block: 0
      Quorum_Version: 21.7.1
      Tm_Version: 21.7.2
      Chain_Id: 1000
    nodes:
    `
)

type QuorumEntry struct {
	Quorum Quorum
	Tm     Tm
}

type GethEntry struct {
	GetStartupParams string `yaml:"Geth_Startup_Params"` //--raftjoinexisting 7
}

type Quorum struct {
	Consensus      string `yaml:"consensus"`
	QuorumVersion  string `yaml:"Quorum_Version"`
	DockerRepoFull string `yaml:"Docker_Repo_Full,omitempty"` //quorum-local
}

type Tm struct {
	Name           string `yaml:"Name"`
	TmVersion      string `yaml:"Tm_Version"`
	DockerRepoFull string `yaml:"Docker_Repo_Full,omitempty"` //tessera-local
}

type NodeEntry struct {
	NodeUserIdent string      `yaml:"Node_UserIdent"`
	KeyDir        string      `yaml:"Key_Dir"`
	QuorumEntry   QuorumEntry `yaml:"quorum"`
	GethEntry     GethEntry   `yaml:"geth"`
}

type ExternalNodeEntry struct {
	NodeUserIdent string `yaml:"Node_UserIdent"`
	EnodeUrl      string `yaml:"Enode_Url"`
	TmUrl         string `yaml:"Tm_Url"`
	// must be set in the yaml without quotes.
	// The hex number will be evaluted to a BigNum and
	// template/istanbul-validator.toml.erb will convert back to hex
	// https://github.com/mikefarah/yq/issues/19
	NodekeyAddress string `yaml:"Node_Acct_Addr,omitempty"`
}

type Prometheus struct {
	//#monitor_params_geth: --metrics --metrics.expensive --pprof --pprofaddr=0.0.0.0
	//monitorParamsGeth string `yaml:"monitor_params_geth"`
	NodePort string `yaml:"nodePort_prom,omitempty"`
	Enabled  bool   `yaml:"enabled,omitempty"`
}

type Cakeshop struct {
	Version string `yaml:"version,omitempty"`
	Service struct {
		Type     string `yaml:"type,omitempty"`
		NodePort string `yaml:"nodePort,omitempty"`
	}
}

type Ingress struct {
	//OneToMany | OneToOne
	Strategy string `yaml:"Strategy,omitempty"`
	Host     string `yaml:"Host,omitempty"`
}

type K8s struct {
	Service struct {
		Type    string  `yaml:"type,omitempty"`
		Ingress Ingress `yaml:"Ingress,omitempty"`
	}
}

type QConfig struct {
	Genesis struct {
		Consensus     string `yaml:"consensus"`
		TestQBFTBlock string `yaml:"Test_QBFT_Block,omitempty"`
		QuorumVersion string `yaml:"Quorum_Version"`
		TmVersion     string `yaml:"Tm_Version"`
		Chain_Id      string `yaml:"Chain_Id"`
	}
	Prometheus Prometheus `yaml:"prometheus,omitempty"`
	Cakeshop   Cakeshop   `yaml:"cakeshop,omitempty"`
	K8s        K8s        `yaml:"k8s,omitempty"`

	Nodes         []NodeEntry
	ExternalNodes []ExternalNodeEntry `yaml:"external_nodes,omitempty"`
}

func GetYamlConfig() QConfig {
	config := QConfig{}
	err := yaml.Unmarshal([]byte(qubeTemplateYaml), &config)
	if err != nil {
		log.Fatalf("error: %v", err)
	}
	return config
}

func LoadYamlConfig(filename string) (QConfig, error) {
	config := QConfig{}
	fileBytes, err := ioutil.ReadFile(filename)
	if err != nil {
		log.Fatalf("error reading file: %v err: %v", filename, err)
		return config, err
	}

	err = yaml.Unmarshal(fileBytes, &config)
	if err != nil {
		log.Fatalf("error unmarshalling file: %v err: %v", filename, err)
		return config, err
	}
	return config, nil
}

func WriteYamlConfig(qconfig QConfig, filename string) (QConfig, error) {
	bs, err := yaml.Marshal(&qconfig)
	if err != nil {
		log.Fatalf("error: %v", err)
		return qconfig, err
	}
	err = ioutil.WriteFile(filename, bs, os.ModePerm)
	return qconfig, err
}

func (q QConfig) ToString() string {
	d, err := yaml.Marshal(&q)
	if err != nil {
		log.Fatalf("error: %v", err)
	}
	return string(d)
}