enable supplying extra headers to the authenticate transport

Author: postables

shell.go

@@ -115,11 +115,11 @@ func NewShellWithClient(url string, c *gohttp.Client) *Shell {
 //        WithAuthorization(token).
 //        Cat(hash)
 //
-func (s *Shell) WithAuthorization(token string) *Shell {
+func (s *Shell) WithAuthorization(token string, extraHeaders map[string]string) *Shell {
 	return &Shell{
 		url: s.url,
 		httpcli: gohttp.Client{
-			Transport: newAuthenticatedTransport(s.httpcli.Transport, token),
+			Transport: newAuthenticatedTransport(s.httpcli.Transport, token, extraHeaders),
 		},
 	}
 }

shell_test.go

@@ -23,7 +23,9 @@ const (
 
 func TestAdd(t *testing.T) {
 	is := is.New(t)
-	s := NewShell(shellUrl)
+	s := NewShell(shellUrl).WithAuthorization("", map[string]string{
+		"X-Hold-Time": "1",
+	})
 
 	mhash, err := s.Add(bytes.NewBufferString("Hello IPFS Shell tests"))
 	is.Nil(err)

transport.go

@@ -3,18 +3,25 @@ package shell
 import "net/http"
 
 type transport struct {
-	token  string
-	httptr http.RoundTripper
+	token        string
+	httptr       http.RoundTripper
+	extraHeaders map[string]string
 }
 
-func newAuthenticatedTransport(tr http.RoundTripper, token string) *transport {
+func newAuthenticatedTransport(tr http.RoundTripper, token string, extraHeaders map[string]string) *transport {
 	return &transport{
-		token:  token,
-		httptr: tr,
+		token:        token,
+		httptr:       tr,
+		extraHeaders: extraHeaders,
 	}
 }
 
 func (t *transport) RoundTrip(req *http.Request) (*http.Response, error) {
+	if t.extraHeaders != nil {
+		for k, v := range t.extraHeaders {
+			req.Header.Set(k, v)
+		}
+	}
 	req.Header.Set("Authorization", "Bearer "+t.token)
 	return t.httptr.RoundTrip(req)
 }