fix(edgestacks): return 400 instead of 500 on edge stack create when an invalid payload is provided EE-4429 (#7880)

pull/8055/head
matias-portainer 2 years ago committed by GitHub
parent 573e05d1c7
commit c21921a08d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -19,6 +19,14 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
) )
type InvalidPayloadError struct {
msg string
}
func (e *InvalidPayloadError) Error() string {
return e.msg
}
// @id EdgeStackCreate // @id EdgeStackCreate
// @summary Create an EdgeStack // @summary Create an EdgeStack
// @description **Access policy**: administrator // @description **Access policy**: administrator
@ -42,7 +50,13 @@ func (handler *Handler) edgeStackCreate(w http.ResponseWriter, r *http.Request)
edgeStack, err := handler.createSwarmStack(method, r) edgeStack, err := handler.createSwarmStack(method, r)
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to create Edge stack", err) var payloadError *InvalidPayloadError
switch {
case errors.As(err, &payloadError):
return httperror.BadRequest("Invalid payload", err)
default:
return httperror.InternalServerError("Unable to create Edge stack", err)
}
} }
return response.JSON(w, edgeStack) return response.JSON(w, edgeStack)
@ -76,13 +90,13 @@ type swarmStackFromFileContentPayload struct {
func (payload *swarmStackFromFileContentPayload) Validate(r *http.Request) error { func (payload *swarmStackFromFileContentPayload) Validate(r *http.Request) error {
if govalidator.IsNull(payload.Name) { if govalidator.IsNull(payload.Name) {
return errors.New("Invalid stack name") return &InvalidPayloadError{msg: "Invalid stack name"}
} }
if govalidator.IsNull(payload.StackFileContent) { if govalidator.IsNull(payload.StackFileContent) {
return errors.New("Invalid stack file content") return &InvalidPayloadError{msg: "Invalid stack file content"}
} }
if payload.EdgeGroups == nil || len(payload.EdgeGroups) == 0 { if payload.EdgeGroups == nil || len(payload.EdgeGroups) == 0 {
return errors.New("Edge Groups are mandatory for an Edge stack") return &InvalidPayloadError{msg: "Edge Groups are mandatory for an Edge stack"}
} }
return nil return nil
} }
@ -194,19 +208,19 @@ type swarmStackFromGitRepositoryPayload struct {
func (payload *swarmStackFromGitRepositoryPayload) Validate(r *http.Request) error { func (payload *swarmStackFromGitRepositoryPayload) Validate(r *http.Request) error {
if govalidator.IsNull(payload.Name) { if govalidator.IsNull(payload.Name) {
return errors.New("Invalid stack name") return &InvalidPayloadError{msg: "Invalid stack name"}
} }
if govalidator.IsNull(payload.RepositoryURL) || !govalidator.IsURL(payload.RepositoryURL) { if govalidator.IsNull(payload.RepositoryURL) || !govalidator.IsURL(payload.RepositoryURL) {
return errors.New("Invalid repository URL. Must correspond to a valid URL format") return &InvalidPayloadError{msg: "Invalid repository URL. Must correspond to a valid URL format"}
} }
if payload.RepositoryAuthentication && (govalidator.IsNull(payload.RepositoryUsername) || govalidator.IsNull(payload.RepositoryPassword)) { if payload.RepositoryAuthentication && (govalidator.IsNull(payload.RepositoryUsername) || govalidator.IsNull(payload.RepositoryPassword)) {
return errors.New("Invalid repository credentials. Username and password must be specified when authentication is enabled") return &InvalidPayloadError{msg: "Invalid repository credentials. Password must be specified when authentication is enabled"}
} }
if govalidator.IsNull(payload.FilePathInRepository) { if govalidator.IsNull(payload.FilePathInRepository) {
payload.FilePathInRepository = filesystem.ComposeFileDefaultName payload.FilePathInRepository = filesystem.ComposeFileDefaultName
} }
if payload.EdgeGroups == nil || len(payload.EdgeGroups) == 0 { if payload.EdgeGroups == nil || len(payload.EdgeGroups) == 0 {
return errors.New("Edge Groups are mandatory for an Edge stack") return &InvalidPayloadError{msg: "Edge Groups are mandatory for an Edge stack"}
} }
return nil return nil
} }
@ -293,26 +307,26 @@ type swarmStackFromFileUploadPayload struct {
func (payload *swarmStackFromFileUploadPayload) Validate(r *http.Request) error { func (payload *swarmStackFromFileUploadPayload) Validate(r *http.Request) error {
name, err := request.RetrieveMultiPartFormValue(r, "Name", false) name, err := request.RetrieveMultiPartFormValue(r, "Name", false)
if err != nil { if err != nil {
return errors.New("Invalid stack name") return &InvalidPayloadError{msg: "Invalid stack name"}
} }
payload.Name = name payload.Name = name
composeFileContent, _, err := request.RetrieveMultiPartFormFile(r, "file") composeFileContent, _, err := request.RetrieveMultiPartFormFile(r, "file")
if err != nil { if err != nil {
return errors.New("Invalid Compose file. Ensure that the Compose file is uploaded correctly") return &InvalidPayloadError{msg: "Invalid Compose file. Ensure that the Compose file is uploaded correctly"}
} }
payload.StackFileContent = composeFileContent payload.StackFileContent = composeFileContent
var edgeGroups []portainer.EdgeGroupID var edgeGroups []portainer.EdgeGroupID
err = request.RetrieveMultiPartFormJSONValue(r, "EdgeGroups", &edgeGroups, false) err = request.RetrieveMultiPartFormJSONValue(r, "EdgeGroups", &edgeGroups, false)
if err != nil || len(edgeGroups) == 0 { if err != nil || len(edgeGroups) == 0 {
return errors.New("Edge Groups are mandatory for an Edge stack") return &InvalidPayloadError{msg: "Edge Groups are mandatory for an Edge stack"}
} }
payload.EdgeGroups = edgeGroups payload.EdgeGroups = edgeGroups
deploymentType, err := request.RetrieveNumericMultiPartFormValue(r, "DeploymentType", true) deploymentType, err := request.RetrieveNumericMultiPartFormValue(r, "DeploymentType", true)
if err != nil { if err != nil {
return errors.New("Invalid deployment type") return &InvalidPayloadError{msg: "Invalid deployment type"}
} }
payload.DeploymentType = portainer.EdgeStackDeploymentType(deploymentType) payload.DeploymentType = portainer.EdgeStackDeploymentType(deploymentType)

@ -322,19 +322,19 @@ func TestCreateWithInvalidPayload(t *testing.T) {
Name: "Empty swarmStackFromFileContentPayload with string method", Name: "Empty swarmStackFromFileContentPayload with string method",
Payload: swarmStackFromFileContentPayload{}, Payload: swarmStackFromFileContentPayload{},
QueryString: "method=string", QueryString: "method=string",
ExpectedStatusCode: 500, ExpectedStatusCode: 400,
}, },
{ {
Name: "Empty swarmStackFromFileContentPayload with repository method", Name: "Empty swarmStackFromFileContentPayload with repository method",
Payload: swarmStackFromFileContentPayload{}, Payload: swarmStackFromFileContentPayload{},
QueryString: "method=repository", QueryString: "method=repository",
ExpectedStatusCode: 500, ExpectedStatusCode: 400,
}, },
{ {
Name: "Empty swarmStackFromFileContentPayload with file method", Name: "Empty swarmStackFromFileContentPayload with file method",
Payload: swarmStackFromFileContentPayload{}, Payload: swarmStackFromFileContentPayload{},
QueryString: "method=file", QueryString: "method=file",
ExpectedStatusCode: 500, ExpectedStatusCode: 400,
}, },
{ {
Name: "Duplicated EdgeStack Name", Name: "Duplicated EdgeStack Name",
@ -356,7 +356,7 @@ func TestCreateWithInvalidPayload(t *testing.T) {
DeploymentType: edgeStack.DeploymentType, DeploymentType: edgeStack.DeploymentType,
}, },
QueryString: "method=string", QueryString: "method=string",
ExpectedStatusCode: 500, ExpectedStatusCode: 400,
}, },
{ {
Name: "EdgeStackDeploymentKubernetes with Docker endpoint", Name: "EdgeStackDeploymentKubernetes with Docker endpoint",
@ -378,7 +378,7 @@ func TestCreateWithInvalidPayload(t *testing.T) {
DeploymentType: portainer.EdgeStackDeploymentCompose, DeploymentType: portainer.EdgeStackDeploymentCompose,
}, },
QueryString: "method=string", QueryString: "method=string",
ExpectedStatusCode: 500, ExpectedStatusCode: 400,
}, },
{ {
Name: "Clone Git respository error", Name: "Clone Git respository error",

Loading…
Cancel
Save