2018-02-14 12:53:06 -08:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
2018-10-19 11:00:27 -07:00
|
|
|
"errors"
|
2018-03-28 08:03:17 -07:00
|
|
|
"fmt"
|
2018-10-10 15:45:41 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/crypto/keys"
|
2018-06-28 17:54:47 -07:00
|
|
|
"github.com/tendermint/tendermint/crypto"
|
2018-07-02 13:34:06 -07:00
|
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
2018-10-19 11:00:27 -07:00
|
|
|
"github.com/tendermint/tendermint/types"
|
2018-04-25 21:27:40 -07:00
|
|
|
|
|
|
|
clkeys "github.com/cosmos/cosmos-sdk/client/keys"
|
2018-09-13 11:17:32 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
2018-04-25 21:27:40 -07:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2018-10-19 11:00:27 -07:00
|
|
|
tmtypes "github.com/tendermint/tendermint/types"
|
2018-02-14 12:53:06 -08:00
|
|
|
)
|
|
|
|
|
2018-04-23 17:05:58 -07:00
|
|
|
// Core functionality passed from the application to the server init command
|
|
|
|
type AppInit struct {
|
2018-04-25 21:49:53 -07:00
|
|
|
// AppGenState creates the core parameters initialization. It takes in a
|
|
|
|
// pubkey meant to represent the pubkey of the validator of this machine.
|
2018-10-19 11:00:27 -07:00
|
|
|
AppGenState func(cdc *codec.Codec, appGenTx []json.RawMessage) (appState json.RawMessage, err error)
|
|
|
|
}
|
|
|
|
|
|
|
|
// SimpleGenTx is a simple genesis tx
|
|
|
|
type SimpleGenTx struct {
|
|
|
|
Addr sdk.AccAddress `json:"addr"`
|
2018-04-23 17:05:58 -07:00
|
|
|
}
|
|
|
|
|
2018-04-25 13:12:14 -07:00
|
|
|
//_____________________________________________________________________
|
|
|
|
|
2018-04-23 17:05:58 -07:00
|
|
|
// simple default application init
|
|
|
|
var DefaultAppInit = AppInit{
|
2018-04-25 21:49:53 -07:00
|
|
|
AppGenState: SimpleAppGenState,
|
2018-04-23 17:05:58 -07:00
|
|
|
}
|
2018-04-22 10:36:35 -07:00
|
|
|
|
2018-04-25 13:12:14 -07:00
|
|
|
// Generate a genesis transaction
|
2018-10-19 11:00:27 -07:00
|
|
|
func SimpleAppGenTx(cdc *codec.Codec, pk crypto.PubKey) (appGenTx, cliPrint json.RawMessage, validator types.GenesisValidator, err error) {
|
2018-07-06 00:06:53 -07:00
|
|
|
var addr sdk.AccAddress
|
2018-04-25 13:12:14 -07:00
|
|
|
var secret string
|
2018-07-06 00:06:53 -07:00
|
|
|
addr, secret, err = GenerateCoinKey()
|
2018-04-25 13:12:14 -07:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
var bz []byte
|
2018-10-19 11:00:27 -07:00
|
|
|
simpleGenTx := SimpleGenTx{Addr: addr}
|
2018-04-25 13:12:14 -07:00
|
|
|
bz, err = cdc.MarshalJSON(simpleGenTx)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
appGenTx = json.RawMessage(bz)
|
|
|
|
mm := map[string]string{"secret": secret}
|
|
|
|
bz, err = cdc.MarshalJSON(mm)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
cliPrint = json.RawMessage(bz)
|
|
|
|
validator = tmtypes.GenesisValidator{
|
|
|
|
PubKey: pk,
|
|
|
|
Power: 10,
|
|
|
|
}
|
2018-04-20 16:55:22 -07:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-04-25 21:49:53 -07:00
|
|
|
// create the genesis app state
|
2018-09-13 11:17:32 -07:00
|
|
|
func SimpleAppGenState(cdc *codec.Codec, appGenTxs []json.RawMessage) (appState json.RawMessage, err error) {
|
2018-04-25 21:49:53 -07:00
|
|
|
|
|
|
|
if len(appGenTxs) != 1 {
|
|
|
|
err = errors.New("must provide a single genesis transaction")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-10-19 11:00:27 -07:00
|
|
|
var tx SimpleGenTx
|
|
|
|
err = cdc.UnmarshalJSON(appGenTxs[0], &tx)
|
2018-04-25 21:49:53 -07:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
appState = json.RawMessage(fmt.Sprintf(`{
|
|
|
|
"accounts": [{
|
|
|
|
"address": "%s",
|
|
|
|
"coins": [
|
|
|
|
{
|
|
|
|
"denom": "mycoin",
|
2018-07-02 14:29:47 -07:00
|
|
|
"amount": "9007199254740992"
|
2018-04-25 21:49:53 -07:00
|
|
|
}
|
|
|
|
]
|
|
|
|
}]
|
2018-10-19 11:00:27 -07:00
|
|
|
}`, tx.Addr))
|
2018-04-25 21:49:53 -07:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
//___________________________________________________________________________________________
|
|
|
|
|
2018-04-20 14:42:56 -07:00
|
|
|
// GenerateCoinKey returns the address of a public key, along with the secret
|
|
|
|
// phrase to recover the private key.
|
2018-07-06 00:06:53 -07:00
|
|
|
func GenerateCoinKey() (sdk.AccAddress, string, error) {
|
2018-04-20 16:55:22 -07:00
|
|
|
|
2018-04-05 03:31:33 -07:00
|
|
|
// construct an in-memory key store
|
|
|
|
keybase := keys.New(
|
|
|
|
dbm.NewMemDB(),
|
|
|
|
)
|
|
|
|
|
|
|
|
// generate a private key, with recovery phrase
|
2018-06-28 17:54:47 -07:00
|
|
|
info, secret, err := keybase.CreateMnemonic("name", keys.English, "pass", keys.Secp256k1)
|
2018-04-05 03:31:33 -07:00
|
|
|
if err != nil {
|
2018-07-09 01:47:38 -07:00
|
|
|
return sdk.AccAddress([]byte{}), "", err
|
2018-04-05 03:31:33 -07:00
|
|
|
}
|
2018-06-28 17:54:47 -07:00
|
|
|
addr := info.GetPubKey().Address()
|
2018-07-06 00:06:53 -07:00
|
|
|
return sdk.AccAddress(addr), secret, nil
|
2018-04-05 03:31:33 -07:00
|
|
|
}
|
2018-04-25 21:27:40 -07:00
|
|
|
|
|
|
|
// GenerateSaveCoinKey returns the address of a public key, along with the secret
|
|
|
|
// phrase to recover the private key.
|
2018-07-06 00:06:53 -07:00
|
|
|
func GenerateSaveCoinKey(clientRoot, keyName, keyPass string, overwrite bool) (sdk.AccAddress, string, error) {
|
2018-04-25 21:27:40 -07:00
|
|
|
|
|
|
|
// get the keystore from the client
|
|
|
|
keybase, err := clkeys.GetKeyBaseFromDir(clientRoot)
|
|
|
|
if err != nil {
|
2018-07-09 01:47:38 -07:00
|
|
|
return sdk.AccAddress([]byte{}), "", err
|
2018-04-25 21:27:40 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// ensure no overwrite
|
|
|
|
if !overwrite {
|
|
|
|
_, err := keybase.Get(keyName)
|
|
|
|
if err == nil {
|
2018-10-19 11:00:27 -07:00
|
|
|
return sdk.AccAddress([]byte{}), "", fmt.Errorf(
|
|
|
|
"key already exists, overwrite is disabled (clientRoot: %s)", clientRoot)
|
2018-04-25 21:27:40 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// generate a private key, with recovery phrase
|
2018-06-28 17:54:47 -07:00
|
|
|
info, secret, err := keybase.CreateMnemonic(keyName, keys.English, keyPass, keys.Secp256k1)
|
2018-04-25 21:27:40 -07:00
|
|
|
if err != nil {
|
2018-07-09 01:47:38 -07:00
|
|
|
return sdk.AccAddress([]byte{}), "", err
|
2018-04-25 21:27:40 -07:00
|
|
|
}
|
2018-06-28 17:54:47 -07:00
|
|
|
addr := info.GetPubKey().Address()
|
2018-07-06 00:06:53 -07:00
|
|
|
return sdk.AccAddress(addr), secret, nil
|
2018-04-25 21:27:40 -07:00
|
|
|
}
|