/
types.go
59 lines (48 loc) · 1.72 KB
/
types.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 messages
import (
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/forbole/juno/v5/types"
)
// MessageAddressesParser represents a function that extracts all the
// involved addresses from a provided message (both accounts and validators)
type MessageAddressesParser = func(tx *types.Tx) ([]string, error)
// CosmosMessageAddressesParser represents a MessageAddressesParser that parses a
// Chain message and returns all the involved addresses (both accounts and validators)
var CosmosMessageAddressesParser = DefaultMessagesParser
// DefaultMessagesParser represents the default messages parser that simply returns the list
// of all the signers of a message
func DefaultMessagesParser(tx *types.Tx) ([]string, error) {
return parseAddressesFromEvents(tx), nil
}
// function to remove duplicate values
func removeDuplicates(s []string) []string {
bucket := make(map[string]bool)
var result []string
for _, str := range s {
if _, ok := bucket[str]; !ok {
bucket[str] = true
result = append(result, str)
}
}
return result
}
func parseAddressesFromEvents(tx *types.Tx) []string {
var addresses []string
for _, event := range tx.Events {
for _, attribute := range event.Attributes {
// Try parsing the address as a validator address
validatorAddress, _ := sdk.ValAddressFromBech32(attribute.Value)
if validatorAddress != nil {
addresses = append(addresses, validatorAddress.String())
}
// Try parsing the address as an account address
accountAddress, err := sdk.AccAddressFromBech32(attribute.Value)
if err != nil {
// Skip if the address is not an account address
continue
}
addresses = append(addresses, accountAddress.String())
}
}
return removeDuplicates(addresses)
}