AlexsJones/gravitywell

package gcp

import (
    "fmt"
    "github.com/AlexsJones/gravitywell/kinds"
    "time"

    "github.com/fatih/color"
    containerpb "google.golang.org/genproto/googleapis/container/v1"
)

func (g *GCPProvider) Create(clusterp kinds.ProviderCluster) error {

    //Convert generic node pool type into a specific GCP resource
    var convertedNodePool []*containerpb.NodePool

    for _, model := range clusterp.NodePools {
        nodePool := new(containerpb.NodePool)
        nodePool.Name = model.NodePool.Name
        nodePool.InitialNodeCount = int32(model.NodePool.Count)
        nodePool.Config = &containerpb.NodeConfig{
            MachineType: model.NodePool.NodeType,
            OauthScopes: clusterp.OauthScopes,
        }
        var labels = map[string]string{}

        if len(clusterp.Labels) > 0 {
            for index, element := range clusterp.Labels {
                labels[index] = element
            }
        }

        if len(model.NodePool.Labels) > 0 {
            for index, element := range model.NodePool.Labels {
                labels[index] = element
            }
        }
        nodePool.Config.Labels = labels

        convertedNodePool = append(convertedNodePool, nodePool)
    }

    // ----------------------------------------------------------------
    var cluster *containerpb.Cluster
    if len(clusterp.NodePools) == 0 {

        cluster = &containerpb.Cluster{
            Name:             clusterp.Name,
            Locations:        clusterp.Zones,
            InitialNodeCount: int32(clusterp.InitialNodeCount),
            NodeConfig: &containerpb.NodeConfig{
                MachineType: clusterp.InitialNodeType,
                OauthScopes: clusterp.OauthScopes,
            },
            ResourceLabels: clusterp.Labels,
        }

    } else {
        cluster = &containerpb.Cluster{
            Name:           clusterp.Name,
            Locations:      clusterp.Zones,
            NodePools:      convertedNodePool,
            ResourceLabels: clusterp.Labels,
        }
    }
    clusterReq := &containerpb.CreateClusterRequest{
        Parent: fmt.Sprintf("projects/%s/locations/%s", clusterp.Project,
            clusterp.Region),
        Cluster: cluster,
    }
    clusterResponse, err := g.ClusterManagerClient.CreateCluster(g.Context, clusterReq)
    if err != nil {
        color.Red(err.Error())
        return err
    }
    color.Blue(fmt.Sprintf("Started cluster build at %s", clusterResponse.StartTime))

    for {
        clust, err :=
            g.ClusterManagerClient.GetCluster(g.Context, &containerpb.GetClusterRequest{Name: fmt.Sprintf("projects/%s/locations/%s/clusters/%s", clusterp.Project,
                clusterp.Region, clusterp.Name)})
        if err != nil {
            return err
        }
        if clust.GetStatus() == containerpb.Cluster_RUNNING {
            color.Green("Cluster running")
            return nil
        }
        time.Sleep(time.Second)
    }

}