gce.go

/*
Copyright AppsCode Inc. and Contributors

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

    http://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 gce

import (
	"context"

	"go.bytebuilders.dev/resource-model/apis/cloud"
	"go.bytebuilders.dev/resource-model/apis/cloud/v1alpha1"
	"go.bytebuilders.dev/resource-model/pkg/credential"

	"google.golang.org/api/compute/v1"
	"google.golang.org/api/option"
)

type Client struct {
	GceProjectID   string
	ComputeService *compute.Service
	Ctx            context.Context
}

func NewClient(opts credential.GCE) (*Client, error) {
	g := &Client{
		GceProjectID: opts.ProjectID,
		Ctx:          context.Background(),
	}
	var err error
	g.ComputeService, err = getComputeService(g.Ctx, opts.ServiceAccountJson())
	if err != nil {
		return nil, err
	}
	return g, nil
}

func (g *Client) GetName() string {
	return cloud.GCE
}

func (g *Client) ListRegions() ([]v1alpha1.Region, error) {
	req := g.ComputeService.Regions.List(g.GceProjectID)

	var regions []v1alpha1.Region
	err := req.Pages(g.Ctx, func(list *compute.RegionList) error {
		for _, region := range list.Items {
			res, err := ParseRegion(region)
			if err != nil {
				return err
			}
			regions = append(regions, *res)
		}
		return nil
	})
	if err != nil {
		return nil, err
	}
	return regions, err
}

func (g *Client) ListZones() ([]string, error) {
	req := g.ComputeService.Zones.List(g.GceProjectID)
	var zones []string
	err := req.Pages(g.Ctx, func(list *compute.ZoneList) error {
		for _, zone := range list.Items {
			zones = append(zones, zone.Name)
		}
		return nil
	})
	if err != nil {
		return nil, err
	}
	return zones, nil
}

func (g *Client) ListMachineTypes() ([]v1alpha1.MachineType, error) {
	zoneList, err := g.ListZones()
	if err != nil {
		return nil, err
	}
	// machinesZone to keep zone to corresponding machine
	machinesZone := map[string][]string{}
	var machineTypes []v1alpha1.MachineType
	for _, zone := range zoneList {
		req := g.ComputeService.MachineTypes.List(g.GceProjectID, zone)
		err := req.Pages(g.Ctx, func(list *compute.MachineTypeList) error {
			for _, machine := range list.Items {
				res, err := ParseMachine(machine)
				if err != nil {
					return err
				}
				// to check whether we added this machine to machineTypes
				// if we found it then add this zone to machinesZone, else add the machine to machineTypes and also add this zone to machinesZone
				if zones, found := machinesZone[res.Spec.SKU]; found {
					machinesZone[res.Spec.SKU] = append(zones, zone)
				} else {
					machinesZone[res.Spec.SKU] = []string{zone}
					machineTypes = append(machineTypes, *res)
				}
			}
			return nil
		})
		if err != nil {
			return nil, err
		}
	}
	// update g.Data.MachineTypes[].Zones
	for index, instanceType := range machineTypes {
		machineTypes[index].Spec.Zones = machinesZone[instanceType.Spec.SKU]
	}
	return machineTypes, nil
}

func getComputeService(ctx context.Context, sajson string) (*compute.Service, error) {
	return compute.NewService(ctx, option.WithCredentialsJSON([]byte(sajson)), option.WithScopes(compute.ComputeScope))
}