added ValidBehaviorCaptcha

Author: feiin

README.md

@@ -1,2 +1,8 @@
 # go-captcha
 golang 验证码
+
+
+## Preview
+[](./previews/test.png)
+[](./previews/test_1.png)
+[](./previews/test_2.png)

alphabet.go

@@ -0,0 +1,79 @@
+package captcha
+
+var Alphabets = "abcdefghijklmnopqrstuvwxyxABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+
+var CNChars = []string{
+	"热情",
+	"帮助",
+	"聪明",
+	"活泼",
+	"笑容",
+	"风景",
+	"眼睛",
+	"太阳",
+	"棉花",
+	"落叶",
+	"翅膀",
+	"松果",
+	"表情",
+	"船只",
+	"灯笼",
+	"天气",
+	"时光",
+	"鲜花",
+	"火车",
+	"街道",
+	"问题",
+	"奇怪",
+	"欢乐",
+	"端正",
+	"大方",
+	"清楚",
+	"欢喜",
+	"完整",
+	"高兴",
+	"漂亮",
+	"干净",
+	"仔细",
+	"完全",
+	"前后",
+	"马虎",
+	"冷清",
+	"普通",
+	"高贵",
+	"优雅",
+	"热闹",
+	"云彩",
+	"同学",
+	"黑板",
+	"篮球",
+	"足球",
+	"城市",
+	"组合",
+	"玩具",
+	"表格",
+	"希望",
+	"和平",
+	"同意",
+	"目标",
+	"工作",
+	"学习",
+	"顺利",
+	"谈话",
+	"风格",
+	"如果",
+	"天气",
+	"严格",
+	"教育",
+	"讨论",
+	"明亮",
+	"相信",
+	"犹豫",
+	"坚定",
+	"缓慢",
+	"勇敢",
+	"提前",
+	"杰出",
+	"安全",
+	"宽容",
+}

canvas.go

@@ -8,10 +8,9 @@ import (
 	"image/color"
 	"image/draw"
 	"math"
-	"unicode/utf8"
-
 	"math/rand"
 	"time"
+	"unicode/utf8"
 )
 
 //go:embed fonts/*
@@ -178,7 +177,8 @@ func (c *Canvas) DrawString(text string) []DrawRect {
 	var drawRects []DrawRect
 	textPos := c.getFontPostions(text)
 
-	for index, ch := range text {
+	var index = 0
+	for _, ch := range text {
 
 		pos := textPos[index]
 
@@ -217,6 +217,8 @@ func (c *Canvas) DrawString(text string) []DrawRect {
 
 		drawRects = append(drawRects, dp)
 
+		index = index + 1
+
 	}
 
 	return drawRects
@@ -415,4 +417,4 @@ func (c *Canvas) DrawCircle(xc, yc, r int, fill bool, cc color.Color) {
 		}
 		x++
 	}
-}
+}

captcha.go

@@ -8,6 +8,7 @@ import (
 	"image"
 	"image/color"
 	"image/png"
+	"math/rand"
 	"os"
 )
 
@@ -94,6 +95,7 @@ func (cp *Captcha) SetBackgroundColor(color color.Color) {
 	cp.Config.BackgroupColor = color
 }
 
