golang 生成圖片驗證碼

zhuCheer發表於2020-05-21
Golang

核心繪圖方案利用 "github.com/fogleman/gg" 這個包來處理

1.隨機獲取驗證碼文字

func GetRandStr(n int) (randStr string) {
    chars := "ABCDEFGHIJKMNPQRSTUVWXYZabcdefghijkmnpqrstuvwxyz23456789"
    charsLen := len(chars)
    if n > 10 {
        n = 10
    }

    rand.Seed(time.Now().UnixNano())
    for i := 0; i < n; i++ {
        randIndex := rand.Intn(charsLen)
        randStr += chars[randIndex : randIndex+1]
    }
    return randStr
}

2.繪製圖片將文字寫入圖片

func ImgText(width, height int, text string) (b []byte) {
    textLen := len(text)
    dc := gg.NewContext(width, height)
    bgR, bgG, bgB, bgA := getRandColorRange(240, 255)
    dc.SetRGBA255(bgR, bgG, bgB, bgA)
    dc.Clear()

    // 干擾線
    for i := 0; i < 10; i++ {
        x1, y1 := getRandPos(width, height)
        x2, y2 := getRandPos(width, height)
        r, g, b, a := getRandColor(255)
        w := float64(rand.Intn(3) + 1)
        dc.SetRGBA255(r, g, b, a)
        dc.SetLineWidth(w)
        dc.DrawLine(x1, y1, x2, y2)
        dc.Stroke()
    }

    fontSize := float64(height/2) + 5
    face := loadFontFace(fontSize)
    dc.SetFontFace(face)

    for i := 0; i < len(text); i++ {
        r, g, b, _ := getRandColor(100)
        dc.SetRGBA255(r, g, b, 255)
        fontPosX := float64(width/textLen*i) + fontSize*0.6

        writeText(dc, text[i:i+1], float64(fontPosX), float64(height/2))
    }

    buffer := bytes.NewBuffer(nil)
    dc.EncodePNG(buffer)
    b = buffer.Bytes()
    return
}

// 渲染文字
func writeText(dc *gg.Context, text string, x, y float64) {
    xfload := 5 - rand.Float64()*10 + x
    yfload := 5 - rand.Float64()*10 + y

    radians := 40 - rand.Float64()*80
    dc.RotateAbout(gg.Radians(radians), x, y)
    dc.DrawStringAnchored(text, xfload, yfload, 0.2, 0.5)
    dc.RotateAbout(-1*gg.Radians(radians), x, y)
    dc.Stroke()
}

// 隨機座標
func getRandPos(width, height int) (x float64, y float64) {
    x = rand.Float64() * float64(width)
    y = rand.Float64() * float64(height)
    return x, y
}

// 隨機顏色
func getRandColor(maxColor int) (r, g, b, a int) {
    r = int(uint8(rand.Intn(maxColor)))
    g = int(uint8(rand.Intn(maxColor)))
    b = int(uint8(rand.Intn(maxColor)))
    a = int(uint8(rand.Intn(255)))
    return r, g, b, a
}

// 隨機顏色範圍
func getRandColorRange(miniColor, maxColor int) (r, g, b, a int) {
    if miniColor > maxColor {
        miniColor = 0
        maxColor = 255
    }
    r = int(uint8(rand.Intn(maxColor-miniColor) + miniColor))
    g = int(uint8(rand.Intn(maxColor-miniColor) + miniColor))
    b = int(uint8(rand.Intn(maxColor-miniColor) + miniColor))
    a = int(uint8(rand.Intn(maxColor-miniColor) + miniColor))
    return r, g, b, a
}

// 載入字型
func loadFontFace(points float64) font.Face {
     // 這裡是將字型TTF檔案轉換成了 byte 資料儲存成了一個 go 檔案 檔案較大可以到附錄下
    // 通過truetype.Parse可以將 byte 型別的資料轉換成TTF字型型別
    f, err := truetype.Parse(COMICSAN)載

    if err != nil {
        panic(err)
    }
    face := truetype.NewFace(f, &truetype.Options{
        Size: points,
    })
    return face
}

3.圖片生成好了將驗證碼文字資訊暫存到session或redis中等等，然後通過將圖片輸出到頁面，可以是base64也可以是配置content-type輸出圖片，可以按自己的業務來;
下面是驗證碼圖片效果；

avatar

給大家接受一個 golang web 框架，能快速生成圖片驗證碼完整方案；享受實用主義帶來的快樂吧!
快速開始：https://www.kancloud.cn/chase688/orange_framework/content/快速開始.md
圖片驗證碼：https://www.kancloud.cn/chase688/orange_framework/content/tools/圖片驗證碼.md

附件

字型 go 檔案 gitee.com/zhucheer/orange/blob/mas...

本作品採用《CC 協議》，轉載必須註明作者和本文連結

相關文章