planetmint-go/lib/config.go

package lib

import (
	"os"
	"sync"

	"github.com/cosmos/cosmos-sdk/client"
	"github.com/cosmos/cosmos-sdk/crypto/keyring"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/planetmint/planetmint-go/lib/params"
)

// Config defines library top level configuration.
type Config struct {
	chainID        string
	clientCtx      client.Context
	encodingConfig params.EncodingConfig
	feeDenom       string
	rootDir        string
	rpcEndpoint    string
	txGas          uint64
	serialPort     string
}

// lib wide global singleton
var (
	libConfig  *Config
	sdkConfig  *sdk.Config
	initConfig sync.Once
	changeLock sync.Mutex
)

// DefaultConfig returns library default configuration.
func DefaultConfig() *Config {
	return &Config{
		chainID:        "planetmint-testnet-1",
		clientCtx:      client.Context{},
		encodingConfig: params.EncodingConfig{},
		feeDenom:       "plmnt",
		rootDir:        "~/.planetmint-go/",
		rpcEndpoint:    "http://127.0.0.1:26657",
		txGas:          200000,
		serialPort:     "",
	}
}

// GetConfig returns the config instance for the SDK.
func GetConfig() *Config {
	initConfig.Do(func() {
		libConfig = DefaultConfig()
		sdkConfig = sdk.GetConfig()
		libConfig.SetBech32PrefixForAccount("plmnt")

		encodingConfig := MakeEncodingConfig()
		libConfig.SetEncodingConfig(encodingConfig)
	})
	return libConfig
}

// SetBech32PrefixForAccount sets the bech32 account prefix.
func (config *Config) SetBech32PrefixForAccount(bech32Prefix string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	sdkConfig.SetBech32PrefixForAccount(bech32Prefix, "pub")
	return config
}

// SetEncodingConfig sets the encoding config and must not be nil.
func (config *Config) SetEncodingConfig(encodingConfig params.EncodingConfig) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.encodingConfig = encodingConfig
	return config
}

// SetChainID sets the chain ID parameter.
func (config *Config) SetChainID(chainID string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.chainID = chainID
	return config
}

// SetClientCtx sets the client context parameter.
func (config *Config) SetClientCtx(clientCtx client.Context) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.clientCtx = clientCtx
	return config
}

// SetFeeDenom sets the fee denominator parameter.
func (config *Config) SetFeeDenom(feeDenom string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.feeDenom = feeDenom
	return config
}

// SetRoot sets the root directory where to find the keyring.
func (config *Config) SetRoot(root string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.rootDir = root
	return config
}

// SetRPCEndpoint sets the RPC endpoint to send requests to.
func (config *Config) SetRPCEndpoint(rpcEndpoint string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.rpcEndpoint = rpcEndpoint
	return config
}

// SetTxGas sets the amount of Gas for the TX that is send to the network
func (config *Config) SetTxGas(txGas uint64) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.txGas = txGas
	return config
}

func (config *Config) SetSerialPort(port string) *Config {
	changeLock.Lock()
	defer changeLock.Unlock()
	config.serialPort = port
	return config
}

func (config *Config) GetSerialPort() string {
	return config.serialPort
}

func (config *Config) getLibKeyring() (keyring.Keyring, error) {
	return keyring.New("lib", keyring.BackendTest, config.rootDir, os.Stdin, config.encodingConfig.Marshaler, []keyring.Option{}...)
}

func (config *Config) GetDefaultValidatorRecord() (*keyring.Record, error) {
	keyring, err := config.getLibKeyring()
	if err != nil {
		return nil, err
	}

	records, err := keyring.List()
	if err != nil {
		return nil, err
	}

	return records[0], nil
}