|
|
|
@ -19,6 +19,7 @@ package captcha
|
|
|
|
import (
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"fmt"
|
|
|
|
"html/template"
|
|
|
|
"html/template"
|
|
|
|
|
|
|
|
"image/color"
|
|
|
|
"path"
|
|
|
|
"path"
|
|
|
|
"strings"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
|
|
|
|
@ -49,6 +50,7 @@ type Captcha struct {
|
|
|
|
ChallengeNums int
|
|
|
|
ChallengeNums int
|
|
|
|
Expiration int64
|
|
|
|
Expiration int64
|
|
|
|
CachePrefix string
|
|
|
|
CachePrefix string
|
|
|
|
|
|
|
|
ColorPalette color.Palette
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// generate key string
|
|
|
|
// generate key string
|
|
|
|
@ -61,16 +63,21 @@ func (c *Captcha) genRandChars() string {
|
|
|
|
return string(com.RandomCreateBytes(c.ChallengeNums, defaultChars...))
|
|
|
|
return string(com.RandomCreateBytes(c.ChallengeNums, defaultChars...))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// tempalte func for output html
|
|
|
|
// CreateHTML outputs HTML for display and fetch new captcha images.
|
|
|
|
func (c *Captcha) CreateHtml() template.HTML {
|
|
|
|
func (c *Captcha) CreateHTML() template.HTML {
|
|
|
|
value, err := c.CreateCaptcha()
|
|
|
|
value, err := c.CreateCaptcha()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
panic(fmt.Errorf("fail to create captcha: %v", err))
|
|
|
|
panic(fmt.Errorf("fail to create captcha: %v", err))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return template.HTML(fmt.Sprintf(`<input type="hidden" name="%s" value="%s">
|
|
|
|
return template.HTML(fmt.Sprintf(`<input type="hidden" name="%[1]s" value="%[2]s">
|
|
|
|
<a class="captcha" href="javascript:">
|
|
|
|
<a class="captcha" href="javascript:" tabindex="-1">
|
|
|
|
<img onclick="this.src=('%s%s%s.png?reload='+(new Date()).getTime())" class="captcha-img" src="%s%s%s.png">
|
|
|
|
<img onclick="this.src=('%[3]s%[4]s%[2]s.png?reload='+(new Date()).getTime())" class="captcha-img" src="%[3]s%[4]s%[2]s.png">
|
|
|
|
</a>`, c.FieldIdName, value, c.SubURL, c.URLPrefix, value, c.SubURL, c.URLPrefix, value))
|
|
|
|
</a>`, c.FieldIdName, value, c.SubURL, c.URLPrefix))
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// DEPRECATED
|
|
|
|
|
|
|
|
func (c *Captcha) CreateHtml() template.HTML {
|
|
|
|
|
|
|
|
return c.CreateHTML()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// create a new captcha id
|
|
|
|
// create a new captcha id
|
|
|
|
@ -139,6 +146,9 @@ type Options struct {
|
|
|
|
Expiration int64
|
|
|
|
Expiration int64
|
|
|
|
// Cache key prefix captcha characters. Default is "captcha_".
|
|
|
|
// Cache key prefix captcha characters. Default is "captcha_".
|
|
|
|
CachePrefix string
|
|
|
|
CachePrefix string
|
|
|
|
|
|
|
|
// ColorPalette holds a collection of primary colors used for
|
|
|
|
|
|
|
|
// the captcha's text. If not defined, a random color will be generated.
|
|
|
|
|
|
|
|
ColorPalette color.Palette
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func prepareOptions(options []Options) Options {
|
|
|
|
func prepareOptions(options []Options) Options {
|
|
|
|
@ -192,6 +202,7 @@ func NewCaptcha(opt Options) *Captcha {
|
|
|
|
ChallengeNums: opt.ChallengeNums,
|
|
|
|
ChallengeNums: opt.ChallengeNums,
|
|
|
|
Expiration: opt.Expiration,
|
|
|
|
Expiration: opt.Expiration,
|
|
|
|
CachePrefix: opt.CachePrefix,
|
|
|
|
CachePrefix: opt.CachePrefix,
|
|
|
|
|
|
|
|
ColorPalette: opt.ColorPalette,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@ -229,9 +240,10 @@ func Captchaer(options ...Options) macaron.Handler {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if _, err := NewImage([]byte(chars), cpt.StdWidth, cpt.StdHeight).WriteTo(ctx.Resp); err != nil {
|
|
|
|
if _, err := NewImage([]byte(chars), cpt.StdWidth, cpt.StdHeight, cpt.ColorPalette).WriteTo(ctx.Resp); err != nil {
|
|
|
|
panic(fmt.Errorf("write captcha: %v", err))
|
|
|
|
panic(fmt.Errorf("write captcha: %v", err))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx.Status(200)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
Lunny Xiao
5 years ago
committed by
GitHub