mirror of
https://github.com/moul/sshportal.git
synced 2025-01-01 13:12:45 +08:00
33 lines
785 B
Go
33 lines
785 B
Go
package dbmodels
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
"github.com/asaskevich/govalidator"
|
|
)
|
|
|
|
func InitValidator() {
|
|
unixUserRegexp := regexp.MustCompile("[a-z_][a-z0-9_-]*")
|
|
|
|
govalidator.CustomTypeTagMap.Set("unix_user", govalidator.CustomTypeValidator(func(i interface{}, context interface{}) bool {
|
|
name, ok := i.(string)
|
|
if !ok {
|
|
return false
|
|
}
|
|
return unixUserRegexp.MatchString(name)
|
|
}))
|
|
govalidator.CustomTypeTagMap.Set("host_logging_mode", govalidator.CustomTypeValidator(func(i interface{}, context interface{}) bool {
|
|
name, ok := i.(string)
|
|
if !ok {
|
|
return false
|
|
}
|
|
if name == "" {
|
|
return true
|
|
}
|
|
return IsValidHostLoggingMode(name)
|
|
}))
|
|
}
|
|
|
|
func IsValidHostLoggingMode(name string) bool {
|
|
return name == "disabled" || name == "input" || name == "everything"
|
|
}
|