mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
138 lines
4.7 KiB
Go
138 lines
4.7 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package gcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/cloud/metadata"
|
|
"github.com/edgelesssys/constellation/v2/internal/gcpshared"
|
|
)
|
|
|
|
// API handles all GCP API requests.
|
|
type API interface {
|
|
// UID retrieves the current instances uid.
|
|
UID() (string, error)
|
|
// RetrieveInstances retrieves a list of all accessible GCP instances with their metadata.
|
|
RetrieveInstances(ctx context.Context, project, zone string) ([]metadata.InstanceMetadata, error)
|
|
// RetrieveInstances retrieves a single GCP instances with its metadata.
|
|
RetrieveInstance(ctx context.Context, project, zone, instanceName string) (metadata.InstanceMetadata, error)
|
|
// RetrieveInstanceMetadata retrieves the GCP instance metadata of the current instance.
|
|
RetrieveInstanceMetadata(attr string) (string, error)
|
|
// RetrieveProjectID retrieves the GCP projectID containing the current instance.
|
|
RetrieveProjectID() (string, error)
|
|
// RetrieveZone retrieves the GCP zone containing the current instance.
|
|
RetrieveZone() (string, error)
|
|
// RetrieveInstanceName retrieves the instance name of the current instance.
|
|
RetrieveInstanceName() (string, error)
|
|
// RetrieveSubnetworkAliasCIDR retrieves the subnetwork CIDR of the current instance.
|
|
RetrieveSubnetworkAliasCIDR(ctx context.Context, project, zone, instanceName string) (string, error)
|
|
// RetrieveLoadBalancerEndpoint retrieves the load balancer endpoint of the current instance.
|
|
RetrieveLoadBalancerEndpoint(ctx context.Context, project string) (string, error)
|
|
// SetInstanceMetadata sets metadata key: value of the instance specified by project, zone and instanceName.
|
|
SetInstanceMetadata(ctx context.Context, project, zone, instanceName, key, value string) error
|
|
// UnsetInstanceMetadata removes a metadata key-value pair of the instance specified by project, zone and instanceName.
|
|
UnsetInstanceMetadata(ctx context.Context, project, zone, instanceName, key string) error
|
|
}
|
|
|
|
// Metadata implements core.ProviderMetadata interface.
|
|
type Metadata struct {
|
|
api API
|
|
}
|
|
|
|
// New creates a new Provider with real API and FS.
|
|
func New(api API) *Metadata {
|
|
return &Metadata{
|
|
api: api,
|
|
}
|
|
}
|
|
|
|
// List retrieves all instances belonging to the current constellation.
|
|
func (m *Metadata) List(ctx context.Context) ([]metadata.InstanceMetadata, error) {
|
|
project, err := m.api.RetrieveProjectID()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
zone, err := m.api.RetrieveZone()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
instances, err := m.api.RetrieveInstances(ctx, project, zone)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("retrieving instances list from GCP api: %w", err)
|
|
}
|
|
return instances, nil
|
|
}
|
|
|
|
// Self retrieves the current instance.
|
|
func (m *Metadata) Self(ctx context.Context) (metadata.InstanceMetadata, error) {
|
|
project, err := m.api.RetrieveProjectID()
|
|
if err != nil {
|
|
return metadata.InstanceMetadata{}, err
|
|
}
|
|
zone, err := m.api.RetrieveZone()
|
|
if err != nil {
|
|
return metadata.InstanceMetadata{}, err
|
|
}
|
|
instanceName, err := m.api.RetrieveInstanceName()
|
|
if err != nil {
|
|
return metadata.InstanceMetadata{}, err
|
|
}
|
|
return m.api.RetrieveInstance(ctx, project, zone, instanceName)
|
|
}
|
|
|
|
// GetInstance retrieves an instance using its providerID.
|
|
func (m *Metadata) GetInstance(ctx context.Context, providerID string) (metadata.InstanceMetadata, error) {
|
|
project, zone, instanceName, err := gcpshared.SplitProviderID(providerID)
|
|
if err != nil {
|
|
return metadata.InstanceMetadata{}, fmt.Errorf("invalid providerID: %w", err)
|
|
}
|
|
return m.api.RetrieveInstance(ctx, project, zone, instanceName)
|
|
}
|
|
|
|
// GetSubnetworkCIDR returns the subnetwork CIDR of the current instance.
|
|
func (m *Metadata) GetSubnetworkCIDR(ctx context.Context) (string, error) {
|
|
project, err := m.api.RetrieveProjectID()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
zone, err := m.api.RetrieveZone()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
instanceName, err := m.api.RetrieveInstanceName()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return m.api.RetrieveSubnetworkAliasCIDR(ctx, project, zone, instanceName)
|
|
}
|
|
|
|
// SupportsLoadBalancer returns true if the cloud provider supports load balancers.
|
|
func (m *Metadata) SupportsLoadBalancer() bool {
|
|
return true
|
|
}
|
|
|
|
// GetLoadBalancerEndpoint returns the endpoint of the load balancer.
|
|
func (m *Metadata) GetLoadBalancerEndpoint(ctx context.Context) (string, error) {
|
|
project, err := m.api.RetrieveProjectID()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return m.api.RetrieveLoadBalancerEndpoint(ctx, project)
|
|
}
|
|
|
|
// UID retrieves the UID of the constellation.
|
|
func (m *Metadata) UID(ctx context.Context) (string, error) {
|
|
return m.api.UID()
|
|
}
|
|
|
|
// Supported is used to determine if metadata API is implemented for this cloud provider.
|
|
func (m *Metadata) Supported() bool {
|
|
return true
|
|
}
|