ForcePilot/web/src/composables/useApproval.js

import { reactive } from 'vue'
import { message } from 'ant-design-vue'
import { handleChatError } from '@/utils/errorHandler'
import { agentApi } from '@/apis'

export function useApproval({ getThreadState, resetOnGoingConv, fetchThreadMessages }) {
  // 审批状态
  const approvalState = reactive({
    showModal: false,
    question: '',
    operation: '',
    threadId: null,
    interruptInfo: null
  })

  // 处理审批逻辑
  const handleApproval = async (approved, currentAgentId, agentConfigId = null) => {
    const threadId = approvalState.threadId
    if (!threadId) {
      message.error('无效的审批请求')
      approvalState.showModal = false
      return
    }

    const threadState = getThreadState(threadId)
    if (!threadState) {
      message.error('无法找到对应的对话线程')
      approvalState.showModal = false
      return
    }

    // 关闭弹窗
    approvalState.showModal = false

    // 清理旧的流式控制器（如果存在）
    if (threadState.streamAbortController) {
      threadState.streamAbortController.abort()
      threadState.streamAbortController = null
    }

    // 标记为处理中
    threadState.isStreaming = true
    resetOnGoingConv(threadId)
    threadState.streamAbortController = new AbortController()

    console.log('🔄 [APPROVAL] Starting resume process:', { approved, threadId, currentAgentId })

    try {
      // 调用恢复接口
      const response = await agentApi.resumeAgentChat(
        currentAgentId,
        {
          thread_id: threadId,
          approved: approved,
          config: agentConfigId ? { agent_config_id: agentConfigId } : {}
        },
        {
          signal: threadState.streamAbortController?.signal
        }
      )

      console.log('🔄 [APPROVAL] Resume API response received')

      if (!response.ok) {
        const errorText = await response.text()
        console.error('Resume API error:', response.status, errorText)
        throw new Error(`HTTP error! status: ${response.status}, details: ${errorText}`)
      }

      console.log('🔄 [APPROVAL] Resume API successful, returning response for stream processing')
      return response // 返回响应供调用方处理流式数据
    } catch (error) {
      console.error('❌ [APPROVAL] Resume failed:', error)
      if (error.name !== 'AbortError') {
        handleChatError(error, 'resume')
        message.error(`恢复对话失败: ${error.message || '未知错误'}`)
      }
      // 重置状态 - 只在错误时重置
      threadState.isStreaming = false
      threadState.streamAbortController = null
      throw error // 重新抛出错误让调用方处理
    }
    // 移除 finally 块 - 让组件管理流式状态的生命周期
  }

  // 在流式处理中处理审批请求
  const processApprovalInStream = (chunk, threadId, currentAgentId) => {
    if (chunk.status !== 'human_approval_required') {
      return false
    }

    const { interrupt_info } = chunk
    const threadState = getThreadState(threadId)

    if (!threadState) return false

    // 停止显示"处理中"状态，让用户可以看到并操作审批弹窗
    threadState.isStreaming = false

    // 显示审批弹窗
    approvalState.showModal = true
    approvalState.question = interrupt_info?.question || '是否批准以下操作？'
    approvalState.operation = interrupt_info?.operation || '未知操作'
    approvalState.threadId = chunk.thread_id || threadId
    approvalState.interruptInfo = interrupt_info

    // 刷新消息历史显示已执行的部分
    fetchThreadMessages({ agentId: currentAgentId, threadId: threadId })

    return true // 表示已处理审批请求，应停止流式处理
  }

  // 重置审批状态
  const resetApprovalState = () => {
    approvalState.showModal = false
    approvalState.question = ''
    approvalState.operation = ''
    approvalState.threadId = null
    approvalState.interruptInfo = null
  }

  return {
    approvalState,
    handleApproval,
    processApprovalInStream,
    resetApprovalState
  }
}