144 lines
3.3 KiB
Go
144 lines
3.3 KiB
Go
package keys
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/99designs/keyring"
|
|
"github.com/spf13/viper"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
"gopkg.in/yaml.v2"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
"github.com/cosmos/cosmos-sdk/crypto/keys"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// available output formats.
|
|
const (
|
|
OutputFormatText = "text"
|
|
OutputFormatJSON = "json"
|
|
|
|
// defaultKeyDBName is the client's subdirectory where keys are stored.
|
|
defaultKeyDBName = "keys"
|
|
)
|
|
|
|
type bechKeyOutFn func(keyInfo keys.Info) (keys.KeyOutput, error)
|
|
|
|
// NewKeyBaseFromHomeFlag initializes a Keybase based on the configuration.
|
|
func NewKeyBaseFromHomeFlag() (keys.Keybase, error) {
|
|
rootDir := viper.GetString(flags.FlagHome)
|
|
return NewKeyBaseFromDir(rootDir)
|
|
}
|
|
|
|
// NewKeyBaseFromDir initializes a keybase at a particular dir.
|
|
func NewKeyBaseFromDir(rootDir string) (keys.Keybase, error) {
|
|
return getLazyKeyBaseFromDir(rootDir)
|
|
}
|
|
|
|
// NewInMemoryKeyBase returns a storage-less keybase.
|
|
func NewInMemoryKeyBase() keys.Keybase { return keys.NewInMemory() }
|
|
|
|
// NewKeyBaseFromHomeFlag initializes a keyring based on configuration.
|
|
func NewKeyringFromHomeFlag(input io.Reader) (keys.Keybase, error) {
|
|
return NewKeyringFromDir(viper.GetString(flags.FlagHome), input)
|
|
}
|
|
|
|
// NewKeyBaseFromDir initializes a keybase at a particular dir.
|
|
func NewKeyringFromDir(rootDir string, input io.Reader) (keys.Keybase, error) {
|
|
if os.Getenv("COSMOS_SDK_TEST_KEYRING") != "" {
|
|
return keys.NewTestKeyring(sdk.GetConfig().GetKeyringServiceName(), rootDir)
|
|
}
|
|
return keys.NewKeyring(sdk.GetConfig().GetKeyringServiceName(), rootDir, input)
|
|
}
|
|
|
|
func getLazyKeyBaseFromDir(rootDir string) (keys.Keybase, error) {
|
|
return keys.New(defaultKeyDBName, filepath.Join(rootDir, "keys")), nil
|
|
}
|
|
|
|
func printKeyInfo(keyInfo keys.Info, bechKeyOut bechKeyOutFn) {
|
|
ko, err := bechKeyOut(keyInfo)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
switch viper.Get(cli.OutputFlag) {
|
|
case OutputFormatText:
|
|
printTextInfos([]keys.KeyOutput{ko})
|
|
|
|
case OutputFormatJSON:
|
|
var out []byte
|
|
var err error
|
|
if viper.GetBool(flags.FlagIndentResponse) {
|
|
out, err = cdc.MarshalJSONIndent(ko, "", " ")
|
|
} else {
|
|
out, err = cdc.MarshalJSON(ko)
|
|
}
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
fmt.Println(string(out))
|
|
}
|
|
}
|
|
|
|
func printInfos(infos []keys.Info) {
|
|
kos, err := keys.Bech32KeysOutput(infos)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
switch viper.Get(cli.OutputFlag) {
|
|
case OutputFormatText:
|
|
printTextInfos(kos)
|
|
|
|
case OutputFormatJSON:
|
|
var out []byte
|
|
var err error
|
|
|
|
if viper.GetBool(flags.FlagIndentResponse) {
|
|
out, err = cdc.MarshalJSONIndent(kos, "", " ")
|
|
} else {
|
|
out, err = cdc.MarshalJSON(kos)
|
|
}
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Printf("%s", out)
|
|
}
|
|
}
|
|
|
|
func printTextInfos(kos []keys.KeyOutput) {
|
|
out, err := yaml.Marshal(&kos)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Println(string(out))
|
|
}
|
|
|
|
func printKeyAddress(info keys.Info, bechKeyOut bechKeyOutFn) {
|
|
ko, err := bechKeyOut(info)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
fmt.Println(ko.Address)
|
|
}
|
|
|
|
func printPubKey(info keys.Info, bechKeyOut bechKeyOutFn) {
|
|
ko, err := bechKeyOut(info)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
fmt.Println(ko.PubKey)
|
|
}
|
|
|
|
func isRunningUnattended() bool {
|
|
backends := keyring.AvailableBackends()
|
|
return len(backends) == 2 && backends[1] == keyring.BackendType("file")
|
|
}
|