mirror of https://github.com/portainer/portainer
68 lines
2.3 KiB
Go
68 lines
2.3 KiB
Go
package customtemplates
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
httperror "github.com/portainer/libhttp/error"
|
|
"github.com/portainer/libhttp/request"
|
|
"github.com/portainer/libhttp/response"
|
|
"github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/http/security"
|
|
"github.com/portainer/portainer/api/internal/authorization"
|
|
)
|
|
|
|
func (handler *Handler) customTemplateList(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
|
|
customTemplates, err := handler.DataStore.CustomTemplate().CustomTemplates()
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve custom templates from the database", err}
|
|
}
|
|
|
|
stackType, _ := request.RetrieveNumericQueryParameter(r, "type", true)
|
|
|
|
resourceControls, err := handler.DataStore.ResourceControl().ResourceControls()
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve resource controls from the database", err}
|
|
}
|
|
|
|
customTemplates = authorization.DecorateCustomTemplates(customTemplates, resourceControls)
|
|
|
|
customTemplates = filterTemplatesByEngineType(customTemplates, portainer.StackType(stackType))
|
|
|
|
securityContext, err := security.RetrieveRestrictedRequestContext(r)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve info from request context", err}
|
|
}
|
|
|
|
if !securityContext.IsAdmin {
|
|
user, err := handler.DataStore.User().User(securityContext.UserID)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve user information from the database", err}
|
|
}
|
|
|
|
userTeamIDs := make([]portainer.TeamID, 0)
|
|
for _, membership := range securityContext.UserMemberships {
|
|
userTeamIDs = append(userTeamIDs, membership.TeamID)
|
|
}
|
|
|
|
customTemplates = authorization.FilterAuthorizedCustomTemplates(customTemplates, user, userTeamIDs)
|
|
}
|
|
|
|
return response.JSON(w, customTemplates)
|
|
}
|
|
|
|
func filterTemplatesByEngineType(templates []portainer.CustomTemplate, stackType portainer.StackType) []portainer.CustomTemplate {
|
|
if stackType == 0 {
|
|
return templates
|
|
}
|
|
|
|
filteredTemplates := []portainer.CustomTemplate{}
|
|
|
|
for _, template := range templates {
|
|
if template.Type == stackType {
|
|
filteredTemplates = append(filteredTemplates, template)
|
|
}
|
|
}
|
|
|
|
return filteredTemplates
|
|
}
|