-
Notifications
You must be signed in to change notification settings - Fork 13
/
msg_server_claim.go
64 lines (50 loc) · 2.02 KB
/
msg_server_claim.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package keeper
import (
"context"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
"github.com/gitopia/gitopia/v2/x/rewards/types"
)
func (k msgServer) Claim(goCtx context.Context, msg *types.MsgClaim) (*types.MsgClaimResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
toAddr, err := sdk.AccAddressFromBech32(msg.Creator)
if err != nil {
return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidAddress, "bad address "+msg.Creator)
}
params := k.GetParams(ctx)
if !params.RewardSeries.SeriesOne.EndTime.IsZero() &&
params.RewardSeries.SeriesOne.EndTime.Before(ctx.BlockTime()) {
return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidRequest, "series 1 reward pool expired")
}
reward, found := k.GetReward(ctx, msg.Creator)
if !found {
return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidRequest, "reward not found for address "+msg.Creator)
}
if reward.Amount.IsZero() {
return nil, sdkerrors.Wrapf(sdkerrors.ErrInvalidRequest, "(%s) address not eligible for reward ", msg.Creator)
}
if reward.Amount.IsEqual(reward.ClaimedAmount) {
return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidRequest, "reward already claimed by address "+msg.Creator)
}
claimableAmount, err := k.GetTotalClaimableAmount(ctx, msg.Creator, reward.Amount)
if err != nil {
return nil, err
}
// claimedAmount cannot be greater than claimable amount
if reward.ClaimedAmount.IsEqual(claimableAmount) {
return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidRequest, "eligible reward already claimed. must complete more tasks")
}
balance := claimableAmount.Sub(reward.ClaimedAmount)
balanceWithDecay, err := k.GetDecayedRewardAmount(ctx, balance)
if err != nil {
return nil, err
}
err = k.bankKeeper.SendCoinsFromModuleToAccount(ctx, types.RewardsSeriesOneAccount, toAddr, sdk.Coins{balanceWithDecay})
if err != nil {
return nil, err
}
reward.ClaimedAmountWithDecay = reward.ClaimedAmount.Add(balanceWithDecay)
reward.ClaimedAmount = claimableAmount
k.SetReward(ctx, reward)
return &types.MsgClaimResponse{}, nil
}