From 4dcf624d6ab58ab847d2e1b7e57e88324bb7c59b Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期五, 02 六月 2023 16:25:45 +0800 Subject: [PATCH] fix --- src/cluster/cluster.go | 64 ++++++++++++++++++++++++++++++++ 1 files changed, 64 insertions(+), 0 deletions(-) diff --git a/src/cluster/cluster.go b/src/cluster/cluster.go index 82aa4bc..3af2437 100644 --- a/src/cluster/cluster.go +++ b/src/cluster/cluster.go @@ -4,8 +4,11 @@ "bytes" "encoding/json" "fmt" + "io/ioutil" "log" "net/http" + "os" + "path/filepath" "strings" "basic.com/aps/aps_deploy.git/src/rancher" @@ -19,6 +22,12 @@ type ClustersResponse struct { Data []Cluster `json:"data"` +} + +type KubectlConfigResponse struct { + BaseType string `json:"baseType"` + Config string `json:"config"` + Type string `json:"type"` } type RegistrationTokenResponse struct { @@ -172,6 +181,9 @@ // Parse the API response var clustersResponse ClustersResponse err = json.NewDecoder(resp.Body).Decode(&clustersResponse) + + log.Println("3333333333333333333333333") + log.Printf("%v", resp.Body) if err != nil { return nil, fmt.Errorf("failed to decode cluster list response: %v", err) } @@ -186,6 +198,58 @@ return clusters, nil } +func GetKubectlConfig(serverURL, bearerToken, clusterName string) (string, error) { + + url := fmt.Sprintf("%s/v3/clusters/%s?action=generateKubeconfig", serverURL, clusterName) + + req, err := http.NewRequest("POST", url, nil) + if err != nil { + return "", fmt.Errorf("failed to get cluster list request: %v", err) + } + + req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", bearerToken)) + req.Header.Set("Content-Type", "application/json") + req.Header.Set("Accept", "application/json") + + client := util.CreateHTTPClient() + + resp, err := client.Do(req) + if err != nil { + return "", fmt.Errorf("failed to get cluster list: %v", err) + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + return "", fmt.Errorf("failed to get cluster list: unexpected status code %d", resp.StatusCode) + } + + // Parse the API response + var kubectlConfig KubectlConfigResponse + err = json.NewDecoder(resp.Body).Decode(&kubectlConfig) + + if err != nil { + return "", fmt.Errorf("failed to decode cluster list response: %v", err) + } + + // 淇濆瓨 kubeconfig 鍒版枃浠� + homeDir, err := os.UserHomeDir() + if err != nil { + log.Fatalf("Failed to get home directory: %v", err) + } + kubeconfigDir := filepath.Join(homeDir, ".kube") + err = os.MkdirAll(kubeconfigDir, 0700) + if err != nil { + log.Fatalf("Failed to create .kube directory: %v", err) + } + kubeconfigPath := filepath.Join(kubeconfigDir, "config") + err = ioutil.WriteFile(kubeconfigPath, []byte(kubectlConfig.Config), 0600) + if err != nil { + log.Fatalf("Failed to save kubeconfig: %v", err) + } + + return kubectlConfig.Config, nil +} + type Node struct { ClusterName string `json:"clusterName"` Roles []string `json:"roles"` -- Gitblit v1.8.0