Merge branch 'master' into fly_develop

wangpengfei

2023-06-02 064c0874e5fd041c4641ef873d1bf72ac98a184d

 src/cluster/cluster.go

@@ -4,12 +4,15 @@
   "bytes"
   "encoding/json"
   "fmt"
   "io/ioutil"
   "log"
   "net/http"
   "os"
   "path/filepath"
   "strings"

   "basic.com/aps/aps_deploy.git/src/util"
   "basic.com/aps/aps_deploy.git/src/rancher"
   "basic.com/aps/aps_deploy.git/src/util"
)

type Cluster struct {
@@ -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 {
@@ -111,7 +120,7 @@

   req, err := http.NewRequest("GET", url, nil)
   if err != nil {
      return "", fmt.Errorf("failed to create cluster list request: %v", err)
      return "", fmt.Errorf("failed to get cluster list request: %v", err)
   }

   req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", bearerToken))
@@ -146,6 +155,101 @@
   return "", fmt.Errorf("cluster '%s' not found", clusterName)
}

func GetClusters(serverURL, bearerToken string) ([]string, error) {

   url := fmt.Sprintf("%s/v3/clusters", serverURL)

   req, err := http.NewRequest("GET", url, nil)
   if err != nil {
      return nil, fmt.Errorf("failed to get cluster list request: %v", err)
   }

   req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", bearerToken))

   client := util.CreateHTTPClient()

   resp, err := client.Do(req)
   if err != nil {
      return nil, fmt.Errorf("failed to get cluster list: %v", err)
   }
   defer resp.Body.Close()

   if resp.StatusCode != http.StatusOK {
      return nil, fmt.Errorf("failed to get cluster list: unexpected status code %d", resp.StatusCode)
   }

   // 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)
   }

   var clusters = make([]string, 0)
   // Print cluster names
   for _, cluster := range clustersResponse.Data {
      clusters = append(clusters, cluster.ID)
      fmt.Printf("Cluster ID: %s, Name: %s\n", cluster.ID, cluster.Name)
   }

   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"`
@@ -154,7 +258,6 @@
   SSHPassword string   `json:"sshPassword"`
   SSHPort     int      `json:"sshPort"`
}


type ClusterCreateRequest struct {
   Name                          string      `json:"name"`
@@ -247,16 +350,17 @@
   return true, nil
}

func CreateCluster(rancherConfig rancher.RancherConfig, clusterName string) error {
func CreateCluster(rancherClusterConfig rancher.RancherClusterConfig, clusterName string) error {
   requestBody := createClusterData(clusterName)
   fmt.Println(rancherClusterConfig.RancherURL)

   url := fmt.Sprintf("%s/v3/clusters", rancherConfig.RancherURL)
   url := fmt.Sprintf("%s/v3/clusters", rancherClusterConfig.RancherURL)
   req, err := http.NewRequest("POST", url, bytes.NewBuffer(requestBody))
   if err != nil {
      return fmt.Errorf("Failed to create HTTP request: %v", err)
   }

   req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", rancherConfig.BearerToken))
   req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", rancherClusterConfig.BearerToken))
   req.Header.Set("Content-Type", "application/json")

   client := util.CreateHTTPClient()