package domain

import (
	"encoding/base64"
	"fmt"
	"github.com/forgoer/openssl"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-gateway/pkg/constant"
	"time"

	jwt "github.com/dgrijalva/jwt-go"
)

const (
	qrcodeTokenSecret string = "bbe35ad433dd8e67"
	qrcodeCodeExpire  int64  = 60 * 30 //15分钟过期
)

var aecSecret = []byte("mmm.qrcode.ecb.1")
var qrcodeLogin = "/v1/auth/login/qrcode?key="

type QrcodeMessage struct {
	jwt.StandardClaims
	Id      string `json:"id"`
	Token   string `json:"token"`
	IsLogin bool   `json:"isLogin"`
	//用户id
	UserId     int64 `json:"userId"`
	UserBaseId int64 `json:"userBaseId"`
	// 账号
	Account string `json:"account"`
	// 公司id
	CompanyId int64 `json:"companyId"`
	// 组织id
	OrgId int64 `json:"orgId"`
}

func (qrmsg *QrcodeMessage) GenerateImageBase64() ([]byte, error) {
	nowTime := time.Now().Unix()
	qrmsg.StandardClaims = jwt.StandardClaims{
		NotBefore: nowTime,
		IssuedAt:  nowTime,
		ExpiresAt: nowTime + qrcodeCodeExpire,
		Issuer:    "allied_creation_gateway",
	}
	qrmsg.Id = fmt.Sprintf("%d", time.Now().UnixNano())
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, *qrmsg)
	str, err := token.SignedString([]byte(qrcodeTokenSecret))
	if err != nil {
		return nil, err
	}
	key := constant.ALLIED_CREATION_GATEWAY_HOST + qrcodeLogin + str
	encryptedData, err := openssl.AesECBEncrypt([]byte(key), aecSecret, openssl.PKCS7_PADDING)
	if err != nil {
		return nil, err
	}
	qrmsg.Token = base64.StdEncoding.EncodeToString(encryptedData)
	qrmsg.IsLogin = false
	return encryptedData, err
}

func (qrmsg *QrcodeMessage) ParseToken(str string) error {
	tokenClaims, err := jwt.ParseWithClaims(
		str,
		qrmsg,
		func(token *jwt.Token) (interface{}, error) {
			return []byte(loginTokenSecret), nil
		})
	if err != nil {
		return err
	}
	if claim, ok := tokenClaims.Claims.(*QrcodeMessage); ok && tokenClaims.Valid {
		*qrmsg = *claim
	}
	return nil
}

func (qrmsg *QrcodeMessage) BindUser(operator Operator) error {
	if qrmsg.IsLogin {
		return fmt.Errorf("登录中")
	}
	qrmsg.UserId = operator.UserId
	qrmsg.OrgId = operator.OrgId
	qrmsg.CompanyId = operator.CompanyId
	qrmsg.UserBaseId = operator.UserBaseId
	qrmsg.Account = operator.Phone
	qrmsg.IsLogin = true
	return nil
}