290 lines
10 KiB
Go
290 lines
10 KiB
Go
package simapp
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
cmn "github.com/tendermint/tendermint/libs/common"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
dbm "github.com/tendermint/tm-db"
|
|
|
|
bam "github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
"github.com/cosmos/cosmos-sdk/version"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
"github.com/cosmos/cosmos-sdk/x/bank"
|
|
"github.com/cosmos/cosmos-sdk/x/crisis"
|
|
distr "github.com/cosmos/cosmos-sdk/x/distribution"
|
|
"github.com/cosmos/cosmos-sdk/x/genaccounts"
|
|
"github.com/cosmos/cosmos-sdk/x/genutil"
|
|
"github.com/cosmos/cosmos-sdk/x/gov"
|
|
"github.com/cosmos/cosmos-sdk/x/mint"
|
|
"github.com/cosmos/cosmos-sdk/x/nft"
|
|
"github.com/cosmos/cosmos-sdk/x/params"
|
|
paramsclient "github.com/cosmos/cosmos-sdk/x/params/client"
|
|
"github.com/cosmos/cosmos-sdk/x/slashing"
|
|
"github.com/cosmos/cosmos-sdk/x/staking"
|
|
"github.com/cosmos/cosmos-sdk/x/supply"
|
|
)
|
|
|
|
const appName = "SimApp"
|
|
|
|
var (
|
|
// default home directories for the application CLI
|
|
DefaultCLIHome = os.ExpandEnv("$HOME/.simapp")
|
|
|
|
// default home directories for the application daemon
|
|
DefaultNodeHome = os.ExpandEnv("$HOME/.simapp")
|
|
|
|
// The module BasicManager is in charge of setting up basic,
|
|
// non-dependant module elements, such as codec registration
|
|
// and genesis verification.
|
|
ModuleBasics = module.NewBasicManager(
|
|
genaccounts.AppModuleBasic{},
|
|
genutil.AppModuleBasic{},
|
|
auth.AppModuleBasic{},
|
|
bank.AppModuleBasic{},
|
|
staking.AppModuleBasic{},
|
|
mint.AppModuleBasic{},
|
|
distr.AppModuleBasic{},
|
|
gov.NewAppModuleBasic(paramsclient.ProposalHandler, distr.ProposalHandler),
|
|
params.AppModuleBasic{},
|
|
crisis.AppModuleBasic{},
|
|
slashing.AppModuleBasic{},
|
|
nft.AppModuleBasic{},
|
|
supply.AppModuleBasic{},
|
|
)
|
|
|
|
// module account permissions
|
|
maccPerms = map[string][]string{
|
|
auth.FeeCollectorName: nil,
|
|
distr.ModuleName: nil,
|
|
mint.ModuleName: {supply.Minter},
|
|
staking.BondedPoolName: {supply.Burner, supply.Staking},
|
|
staking.NotBondedPoolName: {supply.Burner, supply.Staking},
|
|
gov.ModuleName: {supply.Burner},
|
|
}
|
|
)
|
|
|
|
// custom tx codec
|
|
func MakeCodec() *codec.Codec {
|
|
var cdc = codec.New()
|
|
ModuleBasics.RegisterCodec(cdc)
|
|
sdk.RegisterCodec(cdc)
|
|
codec.RegisterCrypto(cdc)
|
|
return cdc
|
|
}
|
|
|
|
// SimApp extends an ABCI application, but with most of its parameters exported.
|
|
// They are exported for convenience in creating helper functions, as object
|
|
// capabilities aren't needed for testing.
|
|
type SimApp struct {
|
|
*bam.BaseApp
|
|
cdc *codec.Codec
|
|
|
|
invCheckPeriod uint
|
|
|
|
// keys to access the substores
|
|
keys map[string]*sdk.KVStoreKey
|
|
tkeys map[string]*sdk.TransientStoreKey
|
|
|
|
// keepers
|
|
AccountKeeper auth.AccountKeeper
|
|
BankKeeper bank.Keeper
|
|
SupplyKeeper supply.Keeper
|
|
StakingKeeper staking.Keeper
|
|
SlashingKeeper slashing.Keeper
|
|
MintKeeper mint.Keeper
|
|
DistrKeeper distr.Keeper
|
|
GovKeeper gov.Keeper
|
|
CrisisKeeper crisis.Keeper
|
|
ParamsKeeper params.Keeper
|
|
NFTKeeper nft.Keeper
|
|
|
|
// the module manager
|
|
mm *module.Manager
|
|
|
|
// simulation manager
|
|
sm *module.SimulationManager
|
|
}
|
|
|
|
// NewSimApp returns a reference to an initialized SimApp.
|
|
func NewSimApp(
|
|
logger log.Logger, db dbm.DB, traceStore io.Writer, loadLatest bool,
|
|
invCheckPeriod uint, baseAppOptions ...func(*bam.BaseApp),
|
|
) *SimApp {
|
|
|
|
cdc := MakeCodec()
|
|
|
|
bApp := bam.NewBaseApp(appName, logger, db, auth.DefaultTxDecoder(cdc), baseAppOptions...)
|
|
bApp.SetCommitMultiStoreTracer(traceStore)
|
|
bApp.SetAppVersion(version.Version)
|
|
|
|
keys := sdk.NewKVStoreKeys(bam.MainStoreKey, auth.StoreKey, staking.StoreKey,
|
|
supply.StoreKey, mint.StoreKey, distr.StoreKey, slashing.StoreKey,
|
|
gov.StoreKey, params.StoreKey, nft.StoreKey)
|
|
tkeys := sdk.NewTransientStoreKeys(staking.TStoreKey, params.TStoreKey)
|
|
|
|
app := &SimApp{
|
|
BaseApp: bApp,
|
|
cdc: cdc,
|
|
invCheckPeriod: invCheckPeriod,
|
|
keys: keys,
|
|
tkeys: tkeys,
|
|
}
|
|
|
|
// init params keeper and subspaces
|
|
app.ParamsKeeper = params.NewKeeper(app.cdc, keys[params.StoreKey], tkeys[params.TStoreKey], params.DefaultCodespace)
|
|
authSubspace := app.ParamsKeeper.Subspace(auth.DefaultParamspace)
|
|
bankSubspace := app.ParamsKeeper.Subspace(bank.DefaultParamspace)
|
|
stakingSubspace := app.ParamsKeeper.Subspace(staking.DefaultParamspace)
|
|
mintSubspace := app.ParamsKeeper.Subspace(mint.DefaultParamspace)
|
|
distrSubspace := app.ParamsKeeper.Subspace(distr.DefaultParamspace)
|
|
slashingSubspace := app.ParamsKeeper.Subspace(slashing.DefaultParamspace)
|
|
govSubspace := app.ParamsKeeper.Subspace(gov.DefaultParamspace).WithKeyTable(gov.ParamKeyTable())
|
|
crisisSubspace := app.ParamsKeeper.Subspace(crisis.DefaultParamspace)
|
|
|
|
// add keepers
|
|
app.AccountKeeper = auth.NewAccountKeeper(app.cdc, keys[auth.StoreKey], authSubspace, auth.ProtoBaseAccount)
|
|
app.BankKeeper = bank.NewBaseKeeper(app.AccountKeeper, bankSubspace, bank.DefaultCodespace, app.ModuleAccountAddrs())
|
|
app.SupplyKeeper = supply.NewKeeper(app.cdc, keys[supply.StoreKey], app.AccountKeeper, app.BankKeeper, maccPerms)
|
|
stakingKeeper := staking.NewKeeper(app.cdc, keys[staking.StoreKey], tkeys[staking.TStoreKey],
|
|
app.SupplyKeeper, stakingSubspace, staking.DefaultCodespace)
|
|
app.MintKeeper = mint.NewKeeper(app.cdc, keys[mint.StoreKey], mintSubspace, &stakingKeeper, app.SupplyKeeper, auth.FeeCollectorName)
|
|
app.DistrKeeper = distr.NewKeeper(app.cdc, keys[distr.StoreKey], distrSubspace, &stakingKeeper,
|
|
app.SupplyKeeper, distr.DefaultCodespace, auth.FeeCollectorName, app.ModuleAccountAddrs())
|
|
app.SlashingKeeper = slashing.NewKeeper(app.cdc, keys[slashing.StoreKey], &stakingKeeper,
|
|
slashingSubspace, slashing.DefaultCodespace)
|
|
app.CrisisKeeper = crisis.NewKeeper(crisisSubspace, invCheckPeriod, app.SupplyKeeper, auth.FeeCollectorName)
|
|
app.NFTKeeper = nft.NewKeeper(app.cdc, keys[nft.StoreKey])
|
|
|
|
// register the proposal types
|
|
govRouter := gov.NewRouter()
|
|
govRouter.AddRoute(gov.RouterKey, gov.ProposalHandler).
|
|
AddRoute(params.RouterKey, params.NewParamChangeProposalHandler(app.ParamsKeeper)).
|
|
AddRoute(distr.RouterKey, distr.NewCommunityPoolSpendProposalHandler(app.DistrKeeper))
|
|
app.GovKeeper = gov.NewKeeper(app.cdc, keys[gov.StoreKey], govSubspace,
|
|
app.SupplyKeeper, &stakingKeeper, gov.DefaultCodespace, govRouter)
|
|
|
|
// register the staking hooks
|
|
// NOTE: stakingKeeper above is passed by reference, so that it will contain these hooks
|
|
app.StakingKeeper = *stakingKeeper.SetHooks(
|
|
staking.NewMultiStakingHooks(app.DistrKeeper.Hooks(), app.SlashingKeeper.Hooks()),
|
|
)
|
|
|
|
// NOTE: Any module instantiated in the module manager that is later modified
|
|
// must be passed by reference here.
|
|
app.mm = module.NewManager(
|
|
genaccounts.NewAppModule(app.AccountKeeper),
|
|
genutil.NewAppModule(app.AccountKeeper, app.StakingKeeper, app.BaseApp.DeliverTx),
|
|
auth.NewAppModule(app.AccountKeeper),
|
|
bank.NewAppModule(app.BankKeeper, app.AccountKeeper),
|
|
crisis.NewAppModule(&app.CrisisKeeper),
|
|
supply.NewAppModule(app.SupplyKeeper, app.AccountKeeper),
|
|
distr.NewAppModule(app.DistrKeeper, app.SupplyKeeper),
|
|
gov.NewAppModule(app.GovKeeper, app.SupplyKeeper),
|
|
mint.NewAppModule(app.MintKeeper),
|
|
slashing.NewAppModule(app.SlashingKeeper, app.StakingKeeper),
|
|
staking.NewAppModule(app.StakingKeeper, app.AccountKeeper, app.SupplyKeeper),
|
|
nft.NewAppModule(app.NFTKeeper),
|
|
)
|
|
|
|
// During begin block slashing happens after distr.BeginBlocker so that
|
|
// there is nothing left over in the validator fee pool, so as to keep the
|
|
// CanWithdrawInvariant invariant.
|
|
app.mm.SetOrderBeginBlockers(mint.ModuleName, distr.ModuleName, slashing.ModuleName)
|
|
|
|
app.mm.SetOrderEndBlockers(crisis.ModuleName, gov.ModuleName, staking.ModuleName)
|
|
|
|
// NOTE: The genutils moodule must occur after staking so that pools are
|
|
// properly initialized with tokens from genesis accounts.
|
|
app.mm.SetOrderInitGenesis(
|
|
genaccounts.ModuleName, distr.ModuleName, staking.ModuleName,
|
|
auth.ModuleName, bank.ModuleName, slashing.ModuleName, gov.ModuleName,
|
|
mint.ModuleName, supply.ModuleName, crisis.ModuleName, genutil.ModuleName,
|
|
)
|
|
|
|
app.mm.RegisterInvariants(&app.CrisisKeeper)
|
|
app.mm.RegisterRoutes(app.Router(), app.QueryRouter())
|
|
|
|
app.sm = module.NewSimulationManager(app.mm.Modules)
|
|
app.sm.RegisterStoreDecoders()
|
|
|
|
// initialize stores
|
|
app.MountKVStores(keys)
|
|
app.MountTransientStores(tkeys)
|
|
|
|
// initialize BaseApp
|
|
app.SetInitChainer(app.InitChainer)
|
|
app.SetBeginBlocker(app.BeginBlocker)
|
|
app.SetAnteHandler(auth.NewAnteHandler(app.AccountKeeper, app.SupplyKeeper, auth.DefaultSigVerificationGasConsumer))
|
|
app.SetEndBlocker(app.EndBlocker)
|
|
|
|
if loadLatest {
|
|
err := app.LoadLatestVersion(app.keys[bam.MainStoreKey])
|
|
if err != nil {
|
|
cmn.Exit(err.Error())
|
|
}
|
|
}
|
|
return app
|
|
}
|
|
|
|
// application updates every begin block
|
|
func (app *SimApp) BeginBlocker(ctx sdk.Context, req abci.RequestBeginBlock) abci.ResponseBeginBlock {
|
|
return app.mm.BeginBlock(ctx, req)
|
|
}
|
|
|
|
// application updates every end block
|
|
func (app *SimApp) EndBlocker(ctx sdk.Context, req abci.RequestEndBlock) abci.ResponseEndBlock {
|
|
return app.mm.EndBlock(ctx, req)
|
|
}
|
|
|
|
// application update at chain initialization
|
|
func (app *SimApp) InitChainer(ctx sdk.Context, req abci.RequestInitChain) abci.ResponseInitChain {
|
|
var genesisState GenesisState
|
|
app.cdc.MustUnmarshalJSON(req.AppStateBytes, &genesisState)
|
|
return app.mm.InitGenesis(ctx, genesisState)
|
|
}
|
|
|
|
// load a particular height
|
|
func (app *SimApp) LoadHeight(height int64) error {
|
|
return app.LoadVersion(height, app.keys[bam.MainStoreKey])
|
|
}
|
|
|
|
// ModuleAccountAddrs returns all the app's module account addresses.
|
|
func (app *SimApp) ModuleAccountAddrs() map[string]bool {
|
|
modAccAddrs := make(map[string]bool)
|
|
for acc := range maccPerms {
|
|
modAccAddrs[supply.NewModuleAddress(acc).String()] = true
|
|
}
|
|
|
|
return modAccAddrs
|
|
}
|
|
|
|
// Codec returns simapp's codec
|
|
func (app *SimApp) Codec() *codec.Codec {
|
|
return app.cdc
|
|
}
|
|
|
|
// GetKey returns the KVStoreKey for the provided store key
|
|
func (app *SimApp) GetKey(storeKey string) *sdk.KVStoreKey {
|
|
return app.keys[storeKey]
|
|
}
|
|
|
|
// GetTKey returns the TransientStoreKey for the provided store key
|
|
func (app *SimApp) GetTKey(storeKey string) *sdk.TransientStoreKey {
|
|
return app.tkeys[storeKey]
|
|
}
|
|
|
|
// GetMaccPerms returns a copy of the module account permissions
|
|
func GetMaccPerms() map[string][]string {
|
|
dupMaccPerms := make(map[string][]string)
|
|
for k, v := range maccPerms {
|
|
dupMaccPerms[k] = v
|
|
}
|
|
return dupMaccPerms
|
|
}
|