-
Notifications
You must be signed in to change notification settings - Fork 43
/
msgs.go
123 lines (98 loc) · 3.2 KB
/
msgs.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
117
118
119
120
121
122
123
package types
import (
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
)
// ensure Msg interface compliance at compile time
var (
_ sdk.Msg = &MsgSwap{}
_ sdk.Msg = &MsgSwapSend{}
)
// market message types
const (
TypeMsgSwap = "swap"
TypeMsgSwapSend = "swap_send"
)
//--------------------------------------------------------
//--------------------------------------------------------
// NewMsgSwap creates a MsgSwap instance
func NewMsgSwap(traderAddress sdk.AccAddress, offerCoin sdk.Coin, askCoin string) *MsgSwap {
return &MsgSwap{
Trader: traderAddress.String(),
OfferCoin: offerCoin,
AskDenom: askCoin,
}
}
// Route Implements Msg
func (msg MsgSwap) Route() string { return RouterKey }
// Type implements sdk.Msg
func (msg MsgSwap) Type() string { return TypeMsgSwap }
// GetSignBytes Implements Msg
func (msg MsgSwap) GetSignBytes() []byte {
return sdk.MustSortJSON(ModuleCdc.MustMarshalJSON(&msg))
}
// GetSigners Implements Msg
func (msg MsgSwap) GetSigners() []sdk.AccAddress {
trader, err := sdk.AccAddressFromBech32(msg.Trader)
if err != nil {
panic(err)
}
return []sdk.AccAddress{trader}
}
// ValidateBasic Implements Msg
func (msg MsgSwap) ValidateBasic() error {
_, err := sdk.AccAddressFromBech32(msg.Trader)
if err != nil {
return sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "Invalid trader address (%s)", err)
}
if msg.OfferCoin.Amount.LTE(sdk.ZeroInt()) || msg.OfferCoin.Amount.BigInt().BitLen() > 100 {
return sdkerrors.Wrap(sdkerrors.ErrInvalidCoins, msg.OfferCoin.String())
}
if msg.OfferCoin.Denom == msg.AskDenom {
return sdkerrors.Wrap(ErrRecursiveSwap, msg.AskDenom)
}
return nil
}
// NewMsgSwapSend conducts market swap and send all the result coins to recipient
func NewMsgSwapSend(fromAddress sdk.AccAddress, toAddress sdk.AccAddress, offerCoin sdk.Coin, askCoin string) *MsgSwapSend {
return &MsgSwapSend{
FromAddress: fromAddress.String(),
ToAddress: toAddress.String(),
OfferCoin: offerCoin,
AskDenom: askCoin,
}
}
// Route Implements Msg
func (msg MsgSwapSend) Route() string { return RouterKey }
// Type implements sdk.Msg
func (msg MsgSwapSend) Type() string { return TypeMsgSwapSend }
// GetSignBytes Implements Msg
func (msg MsgSwapSend) GetSignBytes() []byte {
return sdk.MustSortJSON(ModuleCdc.MustMarshalJSON(&msg))
}
// GetSigners Implements Msg
func (msg MsgSwapSend) GetSigners() []sdk.AccAddress {
from, err := sdk.AccAddressFromBech32(msg.FromAddress)
if err != nil {
panic(err)
}
return []sdk.AccAddress{from}
}
// ValidateBasic Implements Msg
func (msg MsgSwapSend) ValidateBasic() error {
_, err := sdk.AccAddressFromBech32(msg.FromAddress)
if err != nil {
return sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "Invalid from address (%s)", err)
}
_, err = sdk.AccAddressFromBech32(msg.ToAddress)
if err != nil {
return sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "Invalid to address (%s)", err)
}
if msg.OfferCoin.Amount.LTE(sdk.ZeroInt()) || msg.OfferCoin.Amount.BigInt().BitLen() > 100 {
return sdkerrors.Wrap(sdkerrors.ErrInvalidCoins, msg.OfferCoin.String())
}
if msg.OfferCoin.Denom == msg.AskDenom {
return sdkerrors.Wrap(ErrRecursiveSwap, msg.AskDenom)
}
return nil
}