Merge #27

Add ability to change user agent.

Author: melbahja

cmd/got/main.go

@@ -72,9 +72,14 @@ func main() {
 			},
 			&cli.StringSliceFlag{
 				Name:    "header",
-				Usage:   "Set these HTTP-Headers on the requests. The format has to be Key: Value",
+				Usage:   `Set these HTTP-Headers on the requests. The format has to be: -H "Key: Value"`,
 				Aliases: []string{"H"},
 			},
+			&cli.StringFlag{
+				Name:    "agent",
+				Usage:   `Set user agent for got HTTP requests.`,
+				Aliases: []string{"u"},
+			},
 		},
 		Version: version,
 		Authors: []*cli.Author{
@@ -147,6 +152,11 @@ func run(ctx context.Context, c *cli.Context) error {
 		}
 	}
 
+	// Set default user agent.
+	if c.String("agent") != "" {
+		got.UserAgent = c.String("agent")
+	}
+
 	// Piped stdin
 	if info.Mode()&os.ModeNamedPipe > 0 || info.Size() > 0 {
 

got.go

@@ -16,8 +16,8 @@ type Got struct {
 	ctx context.Context
 }
 
-// DefaulUserAgent is the default Got user agent to send http requests.
-const DefaulUserAgent = "Got/1.0"
+// UserAgent is the default Got user agent to send http requests.
+var UserAgent = "Got/1.0"
 
 // ErrDownloadAborted - When download is aborted by the OS before it is completed, ErrDownloadAborted will be triggered
 var ErrDownloadAborted = errors.New("Operation aborted")
@@ -82,7 +82,7 @@ func NewRequest(ctx context.Context, method, URL string, header []GotHeader) (re
 		return
 	}
 
-	req.Header.Set("User-Agent", DefaulUserAgent)
+	req.Header.Set("User-Agent", UserAgent)
 
 	for _, h := range header {
 		req.Header.Set(h.Key, h.Value)