add CLI

2022-07-25 23:35:12 +02:00
parent 8adeeb9665
commit 90ef447d52
3 changed files with 1129 additions and 0 deletions
@@ -0,0 +1,117 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"sort"
+	"strconv"
+	"strings"
+
+	helmctrlapi "github.com/fluxcd/helm-controller/api/v2beta1"
+	ksctrlapi "github.com/fluxcd/kustomize-controller/api/v1beta2"
+	appsv1 "k8s.io/api/apps/v1"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/client/config"
+)
+
+const (
+	PipelineNameLabel  = "pipelines.weave.works/name"
+	PipelineOrderLabel = "pipelines.weave.works/stage"
+)
+
+func main() {
+	cfg := config.GetConfigOrDie()
+	c, err := client.New(cfg, client.Options{})
+	if err != nil {
+		panic(err)
+	}
+
+	if err := ksctrlapi.AddToScheme(c.Scheme()); err != nil {
+		panic(err)
+	}
+	if err := helmctrlapi.AddToScheme(c.Scheme()); err != nil {
+		panic(err)
+	}
+
+	// fetch all Kustomizations that define a pipeline
+
+	pipelines := make(map[string][]ksctrlapi.Kustomization)
+	k := ksctrlapi.KustomizationList{}
+	if err := c.List(context.Background(), &k, client.HasLabels{
+		PipelineNameLabel,
+	}); err != nil {
+		panic(err)
+	}
+
+	for _, ks := range k.Items {
+		ks := ks
+		pName, ok := ks.GetLabels()[PipelineNameLabel]
+		if !ok {
+			continue
+		}
+		pks, ok := pipelines[pName]
+		if !ok {
+			pks = make([]ksctrlapi.Kustomization, 0)
+		}
+		pks = append(pks, ks)
+		pipelines[pName] = pks
+	}
+
+	// sort the Kustomizations by the pipeline stage they're in
+
+	for _, kss := range pipelines {
+		sort.SliceStable(kss, func(i, j int) bool {
+			ksi := kss[i]
+			ksj := kss[j]
+			ksiOrder, err := strconv.Atoi(ksi.GetLabels()[PipelineOrderLabel])
+			if err != nil {
+				return false
+			}
+			ksjOrder, err := strconv.Atoi(ksj.GetLabels()[PipelineOrderLabel])
+			if err != nil {
+				return false
+			}
+			return ksiOrder < ksjOrder
+		})
+	}
+
+	for pName, pKss := range pipelines {
+		fmt.Printf("%s:\n", pName)
+		for _, ks := range pKss {
+			fmt.Printf("\t%s/%s\n", ks.GetNamespace(), ks.GetName())
+
+			// fetch Deployments in pipeline
+
+			deploys := appsv1.DeploymentList{}
+			if err := c.List(context.Background(), &deploys, client.MatchingLabels{
+				"kustomize.toolkit.fluxcd.io/name":      ks.Name,
+				"kustomize.toolkit.fluxcd.io/namespace": ks.Namespace,
+			}); err != nil {
+				panic(err)
+			}
+			for _, deploy := range deploys.Items {
+				fmt.Printf("\t\tDeployment/%s/%s: ", deploy.Namespace, deploy.Name)
+				for idx, ctr := range deploy.Spec.Template.Spec.Containers {
+					fmt.Printf("%s", strings.Split(ctr.Image, ":")[1])
+					if idx < len(deploy.Spec.Template.Spec.Containers)-1 {
+						fmt.Printf(", ")
+					}
+				}
+				fmt.Println()
+			}
+
+			// fetch HelmReleases in pipeline
+
+			hrs := helmctrlapi.HelmReleaseList{}
+			if err := c.List(context.Background(), &hrs, client.MatchingLabels{
+				"kustomize.toolkit.fluxcd.io/name":      ks.Name,
+				"kustomize.toolkit.fluxcd.io/namespace": ks.Namespace,
+			}); err != nil {
+				panic(err)
+			}
+			for _, hr := range hrs.Items {
+				fmt.Printf("\t\tHelmRelease/%s/%s: %s\n", hr.Namespace, hr.Name, hr.Spec.Chart.Spec.Version)
+			}
+		}
+	}
+}