Add debug util for gateway

Signed-off-by: Arnob kumar saha <arnob@appscode.com>

Author: ArnobKumarSaha

go.mod

@@ -0,0 +1,41 @@
+/*
+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 (
+	"go.bytebuilders.dev/cli/pkg/cmds/debug/gateway"
+
+	"github.com/spf13/cobra"
+	"k8s.io/cli-runtime/pkg/genericclioptions"
+	cmdutil "k8s.io/kubectl/pkg/cmd/util"
+)
+
+func NewCmdDebug() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:               "debug",
+		Short:             "Debug utilities",
+		DisableAutoGenTag: true,
+	}
+
+	kubeConfigFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
+	matchVersionKubeConfigFlags := cmdutil.NewMatchVersionFlags(kubeConfigFlags)
+	f := cmdutil.NewFactory(matchVersionKubeConfigFlags)
+
+	cmd.AddCommand(gateway.NewCmdGateway(f))
+
+	return cmd
+}

go.sum

@@ -0,0 +1,210 @@
+package gateway
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"os"
+	"path"
+
+	catalogapi "go.bytebuilders.dev/catalog/api/catalog/v1alpha1"
+
+	flux "github.com/fluxcd/helm-controller/api/v2"
+	"github.com/spf13/cobra"
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/labels"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/types"
+	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
+	"k8s.io/client-go/kubernetes"
+	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
+	"k8s.io/klog/v2"
+	cmdutil "k8s.io/kubectl/pkg/cmd/util"
+	"kmodules.xyz/client-go/meta"
+	dbapi "kubedb.dev/apimachinery/apis/kubedb/v1"
+	kubedbscheme "kubedb.dev/apimachinery/client/clientset/versioned/scheme"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	gwapiv1 "sigs.k8s.io/gateway-api/apis/v1"
+	vgapi "voyagermesh.dev/gateway-api/apis/gateway/v1alpha1"
+)
+
+var scheme = runtime.NewScheme()
+
+func init() {
+	utilruntime.Must(clientgoscheme.AddToScheme(scheme))
+	utilruntime.Must(catalogapi.AddToScheme(scheme))
+	utilruntime.Must(kubedbscheme.AddToScheme(scheme))
+	utilruntime.Must(gwapiv1.Install(scheme))
+	utilruntime.Must(vgapi.AddToScheme(scheme))
+	utilruntime.Must(flux.AddToScheme(scheme))
+}
+
+func NewCmdGateway(f cmdutil.Factory) *cobra.Command {
+	opt := newGatewayOpts(f)
+	cmd := &cobra.Command{
+		Use:               "gateway",
+		Short:             "Gateway related info",
+		DisableAutoGenTag: true,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			klog.Infof("========> %s %s %s \n", opt.db.resource, opt.db.name, opt.db.namespace)
+			return opt.run()
+		},
+	}
+
+	cmd.Flags().StringVarP(&opt.db.resource, "db-type", "t", "mongodb", "Database type")
+	cmd.Flags().StringVarP(&opt.db.name, "name", "m", "mg-test", "Database name")
+	cmd.Flags().StringVarP(&opt.db.namespace, "namespace", "n", "demo", "Database namespace")
+	return cmd
+}
+
+type gatewayOpts struct {
+	kc         client.Client
+	kubeClient kubernetes.Interface
+	db         dbInfo
+	dir        string
+}
+
+type dbInfo struct {
+	resource  string
+	name      string
+	namespace string
+}
+
+func newGatewayOpts(f cmdutil.Factory) *gatewayOpts {
+	config, err := f.ToRESTConfig()
+	if err != nil {
+		log.Fatal(err)
+	}
+	kc, err := client.New(config, client.Options{Scheme: scheme})
+	if err != nil {
+		log.Fatalf("failed to create client: %v", err)
+	}
+
+	cs, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		log.Fatalf("failed to create kube client: %v", err)
+	}
+
+	return &gatewayOpts{kc: kc, kubeClient: cs}
+}
+
+func (g *gatewayOpts) run() error {
+	pwd, _ := os.Getwd()
+	dir := path.Join(pwd, g.db.name)
+	err := os.MkdirAll(path.Join(dir, logsDir), dirPerm)
+	if err != nil {
+		return err
+	}
+	err = os.MkdirAll(path.Join(dir, yamlsDir), dirPerm)
+	if err != nil {
+		return err
+	}
+	g.dir = dir
+
+	if err := g.collectGateways(); err != nil {
+		return err
+	}
+	if err := g.collectDatabase(); err != nil {
+		return err
+	}
+
+	if err := g.collectOperatorLogs(); err != nil {
+		return err
+	}
+	fmt.Println("Success !!!")
+	return nil
+}
+
+func (g *gatewayOpts) collectGateways() error {
+	var gwList gwapiv1.GatewayList
+	err := g.kc.List(context.TODO(), &gwList, client.InNamespace(g.db.namespace))
+	if err != nil {
+		log.Fatalf("failed to get gateways: %v", err)
+	}
+
+	psYamlDir := path.Join(g.dir, yamlsDir, "gateways")
+	err = os.MkdirAll(psYamlDir, dirPerm)
+	if err != nil {
+		return err
+	}
+
+	for _, p := range gwList.Items {
+		err = writeYaml(&p, psYamlDir)
+		if err != nil {
+			return err
+		}
+
+	}
+	return nil
+}
+
+func (g *gatewayOpts) collectDatabase() error {
+	var uns unstructured.Unstructured
+	uns.SetGroupVersionKind(dbapi.SchemeGroupVersion.WithKind(getKindFromResource(g.db.resource)))
+	err := g.kc.Get(context.Background(), types.NamespacedName{
+		Namespace: g.db.namespace,
+		Name:      g.db.name,
+	}, &uns)
+	if err != nil {
+		log.Fatalf("failed to get gateways: %v", err)
+	}
+
+	return writeYaml(&uns, path.Join(g.dir, yamlsDir))
+}
+
+func (g *gatewayOpts) collectOperatorLogs() error {
+	labels := labels.SelectorFromSet(map[string]string{
+		meta.InstanceLabelKey: catalogManager,
+	}).String()
+	pods, err := g.kubeClient.CoreV1().Pods("ace").List(context.TODO(), metav1.ListOptions{
+		LabelSelector: labels,
+	})
+	if err != nil {
+		return err
+	}
+	for _, pod := range pods.Items {
+		isOperatorPod := false
+		for _, container := range pod.Spec.Containers {
+			if container.Name == catalogManager {
+				isOperatorPod = true
+			}
+		}
+		if isOperatorPod {
+			err = g.writeLogs(pod.Name, pod.Namespace, catalogManager)
+			if err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+func (g *gatewayOpts) writeLogs(podName, ns, container string) error {
+	req := g.kubeClient.CoreV1().Pods(ns).GetLogs(podName, &corev1.PodLogOptions{
+		Container: container,
+	})
+
+	podLogs, err := req.Stream(context.TODO())
+	if err != nil {
+		return err
+	}
+	defer podLogs.Close()
+
+	logFile, err := os.Create(path.Join(g.dir, logsDir, podName+"_"+container+".log"))
+	if err != nil {
+		return err
+	}
+	defer logFile.Close()
+
+	buf := make([]byte, 1024)
+	for {
+		bytesRead, err := podLogs.Read(buf)
+		if err != nil {
+			break
+		}
+		_, _ = logFile.Write(buf[:bytesRead])
+	}
+	return nil
+}

pkg/cmds/debug/debug.go

@@ -0,0 +1,37 @@
+package gateway
+
+import (
+	"fmt"
+	"os"
+	"path"
+
+	dbapi "kubedb.dev/apimachinery/apis/kubedb/v1"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/yaml"
+)
+
+const (
+	logsDir  = "logs"
+	yamlsDir = "yamls"
+	dirPerm  = 0o755
+	filePerm = 0o644
+
+	catalogManager = "catalog-manager"
+)
+
+func writeYaml(obj client.Object, fullPath string) error {
+	b, err := yaml.Marshal(obj)
+	if err != nil {
+		return err
+	}
+	return os.WriteFile(path.Join(fullPath, obj.GetName()+".yaml"), b, filePerm)
+}
+
+func getKindFromResource(res string) string {
+	switch res {
+	case "postgres", "pg":
+		return dbapi.ResourceKindPostgres
+	}
+	_ = fmt.Errorf("resource %s not supported", res)
+	return ""
+}

pkg/cmds/debug/gateway/gateway.go

@@ -22,9 +22,9 @@ import (
 	"syscall"
 
 	"go.bytebuilders.dev/cli/pkg/cmds/auth"
-	"go.bytebuilders.dev/cli/pkg/cmds/cloud_swap"
 	"go.bytebuilders.dev/cli/pkg/cmds/cluster"
 	cmdconfig "go.bytebuilders.dev/cli/pkg/cmds/config"
+	"go.bytebuilders.dev/cli/pkg/cmds/debug"
 	"go.bytebuilders.dev/cli/pkg/cmds/installer"
 	"go.bytebuilders.dev/cli/pkg/config"
 	ace "go.bytebuilders.dev/client"
@@ -51,9 +51,10 @@ func NewRootCmd() *cobra.Command {
 	rootCmd.AddCommand(cluster.NewCmdCluster(f))
 	rootCmd.AddCommand(auth.NewCmdAuth())
 
-	rootCmd.AddCommand(cloud_swap.NewCmdCloudSwap())
+	// rootCmd.AddCommand(cloud_swap.NewCmdCloudSwap())
 
 	rootCmd.AddCommand(installer.NewCmdInstaller())
+	rootCmd.AddCommand(debug.NewCmdDebug())
 
 	rootCmd.AddCommand(v.NewCmdVersion())
 	rootCmd.AddCommand(NewCmdCompletion())