add user-agent

Signed-off-by: Karl Piplies <[email protected]>
mercedes-benz · Jul 31, 2024 · 48343e4 · 48343e4
1 parent 0fb59d4
commit 48343e4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/conf/config.go b/conf/config.go
@@ -15,7 +15,10 @@ import (
 	"github.com/jinzhu/configor"
 )
 
-const DefaultApiVersion = "v1"
+const (
+	DefaultApiVersion = "v1"
+	CLIVersion        = "v0.7"
+)
 
 var Config = struct {
 	ProjectToken   string `default:""`
@@ -24,6 +27,10 @@ var Config = struct {
 	Host           string `default:""`
 }{}
 
+func UserAgent() string {
+	return "disclosure-cli / " + CLIVersion
+}
+
 func LoadConfig(configFileLocation string) {
 	if _, err := os.Stat(configFileLocation); errors.Is(err, os.ErrNotExist) {
 		fmt.Println("No config file available in location. Proceeding to check environment variables.")

diff --git a/disclosure-cli b/disclosure-cli
diff --git a/pkg/helper/apiRequests.go b/pkg/helper/apiRequests.go
@@ -57,6 +57,7 @@ func SbomUploadFormData(url string, completeFilename string, tag string) string
 	req.Header.Set("accept", "application/json")
 	req.Header.Set("Content-Type", m.FormDataContentType())
 	req.Header.Set("Authorization", "DISCO"+" "+conf.Config.ProjectToken)
+	req.Header.Set("User-Agent", conf.UserAgent())
 
 	resp, err := client.Do(req)
 	if err != nil {
@@ -96,6 +97,7 @@ func DiscoApiPost(url string, v interface{}) string {
 	req.Header.Set("accept", "application/json")
 	req.Header.Set("Content-Type", "application/json")
 	req.Header.Set("Authorization", "DISCO"+" "+conf.Config.ProjectToken)
+	req.Header.Set("User-Agent", conf.UserAgent())
 	resp, err := client.Do(req)
 	if err != nil {
 		fmt.Println("Error on requesting url " + url)
@@ -136,6 +138,7 @@ func DiscoApiPut(url string, v interface{}) string {
 	req.Header.Set("accept", "application/json")
 	req.Header.Set("Content-Type", "application/json")
 	req.Header.Set("Authorization", "DISCO"+" "+conf.Config.ProjectToken)
+	req.Header.Set("User-Agent", conf.UserAgent())
 	resp, err := client.Do(req)
 	if err != nil {
 		fmt.Printf("\n Error on requesting url %s \n Error: %s", url, err.Error())
@@ -168,6 +171,7 @@ func DiscoApiGet(url string) string {
 
 	req.Header.Set("accept", "application/json")
 	req.Header.Set("Authorization", "DISCO"+" "+conf.Config.ProjectToken)
+	req.Header.Set("User-Agent", conf.UserAgent())
 	resp, err := client.Do(req)
 	if err != nil {
 		fmt.Printf("\n Error on requesting url %s \n Error: %s ", url, err.Error())