Add gitops CLI (#804)

Signed-off-by: obaydullahmhs <obaydullah@appscode.com>

Author: obaydullahmhs

pkg/cmds/debug.go

@@ -36,6 +36,7 @@ var (
 			* mysql
 			* postgres
 			* redis
+			* gitops
 `)
 )
 

pkg/debug/cassandra.go

@@ -73,6 +73,7 @@ func CassandraDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/clickhouse.go

@@ -73,6 +73,7 @@ func ClickHouseDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/common.go

@@ -117,6 +117,16 @@ func (opts *dbOpts) collectALl() error {
 	if err != nil {
 		return err
 	}
+	if IsOwnByGitOpsObject(opts.db.GetOwnerReferences(), opts.db.GetName(), opts.kind) {
+		gitOpsOpts, err := newGitOpsOpts(opts.kc, opts.db.GetName(), opts.db.GetNamespace(), opts.kind, path.Join(opts.dir, "gitops"))
+		if err != nil {
+			return err
+		}
+		err = gitOpsOpts.collectGitOpsYamls()
+		if err != nil {
+			return err
+		}
+	}
 	return nil
 }
 

pkg/debug/druid.go

@@ -73,6 +73,7 @@ func DruidDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/elasticsearch.go

@@ -73,6 +73,7 @@ func ElasticsearchDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/ferretdb.go

@@ -73,6 +73,7 @@ func FerretDBDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/gitops.go

@@ -0,0 +1,171 @@
+/*
+Copyright AppsCode Inc. and Contributors
+
+Licensed under the AppsCode Community License 1.0.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    https://github.com/appscode/licenses/raw/1.0.0/AppsCode-Community-1.0.0.md
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package debug
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path"
+
+	gitops "kubedb.dev/apimachinery/apis/gitops/v1alpha1"
+	opsapi "kubedb.dev/apimachinery/apis/ops/v1alpha1"
+
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/apimachinery/pkg/types"
+	_ "k8s.io/client-go/plugin/pkg/client/auth"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+type GitOpsStatus struct {
+	GitOps gitops.GitOpsStatus `json:"gitops,omitempty" yaml:"gitops,omitempty"`
+}
+
+type GitOps struct {
+	Status GitOpsStatus `json:"status,omitempty" yaml:"status,omitempty"`
+}
+
+type Ops struct {
+	Status *opsapi.OpsRequestStatus `json:"status,omitempty" yaml:"status,omitempty"`
+}
+
+type dbInfo struct {
+	kind      string
+	name      string
+	namespace string
+}
+
+type gitOpsOpts struct {
+	kc client.Client
+	db dbInfo
+
+	dir     string
+	summary []string
+}
+
+func (g *gitOpsOpts) collectGitOpsYamls() error {
+	err := g.collectGitOpsDatabase()
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("Summary:")
+	for _, line := range g.summary {
+		fmt.Println("- ", line)
+	}
+	fmt.Println("--------------- Done ---------------")
+	return nil
+}
+
+func newGitOpsOpts(kc client.Client, name, namespace, kind, dir string) (*gitOpsOpts, error) {
+	err := os.MkdirAll(dir, dirPerm)
+	if err != nil {
+		return nil, err
+	}
+	opts := &gitOpsOpts{
+		kc: kc,
+		db: dbInfo{
+			kind:      kind,
+			name:      name,
+			namespace: namespace,
+		},
+		dir:     dir,
+		summary: make([]string, 0),
+	}
+	return opts, nil
+}
+
+func (g *gitOpsOpts) collectGitOpsDatabase() error {
+	var uns unstructured.Unstructured
+	uns.SetGroupVersionKind(schema.GroupVersionKind{
+		Group:   gitops.SchemeGroupVersion.Group,
+		Version: gitops.SchemeGroupVersion.Version,
+		Kind:    g.db.kind,
+	})
+	err := g.kc.Get(context.Background(), types.NamespacedName{
+		Namespace: g.db.namespace,
+		Name:      g.db.name,
+	}, &uns)
+	if err != nil {
+		return err
+	}
+
+	var gitOpsObj GitOps
+	err = runtime.DefaultUnstructuredConverter.FromUnstructured(uns.Object, &gitOpsObj)
+	if err != nil {
+		return err
+	}
+
+	status := string(gitops.ChangeRequestStatusInCurrent)
+	// last status is the latest status
+	if len(gitOpsObj.Status.GitOps.GitOpsInfo) > 0 {
+		status = string(gitOpsObj.Status.GitOps.GitOpsInfo[len(gitOpsObj.Status.GitOps.GitOpsInfo)-1].ChangeRequestStatus)
+	}
+
+	g.summary = append(g.summary, fmt.Sprintf("GitOps Database Status for: %s/%s is %s", g.db.namespace, g.db.name, status))
+
+	if err := g.collectOpsRequests(gitOpsObj.Status); err != nil {
+		return err
+	}
+
+	return writeYaml(&uns, g.dir)
+}
+
+func (g *gitOpsOpts) collectOpsRequests(gitOpsStatus GitOpsStatus) error {
+	opsYamlDir := path.Join(g.dir, "ops")
+	err := os.MkdirAll(opsYamlDir, dirPerm)
+	if err != nil {
+		return err
+	}
+	for _, info := range gitOpsStatus.GitOps.GitOpsInfo {
+		for _, op := range info.Operations {
+			var uns unstructured.Unstructured
+			uns.SetGroupVersionKind(schema.GroupVersionKind{
+				Group:   opsapi.SchemeGroupVersion.Group,
+				Version: opsapi.SchemeGroupVersion.Version,
+				Kind:    g.db.kind + "OpsRequest",
+			})
+			err := g.kc.Get(context.Background(), types.NamespacedName{
+				Namespace: g.db.namespace,
+				Name:      op.Name,
+			}, &uns)
+			if err != nil {
+				return err
+			}
+			err = writeYaml(&uns, opsYamlDir)
+			if err != nil {
+				return err
+			}
+			var ops Ops
+			err = runtime.DefaultUnstructuredConverter.FromUnstructured(uns.Object, &ops)
+			if err != nil {
+				return err
+			}
+			if ops.Status.Phase == opsapi.OpsRequestPhaseFailed {
+				for _, cond := range ops.Status.Conditions {
+					if cond.Reason == opsapi.Failed {
+						g.summary = append(g.summary, fmt.Sprintf("RequestName %s: %s", op.Name, cond.Message))
+					}
+				}
+			}
+		}
+	}
+
+	return nil
+}

pkg/debug/hazelcast.go

@@ -73,6 +73,7 @@ func HazelcastDebugCMD(f cmdutil.Factory) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			opts.db.OwnerReferences = db.OwnerReferences
 
 			err = writeYaml(&db, getDir(db.GetName()))
 			if err != nil {

pkg/debug/helpers.go

@@ -20,10 +20,20 @@ import (
 	"os"
 	"path"
 
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/yaml"
 )
 
+func IsOwnByGitOpsObject(owners []metav1.OwnerReference, name, kind string) bool {
+	for _, owner := range owners {
+		if owner.Kind == kind && owner.Name == name && owner.APIVersion == "gitops.kubedb.com/v1alpha1" {
+			return true
+		}
+	}
+	return false
+}
+
 func writeYaml(obj client.Object, fullPath string) error {
 	b, err := yaml.Marshal(obj)
 	if err != nil {