From 90b05bac5fbd1c5209c32b0e86a67a885c938b26 Mon Sep 17 00:00:00 2001
From: AJ ONeal <coolaj86@gmail.com>
Date: Fri, 8 Mar 2019 14:28:23 -0700
Subject: [PATCH] add user agent

---
 keyfetch/fetch_test.go     |  2 ++
 keyfetch/uncached/fetch.go | 10 +++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/keyfetch/fetch_test.go b/keyfetch/fetch_test.go
index aa8cedc..76d2ffe 100644
--- a/keyfetch/fetch_test.go
+++ b/keyfetch/fetch_test.go
@@ -12,6 +12,8 @@ func TestCachesKey(t *testing.T) {
 	testCachesKey(t, "https://bigsquid.auth0.com/")
 	clear()
 	testCachesKey(t, "https://bigsquid.auth0.com")
+	clear()
+	testCachesKey(t, "https://big-squid.github.io/")
 }
 
 func testCachesKey(t *testing.T, url string) {
diff --git a/keyfetch/uncached/fetch.go b/keyfetch/uncached/fetch.go
index 11309b3..e909bab 100644
--- a/keyfetch/uncached/fetch.go
+++ b/keyfetch/uncached/fetch.go
@@ -36,8 +36,9 @@ func OIDCJWKs(baseURL string) (map[string]map[string]string, map[string]keypairs
 // WellKnownJWKs calls JWKs with baseURL + /.well-known/jwks.json as constructs the jwks_uri
 func WellKnownJWKs(baseURL string) (map[string]map[string]string, map[string]keypairs.PublicKey, error) {
 	baseURL = normalizeBaseURL(baseURL)
+	url := baseURL + ".well-known/jwks.json"
 
-	return JWKs(baseURL + ".well-known/jwks.json")
+	return JWKs(url)
 }
 
 // JWKs fetches and parses a jwks.json (assuming well-known format)
@@ -121,12 +122,15 @@ func safeFetch(url string, decoder decodeFunc) error {
 		}).Dial,
 		TLSHandshakeTimeout: 5 * time.Second,
 	}
-	var netClient = &http.Client{
+	var client = &http.Client{
 		Timeout:   time.Second * 10,
 		Transport: netTransport,
 	}
 
-	res, err := netClient.Get(url)
+	req, err := http.NewRequest("GET", url, nil)
+	req.Header.Set("User-Agent", "go-keypairs/keyfetch")
+	req.Header.Set("Accept", "application/json;q=0.9,*/*;q=0.8")
+	res, err := client.Do(req)
 	if nil != err {
 		return err
 	}