support custom user agent

Author: zachmann

context.go

@@ -0,0 +1,27 @@
+package mytokenlib
+
+import (
+	"context"
+	"net/http"
+)
+
+var ctx = context.Background()
+var httpClient = &http.Client{}
+var userAgent = "mytokenlib"
+
+// ContextKeyUserAgent is used to set a useragent string in the context
+const ContextKeyUserAgent = "mytokenlib-user-agent"
+
+// SetClient sets the http.Client used to make API requests
+func SetClient(client *http.Client) {
+	httpClient = client
+}
+
+// SetContext sets a context.Context used for all API requests
+func SetContext(contxt context.Context) {
+	ctx = contxt
+	s, ok := ctx.Value(ContextKeyUserAgent).(string)
+	if ok {
+		userAgent = s
+	}
+}

http.go

@@ -33,6 +33,9 @@ func doHTTPRequestWithAuth(
 	if err != nil {
 		return newMytokenErrorFromError(errSendingHttpRequest, err)
 	}
+	if userAgent != "" {
+		req.Header.Set("User-Agent", userAgent)
+	}
 	if reqBody != nil {
 		req.Header.Set("Content-Type", mimetypeJSON)
 	}

server.go

@@ -1,9 +1,6 @@
 package mytokenlib
 
 import (
-	"context"
-	"net/http"
-
 	"github.com/oidc-mytoken/api/v0"
 )
 
@@ -23,9 +20,6 @@ type Endpoint interface {
 	DoHTTPRequest(method string, req interface{}, resp interface{}) error
 }
 
-var httpClient = &http.Client{}
-var ctx = context.Background()
-
 // NewMytokenServer creates a new MytokenServer
 func NewMytokenServer(url string) (*MytokenServer, error) {
 	configEndpoint := url
@@ -52,13 +46,3 @@ func NewMytokenServer(url string) (*MytokenServer, error) {
 	}
 	return server, err
 }
-
-// SetClient sets the http.Client used to make API requests
-func SetClient(client *http.Client) {
-	httpClient = client
-}
-
-// SetContext sets a context.Context used for all API requests
-func SetContext(contxt context.Context) {
-	ctx = contxt
-}