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.Provider, 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() }