package headmaster

import (
	"context"
	"fmt"
	"strings"
	"time"

	"git.lxtend.com/qqbot/config"
	"git.lxtend.com/qqbot/constants"
	"git.lxtend.com/qqbot/handler"
	"git.lxtend.com/qqbot/model"
	"git.lxtend.com/qqbot/util"
	"github.com/sashabaranov/go-openai"
)

var (
	histories      = make(map[string][]openai.ChatCompletionMessage)
	histories_time = make(map[string]time.Time)
)

func init() {
	handler.RegisterHandler("校长", headmasterHandler, constants.LEVEL_USER)
	handler.RegisterPrivateHandler(headmasterHandler)
}

func headmasterHandler(msg model.Message) (reply model.Reply) {
	var from string
	if msg.GroupInfo.IsGroupMsg {
		from = util.From(msg.GroupInfo.GroupId, 0)
	} else {
		from = util.From(0, msg.UserId)
	}
	if len(msg.RawMsg) > 7 && msg.RawMsg[0:7] == "校长 " {
		return model.Reply{
			ReplyMsg:       ask(from, fmt.Sprintf("{\"qqid\":%d,\"group_nickname\":\"%s\",\"nickname\":\"%s\",\"referid\":%d,\"received_time\":%s,\"msg\":\"%s\"}", msg.UserId, msg.GroupInfo.UserCard, msg.UserNickName, msg.OriginMsgId, time.Now().Format("2006-01-02 15:04:05"), msg.RawMsg[7:])),
			ReferOriginMsg: false,
			FromMsg:        msg,
		}
	}
	return model.Reply{
		ReplyMsg:       ask(from, fmt.Sprintf("{\"qqid\":%d,\"group_nickname\":\"%s\",\"nickname\":\"%s\",\"referid\":%d,\"received_time\":%s,\"msg\":\"%s\"}", msg.UserId, msg.GroupInfo.UserCard, msg.UserNickName, msg.OriginMsgId, time.Now().Format("2006-01-02 15:04:05"), msg.RawMsg)),
		ReferOriginMsg: false,
		FromMsg:        msg,
	}
}

func ask(from string, question string) (reply string) {
	llmConfig := openai.DefaultAzureConfig(config.ConfigManager.GetProperty("openai_api_key"), config.ConfigManager.GetProperty("openai_api_base_url"))
	llmConfig.APIType = openai.APITypeOpenAI
	llmConfig.APIVersion = ""
	client := openai.NewClientWithConfig(llmConfig)
	resp, err := client.CreateChatCompletion(
		context.Background(),
		openai.ChatCompletionRequest{
			Model:    config.ConfigManager.GetProperty("openai_model_name"),
			Messages: GenRequestFromUsr(from, question),
			// Tools: []openai.Tool{
			// 	{
			// 		Type: openai.ToolTypeFunction,
			// 		Function: &openai.FunctionDefinition{
			// 			Name:        "try_get_qqid_by_nickname",
			// 			Description: "尝试在群内获取昵称对应的qq号",
			// 			Parameters:  json.RawMessage(`{"type": "object", "properties": {"nickname": {"type": "string", "description": "用户的昵称"}}}`),
			// 		},
			// 	},
			// },
			// ToolChoice: &openai.ToolChoice{
			// 	Type: openai.ToolTypeFunction,
			// 	Function: openai.ToolFunction{
			// 		Name: "try_get_qqid_by_nickname",
			// 	},
			// },
		},
	)

	if err != nil {
		fmt.Printf("ChatCompletion error: %v\n", err)
		return
	}
	AppendReplyToHistory(from, resp.Choices[0].Message.Content)
	return enterFormatter(resp.Choices[0].Message.Content)
}

func enterFormatter(msgIn string) string {
	return strings.ReplaceAll(msgIn, "\\n", "\n")
}

func GenRequestFromUsr(from string, question string) []openai.ChatCompletionMessage {
	if _, ok := histories[from]; !ok || histories_time[from].Add(10*time.Minute).Before(time.Now()) {
		histories[from] = make([]openai.ChatCompletionMessage, 0)
		histories[from] = append(histories[from], openai.ChatCompletionMessage{
			Role:    openai.ChatMessageRoleSystem,
			Content: config.ConfigManager.GetProperty("openai_prompt"),
		},
		)
	}
	histories[from] = append(histories[from], openai.ChatCompletionMessage{
		Role:    openai.ChatMessageRoleUser,
		Content: question,
	})
	return histories[from]
}

func AppendReplyToHistory(from string, reply string) {
	histories[from] = append(histories[from], openai.ChatCompletionMessage{
		Role:    openai.ChatMessageRoleAssistant,
		Content: reply,
	})
	histories_time[from] = time.Now()
	for len(histories[from]) > 20 {
		histories[from] = histories[from][1:]
		histories[from][0] = openai.ChatCompletionMessage{
			Role:    openai.ChatMessageRoleSystem,
			Content: config.ConfigManager.GetProperty("openai_prompt"),
		}
	}
}