package allhelp

import (
	"yunyan/comm"
	"yunyan/pb"
	"yunyan/utils"
	"fmt"
	"strings"
	"time"
)

// SubmitChatSummary 提交每日聊天内容，异步生成AI总结
// @Summary 提交每日聊天总结任务
// @Description 提交某一天的聊天消息列表，异步调用AI生成总结；同一(uid,日期)重复提交会覆盖重算
// @Tags Allhelp
// @Accept json
// @Produce json
// @Security BearerAuth
// @Param user body pb.AllhelpSubmitChatSummaryReq true "请求数据"
// @Success 200 {object} comm.HttpResult{data=pb.AllhelpSubmitChatSummaryResp} "响应数据"
// @Router /api/home/allhelp_submitchatsummary [post]
func (this *apiComp) SubmitChatSummary(session comm.IUserSession, req *pb.AllhelpSubmitChatSummaryReq) (resp *pb.AllhelpSubmitChatSummaryResp, errdata *pb.ErrorData) {
	uid := session.GetUserId()
	if uid == "" {
		errdata = &pb.ErrorData{Code: pb.ErrorCode_ReqParameterError, Message: "用户未登录"}
		return
	}
	if !isValidDate(req.SummaryDate) {
		errdata = &pb.ErrorData{Code: pb.ErrorCode_ReqParameterError, Message: "summary_date 必须为 YYYY-MM-DD"}
		return
	}
	if len(req.Msgs) == 0 {
		errdata = &pb.ErrorData{Code: pb.ErrorCode_ReqParameterError, Message: "聊天内容不能为空"}
		return
	}

	rec := &pb.DBChatSummary{
		Uid:         uid,
		SummaryDate: req.SummaryDate,
		ChatContent: joinChatMsgs(req.Msgs),
		ChatMsgs:    utils.ToString(req.Msgs),
		Summary:     "",
		State:       pb.ChatSummaryState_ChatSummaryState_Pending,
		ErrorMsg:    "",
		FinishTime:  0,
	}
	if err := this.module.model.upsertChatSummary(rec); err != nil {
		this.module.Errorln("SubmitChatSummary upsert error:", err)
		errdata = &pb.ErrorData{Code: pb.ErrorCode_DBError, Message: "入库失败"}
		return
	}
	if err := this.module.summary.Enqueue(rec.Id); err != nil {
		this.module.Errorln("SubmitChatSummary enqueue error:", err)
		errdata = &pb.ErrorData{Code: pb.ErrorCode_DBError, Message: "入队失败"}
		return
	}
	resp = &pb.AllhelpSubmitChatSummaryResp{Record: rec}
	return
}

// isValidDate 检查 YYYY-MM-DD
func isValidDate(s string) bool {
	_, err := time.Parse("2006-01-02", s)
	return err == nil
}

// joinChatMsgs 将结构化消息拼接为 AI 可读的文本
func joinChatMsgs(msgs []*pb.ChatMsg) string {
	var b strings.Builder
	for _, m := range msgs {
		if m == nil {
			continue
		}
		speaker := m.Speaker
		if speaker == "" {
			speaker = "user"
		}
		if m.Time > 0 {
			b.WriteString(fmt.Sprintf("[%s %s] %s\n", time.Unix(m.Time, 0).Format("15:04"), speaker, m.Content))
		} else {
			b.WriteString(fmt.Sprintf("[%s] %s\n", speaker, m.Content))
		}
	}
	return b.String()
}