forked from golang/hotime
107 lines
4.1 KiB
Go
107 lines
4.1 KiB
Go
|
package qrcode
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/silenceper/wechat/v2/miniprogram/context"
|
|||
|
|
"github.com/silenceper/wechat/v2/util"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
createWXAQRCodeURL = "https://api.weixin.qq.com/cgi-bin/wxaapp/createwxaqrcode?access_token=%s"
|
|||
|
|
getWXACodeURL = "https://api.weixin.qq.com/wxa/getwxacode?access_token=%s"
|
|||
|
|
getWXACodeUnlimitURL = "https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=%s"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// QRCode struct
|
|||
|
|
type QRCode struct {
|
|||
|
|
*context.Context
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewQRCode 实例
|
|||
|
|
func NewQRCode(context *context.Context) *QRCode {
|
|||
|
|
qrCode := new(QRCode)
|
|||
|
|
qrCode.Context = context
|
|||
|
|
return qrCode
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Color QRCode color
|
|||
|
|
type Color struct {
|
|||
|
|
R string `json:"r"`
|
|||
|
|
G string `json:"g"`
|
|||
|
|
B string `json:"b"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// QRCoder 小程序码参数
|
|||
|
|
type QRCoder struct {
|
|||
|
|
// page 必须是已经发布的小程序存在的页面,根路径前不要填加 /,不能携带参数（参数请放在scene字段里），如果不填写这个字段，默认跳主页面
|
|||
|
|
Page string `json:"page,omitempty"`
|
|||
|
|
// path 扫码进入的小程序页面路径
|
|||
|
|
Path string `json:"path,omitempty"`
|
|||
|
|
// checkPath 检查page 是否存在，为 true 时 page 必须是已经发布的小程序存在的页面（否则报错）；为 false 时允许小程序未发布或者 page 不存在， 但page 有数量上限（60000个）请勿滥用
|
|||
|
|
CheckPath bool `json:"check_path,omitempty"`
|
|||
|
|
// width 图片宽度
|
|||
|
|
Width int `json:"width,omitempty"`
|
|||
|
|
// scene 最大32个可见字符，只支持数字，大小写英文以及部分特殊字符：!#$&'()*+,/:;=?@-._~，其它字符请自行编码为合法字符（因不支持%，中文无法使用 urlencode 处理，请使用其他编码方式）
|
|||
|
|
Scene string `json:"scene,omitempty"`
|
|||
|
|
// autoColor 自动配置线条颜色，如果颜色依然是黑色，则说明不建议配置主色调
|
|||
|
|
AutoColor bool `json:"auto_color,omitempty"`
|
|||
|
|
// lineColor AutoColor 为 false 时生效，使用 rgb 设置颜色 例如 {"r":"xxx","g":"xxx","b":"xxx"},十进制表示
|
|||
|
|
LineColor Color `json:"line_color,omitempty"`
|
|||
|
|
// isHyaline 是否需要透明底色
|
|||
|
|
IsHyaline bool `json:"is_hyaline,omitempty"`
|
|||
|
|
// envVersion 要打开的小程序版本。正式版为 "release"，体验版为 "trial"，开发版为 "develop"
|
|||
|
|
EnvVersion string `json:"env_version,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// fetchCode 请求并返回二维码二进制数据
|
|||
|
|
func (qrCode *QRCode) fetchCode(urlStr string, body interface{}) (response []byte, err error) {
|
|||
|
|
var accessToken string
|
|||
|
|
accessToken, err = qrCode.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
urlStr = fmt.Sprintf(urlStr, accessToken)
|
|||
|
|
var contentType string
|
|||
|
|
response, contentType, err = util.PostJSONWithRespContentType(urlStr, body)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if strings.HasPrefix(contentType, "application/json") {
|
|||
|
|
// 返回错误信息
|
|||
|
|
var result util.CommonError
|
|||
|
|
err = json.Unmarshal(response, &result)
|
|||
|
|
if err == nil && result.ErrCode != 0 {
|
|||
|
|
err = fmt.Errorf("fetchCode error : errcode=%v , errmsg=%v", result.ErrCode, result.ErrMsg)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if contentType == "image/jpeg" {
|
|||
|
|
// 返回文件
|
|||
|
|
return response, nil
|
|||
|
|
}
|
|||
|
|
err = fmt.Errorf("fetchCode error : unknown response content type - %v", contentType)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CreateWXAQRCode 获取小程序二维码，适用于需要的码数量较少的业务场景
|
|||
|
|
// 文档地址： https://developers.weixin.qq.com/miniprogram/dev/api/createWXAQRCode.html
|
|||
|
|
func (qrCode *QRCode) CreateWXAQRCode(coderParams QRCoder) (response []byte, err error) {
|
|||
|
|
return qrCode.fetchCode(createWXAQRCodeURL, coderParams)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetWXACode 获取小程序码，适用于需要的码数量较少的业务场景
|
|||
|
|
// 文档地址： https://developers.weixin.qq.com/miniprogram/dev/api/getWXACode.html
|
|||
|
|
func (qrCode *QRCode) GetWXACode(coderParams QRCoder) (response []byte, err error) {
|
|||
|
|
return qrCode.fetchCode(getWXACodeURL, coderParams)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetWXACodeUnlimit 获取小程序码，适用于需要的码数量极多的业务场景
|
|||
|
|
// 文档地址： https://developers.weixin.qq.com/miniprogram/dev/api/getWXACodeUnlimit.html
|
|||
|
|
func (qrCode *QRCode) GetWXACodeUnlimit(coderParams QRCoder) (response []byte, err error) {
|
|||
|
|
return qrCode.fetchCode(getWXACodeUnlimitURL, coderParams)
|
|||
|
|
}
|