import { EventSourcePolyfill } from 'event-source-polyfill'
let connectEventTimeout = 60000 * 5
let ev = null as any
let connectEventTimer = null as any
let waitMsgTimer = null as any
const requestEv = () => {
  const getEvFlow= (params: any, callback: any) => {
    let data = ''
    // 接口可作為參數(shù)傳遞進來
    ev = new EventSourcePolyfill(params.fullUrl || (params.url), {
      headers: {
        token: '123' // 攜帶參數(shù)
      },
      heartbeatTimeout: connectEventTimeout  // 心跳超時時長端壳，也可傳參
    })
    connectEventTimer= setTimeout(() => {
      ev?.close()
      connectEventTimer = null
      clearTimeout(connectEventTimer )
    }, connectEventTimeout )
    ev.onopen = (e: any) => {
      connectEventTimer = null
      clearTimeout(connectEventTimer )
    }
    ev.onmessage = (e: any) => {
      // e中有個status屬性是用來記錄當前流的狀態(tài)告丢，可捕獲到流是正在輸出還是輸出結束等
      clearTimeout(waitMsgTimer)
      let txt = e.data
      data += txt
      data = data.replace('[ERROR]', '').replace('[ILLEGAL]', '').replace('[END]', '')
      callback.doingCallback?.(data)
      if (txt === '[ERROR]') { // gpt異常處理會返回[ERROR]
        callback.errCallback?.(data.replace('[ERROR]', ''))
        ev?.close()
        return
      } else if (txt === '[ILLEGAL]') { // gpt違法處理會返回[ILLEGAL]
        callback.illegalCallback?.(data.replace('[ILLEGAL]', ''))
        ev?.close()
        return
      } else if (txt === '[EV_END]') { // gpt結束處理會返回[EV_END]
        callback.endCallback?.(data.replace('[EV_END]', '')) // 吐出去的回調函數(shù)
        ev?.close()
        return
      }
      // 10秒長時間沒反應的時候自動停止流的狀態(tài)（此處還有一個未解決的問題就是如果手動停止流的輸出會跟10秒后的停止流函數(shù)重合）
      waitMsgTimer = setTimeout(() => {
        callback.endCallback?.(data.replace('[ERROR]', ''))
        ev?.close()
      }, 10000)
    }
    ev.onerror = (e: any) => {
      ev?.close()
      connectEventTimer= null
      callback.evErrCallback?.(e)
      clearTimeout(connectEventTimer)
    }
    ev.onclose = (e: any) => {
      console.log(e, 'renderEvFlow onclose')
    }
    ev.addEventListener('close', (e: any) => {
      console.log(e, 'renderEvFlow close')
    })
  }

  const stopFlow = () => {
    ev?.close()
    connectEventTimer= null
    clearTimeout(connectEventTimer)
  }
  return { getEvFlow, stopFlow }
}
export default requestEv