TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add entity type validation

Browse Source
This commit is contained in:
Neeraj Gupta
2024-08-16 11:40:31 +05:30
parent 05df5962ef
commit c759d37fd3
2 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/pkg/api/userentity.go
View File

@@ -25,6 +25,10 @@ func (h *UserEntityHandler) CreateKey(c *gin.Context) {
stacktrace.Propagate(ente.ErrBadRequest, fmt.Sprintf("Request binding failed %s", err)))
return
}
if err := request.Type.IsValid(); err != nil {
handler.Error(c, stacktrace.Propagate(err, "Invalid EntityType"))
return
}
err := h.Controller.CreateKey(c, request)
if err != nil {
handler.Error(c, stacktrace.Propagate(err, "Failed to create CreateKey"))
@@ -57,6 +61,10 @@ func (h *UserEntityHandler) CreateEntity(c *gin.Context) {
stacktrace.Propagate(ente.ErrBadRequest, fmt.Sprintf("Request binding failed %s", err)))
return
}
if err := request.Type.IsValid(); err != nil {
handler.Error(c, stacktrace.Propagate(err, "Invalid EntityType"))
return
}
resp, err := h.Controller.CreateEntity(c, request)
if err != nil {
handler.Error(c, stacktrace.Propagate(err, "Failed to create CreateEntity"))