添加文件: intelligence_finance_v1/client.go
This commit is contained in:
parent
78fce20c2e
commit
5ff9752895
|
|
@ -0,0 +1,345 @@
|
||||||
|
package intelligence_finance_v1
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// DefaultTimeout 默认请求超时时间
|
||||||
|
DefaultTimeout = 30 * time.Second
|
||||||
|
// DefaultNonceLength 默认随机数长度
|
||||||
|
DefaultNonceLength = 16
|
||||||
|
// HeaderTenantID 租户ID请求头
|
||||||
|
HeaderTenantID = "tenant-id"
|
||||||
|
// HeaderClientID 应用标识请求头
|
||||||
|
HeaderClientID = "client-id"
|
||||||
|
// HeaderSignatureTimestamp 签名时间戳请求头
|
||||||
|
HeaderSignatureTimestamp = "x-bfl-signature-timestamp"
|
||||||
|
// HeaderSignatureNonce 签名随机数请求头
|
||||||
|
HeaderSignatureNonce = "x-bfl-signature-nonce"
|
||||||
|
// HeaderSignature 签名信息请求头
|
||||||
|
HeaderSignature = "x-bfl-signature"
|
||||||
|
// ClientIDDingTalk 钉钉AI表格的固定 client-id
|
||||||
|
ClientIDDingTalk = "dd-ai-table"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Client 业财连接 SDK 客户端
|
||||||
|
// 用于调用平台提供的接口（订单开票、开票状态查询、创建付款单据、支付状态查询）
|
||||||
|
// 以及处理平台回调通知
|
||||||
|
type Client struct {
|
||||||
|
// BaseURL 平台接口基础地址
|
||||||
|
BaseURL string
|
||||||
|
// TenantID 平台分配的租户唯一标识
|
||||||
|
TenantID string
|
||||||
|
// ClientID 平台分配的应用标识
|
||||||
|
ClientID string
|
||||||
|
// ClientSecret 平台分配的密钥，用于签名
|
||||||
|
ClientSecret string
|
||||||
|
// HTTPClient HTTP 客户端
|
||||||
|
HTTPClient *http.Client
|
||||||
|
// NonceLength 随机数长度
|
||||||
|
NonceLength int
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClientOption 客户端配置选项
|
||||||
|
type ClientOption func(*Client)
|
||||||
|
|
||||||
|
// WithHTTPClient 设置自定义 HTTP 客户端
|
||||||
|
func WithHTTPClient(httpClient *http.Client) ClientOption {
|
||||||
|
return func(c *Client) {
|
||||||
|
c.HTTPClient = httpClient
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithNonceLength 设置随机数长度
|
||||||
|
func WithNonceLength(length int) ClientOption {
|
||||||
|
return func(c *Client) {
|
||||||
|
c.NonceLength = length
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewClient 创建一个新的业财连接客户端
|
||||||
|
// baseURL: 平台接口基础地址（如 "https://api.example.com"）
|
||||||
|
// tenantID: 平台分配的租户唯一标识
|
||||||
|
// clientID: 平台分配的应用标识（钉钉AI表格固定为 "dd-ai-table"）
|
||||||
|
// clientSecret: 平台分配的密钥
|
||||||
|
func NewClient(baseURL, tenantID, clientID, clientSecret string, opts ...ClientOption) *Client {
|
||||||
|
c := &Client{
|
||||||
|
BaseURL: baseURL,
|
||||||
|
TenantID: tenantID,
|
||||||
|
ClientID: clientID,
|
||||||
|
ClientSecret: clientSecret,
|
||||||
|
HTTPClient: &http.Client{
|
||||||
|
Timeout: DefaultTimeout,
|
||||||
|
},
|
||||||
|
NonceLength: DefaultNonceLength,
|
||||||
|
}
|
||||||
|
for _, opt := range opts {
|
||||||
|
opt(c)
|
||||||
|
}
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewDingTalkClient 创建一个钉钉AI表格专用的客户端
|
||||||
|
// baseURL: 平台接口基础地址
|
||||||
|
// tenantID: 平台分配的租户唯一标识
|
||||||
|
// clientSecret: 平台分配的密钥（在钉钉AI表格中配置为 APPSecret）
|
||||||
|
// 注意：钉钉AI表格的 client-id 固定为 "dd-ai-table"，签名由AI表格自动完成
|
||||||
|
func NewDingTalkClient(baseURL, tenantID, clientSecret string, opts ...ClientOption) *Client {
|
||||||
|
return NewClient(baseURL, tenantID, ClientIDDingTalk, clientSecret, opts...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 签名相关方法
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// buildSignature 构建请求签名
|
||||||
|
// 签名算法：HmacSHA256(client-secret, timestamp + nonce)，结果 Base64 编码
|
||||||
|
func (c *Client) buildSignature(timestamp, nonce string) string {
|
||||||
|
data := timestamp + nonce
|
||||||
|
return HmacSHA256Sign(c.ClientSecret, data)
|
||||||
|
}
|
||||||
|
|
||||||
|
// setAuthHeaders 设置认证请求头
|
||||||
|
func (c *Client) setAuthHeaders(req *http.Request, timestamp, nonce string) {
|
||||||
|
req.Header.Set(HeaderTenantID, c.TenantID)
|
||||||
|
req.Header.Set(HeaderClientID, c.ClientID)
|
||||||
|
req.Header.Set(HeaderSignatureTimestamp, timestamp)
|
||||||
|
req.Header.Set(HeaderSignatureNonce, nonce)
|
||||||
|
req.Header.Set(HeaderSignature, c.buildSignature(timestamp, nonce))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 内部 HTTP 请求方法
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// doRequest 执行带签名的 POST 请求
|
||||||
|
func (c *Client) doRequest(ctx context.Context, path string, requestBody interface{}) (*commonResponse, error) {
|
||||||
|
bodyBytes, err := json.Marshal(requestBody)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal request body: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
url := c.BaseURL + path
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(bodyBytes))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("create request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
timestamp := GenerateTimestamp()
|
||||||
|
nonce, err := GenerateNonce(c.NonceLength)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("generate nonce: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
c.setAuthHeaders(req, timestamp, nonce)
|
||||||
|
|
||||||
|
resp, err := c.HTTPClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("execute request: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
respBody, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("read response body: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
|
return nil, fmt.Errorf("unexpected HTTP status: %d, body: %s", resp.StatusCode, string(respBody))
|
||||||
|
}
|
||||||
|
|
||||||
|
var commonResp commonResponse
|
||||||
|
if err := json.Unmarshal(respBody, &commonResp); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if commonResp.Code != 0 {
|
||||||
|
return nil, NewAPIError(commonResp.Code, commonResp.Msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &commonResp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 接口 1：订单开票
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// CreateInvoice 提交订单开票请求
|
||||||
|
// path: 接口路径（接口短码，由平台对接时分配）
|
||||||
|
// req: 开票请求参数
|
||||||
|
// 返回开票响应，包含开票状态和发票数据列表
|
||||||
|
func (c *Client) CreateInvoice(ctx context.Context, path string, req *InvoiceRequest) (*InvoiceResponse, error) {
|
||||||
|
commonResp, err := c.doRequest(ctx, path, req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将 data 字段重新序列化后反序列化为 InvoiceResponse
|
||||||
|
dataBytes, err := json.Marshal(commonResp.Data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal response data: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var invoiceResp InvoiceResponse
|
||||||
|
if err := json.Unmarshal(dataBytes, &invoiceResp); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal invoice response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &invoiceResp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 接口 2：开票状态查询
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// QueryInvoiceStatus 查询订单开票状态
|
||||||
|
// path: 接口路径（接口短码，由平台对接时分配）
|
||||||
|
// req: 查询请求参数
|
||||||
|
// 返回开票状态查询响应
|
||||||
|
func (c *Client) QueryInvoiceStatus(ctx context.Context, path string, req *InvoiceStatusQueryRequest) (*InvoiceStatusQueryResponse, error) {
|
||||||
|
commonResp, err := c.doRequest(ctx, path, req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
dataBytes, err := json.Marshal(commonResp.Data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal response data: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var statusResp InvoiceStatusQueryResponse
|
||||||
|
if err := json.Unmarshal(dataBytes, &statusResp); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal invoice status response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &statusResp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 接口 3：创建付款单据
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// CreatePaymentDocument 创建付款单据
|
||||||
|
// path: 接口路径（接口短码，由平台对接时分配）
|
||||||
|
// req: 付款单据请求参数
|
||||||
|
// 返回创建结果，包含单据唯一标识
|
||||||
|
func (c *Client) CreatePaymentDocument(ctx context.Context, path string, req *PaymentDocumentRequest) (*PaymentDocumentResponse, error) {
|
||||||
|
commonResp, err := c.doRequest(ctx, path, req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
dataBytes, err := json.Marshal(commonResp.Data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal response data: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var paymentResp PaymentDocumentResponse
|
||||||
|
if err := json.Unmarshal(dataBytes, &paymentResp); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal payment document response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &paymentResp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 接口 5：支付状态查询
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// QueryPaymentStatus 查询支付状态
|
||||||
|
// path: 接口路径（接口短码，由平台对接时分配）
|
||||||
|
// req: 查询请求参数
|
||||||
|
// 返回支付状态信息，结构与支付通知一致
|
||||||
|
func (c *Client) QueryPaymentStatus(ctx context.Context, path string, req *PaymentStatusQueryRequest) (*PaymentStatusQueryResponse, error) {
|
||||||
|
commonResp, err := c.doRequest(ctx, path, req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
dataBytes, err := json.Marshal(commonResp.Data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal response data: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var paymentStatusResp PaymentStatusQueryResponse
|
||||||
|
if err := json.Unmarshal(dataBytes, &paymentStatusResp); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal payment status response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &paymentStatusResp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// 回调通知处理（接口 4：支付完成通知）
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// ParsePaymentNotification 从 HTTP 请求中解析支付完成通知
|
||||||
|
// 该方法用于客户接收平台回调时使用
|
||||||
|
func ParsePaymentNotification(r *http.Request) (*PaymentNotification, error) {
|
||||||
|
if r.Method != http.MethodPost {
|
||||||
|
return nil, fmt.Errorf("invalid HTTP method: %s, expected POST", r.Method)
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := io.ReadAll(r.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("read notification body: %w", err)
|
||||||
|
}
|
||||||
|
defer r.Body.Close()
|
||||||
|
|
||||||
|
var notification PaymentNotification
|
||||||
|
if err := json.Unmarshal(body, ¬ification); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal payment notification: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ¬ification, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseGeneralNotification 从 HTTP 请求中解析通用通知
|
||||||
|
func ParseGeneralNotification(r *http.Request) (*GeneralNotification, error) {
|
||||||
|
if r.Method != http.MethodPost {
|
||||||
|
return nil, fmt.Errorf("invalid HTTP method: %s, expected POST", r.Method)
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := io.ReadAll(r.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("read notification body: %w", err)
|
||||||
|
}
|
||||||
|
defer r.Body.Close()
|
||||||
|
|
||||||
|
var notification GeneralNotification
|
||||||
|
if err := json.Unmarshal(body, ¬ification); err != nil {
|
||||||
|
return nil, fmt.Errorf("unmarshal general notification: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ¬ification, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// VerifyNotificationSignature 验证回调通知签名
|
||||||
|
// payload: 原始请求体
|
||||||
|
// signature: 请求中的签名
|
||||||
|
// secret: 客户在平台配置的验签密钥
|
||||||
|
// 使用 HmacSHA256 算法验证
|
||||||
|
func VerifyNotificationSignature(payload []byte, signature, secret string) bool {
|
||||||
|
return HmacSHA256Verify(secret, string(payload), signature)
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriteNotificationResponse 写入通知响应
|
||||||
|
// success: true 返回 "SUCCESS"，false 返回 "FAILED"
|
||||||
|
func WriteNotificationResponse(w http.ResponseWriter, success bool) {
|
||||||
|
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||||||
|
if success {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
w.Write([]byte(NotificationSuccess))
|
||||||
|
} else {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
w.Write([]byte(NotificationFailed))
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue