aiforge/routers/authentication/wechat.go

package authentication

import (
	"code.gitea.io/gitea/modules/auth/wechat"
	"code.gitea.io/gitea/modules/context"
	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/redis/redis_client"
	"code.gitea.io/gitea/modules/redis/redis_key"
	"code.gitea.io/gitea/modules/setting"
	"encoding/json"
	"errors"
	gouuid "github.com/satori/go.uuid"
	"time"
)

type QRCodeResponse struct {
	Url           string
	Ticket        string
	SceneStr      string
	ExpireSeconds int
}

// GetQRCode4Bind get QR code for wechat binding
func GetQRCode4Bind(ctx *context.Context) {
	userId := ctx.User.ID

	r, err := createQRCode4Bind(userId)
	if err != nil {
		ctx.JSON(200, map[string]interface{}{
			"code": "9999",
			"msg":  "Get QR code failed",
		})
		return
	}

	ctx.JSON(200, map[string]interface{}{
		"code": "00",
		"msg":  "success",
		"data": r,
	})
}

// GetQRCode4Bind get QR code for wechat binding
func GetBindStatus(ctx *context.Context) {
	sceneStr := ctx.Query("sceneStr")
	val, _ := redis_client.Get(redis_key.WechatBindingUserIdKey(sceneStr))
	if val == "" {
		ctx.JSON(200, map[string]interface{}{
			"code": "00",
			"msg":  "QR code expired",
			"data": map[string]interface{}{
				"status": wechat.BIND_STATUS_EXPIRED,
			},
		})
		return
	}
	qrCache := new(wechat.QRCode4BindCache)
	json.Unmarshal([]byte(val), qrCache)
	ctx.JSON(200, map[string]interface{}{
		"code": "00",
		"msg":  "success",
		"data": map[string]interface{}{
			"status": qrCache.Status,
		},
	})
}

func createQRCode4Bind(userId int64) (*QRCodeResponse, error) {
	sceneStr := gouuid.NewV4().String()
	r := wechat.GetWechatQRCode4Bind(sceneStr)
	if r == nil {
		return nil, errors.New("createQRCode4Bind failed")
	}

	jsonStr, _ := json.Marshal(&wechat.QRCode4BindCache{
		UserId: userId,
		Status: wechat.BIND_STATUS_UNBIND,
	})
	isOk, err := redis_client.Setex(redis_key.WechatBindingUserIdKey(sceneStr), string(jsonStr), time.Duration(setting.WechatQRCodeExpireSeconds)*time.Second)
	if err != nil {
		log.Error("createQRCode4Bind failed.e=%v", err)
		return nil, err
	}
	if !isOk {
		log.Error("createQRCode4Bind failed.redis reply is not ok")
		return nil, errors.New("reply is not ok when set WechatBindingUserIdKey")
	}

	result := &QRCodeResponse{
		Url:           r.Url,
		Ticket:        r.Ticket,
		SceneStr:      sceneStr,
		ExpireSeconds: setting.WechatQRCodeExpireSeconds,
	}
	return result, nil
}