-
Notifications
You must be signed in to change notification settings - Fork 2
/
handler_map.go
59 lines (49 loc) · 1.71 KB
/
handler_map.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
package signing
import (
"fmt"
ibctx "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types/ibc-adapter"
"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types/tx/signing"
)
// SignModeHandlerMap is SignModeHandler that aggregates multiple SignModeHandler's into
// a single handler
type SignModeHandlerMap struct {
defaultMode signing.SignMode
modes []signing.SignMode
signModeHandlers map[signing.SignMode]SignModeHandler
}
var _ SignModeHandler = SignModeHandlerMap{}
// NewSignModeHandlerMap returns a new SignModeHandlerMap with the provided defaultMode and handlers
func NewSignModeHandlerMap(defaultMode signing.SignMode, handlers []SignModeHandler) SignModeHandlerMap {
handlerMap := make(map[signing.SignMode]SignModeHandler)
var modes []signing.SignMode
for _, h := range handlers {
for _, m := range h.Modes() {
if _, have := handlerMap[m]; have {
panic(fmt.Errorf("duplicate sign mode handler for mode %s", m))
}
handlerMap[m] = h
modes = append(modes, m)
}
}
return SignModeHandlerMap{
defaultMode: defaultMode,
modes: modes,
signModeHandlers: handlerMap,
}
}
// DefaultMode implements SignModeHandler.DefaultMode
func (h SignModeHandlerMap) DefaultMode() signing.SignMode {
return h.defaultMode
}
// Modes implements SignModeHandler.Modes
func (h SignModeHandlerMap) Modes() []signing.SignMode {
return h.modes
}
// DefaultMode implements SignModeHandler.GetSignBytes
func (h SignModeHandlerMap) GetSignBytes(mode signing.SignMode, data SignerData, tx ibctx.Tx) ([]byte, error) {
handler, found := h.signModeHandlers[mode]
if !found {
return nil, fmt.Errorf("can't verify sign mode %s", mode.String())
}
return handler.GetSignBytes(mode, data, tx)
}