third-platform-sdk/platform/t3-union/api.go

package t3_union

import (
	"context"
	"errors"
	"net/http"

	"github.com/zeromicro/go-zero/core/logx"

	"repository.lenntc.com/lenntc/third-platform-sdk/client"
	"repository.lenntc.com/lenntc/third-platform-sdk/util"
)

// T3UnionApi 调用第三方平台的api
// Api defines the interface of t3_union api
type T3UnionApi interface {
	// Sign 签名
	Sign(methodType string, url string, data map[string]interface{}) string
	// GenerateLink 生成短链
	GenerateLink(ctx context.Context, req GenerateLinkRequest) (*GenerateLinkData, error)
	// GenerateCode 生成二维码
	GenerateCode(ctx context.Context, req GenerateCodeRequest) (*GenerateCodeData, error)
	// GeneratePoster 生成推广海报
	GeneratePoster(ctx context.Context, req GeneratePosterRequest) (*GeneratePosterData, error)
	// QueryOrderList 查询订单列表
	QueryOrderList(ctx context.Context, req QueryOrderListRequest) (*QueryOrderData, error)
}

type t3UnionApiImpl struct {
	log    logx.Logger
	client *Client
	sign   *Sign
}

func newT3UnionApiImpl(log logx.Logger, client *Client) T3UnionApi {
	sign := newSign(client.authConfig.AppKey, client.authConfig.AppSecret)
	return &t3UnionApiImpl{
		log:    log,
		client: client,
		sign:   sign,
	}
}

// Sign 签名
func (a *t3UnionApiImpl) Sign(methodType string, uri string, dataMap map[string]any) string {
	headers := a.sign.BuildHeader(methodType, uri, dataMap)
	if sign, ok := headers[XT3Signature]; ok {
		return sign
	}
	return ""
}

// GenerateLink 生成短链
func (a *t3UnionApiImpl) GenerateLink(ctx context.Context, req GenerateLinkRequest) (*GenerateLinkData, error) {
	args := util.StructToMap(req)
	headers := a.sign.BuildHeader(http.MethodPost, GetLinkUri, args)
	for k, v := range a.client.headers {
		headers[k] = v
	}
	request := &client.HttpRequest{Headers: headers, BodyArgs: args}
	response := new(GenerateLinkResponse)
	if err := a.client.HttpPost(GetLinkUrl, request, &client.HttpResponse{Result: response}); err != nil {
		return nil, err
	}
	if !response.Success || response.ErrCode != 0 {
		a.log.WithFields(logx.LogField{Key: "data", Value: map[string]any{"req": req, "resp": response}}).
			Errorf("[t3UnionApiImpl][GenerateLink] response result error: %s", response.Msg)
		return nil, errors.New(response.Msg)
	}
	return response.Data, nil
}

// GenerateCode 生成二维码
func (a *t3UnionApiImpl) GenerateCode(ctx context.Context, req GenerateCodeRequest) (*GenerateCodeData, error) {
	args := util.StructToMap(req)
	headers := a.sign.BuildHeader(http.MethodPost, GetMiniCodeUri, args)
	for k, v := range a.client.headers {
		headers[k] = v
	}
	request := &client.HttpRequest{Headers: headers, BodyArgs: args}
	response := new(GenerateCodeResponse)
	if err := a.client.HttpPost(GetMiniCodeUrl, request, &client.HttpResponse{Result: response}); err != nil {
		return nil, err
	}
	if !response.Success || response.ErrCode != 0 {
		a.log.WithFields(logx.LogField{Key: "data", Value: map[string]any{"req": req, "resp": response}}).
			Errorf("[t3UnionApiImpl][GenerateCode] response result error: %s", response.Msg)
		return nil, errors.New(response.Msg)
	}
	return response.Data, nil
}

// GeneratePoster 生成推广海报
func (a *t3UnionApiImpl) GeneratePoster(ctx context.Context, req GeneratePosterRequest) (*GeneratePosterData, error) {
	args := util.StructToMap(req)
	headers := a.sign.BuildHeader(http.MethodPost, GetPosterUri, args)
	for k, v := range a.client.headers {
		headers[k] = v
	}
	request := &client.HttpRequest{Headers: headers, BodyArgs: args}
	response := new(GeneratePosterResponse)
	if err := a.client.HttpPost(GetPosterUrl, request, &client.HttpResponse{Result: response}); err != nil {
		return nil, err
	}
	if !response.Success || response.ErrCode != 0 {
		a.log.WithFields(logx.LogField{Key: "data", Value: map[string]any{"req": req, "resp": response}}).
			Errorf("[t3UnionApiImpl][GeneratePoster] response result error: %s", response.Msg)
		return nil, errors.New(response.Msg)
	}
	return response.Data, nil
}

// QueryOrderList 查询订单列表
func (a *t3UnionApiImpl) QueryOrderList(ctx context.Context, req QueryOrderListRequest) (*QueryOrderData, error) {
	if len(req.SupplierUuid) == 0 {
		req.SupplierUuid = a.client.authConfig.SupplierID
	}
	args := util.StructToMap(req)
	headers := a.sign.BuildHeader(http.MethodPost, GetOrderListUri, args)
	for k, v := range a.client.headers {
		headers[k] = v
	}
	request := &client.HttpRequest{Headers: headers, BodyArgs: args}
	response := new(QueryOrderListResponse)
	if err := a.client.HttpPost(GetOrderListUrl, request, &client.HttpResponse{Result: response}); err != nil {
		return nil, err
	}
	if !response.Success || response.ErrCode != 0 {
		a.log.WithFields(logx.LogField{Key: "data", Value: map[string]any{"req": req, "resp": response}}).
			Errorf("[t3UnionApiImpl][QueryOrderList] response result error: %s", response.Msg)
		return nil, errors.New(response.Msg)
	}
	return response.Data, nil
}