AgentCoord/frontend/src/api/index.ts

import request from '@/utils/request'
import type { Agent, IApiStepTask, IRawPlanResponse } from '@/stores'
import {
  mockBackendAgentSelectModifyInit,
  mockBackendAgentSelectModifyAddAspect,
  type BackendAgentScoreResponse,
} from '@/layout/components/Main/TaskTemplate/TaskSyllabus/components/mock/AgentAssignmentBackendMock'
import {
  mockBackendFillAgentTaskProcess,
  type RawAgentTaskProcessResponse,
} from '@/layout/components/Main/TaskTemplate/TaskProcess/components/mock/AgentTaskProcessBackendMock'

export interface ActionHistory {
  ID: string
  ActionType: string
  AgentName: string
  Description: string
  ImportantInput: string[]
  Action_Result: string
}

export type IExecuteRawResponse = {
  LogNodeType: string
  NodeId: string
  InputName_List?: string[] | null
  OutputName?: string
  content?: string
  ActionHistory: ActionHistory[]
}

export interface IFillAgentSelectionRequest {
  goal: string
  stepTask: IApiStepTask
  agents: string[]
}

class Api {
  setAgents = (data: Pick<Agent, 'Name' | 'Profile' | 'apiUrl' | 'apiKey' | 'apiModel'>[]) => {
    return request({
      url: '/setAgents',
      data,
      method: 'POST',
    })
  }

  generateBasePlan = (data: {
    goal: string
    inputs: string[]
    apiUrl?: string
    apiKey?: string
    apiModel?: string
  }) => {
    return request<unknown, IRawPlanResponse>({
      url: '/generate_basePlan',
      method: 'POST',
      data: {
        'General Goal': data.goal,
        'Initial Input Object': data.inputs,
        apiUrl: data.apiUrl,
        apiKey: data.apiKey,
        apiModel: data.apiModel,
      },
    })
  }

  executePlan = (plan: IRawPlanResponse) => {
    return request<unknown, IExecuteRawResponse[]>({
      url: '/executePlan',
      method: 'POST',
      data: {
        RehearsalLog: [],
        num_StepToRun: null,
        plan: {
          'Initial Input Object': plan['Initial Input Object'],
          'General Goal': plan['General Goal'],
          'Collaboration Process': plan['Collaboration Process']?.map((step) => ({
            StepName: step.StepName,
            TaskContent: step.TaskContent,
            InputObject_List: step.InputObject_List,
            OutputObject: step.OutputObject,
            AgentSelection: step.AgentSelection,
            Collaboration_Brief_frontEnd: step.Collaboration_Brief_frontEnd,
            TaskProcess: step.TaskProcess.map((action) => ({
              ActionType: action.ActionType,
              AgentName: action.AgentName,
              Description: action.Description,
              ID: action.ID,
              ImportantInput: action.ImportantInput,
            })),
          })),
        },
      },
    })
  }

  // 分支任务大纲（根节点级别）
  branchPlanOutline = (data: {
    branch_Number: number
    Modification_Requirement: string
    Existing_Steps: string[]
    Baseline_Completion: number
    initialInputs: string[]
    goal: string
  }) => {
    return request<unknown, IRawPlanResponse>({
      url: '/branch_PlanOutline',
      method: 'POST',
      data: {
        branch_Number: data.branch_Number,
        Modification_Requirement: data.Modification_Requirement,
        Existing_Steps: data.Existing_Steps,
        Baseline_Completion: data.Baseline_Completion,
        'Initial Input Object': data.initialInputs,
        'General Goal': data.goal,
      },
    })
  }

  // 分支任务流程（任务节点级别）
  branchTaskProcess = (data: {
    branch_Number: number
    Modification_Requirement: string
    Existing_Steps: string[]
    Baseline_Completion: number
    stepTaskExisting: any
    goal: string
  }) => {
    return request<unknown, IRawPlanResponse>({
      url: '/branch_TaskProcess',
      method: 'POST',
      data: {
        branch_Number: data.branch_Number,
        Modification_Requirement: data.Modification_Requirement,
        Existing_Steps: data.Existing_Steps,
        Baseline_Completion: data.Baseline_Completion,
        stepTaskExisting: data.stepTaskExisting,
        'General Goal': data.goal,
      },
    })
  }

  fillStepTask = (data: { goal: string; stepTask: any }) => {
    return request<unknown, any>({
      url: '/fill_stepTask',
      method: 'POST',
      data: {
        'General Goal': data.goal,
        stepTask: data.stepTask,
      },
    })
  }

