-
Notifications
You must be signed in to change notification settings - Fork 45
/
claim_reward.go
88 lines (77 loc) · 2.67 KB
/
claim_reward.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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package keeper
import (
errorsmod "cosmossdk.io/errors"
"cosmossdk.io/math"
sdk "github.com/cosmos/cosmos-sdk/types"
aptypes "github.com/elys-network/elys/x/assetprofile/types"
"github.com/elys-network/elys/x/commitment/types"
)
// Update commitments for claim reward operation
func (k Keeper) RecordClaimReward(ctx sdk.Context, creator string, denom string, amount math.Int, withdrawMode types.EarnType) error {
assetProfile, found := k.assetProfileKeeper.GetEntry(ctx, denom)
if !found {
return errorsmod.Wrapf(aptypes.ErrAssetProfileNotFound, "denom: %s", denom)
}
if !assetProfile.WithdrawEnabled {
return errorsmod.Wrapf(types.ErrWithdrawDisabled, "denom: %s", denom)
}
// uses asset profile denom
denom = assetProfile.Denom
// Get the Commitments for the creator
commitments := k.GetCommitments(ctx, creator)
if !found {
return errorsmod.Wrapf(types.ErrCommitmentsNotFound, "creator: %s", creator)
}
// Withdraw reward not depending on program type
switch withdrawMode {
case types.EarnType_ALL_PROGRAM:
// Subtract the withdrawn amount from the unclaimed balance
err := commitments.SubRewardsUnclaimed(sdk.NewCoin(denom, amount))
if err != nil {
return err
}
case types.EarnType_USDC_PROGRAM:
// Subtract the withdrawn amount from the unclaimed balance
err := commitments.SubRewardsUnclaimedForUSDCDeposit(sdk.NewCoin(denom, amount))
if err != nil {
return err
}
case types.EarnType_ELYS_PROGRAM:
// Subtract the withdrawn amount from the unclaimed balance
err := commitments.SubRewardsUnclaimedForElysStaking(sdk.NewCoin(denom, amount))
if err != nil {
return err
}
case types.EarnType_EDEN_PROGRAM:
// Subtract the withdrawn amount from the unclaimed balance
err := commitments.SubRewardsUnclaimedForEdenCommitted(sdk.NewCoin(denom, amount))
if err != nil {
return err
}
case types.EarnType_EDENB_PROGRAM:
// Subtract the withdrawn amount from the unclaimed balance
err := commitments.SubRewardsUnclaimedForEdenBCommitted(sdk.NewCoin(denom, amount))
if err != nil {
return err
}
default:
return errorsmod.Wrapf(types.ErrUnsupportedWithdrawMode, "creator: %s", creator)
}
// Update the commitments
k.SetCommitments(ctx, commitments)
withdrawCoins := sdk.NewCoins(sdk.NewCoin(denom, amount))
err := k.HandleWithdrawFromCommitment(ctx, &commitments, withdrawCoins, false, sdk.AccAddress{})
if err != nil {
return err
}
// Emit blockchain event
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeCommitmentChanged,
sdk.NewAttribute(types.AttributeCreator, creator),
sdk.NewAttribute(types.AttributeAmount, amount.String()),
sdk.NewAttribute(types.AttributeDenom, denom),
),
)
return nil
}