package api

import (
	"chat/types"
	"chat/utils"
	"chat/web"
	"fmt"
	"math/rand"
	"strings"
	"time"
)

func ChatWithWeb(message []types.ChatGPTMessage, long bool) (string, []types.ChatGPTMessage) {
	keyword := strings.TrimSpace(SearchWeb(message))

	if len(keyword) == 0 {
		return keyword, message
	}

	data := web.SearchBing(keyword)
	if long {
		data = utils.GetSegmentString(data, 6000)
	} else {
		data = utils.GetSegmentString(data, 3000)
	}
	return keyword, utils.Insert(message, 0, types.ChatGPTMessage{
		Role: "system",
		Content: fmt.Sprintf("你将扮演AI问答助手，你的知识库不是截止于2021年9月，而是可以实时联网的，你可以提供实时联网的知识库。"+
			"网站使用markdown链接包裹，多使用emoji和颜文字，给予用户精确的答复。"+
			"公式使用 $$ [公式] $$ 包裹，使用latex语法，例如：$$ \\frac{1}{2} $$。"+
			"当前时间: %s, 实时联网搜索结果：%s",
			time.Now().Format("2006-01-02 15:04:05"), data,
		),
	})
}

func GetRandomKey(apikey string) string {
	arr := strings.Split(apikey, "|")
	idx := rand.Intn(len(arr))
	return arr[idx]
}

func StringCleaner(content string) string {
	for _, replacer := range []string{",", "、", "，", "。", "：", ":", "；", ";", "！", "!", "=", "？", "?", "（", "）", "(", ")", "关键字", "空", "1+1"} {
		content = strings.ReplaceAll(content, replacer, " ")
	}
	return strings.TrimSpace(content)
}

func SearchWeb(message []types.ChatGPTMessage) string {
	resp, _ := GetChatGPTResponse([]types.ChatGPTMessage{{
		Role:    "system",
		Content: "If the user input content require ONLINE SEARCH to get the results, please output these keywords to refine the data Interval with space, remember not to answer other content, json format return, format {\"keyword\": \"...\" }",
	}, {
		Role:    "user",
		Content: "你是谁",
	}, {
		Role:    "assistant",
		Content: "{\"keyword\":\"\"}",
	}, {
		Role:    "user",
		Content: "那fystart起始页是什么 和深能科创有什么关系",
	}, {
		Role:    "assistant",
		Content: "{\"keyword\":\"fystart起始页 深能科创 关系\"}",
	}, {
		Role:    "user",
		Content: "1+1=?",
	}, {
		Role:    "assistant",
		Content: "{\"keyword\":\"\"}",
	}, {
		Role:    "user",
		Content: "?",
	}, {
		Role:    "assistant",
		Content: "{\"keyword\":\"\"}",
	}, {
		Role:    "user",
		Content: message[len(message)-1].Content,
	}}, 40)
	keyword := utils.UnmarshalJson[map[string]interface{}](resp)
	if keyword == nil {
		return ""
	}
	return StringCleaner(keyword["keyword"].(string))
}
