2022-03-22 15:03:15 +00:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"strconv"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/edgelesssys/constellation/cli/azure"
|
|
|
|
"github.com/edgelesssys/constellation/cli/ec2"
|
|
|
|
"github.com/edgelesssys/constellation/cli/gcp"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
)
|
|
|
|
|
|
|
|
// isIntArg checks if argument at position arg is an integer.
|
|
|
|
func isIntArg(arg int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
if _, err := strconv.Atoi(args[arg]); err != nil {
|
|
|
|
return fmt.Errorf("argument %d must be an integer", arg)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-04-04 14:44:15 +00:00
|
|
|
// isIntGreaterArg checks if argument at position arg is an integer and greater i.
|
2022-03-22 15:03:15 +00:00
|
|
|
func isIntGreaterArg(arg int, i int) cobra.PositionalArgs {
|
|
|
|
return cobra.MatchAll(isIntArg(arg), func(cmd *cobra.Command, args []string) error {
|
|
|
|
if v, _ := strconv.Atoi(args[arg]); v <= i {
|
|
|
|
return fmt.Errorf("argument %d must be greater %d, but it's %d", arg, i, v)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2022-04-04 13:55:58 +00:00
|
|
|
// isValidAWSCoordinatorCount checks additional conditions for the AWS coordinator count.
|
|
|
|
func isValidAWSCoordinatorCount(coordCountPos, providerPos int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
if strings.ToLower(args[providerPos]) != "aws" {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
v, err := strconv.Atoi(args[coordCountPos])
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("argument %d must be an integer", coordCountPos)
|
|
|
|
}
|
|
|
|
if v != 1 {
|
|
|
|
return fmt.Errorf(
|
|
|
|
"argument %d is %d, invalid coordinator count for AWS, has to be 1",
|
|
|
|
coordCountPos, v,
|
|
|
|
)
|
2022-04-04 14:44:15 +00:00
|
|
|
}
|
|
|
|
return nil
|
2022-04-04 13:55:58 +00:00
|
|
|
}
|
2022-04-04 14:44:15 +00:00
|
|
|
}
|
|
|
|
|
2022-03-22 15:03:15 +00:00
|
|
|
// isIntGreaterZeroArg checks if argument at position arg is a positive non zero integer.
|
|
|
|
func isIntGreaterZeroArg(arg int) cobra.PositionalArgs {
|
2022-04-04 13:55:58 +00:00
|
|
|
return isIntGreaterArg(arg, 0)
|
2022-03-22 15:03:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// isEC2InstanceType checks if argument at position arg is a key in m.
|
|
|
|
// The argument will always be converted to lower case letters.
|
|
|
|
func isEC2InstanceType(arg int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
if _, ok := ec2.InstanceTypes[strings.ToLower(args[arg])]; !ok {
|
|
|
|
return fmt.Errorf("'%s' isn't an AWS EC2 instance type", args[arg])
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func isGCPInstanceType(arg int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
for _, instanceType := range gcp.InstanceTypes {
|
|
|
|
if args[arg] == instanceType {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fmt.Errorf("argument %s isn't a valid GCP instance type", args[arg])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func isAzureInstanceType(arg int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
for _, instanceType := range azure.InstanceTypes {
|
|
|
|
if args[arg] == instanceType {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fmt.Errorf("argument %s isn't a valid Azure instance type", args[arg])
|
|
|
|
}
|
|
|
|
}
|
2022-04-04 13:55:58 +00:00
|
|
|
|
|
|
|
// isInstanceTypeForProvider returns a argument validation function that checks if the argument
|
|
|
|
// at position typePos is a valid instance type for the cloud provider string at position
|
|
|
|
// providerPos.
|
|
|
|
func isInstanceTypeForProvider(typePos, providerPos int) cobra.PositionalArgs {
|
|
|
|
return func(cmd *cobra.Command, args []string) error {
|
|
|
|
if len(args) < 2 {
|
|
|
|
return fmt.Errorf("requires 2 arguments, but only %d are provided", len(args))
|
|
|
|
}
|
|
|
|
if len(args) <= typePos {
|
|
|
|
return fmt.Errorf(
|
|
|
|
"%d arguments provided, but index %d of typePos is out of bound",
|
|
|
|
len(args), typePos,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
if len(args) <= providerPos {
|
|
|
|
return fmt.Errorf(
|
|
|
|
"%d arguments provided, but index %d of providerPos is out of bound",
|
|
|
|
len(args), providerPos,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
switch strings.ToLower(args[providerPos]) {
|
|
|
|
case "aws":
|
|
|
|
return isEC2InstanceType(typePos)(cmd, args)
|
|
|
|
case "gcp":
|
|
|
|
return isGCPInstanceType(typePos)(cmd, args)
|
|
|
|
case "azure":
|
|
|
|
return isAzureInstanceType(typePos)(cmd, args)
|
|
|
|
default:
|
|
|
|
return fmt.Errorf("argument %s isn't a valid cloud platform", args[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|