<script setup lang="ts">
import { computed, inject, onMounted, onUnmounted, ref } from 'vue'
import { Emitter } from 'mitt'
import { useI18n } from 'vue-i18n'
import { nanoid } from 'nanoid'
import { useRoute } from 'vue-router'
import { fetchCustomEventSource } from '@/composables/useEventSource'
import { useUserStore } from '@/store/modules/user'
import { UploadStatus } from '@/enums/upload-status'
import { useDialogueFile } from '@/composables/useDialogueFile'
import { useLayoutConfig } from '@/composables/useLayoutConfig'
import { TEXTTOSPEECH_WS_URL } from '@/config/base-url'
import WebSocketCtr from '@/utils/web-socket-ctr'
import { ChannelType } from '@/enums/channel'

interface Props {
  agentId: string
  dialogsId: string
  messageList: Map<string, ConversationMessageItem>
  continuousQuestionStatus: 'default' | 'close'
  isEnableDocumentParse: boolean
  isEnableVoice: boolean
  answerAudioAutoPlay: boolean
  answerAudioPlaying: boolean
  timbreId: string
}

const { t } = useI18n()

const { query } = useRoute()

const props = defineProps<Props>()

const emit = defineEmits<{
  addMessageItem: [messageId: string, value: ConversationMessageItem]
  updateSpecifyMessageItem: [messageId: string, newObj: Partial<ConversationMessageItem>]
  deleteMessageItem: [messageId: string]
  updatePageScroll: []
  clearAllMessage: []
  toLogin: []
  createContinueQuestions: [value: string]
  resetContinueQuestionList: []
  audioPlay: [messageItem: ConversationMessageItem, requestId?: string]
  audioPause: []
}>()

const { isMobile } = useLayoutConfig()

const userStore = useUserStore()

const { uploadFileList, handleLimitUpload, handleUpload, handleRemoveFile } = useDialogueFile()

const emitter = inject<Emitter<MittEvents>>('emitter')

const inputMessageContent = ref('')
const isAnswerResponseWait = ref(false)
const currentReplyContentSentenceExtractIndex = ref(0)
const sentenceFragmentSerialNo = ref(0)
const sentenceExtractCheckEnabled = ref(false)
const assistantFullAnswerContent = ref('')
const sentenceSpeechException = ref(false)
const messageAudioLoading = ref(false)
const currentLatestMessageItemKeyMap = ref(new Map<'assistant' | 'user', string>())

let controller: AbortController | null = null

const isLogin = computed(() => {
  return userStore.isLogin
})

const isAllowClearMessage = computed(() => {
  return props.messageList.size > 0
})

const isSendBtnDisabled = computed(() => {
  return !inputMessageContent.value.trim()
})

const inputPlaceholder = computed(() => {
  return isLogin.value ? t('common_module.dialogue_module.question_input_placeholder') : ''
})

const isCreateContinueQuestions = computed(() => {
  return props.continuousQuestionStatus === 'default'
})

const isInputMessageDisabled = computed(() => {
  return uploadFileList.value.some((fileItem) => fileItem.status !== UploadStatus.FINISHED)
})

const isUploadFileDisabled = computed(() => {
  return uploadFileList.value.length === 1
})

const uploadFileIcon = (type: string) => {
  return `https://gsst-poe-sit.gz.bcebos.com/icon/${type}.svg`
}

onMounted(() => {
  emitter?.on('selectQuestion', (featuredQuestion) => {
    if (!isLogin.value) {
      window.$message.warning(t('common_module.not_login_text'))
      return
    }

    inputMessageContent.value = featuredQuestion
    handleMessageSend()
  })
})

onUnmounted(() => {
  blockMessageResponse()
  emitter?.off('selectQuestion')
})

function messageItemFactory(): ConversationMessageItem {
  return {
    timestamp: Date.now(),
    role: 'user',
    textContent: '',
    isEmptyContent: false,
    isTextContentLoading: false,
    isAnswerResponseLoading: false,
    isVoiceLoading: false,
    isVoicePlaying: false,
    voiceFragmentUrlList: [],
  }
}

function handleInputMessageEnter(event: KeyboardEvent) {
  if (event.key === 'Enter') {
    event.preventDefault()
    handleMessageSend()
  }
}

