chore: adapt cluster connect command with 1.0 api (#437)

Co-authored-by: wangyelei <[email protected]>

docs/user_docs/cli/kbcli_cluster_connect.md

@@ -11,41 +11,26 @@ kbcli cluster connect (NAME | -i INSTANCE-NAME) [flags]
 ### Examples
 
 ```
-  # connect to a specified cluster, default connect to the leader/primary instance
+  # connect to a specified cluster
   kbcli cluster connect mycluster
   
-  # connect to cluster as user
-  kbcli cluster connect mycluster --as-user myuser
-  
   # connect to a specified instance
   kbcli cluster connect -i mycluster-instance-0
   
   # connect to a specified component
   kbcli cluster connect mycluster --component mycomponent
   
-  # show cli connection example with password mask
-  kbcli cluster connect mycluster --show-example --client=cli
-  
-  # show java connection example with password mask
-  kbcli cluster connect mycluster --show-example --client=java
-  
-  # show all connection examples with password mask
-  kbcli cluster connect mycluster --show-example
-  
-  # show cli connection examples with real password
-  kbcli cluster connect mycluster --show-example --client=cli --show-password
+  # show cli connection example, supported client: [cli, java, python, rust, php, node.js, go, .net, django] and more.
+  kbcli cluster connect mycluster --client=cli
 ```
 
 ### Options
 
 ```
-      --as-user string     Connect to cluster as user
-      --client string      Which client connection example should be output, only valid if --show-example is true.
-      --component string   The component to connect. If not specified, pick up the first one.
+      --client string      Which client connection example should be output.
+      --component string   The component to connect. If not specified and no any cluster scope services, pick up the first one.
   -h, --help               help for connect
   -i, --instance string    The instance name to connect.
-      --show-example       Show how to connect to cluster/instance from different clients.
-      --show-password      Show password in example.
 ```
 
 ### Options inherited from parent commands

pkg/cluster/cluster_chart.go

@@ -22,11 +22,11 @@ package cluster
 import (
 	"compress/gzip"
 	"fmt"
+	"slices"
 	"strings"
 
 	"github.com/pkg/errors"
 	"golang.org/x/exp/maps"
-	"golang.org/x/exp/slices"
 	"helm.sh/helm/v3/pkg/action"
 	"helm.sh/helm/v3/pkg/chart"
 	"helm.sh/helm/v3/pkg/chart/loader"
@@ -219,8 +219,8 @@ func loadHelmChart(ci *ChartInfo, t ClusterType) error {
 
 func SupportedTypes() []ClusterType {
 	types := maps.Keys(ClusterTypeCharts)
-	slices.SortFunc(types, func(i, j ClusterType) bool {
-		return i.String() < j.String()
+	slices.SortFunc(types, func(i, j ClusterType) int {
+		return strings.Compare(i.String(), j.String())
 	})
 	return types
 }

pkg/cluster/helper.go

@@ -32,6 +32,7 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/client-go/dynamic"
+	"k8s.io/client-go/kubernetes"
 	"k8s.io/klog/v2"
 
 	"github.com/apecloud/kbcli/pkg/testing"
@@ -377,6 +378,9 @@ func GetPodComponentName(pod *corev1.Pod) string {
 	if pod.Labels == nil {
 		return ""
 	}
+	if compName, ok := pod.Labels[constant.KBAppShardingNameLabelKey]; ok {
+		return compName
+	}
 	return pod.Labels[constant.KBAppComponentLabelKey]
 }
 
@@ -396,6 +400,24 @@ func GetConfigConstraintByName(dynamic dynamic.Interface, name string) (*appsv1b
 	return ccObj, nil
 }
 
+func ListShardingComponents(dynamic dynamic.Interface, clusterName, clusterNamespace, componentName string) ([]*appsv1alpha1.Component, error) {
+	unstructuredObjList, err := dynamic.Resource(types.ComponentGVR()).Namespace(clusterNamespace).List(context.TODO(), metav1.ListOptions{
+		LabelSelector: fmt.Sprintf("%s=%s,%s=%s", constant.AppInstanceLabelKey, clusterName, constant.KBAppShardingNameLabelKey, componentName),
+	})
+	if err != nil {
+		return nil, nil
+	}
+	var components []*appsv1alpha1.Component
+	for i := range unstructuredObjList.Items {
+		comp := &appsv1alpha1.Component{}
+		if err = runtime.DefaultUnstructuredConverter.FromUnstructured(unstructuredObjList.Items[i].UnstructuredContent(), comp); err != nil {
+			return nil, err
+		}
+		components = append(components, comp)
+	}
+	return components, nil
+}
+
 func GetServiceRefs(cd *appsv1alpha1.ClusterDefinition) []string {
 	var serviceRefs []string
 	for _, compDef := range cd.Spec.ComponentDefs {
@@ -441,3 +463,27 @@ func GetDefaultVersionByCompDefs(dynamic dynamic.Interface, compDefs []string) (
 	}
 	return cv, nil
 }
+
+func GetReadyNodeForNodePort(client *kubernetes.Clientset) (*corev1.Node, error) {
+	nodes, err := client.CoreV1().Nodes().List(context.TODO(), metav1.ListOptions{
+		Limit: 20,
+	})
+	if err != nil {
+		return nil, err
+	}
+	var readyNode *corev1.Node
+	for _, node := range nodes.Items {
+		var nodeIsReady bool
+		for _, con := range node.Status.Conditions {
+			if con.Type == corev1.NodeReady {
+				nodeIsReady = con.Status == corev1.ConditionTrue
+				break
+			}
+		}
+		if nodeIsReady {
+			readyNode = &node
+			break
+		}
+	}
+	return readyNode, nil
+}