Add useragent header

replicatedhq · Jul 24, 2019 · 93e9592 · 93e9592
1 parent ad14283
commit 93e9592
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/cmd/preflight/cli/run_nocrd.go b/cmd/preflight/cli/run_nocrd.go
@@ -44,7 +44,12 @@ func runPreflightsNoCRD(v *viper.Viper, arg string) error {
 
 		preflightContent = string(b)
 	} else {
-		resp, err := http.Get(arg)
+		req, err := http.NewRequest("GET", arg, nil)
+		if err != nil {
+			return err
+		}
+		req.Header.Set("User-Agent", "Replicated_Preflight/v1beta1")
+		resp, err := http.DefaultClient.Do(req)
 		if err != nil {
 			return err
 		}

diff --git a/cmd/troubleshoot/cli/run_nocrd.go b/cmd/troubleshoot/cli/run_nocrd.go
@@ -43,7 +43,12 @@ func runTroubleshootNoCRD(v *viper.Viper, arg string) error {
 
 		collectorContent = string(b)
 	} else {
-		resp, err := http.Get(arg)
+		req, err := http.NewRequest("GET", arg, nil)
+		if err != nil {
+			return err
+		}
+		req.Header.Set("User-Agent", "Replicated_Troubleshoot/v1beta1")
+		resp, err := http.DefaultClient.Do(req)
 		if err != nil {
 			return err
 		}