Skip to content

Commit

Permalink
update migration code to add burn permission to treasury module
Browse files Browse the repository at this point in the history
  • Loading branch information
yun-yeo committed Jun 2, 2021
1 parent d04903f commit 24ada49
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion custom/auth/legacy/v040/migrate.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ import (
v40mint "github.com/cosmos/cosmos-sdk/x/mint/types"

v039authcustom "github.com/terra-money/core/custom/auth/legacy/v039"
v40treasury "github.com/terra-money/core/x/treasury/types"
v040vesting "github.com/terra-money/core/x/vesting/types"
)

Expand Down Expand Up @@ -54,6 +55,7 @@ func convertBaseVestingAccount(old *v039auth.BaseVestingAccount) *v040authvestin
// - Re-encode in v0.40 GenesisState.
func Migrate(authGenState v039auth.GenesisState) *v040auth.GenesisState {
mintModuleAddress := v040auth.NewModuleAddress(v40mint.ModuleName)
treasuryModuleAddress := v040auth.NewModuleAddress(v40treasury.ModuleName)

// Convert v0.39 accounts to v0.40 ones.
var v040Accounts = make([]v040auth.GenesisAccount, len(authGenState.Accounts))
Expand All @@ -75,10 +77,16 @@ func Migrate(authGenState v039auth.GenesisState) *v040auth.GenesisState {
}
case *v039auth.ModuleAccount:
{
// burn permission must be added to treasury module
permissions := v039Account.Permissions
if !v039Account.GetAddress().Equals(treasuryModuleAddress) {
permissions = append(permissions, v040auth.Burner)
}

v040Accounts[i] = &v040auth.ModuleAccount{
BaseAccount: convertBaseAccount(v039Account.BaseAccount),
Name: v039Account.Name,
Permissions: v039Account.Permissions,
Permissions: permissions,
}
}
case *v039auth.BaseVestingAccount:
Expand Down

0 comments on commit 24ada49

Please sign in to comment.