Set User-Agent

cmd/mstdn/main.go

@@ -370,6 +370,7 @@ func run() int {
 		}
 
 		client := mastodon.NewClient(config)
+		client.UserAgent = "mstdn"
 		app.Metadata = map[string]interface{}{
 			"client":      client,
 			"config":      config,

mastodon.go

@@ -32,6 +32,7 @@ type Config struct {
 type Client struct {
 	http.Client
 	config    *Config
+	UserAgent string
 }
 
 func (c *Client) doAPI(ctx context.Context, method string, uri string, params interface{}, res interface{}, pg *Pagination) error {
@@ -117,6 +118,9 @@ func (c *Client) doAPI(ctx context.Context, method string, uri string, params in
 	if params != nil {
 		req.Header.Set("Content-Type", ct)
 	}
+	if c.UserAgent != "" {
+		req.Header.Set("User-Agent", c.UserAgent)
+	}
 
 	var resp *http.Response
 	backoff := 1000 * time.Millisecond
@@ -211,6 +215,9 @@ func (c *Client) authenticate(ctx context.Context, params url.Values) error {
 	}
 	req = req.WithContext(ctx)
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	if c.UserAgent != "" {
+		req.Header.Set("User-Agent", c.UserAgent)
+	}
 	resp, err := c.Do(req)
 	if err != nil {
 		return err