mirror of https://github.com/portainer/portainer
76 lines
2.7 KiB
Go
76 lines
2.7 KiB
Go
package users
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/portainer/portainer"
|
|
httperror "github.com/portainer/portainer/http/error"
|
|
"github.com/portainer/portainer/http/request"
|
|
"github.com/portainer/portainer/http/response"
|
|
"github.com/portainer/portainer/http/security"
|
|
)
|
|
|
|
type userUpdatePayload struct {
|
|
Password string
|
|
Role int
|
|
}
|
|
|
|
func (payload *userUpdatePayload) Validate(r *http.Request) error {
|
|
if payload.Role != 0 && payload.Role != 1 && payload.Role != 2 {
|
|
return portainer.Error("Invalid role value. Value must be one of: 1 (administrator) or 2 (regular user)")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// PUT request on /api/users/:id
|
|
func (handler *Handler) userUpdate(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
|
|
userID, err := request.RetrieveNumericRouteVariableValue(r, "id")
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusBadRequest, "Invalid user identifier route variable", err}
|
|
}
|
|
|
|
tokenData, err := security.RetrieveTokenData(r)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve user authentication token", err}
|
|
}
|
|
|
|
if tokenData.Role != portainer.AdministratorRole && tokenData.ID != portainer.UserID(userID) {
|
|
return &httperror.HandlerError{http.StatusForbidden, "Permission denied to update user", portainer.ErrUnauthorized}
|
|
}
|
|
|
|
var payload userUpdatePayload
|
|
err = request.DecodeAndValidateJSONPayload(r, &payload)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusBadRequest, "Invalid request payload", err}
|
|
}
|
|
|
|
if tokenData.Role != portainer.AdministratorRole && payload.Role != 0 {
|
|
return &httperror.HandlerError{http.StatusForbidden, "Permission denied to update user to administrator role", portainer.ErrResourceAccessDenied}
|
|
}
|
|
|
|
user, err := handler.UserService.User(portainer.UserID(userID))
|
|
if err == portainer.ErrUserNotFound {
|
|
return &httperror.HandlerError{http.StatusNotFound, "Unable to find a user with the specified identifier inside the database", err}
|
|
} else if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to find a user with the specified identifier inside the database", err}
|
|
}
|
|
|
|
if payload.Password != "" {
|
|
user.Password, err = handler.CryptoService.Hash(payload.Password)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to hash user password", portainer.ErrCryptoHashFailure}
|
|
}
|
|
}
|
|
|
|
if payload.Role != 0 {
|
|
user.Role = portainer.UserRole(payload.Role)
|
|
}
|
|
|
|
err = handler.UserService.UpdateUser(user.ID, user)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to persist user changes inside the database", err}
|
|
}
|
|
|
|
return response.JSON(w, user)
|
|
}
|