Query UCP Kubernetes user namespaces endpoint

mat007 · mat007 · commit 0bf781e9123a · 2018-05-15T15:25:28.000+02:00
Signed-off-by: Mathieu Champlon &lt;mathieu.champlon@docker.com&gt;
diff --git a/cli/command/cli.go b/cli/command/cli.go
@@ -174,6 +174,7 @@ func (cli *DockerCli) Initialize(opts *cliflags.ClientOptions) error {
 		DefaultVersion:  cli.client.ClientVersion(),
 		HasExperimental: hasExperimental,
 		Orchestrator:    orchestrator,
+		TLSOptions:      opts.Common.TLSOptions,
 	}
 	cli.initializeFromClient()
 	return nil
@@ -240,6 +241,7 @@ type ClientInfo struct {
 	HasExperimental bool
 	DefaultVersion  string
 	Orchestrator    Orchestrator
+	TLSOptions      *tlsconfig.Options
 }
 
 // HasKubernetes checks if kubernetes orchestrator is enabled
diff --git a/cli/command/stack/kubernetes/list.go b/cli/command/stack/kubernetes/list.go
@@ -1,9 +1,20 @@
 package kubernetes
 
 import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+
 	"github.com/docker/cli/cli/command"
 	"github.com/docker/cli/cli/command/formatter"
 	"github.com/docker/cli/cli/command/stack/options"
+	"github.com/docker/go-connections/tlsconfig"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+	core_v1 "k8s.io/api/core/v1"
+	apierrs "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
@@ -17,7 +28,7 @@ func GetStacks(dockerCli command.Cli, opts options.List, kopts Options) ([]*form
 		if dockerCli.ClientInfo().HasAll() {
 			opts.AllNamespaces = true
 		}
-		return getStacks(kubeCli, opts)
+		return getStacksWithAllNamespaces(dockerCli, kubeCli, opts)
 	}
 	return getStacksWithNamespaces(kubeCli, opts)
 }
@@ -47,6 +58,74 @@ func getStacks(kubeCli *KubeCli, opts options.List) ([]*formatter.Stack, error)
 	return formattedStacks, nil
 }
 
+func getStacksWithAllNamespaces(dockerCli command.Cli, kubeCli *KubeCli, opts options.List) ([]*formatter.Stack, error) {
+	stacks, err := getStacks(kubeCli, opts)
+	if err == nil || !apierrs.IsForbidden(err) {
+		return stacks, err
+	}
+	nms, err2 := getUserVisibleNamespaces(dockerCli)
+	if err2 != nil {
+		logrus.Debugf("Failed to query user visible namespaces: %s", err2)
+		return nil, err
+	}
+	opts.AllNamespaces = false
+	for _, namespace := range nms.Items {
+		kubeCli.kubeNamespace = namespace.Name
+		ss, err := getStacks(kubeCli, opts)
+		if err != nil {
+			return nil, err
+		}
+		stacks = append(stacks, ss...)
+	}
+	return stacks, nil
+}
+
+func getUserVisibleNamespaces(dockerCli command.Cli) (*core_v1.NamespaceList, error) {
+	host := dockerCli.Client().DaemonHost()
+	endpoint, err := url.Parse(host)
+	if err != nil {
+		return nil, err
+	}
+	endpoint.Scheme = "https"
+	endpoint.Path = "/kubernetesNamespaces"
+	res := core_v1.NamespaceList{}
+	return &res, makeRequest(dockerCli, *endpoint, func(resp http.Response) error {
+		body, err := ioutil.ReadAll(resp.Body)
+		if err != nil {
+			return errors.Wrapf(err, "received %d status and unable to read response", resp.StatusCode)
+		}
+		if resp.StatusCode != http.StatusOK {
+			return fmt.Errorf(string(body))
+		}
+		if err := json.Unmarshal(body, &res); err != nil {
+			return errors.Wrapf(err, "unmarshal failed: %s", string(body))
+		}
+		return nil
+	})
+}
+
+func makeRequest(dockerCli command.Cli, endpoint url.URL, handler func(resp http.Response) error) error {
+	tlsOptions := dockerCli.ClientInfo().TLSOptions
+	if tlsOptions == nil {
+		return fmt.Errorf("missing TLS options for https")
+	}
+	tlsConfig, err := tlsconfig.Client(*tlsOptions)
+	if err != nil {
+		return err
+	}
+	httpClient := http.Client{
+		Transport: &http.Transport{
+			TLSClientConfig: tlsConfig,
+		},
+	}
+	resp, err := httpClient.Get(endpoint.String())
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+	return handler(*resp)
+}
+
 func getStacksWithNamespaces(kubeCli *KubeCli, opts options.List) ([]*formatter.Stack, error) {
 	stacks := []*formatter.Stack{}
 	for _, namespace := range removeDuplicates(opts.Namespaces) {

Original file line number	Diff line number	Diff line change
`@@ -174,6 +174,7 @@ func (cli DockerCli) Initialize(opts cliflags.ClientOptions) error {`
`174`	`174`	`DefaultVersion: cli.client.ClientVersion(),`
`175`	`175`	`HasExperimental: hasExperimental,`
`176`	`176`	`Orchestrator: orchestrator,`
	`177`	`+ TLSOptions: opts.Common.TLSOptions,`
`177`	`178`	`}`
`178`	`179`	`cli.initializeFromClient()`
`179`	`180`	`return nil`
`@@ -240,6 +241,7 @@ type ClientInfo struct {`
`240`	`241`	`HasExperimental bool`
`241`	`242`	`DefaultVersion string`
`242`	`243`	`Orchestrator Orchestrator`
	`244`	`+ TLSOptions *tlsconfig.Options`
`243`	`245`	`}`
`244`	`246`
`245`	`247`	`// HasKubernetes checks if kubernetes orchestrator is enabled`