package captcha import ( "fmt" "git.bvbej.com/bvbej/base-golang/pkg/cache" "github.com/mojocn/base64Captcha" "go.uber.org/zap" "strings" "time" ) var _ base64Captcha.Store = (*store)(nil) type store struct { cache cache.Repo ttl time.Duration logger *zap.Logger ns string prefix string } func (s *store) Set(id string, value string) error { err := s.cache.Set(fmt.Sprintf("%s%s%s", s.ns, s.prefix, id), value, s.ttl) if err != nil { return err } return nil } func (s *store) Get(id string, clear bool) string { value, err := s.cache.Get(fmt.Sprintf("%s%s%s", s.ns, s.prefix, id)) if err == nil && clear { s.cache.Del(fmt.Sprintf("%s%s%s", s.ns, s.prefix, id)) } return value } func (s *store) Verify(id, answer string, clear bool) bool { value := s.Get(id, clear) if value == "" || answer == "" { return false } return strings.ToLower(value) == strings.ToLower(answer) } func NewStore(cache cache.Repo, ttl time.Duration, namespace string) base64Captcha.Store { return &store{ cache: cache, ttl: ttl, ns: namespace, prefix: "captcha:base64:", } } var _ Captcha = (*captcha)(nil) type Captcha interface { Generate() (id, b64s, answer string, err error) Verify(id, value string) bool } type captcha struct { driver base64Captcha.Driver store base64Captcha.Store } func NewStringCaptcha(store base64Captcha.Store, height, width, length int) Captcha { conf := &base64Captcha.DriverString{ Height: height, Width: width, NoiseCount: length, ShowLineOptions: base64Captcha.OptionShowHollowLine, Length: length, Source: "ABCDEFGHIJKMNPQRSTUVWXYZabcdefghijkmnpqrstuvwxyz0123456789", } return &captcha{ driver: conf.ConvertFonts(), store: store, } } func NewDigitCaptcha(store base64Captcha.Store, height, width, length int) Captcha { conf := base64Captcha.NewDriverDigit(height, width, length, 0.7, height) return &captcha{ driver: conf, store: store, } } func (c *captcha) Generate() (id, b64s, answer string, err error) { newCaptcha := base64Captcha.NewCaptcha(c.driver, c.store) return newCaptcha.Generate() } func (c *captcha) Verify(id, value string) bool { return c.store.Verify(id, value, true) }