function handleMessageSend() {
  if (!isLogin.value) {
    window.$message.warning(t('common_module.not_login_text'))
    return
  }

  if (!inputMessageContent.value.trim() || isInputMessageDisabled.value) {
    return
  }

  if (isAnswerResponseWait.value || messageAudioLoading.value) {
    window.$message.warning(t('common_module.dialogue_module.do_not_operate_until_the_reply_is_complete'))
    return
  }

  if (props.answerAudioPlaying) {
    window.$message.warning(t('common_module.dialogue_module.stop_playing_and_then_operate'))
    return
  }

  const latestUserMessageKey = nanoid()
  const latestAssistantMessageKey = nanoid()
  currentLatestMessageItemKeyMap.value.set('user', latestUserMessageKey)
  currentLatestMessageItemKeyMap.value.set('assistant', latestAssistantMessageKey)

  emit('resetContinueQuestionList')
  emit('addMessageItem', latestUserMessageKey, { ...messageItemFactory(), textContent: inputMessageContent.value })
  emit('updatePageScroll')

  emit('addMessageItem', latestAssistantMessageKey, {
    ...messageItemFactory(),
    role: 'assistant',
    isTextContentLoading: true,
    isAnswerResponseLoading: true,
    isVoiceLoading: true,
  })
  emit('updatePageScroll')

  const input = inputMessageContent.value

  let replyTextContent = ''
  isAnswerResponseWait.value = true
  inputMessageContent.value = ''
  currentReplyContentSentenceExtractIndex.value = 0
  sentenceFragmentSerialNo.value = 0
  sentenceExtractCheckEnabled.value = false
  assistantFullAnswerContent.value = ''
  sentenceSpeechException.value = false
  messageAudioLoading.value = false

  controller = new AbortController()

  fetchCustomEventSource({
    path: '/api/rest/agentApplicationRest/callAgentApplication.json',
    payload: {
      agentId: props.agentId,
      dialogsId: props.dialogsId,
      fileUrls: uploadFileList.value.map((item) => item.url),
      input,
      channel: query.channel || ChannelType.link_share,
    },
    controller,
    onMessage: (data: any) => {
      if (data === '[DONE]') {
        emit('updateSpecifyMessageItem', latestAssistantMessageKey, {
          isEmptyContent: !replyTextContent,
          isTextContentLoading: false,
          isAnswerResponseLoading: false,
        })

        isCreateContinueQuestions.value && emit('createContinueQuestions', replyTextContent)
        emit('updatePageScroll')
        blockMessageResponse()
        return
      }

      if (data) {
        replyTextContent += data

        assistantFullAnswerContent.value = (assistantFullAnswerContent.value + data).replace(
          /\^\[[\d\\[\]-]+?\]\^/g,
          '',
        )

        if (!sentenceExtractCheckEnabled.value && props.isEnableVoice) {
          sentenceExtract(latestAssistantMessageKey)
          sentenceExtractCheckEnabled.value = true
          messageAudioLoading.value = true
        }

        emit('updateSpecifyMessageItem', latestAssistantMessageKey, {
          textContent: replyTextContent,
          isTextContentLoading: false,
        })
        emit('updatePageScroll')
      }
    },
    onRequestError: () => {
      errorMessageResponse()
    },
    onError: () => {
      errorMessageResponse()
    },
    onFinally: () => {
      controller = null
    },
  })
}

function errorMessageResponse() {
  emit('updateSpecifyMessageItem', currentLatestMessageItemKeyMap.value.get('assistant')!, {
    isTextContentLoading: false,
    textContent: '',
  })
  emit('deleteMessageItem', currentLatestMessageItemKeyMap.value.get('user')!)
  emit('deleteMessageItem', currentLatestMessageItemKeyMap.value.get('assistant')!)
  emit('audioPause')
  blockMessageResponse()
}

function handleClearAllMessage() {
  if (!isAllowClearMessage.value) return

  emit('clearAllMessage')
}

function blockMessageResponse() {
  controller?.abort()
  isAnswerResponseWait.value = false
}

function handleToLogin() {
  emit('toLogin')
}

function handleSelectFile(cb: () => void) {
  if (isUploadFileDisabled.value) {
    window.$message.ctWarning('', t('common_module.dialogue_module.overwrite_file_tip')).then(() => {
      cb()
    })
    return
  }

  cb()
}

