package wechat

import (
	"fmt"
	"strings"

	"wechatbot/config"
	"wechatbot/openai"
	"wechatbot/utils"

	"github.com/eatmoreapple/openwechat"
	log "github.com/sirupsen/logrus"
)

var _ MessageHandlerInterface = (*GroupMessageHandler)(nil)

type GroupMessageHandler struct {
}

func (gmh *GroupMessageHandler) handle(msg *openwechat.Message) error {
	if !msg.IsText() {
		return nil
	}

	return gmh.ReplyText(msg)
}

func NewGroupMessageHandler() MessageHandlerInterface {
	return &GroupMessageHandler{}
}

func (gmh *GroupMessageHandler) ReplyText(msg *openwechat.Message) error {
	sender, err := msg.Sender()
	group := openwechat.Group{User: sender}
	log.Printf("Received Group %v Text Msg : %v", group.NickName, msg.Content)

	wechat := config.GetWechatKeyword()
	requestText := msg.Content
	if wechat != nil {
		content, key := utils.ContainsI(requestText, *wechat)
		if len(key) == 0 {
			return nil
		}

		splitItems := strings.Split(content, key)
		if len(splitItems) < 2 {
			return nil
		}

		requestText = strings.TrimSpace(splitItems[1])
	}

	log.Println("问题：", requestText)
	reply, err := openai.Completions(requestText)
	if err != nil {
		log.Println(err)
		if reply != nil {
			result := *reply
			// 如果文字超过4000个字会回错，截取前4000个文字进行回复
			if len(result) > 4000 {
				_, err = msg.ReplyText(result[:4000])
				if err != nil {
					log.Println("回复出错：", err.Error())
					return err
				}
			}
		}

		text, err := msg.ReplyText(fmt.Sprintf("bot error: %s", err.Error()))
		log.Println(text)
		return err
	}

	// 如果在提问的时候没有包含？,AI会自动在开头补充个？看起来很奇怪
	result := *reply
	if strings.HasPrefix(result, "?") {
		result = strings.Replace(result, "?", "", -1)
	}

	if strings.HasPrefix(result, "？") {
		result = strings.Replace(result, "？", "", -1)
	}

	// 微信不支持markdown格式，所以把反引号直接去掉
	if strings.Contains(result, "`") {
		result = strings.Replace(result, "`", "", -1)
	}

	if reply != nil {
		_, err = msg.ReplyText(*reply)
		if err != nil {
			log.Println(err)
		}
		return err
	}

	return nil
}