132 lines
3.5 KiB
Go
132 lines
3.5 KiB
Go
package tools
|
|
|
|
import (
|
|
"ai_scheduler/internal/config"
|
|
"ai_scheduler/internal/entitys"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
|
|
"gitea.cdlsxd.cn/self-tools/l_request"
|
|
)
|
|
|
|
type ZltxOrderAfterSaleTool struct {
|
|
config config.ToolConfig
|
|
}
|
|
|
|
func (z ZltxOrderAfterSaleTool) Name() string {
|
|
return "zltxOrderAfterSale"
|
|
}
|
|
|
|
func (z ZltxOrderAfterSaleTool) Description() string {
|
|
return "查询直连天下订单售后信息"
|
|
}
|
|
|
|
func (z ZltxOrderAfterSaleTool) Definition() entitys.ToolDefinition {
|
|
return entitys.ToolDefinition{
|
|
Type: "function",
|
|
Function: entitys.FunctionDef{
|
|
Name: z.Name(),
|
|
Description: z.Description(),
|
|
Parameters: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"number": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "账号或分销商号",
|
|
},
|
|
},
|
|
"required": []string{"number"},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
type ZltxOrderAfterSaleRequest struct {
|
|
Number string `json:"number"`
|
|
}
|
|
|
|
func (z ZltxOrderAfterSaleTool) Execute(ctx context.Context, requireData *entitys.RequireData) error {
|
|
var req ZltxOrderAfterSaleRequest
|
|
if err := json.Unmarshal([]byte(requireData.Match.Parameters), &req); err != nil {
|
|
return err
|
|
}
|
|
if req.Number == "" {
|
|
return fmt.Errorf("number is required")
|
|
}
|
|
return z.getZltxOrderAfterSale(req.Number, requireData)
|
|
}
|
|
|
|
type ZltxOrderAfterSaleResponse struct {
|
|
Code int `json:"code"`
|
|
Data struct {
|
|
List []*ZltxOrderAfterSaleData `json:"list"`
|
|
} `json:"data"`
|
|
Error string `json:"error"`
|
|
}
|
|
|
|
type ZltxOrderAfterSaleData struct {
|
|
// 处理方式 价款/扣款
|
|
HandleMethod string `json:"handle_method"`
|
|
// 售后金额
|
|
AfterSaleAmount float64 `json:"after_sale_amount"`
|
|
// 售后原因
|
|
AfterSaleReason string `json:"after_sale_reason"`
|
|
// 原单信息
|
|
OriginalOrderInfo *OrderInfo
|
|
}
|
|
|
|
type OrderInfo struct {
|
|
SerialNumber string `json:"serial_number"` // 订单流水号
|
|
SupplierName string `json:"supplier_name"` // 供应商名称
|
|
SigningBody string `json:"signing_body"` // 签约主体
|
|
ProductName string `json:"product_name"` // 商品名称
|
|
UpstreamPrice float64 `json:"upstream_price"` // 上游价格
|
|
RechargeAccount string `json:"recharge_account"` // 充值账号
|
|
RechargeStatus string `json:"recharge_status"` // 充值状态
|
|
}
|
|
|
|
func (z ZltxOrderAfterSaleTool) getZltxOrderAfterSale(number string, requireData *entitys.RequireData) error {
|
|
//查询订单详情
|
|
|
|
url := fmt.Sprintf("%s%s", z.config.BaseURL, number)
|
|
req := l_request.Request{
|
|
Url: url,
|
|
Headers: map[string]string{
|
|
"Authorization": fmt.Sprintf("Bearer %s", requireData.Auth),
|
|
},
|
|
Method: "GET",
|
|
}
|
|
res, err := req.Send()
|
|
var resData ZltxOrderAfterSaleResponse
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := json.Unmarshal(res.Content, &resData); err != nil {
|
|
return err
|
|
}
|
|
if resData.Code != 200 {
|
|
return fmt.Errorf("为获取到数据，请检查权限: %s", string(res.Content))
|
|
}
|
|
//按照日期排序
|
|
sort.Slice(resData.Data.RecentThreeDays, func(i, j int) bool {
|
|
return resData.Data.RecentThreeDays[i].Date < resData.Data.RecentThreeDays[j].Date
|
|
})
|
|
sort.Slice(resData.Data.RecentOneMonth, func(i, j int) bool {
|
|
return resData.Data.RecentOneMonth[i].Date < resData.Data.RecentOneMonth[j].Date
|
|
})
|
|
jsonByte, err := json.Marshal(resData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
entitys.ResJson(requireData.Ch, z.Name(), string(jsonByte))
|
|
return nil
|
|
}
|
|
|
|
func NewZltxOrderAfterSaleTool(config config.ToolConfig) *ZltxOrderAfterSaleTool {
|
|
return &ZltxOrderAfterSaleTool{
|
|
config: config,
|
|
}
|
|
}
|