package prompts

import (
	"encoding/json"
	"sdk-generator/internal/entitys"

	"github.com/sashabaranov/go-openai"
)

// BuildRefinePrompt 构建文档精炼的请求
func BuildRefinePrompt(rawDoc string) openai.ChatCompletionRequest {
	return openai.ChatCompletionRequest{
		Messages: []openai.ChatCompletionMessage{
			{
				Role:    openai.ChatMessageRoleSystem,
				Content: `你是一个专业的 API 文档分析专家，擅长从各种格式的文档中提取技术信息。你的输出必须准确、完整，因为后续会基于你的输出生成可运行的代码。`,
			},
			{
				Role:    openai.ChatMessageRoleUser,
				Content: RefinePrompt(rawDoc),
			},
		},
		Temperature:      0.1,
		TopP:             0.9,
		MaxTokens:        65536,
		FrequencyPenalty: 0.0,
		PresencePenalty:  0.0,
		// 使用工具调用来强制结构化输出
		ToolChoice: "required",
		Tools: []openai.Tool{
			{
				Type: openai.ToolTypeFunction,
				Function: &openai.FunctionDefinition{
					Name:        "classify_and_extract_doc",
					Description: "分析API文档，判断文档类型并提取所有技术信息",
					Parameters: map[string]interface{}{
						"type": "object",
						"properties": map[string]interface{}{
							"refined_doc": map[string]interface{}{
								"type":        "string",
								"description": "精炼后的文档内容，保留所有技术信息",
							},
							"interfaces": map[string]interface{}{
								"type":        "array",
								"description": "提取的接口列表摘要（用于快速预览）",
								"items": map[string]interface{}{
									"type": "object",
									"properties": map[string]interface{}{
										"method":  map[string]interface{}{"type": "string"},
										"summary": map[string]interface{}{"type": "string"},
									},
								},
							},
						},
						"required": []string{"interfaces", "refined_doc"},
					},
				},
			},
		},
		Stream: false,
	}
}

// RefinePrompt 文档精炼提示词模板
func RefinePrompt(rawDoc string) string {
	return `分析以下 API 文档，为后续代码生成做准备。



## 第一步：精炼文档内容

### 必须保留的内容（一个不漏）
- ✅ 所有接口（路径、方法、完整URL）
- ✅ 所有参数和字段（类型、必填/可选、取值范围、描述）
- ✅ 所有响应字段（成功/失败）
- ✅ 所有错误码及含义
- ✅ 认证、加密、签名的完整流程（算法、密钥来源、拼接顺序）
- ✅ 所有请求/响应示例
- ✅ 回调/通知机制（如果有）
- ✅ 时间戳、随机数、签名等共性参数

### 可以去掉的内容
- ❌ 公司介绍、平台介绍
- ❌ 接入流程步骤（第一步注册、第二步申请...）
- ❌ 联调建议、FAQ
- ❌ 营销文案、联系方式
- ❌ 重复的示例（保留一个最完整的即可）

## 输出要求
1. **refined_doc**：精炼后的文档，按以下结构组织：
1. **interfaces**：提取的接口列表摘要（用于快速预览）：

### 精炼文档结构
` + "```" + `
## 文档概述
- 接口总数：[N]个

## 认证与安全
[完整的认证/加密/签名流程]

## 接口列表
### 接口 N：[名称]
- 路径：[完整路径]
- 方法：[GET/POST/...]
- 描述：[功能说明]

#### 请求参数
| 参数名 | 类型 | 必填 | 说明 |
|--------|------|------|------|

#### 响应参数
| 参数名 | 类型 | 说明 |
|--------|------|------|

#### 示例
[请求示例/响应示例]

## 错误码
| 错误码 | 说明 | 处理建议 |
|--------|------|----------|

## 回调通知（如有）
[回调机制说明]
` + "```" + `

## 关键原则

1. **完整性优先**：不确定是否重要的信息，一律保留
2. **忠于原文**：类型、字段名、示例值保持原样，不要修改或猜测
3. **不编造**：文档中没有的信息，绝不补充
4. **类型明确**：如果文档没有明确标注"必填"，视为可选

原始文档：
` + rawDoc
}

func ExtractDocTypeFromToolCall(response openai.ChatCompletionResponse) (*entitys.ToolRes, bool) {
	var toolRes entitys.ToolRes
	if len(response.Choices) == 0 {
		return nil, false
	}
	choice := response.Choices[0]
	if choice.Message.ToolCalls == nil || len(choice.Message.ToolCalls) == 0 {
		return nil, false
	}
	if err := json.Unmarshal([]byte(choice.Message.ToolCalls[0].Function.Arguments), &toolRes); err != nil {
		return nil, false
	}
	// 这里需要根据实际的 JSON 解析逻辑实现
	return &toolRes, true
}