From 7be8d01ee732be71f08f4482e0d5c5b61731ee81 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Magiera?= <magik6k@gmail.com>
Date: Mon, 18 Feb 2019 17:27:12 +0100
Subject: [PATCH] response: option to disable output draining

This commit was moved from ipfs/go-ipfs-http-client@9f3d9635fa4e977b6ec202c9e4f78ca1c64fc0dd
---
 client/httpapi/api.go            |  1 +
 client/httpapi/apifile.go        |  2 +-
 client/httpapi/pubsub.go         |  4 ++--
 client/httpapi/request.go        |  2 ++
 client/httpapi/requestbuilder.go |  7 +++++++
 client/httpapi/response.go       | 10 ++++++++--
 6 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/client/httpapi/api.go b/client/httpapi/api.go
index e106d8d87..f74300216 100644
--- a/client/httpapi/api.go
+++ b/client/httpapi/api.go
@@ -131,6 +131,7 @@ func (api *HttpApi) request(command string, args ...string) *RequestBuilder {
 		command: command,
 		args:    args,
 		shell:   api,
+		drainOut: true,
 	}
 }
 
diff --git a/client/httpapi/apifile.go b/client/httpapi/apifile.go
index 864760684..e3cb85ea4 100644
--- a/client/httpapi/apifile.go
+++ b/client/httpapi/apifile.go
@@ -57,7 +57,7 @@ func (f *apiFile) reset() error {
 	if f.r != nil {
 		f.r.Close()
 	}
-	req := f.core.request("cat", f.path.String())
+	req := f.core.request("cat", f.path.String()).NoDrain()
 	if f.at != 0 {
 		req.Option("offset", f.at)
 	}
diff --git a/client/httpapi/pubsub.go b/client/httpapi/pubsub.go
index edc1a9709..b3e45ed36 100644
--- a/client/httpapi/pubsub.go
+++ b/client/httpapi/pubsub.go
@@ -113,8 +113,8 @@ func (api *PubsubAPI) Subscribe(ctx context.Context, topic string, opts ...caopt
 	}
 
 	resp, err := api.core().request("pubsub/sub", topic).
-		Option("discover", options.Discover).
-		Send(ctx)
+		Option("discover", options.Discover).NoDrain().Send(ctx)
+
 	if err != nil {
 		return nil, err
 	}
diff --git a/client/httpapi/request.go b/client/httpapi/request.go
index 58c61ac67..18cfb7fd0 100644
--- a/client/httpapi/request.go
+++ b/client/httpapi/request.go
@@ -13,6 +13,7 @@ type Request struct {
 	Opts    map[string]string
 	Body    io.Reader
 	Headers map[string]string
+	DrainOut bool // if set, resp.Close will read all remaining data
 }
 
 func NewRequest(ctx context.Context, url, command string, args ...string) *Request {
@@ -30,5 +31,6 @@ func NewRequest(ctx context.Context, url, command string, args ...string) *Reque
 		Args:    args,
 		Opts:    opts,
 		Headers: make(map[string]string),
+		DrainOut: true,
 	}
 }
diff --git a/client/httpapi/requestbuilder.go b/client/httpapi/requestbuilder.go
index af43ce236..8b040522e 100644
--- a/client/httpapi/requestbuilder.go
+++ b/client/httpapi/requestbuilder.go
@@ -19,6 +19,7 @@ type RequestBuilder struct {
 	opts    map[string]string
 	headers map[string]string
 	body    io.Reader
+	drainOut bool
 
 	shell *HttpApi
 }
@@ -84,6 +85,12 @@ func (r *RequestBuilder) Header(name, value string) *RequestBuilder {
 	return r
 }
 
+// NoDrain disables output draining in response closer
+func (r *RequestBuilder) NoDrain() *RequestBuilder {
+	r.drainOut = false
+	return r
+}
+
 // Send sends the request and return the response.
 func (r *RequestBuilder) Send(ctx context.Context) (*Response, error) {
 	r.shell.applyGlobal(r)
diff --git a/client/httpapi/response.go b/client/httpapi/response.go
index f773130e8..cd3cc2b71 100644
--- a/client/httpapi/response.go
+++ b/client/httpapi/response.go
@@ -35,13 +35,18 @@ func (r *trailerReader) Close() error {
 type Response struct {
 	Output io.ReadCloser
 	Error  *Error
+
+	drainOutput bool
 }
 
 func (r *Response) Close() error {
 	if r.Output != nil {
 
-		// always drain output (response body) //TODO: make optional for things like cat
-		_, err1 := io.Copy(ioutil.Discard, r.Output)
+		// always drain output (response body)
+		var err1 error
+		if r.drainOutput {
+			_, err1 = io.Copy(ioutil.Discard, r.Output)
+		}
 		err2 := r.Output.Close()
 		if err1 != nil {
 			return err1
@@ -114,6 +119,7 @@ func (r *Request) Send(c *http.Client) (*Response, error) {
 
 	nresp := new(Response)
 
+	nresp.drainOutput = r.DrainOut
 	nresp.Output = &trailerReader{resp}
 	if resp.StatusCode >= http.StatusBadRequest {
 		e := &Error{