base-golang/pkg/captcha/base64.go

98 lines
2.2 KiB
Go
Raw Permalink Normal View History

2024-07-23 10:23:43 +08:00
package captcha
import (
"fmt"
2024-07-31 16:49:14 +08:00
"gitea.bvbej.com/bvbej/base-golang/pkg/cache"
2024-07-23 10:23:43 +08:00
"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)
}