Refactoring

pkg/cloudapi/compute/pfw_add.go

@@ -0,0 +1,69 @@
+package compute
+
+import (
+	"context"
+	"errors"
+	"strconv"
+
+	"github.com/rudecs/decort-sdk/internal/validators"
+	"github.com/rudecs/decort-sdk/opts"
+	"github.com/rudecs/decort-sdk/typed"
+)
+
+type PFWAddRequest struct {
+	ComputeId       uint64 `url:"computeId"`
+	PublicPortStart uint64 `url:"publicPortStart"`
+	PublicPortEnd   uint64 `url:"publicPortEnd,omitempty"`
+	LocalBasePort   uint64 `url:"localBasePort"`
+	Proto           string `url:"proto"`
+}
+
+func (crq PFWAddRequest) Validate() error {
+	if crq.ComputeId == 0 {
+		return errors.New("validation-error: field ComputeId can not be empty or equal to 0")
+	}
+	if crq.PublicPortStart == 0 {
+		return errors.New("validation-error: field PublicPortStart can not be empty or equal to 0")
+	}
+	if crq.LocalBasePort == 0 {
+		return errors.New("validation-error: field LocalBasePort can not be empty or equal to 0")
+	}
+	if crq.Proto == "" {
+		return errors.New("validation-error: field Proto can not be empty")
+	}
+	validator := validators.StringInSlice(crq.Proto, []string{"tcp", "udp"})
+	if !validator {
+		return errors.New("validation-error: field Proto can be only tcp or udp")
+	}
+
+	return nil
+}
+
+func (c Compute) PFWAdd(ctx context.Context, req PFWAddRequest, options ...opts.DecortOpts) (uint64, error) {
+	err := req.Validate()
+	if err != nil {
+		return 0, err
+	}
+
+	url := "/compute/pfwAdd"
+	prefix := "/cloudapi"
+
+	option := opts.New(options)
+	if option != nil {
+		if option.IsAdmin {
+			prefix = "/" + option.AdminValue
+		}
+	}
+	url = prefix + url
+	res, err := c.client.DecortApiCall(ctx, typed.POST, url, req)
+	if err != nil {
+		return 0, err
+	}
+
+	result, err := strconv.ParseUint(string(res), 10, 64)
+	if err != nil {
+		return 0, err
+	}
+
+	return result, nil
+}