-
Notifications
You must be signed in to change notification settings - Fork 10
/
contract_extension.go
78 lines (70 loc) · 2.24 KB
/
contract_extension.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
package types
import (
"math"
wasmtypes "github.com/CosmWasm/wasmd/x/wasm/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
)
//AddRegisteredPrivilege add privilege type to list
func (d *TgradeContractDetails) AddRegisteredPrivilege(t PrivilegeType, pos uint8) {
d.RegisteredPrivileges = append(d.RegisteredPrivileges, RegisteredPrivilege{
PrivilegeType: t.String(),
Position: uint32(pos),
})
}
// RemoveRegisteredPrivilege remove privilege type from list
func (d *TgradeContractDetails) RemoveRegisteredPrivilege(t PrivilegeType, pos uint8) {
src := &RegisteredPrivilege{
PrivilegeType: t.String(),
Position: uint32(pos),
}
for i, v := range d.RegisteredPrivileges {
if src.Equal(v) {
d.RegisteredPrivileges = append(d.RegisteredPrivileges[0:i], d.RegisteredPrivileges[i+1:]...)
}
}
}
// HasRegisteredPrivilege returs true when given type was registered by this contract
func (d *TgradeContractDetails) HasRegisteredPrivilege(c PrivilegeType) bool {
for _, v := range d.RegisteredPrivileges {
if v.PrivilegeType == c.String() {
return true
}
}
return false
}
func (d TgradeContractDetails) IterateRegisteredPrivileges(cb func(c PrivilegeType, pos uint8) bool) {
for _, v := range d.RegisteredPrivileges {
if cb(*PrivilegeTypeFrom(v.PrivilegeType), uint8(v.Position)) {
return
}
}
}
// ValidateBasic syntax checks
func (d TgradeContractDetails) ValidateBasic() error {
unique := make(map[PrivilegeType]struct{})
for i, c := range d.RegisteredPrivileges {
if err := c.ValidateBasic(); err != nil {
return sdkerrors.Wrapf(err, "privilege %d", i)
}
privilegeType := *PrivilegeTypeFrom(c.PrivilegeType)
if _, exists := unique[privilegeType]; exists {
return sdkerrors.Wrapf(wasmtypes.ErrDuplicate, "privilege %q", privilegeType.String())
}
unique[privilegeType] = struct{}{}
}
return nil
}
// ValidateBasic syntax checks
func (r RegisteredPrivilege) ValidateBasic() error {
if r.Position > math.MaxUint8 {
return sdkerrors.Wrap(wasmtypes.ErrInvalid, "position exceeds max")
}
if r.Position == 0 {
return sdkerrors.Wrap(wasmtypes.ErrEmpty, "position")
}
tp := PrivilegeTypeFrom(r.PrivilegeType)
if tp == nil {
return sdkerrors.Wrap(wasmtypes.ErrInvalid, "privilege type")
}
return nil
}