//go:build ignore

package main

import (
	"os"
	"strings"
)

func main() {
	f := "internal/biz/advice_chat.go"
	b, _ := os.ReadFile(f)
	s := string(b)
	ok := true

	// === 1. Regis: 存 ChatData 到 Redis ===
	old1 := `	err = a.rdb.Rdb.SetEx(ctx, sessionId, promptJson, 3600*time.Second).Err()

	return sessionId, err
}`
	new1 := `	err = a.rdb.Rdb.SetEx(ctx, sessionId, promptJson, 3600*time.Second).Err()
	// 存储 ChatData 供后续 Chat 全量注入提示词
	if chatDataJson, e := json.Marshal(chatData); e == nil {
		_ = a.rdb.Rdb.SetEx(ctx, "chatdata:"+sessionId, string(chatDataJson), 3600*time.Second).Err()
	}
	return sessionId, err
}`
	if strings.Contains(s, old1) {
		s = strings.Replace(s, old1, new1, 1)
		println("1. Regis: ChatData store added")
	} else {
		println("1. ERROR: Regis block not found")
		ok = false
	}

	// === 2. Chat: 从 Redis 取 ChatData，传给 buildChatPromptResponse ===
	old2 := `	prompt, err := a.buildChatPromptResponse(ctx, chat, &session, chatHis)
	if err != nil {
		return assistant, err
	}
	log.Infof("[提示词] sessionId=%s prompt=%s", session.SessionID, pkg.JsonStringIgonErr(prompt))
	resContent, err := a.callLlmResponse(ctx, prompt, modelInfo.Key, modelInfo.URL, modelInfo.ChatModel, session.ContextCache)`
	new2 := `	var chatData entitys.ChatData
	if cdStr, e := a.rdb.Rdb.Get(ctx, "chatdata:"+session.SessionID).Result(); e == nil {
		_ = json.Unmarshal([]byte(cdStr), &chatData)
	}
	prompt, err := a.buildChatPromptResponse(ctx, chat, &session, chatHis, &chatData)
	if err != nil {
		return assistant, err
	}
	log.Infof("[提示词] sessionId=%s prompt=%s", session.SessionID, pkg.JsonStringIgonErr(prompt))
	resContent, err := a.callLlmResponse(ctx, prompt, modelInfo.Key, modelInfo.URL, modelInfo.ChatModel, session.ContextCache)`
	if strings.Contains(s, old2) {
		s = strings.Replace(s, old2, new2, 1)
		println("2. Chat: ChatData load + pass added")
	} else {
		println("2. ERROR: Chat block not found")
		ok = false
	}

	// === 3. ChatWithSession: 同样从 Redis 取 ChatData ===
	old3 := `	prompt, err := a.buildChatPromptResponse(ctx, chat, session, chatHis)
	if err != nil {
		return assistant, err
	}
	log.Infof("[提示词] sessionId=%s prompt=%s", session.SessionID, pkg.JsonStringIgonErr(prompt))
	resContent, err := a.callLlmResponse(ctx, prompt, modelInfo.Key, modelInfo.URL, modelInfo.ChatModel, session.ContextCache)`
	new3 := `	var chatData entitys.ChatData
	if cdStr, e := a.rdb.Rdb.Get(ctx, "chatdata:"+session.SessionID).Result(); e == nil {
		_ = json.Unmarshal([]byte(cdStr), &chatData)
	}
	prompt, err := a.buildChatPromptResponse(ctx, chat, session, chatHis, &chatData)
	if err != nil {
		return assistant, err
	}
	log.Infof("[提示词] sessionId=%s prompt=%s", session.SessionID, pkg.JsonStringIgonErr(prompt))
	resContent, err := a.callLlmResponse(ctx, prompt, modelInfo.Key, modelInfo.URL, modelInfo.ChatModel, session.ContextCache)`
	if strings.Contains(s, old3) {
		s = strings.Replace(s, old3, new3, 1)
		println("3. ChatWithSession: ChatData load + pass added")
	} else {
		println("3. ERROR: ChatWithSession block not found")
		ok = false
	}

	// === 4. buildChatPromptResponse: 全量数据注入 ===
	old4 := `func (a *AdviceChatBiz) buildChatPromptResponse(ctx context.Context, chat *entitys.AdvicerChatReq, session *dbmodel.AiAdviceSession, chatList []mongo_model.AdvicerChatHisMongoEntity) ([]openai.ResponseInputMessage, error) {
	// 输出格式指令：模型使用 response_format=json_object 时，prompt 必须包含 "json" 关键词，
	// 同时明确告诉模型返回 JSON 结构，便于下游 json.Unmarshal 解析为 Assistant。
	const outputFormatHint = "\n[输出格式]请用 json 格式返回你的回复，结构为 {\"result\":\"回复内容\",\"mission_status\":\"in_progress|completed|fail\",\"mission_complete_desc\":\"本轮进展小结\"}。"
	message := []openai.ResponseInputMessage{
		{
			Role:    openai.ChatMessageRoleSystem,
			Content: a.taskPrompt(session) + outputFormatHint,
		},
		{
			Role:    openai.ChatMessageRoleSystem,
			Content: formatChatHis(chatList),
		},
		{
			Role:    openai.ChatMessageRoleUser,
			Content: chat.Content,
		},
	}
	return message, nil
}`
	new4 := `func (a *AdviceChatBiz) buildChatPromptResponse(ctx context.Context, chat *entitys.AdvicerChatReq, session *dbmodel.AiAdviceSession, chatList []mongo_model.AdvicerChatHisMongoEntity, chatData *entitys.ChatData) ([]openai.ResponseInputMessage, error) {
	const outputFormatHint = "\n[输出格式]请用 json 格式返回你的回复，结构为 {\"result\":\"回复内容\",\"mission_status\":\"in_progress|completed|fail\",\"mission_complete_desc\":\"本轮进展小结\"}。"
	var sys strings.Builder
	// 基础规则 + 任务 + 风控红线
	sys.WriteString(constants.BasePrompt)
	if session.Mission != "" {
		sys.WriteString(session.Mission)
		sys.WriteString("\n")
	}
	if chatData != nil && chatData.RuleDimension != "" {
		sys.WriteString(constants.RulePromptPrefix)
		sys.WriteString(chatData.RuleDimension)
		sys.WriteString("\n")
	}
	sys.WriteString(constants.BasePrompt2)
	// 全量结构化数据注入
	if chatData != nil {
		if chatData.ProjectInfo != nil {
			sys.WriteString("\n[项目信息]\n")
			sys.WriteString(pkg.JsonStringIgonErr(chatData.ProjectInfo))
		}
		if chatData.AdvicerInfo != nil {
			sys.WriteString("\n[销售信息]\n")
			sys.WriteString(pkg.JsonStringIgonErr(chatData.AdvicerInfo))
		}
		if chatData.AdvicerVersion != nil && len(chatData.AdvicerVersion) > 0 {
			sys.WriteString("\n[销售人设风格]\n")
			sys.WriteString(pkg.JsonStringIgonErr(chatData.AdvicerVersion))
		}
		if chatData.ClientInfo != nil {
			sys.WriteString("\n[客户信息]\n")
			sys.WriteString(pkg.JsonStringIgonErr(chatData.ClientInfo))
		}
		if chatData.TalkSkill != nil {
			sys.WriteString("\n[销售技巧]\n")
			sys.WriteString(pkg.JsonStringIgonErr(chatData.TalkSkill))
		}
	}
	sys.WriteString(outputFormatHint)
	message := []openai.ResponseInputMessage{
		{
			Role:    openai.ChatMessageRoleSystem,
			Content: sys.String(),
		},
		{
			Role:    openai.ChatMessageRoleSystem,
			Content: formatChatHis(chatList),
		},
		{
			Role:    openai.ChatMessageRoleUser,
			Content: chat.Content,
		},
	}
	return message, nil
}`
	if strings.Contains(s, old4) {
		s = strings.Replace(s, old4, new4, 1)
		println("4. buildChatPromptResponse: full data injection done")
	} else {
		println("4. ERROR: buildChatPromptResponse block not found")
		ok = false
	}

	// === 5. 删掉 taskPrompt 中的 session.Mission 注入（避免重复，已在 buildChatPromptResponse 中处理）===
	old5 := `	// 注入 mission（含人格 + 项目信息），确保每次 Chat 模型都能看到
	if session.Mission != "" {
		b.WriteString("\n" + session.Mission)
	}
	return b.String()
}

func (a *AdviceChatBiz) buildBasePromptResponse`
	new5 := `	return b.String()
}

func (a *AdviceChatBiz) buildBasePromptResponse`
	if strings.Contains(s, old5) {
		s = strings.Replace(s, old5, new5, 1)
		println("5. taskPrompt: removed redundant mission injection")
	} else {
		println("5. SKIP: taskPrompt already clean")
	}

	if ok {
		os.WriteFile(f, []byte(s), 0644)
		println("\nALL DONE - advice_chat.go updated")
	} else {
		println("\nFAILED - some blocks not found")
	}
}