-
Notifications
You must be signed in to change notification settings - Fork 40
/
msg.go
118 lines (102 loc) · 3.58 KB
/
msg.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
package freeze
import (
"encoding/json"
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/bnb-chain/node/common/types"
"github.com/bnb-chain/node/common/upgrade"
)
// TODO: "route expressions can only contain alphanumeric characters", we need to change the cosmos sdk to support slash
// const FreezeRoute = "tokens/freeze"
const FreezeRoute = "tokensFreeze"
var _ sdk.Msg = FreezeMsg{}
type FreezeMsg struct {
From sdk.AccAddress `json:"from"`
Symbol string `json:"symbol"`
Amount int64 `json:"amount"`
}
func NewFreezeMsg(from sdk.AccAddress, symbol string, amount int64) FreezeMsg {
return FreezeMsg{
From: from,
Symbol: symbol,
Amount: amount,
}
}
func (msg FreezeMsg) Route() string { return FreezeRoute }
func (msg FreezeMsg) Type() string { return FreezeRoute }
func (msg FreezeMsg) String() string {
return fmt.Sprintf("Freeze{%v#%v%v}", msg.From, msg.Amount, msg.Symbol)
}
func (msg FreezeMsg) GetInvolvedAddresses() []sdk.AccAddress { return msg.GetSigners() }
func (msg FreezeMsg) GetSigners() []sdk.AccAddress { return []sdk.AccAddress{msg.From} }
// ValidateBasic does a simple validation check that
// doesn't require access to any other information.
func (msg FreezeMsg) ValidateBasic() sdk.Error {
if sdk.IsUpgrade(upgrade.BEP8) && types.IsValidMiniTokenSymbol(msg.Symbol) {
if msg.Amount <= 0 {
return sdk.ErrInsufficientFunds("amount should be more than 0")
}
return nil
}
// if BEP8 not upgraded, we rely on `ValidateTokenSymbol` rejecting the MiniToken.
// expect all msgs that reference a token after issue to use the suffixed form (e.g. "BNB-ABC")
err := types.ValidateTokenSymbol(msg.Symbol)
if err != nil {
return sdk.ErrInvalidCoins(err.Error())
}
if msg.Amount <= 0 {
// TODO: maybe we need to define our own errors
return sdk.ErrInsufficientFunds("amount should be more than 0")
}
return nil
}
func (msg FreezeMsg) GetSignBytes() []byte {
b, err := json.Marshal(msg) // XXX: ensure some canonical form
if err != nil {
panic(err)
}
return b
}
var _ sdk.Msg = UnfreezeMsg{}
type UnfreezeMsg struct {
From sdk.AccAddress `json:"from"`
Symbol string `json:"symbol"`
Amount int64 `json:"amount"`
}
func NewUnfreezeMsg(from sdk.AccAddress, symbol string, amount int64) UnfreezeMsg {
return UnfreezeMsg{
From: from, Symbol: symbol, Amount: amount}
}
func (msg UnfreezeMsg) Route() string { return FreezeRoute }
func (msg UnfreezeMsg) Type() string { return FreezeRoute }
func (msg UnfreezeMsg) String() string {
return fmt.Sprintf("Unfreeze{%v#%v%v}", msg.From, msg.Amount, msg.Symbol)
}
func (msg UnfreezeMsg) GetInvolvedAddresses() []sdk.AccAddress { return msg.GetSigners() }
func (msg UnfreezeMsg) GetSigners() []sdk.AccAddress { return []sdk.AccAddress{msg.From} }
func (msg UnfreezeMsg) ValidateBasic() sdk.Error {
if sdk.IsUpgrade(upgrade.BEP8) && types.IsValidMiniTokenSymbol(msg.Symbol) {
if msg.Amount <= 0 {
return sdk.ErrInsufficientFunds("amount should be more than 0")
}
return nil
}
// if BEP8 not upgraded, we rely on `ValidateTokenSymbol` rejecting the MiniToken.
// expect all msgs that reference a token after issue to use the suffixed form (e.g. "BNB-ABC")
err := types.ValidateTokenSymbol(msg.Symbol)
if err != nil {
return sdk.ErrInvalidCoins(err.Error())
}
if msg.Amount <= 0 {
// TODO: maybe we need to define our own errors
return sdk.ErrInsufficientFunds("amount should be more than 0")
}
return nil
}
func (msg UnfreezeMsg) GetSignBytes() []byte {
b, err := json.Marshal(msg) // XXX: ensure some canonical form
if err != nil {
panic(err)
}
return b
}