v1.0.0

pkg/cloudbroker/account/create.go

@@ -5,34 +5,92 @@ import (
 	"errors"
 	"net/http"
 	"strconv"
+
+	"github.com/rudecs/decort-sdk/internal/validators"
 )
 
+// Request struct for creating account
 type CreateRequest struct {
-	Name                   string `url:"name"`
-	Username               string `url:"username"`
-	EmailAddress           string `url:"emailaddress,omitempty"`
-	MaxMemoryCapacity      uint   `url:"maxMemoryCapacity,omitempty"`
-	MaxVDiskCapacity       uint   `url:"maxVDiskCapacity,omitempty"`
-	MaxCPUCapacity         uint   `url:"maxCPUCapacity,omitempty"`
-	MaxNetworkPeerTransfer uint   `url:"maxNetworkPeerTransfer,omitempty"`
-	MaxNumPublicIP         uint   `url:"maxNumPublicIP,omitempty"`
-	SendAccessEmails       bool   `url:"sendAccessEmails,omitempty"`
-	GPUUnits               uint   `url:"gpu_units,omitempty"`
+	// Display name
+	// Required: true
+	Name string `url:"name"`
+
+	// Name of the account
+	// Required: true
+	Username string `url:"username"`
+
+	// Email
+	// Required: false
+	EmailAddress string `url:"emailaddress,omitempty"`
+
+	// Max size of memory in MB
+	// Required: false
+	MaxMemoryCapacity uint64 `url:"maxMemoryCapacity,omitempty"`
+
+	// Max size of aggregated vdisks in GB
+	// Required: false
+	MaxVDiskCapacity uint64 `url:"maxVDiskCapacity,omitempty"`
+
+	// Max number of CPU cores
+	// Required: false
+	MaxCPUCapacity uint64 `url:"maxCPUCapacity,omitempty"`
+
+	// Max sent/received network transfer peering
+	// Required: false
+	MaxNetworkPeerTransfer uint64 `url:"maxNetworkPeerTransfer,omitempty"`
+
+	// Max number of assigned public IPs
+	// Required: false
+	MaxNumPublicIP uint64 `url:"maxNumPublicIP,omitempty"`
+
+	// If true send emails when a user is granted access to resources
+	// Required: false
+	SendAccessEmails bool `url:"sendAccessEmails,omitempty"`
+
+	// Limit (positive) or disable (0) GPU resources
+	// Required: false
+	GPUUnits uint64 `url:"gpu_units,omitempty"`
+
+	// List of strings with pools
+	// i.e.: ["sep1_poolName1", "sep2_poolName2", etc]
+	// Required: false
+	UniqPools []string `url:"uniqPools,omitempty"`
+
+	// Resource types available to create in this account
+	// Each element in a resource type slice must be one of:
+	//	- compute
+	//	- vins
+	//	- k8s
+	//	- openshift
+	//	- lb
+	//	- flipgroup
+	// Required: false
+	ResTypes []string `url:"resourceTypes,omitempty"`
 }
 
-func (arq CreateRequest) Validate() error {
+func (arq CreateRequest) validate() error {
 	if arq.Name == "" {
 		return errors.New("validation-error: field Name can not be empty")
 	}
-
 	if arq.Username == "" {
 		return errors.New("validation-error: field Username can not be empty")
 	}
+	if len(arq.ResTypes) > 0 {
+		for _, value := range arq.ResTypes {
+			validate := validators.StringInSlice(value, []string{"compute", "vins", "k8s", "openshift", "lb", "flipgroup"})
+			if !validate {
+				return errors.New("validation-error: Every resource type specified should be one of [compute, vins, k8s, openshift, lb, flipgroup]")
+			}
+		}
+	}
+
 	return nil
 }
 
+// Create creates account
+// Setting a cloud unit maximum to -1 or empty will not put any restrictions on the resource
 func (a Account) Create(ctx context.Context, req CreateRequest) (uint64, error) {
-	err := req.Validate()
+	err := req.validate()
 	if err != nil {
 		return 0, err
 	}
@@ -44,10 +102,10 @@ func (a Account) Create(ctx context.Context, req CreateRequest) (uint64, error)
 		return 0, err
 	}
 
-	id, err := strconv.ParseUint(string(res), 10, 64)
+	result, err := strconv.ParseUint(string(res), 10, 64)
 	if err != nil {
 		return 0, err
 	}
 
-	return id, nil
+	return result, nil
 }