1Panel/core/init/validator/validator.go

package validator

import (
	"regexp"
	"unicode"

	"github.com/1Panel-dev/1Panel/core/global"
	"github.com/1Panel-dev/1Panel/core/utils/re"
	"github.com/go-playground/validator/v10"
)

func Init() {
	validator := validator.New()
	if err := validator.RegisterValidation("name", checkNamePattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("ip", checkIpPattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("password", checkPasswordPattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("base_setting_key", checkBaseSettingKey); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("vm_name", checkVMNamePattern); err != nil {
		panic(err)
	}
	if err := validator.RegisterValidation("vm_common", checkVMCommonPattern); err != nil {
		panic(err)
	}
	global.VALID = validator
}

var baseSettingKeys = map[string]struct{}{
	"PanelName":                  {},
	"Theme":                      {},
	"MenuTabs":                   {},
	"MenuAccordion":              {},
	"Language":                   {},
	"SessionTimeout":             {},
	"ExpirationDays":             {},
	"DeveloperMode":              {},
	"UpgradeBackupCopies":        {},
	"SecurityEntrance":           {},
	"BindDomain":                 {},
	"AllowIPs":                   {},
	"AllowIPTrustedProxies":      {},
	"PasskeyTrustedProxies":      {},
	"ComplexityVerification":     {},
	"NoAuthSetting":              {},
	"DashboardMemoVisible":       {},
	"DashboardSimpleNodeVisible": {},
	"Edition":                    {},
	"DocSource":                  {},
	"AppStoreLastModified":       {},
	"ScriptSync":                 {},
	"HideMenu":                   {},
}

func checkNamePattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	result, err := regexp.MatchString("^[a-zA-Z\u4e00-\u9fa5]{1}[a-zA-Z0-9_\u4e00-\u9fa5]{0,30}$", value)
	if err != nil {
		global.LOG.Errorf("regexp matchString failed, %v", err)
	}
	return result
}

func checkIpPattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	result, err := regexp.MatchString(`^((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})(\.((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})){3}$`, value)
	if err != nil {
		global.LOG.Errorf("regexp check ip matchString failed, %v", err)
	}
	return result
}

func checkPasswordPattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	if len(value) < 8 || len(value) > 30 {
		return false
	}

	hasNum := false
	hasLetter := false
	for _, r := range value {
		if unicode.IsLetter(r) && !hasLetter {
			hasLetter = true
		}
		if unicode.IsNumber(r) && !hasNum {
			hasNum = true
		}
		if hasLetter && hasNum {
			return true
		}
	}

	return false
}

func checkBaseSettingKey(fl validator.FieldLevel) bool {
	_, ok := baseSettingKeys[fl.Field().String()]
	return ok
}

func checkVMNamePattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	return re.GetRegex(re.VMNameValidationPattern).MatchString(value)
}
func checkVMCommonPattern(fl validator.FieldLevel) bool {
	value := fl.Field().String()
	return re.GetRegex(re.VMCommonPattern).MatchString(value)
}