-
Notifications
You must be signed in to change notification settings - Fork 202
/
upgrades.go
37 lines (32 loc) · 1.09 KB
/
upgrades.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package v6
import (
"fmt"
errorsmod "cosmossdk.io/errors"
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/types/module"
upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
claimkeeper "github.com/Stride-Labs/stride/v22/x/claim/keeper"
)
// Note: ensure these values are properly set before running upgrade
var (
UpgradeName = "v6"
)
// CreateUpgradeHandler creates an SDK upgrade handler for v6
func CreateUpgradeHandler(
mm *module.Manager,
configurator module.Configurator,
cdc codec.Codec,
claimKeeper claimkeeper.Keeper,
) upgradetypes.UpgradeHandler {
return func(ctx sdk.Context, _ upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
// Reset Claims
airdropClaimTypes := []string{"stride", "gaia", "osmosis", "juno", "stars"}
for _, claimType := range airdropClaimTypes {
if err := claimKeeper.ResetClaimStatus(ctx, claimType); err != nil {
return vm, errorsmod.Wrapf(err, fmt.Sprintf("unable to reset %s claim status", claimType))
}
}
return mm.RunMigrations(ctx, configurator, vm)
}
}