You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
2.0 KiB
59 lines
2.0 KiB
2 years ago
|
// Copyright 2017 Eric Zhou. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
// Package base64Captcha supports digits, numbers,alphabet, arithmetic, audio and digit-alphabet captcha.
|
||
|
|
// base64Captcha is used for fast development of RESTful APIs, web apps and backend services in Go. give a string identifier to the package and it returns with a base64-encoding-png-string
|
||
|
|
package base64Captcha
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Captcha captcha basic information.
|
||
|
|
type Captcha struct {
|
||
|
|
Driver Driver
|
||
|
|
Store Store
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewCaptcha creates a captcha instance from driver and store
|
||
|
|
func NewCaptcha(driver Driver, store Store) *Captcha {
|
||
|
|
return &Captcha{Driver: driver, Store: store}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Generate generates a random id, base64 image string or an error if any
|
||
|
|
func (c *Captcha) Generate() (id, b64s string, err error) {
|
||
|
|
id, content, answer := c.Driver.GenerateIdQuestionAnswer()
|
||
|
|
item, err := c.Driver.DrawCaptcha(content)
|
||
|
|
if err != nil {
|
||
|
|
return "", "", err
|
||
|
|
}
|
||
|
|
err = c.Store.Set(id, answer)
|
||
|
|
if err != nil {
|
||
|
|
return "", "", err
|
||
|
|
}
|
||
|
|
b64s = item.EncodeB64string()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Verify by a given id key and remove the captcha value in store,
|
||
|
|
// return boolean value.
|
||
|
|
// if you has multiple captcha instances which share a same store.
|
||
|
|
// You may want to call `store.Verify` method instead.
|
||
|
|
func (c *Captcha) Verify(id, answer string, clear bool) (match bool) {
|
||
|
|
vv := c.Store.Get(id, clear)
|
||
|
|
//fix issue for some redis key-value string value
|
||
|
|
vv = strings.TrimSpace(vv)
|
||
|
|
return vv == strings.TrimSpace(answer)
|
||
|
|
}
|