feat: [ML-1270]: Add user-agent header for tracking (#189)

* feat: [ML-1270]: Add user-agent header for tracking

Author: David-Kreiner

client/client.go

@@ -47,6 +47,9 @@ type Client struct {
 
 	// AuthProvider used for authentication
 	AuthProvider auth.Provider
+
+	// UserAgent is the User-Agent header string sent with requests
+	UserAgent string
 }
 
 type service struct {
@@ -64,16 +67,25 @@ func defaultHTTPClient(timeout ...time.Duration) *http.Client {
 	}
 }
 
-// NewWithToken creates a new client with the specified base URL and API token
-func NewWithAuthProvider(uri string, authProvider auth.Provider, timeout ...time.Duration) (*Client, error) {
+// NewWithAuthProvider creates a new client with the specified base URL, auth provider, and optional version
+// If version is an empty string, "unknown" will be used in the User-Agent header
+func NewWithAuthProvider(uri string, authProvider auth.Provider, version string, timeout ...time.Duration) (*Client, error) {
 	parsedURL, err := url.Parse(uri)
 	if err != nil {
 		return nil, err
 	}
+
+	versionStr := "unknown"
+	if version != "" {
+		versionStr = version
+	}
+	userAgent := fmt.Sprintf("harness-mcp-client/%s", versionStr)
+
 	c := &Client{
 		client:       defaultHTTPClient(timeout...),
 		BaseURL:      parsedURL,
 		AuthProvider: authProvider,
+		UserAgent:    userAgent,
 	}
 	return c, nil
 }
@@ -474,6 +486,8 @@ func (c *Client) Do(r *http.Request) (*http.Response, error) {
 	}
 	r.Header.Set(k, v)
 
+	r.Header.Set("User-Agent", c.UserAgent)
+
 	// Check for scope in context and add account ID to headers if present
 	if scope, err := common.GetScopeFromContext(ctx); err == nil {
 		if scope.AccountID != "" {

pkg/modules/har.go

@@ -80,10 +80,10 @@ func RegisterRegistries(config *config.Config, tsg *toolsets.ToolsetGroup) error
 
 	if config.Internal {
 		authProvider := auth.NewJWTProvider(secret, "Basic", &utils.DefaultJWTLifetime)
-		c, err = client.NewWithAuthProvider(baseURL, authProvider)
+		c, err = client.NewWithAuthProvider(baseURL, authProvider, config.Version)
 	} else {
 		authProvider := auth.NewAPIKeyProvider(config.APIKey)
-		c, err = client.NewWithAuthProvider(baseURL, authProvider)
+		c, err = client.NewWithAuthProvider(baseURL, authProvider, config.Version)
 	}
 
 	if err != nil {

pkg/modules/utils/clients.go

@@ -50,7 +50,7 @@ func CreateClientWithIdentity(baseURL string, config *config.Config, secret stri
 		authProvider = auth.NewAPIKeyProvider(config.APIKey)
 	}
 
-	client, err := client.NewWithAuthProvider(baseURL, authProvider, timeout...)
+	client, err := client.NewWithAuthProvider(baseURL, authProvider, config.Version, timeout...)
 	if err != nil {
 		slog.Error("Failed to create client", "error", err)
 		return nil, fmt.Errorf("failed to create client: %w", err)