forked from umee-network/umee
-
Notifications
You must be signed in to change notification settings - Fork 0
/
validate.go
102 lines (92 loc) · 2.81 KB
/
validate.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
package keeper
import (
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/mokitanetwork/katana/x/leverage/types"
)
// validateAcceptedDenom validates an sdk.Coin and ensures it is a registered Token
// with Blacklisted == false
func (k Keeper) validateAcceptedDenom(ctx sdk.Context, denom string) error {
if types.HasUTokenPrefix(denom) {
return types.ErrUToken.Wrap(denom)
}
token, err := k.GetTokenSettings(ctx, denom)
if err != nil {
return err
}
return token.AssertNotBlacklisted()
}
// validateAcceptedAsset validates an sdk.Coin and ensures it is a registered Token
// with Blacklisted == false
func (k Keeper) validateAcceptedAsset(ctx sdk.Context, coin sdk.Coin) error {
if err := coin.Validate(); err != nil {
return err
}
return k.validateAcceptedDenom(ctx, coin.Denom)
}
// validateSupply validates an sdk.Coin and ensures its Denom is a Token with EnableMsgSupply
func (k Keeper) validateSupply(ctx sdk.Context, coin sdk.Coin) error {
if err := validateBaseToken(coin); err != nil {
return err
}
token, err := k.GetTokenSettings(ctx, coin.Denom)
if err != nil {
return err
}
return token.AssertSupplyEnabled()
}
// validateBorrow validates an sdk.Coin and ensures its Denom is a Token with EnableMsgBorrow
func (k Keeper) validateBorrow(ctx sdk.Context, borrow sdk.Coin) error {
if err := validateBaseToken(borrow); err != nil {
return err
}
token, err := k.GetTokenSettings(ctx, borrow.Denom)
if err != nil {
return err
}
return token.AssertBorrowEnabled()
}
// validateCollateralize validates an sdk.Coin and ensures it is a uToken of an accepted
// Token with EnableMsgSupply and CollateralWeight > 0
func (k Keeper) validateCollateralize(ctx sdk.Context, collateral sdk.Coin) error {
if err := validateUToken(collateral); err != nil {
return err
}
token, err := k.GetTokenSettings(ctx, types.ToTokenDenom(collateral.Denom))
if err != nil {
return err
}
if token.CollateralWeight.IsZero() {
return types.ErrCollateralWeightZero
}
return token.AssertSupplyEnabled()
}
// validateBaseToken validates an sdk.Coin and ensures its Denom is not a uToken.
func validateBaseToken(coin sdk.Coin) error {
if err := coin.Validate(); err != nil {
return err
}
if types.HasUTokenPrefix(coin.Denom) {
return types.ErrUToken.Wrap(coin.Denom)
}
return nil
}
// validateUToken validates an sdk.Coin and ensures its Denom is a uToken.
func validateUToken(coin sdk.Coin) error {
if err := coin.Validate(); err != nil {
return err
}
if !types.HasUTokenPrefix(coin.Denom) {
return types.ErrNotUToken.Wrap(coin.Denom)
}
return nil
}
// validateBaseDenom validates a denom and ensures it is not a uToken.
func validateBaseDenom(denom string) error {
if err := sdk.ValidateDenom(denom); err != nil {
return err
}
if types.HasUTokenPrefix(denom) {
return types.ErrUToken.Wrap(denom)
}
return nil
}