Split cmd package

cli/cloud/cloudcmd/create.go

@@ -0,0 +1,123 @@
+package cloudcmd
+
+import (
+	"context"
+	"fmt"
+	"io"
+
+	azurecl "github.com/edgelesssys/constellation/cli/azure/client"
+	"github.com/edgelesssys/constellation/cli/cloudprovider"
+	"github.com/edgelesssys/constellation/cli/gcp"
+	"github.com/edgelesssys/constellation/cli/gcp/client"
+	gcpcl "github.com/edgelesssys/constellation/cli/gcp/client"
+	"github.com/edgelesssys/constellation/internal/config"
+	"github.com/edgelesssys/constellation/internal/state"
+)
+
+// Creator creates cloud resources.
+type Creator struct {
+	out            io.Writer
+	newGCPClient   func(ctx context.Context, project, zone, region, name string) (gcpclient, error)
+	newAzureClient func(subscriptionID, tenantID, name, location string) (azureclient, error)
+}
+
+// NewCreator creates a new creator.
+func NewCreator(out io.Writer) *Creator {
+	return &Creator{
+		out: out,
+		newGCPClient: func(ctx context.Context, project, zone, region, name string) (gcpclient, error) {
+			return gcpcl.NewInitialized(ctx, project, zone, region, name)
+		},
+		newAzureClient: func(subscriptionID, tenantID, name, location string) (azureclient, error) {
+			return azurecl.NewInitialized(subscriptionID, tenantID, name, location)
+		},
+	}
+}
+
+// Create creates the handed amount of instances and all the needed resources.
+func (c *Creator) Create(ctx context.Context, provider cloudprovider.CloudProvider, config *config.Config, name, insType string, coordCount, nodeCount int,
+) (state.ConstellationState, error) {
+	switch provider {
+	case cloudprovider.GCP:
+		cl, err := c.newGCPClient(
+			ctx,
+			*config.Provider.GCP.Project,
+			*config.Provider.GCP.Zone,
+			*config.Provider.GCP.Region,
+			name,
+		)
+		if err != nil {
+			return state.ConstellationState{}, err
+		}
+		defer cl.Close()
+		return c.createGCP(ctx, cl, config, insType, coordCount, nodeCount)
+	case cloudprovider.Azure:
+		cl, err := c.newAzureClient(
+			*config.Provider.Azure.SubscriptionID,
+			*config.Provider.Azure.TenantID,
+			name,
+			*config.Provider.Azure.Location,
+		)
+		if err != nil {
+			return state.ConstellationState{}, err
+		}
+		return c.createAzure(ctx, cl, config, insType, coordCount, nodeCount)
+	default:
+		return state.ConstellationState{}, fmt.Errorf("unsupported cloud provider: %s", provider)
+	}
+}
+
+func (c *Creator) createGCP(ctx context.Context, cl gcpclient, config *config.Config, insType string, coordCount, nodeCount int,
+) (stat state.ConstellationState, retErr error) {
+	defer rollbackOnError(context.Background(), c.out, &retErr, &rollbackerGCP{client: cl})
+
+	if err := cl.CreateVPCs(ctx, *config.Provider.GCP.VPCsInput); err != nil {
+		return state.ConstellationState{}, err
+	}
+	if err := cl.CreateFirewall(ctx, *config.Provider.GCP.FirewallInput); err != nil {
+		return state.ConstellationState{}, err
+	}
+
+	createInput := client.CreateInstancesInput{
+		CountCoordinators: coordCount,
+		CountNodes:        nodeCount,
+		ImageId:           *config.Provider.GCP.Image,
+		InstanceType:      insType,
+		StateDiskSizeGB:   *config.StateDiskSizeGB,
+		KubeEnv:           gcp.KubeEnv,
+		DisableCVM:        *config.Provider.GCP.DisableCVM,
+	}
+	if err := cl.CreateInstances(ctx, createInput); err != nil {
+		return state.ConstellationState{}, err
+	}
+
+	return cl.GetState()
+}
+
+func (c *Creator) createAzure(ctx context.Context, cl azureclient, config *config.Config, insType string, coordCount, nodeCount int,
+) (stat state.ConstellationState, retErr error) {
+	defer rollbackOnError(context.Background(), c.out, &retErr, &rollbackerAzure{client: cl})
+
+	if err := cl.CreateResourceGroup(ctx); err != nil {
+		return state.ConstellationState{}, err
+	}
+	if err := cl.CreateVirtualNetwork(ctx); err != nil {
+		return state.ConstellationState{}, err
+	}
+	if err := cl.CreateSecurityGroup(ctx, *config.Provider.Azure.NetworkSecurityGroupInput); err != nil {
+		return state.ConstellationState{}, err
+	}
+	createInput := azurecl.CreateInstancesInput{
+		CountCoordinators:    coordCount,
+		CountNodes:           nodeCount,
+		InstanceType:         insType,
+		StateDiskSizeGB:      *config.StateDiskSizeGB,
+		Image:                *config.Provider.Azure.Image,
+		UserAssingedIdentity: *config.Provider.Azure.UserAssignedIdentity,
+	}
+	if err := cl.CreateInstances(ctx, createInput); err != nil {
+		return state.ConstellationState{}, err
+	}
+
+	return cl.GetState()
+}