package services

import (
	"encoding/json"
	"regexp"
	"strings"
)

// parseFramePromptJSON 解析AI返回的JSON格式提示词
func (s *FramePromptService) parseFramePromptJSON(aiResponse string) *SingleFramePrompt {
	// 清理可能的markdown代码块标记
	cleaned := strings.TrimSpace(aiResponse)

	// 移除 ```json 和 ``` 标记
	re := regexp.MustCompile("(?s)```json\\s*(.+?)\\s*```")
	if matches := re.FindStringSubmatch(cleaned); len(matches) > 1 {
		cleaned = strings.TrimSpace(matches[1])
	} else {
		// 移除单独的 ``` 标记
		cleaned = strings.Trim(cleaned, "`")
		cleaned = strings.TrimSpace(cleaned)
	}

	// 尝试解析JSON
	var result SingleFramePrompt
	if err := json.Unmarshal([]byte(cleaned), &result); err != nil {
		s.log.Warnw("Failed to parse JSON", "error", err, "cleaned_response", cleaned)
		return nil
	}

	// 验证必需字段
	if result.Prompt == "" {
		s.log.Warnw("Parsed JSON missing prompt field", "response", cleaned)
		return nil
	}

	return &result
}