config.go

package main

import (
	"path/filepath"
	"strings"

	"github.com/spf13/viper"
)

const (
	ConfigName = "fabric.toml"
)

type Config struct {
	Fabric   Fabric    `toml:"fabric" json:"fabric"`
	Services []Service `mapstructure:"services" json:"services"`
}
type Fabric struct {
	Name          string `toml:"name" json:"name"`
	Username      string `toml:"username" json:"username"`
	CCID          string `toml:"ccid" json:"ccid"`
	ChannelId     string `mapstructure:"channel_id" toml:"channel_id" json:"channel_id"`
	Org           string `toml:"org" json:"org"`
	ServerPort    string `toml:"server_port" json:"server_port"`
	TimeoutHeight int64  `mapstructure:"timeout_height" json:"timeout_height"`
	TimeoutPeriod uint64 `mapstructure:"timeout_period" json:"timeout_period"`
}

type Service struct {
	ID   string `toml:"id" json:"id"`
	Name string `toml:"name" json:"name"`
	Type string `toml:"type" json:"type"`
}

func DefaultConfig() *Config {
	return &Config{
		Fabric: Fabric{
			Name:          "fabric",
			Username:      "Admin",
			CCID:          "broker",
			ChannelId:     "mychannel",
			Org:           "org2",
			TimeoutHeight: 30,
			TimeoutPeriod: 60,
		},
		Services: nil,
	}
}

func UnmarshalConfig(configPath string) (*Config, error) {
	viper.SetConfigFile(filepath.Join(configPath, ConfigName))
	viper.SetConfigType("toml")
	viper.AutomaticEnv()
	viper.SetEnvPrefix("FABRIC")
	replacer := strings.NewReplacer(".", "_")
	viper.SetEnvKeyReplacer(replacer)
	if err := viper.ReadInConfig(); err != nil {
		return nil, err
	}

	config := DefaultConfig()

	if err := viper.Unmarshal(config); err != nil {
		return nil, err
	}

	return config, nil
}