添加文件: intelligence_finance_v1_ga/client.go
This commit is contained in:
parent
d37265a894
commit
75d380038b
|
|
@ -0,0 +1,237 @@
|
|||
package intelligence_finance_v1_ga
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 默认接口路径（实际路径以接口对接时分配的短码为准，可通过 Option 覆盖）。
|
||||
const (
|
||||
DefaultInvoiceOrderPath = "/invoice/order"
|
||||
DefaultQueryInvoiceStatusPath = "/invoice/status/query"
|
||||
DefaultCreatePaymentPath = "/payment/create"
|
||||
DefaultQueryPaymentStatusPath = "/payment/status/query"
|
||||
)
|
||||
|
||||
// Client 是业财连接接口的客户端。
|
||||
type Client struct {
|
||||
baseURL string
|
||||
tenantID string
|
||||
clientID string
|
||||
clientSecret string
|
||||
httpClient *http.Client
|
||||
signEnabled bool
|
||||
|
||||
invoiceOrderPath string
|
||||
queryInvoiceStatusPath string
|
||||
createPaymentPath string
|
||||
queryPaymentStatusPath string
|
||||
}
|
||||
|
||||
// Option 定义客户端配置选项。
|
||||
type Option func(*Client)
|
||||
|
||||
// WithHTTPClient 设置自定义 HTTP 客户端。
|
||||
func WithHTTPClient(c *http.Client) Option {
|
||||
return func(cli *Client) {
|
||||
cli.httpClient = c
|
||||
}
|
||||
}
|
||||
|
||||
// WithTimeout 设置请求超时时间。
|
||||
func WithTimeout(d time.Duration) Option {
|
||||
return func(cli *Client) {
|
||||
cli.httpClient.Timeout = d
|
||||
}
|
||||
}
|
||||
|
||||
// WithSignEnabled 设置是否启用签名。
|
||||
// 客户应用需启用签名（默认启用）；钉钉AI表格无需配置签名，可设置为 false。
|
||||
func WithSignEnabled(enabled bool) Option {
|
||||
return func(cli *Client) {
|
||||
cli.signEnabled = enabled
|
||||
}
|
||||
}
|
||||
|
||||
// WithInvoiceOrderPath 覆盖订单开票接口路径。
|
||||
func WithInvoiceOrderPath(p string) Option {
|
||||
return func(cli *Client) { cli.invoiceOrderPath = p }
|
||||
}
|
||||
|
||||
// WithQueryInvoiceStatusPath 覆盖开票状态查询接口路径。
|
||||
func WithQueryInvoiceStatusPath(p string) Option {
|
||||
return func(cli *Client) { cli.queryInvoiceStatusPath = p }
|
||||
}
|
||||
|
||||
// WithCreatePaymentPath 覆盖创建付款单据接口路径。
|
||||
func WithCreatePaymentPath(p string) Option {
|
||||
return func(cli *Client) { cli.createPaymentPath = p }
|
||||
}
|
||||
|
||||
// WithQueryPaymentStatusPath 覆盖支付状态查询接口路径。
|
||||
func WithQueryPaymentStatusPath(p string) Option {
|
||||
return func(cli *Client) { cli.queryPaymentStatusPath = p }
|
||||
}
|
||||
|
||||
// NewClient 创建一个新的业财连接接口客户端。
|
||||
// baseURL 为平台接口地址，tenantID 为平台分配的租户唯一标识，
|
||||
// clientID 为平台分配的应用标识（钉钉AI表格固定为 "dd-ai-table"），
|
||||
// clientSecret 为平台分配的密钥。
|
||||
func NewClient(baseURL, tenantID, clientID, clientSecret string, opts ...Option) *Client {
|
||||
c := &Client{
|
||||
baseURL: strings.TrimRight(baseURL, "/"),
|
||||
tenantID: tenantID,
|
||||
clientID: clientID,
|
||||
clientSecret: clientSecret,
|
||||
httpClient: &http.Client{Timeout: 30 * time.Second},
|
||||
signEnabled: true,
|
||||
invoiceOrderPath: DefaultInvoiceOrderPath,
|
||||
queryInvoiceStatusPath: DefaultQueryInvoiceStatusPath,
|
||||
createPaymentPath: DefaultCreatePaymentPath,
|
||||
queryPaymentStatusPath: DefaultQueryPaymentStatusPath,
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(c)
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
// doRequest 发送 POST 请求并处理通用响应结构，返回 data 字段的原始字节。
|
||||
func (c *Client) doRequest(ctx context.Context, path string, reqBody interface{}) (json.RawMessage, error) {
|
||||
bodyBytes, err := json.Marshal(reqBody)
|
||||
if err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("marshal request: %w", err)}
|
||||
}
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.baseURL+path, bytes.NewReader(bodyBytes))
|
||||
if err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("new request: %w", err)}
|
||||
}
|
||||
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("tenant-id", c.tenantID)
|
||||
req.Header.Set("client-id", c.clientID)
|
||||
|
||||
if c.signEnabled {
|
||||
timestamp := GenerateTimestamp()
|
||||
nonce, err := GenerateNonce(16)
|
||||
if err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("generate nonce: %w", err)}
|
||||
}
|
||||
// 签名算法：HmacSHA256，密钥为 client-secret，签名数据为 timestamp + nonce，结果 Base64 编码。
|
||||
signature := HmacSHA256Base64(c.clientSecret, timestamp+nonce)
|
||||
req.Header.Set("x-bfl-signature-timestamp", timestamp)
|
||||
req.Header.Set("x-bfl-signature-nonce", nonce)
|
||||
req.Header.Set("x-bfl-signature", signature)
|
||||
}
|
||||
|
||||
resp, err := c.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("do request: %w", err)}
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBytes, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("read response: %w", err)}
|
||||
}
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, &APIError{HTTPStatus: resp.StatusCode, Body: string(respBytes)}
|
||||
}
|
||||
|
||||
var base BaseResponse
|
||||
if err := json.Unmarshal(respBytes, &base); err != nil {
|
||||
return nil, &SDKError{Op: "doRequest", Err: fmt.Errorf("unmarshal base response: %w", err)}
|
||||
}
|
||||
if base.Code != 0 && base.Code != 200 {
|
||||
return nil, &APIError{Code: base.Code, Msg: base.Msg}
|
||||
}
|
||||
return base.Data, nil
|
||||
}
|
||||
|
||||
// InvoiceOrder 提交订单开票请求。
|
||||
func (c *Client) InvoiceOrder(ctx context.Context, req *InvoiceOrderRequest) (*InvoiceOrderData, error) {
|
||||
data, err := c.doRequest(ctx, c.invoiceOrderPath, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var out InvoiceOrderData
|
||||
if len(data) > 0 {
|
||||
if err := json.Unmarshal(data, &out); err != nil {
|
||||
return nil, &SDKError{Op: "InvoiceOrder", Err: fmt.Errorf("unmarshal data: %w", err)}
|
||||
}
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
// QueryInvoiceStatus 查询订单开票状态。
|
||||
func (c *Client) QueryInvoiceStatus(ctx context.Context, req *QueryInvoiceStatusRequest) (*QueryInvoiceStatusData, error) {
|
||||
data, err := c.doRequest(ctx, c.queryInvoiceStatusPath, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var out QueryInvoiceStatusData
|
||||
if len(data) > 0 {
|
||||
if err := json.Unmarshal(data, &out); err != nil {
|
||||
return nil, &SDKError{Op: "QueryInvoiceStatus", Err: fmt.Errorf("unmarshal data: %w", err)}
|
||||
}
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
// CreatePayment 创建付款单据。
|
||||
func (c *Client) CreatePayment(ctx context.Context, req *CreatePaymentRequest) (*CreatePaymentData, error) {
|
||||
data, err := c.doRequest(ctx, c.createPaymentPath, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var out CreatePaymentData
|
||||
if len(data) > 0 {
|
||||
if err := json.Unmarshal(data, &out); err != nil {
|
||||
return nil, &SDKError{Op: "CreatePayment", Err: fmt.Errorf("unmarshal data: %w", err)}
|
||||
}
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
// QueryPaymentStatus 查询支付状态。
|
||||
// 响应数据与支付完成通知的数据结构一致。
|
||||
func (c *Client) QueryPaymentStatus(ctx context.Context, req *QueryPaymentStatusRequest) (*PaymentInfo, error) {
|
||||
data, err := c.doRequest(ctx, c.queryPaymentStatusPath, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var out PaymentInfo
|
||||
if len(data) > 0 {
|
||||
if err := json.Unmarshal(data, &out); err != nil {
|
||||
return nil, &SDKError{Op: "QueryPaymentStatus", Err: fmt.Errorf("unmarshal data: %w", err)}
|
||||
}
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
// ParsePaymentNotify 解析支付完成通知请求体。
|
||||
// 该接口由客户提供、平台调用，用于在支付完成后接收平台回调。
|
||||
func ParsePaymentNotify(body []byte) (*PaymentInfo, error) {
|
||||
var info PaymentInfo
|
||||
if err := json.Unmarshal(body, &info); err != nil {
|
||||
return nil, &SDKError{Op: "ParsePaymentNotify", Err: fmt.Errorf("unmarshal notify: %w", err)}
|
||||
}
|
||||
return &info, nil
|
||||
}
|
||||
|
||||
// ParseNotify 解析通用通知请求体（bizType / bizId / data）。
|
||||
func ParseNotify(body []byte) (*NotifyRequest, error) {
|
||||
var req NotifyRequest
|
||||
if err := json.Unmarshal(body, &req); err != nil {
|
||||
return nil, &SDKError{Op: "ParseNotify", Err: fmt.Errorf("unmarshal notify: %w", err)}
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
Loading…
Reference in New Issue