package util

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

// 占位符替换  xxx{placeholder}xxx
func ReplacePlaceholder(str string, placeholder string, value string) string {
	return strings.ReplaceAll(str, "{"+placeholder+"}", value)
}

// 构建跳转a标签，新tab打开
func BuildJumpLink(url string, text string) string {
	return "<a href=\"" + url + "\" target=\"_blank\">" + text + "</a>"
}

func EscapeJSONString(s string) string {
	b, _ := json.Marshal(s)
	return string(b[1 : len(b)-1])
}

// string 转 int
func StringToInt(s string) int {
	i, _ := strconv.Atoi(s)
	return i
}

// string 转 float64
func StringToFloat64(s string) float64 {
	i, _ := strconv.ParseFloat(s, 64)
	return i
}

// 是否包含在数组中
func Contains[T comparable](strings []T, str T) bool {
	for _, s := range strings {
		if s == str {
			return true
		}
	}
	return false
}

// json LLM专用字符串修复
func JSONRepair(input string) (string, error) {
	s := strings.TrimSpace(input)

	s = trimToJSONObject(s)
	s = normalizeQuotes(s)
	s = removeTrailingCommas(s)
	s = quoteObjectKeys(s)
	s = balanceBrackets(s)

	// 最终校验
	var js any
	if err := json.Unmarshal([]byte(s), &js); err != nil {
		return "", fmt.Errorf("json repair failed: %w", err)
	}
	return s, nil
}

// 裁剪前后垃圾文本
func trimToJSONObject(s string) string {
	start := strings.IndexAny(s, "{[")
	end := strings.LastIndexAny(s, "}]")
	if start == -1 || end == -1 || start >= end {
		return s
	}
	return s[start : end+1]
}

// 引号统一
func normalizeQuotes(s string) string {
	// 只替换“看起来像字符串的单引号”
	re := regexp.MustCompile(`'([^']*)'`)
	return re.ReplaceAllString(s, `"$1"`)
}

// 删除尾随逗号
func removeTrailingCommas(s string) string {
	re := regexp.MustCompile(`,(\s*[}\]])`)
	return re.ReplaceAllString(s, `$1`)
}

// 给 object key 自动补双引号
func quoteObjectKeys(s string) string {
	re := regexp.MustCompile(`([{,]\s*)([a-zA-Z0-9_]+)\s*:`)
	return re.ReplaceAllString(s, `$1"$2":`)
}

// 括号补齐
func balanceBrackets(s string) string {
	var stack []rune
	for _, r := range s {
		switch r {
		case '{', '[':
			stack = append(stack, r)
		case '}', ']':
			if len(stack) > 0 {
				stack = stack[:len(stack)-1]
			}
		}
	}
	for i := len(stack) - 1; i >= 0; i-- {
		switch stack[i] {
		case '{':
			s += "}"
		case '[':
			s += "]"
		}
	}
	return s
}