Merge pull request #26 from replicatedhq/user-agent

Add useragent header
replicatedhq · Jul 24, 2019 · d9990f8 · d9990f8
2 parents ad14283 + 93e9592
commit d9990f8
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
 		}