58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package keeper
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/protocolpool/types"
|
|
)
|
|
|
|
func (k Keeper) InitGenesis(ctx sdk.Context, data *types.GenesisState) error {
|
|
currentTime := ctx.BlockTime()
|
|
|
|
err := k.Params.Set(ctx, data.Params)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to set params: %w", err)
|
|
}
|
|
|
|
for _, cf := range data.ContinuousFunds {
|
|
recipientAddress, err := k.authKeeper.AddressCodec().StringToBytes(cf.Recipient)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to decode recipient address: %w", err)
|
|
}
|
|
|
|
if k.bankKeeper.BlockedAddr(recipientAddress) {
|
|
return fmt.Errorf("recipient is blocked in the bank keeper: %s", recipientAddress)
|
|
}
|
|
|
|
// ignore expired ContinuousFunds
|
|
if cf.Expiry != nil && cf.Expiry.Before(currentTime) {
|
|
continue
|
|
}
|
|
|
|
if err := k.ContinuousFunds.Set(ctx, recipientAddress, cf); err != nil {
|
|
return fmt.Errorf("failed to set continuous fund for recipient %s: %w", recipientAddress, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (k Keeper) ExportGenesis(ctx sdk.Context) (*types.GenesisState, error) {
|
|
cf, err := k.GetAllContinuousFunds(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
genState := types.NewGenesisState(cf)
|
|
|
|
params, err := k.Params.Get(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
genState.Params = params
|
|
|
|
return genState, nil
|
|
}
|