mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-11 02:15:48 +02:00
enhance: validate settings value before save
This commit is contained in:
parent
d70e37c857
commit
d67b8428e4
8 changed files with 83 additions and 26 deletions
73
api/api.go
73
api/api.go
|
@ -4,11 +4,28 @@ import (
|
|||
"errors"
|
||||
"github.com/0xJacky/Nginx-UI/internal/logger"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gin-gonic/gin/binding"
|
||||
val "github.com/go-playground/validator/v10"
|
||||
"net/http"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func init() {
|
||||
if v, ok := binding.Validator.Engine().(*val.Validate); ok {
|
||||
err := v.RegisterValidation("alphanumdash", func(fl val.FieldLevel) bool {
|
||||
return regexp.MustCompile(`^[a-zA-Z0-9-]+$`).MatchString(fl.Field().String())
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
logger.Fatal("binding validator engine is not initialized")
|
||||
}
|
||||
|
||||
func ErrHandler(c *gin.Context, err error) {
|
||||
logger.GetLogger().Errorln(err)
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
|
@ -22,7 +39,6 @@ type ValidError struct {
|
|||
}
|
||||
|
||||
func BindAndValid(c *gin.Context, target interface{}) bool {
|
||||
errs := make(map[string]string)
|
||||
err := c.ShouldBindJSON(target)
|
||||
if err != nil {
|
||||
logger.Error("bind err", err)
|
||||
|
@ -31,24 +47,23 @@ func BindAndValid(c *gin.Context, target interface{}) bool {
|
|||
ok := errors.As(err, &verrs)
|
||||
|
||||
if !ok {
|
||||
logger.Error("valid err", verrs)
|
||||
c.JSON(http.StatusNotAcceptable, gin.H{
|
||||
"message": "Requested with wrong parameters",
|
||||
"code": http.StatusNotAcceptable,
|
||||
"error": verrs,
|
||||
})
|
||||
return false
|
||||
}
|
||||
|
||||
t := reflect.TypeOf(target).Elem()
|
||||
errorsMap := make(map[string]interface{})
|
||||
for _, value := range verrs {
|
||||
t := reflect.ValueOf(target)
|
||||
realType := t.Type().Elem()
|
||||
field, _ := realType.FieldByName(value.StructField())
|
||||
errs[field.Tag.Get("json")] = value.Tag()
|
||||
var path []string
|
||||
getJsonPath(t, value.StructNamespace(), &path)
|
||||
insertError(errorsMap, path, value.Tag())
|
||||
}
|
||||
|
||||
c.JSON(http.StatusNotAcceptable, gin.H{
|
||||
"errors": errs,
|
||||
"errors": errorsMap,
|
||||
"message": "Requested with wrong parameters",
|
||||
"code": http.StatusNotAcceptable,
|
||||
})
|
||||
|
@ -58,3 +73,45 @@ func BindAndValid(c *gin.Context, target interface{}) bool {
|
|||
|
||||
return true
|
||||
}
|
||||
|
||||
// findField recursively finds the field in a nested struct
|
||||
func getJsonPath(t reflect.Type, namespace string, path *[]string) {
|
||||
fields := strings.Split(namespace, ".")
|
||||
if len(fields) == 0 {
|
||||
return
|
||||
}
|
||||
f, ok := t.FieldByName(fields[0])
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
*path = append(*path, f.Tag.Get("json"))
|
||||
|
||||
if len(fields) > 1 {
|
||||
subFields := strings.Join(fields[1:], ".")
|
||||
getJsonPath(f.Type, subFields, path)
|
||||
}
|
||||
}
|
||||
|
||||
// insertError inserts an error into the errors map
|
||||
func insertError(errorsMap map[string]interface{}, path []string, errorTag string) {
|
||||
if len(path) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
jsonTag := path[0]
|
||||
if len(path) == 1 {
|
||||
// Last element in the path, set the error
|
||||
errorsMap[jsonTag] = errorTag
|
||||
return
|
||||
}
|
||||
|
||||
// Create a new map if necessary
|
||||
if _, ok := errorsMap[jsonTag]; !ok {
|
||||
errorsMap[jsonTag] = make(map[string]interface{})
|
||||
}
|
||||
|
||||
// Recursively insert into the nested map
|
||||
subMap, _ := errorsMap[jsonTag].(map[string]interface{})
|
||||
insertError(subMap, path[1:], errorTag)
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue