internal/gcebeta/gcebeta.go

/*
Copyright 2023 Google LLC

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

// Package gcebeta is a lightweight testable wrapper around GCE compute beta APIs.
package gcebeta

import (
	"context"

	"github.com/pkg/errors"
	"google.golang.org/api/compute/v0.beta"
)

// GCEBeta is a wrapper for Google Compute Engine services.
type GCEBeta struct {
	service *compute.Service
}

// NewGCEClient creates a new GCE service wrapper.
func NewGCEClient(ctx context.Context) (*GCEBeta, error) {
	s, err := compute.NewService(ctx)
	if err != nil {
		return nil, errors.Wrap(err, "error creating GCE beta client")
	}

	return &GCEBeta{s}, nil
}

// Initialized checks if the compute service has been initialized.
func (g *GCEBeta) Initialized() bool {
	return g.service != nil
}

// OverrideComputeBasePath overrides the base path of the GCE clients.
func (g *GCEBeta) OverrideComputeBasePath(basePath string) {
	g.service.BasePath = basePath
}

// GetInstance retrieves a GCE Instance defined by the project, zone, and name provided.
func (g *GCEBeta) GetInstance(project, zone, instance string) (*compute.Instance, error) {
	return g.service.Instances.Get(project, zone, instance).Do()
}

// ListNodeGroups retrieves the node groups for a given project and zone.
func (g *GCEBeta) ListNodeGroups(project, zone string) (*compute.NodeGroupList, error) {
	return g.service.NodeGroups.List(project, zone).Do()
}

// ListNodeGroupNodes lists the nodes in a given node group.
func (g *GCEBeta) ListNodeGroupNodes(project, zone, nodeGroup string) (*compute.NodeGroupsListNodes, error) {
	return g.service.NodeGroups.ListNodes(project, zone, nodeGroup).Do()
}