Merge pull request #103 from ckeyer/support_qr

生成带参数的二维码

qr/qr.go

@@ -0,0 +1,122 @@
+package qr
+
+import (
+	"encoding/json"
+	"fmt"
+	"reflect"
+	"time"
+
+	"github.com/silenceper/wechat/context"
+	"github.com/silenceper/wechat/util"
+)
+
+const (
+	qrCreateURL = "https://api.weixin.qq.com/cgi-bin/qrcode/create?access_token=%s"
+	getQRImgURL = "https://mp.weixin.qq.com/cgi-bin/showqrcode?ticket=%s"
+)
+
+const (
+	actionID  = "QR_SCENE"
+	actionStr = "QR_STR_SCENE"
+
+	actionLimitID  = "QR_LIMIT_SCENE"
+	actionLimitStr = "QR_LIMIT_STR_SCENE"
+)
+
+// QR 二维码
+type QR struct {
+	*context.Context
+}
+
+//NewQR 二维码实例
+func NewQR(context *context.Context) *QR {
+	q := new(QR)
+	q.Context = context
+	return q
+}
+
+// Request 临时二维码
+type Request struct {
+	ExpireSeconds int64  `json:"expire_seconds,omitempty"`
+	ActionName    string `json:"action_name"`
+	ActionInfo    struct {
+		Scene struct {
+			SceneStr string `json:"scene_str,omitempty"`
+			SceneID  int    `json:"scene_id,omitempty"`
+		} `json:"scene"`
+	} `json:"action_info"`
+}
+
+// Ticket 二维码ticket
+type Ticket struct {
+	util.CommonError `json:",inline"`
+	Ticket           string `json:"ticket"`
+	ExpireSeconds    int64  `json:"expire_seconds"`
+	URL              string `json:"url"`
+}
+
+// GetQRTicket 获取二维码 Ticket
+func (q *QR) GetQRTicket(tq *Request) (t *Ticket, err error) {
+	accessToken, err := q.GetAccessToken()
+	if err != nil {
+		return
+	}
+
+	uri := fmt.Sprintf(qrCreateURL, accessToken)
+	response, err := util.PostJSON(uri, tq)
+	if err != nil {
+		err = fmt.Errorf("get qr ticket failed, %s", err)
+		return
+	}
+
+	t = new(Ticket)
+	err = json.Unmarshal(response, &t)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+// ShowQRCode 通过ticket换取二维码
+func ShowQRCode(tk *Ticket) string {
+	return fmt.Sprintf(getQRImgURL, tk.Ticket)
+}
+
+// NewTmpQrRequest 新建临时二维码请求实例
+func NewTmpQrRequest(exp time.Duration, scene interface{}) *Request {
+	tq := &Request{
+		ExpireSeconds: int64(exp.Seconds()),
+	}
+	switch reflect.ValueOf(scene).Kind() {
+	case reflect.String:
+		tq.ActionName = actionStr
+		tq.ActionInfo.Scene.SceneStr = scene.(string)
+	case reflect.Int, reflect.Int8, reflect.Int16,
+		reflect.Int32, reflect.Int64,
+		reflect.Uint, reflect.Uint8, reflect.Uint16,
+		reflect.Uint32, reflect.Uint64:
+		tq.ActionName = actionID
+		tq.ActionInfo.Scene.SceneID = scene.(int)
+	}
+
+	return tq
+}
+
+// NewLimitQrRequest 新建永久二维码请求实例
+func NewLimitQrRequest(scene interface{}) *Request {
+	tq := &Request{}
+	switch reflect.ValueOf(scene).Kind() {
+	case reflect.String:
+		tq.ActionName = actionLimitStr
+		tq.ActionInfo.Scene.SceneStr = scene.(string)
+	case reflect.Int, reflect.Int8, reflect.Int16,
+		reflect.Int32, reflect.Int64,
+		reflect.Uint, reflect.Uint8, reflect.Uint16,
+		reflect.Uint32, reflect.Uint64:
+		tq.ActionName = actionLimitID
+		tq.ActionInfo.Scene.SceneID = scene.(int)
+	}
+
+	return tq
+}

wechat.go

@@ -12,6 +12,7 @@ import (
 	"github.com/silenceper/wechat/menu"
 	"github.com/silenceper/wechat/oauth"
 	"github.com/silenceper/wechat/pay"
+	"github.com/silenceper/wechat/qr"
 	"github.com/silenceper/wechat/server"
 	"github.com/silenceper/wechat/template"
 	"github.com/silenceper/wechat/user"
@@ -101,6 +102,11 @@ func (wc *Wechat) GetPay() *pay.Pay {
 	return pay.NewPay(wc.Context)
 }
 
+// GetQR 返回二维码的实例
+func (wc *Wechat) GetQR() *qr.QR {
+	return qr.NewQR(wc.Context)
+}
+
 // GetMiniProgram 获取小程序的实例
 func (wc *Wechat) GetMiniProgram() *miniprogram.MiniProgram {
 	return miniprogram.NewMiniProgram(wc.Context)