+//GenCaptchaImage 生成文字的验证码图片
 func (cp *Captcha) GenCaptchaImage(text string) (CaptchaResult, error) {
 
 	result := CaptchaResult{}
@@ -126,11 +128,65 @@ func (cp *Captcha) GenCaptchaImage(text string) (CaptchaResult, error) {
 	result.ImageBase64 = fmt.Sprintf("data:%s;base64,%s", "image/png", base64.StdEncoding.EncodeToString(imageBytes))
 	result.Text = text
 
-	// writeImageFile("./test.png", imageBytes)
+	// writeImageFile("./previews/test.png", imageBytes)
 	return result, nil
 
 }
 
+//GenNormalRandomCaptcha 随机验证码 - 普通验证码
+func (cp *Captcha) GenRandomNormalCaptcha(length int) (CaptchaResult, error) {
+	var buff bytes.Buffer
+
+	for i := 0; i < length; i++ {
+		ix := rand.Intn(length)
+		buff.WriteByte(Alphabets[ix])
+	}
+
+	return cp.GenCaptchaImage(buff.String())
+
+}
+
+//GenBehaviorCaptcha 生成中文点击验证码 - 点击行为验证码
+func (cp *Captcha) GenBehaviorCaptcha() (CaptchaResult, error) {
+
+	ix := rand.Intn(len(CNChars))
+	return cp.GenCaptchaImage(CNChars[ix])
+
+}
+
+//ValidBehaviorCaptcha 验证行为点击验证码是否正确
+func ValidBehaviorCaptcha(cr CaptchaResult, userPoints []Point) bool {
+	//字符数量不相等
+	return ValidBehaviorRects(cr.DrawRects, userPoints)
+}
+
+//ValidBehaviorRects 验证行为点是否在验证码矩形中
+func ValidBehaviorRects(rects []DrawRect, userPoints []Point) bool {
+
+	if len(userPoints) != len(rects) {
+		return false
+	}
+
+	for i := 0; i < len(userPoints); i++ {
+		realRect := rects[i]
+		userPos := userPoints[i]
+
+		if !PointInRect(userPos, realRect) {
+			return false
+		}
+	}
+	return true
+
+}
+
+//PointInRect 验证点是否在矩形中
+func PointInRect(p Point, rect DrawRect) bool {
+	if p.X >= rect.X && p.X <= rect.X+rect.Width && p.Y >= rect.Y && p.Y <= rect.Y+rect.Height {
+		return true
+	}
+	return false
+}
+
 func encodingWithPng(img image.Image) (encodeResult []byte, err error) {
 
 	var buf bytes.Buffer
@@ -146,6 +202,7 @@ func encodingWithPng(img image.Image) (encodeResult []byte, err error) {
 
 }
 
+//writeImageFile tmp
 func writeImageFile(filepath string, imageBytes []byte) {
 	file, err := os.OpenFile(filepath, os.O_WRONLY|os.O_CREATE, 0666)
 	if err != nil {

captcha_test.go

@@ -8,8 +8,108 @@ import (
 func TestGenCaptcha(t *testing.T) {
 
 	cpt := NewCaptcha(260, 100)
-	cpt.SetBackgroundColor(color.RGBA{R: uint8(20), G: uint8(8), B: uint8(100), A: uint8(255)})
+	// cpt.SetBackgroundColor(color.RGBA{R: uint8(20), G: uint8(8), B: uint8(100), A: uint8(255)})
 	// cpt.Config.Style = CaptchaStyle_Behavior
 	result, err := cpt.GenCaptchaImage("3567")
 	t.Logf("GenCaptchaImage: %+v %+v", result, err)
 }
+
+func TestGenNormalCaptcha(t *testing.T) {
+
+	cpt := NewCaptcha(260, 100)
+	cpt.SetBackgroundColor(color.RGBA{R: uint8(20), G: uint8(8), B: uint8(100), A: uint8(255)})
+	// cpt.Config.Style = CaptchaStyle_Behavior
+	result, err := cpt.GenRandomNormalCaptcha(4)
+	t.Logf("GenCaptchaImage: %+v %+v", result, err)
+}
+
+func TestGenGenBehaviorCaptcha(t *testing.T) {
+
+	cpt := NewCaptcha(260, 100)
+	cpt.SetBackgroundColor(color.RGBA{R: uint8(20), G: uint8(8), B: uint8(100), A: uint8(255)})
+	// cpt.Config.Style = CaptchaStyle_Behavior
+	cpt.Config.MaxRotate = 20
+	result, err := cpt.GenBehaviorCaptcha()
+	t.Logf("GenCaptchaImage: %+v %+v", result, err)
+}
+
+func TestPointInRect(t *testing.T) {
+	rect := DrawRect{X: 10, Width: 50, Y: 50, Height: 60}
+	point := Point{X: 10, Y: 40}
+
+	if PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+
+	point = Point{X: 10, Y: 50}
+	if !PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+
+	point = Point{X: 10, Y: 60}
+	if !PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+
+	point = Point{X: 10, Y: 110}
+	if !PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+
+	point = Point{X: 30, Y: 110}
+	if !PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+
+	point = Point{X: 60, Y: 110}
+	if !PointInRect(point, rect) {
+		t.Errorf("PointInRect error")
+	}
+}
+
+func TestValidBehaviorCaptcha(t *testing.T) {
+	cr := CaptchaResult{}
+	cr.DrawRects = []DrawRect{
+		DrawRect{X: 10, Y: 10, Width: 60, Height: 60},
+		DrawRect{X: 20, Y: 60, Width: 60, Height: 60},
+		DrawRect{X: 60, Y: 90, Width: 60, Height: 60},
+	}
+
+	ps := []Point{
+		Point{X: 15, Y: 8},
+	}
+
+	if ValidBehaviorCaptcha(cr, ps) {
+		t.Errorf("invalid error")
+	}
+	if !ValidBehaviorCaptcha(cr, ps) {
+		t.Logf("invalid captcha")
+	}
+
+	ps = []Point{
+		Point{X: 15, Y: 8},
+		Point{X: 15, Y: 8},
+		Point{X: 15, Y: 8},
+	}
+	if ValidBehaviorCaptcha(cr, ps) {
+		t.Errorf("invalid error")
+	}
+
+	ps = []Point{
+		Point{X: 15, Y: 12},
+		Point{X: 20, Y: 62},
+		Point{X: 60, Y: 8},
+	}
+	if ValidBehaviorCaptcha(cr, ps) {
+		t.Errorf("invalid error")
+	}
+
+	ps = []Point{
+		Point{X: 15, Y: 12},
+		Point{X: 20, Y: 62},
+		Point{X: 60, Y: 98},
+	}
+	if ValidBehaviorCaptcha(cr, ps) {
+		t.Logf("valid captcha")
+	}
+}