70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"text/template"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const defaultConfigTemplate = `# This is a TOML config file.
|
|
# For more information, see https://github.com/toml-lang/toml
|
|
|
|
###############################################################################
|
|
### Client Configuration ###
|
|
###############################################################################
|
|
|
|
# The network chain ID
|
|
chain-id = "{{ .ChainID }}"
|
|
# The keyring's backend, where the keys are stored (os|file|kwallet|pass|test|memory)
|
|
keyring-backend = "{{ .KeyringBackend }}"
|
|
# CLI output format (text|json)
|
|
output = "{{ .Output }}"
|
|
# <host>:<port> to Tendermint RPC interface for this chain
|
|
node = "{{ .Node }}"
|
|
# Transaction broadcasting mode (sync|async|block)
|
|
broadcast-mode = "{{ .BroadcastMode }}"
|
|
`
|
|
|
|
// writeConfigToFile parses defaultConfigTemplate, renders config using the template and writes it to
|
|
// configFilePath.
|
|
func writeConfigToFile(configFilePath string, config *ClientConfig) error {
|
|
var buffer bytes.Buffer
|
|
|
|
tmpl := template.New("clientConfigFileTemplate")
|
|
configTemplate, err := tmpl.Parse(defaultConfigTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := configTemplate.Execute(&buffer, config); err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.WriteFile(configFilePath, buffer.Bytes(), 0600)
|
|
}
|
|
|
|
// ensureConfigPath creates a directory configPath if it does not exist
|
|
func ensureConfigPath(configPath string) error {
|
|
return os.MkdirAll(configPath, os.ModePerm)
|
|
}
|
|
|
|
// getClientConfig reads values from client.toml file and unmarshalls them into ClientConfig
|
|
func getClientConfig(configPath string, v *viper.Viper) (*ClientConfig, error) {
|
|
v.AddConfigPath(configPath)
|
|
v.SetConfigName("client")
|
|
v.SetConfigType("toml")
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conf := new(ClientConfig)
|
|
if err := v.Unmarshal(conf); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return conf, nil
|
|
}
|