126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
package tools
|
|
|
|
import (
|
|
"ai_scheduler/internal/config"
|
|
"ai_scheduler/internal/entitys"
|
|
"ai_scheduler/internal/pkg/l_request"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
type ZltxOrderAfterSalePreCheckTool struct {
|
|
config config.ToolConfig
|
|
}
|
|
|
|
// NewZltxOrderAfterSalePreCheckTool 创建售后订单预检工具
|
|
func NewZltxOrderAfterSalePreCheckTool(config config.ToolConfig) *ZltxOrderAfterSalePreCheckTool {
|
|
return &ZltxOrderAfterSalePreCheckTool{config: config}
|
|
}
|
|
|
|
// Name 返回工具名称
|
|
func (t *ZltxOrderAfterSalePreCheckTool) Name() string {
|
|
return "zltxOrderAfterSalePreCheck"
|
|
}
|
|
|
|
func (t *ZltxOrderAfterSalePreCheckTool) Description() string {
|
|
return "直连天下售后订单预检工具"
|
|
}
|
|
|
|
func (t *ZltxOrderAfterSalePreCheckTool) Definition() entitys.ToolDefinition {
|
|
return entitys.ToolDefinition{
|
|
Type: "function",
|
|
Function: entitys.FunctionDef{
|
|
Name: t.Name(),
|
|
Description: t.Description(),
|
|
Parameters: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"orderType": map[string]interface{}{
|
|
"type": "integer",
|
|
"description": "售后订单类型",
|
|
},
|
|
"orderNumber": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "售后订单号",
|
|
},
|
|
},
|
|
"required": []string{"orderType", "orderNumber"},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
type ZltxOrderAfterSalePreCheckRequest struct {
|
|
OrderType int `json:"orderType"`
|
|
OrderNumber string `json:"orderNumber"`
|
|
}
|
|
|
|
type ZltxOrderAfterSalePreCheckResponse struct {
|
|
Code int `json:"code"`
|
|
Msg string `json:"msg"`
|
|
Data struct {
|
|
CheckResult bool `json:"checkResult"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
type CheckResult struct {
|
|
OrderType int `json:"orderType"`
|
|
OrderNumber string `json:"orderNumber"`
|
|
OrderAmount float64 `json:"orderAmount"`
|
|
OrderPrice float64 `json:"orderPrice"`
|
|
SignCompany int `json:"signCompany"`
|
|
OrderQuantity int `json:"orderQuantity"`
|
|
ResellerID int `json:"resellerId"`
|
|
ResellerName string `json:"resellerName"`
|
|
OurProductID int `json:"ourProductId"`
|
|
OurProductTitle string `json:"ourProductTitle"`
|
|
Account []string `json:"account"`
|
|
Platforms struct {
|
|
Num4 string `json:"4"`
|
|
} `json:"platforms"`
|
|
}
|
|
|
|
func (t *ZltxOrderAfterSalePreCheckTool) Execute(ctx context.Context, requireData *entitys.RequireData) error {
|
|
var req ZltxOrderAfterSalePreCheckRequest
|
|
if err := json.Unmarshal([]byte(requireData.Match.Parameters), &req); err != nil {
|
|
return err
|
|
}
|
|
if req.OrderType == 0 || req.OrderNumber == "" {
|
|
return fmt.Errorf("orderType and orderNumber are required")
|
|
}
|
|
return t.checkZltxOrderAfterSalePreCheck(req.OrderType, req.OrderNumber, requireData)
|
|
}
|
|
|
|
func (t *ZltxOrderAfterSalePreCheckTool) checkZltxOrderAfterSalePreCheck(orderType int, orderNumber string, requireData *entitys.RequireData) error {
|
|
req := l_request.Request{
|
|
Url: t.config.BaseURL,
|
|
Headers: map[string]string{
|
|
"Authorization": fmt.Sprintf("Bearer %s", requireData.Auth),
|
|
},
|
|
Method: "POST",
|
|
Data: map[string]string{
|
|
"orderType": fmt.Sprintf("%d", orderType),
|
|
"orderNumber": orderNumber,
|
|
},
|
|
}
|
|
res, err := req.Send()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 解析响应
|
|
var resp ZltxOrderAfterSalePreCheckResponse
|
|
if err := json.Unmarshal(res.Content, &resp); err != nil {
|
|
return err
|
|
}
|
|
if resp.Code != 0 {
|
|
return fmt.Errorf("check failed: %s", resp.Msg)
|
|
}
|
|
jsonByte, err := json.Marshal(resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
entitys.ResJson(requireData.Ch, t.Name(), string(jsonByte))
|
|
return nil
|
|
}
|