72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package commands
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/tendermint/go-crypto"
|
|
"github.com/tendermint/tmlibs/cli"
|
|
)
|
|
|
|
//---------------------------------------------
|
|
// simple implementation of a key
|
|
|
|
// Address - public address for a key
|
|
type Address [20]byte
|
|
|
|
// MarshalJSON - marshal the json bytes of the address
|
|
func (a Address) MarshalJSON() ([]byte, error) {
|
|
return []byte(fmt.Sprintf(`"%x"`, a[:])), nil
|
|
}
|
|
|
|
// UnmarshalJSON - unmarshal the json bytes of the address
|
|
func (a *Address) UnmarshalJSON(addrHex []byte) error {
|
|
addr, err := hex.DecodeString(strings.Trim(string(addrHex), `"`))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
copy(a[:], addr)
|
|
return nil
|
|
}
|
|
|
|
// Key - full private key
|
|
type Key struct {
|
|
Address Address `json:"address"`
|
|
PubKey crypto.PubKey `json:"pub_key"`
|
|
PrivKey crypto.PrivKey `json:"priv_key"`
|
|
}
|
|
|
|
// Sign - Implements Signer
|
|
func (k *Key) Sign(msg []byte) crypto.Signature {
|
|
return k.PrivKey.Sign(msg)
|
|
}
|
|
|
|
// LoadKey - load key from json file
|
|
func LoadKey(keyFile string) (*Key, error) {
|
|
filePath := keyFile
|
|
|
|
if !strings.HasPrefix(keyFile, "/") && !strings.HasPrefix(keyFile, ".") {
|
|
rootDir := viper.GetString(cli.HomeFlag)
|
|
filePath = path.Join(rootDir, keyFile)
|
|
}
|
|
|
|
keyJSONBytes, err := ioutil.ReadFile(filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
key := new(Key)
|
|
err = json.Unmarshal(keyJSONBytes, key)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Error reading key from %v: %v", filePath, err) //never stack trace
|
|
}
|
|
|
|
return key, nil
|
|
}
|