user agent

Author: sestinj

internal/registry/registry.go

@@ -10,6 +10,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"rules-cli/internal/utils"
 	"strings"
 )
 
@@ -68,6 +69,8 @@ func (c *Client) GetRule(name, version string) (*RuleInfo, error) {
 		req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))
 	}
 
+	utils.SetUserAgent(req)
+	
 	client := &http.Client{}
 	resp, err := client.Do(req)
 	if err != nil {
@@ -110,6 +113,8 @@ func (c *Client) DownloadRule(name, version, formatDir string) error {
 		req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))
 	}
 
+	utils.SetUserAgent(req)
+	
 	client := &http.Client{}
 	resp, err := client.Do(req)
 	if err != nil {
@@ -167,6 +172,8 @@ func (c *Client) PublishRule(ownerSlug, ruleSlug, content string, visibility str
 	req.Header.Set("Content-Type", "application/json")
 	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))
 
+	utils.SetUserAgent(req)
+	
 	client := &http.Client{}
 	resp, err := client.Do(req)
 	if err != nil {
@@ -193,6 +200,7 @@ func (c *Client) downloadFromGitHub(repoPath string, formatDir string) error {
 		return fmt.Errorf("failed to create request: %w", err)
 	}
 	req.Header.Set("Accept", "application/vnd.github.v3+json")
+	utils.SetUserAgent(req)
 
 	// Send request
 	client := &http.Client{}
@@ -303,4 +311,4 @@ func (c *Client) downloadFromGitHub(repoPath string, formatDir string) error {
 	}
 
 	return nil
-}
+}

internal/utils/user_agent.go

@@ -0,0 +1,27 @@
+package utils
+
+import (
+	"fmt"
+	"net/http"
+	"runtime"
+)
+
+// GetUserAgent returns the user agent string for HTTP requests
+// This includes the CLI name, version, and runtime information
+func GetUserAgent() string {
+	// For now, use a hardcoded version. In the future, this could be injected
+	// during the build process via ldflags
+	version := "1.0.0"
+	
+	// Include runtime information for better analytics
+	return fmt.Sprintf("rules-cli/%s (%s; %s; %s)", 
+		version, 
+		runtime.GOOS, 
+		runtime.GOARCH, 
+		runtime.Version())
+}
+
+// SetUserAgent sets the User-Agent header on an HTTP request
+func SetUserAgent(req *http.Request) {
+	req.Header.Set("User-Agent", GetUserAgent())
+}