106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"cosmossdk.io/x/protocolpool/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
func (k Keeper) InitGenesis(ctx context.Context, data *types.GenesisState) error {
|
|
currentTime := k.HeaderService.HeaderInfo(ctx).Time
|
|
for _, cf := range data.ContinuousFund {
|
|
// ignore expired ContinuousFunds
|
|
if cf.Expiry != nil && cf.Expiry.Before(currentTime) {
|
|
continue
|
|
}
|
|
|
|
recipientAddress, err := k.authKeeper.AddressCodec().StringToBytes(cf.Recipient)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to decode recipient address: %w", err)
|
|
}
|
|
if err := k.ContinuousFund.Set(ctx, recipientAddress, *cf); err != nil {
|
|
return fmt.Errorf("failed to set continuous fund for recipient %s: %w", recipientAddress, err)
|
|
}
|
|
}
|
|
for _, budget := range data.Budget {
|
|
// Validate StartTime
|
|
if budget.LastClaimedAt == nil || budget.LastClaimedAt.IsZero() {
|
|
budget.LastClaimedAt = ¤tTime
|
|
}
|
|
// ignore budgets with period <= 0 || nil
|
|
if budget.Period == nil || (budget.Period != nil && budget.Period.Seconds() <= 0) {
|
|
continue
|
|
}
|
|
|
|
// ignore budget with start time < currentTime
|
|
if budget.LastClaimedAt.Before(currentTime) {
|
|
continue
|
|
}
|
|
|
|
recipientAddress, err := k.authKeeper.AddressCodec().StringToBytes(budget.RecipientAddress)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to decode recipient address: %w", err)
|
|
}
|
|
if err = k.BudgetProposal.Set(ctx, recipientAddress, *budget); err != nil {
|
|
return fmt.Errorf("failed to set budget for recipient %s: %w", recipientAddress, err)
|
|
}
|
|
}
|
|
|
|
if err := k.ToDistribute.Set(ctx, data.ToDistribute); err != nil {
|
|
return fmt.Errorf("failed to set to distribute: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (k Keeper) ExportGenesis(ctx context.Context) (*types.GenesisState, error) {
|
|
var cf []*types.ContinuousFund
|
|
err := k.ContinuousFund.Walk(ctx, nil, func(key sdk.AccAddress, value types.ContinuousFund) (stop bool, err error) {
|
|
recipient, err := k.authKeeper.AddressCodec().BytesToString(key)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
cf = append(cf, &types.ContinuousFund{
|
|
Recipient: recipient,
|
|
Percentage: value.Percentage,
|
|
Expiry: value.Expiry,
|
|
})
|
|
return false, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var budget []*types.Budget
|
|
err = k.BudgetProposal.Walk(ctx, nil, func(key sdk.AccAddress, value types.Budget) (stop bool, err error) {
|
|
recipient, err := k.authKeeper.AddressCodec().BytesToString(key)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
budget = append(budget, &types.Budget{
|
|
RecipientAddress: recipient,
|
|
ClaimedAmount: value.ClaimedAmount,
|
|
LastClaimedAt: value.LastClaimedAt,
|
|
TranchesLeft: value.TranchesLeft,
|
|
Period: value.Period,
|
|
BudgetPerTranche: value.BudgetPerTranche,
|
|
})
|
|
return false, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
genState := types.NewGenesisState(cf, budget)
|
|
|
|
genState.ToDistribute, err = k.ToDistribute.Get(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return genState, nil
|
|
}
|