  fillStepTaskTaskProcess = async (data: {
    goal: string
    stepTask: IApiStepTask
    agents: string[]
  }): Promise<IApiStepTask> => {
    // 后端返回格式: IRawStepTask
    const response = await request<
      {
        'General Goal': string
        stepTask_lackTaskProcess: {
          StepName: string
          TaskContent: string
          InputObject_List: string[]
          OutputObject: string
          AgentSelection: string[]
        }
      },
      {
        StepName?: string
        TaskContent?: string
        InputObject_List?: string[]
        OutputObject?: string
        AgentSelection?: string[]
        TaskProcess?: Array<{
          ID: string
          ActionType: string
          AgentName: string
          Description: string
          ImportantInput: string[]
        }>
        Collaboration_Brief_FrontEnd?: {
          template: string
          data: Record<string, { text: string; color: number[] }>
        }
      }
    >({
      url: '/fill_stepTask_TaskProcess',
      method: 'POST',
      data: {
        'General Goal': data.goal,
        stepTask_lackTaskProcess: {
          StepName: data.stepTask.name,
          TaskContent: data.stepTask.content,
          InputObject_List: data.stepTask.inputs,
          OutputObject: data.stepTask.output,
          AgentSelection: data.agents,
        },
      },
    })

    // 数据转换：后端格式 (IRawStepTask) → 前端格式 (IApiStepTask)
    // 注意：此转换逻辑与Mock API完全一致

    // 1. 转换颜色格式：[h, s, l] → "hsl(h, s%, l%)"
    const vec2Hsl = (color: number[]): string => {
      const [h, s, l] = color
      return `hsl(${h}, ${s}%, ${l}%)`
    }

    // 2. 转换 brief.data: { "0": { text, color: [h,s,l] } } → { "0": { text, style: { background } } }
    const briefData: Record<string, { text: string; style: { background: string } }> = {}
    if (response.Collaboration_Brief_FrontEnd?.data) {
      for (const [key, value] of Object.entries(response.Collaboration_Brief_FrontEnd.data)) {
        briefData[key] = {
          text: value.text,
          style: {
            background: vec2Hsl(value.color),
          },
        }
      }
    }

    // 3. 转换 process: { ID, ActionType, AgentName, Description, ImportantInput } → { id, type, agent, description, inputs }
    const process = (response.TaskProcess || []).map((action) => ({
      id: action.ID,
      type: action.ActionType,
      agent: action.AgentName,
      description: action.Description,
      inputs: action.ImportantInput,
    }))

    // 4. 构建前端格式的 IApiStepTask
    return {
      name: response.StepName || '',
      content: response.TaskContent || '',
      inputs: response.InputObject_List || [],
      output: response.OutputObject || '',
      agents: response.AgentSelection || [],
      brief: {
        template: response.Collaboration_Brief_FrontEnd?.template || '',
        data: briefData,
      },
      process,
    }
  }

  // 为每个智能体评分
  agentSelectModifyInit = async (data: {
    goal: string
    stepTask: any
  }): Promise<Record<string, Record<string, { reason: string; score: number }>>> => {
    // 后端返回：维度 -> agent -> { Reason, Score }
    const response = await request<
      {
        'General Goal': string
        stepTask: any
      },
      Record<string, Record<string, { Reason: string; Score: number }>>
    >({
      url: '/agentSelectModify_init',
      method: 'POST',
      data: {
        'General Goal': data.goal,
        stepTask: {
          StepName: data.stepTask.StepName || data.stepTask.name,
          TaskContent: data.stepTask.TaskContent || data.stepTask.content,
          InputObject_List: data.stepTask.InputObject_List || data.stepTask.inputs,
          OutputObject: data.stepTask.OutputObject || data.stepTask.output,
        },
      },
    })

    // 数据转换：后端格式 (维度->agent) → 前端格式 (agent->维度)
    const transformedData: Record<string, Record<string, { reason: string; score: number }>> = {}

    for (const [aspect, agents] of Object.entries(response)) {
      // aspect: 维度名称, agents: { agentName: { Reason, Score } }
      for (const [agentName, scoreInfo] of Object.entries(agents)) {
        if (!transformedData[agentName]) {
          transformedData[agentName] = {}
        }
        transformedData[agentName][aspect] = {
          reason: scoreInfo.Reason,
          score: scoreInfo.Score,
        }
      }
    }

    return transformedData
  }

  // 添加新的评估维度
  // 定义返回类型（与 Mock 数据格式一致）
  agentSelectModifyAddAspect = async (data: {
    aspectList: string[]
  }): Promise<{
    aspectName: string
    agentScores: Record<string, { score: number; reason: string }>
  }> => {
    // 后端返回：维度 -> agent -> { Reason, Score }
    const response = await request<
      {
        aspectList: string[]
      },
      Record<string, Record<string, { Reason: string; Score: number }>>
    >({
      url: '/agentSelectModify_addAspect',
      method: 'POST',
      data: {
        aspectList: data.aspectList,
      },
    })

    // 获取新添加的维度（最后一个）
    const newAspect = data.aspectList[data.aspectList.length - 1]
    if (!newAspect) {
      throw new Error('aspectList is empty')
    }

    // 提取该维度的数据：维度 -> agent -> { Reason, Score } → agent -> { score, reason }
    const newAspectAgents = response[newAspect]
    const agentScores: Record<string, { score: number; reason: string }> = {}

    if (newAspectAgents) {
      for (const [agentName, scoreInfo] of Object.entries(newAspectAgents)) {
        agentScores[agentName] = {
          score: scoreInfo.Score,
          reason: scoreInfo.Reason,
        }
      }
    }

    return {
      aspectName: newAspect,
      agentScores,
    }
  }

