2018-04-06 22:12:00 -07:00
|
|
|
package app
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
|
|
|
|
abci "github.com/tendermint/abci/types"
|
|
|
|
cmn "github.com/tendermint/tmlibs/common"
|
|
|
|
dbm "github.com/tendermint/tmlibs/db"
|
|
|
|
"github.com/tendermint/tmlibs/log"
|
|
|
|
|
|
|
|
bam "github.com/cosmos/cosmos-sdk/baseapp"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/bank"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/ibc"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/stake"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
appName = "GaiaApp"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Extended ABCI application
|
|
|
|
type GaiaApp struct {
|
|
|
|
*bam.BaseApp
|
|
|
|
cdc *wire.Codec
|
|
|
|
|
|
|
|
// keys to access the substores
|
|
|
|
capKeyMainStore *sdk.KVStoreKey
|
|
|
|
capKeyAccountStore *sdk.KVStoreKey
|
|
|
|
capKeyIBCStore *sdk.KVStoreKey
|
|
|
|
capKeyStakeStore *sdk.KVStoreKey
|
|
|
|
|
|
|
|
// Manage getting and setting accounts
|
|
|
|
accountMapper sdk.AccountMapper
|
2018-04-06 22:50:46 -07:00
|
|
|
coinKeeper bank.CoinKeeper
|
|
|
|
ibcMapper ibc.IBCMapper
|
|
|
|
stakeKeeper stake.Keeper
|
2018-04-06 22:12:00 -07:00
|
|
|
}
|
|
|
|
|
2018-04-14 13:52:58 -07:00
|
|
|
func NewGaiaApp(logger log.Logger, db dbm.DB) *GaiaApp {
|
2018-04-06 22:12:00 -07:00
|
|
|
// create your application object
|
|
|
|
var app = &GaiaApp{
|
2018-04-14 13:52:58 -07:00
|
|
|
BaseApp: bam.NewBaseApp(appName, logger, db),
|
2018-04-06 22:12:00 -07:00
|
|
|
cdc: MakeCodec(),
|
|
|
|
capKeyMainStore: sdk.NewKVStoreKey("main"),
|
|
|
|
capKeyAccountStore: sdk.NewKVStoreKey("acc"),
|
|
|
|
capKeyIBCStore: sdk.NewKVStoreKey("ibc"),
|
|
|
|
capKeyStakeStore: sdk.NewKVStoreKey("stake"),
|
|
|
|
}
|
|
|
|
|
|
|
|
// define the accountMapper
|
2018-04-09 20:08:00 -07:00
|
|
|
app.accountMapper = auth.NewAccountMapper(
|
|
|
|
app.cdc,
|
2018-04-06 22:12:00 -07:00
|
|
|
app.capKeyMainStore, // target store
|
|
|
|
&auth.BaseAccount{}, // prototype
|
2018-04-09 20:08:00 -07:00
|
|
|
).Seal()
|
2018-04-06 22:12:00 -07:00
|
|
|
|
|
|
|
// add handlers
|
2018-04-06 22:50:46 -07:00
|
|
|
app.coinKeeper = bank.NewCoinKeeper(app.accountMapper)
|
|
|
|
app.ibcMapper = ibc.NewIBCMapper(app.cdc, app.capKeyIBCStore)
|
|
|
|
app.stakeKeeper = stake.NewKeeper(app.cdc, app.capKeyStakeStore, app.coinKeeper)
|
2018-04-06 22:12:00 -07:00
|
|
|
app.Router().
|
2018-04-06 22:50:46 -07:00
|
|
|
AddRoute("bank", bank.NewHandler(app.coinKeeper)).
|
|
|
|
AddRoute("ibc", ibc.NewHandler(app.ibcMapper, app.coinKeeper)).
|
|
|
|
AddRoute("stake", stake.NewHandler(app.stakeKeeper))
|
2018-04-06 22:12:00 -07:00
|
|
|
|
|
|
|
// initialize BaseApp
|
|
|
|
app.SetTxDecoder(app.txDecoder)
|
|
|
|
app.SetInitChainer(app.initChainer)
|
2018-04-06 22:50:46 -07:00
|
|
|
app.SetEndBlocker(stake.NewEndBlocker(app.stakeKeeper))
|
2018-04-14 13:52:58 -07:00
|
|
|
app.MountStoresIAVL(app.capKeyMainStore, app.capKeyAccountStore, app.capKeyIBCStore, app.capKeyStakeStore)
|
2018-04-06 22:12:00 -07:00
|
|
|
app.SetAnteHandler(auth.NewAnteHandler(app.accountMapper))
|
|
|
|
err := app.LoadLatestVersion(app.capKeyMainStore)
|
|
|
|
if err != nil {
|
|
|
|
cmn.Exit(err.Error())
|
|
|
|
}
|
|
|
|
|
|
|
|
return app
|
|
|
|
}
|
|
|
|
|
|
|
|
// custom tx codec
|
|
|
|
func MakeCodec() *wire.Codec {
|
2018-04-09 20:08:00 -07:00
|
|
|
var cdc = wire.NewCodec()
|
|
|
|
|
|
|
|
// Register Msgs
|
|
|
|
cdc.RegisterInterface((*sdk.Msg)(nil), nil)
|
2018-04-11 10:27:36 -07:00
|
|
|
|
|
|
|
ibc.RegisterWire(cdc)
|
|
|
|
bank.RegisterWire(cdc)
|
|
|
|
stake.RegisterWire(cdc)
|
2018-04-09 20:08:00 -07:00
|
|
|
|
|
|
|
// Register AppAccount
|
|
|
|
cdc.RegisterInterface((*sdk.Account)(nil), nil)
|
|
|
|
cdc.RegisterConcrete(&auth.BaseAccount{}, "gaia/Account", nil)
|
|
|
|
|
|
|
|
// Register crypto.
|
|
|
|
wire.RegisterCrypto(cdc)
|
|
|
|
|
2018-04-06 22:12:00 -07:00
|
|
|
return cdc
|
|
|
|
}
|
|
|
|
|
|
|
|
// custom logic for transaction decoding
|
|
|
|
func (app *GaiaApp) txDecoder(txBytes []byte) (sdk.Tx, sdk.Error) {
|
|
|
|
var tx = sdk.StdTx{}
|
|
|
|
|
|
|
|
if len(txBytes) == 0 {
|
|
|
|
return nil, sdk.ErrTxDecode("txBytes are empty")
|
|
|
|
}
|
|
|
|
|
|
|
|
// StdTx.Msg is an interface. The concrete types
|
2018-04-14 13:52:58 -07:00
|
|
|
// are registered by MakeTxCodec
|
2018-04-06 22:12:00 -07:00
|
|
|
err := app.cdc.UnmarshalBinary(txBytes, &tx)
|
|
|
|
if err != nil {
|
|
|
|
return nil, sdk.ErrTxDecode("").TraceCause(err, "")
|
|
|
|
}
|
|
|
|
return tx, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// custom logic for gaia initialization
|
|
|
|
func (app *GaiaApp) initChainer(ctx sdk.Context, req abci.RequestInitChain) abci.ResponseInitChain {
|
|
|
|
stateJSON := req.AppStateBytes
|
|
|
|
|
|
|
|
genesisState := new(GenesisState)
|
|
|
|
err := json.Unmarshal(stateJSON, genesisState)
|
|
|
|
if err != nil {
|
|
|
|
panic(err) // TODO https://github.com/cosmos/cosmos-sdk/issues/468
|
|
|
|
// return sdk.ErrGenesisParse("").TraceCause(err, "")
|
|
|
|
}
|
|
|
|
|
2018-04-06 22:50:46 -07:00
|
|
|
// load the accounts
|
2018-04-06 22:12:00 -07:00
|
|
|
for _, gacc := range genesisState.Accounts {
|
|
|
|
acc := gacc.ToAccount()
|
2018-04-06 22:50:46 -07:00
|
|
|
app.accountMapper.SetAccount(ctx, acc)
|
2018-04-06 22:12:00 -07:00
|
|
|
}
|
2018-04-06 22:50:46 -07:00
|
|
|
|
|
|
|
// load the initial stake information
|
|
|
|
stake.InitGenesis(ctx, app.stakeKeeper, genesisState.StakeData)
|
|
|
|
|
2018-04-06 22:12:00 -07:00
|
|
|
return abci.ResponseInitChain{}
|
|
|
|
}
|
2018-04-06 22:50:46 -07:00
|
|
|
|
|
|
|
//__________________________________________________________
|
|
|
|
|
|
|
|
// State to Unmarshal
|
|
|
|
type GenesisState struct {
|
2018-04-11 10:27:36 -07:00
|
|
|
Accounts []GenesisAccount `json:"accounts"`
|
|
|
|
StakeData stake.GenesisState `json:"stake"`
|
2018-04-06 22:50:46 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// GenesisAccount doesn't need pubkey or sequence
|
|
|
|
type GenesisAccount struct {
|
|
|
|
Address sdk.Address `json:"address"`
|
|
|
|
Coins sdk.Coins `json:"coins"`
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewGenesisAccount(acc *auth.BaseAccount) GenesisAccount {
|
|
|
|
return GenesisAccount{
|
|
|
|
Address: acc.Address,
|
|
|
|
Coins: acc.Coins,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// convert GenesisAccount to GaiaAccount
|
|
|
|
func (ga *GenesisAccount) ToAccount() (acc *auth.BaseAccount) {
|
|
|
|
return &auth.BaseAccount{
|
|
|
|
Address: ga.Address,
|
|
|
|
Coins: ga.Coins.Sort(),
|
|
|
|
}
|
|
|
|
}
|
2018-04-11 08:58:11 -07:00
|
|
|
|
|
|
|
// DefaultGenAppState expects two args: an account address
|
|
|
|
// and a coin denomination, and gives lots of coins to that address.
|
|
|
|
func DefaultGenAppState(args []string, addr sdk.Address, coinDenom string) (json.RawMessage, error) {
|
|
|
|
|
|
|
|
accAuth := auth.NewBaseAccountWithAddress(addr)
|
|
|
|
accAuth.Coins = sdk.Coins{{"fermion", 100000}}
|
|
|
|
acc := NewGenesisAccount(&accAuth)
|
|
|
|
genaccs := []GenesisAccount{acc}
|
|
|
|
|
|
|
|
genesisState := GenesisState{
|
|
|
|
Accounts: genaccs,
|
2018-04-11 10:27:36 -07:00
|
|
|
StakeData: stake.GetDefaultGenesisState(),
|
2018-04-11 08:58:11 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
stateBytes, err := json.MarshalIndent(genesisState, "", "\t")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return stateBytes, nil
|
|
|
|
}
|