spec_string.go

package vd

import (
	"github.com/hoisie/mustache"
	"log"
	"regexp"
	"runtime/debug"
)

type StringSpec struct {
	Name string
	Path string
	AllowEmpty bool
	RuneLen uint64
	RuneLenMessage string
	MinRuneLen uint64
	MinRuneLenMessage string
	MaxRuneLen uint64
	MaxRuneLenMessage string
	Pattern []string
	BanPattern []string
	PatternMessage string
	Enum []string
	Ext []StringSpec
}
func (s StringSpec) NameIs(name string) StringSpec {
	s.Name = name
	return s
}
type stringSpecRender struct {
	Value interface{}
	StringSpec
}
func (spec StringSpec) render (message string, value interface{}) string {
	context := stringSpecRender{
		Value: value,
		StringSpec: spec,
	}
	return mustache.Render(message, context)
}
func (r *Rule) String(v string, spec StringSpec) {
	if r.Fail { return }
	if v == "" {
		if spec.AllowEmpty {
			return
		} else {
			r.Break(r.Format.StringNotAllowEmpty(spec.Name, spec.Path), spec.Path)
			return
		}
	}
	if spec.CheckRuneLen(v, r) { return }
	if spec.CheckMinRuneLen(v, r) { return }
	if spec.CheckMaxRuneLen(v, r) { return }
	if spec.CheckPattern   (v, r) { return }
	if spec.CheckBanPattern(v, r) { return }
	if spec.CheckEnum(v, r) {return}
	for _, ext := range spec.Ext {
		ext.Name = spec.Name
		ext.AllowEmpty = spec.AllowEmpty
		if ext.PatternMessage == "" {
			ext.PatternMessage = spec.PatternMessage
		}
		r.String(v, ext)
		if r.Fail {return}
	}
}
func (spec StringSpec) CheckRuneLen(v string, r *Rule) (fail bool) {
	if spec.RuneLen == 0 {
		return false
	}
	length := len([]rune(v))
	pass := uint64(length) == spec.RuneLen
	if !pass {
		message := r.CreateMessage(spec.RuneLenMessage, func() string {
			return r.Format.StringRuneLen(spec.Name, spec.Path, v, spec.RuneLen)
		})
		r.Break(spec.render(message, v), spec.Path)
	}
	return r.Fail
}
func (spec StringSpec) CheckMaxRuneLen(v string, r *Rule) (fail bool) {
	if spec.MaxRuneLen == 0 {
		return false
	}
	length := len([]rune(v))
	pass := uint64(length) <= spec.MaxRuneLen
	if !pass {
		message := r.CreateMessage(spec.MaxRuneLenMessage, func() string {
			return r.Format.StringMaxRuneLen(spec.Name, spec.Path, v, spec.MaxRuneLen)
		})
		r.Break(spec.render(message, v), spec.Path)
	}
	return r.Fail
}

func (spec StringSpec) CheckMinRuneLen(v string, r *Rule) (fail bool) {
	length := len([]rune(v))
	pass := uint64(length) >= spec.MinRuneLen
	if !pass {
		message := r.CreateMessage(spec.MinRuneLenMessage, func() string {
			return r.Format.StringMinRuneLen(spec.Name, spec.Path, v, spec.MinRuneLen)
		})
		r.Break(spec.render(message, v), spec.Path)
	}
	return r.Fail
}
type patternData struct {
	Pattern []string
	PatternMessage string
	Name string
	Path string
}
func checkPattern(data patternData, render func(string, interface{}) string, v string, r *Rule, path string) (fail bool) {
	if len(data.Pattern) == 0 {
		return false
	}
	for _, pattern := range data.Pattern {
		matched, err := regexp.MatchString(pattern, v) ; if err != nil {
			// 此处如果将错误向上传递则会导致接口易用性下降
			// panic 会导致不必要的中断服务
			// 所以这里采取打印错误并验证错误的处理 @nimoc 2021年02月12日01:20:45 (新年快乐)
			log.Print(err)
			debug.PrintStack()
			r.Break("goclub/validator: regexp pattern error(" + pattern + ")", path)
			return
		}
		pass := matched
		if !pass {
			message := r.CreateMessage(data.PatternMessage, func() string {
				return r.Format.Pattern(data.Name, data.Path,  v, data.Pattern, pattern)
			})
			r.Break(render(message, v), path)
			break
		}
	}
	return r.Fail
}

func (spec StringSpec) CheckPattern(v string, r *Rule) (fail bool) {
	return checkPattern(patternData{
		Pattern:        spec.Pattern,
		PatternMessage: spec.PatternMessage,
		Name:           spec.Name,
		Path: spec.Path,
	}, spec.render, v, r, spec.Path)
}
type banPatternData struct {
	BanPattern []string
	PatternMessage string
	Name string
	Path string
}
func checkBanPattern(data banPatternData, render func(string, interface{}) string, v string, r *Rule, path string) (fail bool) {
	if len(data.BanPattern) == 0 {
		return false
	}
	for _, pattern := range data.BanPattern {
		matched, err := regexp.MatchString(pattern, v) ; if err != nil {
			// 此处如果将错误向上传递则会导致接口易用性下降
			// panic 会导致不必要的中断服务
			// 所以这里采取打印错误并验证错误的处理 @nimoc 2021年02月12日01:20:45 (新年快乐)
			log.Print(err)
			debug.PrintStack()
			r.Break("goclub/validator: regexp ban pattern error(" + pattern + ")", path)
		}
		pass := !matched
		if !pass {
			message := r.CreateMessage(data.PatternMessage, func() string {
				return r.Format.BanPattern(data.Name, data.Path,  v, data.BanPattern, pattern)
			})
			r.Break(render(message, v), path)
			break
		}
	}
	return
}
func (spec StringSpec) CheckBanPattern(v string, r *Rule) (fail bool) {
	return checkBanPattern(banPatternData{
		BanPattern:     spec.BanPattern,
		PatternMessage: spec.PatternMessage,
		Name:           spec.Name,
		Path:           spec.Path,
	}, spec.render, v, r, spec.Path)
}
func (spec StringSpec) CheckEnum(v string, r *Rule) (fail bool) {
	if len(spec.Enum) == 0 {
		return false
	}
	pass := false
	for _, enum := range spec.Enum {
		if enum == v {
			pass = true
		}
	}
	if !pass {
		message := r.Format.StringEnum(spec.Name, spec.Path, v, spec.Enum)
		r.Break(spec.render(message, v), spec.Path)
	}
	return r.Fail
}

func uniqueStrings(strings []string) (isRepeat bool, repeatElement string) {
	hash := map[string]bool{}
	for _, s := range strings {
		_,ok := hash[s]; if ok {
			return true, s
		} else {
			hash[s] = true
		}
	}
	return
}