forked from hardiksa/torqueOLD
/
claims_record.go
116 lines (100 loc) · 3.36 KB
/
claims_record.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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package types
import (
"errors"
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
)
// IBCTriggerAmt is the amount required to trigger a merge/migration of claims records
var IBCTriggerAmt = "63743"
// NewClaimsRecord creates a new claims record instance
func NewClaimsRecord(initialClaimableAmt sdk.Int) ClaimsRecord {
return ClaimsRecord{
InitialClaimableAmount: initialClaimableAmt,
ActionsCompleted: []bool{false, false, false, false},
}
}
// Validate performs a stateless validation of the fields
func (cr ClaimsRecord) Validate() error {
if cr.InitialClaimableAmount.IsNil() {
return errors.New("initial claimable amount is nil")
}
if !cr.InitialClaimableAmount.IsPositive() {
return fmt.Errorf("initial claimable amount is not positive, %s", cr.InitialClaimableAmount)
}
if len(cr.ActionsCompleted) == 0 || len(Action_value)-1 != len(cr.ActionsCompleted) {
return fmt.Errorf("action length mismatch, expected %d, got %d", len(Action_value)-1, len(cr.ActionsCompleted))
}
return nil
}
// MarkClaimed marks the given action as completed (i.e claimed). It performs a no-op if the
// action is invalid or if the ActionsCompleted slice has an invalid length.
func (cr *ClaimsRecord) MarkClaimed(action Action) {
switch {
case len(cr.ActionsCompleted) != len(Action_value)-1:
return
case action == ActionUnspecified || int(action) > len(Action_value)-1:
return
default:
cr.ActionsCompleted[action-1] = true
}
}
// HasClaimedAction checks if the user has claimed a given action. It also
// returns false if the action is invalid or if the ActionsCompleted slice has
// an invalid length.
func (cr ClaimsRecord) HasClaimedAction(action Action) bool {
switch {
case len(cr.ActionsCompleted) != len(Action_value)-1:
return false
case action == 0 || int(action) > len(Action_value)-1:
return false
default:
return cr.ActionsCompleted[action-1]
}
}
// HasClaimedAny returns true if the user has claimed at least one reward from
// the available actions
func (cr ClaimsRecord) HasClaimedAny() bool {
for _, completed := range cr.ActionsCompleted {
if completed {
return true
}
}
return false
}
// HasClaimedAll returns true if the user has claimed all the rewards from the
// available actions
func (cr ClaimsRecord) HasClaimedAll() bool {
if len(cr.ActionsCompleted) == 0 {
return false
}
for _, completed := range cr.ActionsCompleted {
if !completed {
return false
}
}
return true
}
// NewClaimsRecordAddress creates a new claims record instance
func NewClaimsRecordAddress(address sdk.AccAddress, initialClaimableAmt sdk.Int) ClaimsRecordAddress {
return ClaimsRecordAddress{
Address: address.String(),
InitialClaimableAmount: initialClaimableAmt,
ActionsCompleted: []bool{false, false, false, false},
}
}
// Validate performs a stateless validation of the fields
func (cra ClaimsRecordAddress) Validate() error {
if _, err := sdk.AccAddressFromBech32(cra.Address); err != nil {
return err
}
if cra.InitialClaimableAmount.IsNil() {
return errors.New("initial claimable amount is nil")
}
if !cra.InitialClaimableAmount.IsPositive() {
return fmt.Errorf("initial claimable amount is not positive, %s", cra.InitialClaimableAmount)
}
if len(Action_value)-1 != len(cra.ActionsCompleted) {
return fmt.Errorf("action length mismatch, expected %d, got %d", len(Action_value)-1, len(cra.ActionsCompleted))
}
return nil
}