feat:三个浮动窗口功能新增

2025-12-31 19:04:58 +08:00
parent d42554ce03
commit 5847365eee
34 changed files with 7471 additions and 647 deletions
--- a/frontend/src/stores/modules/selection.ts
+++ b/frontend/src/stores/modules/selection.ts
@@ -0,0 +1,247 @@
+import { ref } from 'vue'
+import { defineStore } from 'pinia'
+import { v4 as uuidv4 } from 'uuid'
+import { store } from '../index'
+import type { IRawStepTask, IApiStepTask } from './agents'
+import type { Node, Edge } from '@vue-flow/core'
+
+// 分支节点数据接口 - 用于存储流程图中的节点和边
+export interface IBranchData {
+  id: string // 分支唯一ID
+  parentNodeId: string // 父节点ID（根节点或任务节点）
+  branchContent: string // 分支需求内容
+  branchType: 'root' | 'task' // 分支类型
+  nodes: Node[] // 分支包含的所有节点
+  edges: Edge[] // 分支包含的所有边
+  tasks: IRawStepTask[] // 分支的任务数据
+  createdAt: number // 创建时间
+}
+
+export const useSelectionStore = defineStore('selection', () => {
+  // ==================== 任务大纲探索分支数据存储 ====================
+  const flowBranches = ref<IBranchData[]>([])
+
+  // 任务大纲分支管理
+
+  // 添加流程图分支
+  function addFlowBranch(data: {
+    parentNodeId: string
+    branchContent: string
+    branchType: 'root' | 'task'
+    nodes: Node[]
+    edges: Edge[]
+    tasks: IRawStepTask[]
+  }): string {
+    const branchId = `flow-branch-${uuidv4()}`
+    const newBranch: IBranchData = {
+      id: branchId,
+      parentNodeId: data.parentNodeId,
+      branchContent: data.branchContent,
+      branchType: data.branchType,
+      nodes: data.nodes,
+      edges: data.edges,
+      tasks: data.tasks,
+      createdAt: Date.now(),
+    }
+    flowBranches.value.push(newBranch)
+    console.log('📂 保存流程图分支到 store:', newBranch)
+    return branchId
+  }
+
+  // 获取所有流程图分支
+  function getAllFlowBranches(): IBranchData[] {
+    return flowBranches.value
+  }
+
+  // 根据父节点ID获取流程图分支
+  function getFlowBranchesByParent(parentNodeId: string): IBranchData[] {
+    return flowBranches.value.filter((branch) => branch.parentNodeId === parentNodeId)
+  }
+
+  // 删除流程图分支
+  function removeFlowBranch(branchId: string): boolean {
+    const index = flowBranches.value.findIndex((branch) => branch.id === branchId)
+    if (index > -1) {
+      flowBranches.value.splice(index, 1)
+      console.log('🗑️ 删除流程图分支:', branchId)
+      return true
+    }
+    return false
+  }
+
+  // 清除所有流程图分支
+  function clearFlowBranches() {
+    flowBranches.value = []
+    console.log('🗑️ 清除所有流程图分支')
+  }
+
+  // 根据父节点ID清除流程图分支
+  function clearFlowBranchesByParent(parentNodeId: string) {
+    flowBranches.value = flowBranches.value.filter((branch) => branch.parentNodeId !== parentNodeId)
+    console.log('🗑️ 清除父节点为', parentNodeId, '的流程图分支')
+  }
+
+  // ==================== 任务过程探索分支数据存储 ====================
+  // 用于存储任务过程探索中的分支数据
+  // 结构: Map<taskStepId, IBranchData[]>
+  const taskProcessBranchesMap = ref<Map<string, IBranchData[]>>(new Map())
+
+  // 添加任务过程分支
+  function addTaskProcessBranch(
+    taskStepId: string,
+    data: {
+      parentNodeId: string
+      branchContent: string
+      branchType: 'root' | 'task'
+      nodes: Node[]
+      edges: Edge[]
+      tasks: IRawStepTask[]
+    },
+  ): string {
+    const branchId = `task-process-branch-${uuidv4()}`
+    const newBranch: IBranchData = {
+      id: branchId,
+      parentNodeId: data.parentNodeId,
+      branchContent: data.branchContent,
+      branchType: data.branchType,
+      nodes: data.nodes,
+      edges: data.edges,
+      tasks: data.tasks,
+      createdAt: Date.now(),
+    }
+
+    // 获取或创建该任务步骤的分支列表
+    if (!taskProcessBranchesMap.value.has(taskStepId)) {
+      taskProcessBranchesMap.value.set(taskStepId, [])
+    }
+    taskProcessBranchesMap.value.get(taskStepId)!.push(newBranch)
+
+    console.log('📂 保存任务过程分支到 store:', { taskStepId, branch: newBranch })
+    return branchId
+  }
+
+  // 获取指定任务步骤的所有分支
+  function getTaskProcessBranches(taskStepId: string): IBranchData[] {
+    return taskProcessBranchesMap.value.get(taskStepId) || []
+  }
+
+  // 获取所有任务过程分支
+  function getAllTaskProcessBranches(): Map<string, IBranchData[]> {
+    return taskProcessBranchesMap.value
+  }
+
+  // 根据父节点ID获取任务过程分支
+  function getTaskProcessBranchesByParent(taskStepId: string, parentNodeId: string): IBranchData[] {
+    const branches = taskProcessBranchesMap.value.get(taskStepId) || []
+    return branches.filter((branch) => branch.parentNodeId === parentNodeId)
+  }
+
+  // 删除任务过程分支
+  function removeTaskProcessBranch(taskStepId: string, branchId: string): boolean {
+    const branches = taskProcessBranchesMap.value.get(taskStepId)
+    if (branches) {
+      const index = branches.findIndex((branch) => branch.id === branchId)
+      if (index > -1) {
+        branches.splice(index, 1)
+        console.log('🗑️ 删除任务过程分支:', { taskStepId, branchId })
+        return true
+      }
+    }
+    return false
+  }
+
+  // 清除指定任务步骤的所有分支
+  function clearTaskProcessBranches(taskStepId: string) {
+    taskProcessBranchesMap.value.delete(taskStepId)
+    console.log('🗑️ 清除任务步骤的所有分支:', taskStepId)
+  }
+
+  // ==================== Agent 组合 TaskProcess 数据存储 ====================
+  // 用于存储 fill_stepTask_TaskProcess 接口返回的数据
+  // 结构: Map<taskId, Map<agentGroupKey, IApiStepTask>>
+  // - taskId: 任务ID
+  // - agentGroupKey: agent 组合的唯一标识（排序后的 JSON 字符串）
+  // - IApiStepTask: 该 agent 组合的完整 TaskProcess 数据
+  const agentTaskProcessMap = ref<Map<string, Map<string, IApiStepTask>>>(new Map())
+
+  // 生成 agent 组合的唯一 key（排序后保证一致性）
+  function getAgentGroupKey(agents: string[]): string {
+    return JSON.stringify(agents.sort())
+  }
+
+  // 存储 agent 组合的 TaskProcess 数据
+  function setAgentTaskProcess(taskId: string, agents: string[], taskProcess: IApiStepTask) {
+    const groupKey = getAgentGroupKey(agents)
+
+    // 获取或创建该任务的 Map
+    if (!agentTaskProcessMap.value.has(taskId)) {
+      agentTaskProcessMap.value.set(taskId, new Map())
+    }
+
+    // 存储该 agent 组合的 TaskProcess 数据
+    agentTaskProcessMap.value.get(taskId)!.set(groupKey, taskProcess)
+    console.log(`📦 存储 agent 组合 TaskProcess [任务: ${taskId}, agents: ${agents.join(', ')}]`)
+  }
+
+  // 获取 agent 组合的 TaskProcess 数据
+  function getAgentTaskProcess(taskId: string, agents: string[]): IApiStepTask | undefined {
+    const groupKey = getAgentGroupKey(agents)
+    return agentTaskProcessMap.value.get(taskId)?.get(groupKey)
+  }
+
+  // 检查 agent 组合是否已有 TaskProcess 数据
+  function hasAgentTaskProcess(taskId: string, agents: string[]): boolean {
+    const groupKey = getAgentGroupKey(agents)
+    return agentTaskProcessMap.value.get(taskId)?.has(groupKey) || false
+  }
+
+  // 清除指定任务的所有 agent 组合 TaskProcess 数据
+  function clearAgentTaskProcess(taskId: string) {
+    agentTaskProcessMap.value.delete(taskId)
+    console.log(`🗑️ 清除任务的 agent 组合 TaskProcess 数据: ${taskId}`)
+  }
+
+  // 清除所有任务的 agent 组合 TaskProcess 数据
+  function clearAllAgentTaskProcess() {
+    agentTaskProcessMap.value.clear()
+    console.log('🗑️ 清除所有任务的 agent 组合 TaskProcess 数据')
+  }
+
+  return {
+    // 状态
+    flowBranches,
+    taskProcessBranchesMap,
+    agentTaskProcessMap,
+
+    // 任务大纲分支管理方法
+    addFlowBranch,
+    getAllFlowBranches,
+    getFlowBranchesByParent,
+    removeFlowBranch,
+    clearFlowBranches,
+    clearFlowBranchesByParent,
+
+    // 任务过程分支管理方法
+    addTaskProcessBranch,
+    getTaskProcessBranches,
+    getAllTaskProcessBranches,
+    getTaskProcessBranchesByParent,
+    removeTaskProcessBranch,
+    clearTaskProcessBranches,
+
+    // Agent 组合 TaskProcess 数据管理方法
+    getAgentGroupKey,
+    setAgentTaskProcess,
+    getAgentTaskProcess,
+    hasAgentTaskProcess,
+    clearAgentTaskProcess,
+    clearAllAgentTaskProcess,
+  }
+})
+
+/**
+ * 用于在组件外部使用 Selection Store
+ */
+export function useSelectionStoreHook() {
+  return useSelectionStore(store)
+}