2017-11-20 09:30:52 -05:00
|
|
|
package kubernetes
|
|
|
|
|
|
|
|
import (
|
2018-04-26 05:13:14 -04:00
|
|
|
"github.com/docker/cli/cli/command"
|
2017-11-20 09:30:52 -05:00
|
|
|
"github.com/docker/cli/cli/command/formatter"
|
2017-12-04 06:30:39 -05:00
|
|
|
"github.com/docker/cli/cli/command/stack/options"
|
2017-11-20 09:30:52 -05:00
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
)
|
|
|
|
|
2018-04-26 05:13:14 -04:00
|
|
|
// GetStacks lists the kubernetes stacks
|
|
|
|
func GetStacks(dockerCli command.Cli, opts options.List, kopts Options) ([]*formatter.Stack, error) {
|
|
|
|
kubeCli, err := WrapCli(dockerCli, kopts)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if opts.AllNamespaces || len(opts.Namespaces) == 0 {
|
|
|
|
return getStacks(kubeCli, opts)
|
|
|
|
}
|
|
|
|
return getStacksWithNamespaces(kubeCli, opts)
|
|
|
|
}
|
|
|
|
|
|
|
|
func getStacks(kubeCli *KubeCli, opts options.List) ([]*formatter.Stack, error) {
|
2018-04-09 09:07:11 -04:00
|
|
|
composeClient, err := kubeCli.composeClient()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2018-04-25 08:24:46 -04:00
|
|
|
stackSvc, err := composeClient.Stacks(opts.AllNamespaces)
|
2017-11-20 09:30:52 -05:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2018-04-09 09:13:16 -04:00
|
|
|
stacks, err := stackSvc.List(metav1.ListOptions{})
|
2017-11-20 09:30:52 -05:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
var formattedStacks []*formatter.Stack
|
2018-01-31 09:37:14 -05:00
|
|
|
for _, stack := range stacks {
|
2017-11-20 09:30:52 -05:00
|
|
|
formattedStacks = append(formattedStacks, &formatter.Stack{
|
2018-03-27 10:38:47 -04:00
|
|
|
Name: stack.name,
|
|
|
|
Services: len(stack.getServices()),
|
|
|
|
Orchestrator: "Kubernetes",
|
2018-04-09 09:11:45 -04:00
|
|
|
Namespace: stack.namespace,
|
2017-11-20 09:30:52 -05:00
|
|
|
})
|
|
|
|
}
|
|
|
|
return formattedStacks, nil
|
|
|
|
}
|
2018-04-26 05:13:14 -04:00
|
|
|
|
|
|
|
func getStacksWithNamespaces(kubeCli *KubeCli, opts options.List) ([]*formatter.Stack, error) {
|
|
|
|
stacks := []*formatter.Stack{}
|
|
|
|
for _, namespace := range removeDuplicates(opts.Namespaces) {
|
|
|
|
kubeCli.kubeNamespace = namespace
|
|
|
|
ss, err := getStacks(kubeCli, opts)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
stacks = append(stacks, ss...)
|
|
|
|
}
|
|
|
|
return stacks, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func removeDuplicates(namespaces []string) []string {
|
|
|
|
found := make(map[string]bool)
|
|
|
|
results := namespaces[:0]
|
|
|
|
for _, n := range namespaces {
|
|
|
|
if !found[n] {
|
|
|
|
results = append(results, n)
|
|
|
|
found[n] = true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return results
|
|
|
|
}
|