56 lines
1.8 KiB
Go
56 lines
1.8 KiB
Go
package v039
|
|
|
|
import (
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
v038auth "github.com/cosmos/cosmos-sdk/x/auth/legacy/v0_38"
|
|
v039auth "github.com/cosmos/cosmos-sdk/x/auth/legacy/v0_39"
|
|
v038bank "github.com/cosmos/cosmos-sdk/x/bank/legacy/v0_38"
|
|
v039bank "github.com/cosmos/cosmos-sdk/x/bank/legacy/v0_39"
|
|
"github.com/cosmos/cosmos-sdk/x/genutil"
|
|
)
|
|
|
|
func Migrate(appState genutil.AppMap) genutil.AppMap {
|
|
v038Codec := codec.New()
|
|
codec.RegisterCrypto(v038Codec)
|
|
v038auth.RegisterCodec(v038Codec)
|
|
|
|
v039Codec := codec.New()
|
|
codec.RegisterCrypto(v039Codec)
|
|
v038auth.RegisterCodec(v039Codec)
|
|
|
|
// remove balances from existing accounts
|
|
if appState[v038auth.ModuleName] != nil {
|
|
// unmarshal relative source genesis application state
|
|
var authGenState v038auth.GenesisState
|
|
v038Codec.MustUnmarshalJSON(appState[v038auth.ModuleName], &authGenState)
|
|
|
|
// delete deprecated x/auth genesis state
|
|
delete(appState, v038auth.ModuleName)
|
|
|
|
// Migrate relative source genesis application state and marshal it into
|
|
// the respective key.
|
|
appState[v039auth.ModuleName] = v039Codec.MustMarshalJSON(v039auth.Migrate(authGenState))
|
|
}
|
|
|
|
if appState[v038bank.ModuleName] != nil {
|
|
// unmarshal relative source genesis application state
|
|
var bankGenState v038bank.GenesisState
|
|
v038Codec.MustUnmarshalJSON(appState[v038bank.ModuleName], &bankGenState)
|
|
|
|
// unmarshal x/auth genesis state to retrieve all account balances
|
|
var authGenState v038auth.GenesisState
|
|
v038Codec.MustUnmarshalJSON(appState[v038auth.ModuleName], &authGenState)
|
|
|
|
// delete deprecated x/bank genesis state
|
|
delete(appState, v038bank.ModuleName)
|
|
|
|
// Migrate relative source genesis application state and marshal it into
|
|
// the respective key.
|
|
appState[v039bank.ModuleName] = v039Codec.MustMarshalJSON(
|
|
v039bank.Migrate(bankGenState, authGenState),
|
|
)
|
|
}
|
|
|
|
return appState
|
|
}
|