package com.pavis.admin.aigc.model.resp;

import com.pavis.admin.common.model.resp.BaseResp;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.io.Serial;
import java.time.LocalDateTime;
import java.util.List;

import static com.pavis.admin.aigc.enums.AgentState.COMPLETED;
import static com.pavis.admin.aigc.enums.AgentState.IN_PROGRESS;

/**
 * 智能体规划记录信息
 *
 * @author semi
 * @since 2025/05/28 16:59
 */
@Data
@Schema(description = "智能体规划记录信息")
public class AgentPlanResp extends BaseResp {

    @Serial
    private static final long serialVersionUID = 1L;

    /**
     * 用户ID
     */
    @Schema(description = "用户ID")
    private Long userId;

    /**
     * 所属会话ID
     */
    @Schema(description = "所属会话ID")
    private Long conversationId;

    /**
     * 对话消息Id
     */
    @Schema(description = "对话消息Id")
    private Long messageId;

    /**
     * 对话消息内容
     */
    @Schema(description = "对话消息内容")
    private String messageContent;

    /**
     * 规划思考过程内容输入
     */
    @Schema(description = "规划思考过程内容输入")
    private String thinkInput;

    /**
     * 规划思考过程内容输出
     */
    @Schema(description = "规划思考过程内容输出")
    private String thinkOutput;

    /**
     * 计划主题,大模型根据用户消息生成
     */
    @Schema(description = "计划主题,大模型根据用户消息生成")
    private String title;

    /**
     * 开始时间
     */
    @Schema(description = "开始时间")
    private LocalDateTime startTime;

    /**
     * 结束时间
     */
    @Schema(description = "结束时间")
    private LocalDateTime endTime;

    /**
     * 执行时长(秒)
     */
    @Schema(description = "执行时长(秒)")
    private Long duration;

    /**
     * 当前执行的步骤编号
     */
    @Schema(description = "当前执行的步骤编号")
    private Long currentPlanStepId;

    /**
     * 是否完成:0未完成 1已完成
     */
    @Schema(description = "是否完成:0未完成 1已完成")
    private Boolean completed;

    /**
     * 计划总结
     */
    @Schema(description = "执行总结")
    private String summary;

    /**
     * 计划执行步骤
     */
    @Schema(description = "计划执行步骤")
    private List<AgentPlanStepResp> steps;

    /**
     * 修改人
     */
    @Schema(description = "修改人")
    private Long updateUser;

    /**
     * 修改时间
     */
    @Schema(description = "修改时间")
    private LocalDateTime updateTime;

    public String getPlanExecutionStateStringFormat(boolean onlyCompletedAndFirstInProgress) {
        StringBuilder state = new StringBuilder();
        state.append("全局目标: ").append("\n").append(title).append(")\n");
        state.append("- 全局步骤计划:\n");
        state.append(getStepsExecutionStateStringFormat(onlyCompletedAndFirstInProgress));
        return state.toString();
    }

    /**
     * 获取步骤执行状态的字符串格式
     *
     * @param onlyCompletedAndFirstInProgress 当为true时，只输出所有已完成的步骤和第一个进行中的步骤
     * @return 格式化的步骤执行状态字符串
     */
    public String getStepsExecutionStateStringFormat(boolean onlyCompletedAndFirstInProgress) {
        StringBuilder state = new StringBuilder();
        boolean foundInProgress = false;

        for (int i = 0; i < steps.size(); i++) {
            AgentPlanStepResp step = steps.get(i);

            // 如果onlyCompletedAndFirstInProgress为true，则只显示COMPLETED状态的步骤和第一个IN_PROGRESS状态的步骤
            if (onlyCompletedAndFirstInProgress) {
                // 如果是COMPLETED状态，始终显示
                if (step.getAgentRuntimeStatus() == COMPLETED) {
                    // 什么都不做，继续显示
                }
                // 如果是IN_PROGRESS状态，且还没找到其他IN_PROGRESS的步骤
                else if (step.getAgentRuntimeStatus() == IN_PROGRESS && !foundInProgress) {
                    foundInProgress = true; // 标记已找到IN_PROGRESS步骤
                }
                // 其他所有情况（不是COMPLETED且不是第一个IN_PROGRESS）
                else {
                    continue; // 跳过不符合条件的步骤
                }
            }

            String symbol = switch (step.getAgentRuntimeStatus()) {
                case COMPLETED -> "[completed]";
                case IN_PROGRESS -> "[in_progress]";
                case BLOCKED -> "[blocked]";
                case NOT_STARTED -> "[not_started]";
                default -> "[ ]";
            };
            state.append("步骤 ")
                .append(i)
                .append(": ")
                .append(symbol)
                .append(" ")
                .append(step.getDescription())
                .append("\n")
                .append("\n");
            String result = step.getResult();
            if (result != null && !result.isEmpty()) {
                state.append("该步骤的执行结果: ").append("\n").append(result).append("\n");
            }

        }
        return state.toString();
    }
}