287 lines
7.9 KiB
Go
287 lines
7.9 KiB
Go
|
package component_report
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"regexp"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
|
|||
|
|
"tyapi-server/internal/domains/product/entities"
|
|||
|
|
"tyapi-server/internal/domains/product/repositories"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ExampleJSONGenerator 示例JSON生成器
|
|||
|
|
type ExampleJSONGenerator struct {
|
|||
|
|
productRepo repositories.ProductRepository
|
|||
|
|
docRepo repositories.ProductDocumentationRepository
|
|||
|
|
apiConfigRepo repositories.ProductApiConfigRepository
|
|||
|
|
logger *zap.Logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewExampleJSONGenerator 创建示例JSON生成器
|
|||
|
|
func NewExampleJSONGenerator(
|
|||
|
|
productRepo repositories.ProductRepository,
|
|||
|
|
docRepo repositories.ProductDocumentationRepository,
|
|||
|
|
apiConfigRepo repositories.ProductApiConfigRepository,
|
|||
|
|
logger *zap.Logger,
|
|||
|
|
) *ExampleJSONGenerator {
|
|||
|
|
return &ExampleJSONGenerator{
|
|||
|
|
productRepo: productRepo,
|
|||
|
|
docRepo: docRepo,
|
|||
|
|
apiConfigRepo: apiConfigRepo,
|
|||
|
|
logger: logger,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ExampleJSONItem example.json 中的单个项
|
|||
|
|
type ExampleJSONItem struct {
|
|||
|
|
Feature struct {
|
|||
|
|
FeatureName string `json:"featureName"`
|
|||
|
|
Sort int `json:"sort"`
|
|||
|
|
} `json:"feature"`
|
|||
|
|
Data struct {
|
|||
|
|
APIID string `json:"apiID"`
|
|||
|
|
Data interface{} `json:"data"`
|
|||
|
|
} `json:"data"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateExampleJSON 生成 example.json 文件内容
|
|||
|
|
// productID: 产品ID（可以是组合包或单品）
|
|||
|
|
// subProductCodes: 子产品编号列表（如果为空，则处理所有子产品）
|
|||
|
|
func (g *ExampleJSONGenerator) GenerateExampleJSON(ctx context.Context, productID string, subProductCodes []string) ([]byte, error) {
|
|||
|
|
// 1. 获取产品信息
|
|||
|
|
product, err := g.productRepo.GetByID(ctx, productID)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取产品信息失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 构建 example.json 数组
|
|||
|
|
var examples []ExampleJSONItem
|
|||
|
|
|
|||
|
|
if product.IsPackage {
|
|||
|
|
// 组合包：遍历子产品
|
|||
|
|
packageItems, err := g.productRepo.GetPackageItems(ctx, productID)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取组合包子产品失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for sort, item := range packageItems {
|
|||
|
|
// 如果指定了子产品编号列表，只处理列表中的产品
|
|||
|
|
if len(subProductCodes) > 0 {
|
|||
|
|
found := false
|
|||
|
|
for _, code := range subProductCodes {
|
|||
|
|
if item.Product != nil && item.Product.Code == code {
|
|||
|
|
found = true
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if !found {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取子产品信息
|
|||
|
|
var subProduct entities.Product
|
|||
|
|
if item.Product != nil {
|
|||
|
|
subProduct = *item.Product
|
|||
|
|
} else {
|
|||
|
|
subProduct, err = g.productRepo.GetByID(ctx, item.ProductID)
|
|||
|
|
if err != nil {
|
|||
|
|
g.logger.Warn("获取子产品信息失败",
|
|||
|
|
zap.String("product_id", item.ProductID),
|
|||
|
|
zap.Error(err),
|
|||
|
|
)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取响应示例数据
|
|||
|
|
responseData := g.extractResponseExample(ctx, &subProduct)
|
|||
|
|
|
|||
|
|
// 获取产品名称和编号
|
|||
|
|
productName := subProduct.Name
|
|||
|
|
productCode := subProduct.Code
|
|||
|
|
|
|||
|
|
// 构建示例项
|
|||
|
|
example := ExampleJSONItem{
|
|||
|
|
Feature: struct {
|
|||
|
|
FeatureName string `json:"featureName"`
|
|||
|
|
Sort int `json:"sort"`
|
|||
|
|
}{
|
|||
|
|
FeatureName: productName,
|
|||
|
|
Sort: sort + 1,
|
|||
|
|
},
|
|||
|
|
Data: struct {
|
|||
|
|
APIID string `json:"apiID"`
|
|||
|
|
Data interface{} `json:"data"`
|
|||
|
|
}{
|
|||
|
|
APIID: productCode,
|
|||
|
|
Data: responseData,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
examples = append(examples, example)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 单品
|
|||
|
|
responseData := g.extractResponseExample(ctx, &product)
|
|||
|
|
|
|||
|
|
example := ExampleJSONItem{
|
|||
|
|
Feature: struct {
|
|||
|
|
FeatureName string `json:"featureName"`
|
|||
|
|
Sort int `json:"sort"`
|
|||
|
|
}{
|
|||
|
|
FeatureName: product.Name,
|
|||
|
|
Sort: 1,
|
|||
|
|
},
|
|||
|
|
Data: struct {
|
|||
|
|
APIID string `json:"apiID"`
|
|||
|
|
Data interface{} `json:"data"`
|
|||
|
|
}{
|
|||
|
|
APIID: product.Code,
|
|||
|
|
Data: responseData,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
examples = append(examples, example)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 序列化为JSON
|
|||
|
|
jsonData, err := json.MarshalIndent(examples, "", " ")
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("序列化example.json失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return jsonData, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MatchProductCodeToPath 根据产品编码匹配 UI 组件路径，返回路径和类型（folder/file）
|
|||
|
|
func (g *ExampleJSONGenerator) MatchProductCodeToPath(ctx context.Context, productCode string) (string, string, error) {
|
|||
|
|
basePath := filepath.Join("resources", "Pure Component", "src", "ui")
|
|||
|
|
|
|||
|
|
entries, err := os.ReadDir(basePath)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", "", fmt.Errorf("读取组件目录失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, entry := range entries {
|
|||
|
|
name := entry.Name()
|
|||
|
|
|
|||
|
|
// 精确匹配
|
|||
|
|
if name == productCode {
|
|||
|
|
path := filepath.Join(basePath, name)
|
|||
|
|
fileType := "folder"
|
|||
|
|
if !entry.IsDir() {
|
|||
|
|
fileType = "file"
|
|||
|
|
}
|
|||
|
|
return path, fileType, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模糊匹配：文件夹名称包含产品编号，或产品编号包含文件夹名称的核心部分
|
|||
|
|
if strings.Contains(name, productCode) || strings.Contains(productCode, extractCoreCode(name)) {
|
|||
|
|
path := filepath.Join(basePath, name)
|
|||
|
|
fileType := "folder"
|
|||
|
|
if !entry.IsDir() {
|
|||
|
|
fileType = "file"
|
|||
|
|
}
|
|||
|
|
return path, fileType, nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return "", "", fmt.Errorf("未找到匹配的组件文件: %s", productCode)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// extractCoreCode 提取文件名中的核心编码部分
|
|||
|
|
func extractCoreCode(name string) string {
|
|||
|
|
for i, r := range name {
|
|||
|
|
if (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9') {
|
|||
|
|
return name[i:]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return name
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// extractResponseExample 提取产品响应示例数据（优先级：文档 > API配置 > 默认值）
|
|||
|
|
func (g *ExampleJSONGenerator) extractResponseExample(ctx context.Context, product *entities.Product) interface{} {
|
|||
|
|
var responseData interface{}
|
|||
|
|
|
|||
|
|
// 1. 优先从产品文档中获取
|
|||
|
|
doc, err := g.docRepo.FindByProductID(ctx, product.ID)
|
|||
|
|
if err == nil && doc != nil && doc.ResponseExample != "" {
|
|||
|
|
// 尝试直接解析为JSON
|
|||
|
|
err := json.Unmarshal([]byte(doc.ResponseExample), &responseData)
|
|||
|
|
if err == nil {
|
|||
|
|
g.logger.Debug("从产品文档中提取响应示例成功",
|
|||
|
|
zap.String("product_id", product.ID),
|
|||
|
|
zap.String("product_code", product.Code),
|
|||
|
|
)
|
|||
|
|
return responseData
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果解析失败，尝试从Markdown代码块中提取JSON
|
|||
|
|
extractedData := extractJSONFromMarkdown(doc.ResponseExample)
|
|||
|
|
if extractedData != nil {
|
|||
|
|
g.logger.Debug("从Markdown代码块中提取响应示例成功",
|
|||
|
|
zap.String("product_id", product.ID),
|
|||
|
|
zap.String("product_code", product.Code),
|
|||
|
|
)
|
|||
|
|
return extractedData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 如果文档中没有，尝试从产品API配置中获取
|
|||
|
|
apiConfig, err := g.apiConfigRepo.FindByProductID(ctx, product.ID)
|
|||
|
|
if err == nil && apiConfig != nil && apiConfig.ResponseExample != "" {
|
|||
|
|
// API配置的响应示例通常是 JSON 字符串
|
|||
|
|
err := json.Unmarshal([]byte(apiConfig.ResponseExample), &responseData)
|
|||
|
|
if err == nil {
|
|||
|
|
g.logger.Debug("从产品API配置中提取响应示例成功",
|
|||
|
|
zap.String("product_id", product.ID),
|
|||
|
|
zap.String("product_code", product.Code),
|
|||
|
|
)
|
|||
|
|
return responseData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 如果都没有，返回默认空对象
|
|||
|
|
g.logger.Warn("未找到响应示例数据，使用默认空对象",
|
|||
|
|
zap.String("product_id", product.ID),
|
|||
|
|
zap.String("product_code", product.Code),
|
|||
|
|
)
|
|||
|
|
return map[string]interface{}{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// extractJSONFromMarkdown 从Markdown代码块中提取JSON
|
|||
|
|
func extractJSONFromMarkdown(markdown string) interface{} {
|
|||
|
|
// 查找 ```json 代码块
|
|||
|
|
re := regexp.MustCompile("(?s)```json\\s*(.*?)\\s*```")
|
|||
|
|
matches := re.FindStringSubmatch(markdown)
|
|||
|
|
|
|||
|
|
if len(matches) > 1 {
|
|||
|
|
var jsonData interface{}
|
|||
|
|
err := json.Unmarshal([]byte(matches[1]), &jsonData)
|
|||
|
|
if err == nil {
|
|||
|
|
return jsonData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 也尝试查找 ``` 代码块（可能是其他格式）
|
|||
|
|
re2 := regexp.MustCompile("(?s)```\\s*(.*?)\\s*```")
|
|||
|
|
matches2 := re2.FindStringSubmatch(markdown)
|
|||
|
|
if len(matches2) > 1 {
|
|||
|
|
var jsonData interface{}
|
|||
|
|
err := json.Unmarshal([]byte(matches2[1]), &jsonData)
|
|||
|
|
if err == nil {
|
|||
|
|
return jsonData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果提取失败，返回 nil（由调用者决定默认值）
|
|||
|
|
return nil
|
|||
|
|
}
|