添加文件: intelligence_finance_v1_ga/client.go
This commit is contained in:
parent
61cc7b10a7
commit
efdea027f8
|
|
@ -0,0 +1,351 @@
|
|||
package intelligence_finance_v1_ga
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 供应商业务类型常量。
|
||||
const (
|
||||
BizTypeCreateSupplier = "create_supplier"
|
||||
BizTypeQuerySupplierByName = "query_supplier_by_name"
|
||||
BizTypeUpdateSupplier = "update_supplier"
|
||||
BizTypeQuerySupplierByUserDefine = "query_supplier_by_userDefineCode"
|
||||
)
|
||||
|
||||
// Client 业财连接接口客户端。
|
||||
type Client struct {
|
||||
baseURL string
|
||||
tenantID string
|
||||
clientID string
|
||||
clientSecret string
|
||||
signEnabled bool
|
||||
httpClient *http.Client
|
||||
|
||||
// 各接口路径（可通过 Option 覆盖）
|
||||
pathSubmitInvoiceApply string
|
||||
pathQueryInvoiceStatus string
|
||||
pathCreatePayment string
|
||||
pathQueryPaymentStatus string
|
||||
pathCreateSupplier string
|
||||
pathQuerySupplierByName string
|
||||
pathUpdateSupplier string
|
||||
pathQuerySupplierByUserDefineCode string
|
||||
}
|
||||
|
||||
// Option 客户端配置项。
|
||||
type Option func(*Client)
|
||||
|
||||
// WithHTTPClient 自定义 http.Client。
|
||||
func WithHTTPClient(c *http.Client) Option {
|
||||
return func(cli *Client) { cli.httpClient = c }
|
||||
}
|
||||
|
||||
// WithSignEnabled 是否启用请求签名（钉钉AI表格场景可关闭，签名由AI表格自动完成）。
|
||||
func WithSignEnabled(enabled bool) Option {
|
||||
return func(cli *Client) { cli.signEnabled = enabled }
|
||||
}
|
||||
|
||||
// WithPathSubmitInvoiceApply 设置提交订单开票申请接口路径。
|
||||
func WithPathSubmitInvoiceApply(path string) Option {
|
||||
return func(cli *Client) { cli.pathSubmitInvoiceApply = path }
|
||||
}
|
||||
|
||||
// WithPathQueryInvoiceStatus 设置开票状态查询接口路径。
|
||||
func WithPathQueryInvoiceStatus(path string) Option {
|
||||
return func(cli *Client) { cli.pathQueryInvoiceStatus = path }
|
||||
}
|
||||
|
||||
// WithPathCreatePayment 设置创建付款单据接口路径。
|
||||
func WithPathCreatePayment(path string) Option {
|
||||
return func(cli *Client) { cli.pathCreatePayment = path }
|
||||
}
|
||||
|
||||
// WithPathQueryPaymentStatus 设置支付状态查询接口路径。
|
||||
func WithPathQueryPaymentStatus(path string) Option {
|
||||
return func(cli *Client) { cli.pathQueryPaymentStatus = path }
|
||||
}
|
||||
|
||||
// WithPathCreateSupplier 设置创建供应商接口路径。
|
||||
func WithPathCreateSupplier(path string) Option {
|
||||
return func(cli *Client) { cli.pathCreateSupplier = path }
|
||||
}
|
||||
|
||||
// WithPathQuerySupplierByName 设置根据名称查询供应商接口路径。
|
||||
func WithPathQuerySupplierByName(path string) Option {
|
||||
return func(cli *Client) { cli.pathQuerySupplierByName = path }
|
||||
}
|
||||
|
||||
// WithPathUpdateSupplier 设置更新供应商接口路径。
|
||||
func WithPathUpdateSupplier(path string) Option {
|
||||
return func(cli *Client) { cli.pathUpdateSupplier = path }
|
||||
}
|
||||
|
||||
// WithPathQuerySupplierByUserDefineCode 设置根据用户自定义编码查询供应商接口路径。
|
||||
func WithPathQuerySupplierByUserDefineCode(path string) Option {
|
||||
return func(cli *Client) { cli.pathQuerySupplierByUserDefineCode = path }
|
||||
}
|
||||
|
||||
// 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,
|
||||
signEnabled: true,
|
||||
httpClient: &http.Client{Timeout: 30 * time.Second},
|
||||
|
||||
pathSubmitInvoiceApply: "/invoice/apply",
|
||||
pathQueryInvoiceStatus: "/invoice/status",
|
||||
pathCreatePayment: "/payment/create",
|
||||
pathQueryPaymentStatus: "/payment/status",
|
||||
pathCreateSupplier: "/supplier/create",
|
||||
pathQuerySupplierByName: "/supplier/queryByName",
|
||||
pathUpdateSupplier: "/supplier/update",
|
||||
pathQuerySupplierByUserDefineCode: "/supplier/queryByUserDefineCode",
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(c)
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
// doRequest 发送 POST 请求并解析响应。
|
||||
func (c *Client) doRequest(ctx context.Context, path string, reqBody, respBody interface{}) error {
|
||||
body, err := json.Marshal(reqBody)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal request: %w", err)
|
||||
}
|
||||
|
||||
httpReq, err := http.NewRequestWithContext(ctx, http.MethodPost, c.baseURL+path, bytes.NewReader(body))
|
||||
if err != nil {
|
||||
return fmt.Errorf("new request: %w", err)
|
||||
}
|
||||
|
||||
httpReq.Header.Set("Content-Type", "application/json")
|
||||
httpReq.Header.Set("tenant-id", c.tenantID)
|
||||
httpReq.Header.Set("client-id", c.clientID)
|
||||
|
||||
if c.signEnabled {
|
||||
timestamp := GenerateTimestamp()
|
||||
nonce, err := GenerateNonce(32)
|
||||
if err != nil {
|
||||
return fmt.Errorf("generate nonce: %w", err)
|
||||
}
|
||||
sign := HmacSHA256Base64(c.clientSecret, timestamp+nonce)
|
||||
httpReq.Header.Set("x-bfl-signature-timestamp", timestamp)
|
||||
httpReq.Header.Set("x-bfl-signature-nonce", nonce)
|
||||
httpReq.Header.Set("x-bfl-signature", sign)
|
||||
}
|
||||
|
||||
httpResp, err := c.httpClient.Do(httpReq)
|
||||
if err != nil {
|
||||
return fmt.Errorf("do request: %w", err)
|
||||
}
|
||||
defer httpResp.Body.Close()
|
||||
|
||||
respData, err := io.ReadAll(httpResp.Body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read response: %w", err)
|
||||
}
|
||||
|
||||
if httpResp.StatusCode != http.StatusOK {
|
||||
return &APIError{StatusCode: httpResp.StatusCode, Body: string(respData)}
|
||||
}
|
||||
|
||||
if err := json.Unmarshal(respData, respBody); err != nil {
|
||||
return fmt.Errorf("unmarshal response: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// checkBusinessCode 校验通用响应结构中的业务 code。
|
||||
func checkBusinessCode(code int, msg string) error {
|
||||
if code != 0 && code != 200 {
|
||||
return &BusinessError{Code: code, Msg: msg}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// SubmitInvoiceApply 提交订单开票申请（接口1）。
|
||||
func (c *Client) SubmitInvoiceApply(ctx context.Context, req *InvoiceApplyRequest) (*InvoiceApplyResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *InvoiceApplyResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathSubmitInvoiceApply, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// QueryInvoiceStatus 查询订单开票状态（接口2）。
|
||||
func (c *Client) QueryInvoiceStatus(ctx context.Context, req *InvoiceStatusQueryRequest) (*InvoiceStatusQueryResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *InvoiceStatusQueryResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathQueryInvoiceStatus, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// CreatePayment 创建付款单据（接口3）。
|
||||
func (c *Client) CreatePayment(ctx context.Context, req *CreatePaymentRequest) (*CreatePaymentResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *CreatePaymentResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathCreatePayment, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// QueryPaymentStatus 查询支付状态（接口5），响应数据与支付通知一致。
|
||||
func (c *Client) QueryPaymentStatus(ctx context.Context, req *PaymentStatusQueryRequest) (*PaymentNotifyData, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *PaymentNotifyData `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathQueryPaymentStatus, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// CreateSupplier 创建供应商（接口6）。
|
||||
func (c *Client) CreateSupplier(ctx context.Context, req *CreateSupplierRequest) (*CreateSupplierResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *CreateSupplierResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathCreateSupplier, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// QuerySupplierByName 根据名称查询供应商（接口7）。
|
||||
func (c *Client) QuerySupplierByName(ctx context.Context, req *QuerySupplierByNameRequest) (*QuerySupplierByNameResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *QuerySupplierByNameResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathQuerySupplierByName, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// UpdateSupplier 更新供应商（接口8）。
|
||||
func (c *Client) UpdateSupplier(ctx context.Context, req *UpdateSupplierRequest) (*UpdateSupplierResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *UpdateSupplierResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathUpdateSupplier, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// QuerySupplierByUserDefineCode 根据用户自定义编码查询供应商（接口9）。
|
||||
func (c *Client) QuerySupplierByUserDefineCode(ctx context.Context, req *QuerySupplierByUserDefineCodeRequest) (*QuerySupplierByUserDefineCodeResponse, error) {
|
||||
var envelope struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data *QuerySupplierByUserDefineCodeResponse `json:"data"`
|
||||
}
|
||||
if err := c.doRequest(ctx, c.pathQuerySupplierByUserDefineCode, req, &envelope); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := checkBusinessCode(envelope.Code, envelope.Msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return envelope.Data, nil
|
||||
}
|
||||
|
||||
// ==================== 通知机制辅助 ====================
|
||||
|
||||
// ParseNotification 解析通用通知数据。
|
||||
func ParseNotification(body []byte) (*Notification, error) {
|
||||
var n Notification
|
||||
if err := json.Unmarshal(body, &n); err != nil {
|
||||
return nil, fmt.Errorf("parse notification: %w", err)
|
||||
}
|
||||
return &n, nil
|
||||
}
|
||||
|
||||
// ParsePaymentNotify 解析支付完成通知数据（接口4）。
|
||||
func ParsePaymentNotify(body []byte) (*PaymentNotifyData, error) {
|
||||
var n PaymentNotifyData
|
||||
if err := json.Unmarshal(body, &n); err != nil {
|
||||
return nil, fmt.Errorf("parse payment notify: %w", err)
|
||||
}
|
||||
return &n, nil
|
||||
}
|
||||
|
||||
// VerifyNotificationSignature 校验通知签名。
|
||||
//
|
||||
// 平台使用客户提供的验签密钥对原始请求体进行 HmacSHA256 签名并 Base64 编码，
|
||||
// 通过请求头 x-bfl-signature 传递。verifyKey 为客户在平台配置的验签密钥。
|
||||
func VerifyNotificationSignature(verifyKey string, rawBody []byte, signature string) (bool, error) {
|
||||
if signature == "" {
|
||||
return false, &NotificationError{Reason: "empty signature"}
|
||||
}
|
||||
expected := HmacSHA256Base64(verifyKey, string(rawBody))
|
||||
return hmacEqual(expected, signature), nil
|
||||
}
|
||||
|
||||
// hmacEqual 常量时间比较两个 Base64 签名是否一致。
|
||||
func hmacEqual(a, b string) bool {
|
||||
if len(a) != len(b) {
|
||||
return false
|
||||
}
|
||||
var v byte
|
||||
for i := 0; i < len(a); i++ {
|
||||
v |= a[i] ^ b[i]
|
||||
}
|
||||
return v == 0
|
||||
}
|
||||
Loading…
Reference in New Issue