function sentenceExtract(messageId: string) {
  const symbolRegExp = /[。！？；.!?;]/g
  let sentenceDraft = assistantFullAnswerContent.value
    .replace(/\s{5,}/gi, '')
    .slice(currentReplyContentSentenceExtractIndex.value)

  let matchResult = symbolRegExp.exec(sentenceDraft)

  function matchExtract() {
    const article = assistantFullAnswerContent.value.replace(/\s{5,}/gi, '')

    if (matchResult && matchResult.index && matchResult.index > 60) {
      sentenceDraft = article.slice(
        currentReplyContentSentenceExtractIndex.value,
        currentReplyContentSentenceExtractIndex.value + matchResult.index + matchResult['0'].length,
      )

      currentReplyContentSentenceExtractIndex.value += sentenceDraft.length

      ttsSocketSendText(sentenceDraft, sentenceFragmentSerialNo.value, messageId)
      sentenceFragmentSerialNo.value += 1

      if (article.length - currentReplyContentSentenceExtractIndex.value > 60) {
        sentenceDraft = article.slice(currentReplyContentSentenceExtractIndex.value)
        matchResult = symbolRegExp.exec(sentenceDraft)
        matchExtract()
      } else {
        setTimeout(() => sentenceExtract(messageId), 600)
      }
    } else if (!isAnswerResponseWait.value) {
      /* 延时避免最后回复内容没有更新全 */
      setTimeout(() => {
        sentenceDraft = article.slice(currentReplyContentSentenceExtractIndex.value)

        ttsSocketSendText(sentenceDraft, sentenceFragmentSerialNo.value, messageId)
        sentenceFragmentSerialNo.value += 1
      }, 700)
    } else {
      sentenceDraft = assistantFullAnswerContent.value
        .replace(/\s{5,}/gi, '')
        .slice(currentReplyContentSentenceExtractIndex.value)
      matchResult = symbolRegExp.exec(sentenceDraft)

      setTimeout(() => matchExtract(), 500)
    }
  }

  if (matchResult) matchExtract()
  else setTimeout(() => sentenceExtract(messageId), 600)
}

function ttsSocketSendText(text: string, audioUrlSerialNo: number, messageId: string) {
  if (sentenceSpeechException.value) {
    return
  }

  const ttsSocketCtl = new WebSocketCtr(TEXTTOSPEECH_WS_URL)

  ttsSocketCtl.onMessage = (data: { audio: string; replyVoiceUrl: string; final: boolean }) => {
    if (data.replyVoiceUrl) {
      if (props.messageList.get(messageId)?.voiceFragmentUrlList) {
        const voiceFragmentUrlListDraft = [...props.messageList.get(messageId)!.voiceFragmentUrlList]
        voiceFragmentUrlListDraft[audioUrlSerialNo] = data.replyVoiceUrl
        messageAudioLoading.value = false

        emit('updateSpecifyMessageItem', messageId, { voiceFragmentUrlList: voiceFragmentUrlListDraft })

        if (props.answerAudioAutoPlay && audioUrlSerialNo === 0 && voiceFragmentUrlListDraft[audioUrlSerialNo]) {
          emit('audioPlay', props.messageList.get(messageId)!)
        }

        if (!props.answerAudioAutoPlay) {
          emit('updateSpecifyMessageItem', messageId, { isVoiceLoading: false })
        }
      }
    }
  }

  ttsSocketCtl.onMessageError = () => {
    emit('updateSpecifyMessageItem', messageId, { isVoiceLoading: false, voiceFragmentUrlList: [] })
    props.messageList.get(messageId)?.isVoicePlaying && emit('audioPause')
    sentenceSpeechException.value = true
    messageAudioLoading.value = false
    window.$message.error(t('common_module.unplayable_tip'))
  }

  const content = (text || '').replace(/\^\[[\d\\[\]-]+?\]\^/g, '')

  if (content && props.timbreId) {
    ttsSocketCtl.connect(() => {
      ttsSocketCtl.send({
        codec: 'wav',
        sampleRate: 16000,
        speed: 0,
        voiceType: props.timbreId,
        volume: 0,
        content,
      })
    })
  }
}

defineExpose({
  blockMessageResponse,
})
</script>

