-
Notifications
You must be signed in to change notification settings - Fork 17
/
config_chain.go
133 lines (120 loc) · 4.12 KB
/
config_chain.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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
package monitoring
import (
"fmt"
"net/url"
"os"
"time"
"github.com/NethermindEth/juno/core/felt"
starknetutils "github.com/NethermindEth/starknet.go/utils"
relayMonitoring "github.com/smartcontractkit/chainlink-common/pkg/monitoring"
)
type StarknetConfig struct {
rpcEndpoint string
rpcApiKey string
networkName string
networkID string
chainID string
readTimeout time.Duration
pollInterval time.Duration
linkTokenAddress string
strkTokenAddress *felt.Felt
}
var _ relayMonitoring.ChainConfig = StarknetConfig{}
func (s StarknetConfig) GetRPCEndpoint() string { return s.rpcEndpoint }
func (s StarknetConfig) GetRPCApiKey() string { return s.rpcApiKey }
func (s StarknetConfig) GetNetworkName() string { return s.networkName }
func (s StarknetConfig) GetNetworkID() string { return s.networkID }
func (s StarknetConfig) GetChainID() string { return s.chainID }
func (s StarknetConfig) GetReadTimeout() time.Duration { return s.readTimeout }
func (s StarknetConfig) GetPollInterval() time.Duration { return s.pollInterval }
func (s StarknetConfig) GetLinkTokenAddress() string { return s.linkTokenAddress }
func (s StarknetConfig) GetStrkTokenAddress() *felt.Felt { return s.strkTokenAddress }
func (s StarknetConfig) ToMapping() map[string]interface{} {
return map[string]interface{}{
"network_name": s.networkName,
"network_id": s.networkID,
"chain_id": s.chainID,
}
}
func ParseStarknetConfig() (StarknetConfig, error) {
cfg := StarknetConfig{}
if err := parseEnvVars(&cfg); err != nil {
return cfg, err
}
applyDefaults(&cfg)
err := validateConfig(cfg)
return cfg, err
}
func parseEnvVars(cfg *StarknetConfig) error {
if value, isPresent := os.LookupEnv("STARKNET_RPC_ENDPOINT"); isPresent {
cfg.rpcEndpoint = value
}
if value, isPresent := os.LookupEnv("STARKNET_RPC_API_KEY"); isPresent {
cfg.rpcApiKey = value
}
if value, isPresent := os.LookupEnv("STARKNET_NETWORK_NAME"); isPresent {
cfg.networkName = value
}
if value, isPresent := os.LookupEnv("STARKNET_NETWORK_ID"); isPresent {
cfg.networkID = value
}
if value, isPresent := os.LookupEnv("STARKNET_CHAIN_ID"); isPresent {
cfg.chainID = value
}
if value, isPresent := os.LookupEnv("STARKNET_READ_TIMEOUT"); isPresent {
readTimeout, err := time.ParseDuration(value)
if err != nil {
return fmt.Errorf("failed to parse env var STARKNET_READ_TIMEOUT, see https://pkg.go.dev/time#ParseDuration: %w", err)
}
cfg.readTimeout = readTimeout
}
if value, isPresent := os.LookupEnv("STARKNET_POLL_INTERVAL"); isPresent {
pollInterval, err := time.ParseDuration(value)
if err != nil {
return fmt.Errorf("failed to parse env var STARKNET_POLL_INTERVAL, see https://pkg.go.dev/time#ParseDuration: %w", err)
}
cfg.pollInterval = pollInterval
}
if value, isPresent := os.LookupEnv("STARKNET_LINK_TOKEN_ADDRESS"); isPresent {
cfg.linkTokenAddress = value
}
if value, isPresent := os.LookupEnv("STRK_TOKEN_ADDRESS"); isPresent {
feltValue, err := starknetutils.HexToFelt(value)
if err != nil {
return fmt.Errorf("failed to parse env var STRK_TOKEN_ADDRESS %w", err)
}
cfg.strkTokenAddress = feltValue
}
return nil
}
func validateConfig(cfg StarknetConfig) error {
// Required config
for envVarName, currentValue := range map[string]string{
"STARKNET_RPC_ENDPOINT": cfg.rpcEndpoint,
"STARKNET_NETWORK_NAME": cfg.networkName,
"STARKNET_NETWORK_ID": cfg.networkID,
"STARKNET_CHAIN_ID": cfg.chainID,
"STARKNET_LINK_TOKEN_ADDRESS": cfg.linkTokenAddress,
} {
if currentValue == "" {
return fmt.Errorf("'%s' env var is required", envVarName)
}
}
// Validate URLs.
for envVarName, currentValue := range map[string]string{
"STARKNET_RPC_ENDPOINT": cfg.rpcEndpoint,
} {
if _, err := url.ParseRequestURI(currentValue); err != nil {
return fmt.Errorf("%s='%s' is not a valid URL: %w", envVarName, currentValue, err)
}
}
return nil
}
func applyDefaults(cfg *StarknetConfig) {
if cfg.readTimeout == 0 {
cfg.readTimeout = 2 * time.Second
}
if cfg.pollInterval == 0 {
cfg.pollInterval = 5 * time.Second
}
}