  // ==================== Mock API（开发阶段使用）====================
  // Mock API 使用与真实 API 相同的数据转换逻辑，确保将来切换无缝

  // Mock: 为每个智能体评分
  mockAgentSelectModifyInit = async (): Promise<
    Record<string, Record<string, { reason: string; score: number }>>
  > => {
    // 调用Mock后端数据（维度 -> agent -> { Reason, Score }）
    const response: BackendAgentScoreResponse = await mockBackendAgentSelectModifyInit()

    // 数据转换：后端格式 (维度->agent) → 前端格式 (agent->维度)
    // 注意：此转换逻辑与真实API完全一致
    const transformedData: Record<string, Record<string, { reason: string; score: number }>> = {}

    for (const [aspect, agents] of Object.entries(response)) {
      // aspect: 维度名称, agents: { agentName: { Reason, Score } }
      for (const [agentName, scoreInfo] of Object.entries(agents)) {
        if (!transformedData[agentName]) {
          transformedData[agentName] = {}
        }
        transformedData[agentName][aspect] = {
          reason: scoreInfo.Reason,
          score: scoreInfo.Score,
        }
      }
    }

    return transformedData
  }

  // Mock: 添加新的评估维度
  mockAgentSelectModifyAddAspect = async (data: {
    aspectList: string[]
  }): Promise<{
    aspectName: string
    agentScores: Record<string, { score: number; reason: string }>
  }> => {
    // 调用Mock后端数据（维度 -> agent -> { Reason, Score }）
    const response: BackendAgentScoreResponse = await mockBackendAgentSelectModifyAddAspect(
      data.aspectList,
    )

    // 获取新添加的维度（最后一个）
    const newAspect = data.aspectList[data.aspectList.length - 1]
    if (!newAspect) {
      throw new Error('aspectList is empty')
    }

    // 提取该维度的数据：维度 -> agent -> { Reason, Score } → agent -> { score, reason }
    // 注意：此转换逻辑与真实API完全一致
    const newAspectAgents = response[newAspect]
    const agentScores: Record<string, { score: number; reason: string }> = {}

    if (newAspectAgents) {
      for (const [agentName, scoreInfo] of Object.entries(newAspectAgents)) {
        agentScores[agentName] = {
          score: scoreInfo.Score,
          reason: scoreInfo.Reason,
        }
      }
    }

    return {
      aspectName: newAspect,
      agentScores,
    }
  }

  // Mock: 填充智能体任务流程
  mockFillStepTaskTaskProcess = async (data: {
    goal: string
    stepTask: IApiStepTask
    agents: string[]
  }): Promise<IApiStepTask> => {
    // 调用Mock后端数据（后端原始格式）
    const response: RawAgentTaskProcessResponse = await mockBackendFillAgentTaskProcess(
      data.goal,
      data.stepTask,
      data.agents,
    )

    // 数据转换：后端格式 → 前端格式
    // 注意：此转换逻辑与真实API完全一致

    // 1. 转换颜色格式：[h, s, l] → "hsl(h, s%, l%)"
    const vec2Hsl = (color: number[]): string => {
      const [h, s, l] = color
      return `hsl(${h}, ${s}%, ${l}%)`
    }

    // 2. 转换 brief.data: { "0": { text, color: [h,s,l] } } → { "0": { text, style: { background } } }
    const briefData: Record<string, { text: string; style: { background: string } }> = {}
    if (response.Collaboration_Brief_frontEnd?.data) {
      for (const [key, value] of Object.entries(response.Collaboration_Brief_frontEnd.data)) {
        briefData[key] = {
          text: value.text,
          style: {
            background: vec2Hsl(value.color),
          },
        }
      }
    }

    // 3. 转换 process: { ID, ActionType, AgentName, Description, ImportantInput } → { id, type, agent, description, inputs }
    const process = (response.TaskProcess || []).map((action) => ({
      id: action.ID,
      type: action.ActionType,
      agent: action.AgentName,
      description: action.Description,
      inputs: action.ImportantInput,
    }))

    // 4. 构建前端格式的 IApiStepTask
    return {
      name: response.StepName || '',
      content: response.TaskContent || '',
      inputs: response.InputObject_List || [],
      output: response.OutputObject || '',
      agents: response.AgentSelection || [],
      brief: {
        template: response.Collaboration_Brief_frontEnd?.template || '',
        data: briefData,
      },
      process,
    }
  }
}

export default new Api()