From 3fea2f0a91fb62848d481e027fde7109cdb0ebeb Mon Sep 17 00:00:00 2001
From: Brent Baude <bbaude@redhat.com>
Date: Wed, 13 May 2020 09:34:46 -0500
Subject: enable remote image tree

Signed-off-by: Brent Baude <bbaude@redhat.com>
---
 pkg/api/handlers/libpod/images.go | 19 ++++++++-----------
 pkg/bindings/images/images.go     | 18 ++++++++++++++++--
 pkg/domain/infra/tunnel/images.go |  2 +-
 3 files changed, 25 insertions(+), 14 deletions(-)

diff --git a/pkg/api/handlers/libpod/images.go b/pkg/api/handlers/libpod/images.go
index 93b4564a1..1cbcfb52c 100644
--- a/pkg/api/handlers/libpod/images.go
+++ b/pkg/api/handlers/libpod/images.go
@@ -56,13 +56,6 @@ func ImageExists(w http.ResponseWriter, r *http.Request) {
 func ImageTree(w http.ResponseWriter, r *http.Request) {
 	runtime := r.Context().Value("runtime").(*libpod.Runtime)
 	name := utils.GetName(r)
-
-	img, err := runtime.ImageRuntime().NewFromLocal(name)
-	if err != nil {
-		utils.Error(w, "Something went wrong.", http.StatusNotFound, errors.Wrapf(err, "Failed to find image %s", name))
-		return
-	}
-
 	decoder := r.Context().Value("decoder").(*schema.Decoder)
 	query := struct {
 		WhatRequires bool `schema:"whatrequires"`
@@ -74,14 +67,18 @@ func ImageTree(w http.ResponseWriter, r *http.Request) {
 			errors.Wrapf(err, "failed to parse parameters for %s", r.URL.String()))
 		return
 	}
-
-	tree, err := img.GenerateTree(query.WhatRequires)
+	ir := abi.ImageEngine{Libpod: runtime}
+	options := entities.ImageTreeOptions{WhatRequires: query.WhatRequires}
+	report, err := ir.Tree(r.Context(), name, options)
 	if err != nil {
+		if errors.Cause(err) == define.ErrNoSuchImage {
+			utils.Error(w, "Something went wrong.", http.StatusNotFound, errors.Wrapf(err, "Failed to find image %s", name))
+			return
+		}
 		utils.Error(w, "Server error", http.StatusInternalServerError, errors.Wrapf(err, "failed to generate image tree for %s", name))
 		return
 	}
-
-	utils.WriteResponse(w, http.StatusOK, tree)
+	utils.WriteResponse(w, http.StatusOK, report)
 }
 
 func GetImage(w http.ResponseWriter, r *http.Request) {
diff --git a/pkg/bindings/images/images.go b/pkg/bindings/images/images.go
index 034ade618..a15ce56e5 100644
--- a/pkg/bindings/images/images.go
+++ b/pkg/bindings/images/images.go
@@ -74,8 +74,22 @@ func GetImage(ctx context.Context, nameOrID string, size *bool) (*entities.Image
 	return &inspectedData, response.Process(&inspectedData)
 }
 
-func Tree(ctx context.Context, nameOrId string) error {
-	return bindings.ErrNotImplemented
+// Tree retrieves a "tree" based representation of the given image
+func Tree(ctx context.Context, nameOrId string, whatRequires *bool) (*entities.ImageTreeReport, error) {
+	var report entities.ImageTreeReport
+	conn, err := bindings.GetClient(ctx)
+	if err != nil {
+		return nil, err
+	}
+	params := url.Values{}
+	if whatRequires != nil {
+		params.Set("size", strconv.FormatBool(*whatRequires))
+	}
+	response, err := conn.DoRequest(nil, http.MethodGet, "/images/%s/tree", params, nameOrId)
+	if err != nil {
+		return nil, err
+	}
+	return &report, response.Process(&report)
 }
 
 // History returns the parent layers of an image.
diff --git a/pkg/domain/infra/tunnel/images.go b/pkg/domain/infra/tunnel/images.go
index 788752fd8..eb25dc4a3 100644
--- a/pkg/domain/infra/tunnel/images.go
+++ b/pkg/domain/infra/tunnel/images.go
@@ -258,7 +258,7 @@ func (ir *ImageEngine) Build(ctx context.Context, containerFiles []string, opts
 }
 
 func (ir *ImageEngine) Tree(ctx context.Context, nameOrId string, opts entities.ImageTreeOptions) (*entities.ImageTreeReport, error) {
-	return nil, errors.New("not implemented yet")
+	return images.Tree(ir.ClientCxt, nameOrId, &opts.WhatRequires)
 }
 
 // Shutdown Libpod engine
-- 
cgit v1.2.3-54-g00ecf