gke-deploy/core/cluster/cluster.go

// Package cluster contains logic related to talking to a GKE cluster.
package cluster

import (
	"context"
	"fmt"

	"github.com/GoogleCloudPlatform/cloud-builders/gke-deploy/core/resource"
	"github.com/GoogleCloudPlatform/cloud-builders/gke-deploy/services"
)

// AuthorizeAccess authorizes kubectl to the cluster. In doing so, this also verifies the cluster
// exists.
func AuthorizeAccess(ctx context.Context, clusterName, clusterLocation, clusterProject string, useGcloud bool, gs services.GcloudService) error {
	if useGcloud {
		fmt.Println("using gcloud to get cluster credentials")
		if err := gs.ContainerClustersGetCredentials(ctx, clusterName, clusterLocation, clusterProject); err != nil {
			return fmt.Errorf("failed to authorize access: %v", err)
		}
	} else {
		fmt.Println("using go client libraries to get cluster credentials")
		if err := gs.ContainerClustersGetCredentialsGoClient(ctx, clusterName, clusterLocation, clusterProject); err != nil {
			return fmt.Errorf("failed to authorize access: %v", err)
		}
	}
	return nil
}

// ApplyConfigFromString applies a config string to the current context's cluster.
func ApplyConfigFromString(ctx context.Context, configString, namespace string, ks services.KubectlService) error {
	if err := ks.ApplyFromString(ctx, configString, namespace); err != nil {
		return fmt.Errorf("failed to apply config from string: %v", err)
	}
	return nil
}

// GetDeployedObject gets an object deployed to the current context's cluster.
func GetDeployedObject(ctx context.Context, kind, name, namespace string, ks services.KubectlService) (*resource.Object, error) {
	objYaml, err := ks.Get(ctx, kind, name, namespace, "yaml", false)
	if err != nil {
		return nil, fmt.Errorf("failed to get config of deployed object: %v", err)
	}
	return resource.DecodeFromYAML(ctx, []byte(objYaml))
}

// DeployedObjectExists returns true if a deployed object exists in the current context's cluster,
// else false.
func DeployedObjectExists(ctx context.Context, kind, name, namespace string, ks services.KubectlService) (bool, error) {
	objYaml, err := ks.Get(ctx, kind, name, namespace, "yaml", true)
	if err != nil {
		return false, fmt.Errorf("failed to get config of deployed object: %v", err)
	}
	if objYaml == "" {
		return false, nil
	}
	return true, nil
}