diff --git a/app/app.go b/app/app.go index 93da18b2..0e5a283b 100644 --- a/app/app.go +++ b/app/app.go @@ -8,6 +8,7 @@ import ( "path/filepath" "strings" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/x/genmsg" wasmdKeeper "github.com/CosmWasm/wasmd/x/wasm/keeper" @@ -265,7 +266,7 @@ type ArchwayApp struct { memKeys map[string]*sdk.MemoryStoreKey // Keepers - Keepers ArchwayKeepers + Keepers keepers.ArchwayKeepers ScopedIBCKeeper capabilitykeeper.ScopedKeeper ScopedICAHostKeeper capabilitykeeper.ScopedKeeper @@ -326,7 +327,7 @@ func NewArchwayApp( keys: keys, tkeys: tkeys, memKeys: memKeys, - Keepers: ArchwayKeepers{}, + Keepers: keepers.ArchwayKeepers{}, } app.Keepers.ParamsKeeper = initParamsKeeper( @@ -393,7 +394,7 @@ func NewArchwayApp( authtypes.FeeCollectorName, app.ModuleAccountAddrs(), ) - app.Keepers.slashingKeeper = slashingkeeper.NewKeeper( + app.Keepers.SlashingKeeper = slashingkeeper.NewKeeper( appCodec, keys[slashingtypes.StoreKey], &stakingKeeper, @@ -416,7 +417,7 @@ func NewArchwayApp( // register the staking hooks // NOTE: stakingKeeper above is passed by reference, so that it will contain these hooks app.Keepers.StakingKeeper = *stakingKeeper.SetHooks( - stakingtypes.NewMultiStakingHooks(app.Keepers.DistrKeeper.Hooks(), app.Keepers.slashingKeeper.Hooks()), + stakingtypes.NewMultiStakingHooks(app.Keepers.DistrKeeper.Hooks(), app.Keepers.SlashingKeeper.Hooks()), ) app.Keepers.IBCKeeper = ibckeeper.NewKeeper( @@ -475,7 +476,7 @@ func NewArchwayApp( appCodec, keys[evidencetypes.StoreKey], &app.Keepers.StakingKeeper, - app.Keepers.slashingKeeper, + app.Keepers.SlashingKeeper, ) app.Keepers.EvidenceKeeper = *evidenceKeeper @@ -610,7 +611,7 @@ func NewArchwayApp( capability.NewAppModule(appCodec, *app.Keepers.CapabilityKeeper), gov.NewAppModule(appCodec, app.Keepers.GovKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper), mint.NewAppModule(appCodec, app.Keepers.MintKeeper, app.Keepers.AccountKeeper), - slashing.NewAppModule(appCodec, app.Keepers.slashingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), + slashing.NewAppModule(appCodec, app.Keepers.SlashingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), distr.NewAppModule(appCodec, app.Keepers.DistrKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), staking.NewAppModule(appCodec, app.Keepers.StakingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper), upgrade.NewAppModule(app.Keepers.UpgradeKeeper), @@ -756,7 +757,7 @@ func NewArchwayApp( mint.NewAppModule(appCodec, app.Keepers.MintKeeper, app.Keepers.AccountKeeper), staking.NewAppModule(appCodec, app.Keepers.StakingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper), distr.NewAppModule(appCodec, app.Keepers.DistrKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), - slashing.NewAppModule(appCodec, app.Keepers.slashingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), + slashing.NewAppModule(appCodec, app.Keepers.SlashingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper, app.Keepers.StakingKeeper), params.NewAppModule(app.Keepers.ParamsKeeper), evidence.NewAppModule(app.Keepers.EvidenceKeeper), wasm.NewAppModule(appCodec, &app.Keepers.WASMKeeper, app.Keepers.StakingKeeper, app.Keepers.AccountKeeper, app.Keepers.BankKeeper), diff --git a/app/app_upgrades.go b/app/app_upgrades.go index b5a1aaf6..8978ca9f 100644 --- a/app/app_upgrades.go +++ b/app/app_upgrades.go @@ -54,7 +54,7 @@ func (app *ArchwayApp) setUpgradeHandlers() { for _, u := range Upgrades { app.Keepers.UpgradeKeeper.SetUpgradeHandler( u.UpgradeName, - u.CreateUpgradeHandler(app.mm, app.configurator, app.Keepers.AccountKeeper), + u.CreateUpgradeHandler(app.mm, app.configurator, app.Keepers), ) } } diff --git a/app/app_upgrades_test.go b/app/app_upgrades_test.go index 08b1f96b..f59b27ae 100644 --- a/app/app_upgrades_test.go +++ b/app/app_upgrades_test.go @@ -7,10 +7,10 @@ import ( storeTypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" "github.com/stretchr/testify/require" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" e2eTesting "github.com/archway-network/archway/e2e/testing" rewardsTypes "github.com/archway-network/archway/x/rewards/types" @@ -45,7 +45,7 @@ func TestUpgrades(t *testing.T) { proposalExecuted := false fauxUpgrade := upgrades.Upgrade{ UpgradeName: "test-upgrade", - CreateUpgradeHandler: func(manager *module.Manager, configurator module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(manager *module.Manager, configurator module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { proposalExecuted = true return manager.RunMigrations(ctx, configurator, fromVM) diff --git a/app/app_upgrades_util_test.go b/app/app_upgrades_util_test.go index 09ac1976..6b94fced 100644 --- a/app/app_upgrades_util_test.go +++ b/app/app_upgrades_util_test.go @@ -7,6 +7,6 @@ import "github.com/archway-network/archway/app/upgrades" func (app *ArchwayApp) AddUpgradeHandler(upgrade upgrades.Upgrade) { app.Keepers.UpgradeKeeper.SetUpgradeHandler( upgrade.UpgradeName, - upgrade.CreateUpgradeHandler(app.mm, app.configurator, app.Keepers.AccountKeeper), + upgrade.CreateUpgradeHandler(app.mm, app.configurator, app.Keepers), ) } diff --git a/app/export.go b/app/export.go index ac3162b0..5e33883c 100644 --- a/app/export.go +++ b/app/export.go @@ -183,11 +183,11 @@ func (app *ArchwayApp) prepForZeroHeightGenesis(ctx sdk.Context, jailAllowedAddr /* Handle slashing state. */ // reset start height on signing infos - app.Keepers.slashingKeeper.IterateValidatorSigningInfos( + app.Keepers.SlashingKeeper.IterateValidatorSigningInfos( ctx, func(addr sdk.ConsAddress, info slashingtypes.ValidatorSigningInfo) (stop bool) { info.StartHeight = 0 - app.Keepers.slashingKeeper.SetValidatorSigningInfo(ctx, addr, info) + app.Keepers.SlashingKeeper.SetValidatorSigningInfo(ctx, addr, info) return false }, ) diff --git a/app/keepers.go b/app/keepers/keepers.go similarity index 97% rename from app/keepers.go rename to app/keepers/keepers.go index 81fe7a4a..611a7233 100644 --- a/app/keepers.go +++ b/app/keepers/keepers.go @@ -1,4 +1,4 @@ -package app +package keepers import ( "github.com/CosmWasm/wasmd/x/wasm" @@ -29,7 +29,7 @@ type ArchwayKeepers struct { BankKeeper bankkeeper.Keeper CapabilityKeeper *capabilitykeeper.Keeper StakingKeeper stakingkeeper.Keeper - slashingKeeper slashingkeeper.Keeper + SlashingKeeper slashingkeeper.Keeper MintKeeper mintkeeper.Keeper DistrKeeper distrkeeper.Keeper GovKeeper govkeeper.Keeper diff --git a/app/upgrades/06/upgrades.go b/app/upgrades/06/upgrades.go index df9042f9..49c014f0 100644 --- a/app/upgrades/06/upgrades.go +++ b/app/upgrades/06/upgrades.go @@ -4,10 +4,10 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" ibcfeetypes "github.com/cosmos/ibc-go/v4/modules/apps/29-fee/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -15,7 +15,7 @@ const Name = "v0.6.0" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { return mm.RunMigrations(ctx, cfg, fromVM) } diff --git a/app/upgrades/1_0_0_rc_4/upgrades.go b/app/upgrades/1_0_0_rc_4/upgrades.go index 4fb3fe65..01b8cc89 100644 --- a/app/upgrades/1_0_0_rc_4/upgrades.go +++ b/app/upgrades/1_0_0_rc_4/upgrades.go @@ -4,9 +4,9 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -14,7 +14,7 @@ const Name = "v1.0.0-rc.4" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { return mm.RunMigrations(ctx, cfg, fromVM) } diff --git a/app/upgrades/2_0_0/upgrades.go b/app/upgrades/2_0_0/upgrades.go index 20235670..1fda2a6d 100644 --- a/app/upgrades/2_0_0/upgrades.go +++ b/app/upgrades/2_0_0/upgrades.go @@ -5,7 +5,6 @@ import ( sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" banktypes "github.com/cosmos/cosmos-sdk/x/bank/types" distrtypes "github.com/cosmos/cosmos-sdk/x/distribution/types" govtypes "github.com/cosmos/cosmos-sdk/x/gov/types" @@ -17,6 +16,7 @@ import ( icahosttypes "github.com/cosmos/ibc-go/v4/modules/apps/27-interchain-accounts/host/types" icatypes "github.com/cosmos/ibc-go/v4/modules/apps/27-interchain-accounts/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -24,7 +24,7 @@ const Name = "v2.0.0" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { // Set Initial Consensus Version diff --git a/app/upgrades/3_0_0/upgrades.go b/app/upgrades/3_0_0/upgrades.go index 0e9f4f61..77a550eb 100644 --- a/app/upgrades/3_0_0/upgrades.go +++ b/app/upgrades/3_0_0/upgrades.go @@ -4,9 +4,9 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -14,7 +14,7 @@ const Name = "v3.0.0" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { return mm.RunMigrations(ctx, cfg, fromVM) } diff --git a/app/upgrades/4_0_0/upgrades.go b/app/upgrades/4_0_0/upgrades.go index 3598ad19..427c7c8d 100644 --- a/app/upgrades/4_0_0/upgrades.go +++ b/app/upgrades/4_0_0/upgrades.go @@ -4,9 +4,9 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -14,7 +14,7 @@ const Name = "v4.0.0" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, _ keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { return mm.RunMigrations(ctx, cfg, fromVM) } diff --git a/app/upgrades/4_0_2/upgrades.go b/app/upgrades/4_0_2/upgrades.go index 0bca3ebd..c208fe9d 100644 --- a/app/upgrades/4_0_2/upgrades.go +++ b/app/upgrades/4_0_2/upgrades.go @@ -6,10 +6,10 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -17,8 +17,9 @@ const Name = "v4.0.2" var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, accountKeeper keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, keepers keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { + accountKeeper := keepers.AccountKeeper fcAccount := accountKeeper.GetModuleAccount(ctx, authtypes.FeeCollectorName) account, ok := fcAccount.(*authtypes.ModuleAccount) if !ok { diff --git a/app/upgrades/latest/upgrades.go b/app/upgrades/latest/upgrades.go index bf932460..171ecb40 100644 --- a/app/upgrades/latest/upgrades.go +++ b/app/upgrades/latest/upgrades.go @@ -4,9 +4,9 @@ import ( storetypes "github.com/cosmos/cosmos-sdk/store/types" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" + "github.com/archway-network/archway/app/keepers" "github.com/archway-network/archway/app/upgrades" ) @@ -25,7 +25,7 @@ const NameAsciiArt = ` var Upgrade = upgrades.Upgrade{ UpgradeName: Name, - CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, accountKeeper keeper.AccountKeeper) upgradetypes.UpgradeHandler { + CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator, keepers keepers.ArchwayKeepers) upgradetypes.UpgradeHandler { return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { migrations, err := mm.RunMigrations(ctx, cfg, fromVM) if err != nil { diff --git a/app/upgrades/upgrades.go b/app/upgrades/upgrades.go index 45bff679..eb299677 100644 --- a/app/upgrades/upgrades.go +++ b/app/upgrades/upgrades.go @@ -1,9 +1,9 @@ package upgrades import ( + "github.com/archway-network/archway/app/keepers" "github.com/cosmos/cosmos-sdk/store/types" "github.com/cosmos/cosmos-sdk/types/module" - "github.com/cosmos/cosmos-sdk/x/auth/keeper" upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types" ) @@ -16,7 +16,7 @@ type Upgrade struct { UpgradeName string // CreateUpgradeHandler defines the function that creates an upgrade handler - CreateUpgradeHandler func(*module.Manager, module.Configurator, keeper.AccountKeeper) upgradetypes.UpgradeHandler + CreateUpgradeHandler func(*module.Manager, module.Configurator, keepers.ArchwayKeepers) upgradetypes.UpgradeHandler // Store upgrades, should be used for any new modules introduced, new modules deleted, or store names renamed. StoreUpgrades types.StoreUpgrades