<template>
  <div class="my-5">
    <div class="flex items-end gap-2.5">
      <div class="flex flex-1 flex-col">
        <ul v-show="uploadFileList.length > 0" class="mb-1.5 grid gap-1.5">
          <li
            v-for="uploadFileItem in uploadFileList"
            :key="uploadFileItem.id"
            class="group relative flex h-[42px] w-full items-center overflow-hidden rounded-[10px] border bg-white/70"
            :class="uploadFileItem.status === 'error' ? 'border-error-font-color' : 'border-transparent'"
          >
            <div class="flex w-full items-center justify-between px-3.5">
              <div class="flex w-full items-center overflow-hidden">
                <img :src="uploadFileIcon(uploadFileItem.type!)" class="h-7 w-7" />

                <div class="mx-2.5 flex flex-1 flex-col overflow-hidden">
                  <n-ellipsis>
                    {{ uploadFileItem.name }}
                  </n-ellipsis>
                </div>
              </div>

              <n-progress
                v-show="!['finished', 'error'].includes(uploadFileItem.status)"
                class="left-13.5 w-[calc(100%-78px)]! absolute bottom-0"
                type="line"
                rail-color="#F3F3F3"
                :height="4"
                :percentage="uploadFileItem.percentage"
                :show-indicator="false"
              />

              <div
                v-show="['finished', 'error'].includes(uploadFileItem.status)"
                class="group-hover:block"
                :class="isMobile ? 'block' : 'hidden'"
              >
                <n-popover trigger="hover" placement="top-end" :show-arrow="false">
                  <template #trigger>
                    <i
                      class="iconfont icon-close cursor-pointer hover:opacity-80"
                      :class="uploadFileItem.status === 'error' ? 'text-error-font-color' : 'text-font-color'"
                      @click="handleRemoveFile(uploadFileItem.id)"
                    />
                  </template>
                  <span>{{ t('common_module.dialogue_module.cancel_associate_file_tip') }}</span>
                </n-popover>
              </div>
            </div>
          </li>
        </ul>
        <div class="relative flex-1">
          <n-input
            v-model:value="inputMessageContent"
            type="textarea"
            :autosize="{ minRows: 1, maxRows: 5 }"
            :placeholder="inputPlaceholder"
            :disabled="!isLogin || isInputMessageDisabled"
            class="rounded-xl! shadow-[0_1px_#09122105,0_1px_1px_#09122105,0_3px_3px_#09122103,0_9px_9px_#09122103]! py-[4px] pr-[50px]"
            @keydown="handleInputMessageEnter"
          />
          <div
            class="bg-px-send-png absolute bottom-2 right-[20px] h-[24px] w-[24px]"
            :class="
              isSendBtnDisabled ||
              isAnswerResponseWait ||
              !isLogin ||
              isInputMessageDisabled ||
              answerAudioPlaying ||
              messageAudioLoading
                ? 'opacity-60'
                : 'cursor-pointer'
            "
            @click="handleMessageSend"
          />

          <div v-show="!isLogin" class="absolute left-3 top-[5px] flex h-[30px] items-center text-[#84868c]">
            {{ t('share_agent_module.please') }}
            <span class="text-theme-color cursor-pointer px-1 hover:opacity-80" @click="handleToLogin">
              {{ t('common_module.login') }}
            </span>
            {{ t('share_agent_module.after_action') }}
          </div>
        </div>
      </div>

      <n-upload
        :show-file-list="false"
        accept=".doc, .pdf, .docx, .txt, .md"
        :disabled="!isLogin"
        abstract
        @before-upload="handleLimitUpload"
        @change="handleUpload"
      >
        <n-upload-trigger #="{ handleClick }" abstract>
          <n-popover style="width: 210px" trigger="hover">
            <template #trigger>
              <div
                v-show="isEnableDocumentParse"
                class="h-7.5 w-7.5 mb-1 flex items-center justify-center rounded-full bg-white"
                :class="
                  isLogin
                    ? 'hover:text-theme-color text-font-color cursor-pointer'
                    : 'cursor-not-allowed text-[#b8babf]'
                "
                @click="handleSelectFile(handleClick)"
              >
                <i class="iconfont icon-upload flex h-4 w-4 items-center justify-center" />
              </div>
            </template>

            <span class="text-xs"> {{ t('common_module.dialogue_module.upload_file_limit') }} </span>
          </n-popover>
        </n-upload-trigger>
      </n-upload>

      <n-popover trigger="hover">
        <template #trigger>
          <div
            class="h-7.5 w-7.5 mb-1 flex items-center justify-center rounded-full bg-white"
            :class="
              isAllowClearMessage
                ? 'hover:text-theme-color text-font-color cursor-pointer'
                : 'cursor-not-allowed text-[#b8babf]'
            "
            @click="handleClearAllMessage"
          >
            <i class="iconfont icon-clear text-base leading-none" />
          </div>
        </template>
        <span class="text-xs"> {{ t('common_module.dialogue_module.clear_message_popover_message') }}</span>
      </n-popover>
    </div>

    <div class="mt-[9px]">
      <span class="flex w-full justify-center text-xs text-[#b8babf]">
        {{ t('common_module.dialogue_module.generate_warning_message') }}
      </span>
    </div>
  </div>
</template>
