/
msgs.go
117 lines (95 loc) · 2.63 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
package types
import (
"encoding/hex"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
)
const (
TypeMsgCreate = "create"
TypeMsgClaim = "claim"
)
var _ sdk.Msg = &MsgCreate{}
func NewMsgCreate(owner sdk.AccAddress, merkleRoot string, startHeight, endHeight int64, coin sdk.Coin) *MsgCreate {
return &MsgCreate{
Owner: owner.String(),
MerkleRoot: merkleRoot,
StartHeight: startHeight,
EndHeight: endHeight,
Coin: coin,
}
}
func (msg MsgCreate) Route() string { return RouterKey }
func (msg MsgCreate) Type() string { return TypeMsgCreate }
func (msg MsgCreate) ValidateBasic() error {
if msg.EndHeight <= msg.StartHeight {
return sdkerrors.Wrapf(ErrInvalidEndHeight, "end height must be > start height")
}
if err := msg.Coin.Validate(); err != nil {
return err
}
if msg.Coin.Amount.LTE(sdk.ZeroInt()) {
return sdkerrors.Wrapf(ErrInvalidCoin, "invalid coin amount, less then zero")
}
_, err := sdk.AccAddressFromBech32(msg.Owner)
if err != nil {
return sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "invalid owner address (%s)", err)
}
_, err = hex.DecodeString(msg.MerkleRoot)
if err != nil {
return sdkerrors.Wrapf(ErrInvalidMerkleRoot, "invalid merkle root (%s)", err)
}
return nil
}
// GetSignBytes Implements Msg.
func (msg MsgCreate) GetSignBytes() []byte {
b, err := ModuleCdc.MarshalJSON(&msg)
if err != nil {
panic(err)
}
return sdk.MustSortJSON(b)
}
// GetSigners Implements Msg.
func (msg MsgCreate) GetSigners() []sdk.AccAddress {
owner, err := sdk.AccAddressFromBech32(msg.Owner)
if err != nil {
panic(err)
}
return []sdk.AccAddress{owner}
}
var _ sdk.Msg = &MsgClaim{}
func NewMsgClaim(index, mdId uint64, amount sdk.Int, proofs []string, sender sdk.AccAddress) *MsgClaim {
return &MsgClaim{
Index: index,
MerkledropId: mdId,
Amount: amount,
Proofs: proofs,
Sender: sender.String(),
}
}
func (msg MsgClaim) Route() string { return RouterKey }
func (msg MsgClaim) Type() string { return TypeMsgClaim }
func (msg MsgClaim) ValidateBasic() error {
for _, p := range msg.Proofs {
_, err := hex.DecodeString(p)
if err != nil {
return sdkerrors.Wrapf(ErrInvalidMerkleRoot, "invalid merkle proof (%s)", err)
}
}
return nil
}
// GetSignBytes Implements Msg.
func (msg MsgClaim) GetSignBytes() []byte {
b, err := ModuleCdc.MarshalJSON(&msg)
if err != nil {
panic(err)
}
return sdk.MustSortJSON(b)
}
// GetSigners Implements Msg.
func (msg MsgClaim) GetSigners() []sdk.AccAddress {
sender, err := sdk.AccAddressFromBech32(msg.Sender)
if err != nil {
panic(err)
}
return []sdk.AccAddress{sender}
}