package cmd import ( "errors" "fmt" "net" "strconv" "strings" "github.com/edgelesssys/constellation/cli/azure" "github.com/edgelesssys/constellation/cli/cloudprovider" "github.com/edgelesssys/constellation/cli/gcp" "github.com/spf13/cobra" ) // warnAWS warns that AWS isn't supported. func warnAWS(providerPos int) cobra.PositionalArgs { return func(cmd *cobra.Command, args []string) error { if cloudprovider.FromString(args[providerPos]) == cloudprovider.AWS { return errors.New("AWS isn't supported by this version of Constellation") } return nil } } func isCloudProvider(arg int) cobra.PositionalArgs { return func(cmd *cobra.Command, args []string) error { if provider := cloudprovider.FromString(args[arg]); provider == cloudprovider.Unknown { return fmt.Errorf("argument %s isn't a valid cloud provider", args[arg]) } return nil } } func validInstanceTypeForProvider(cmd *cobra.Command, insType string, provider cloudprovider.Provider) error { switch provider { case cloudprovider.GCP: for _, instanceType := range gcp.InstanceTypes { if insType == instanceType { return nil } } cmd.SetUsageTemplate("GCP instance types:\n" + formatInstanceTypes(gcp.InstanceTypes)) cmd.SilenceUsage = false return fmt.Errorf("%s isn't a valid GCP instance type", insType) case cloudprovider.Azure: for _, instanceType := range azure.InstanceTypes { if insType == instanceType { return nil } } cmd.SetUsageTemplate("Azure instance types:\n" + formatInstanceTypes(azure.InstanceTypes)) cmd.SilenceUsage = false return fmt.Errorf("%s isn't a valid Azure instance type", insType) default: return fmt.Errorf("%s isn't a valid cloud platform", provider) } } func validateEndpoint(endpoint string, defaultPort int) (string, error) { _, _, err := net.SplitHostPort(endpoint) if err == nil { return endpoint, nil } if strings.Contains(err.Error(), "missing port in address") { return net.JoinHostPort(endpoint, strconv.Itoa(defaultPort)), nil } return "", err }