From 382aba32a8a1916c2e67cdadb756dfe80c4bde29 Mon Sep 17 00:00:00 2001
From: David Calavera <david.calavera@gmail.com>
Date: Tue, 22 Dec 2015 19:00:27 -0500
Subject: [PATCH] Make `docker load` to output json when the response content
 type is json.

Signed-off-by: David Calavera <david.calavera@gmail.com>
Upstream-commit: 9fd2c0feb0c131d01d727d50baa7183b976c7bdc
Component: engine
---
 components/engine/api/client/client.go         |  2 +-
 components/engine/api/client/lib/image_load.go | 11 ++++++++---
 components/engine/api/client/load.go           | 11 ++++++++---
 components/engine/api/types/client.go          |  6 ++++++
 4 files changed, 23 insertions(+), 7 deletions(-)

diff --git a/components/engine/api/client/client.go b/components/engine/api/client/client.go
index 7da14e2cbb..36aca35c1e 100644
--- a/components/engine/api/client/client.go
+++ b/components/engine/api/client/client.go
@@ -53,7 +53,7 @@ type apiClient interface {
 	ImageImport(options types.ImageImportOptions) (io.ReadCloser, error)
 	ImageInspectWithRaw(imageID string, getSize bool) (types.ImageInspect, []byte, error)
 	ImageList(options types.ImageListOptions) ([]types.Image, error)
-	ImageLoad(input io.Reader) (io.ReadCloser, error)
+	ImageLoad(input io.Reader) (types.ImageLoadResponse, error)
 	ImagePull(options types.ImagePullOptions, privilegeFunc lib.RequestPrivilegeFunc) (io.ReadCloser, error)
 	ImagePush(options types.ImagePushOptions, privilegeFunc lib.RequestPrivilegeFunc) (io.ReadCloser, error)
 	ImageRemove(options types.ImageRemoveOptions) ([]types.ImageDelete, error)
diff --git a/components/engine/api/client/lib/image_load.go b/components/engine/api/client/lib/image_load.go
index 954cfd8d7b..c713262c79 100644
--- a/components/engine/api/client/lib/image_load.go
+++ b/components/engine/api/client/lib/image_load.go
@@ -3,15 +3,20 @@ package lib
 import (
 	"io"
 	"net/url"
+
+	"github.com/docker/docker/api/types"
 )
 
 // ImageLoad loads an image in the docker host from the client host.
 // It's up to the caller to close the io.ReadCloser returned by
 // this function.
-func (cli *Client) ImageLoad(input io.Reader) (io.ReadCloser, error) {
+func (cli *Client) ImageLoad(input io.Reader) (types.ImageLoadResponse, error) {
 	resp, err := cli.postRaw("/images/load", url.Values{}, input, nil)
 	if err != nil {
-		return nil, err
+		return types.ImageLoadResponse{}, err
 	}
-	return resp.body, nil
+	return types.ImageLoadResponse{
+		Body: resp.body,
+		JSON: resp.header.Get("Content-Type") == "application/json",
+	}, nil
 }
diff --git a/components/engine/api/client/load.go b/components/engine/api/client/load.go
index 10681dc71a..745c013c01 100644
--- a/components/engine/api/client/load.go
+++ b/components/engine/api/client/load.go
@@ -5,6 +5,7 @@ import (
 	"os"
 
 	Cli "github.com/docker/docker/cli"
+	"github.com/docker/docker/pkg/jsonmessage"
 	flag "github.com/docker/docker/pkg/mflag"
 )
 
@@ -29,12 +30,16 @@ func (cli *DockerCli) CmdLoad(args ...string) error {
 		input = file
 	}
 
-	responseBody, err := cli.client.ImageLoad(input)
+	response, err := cli.client.ImageLoad(input)
 	if err != nil {
 		return err
 	}
-	defer responseBody.Close()
+	defer response.Body.Close()
 
-	_, err = io.Copy(cli.out, responseBody)
+	if response.JSON {
+		return jsonmessage.DisplayJSONMessagesStream(response.Body, cli.out, cli.outFd, cli.isTerminalOut)
+	}
+
+	_, err = io.Copy(cli.out, response.Body)
 	return err
 }
diff --git a/components/engine/api/types/client.go b/components/engine/api/types/client.go
index 4a0877f97f..c911ad0789 100644
--- a/components/engine/api/types/client.go
+++ b/components/engine/api/types/client.go
@@ -179,6 +179,12 @@ type ImageListOptions struct {
 	Filters   filters.Args
 }
 
+// ImageLoadResponse returns information to the client about a load process.
+type ImageLoadResponse struct {
+	Body io.ReadCloser
+	JSON bool
+}
+
 // ImagePullOptions holds information to pull images.
 type ImagePullOptions struct {
 	ImageID string