Merge pull request #76 from flynn/user-agent

client: Add optional custom User-Agent to HTTPRemoteStore
theupdateframework · Mar 13, 2015 · 1882ab4 · 1882ab4
2 parents 22a1f44 + 3576627
commit 1882ab4
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/client/remote_store.go b/client/remote_store.go
@@ -13,6 +13,7 @@ import (
 type HTTPRemoteOptions struct {
 	MetadataPath string
 	TargetsPath  string
+	UserAgent    string
 }
 
 func HTTPRemoteStore(baseURL string, opts *HTTPRemoteOptions) (RemoteStore, error) {
@@ -40,7 +41,14 @@ func (h *httpRemoteStore) GetTarget(name string) (io.ReadCloser, int64, error) {
 
 func (h *httpRemoteStore) get(s string) (io.ReadCloser, int64, error) {
 	u := h.url(s)
-	res, err := http.Get(u)
+	req, err := http.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, 0, err
+	}
+	if h.opts.UserAgent != "" {
+		req.Header.Set("User-Agent", h.opts.UserAgent)
+	}
+	res, err := http.DefaultClient.Do(req)
 	if err != nil {
 		return nil, 0, err
 	}