-
Notifications
You must be signed in to change notification settings - Fork 3
/
abiCodec.go
106 lines (95 loc) · 2.98 KB
/
abiCodec.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
package rpc
import (
"encoding/json"
"strings"
"github.com/zapproject/pythia/contracts1"
"github.com/ethereum/go-ethereum/accounts/abi"
"github.com/ethereum/go-ethereum/common/hexutil"
"github.com/zapproject/pythia/contracts"
"github.com/zapproject/pythia/util"
)
var abiCodecLog = util.NewLogger("rpc", "ABICodec")
//ABICodec holds abi definitions for encoding/decoding contract methods and events
type ABICodec struct {
abiStruct abi.ABI
methods map[string]*abi.Method
Events map[string]*abi.Event
}
//BuildCodec constructs a merged abi structure representing all methods/events for Zap contracts. This is primarily
//used for mock encoding/decoding parameters but could also be used for manual RPC operations that do not rely on geth's contract impl
func BuildCodec() (*ABICodec, error) {
all := []string{
contracts.ZapDisputeABI,
contracts.ZapGettersABI,
contracts.ZapGettersLibraryABI,
contracts.ZapStakeABI}
parsed := make([]interface{}, 0)
for _, abi := range all {
var f interface{}
if err := json.Unmarshal([]byte(abi), &f); err != nil {
return nil, err
}
asList := f.([]interface{})
for _, parsedABI := range asList {
parsed = append(parsed, parsedABI)
}
}
j, err := json.Marshal(parsed)
if err != nil {
return nil, err
}
abiStruct, err := abi.JSON(strings.NewReader(string(j)))
if err != nil {
return nil, err
}
methodMap := make(map[string]*abi.Method)
eventMap := make(map[string]*abi.Event)
for _, a := range abiStruct.Methods {
sig := hexutil.Encode(a.ID)
abiCodecLog.Debug("Mapping method sig: %s to method: %s", sig, a.Name)
methodMap[sig] = &abi.Method{Name: a.Name, Constant: a.Constant, Inputs: a.Inputs, Outputs: a.Outputs}
}
for _, e := range abiStruct.Events {
sig := hexutil.Encode(e.ID.Bytes())
abiCodecLog.Debug("Mapping event sig: %s to event %s", sig, e.Name)
eventMap[sig] = &abi.Event{Name: e.Name, Anonymous: e.Anonymous, Inputs: e.Inputs}
}
return &ABICodec{abiStruct, methodMap, eventMap}, nil
}
//this is helpful for debugging, lets you quickly find the type of each event
func AllEvents() (map[[32]byte]abi.Event, error) {
all := []string{
contracts1.ZapABI,
contracts.ZapDisputeABI,
contracts.ZapGettersABI,
contracts.ZapGettersLibraryABI,
contracts1.ZapLibraryABI,
contracts.ZapMasterABI,
contracts.ZapStakeABI,
contracts.ZapStorageABI,
}
parsed := make([]interface{}, 0)
for _, abi := range all {
var f interface{}
if err := json.Unmarshal([]byte(abi), &f); err != nil {
return nil, err
}
asList := f.([]interface{})
for _, parsedABI := range asList {
parsed = append(parsed, parsedABI)
}
}
j, err := json.Marshal(parsed)
if err != nil {
return nil, err
}
abiStruct, err := abi.JSON(strings.NewReader(string(j)))
if err != nil {
return nil, err
}
eventMap := make(map[[32]byte]abi.Event)
for _, e := range abiStruct.Events {
eventMap[e.ID] = e
}
return eventMap, nil
}