2018-02-22 07:17:19 -08:00
|
|
|
package keys
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2019-11-14 06:17:21 -08:00
|
|
|
"io"
|
2019-06-22 02:24:59 -07:00
|
|
|
"os"
|
2018-03-28 08:52:18 -07:00
|
|
|
"path/filepath"
|
2018-02-22 07:17:19 -08:00
|
|
|
|
2019-11-14 06:17:21 -08:00
|
|
|
"github.com/99designs/keyring"
|
2018-11-16 14:21:36 -08:00
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/tendermint/tendermint/libs/cli"
|
2019-11-14 06:17:21 -08:00
|
|
|
"gopkg.in/yaml.v2"
|
2018-12-10 06:27:25 -08:00
|
|
|
|
2019-05-28 01:44:04 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
2018-12-10 06:27:25 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/crypto/keys"
|
2019-11-14 06:17:21 -08:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2018-02-22 07:17:19 -08:00
|
|
|
)
|
|
|
|
|
2019-02-11 17:18:58 -08:00
|
|
|
// available output formats.
|
2019-02-08 12:45:23 -08:00
|
|
|
const (
|
|
|
|
OutputFormatText = "text"
|
|
|
|
OutputFormatJSON = "json"
|
2019-02-11 17:18:58 -08:00
|
|
|
|
|
|
|
// defaultKeyDBName is the client's subdirectory where keys are stored.
|
|
|
|
defaultKeyDBName = "keys"
|
2019-02-08 12:45:23 -08:00
|
|
|
)
|
2018-02-28 06:36:04 -08:00
|
|
|
|
2019-03-01 13:29:33 -08:00
|
|
|
type bechKeyOutFn func(keyInfo keys.Info) (keys.KeyOutput, error)
|
2018-08-30 21:06:44 -07:00
|
|
|
|
2019-02-06 11:23:49 -08:00
|
|
|
// NewKeyBaseFromHomeFlag initializes a Keybase based on the configuration.
|
|
|
|
func NewKeyBaseFromHomeFlag() (keys.Keybase, error) {
|
2019-05-28 01:44:04 -07:00
|
|
|
rootDir := viper.GetString(flags.FlagHome)
|
2019-02-06 11:23:49 -08:00
|
|
|
return NewKeyBaseFromDir(rootDir)
|
2018-10-22 14:29:27 -07:00
|
|
|
}
|
|
|
|
|
2019-02-06 11:23:49 -08:00
|
|
|
// NewKeyBaseFromDir initializes a keybase at a particular dir.
|
|
|
|
func NewKeyBaseFromDir(rootDir string) (keys.Keybase, error) {
|
|
|
|
return getLazyKeyBaseFromDir(rootDir)
|
2018-02-22 07:17:19 -08:00
|
|
|
}
|
|
|
|
|
2019-02-07 14:38:45 -08:00
|
|
|
// NewInMemoryKeyBase returns a storage-less keybase.
|
|
|
|
func NewInMemoryKeyBase() keys.Keybase { return keys.NewInMemory() }
|
|
|
|
|
2019-11-14 06:17:21 -08:00
|
|
|
// NewKeyBaseFromHomeFlag initializes a keyring based on configuration.
|
|
|
|
func NewKeyringFromHomeFlag(input io.Reader) (keys.Keybase, error) {
|
|
|
|
return NewKeyringFromDir(viper.GetString(flags.FlagHome), input)
|
|
|
|
}
|
|
|
|
|
2019-11-30 06:32:08 -08:00
|
|
|
// NewKeyBaseFromDir initializes a keyring at a particular dir.
|
|
|
|
// If the COSMOS_SDK_TEST_KEYRING environment variable is set and not empty it will
|
|
|
|
// return an on-disk, password-less keyring that could be used for testing purposes.
|
2019-11-14 06:17:21 -08:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
|
2019-02-06 11:23:49 -08:00
|
|
|
func getLazyKeyBaseFromDir(rootDir string) (keys.Keybase, error) {
|
2019-02-11 17:18:58 -08:00
|
|
|
return keys.New(defaultKeyDBName, filepath.Join(rootDir, "keys")), nil
|
2018-03-05 08:41:50 -08:00
|
|
|
}
|
|
|
|
|
2018-08-30 21:06:44 -07:00
|
|
|
func printKeyInfo(keyInfo keys.Info, bechKeyOut bechKeyOutFn) {
|
|
|
|
ko, err := bechKeyOut(keyInfo)
|
2018-06-05 21:53:04 -07:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2018-08-30 21:06:44 -07:00
|
|
|
|
2018-02-22 07:17:19 -08:00
|
|
|
switch viper.Get(cli.OutputFlag) {
|
2019-02-08 12:45:23 -08:00
|
|
|
case OutputFormatText:
|
2019-05-17 07:15:45 -07:00
|
|
|
printTextInfos([]keys.KeyOutput{ko})
|
2019-03-01 13:29:33 -08:00
|
|
|
|
2019-03-09 13:00:08 -08:00
|
|
|
case OutputFormatJSON:
|
2019-03-13 10:36:52 -07:00
|
|
|
var out []byte
|
|
|
|
var err error
|
2019-05-28 01:44:04 -07:00
|
|
|
if viper.GetBool(flags.FlagIndentResponse) {
|
2019-03-13 10:36:52 -07:00
|
|
|
out, err = cdc.MarshalJSONIndent(ko, "", " ")
|
|
|
|
} else {
|
|
|
|
out, err = cdc.MarshalJSON(ko)
|
|
|
|
}
|
2018-02-22 07:17:19 -08:00
|
|
|
if err != nil {
|
2018-04-20 09:51:47 -07:00
|
|
|
panic(err)
|
2018-02-22 07:17:19 -08:00
|
|
|
}
|
2018-08-30 21:06:44 -07:00
|
|
|
|
2018-04-20 09:51:47 -07:00
|
|
|
fmt.Println(string(out))
|
2018-02-22 07:17:19 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func printInfos(infos []keys.Info) {
|
2019-03-01 13:29:33 -08:00
|
|
|
kos, err := keys.Bech32KeysOutput(infos)
|
2018-06-05 21:53:04 -07:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2019-03-01 13:29:33 -08:00
|
|
|
|
2018-02-22 07:17:19 -08:00
|
|
|
switch viper.Get(cli.OutputFlag) {
|
2019-02-08 12:45:23 -08:00
|
|
|
case OutputFormatText:
|
2019-05-17 07:15:45 -07:00
|
|
|
printTextInfos(kos)
|
2019-03-01 13:29:33 -08:00
|
|
|
|
2019-02-08 12:45:23 -08:00
|
|
|
case OutputFormatJSON:
|
2019-03-13 10:36:52 -07:00
|
|
|
var out []byte
|
|
|
|
var err error
|
|
|
|
|
2019-05-28 01:44:04 -07:00
|
|
|
if viper.GetBool(flags.FlagIndentResponse) {
|
2019-03-13 10:36:52 -07:00
|
|
|
out, err = cdc.MarshalJSONIndent(kos, "", " ")
|
|
|
|
} else {
|
|
|
|
out, err = cdc.MarshalJSON(kos)
|
|
|
|
}
|
|
|
|
|
2018-02-22 07:17:19 -08:00
|
|
|
if err != nil {
|
2018-04-20 09:51:47 -07:00
|
|
|
panic(err)
|
2018-02-22 07:17:19 -08:00
|
|
|
}
|
2019-05-31 06:14:35 -07:00
|
|
|
fmt.Printf("%s", out)
|
2018-02-22 07:17:19 -08:00
|
|
|
}
|
|
|
|
}
|
2018-05-28 16:27:34 -07:00
|
|
|
|
2019-05-17 07:15:45 -07:00
|
|
|
func printTextInfos(kos []keys.KeyOutput) {
|
2019-05-31 06:14:35 -07:00
|
|
|
out, err := yaml.Marshal(&kos)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
2019-03-01 13:29:33 -08:00
|
|
|
}
|
2019-05-31 06:14:35 -07:00
|
|
|
fmt.Println(string(out))
|
2019-03-01 13:29:33 -08:00
|
|
|
}
|
|
|
|
|
2018-08-30 21:06:44 -07:00
|
|
|
func printKeyAddress(info keys.Info, bechKeyOut bechKeyOutFn) {
|
|
|
|
ko, err := bechKeyOut(info)
|
2018-08-10 23:42:57 -07:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2018-08-30 21:06:44 -07:00
|
|
|
|
|
|
|
fmt.Println(ko.Address)
|
2018-08-10 23:42:57 -07:00
|
|
|
}
|
|
|
|
|
2018-08-30 21:06:44 -07:00
|
|
|
func printPubKey(info keys.Info, bechKeyOut bechKeyOutFn) {
|
|
|
|
ko, err := bechKeyOut(info)
|
2018-08-10 23:42:57 -07:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2018-08-30 21:06:44 -07:00
|
|
|
|
2018-08-10 23:42:57 -07:00
|
|
|
fmt.Println(ko.PubKey)
|
|
|
|
}
|
2019-11-14 06:17:21 -08:00
|
|
|
|
|
|
|
func isRunningUnattended() bool {
|
|
|
|
backends := keyring.AvailableBackends()
|
|
|
|
return len(backends) == 2 && backends[1] == keyring.BackendType("file")
|
|
|
|
}
|