mirror of https://github.com/portainer/portainer
fix(edgestacks): return 400 instead of 500 on edge stack create when an invalid payload is provided EE-4429 (#7880)
parent
573e05d1c7
commit
c21921a08d
|
@ -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…
Reference in New Issue