voucher/internal/data/mixrepoimpl/cmb.go

package mixrepoimpl

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"strings"
	"time"
	v1 "voucher/api/v1"
	"voucher/internal/biz/mixrepos"
	"voucher/internal/conf"
	"voucher/internal/pkg/cmb"
	"voucher/internal/pkg/helper"
)

type CmbMixRepoImpl struct {
	bc *conf.Bootstrap
}

func NewCmbMixRepoImpl(bc *conf.Bootstrap) mixrepos.CmbMixRepo {
	return &CmbMixRepoImpl{bc: bc}
}

func (s *CmbMixRepoImpl) OrderVerify(ctx context.Context, req *v1.CmbRequest, funcName string) (*v1.CmbOrderRequest, error) {
	bizStr, err := s.Verify(ctx, req, funcName)
	if err != nil {
		return nil, err
	}

	var resp *v1.CmbOrderRequest
	if err = json.Unmarshal([]byte(bizStr), &resp); err != nil {
		return nil, err
	}

	return resp, nil
}

func (s *CmbMixRepoImpl) ProductQueryVerify(ctx context.Context, req *v1.CmbRequest, funcName string) (*v1.CmbQueryProductRequest, error) {
	bizStr, err := s.Verify(ctx, req, funcName)
	if err != nil {
		return nil, err
	}

	var resp *v1.CmbQueryProductRequest
	if err = json.Unmarshal([]byte(bizStr), &resp); err != nil {
		return nil, err
	}

	return resp, nil
}

func (s *CmbMixRepoImpl) Verify(ctx context.Context, req *v1.CmbRequest, funcName string) (string, error) {
	str, err := s.getSignStr(ctx, req, funcName)
	if err != nil {
		return "", err
	}

	b, err := cmb.Verify(s.bc.Cmb.CmbSm2Puk, str, req.Sign)
	if err != nil {
		return "", err
	}

	if !b {
		return "", errors.New("签名验证失败")
	}

	bizStr, err := cmb.Decrypt(s.bc.Cmb.Sm2Prk, req.EncryptBody)
	if err != nil {
		return "", err
	}

	return bizStr, nil
}

func (s *CmbMixRepoImpl) getSignStr(_ context.Context, req *v1.CmbRequest, funcName string) (string, error) {
	kvRows := helper.SortStructFieldsByKey(req)

	var strToBeSigned strings.Builder
	for _, kv := range kvRows {
		if kv.Key == "sign" {
			continue
		}
		if kv.Value == "" {
			continue
		}
		strToBeSigned.WriteString(fmt.Sprintf("%s=%s&", kv.Key, kv.Value))
	}

	return fmt.Sprintf("%s?%s", funcName, strings.TrimRight(strToBeSigned.String(), "&")), nil
}

func (s *CmbMixRepoImpl) GetRequestData(ctx context.Context, funcName, bizJsonStr string) (*v1.CmbRequest, error) {

	encryptBody, err := cmb.Encrypt(s.bc.Cmb.Sm2Puk, bizJsonStr)
	if err != nil {
		return nil, err
	}

	date := time.Now().Format("20060102150405")

	req := &v1.CmbRequest{
		Mid:         s.bc.Cmb.Mid,
		Aid:         s.bc.Cmb.Aid,
		Date:        date,
		Random:      string(cmb.RandomBytes(16)),
		KeyAlias:    s.bc.Cmb.KeyAlias,
		CmbKeyAlias: s.bc.Cmb.CmbKeyAlias,
		EncryptBody: encryptBody,
		Sign:        "",
	}

	str, err := s.getSignStr(ctx, req, funcName)
	if err != nil {
		return nil, err
	}

	sing, err := cmb.Sign(s.bc.Cmb.CmbSm2Pik, str)
	if err != nil {
		return nil, err
	}
	req.Sign = sing

	return req, nil
}

func (s *CmbMixRepoImpl) GetResponseData(ctx context.Context, funcName, bizJsonStr string) (*v1.CmbRequest, error) {

	encryptBody, err := cmb.Encrypt(s.bc.Cmb.Sm2Puk, bizJsonStr)
	if err != nil {
		return nil, err
	}

	date := time.Now().Format("20060102150405")

	req := &v1.CmbRequest{
		Mid:         s.bc.Cmb.Mid,
		Aid:         s.bc.Cmb.Aid,
		Date:        date,
		Random:      string(cmb.RandomBytes(16)),
		KeyAlias:    s.bc.Cmb.KeyAlias,
		CmbKeyAlias: s.bc.Cmb.CmbKeyAlias,
		EncryptBody: encryptBody,
		Sign:        "",
	}

	str, err := s.getSignStr(ctx, req, funcName)
	if err != nil {
		return nil, err
	}

	sing, err := cmb.Sign(s.bc.Cmb.CmbSm2Pik, str)
	if err != nil {
		return nil, err
	}
	req.Sign = sing

	return req, nil
}

func (s *CmbMixRepoImpl) Request(ctx context.Context, funcName, bizJsonStr string) (*v1.CmbRequest, error) {
	data, err := s.GetRequestData(ctx, funcName, bizJsonStr)
	if err != nil {
		return nil, err
	}
	fmt.Print(data)

	return nil, nil
}