添加文件: xy_sh/client.go

xy_sh/client.go

@@ -0,0 +1,244 @@
+package xy_sh
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+)
+
+// Client 供应商接口客户端
+type Client struct {
+	httpClient  *http.Client
+	sm4Key      []byte // SM4加密密钥，16字节
+	sm3Salt     []byte // SM3签名盐值
+	orderURL    string // 下单接口地址
+	queryURL    string // 查询接口地址
+	rechargeURL string // 微信立减金充值接口地址
+}
+
+// NewClient 创建新的客户端
+//   - sm4Key: SM4加密密钥，必须为16字节
+//   - sm3Salt: SM3签名盐值
+//   - orderURL: 下单接口地址
+//   - queryURL: 查询接口地址
+//   - rechargeURL: 微信立减金充值接口地址
+func NewClient(sm4Key []byte, sm3Salt []byte, orderURL, queryURL, rechargeURL string) (*Client, error) {
+	if len(sm4Key) != 16 {
+		return nil, fmt.Errorf("SM4密钥长度必须为16字节")
+	}
+	if len(sm3Salt) == 0 {
+		return nil, fmt.Errorf("SM3盐值不能为空")
+	}
+	return &Client{
+		httpClient:  &http.Client{},
+		sm4Key:      sm4Key,
+		sm3Salt:     sm3Salt,
+		orderURL:    orderURL,
+		queryURL:    queryURL,
+		rechargeURL: rechargeURL,
+	}, nil
+}
+
+// SetHTTPClient 设置自定义HTTP客户端
+func (c *Client) SetHTTPClient(httpClient *http.Client) {
+	c.httpClient = httpClient
+}
+
+// doRequest 执行加密请求并解密响应
+func (c *Client) doRequest(ctx context.Context, url string, bizReq interface{}) (*RawResponse, error) {
+	// 1. 序列化业务参数为JSON
+	bizBody, err := json.Marshal(bizReq)
+	if err != nil {
+		return nil, fmt.Errorf("序列化业务参数失败: %v", err)
+	}
+
+	// 2. SM4加密业务参数
+	encryptedData, err := SM4Encrypt(bizBody, c.sm4Key)
+	if err != nil {
+		return nil, fmt.Errorf("SM4加密失败: %v", err)
+	}
+
+	// 3. 生成时间戳和签名
+	timestamp := GenerateTimestampMillis()
+	sign := GenerateSign(timestamp, encryptedData, c.sm3Salt)
+
+	// 4. 构建加密请求体
+	reqBody := EncryptedRequest{EncryptedData: encryptedData}
+	reqBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return nil, fmt.Errorf("序列化请求体失败: %v", err)
+	}
+
+	// 5. 创建HTTP请求
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(reqBytes))
+	if err != nil {
+		return nil, fmt.Errorf("创建HTTP请求失败: %v", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("timestamp", timestamp)
+	req.Header.Set("sign", sign)
+
+	// 6. 发送请求
+	resp, err := c.httpClient.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("发送HTTP请求失败: %v", err)
+	}
+	defer resp.Body.Close()
+
+	// 7. 读取响应体
+	respBytes, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("读取响应体失败: %v", err)
+	}
+
+	// 8. 解析原始响应
+	var rawResp RawResponse
+	if err := json.Unmarshal(respBytes, &rawResp); err != nil {
+		return nil, fmt.Errorf("解析响应JSON失败: %v", err)
+	}
+
+	return &rawResp, nil
+}
+
+// decryptData 解密响应中的data字段
+func (c *Client) decryptData(encryptedData string, target interface{}) error {
+	if encryptedData == "" {
+		return nil
+	}
+	plaintext, err := SM4Decrypt(encryptedData, c.sm4Key)
+	if err != nil {
+		return fmt.Errorf("SM4解密失败: %v", err)
+	}
+	if err := json.Unmarshal(plaintext, target); err != nil {
+		return fmt.Errorf("解析解密后数据失败: %v", err)
+	}
+	return nil
+}
+
+// Order 卡券/直充权益下单
+// 适用于卡密、直充商品下单
+func (c *Client) Order(ctx context.Context, req *OrderRequest) (*OrderResponse, error) {
+	rawResp, err := c.doRequest(ctx, c.orderURL, req)
+	if err != nil {
+		return nil, err
+	}
+
+	resp := &OrderResponse{
+		Code: rawResp.Code,
+		Msg:  rawResp.Msg,
+	}
+
+	if rawResp.Data != "" {
+		var data OrderData
+		if err := c.decryptData(rawResp.Data, &data); err != nil {
+			return nil, err
+		}
+		resp.Data = &data
+	}
+
+	return resp, nil
+}
+
+// QueryOrder 卡券/直充/微信立减金订单查询
+// 通过供应商订单号查询订单状态
+func (c *Client) QueryOrder(ctx context.Context, req *QueryRequest) (*QueryResponse, error) {
+	rawResp, err := c.doRequest(ctx, c.queryURL, req)
+	if err != nil {
+		return nil, err
+	}
+
+	resp := &QueryResponse{
+		Code: rawResp.Code,
+		Msg:  rawResp.Msg,
+	}
+
+	if rawResp.Data != "" {
+		var data QueryData
+		if err := c.decryptData(rawResp.Data, &data); err != nil {
+			return nil, err
+		}
+		resp.Data = &data
+	}
+
+	return resp, nil
+}
+
+// WechatRecharge 微信立减金订单充值
+func (c *Client) WechatRecharge(ctx context.Context, req *WechatRechargeRequest) (*WechatRechargeResponse, error) {
+	rawResp, err := c.doRequest(ctx, c.rechargeURL, req)
+	if err != nil {
+		return nil, err
+	}
+
+	resp := &WechatRechargeResponse{
+		Code: rawResp.Code,
+		Msg:  rawResp.Msg,
+	}
+
+	if rawResp.Data != "" {
+		var data WechatRechargeData
+		if err := c.decryptData(rawResp.Data, &data); err != nil {
+			return nil, err
+		}
+		resp.Data = &data
+	}
+
+	return resp, nil
+}
+
+// SendCallback 发送回调通知（供应商调用行方回调接口）
+// 返回的 CallbackResponse 中 StatusCode 为 HTTP 状态码，Body 为响应体内容
+func (c *Client) SendCallback(ctx context.Context, callbackURL string, req *CallbackRequest) (*CallbackResponse, error) {
+	// 1. 序列化业务参数
+	bizBody, err := json.Marshal(req)
+	if err != nil {
+		return nil, fmt.Errorf("序列化回调参数失败: %v", err)
+	}
+
+	// 2. SM4加密
+	encryptedData, err := SM4Encrypt(bizBody, c.sm4Key)
+	if err != nil {
+		return nil, fmt.Errorf("SM4加密失败: %v", err)
+	}
+
+	// 3. 生成时间戳和签名
+	timestamp := GenerateTimestampMillis()
+	sign := GenerateSign(timestamp, encryptedData, c.sm3Salt)
+
+	// 4. 构建请求体
+	reqBody := EncryptedRequest{EncryptedData: encryptedData}
+	reqBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return nil, fmt.Errorf("序列化请求体失败: %v", err)
+	}
+
+	// 5. 创建HTTP请求
+	httpReq, err := http.NewRequestWithContext(ctx, http.MethodPost, callbackURL, bytes.NewReader(reqBytes))
+	if err != nil {
+		return nil, fmt.Errorf("创建HTTP请求失败: %v", err)
+	}
+	httpReq.Header.Set("Content-Type", "application/json")
+	httpReq.Header.Set("timestamp", timestamp)
+	httpReq.Header.Set("sign", sign)
+
+	// 6. 发送请求
+	resp, err := c.httpClient.Do(httpReq)
+	if err != nil {
+		return nil, fmt.Errorf("发送回调请求失败: %v", err)
+	}
+	defer resp.Body.Close()
+
+	// 7. 读取响应体
+	respBytes, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("读取回调响应失败: %v", err)
+	}
+
+	return &CallbackResponse{
+		StatusCode: resp.StatusCode,
+		Body:       string(respBytes),
+	}, nil
+}