Skip to content

Commit 0dceced

Browse files
committed
fix(redteam): add network_id to relay KV requests
1 parent 82fa3c0 commit 0dceced

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

redteam/relay.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,21 +15,26 @@ import (
1515
var (
1616
relayURL string
1717
relayToken string
18+
networkID string
1819
relayEnabled bool
1920
relayClient = &http.Client{Timeout: 10 * time.Second}
2021
)
2122

2223
func initRelay() {
2324
relayURL = os.Getenv("RELAY_URL")
2425
relayToken = os.Getenv("CODEWIRE_TOKEN")
26+
networkID = os.Getenv("NETWORK_ID")
27+
if networkID == "" {
28+
networkID = "redteam"
29+
}
2530
relayEnabled = os.Getenv("CODEWIRE_ENABLED") != "false" && relayURL != "" && relayToken != ""
2631
}
2732

2833
func kvPut(key, value, ttl string) {
2934
if !relayEnabled {
3035
return
3136
}
32-
u := fmt.Sprintf("%s/api/v1/kv/redteam/%s", relayURL, key)
37+
u := fmt.Sprintf("%s/api/v1/kv/redteam/%s?network_id=%s", relayURL, key, url.QueryEscape(networkID))
3338
req, err := http.NewRequest("PUT", u, strings.NewReader(value))
3439
if err != nil {
3540
return
@@ -50,7 +55,7 @@ func kvGet(prefix string) string {
5055
if !relayEnabled {
5156
return ""
5257
}
53-
u := fmt.Sprintf("%s/api/v1/kv/redteam?prefix=%s", relayURL, url.QueryEscape(prefix))
58+
u := fmt.Sprintf("%s/api/v1/kv/redteam?prefix=%s&network_id=%s", relayURL, url.QueryEscape(prefix), url.QueryEscape(networkID))
5459
req, err := http.NewRequest("GET", u, nil)
5560
if err != nil {
5661
return ""

0 commit comments

Comments
 (0)