2020-07-12 23:55:58 -07:00
|
|
|
package keeper
|
2019-01-28 19:06:48 -08:00
|
|
|
|
|
|
|
import (
|
2020-01-30 13:31:16 -08:00
|
|
|
"fmt"
|
|
|
|
|
2019-01-28 19:06:48 -08:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2020-04-20 08:22:12 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/x/bank/types"
|
2019-01-28 19:06:48 -08:00
|
|
|
)
|
|
|
|
|
2020-01-30 13:31:16 -08:00
|
|
|
// InitGenesis initializes the bank module's state from a given genesis state.
|
2020-07-12 23:55:58 -07:00
|
|
|
func (k BaseKeeper) InitGenesis(ctx sdk.Context, genState types.GenesisState) {
|
|
|
|
k.SetParams(ctx, genState.Params)
|
2020-01-30 13:31:16 -08:00
|
|
|
|
2020-04-20 08:22:12 -07:00
|
|
|
var totalSupply sdk.Coins
|
|
|
|
|
2020-06-14 16:06:16 -07:00
|
|
|
genState.Balances = types.SanitizeGenesisBalances(genState.Balances)
|
2020-01-30 13:31:16 -08:00
|
|
|
for _, balance := range genState.Balances {
|
2020-07-12 23:55:58 -07:00
|
|
|
if err := k.ValidateBalance(ctx, balance.Address); err != nil {
|
2020-01-30 13:31:16 -08:00
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
|
2020-07-12 23:55:58 -07:00
|
|
|
if err := k.SetBalances(ctx, balance.Address, balance.Coins); err != nil {
|
2020-03-20 12:24:18 -07:00
|
|
|
panic(fmt.Errorf("error on setting balances %w", err))
|
|
|
|
}
|
2020-04-20 08:22:12 -07:00
|
|
|
|
|
|
|
totalSupply = totalSupply.Add(balance.Coins...)
|
|
|
|
}
|
|
|
|
|
|
|
|
if genState.Supply.Empty() {
|
|
|
|
genState.Supply = totalSupply
|
2020-01-30 13:31:16 -08:00
|
|
|
}
|
2020-04-20 08:22:12 -07:00
|
|
|
|
2020-07-12 23:55:58 -07:00
|
|
|
k.SetSupply(ctx, types.NewSupply(genState.Supply))
|
2019-01-28 19:06:48 -08:00
|
|
|
}
|
|
|
|
|
2020-01-30 13:31:16 -08:00
|
|
|
// ExportGenesis returns the bank module's genesis state.
|
2020-07-12 23:55:58 -07:00
|
|
|
func (k BaseKeeper) ExportGenesis(ctx sdk.Context) types.GenesisState {
|
2020-01-30 13:31:16 -08:00
|
|
|
balancesSet := make(map[string]sdk.Coins)
|
|
|
|
|
2020-07-12 23:55:58 -07:00
|
|
|
k.IterateAllBalances(ctx, func(addr sdk.AccAddress, balance sdk.Coin) bool {
|
2020-01-30 13:31:16 -08:00
|
|
|
balancesSet[addr.String()] = balancesSet[addr.String()].Add(balance)
|
|
|
|
return false
|
|
|
|
})
|
|
|
|
|
2020-06-14 16:06:16 -07:00
|
|
|
balances := []types.Balance{}
|
2020-01-30 13:31:16 -08:00
|
|
|
|
|
|
|
for addrStr, coins := range balancesSet {
|
|
|
|
addr, err := sdk.AccAddressFromBech32(addrStr)
|
|
|
|
if err != nil {
|
|
|
|
panic(fmt.Errorf("failed to convert address from string: %w", err))
|
|
|
|
}
|
|
|
|
|
2020-06-14 16:06:16 -07:00
|
|
|
balances = append(balances, types.Balance{
|
2020-01-30 13:31:16 -08:00
|
|
|
Address: addr,
|
|
|
|
Coins: coins,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-07-12 23:55:58 -07:00
|
|
|
return types.NewGenesisState(k.GetParams(ctx), balances, k.GetSupply(ctx).GetTotal())
|
2019-01-28 19:06:48 -08:00
|
|
|
}
|