package router

import (
	"bytes"
	"net/http"
	"strings"

	"opencatd-open/pkg/claude"
	"opencatd-open/pkg/google"
	"opencatd-open/pkg/openai"
	"opencatd-open/pkg/search"

	"github.com/gin-gonic/gin"
)

func ChatHandler(c *gin.Context) {
	var chatreq openai.ChatCompletionRequest
	if err := c.ShouldBindJSON(&chatreq); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	if chatreq.Messages[len(chatreq.Messages)-1].Role == "user" {
		result, err := search.BingSearch(search.SearchParams{Query: string(chatreq.Messages[len(chatreq.Messages)-1].Content)})
		if err == nil {
			var msgs []openai.ChatCompletionMessage
			for _, m := range chatreq.Messages {
				var buf bytes.Buffer
				buf.WriteString("根据我提问的语言回答我,我将提供一些从搜索引擎获取的信息。你自行判断是否使用搜索引擎获取的内容。不要原封不动照抄,根据你自己的知识库提炼信息之后回答我\n\n")
				if m.Role == "system" {
					buf.Write(m.Content)
					msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: buf.Bytes()})
				} else {
					msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: buf.Bytes()})
				}
				msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: m.Content})
			}
			msgs = append(msgs, openai.ChatCompletionMessage{Role: "tool", Content: []byte(result.(string))})
			chatreq.Messages = msgs
		}
	}

	if strings.HasPrefix(chatreq.Model, "gpt") {
		openai.ChatProxy(c, &chatreq)
		return
	}

	if strings.HasPrefix(chatreq.Model, "claude") {
		claude.ChatProxy(c, &chatreq)
		return
	}

	if strings.HasPrefix(chatreq.Model, "gemini") {
		google.ChatProxy(c, &chatreq)
		return
	}
}