constellation/internal/cloud/cloudprovider/cloudprovider.go

/*
Copyright (c) Edgeless Systems GmbH

SPDX-License-Identifier: AGPL-3.0-only
*/

package cloudprovider

import (
	"encoding/json"
	"strings"
)

//go:generate stringer -type=Provider

// Provider is cloud provider used by the CLI.
type Provider uint32

const (
	// Unknown is default value for Provider.
	Unknown Provider = iota
	// AWS is Amazon Web Services.
	AWS
	// Azure cloud.
	Azure
	// GCP is Google Compute Platform.
	GCP
	// OpenStack is an open standard cloud computing platform.
	OpenStack
	// QEMU for a local emulated installation.
	QEMU
)

// MarshalJSON marshals the Provider to JSON string.
func (p Provider) MarshalJSON() ([]byte, error) {
	return json.Marshal(p.String())
}

// UnmarshalJSON unmarshals the Provider from JSON string.
func (p *Provider) UnmarshalJSON(b []byte) error {
	var s string
	if err := json.Unmarshal(b, &s); err != nil {
		return err
	}
	*p = FromString(s)
	return nil
}

// MarshalYAML marshals the Provider to YAML string.
func (p Provider) MarshalYAML() (any, error) {
	return p.String(), nil
}

// UnmarshalYAML unmarshals the Provider from YAML string.
func (p *Provider) UnmarshalYAML(unmarshal func(any) error) error {
	var s string
	if err := unmarshal(&s); err != nil {
		return err
	}
	*p = FromString(s)
	return nil
}

// FromString returns cloud provider from string.
func FromString(s string) Provider {
	s = strings.ToLower(s)
	if isOpenStackProvider(s) {
		return OpenStack
	}

	switch s {
	case "aws":
		return AWS
	case "azure":
		return Azure
	case "gcp":
		return GCP
	case "qemu":
		return QEMU
	default:
		return Unknown
	}
}

// IsOpenStackProvider returns true if the provider is based on OpenStack.
func isOpenStackProvider(s string) bool {
	switch strings.ToLower(s) {
	case "openstack", "stackit":
		return true
	}
	return false
}