pvs_agent/src/entrypoints/sidepanel/Chat.vue

<!-- Chat.vue -->
<template>
  <div class="chat-container">
    <div v-if="!messages.length" class="message-list">
      <pageMask />
    </div>
    <!-- 消息列表 -->
    <div class="message-list" v-else>
      <el-scrollbar ref="scrollbar" @scroll="handleScroll">
        <!-- 加载更多指示器 -->
        <div v-if="isLoadingMore" class="loading-more-indicator">
          <div class="loading-spinner"></div>
          <span>加载更多消息...</span>
        </div>
        // 添加一个 ref 用于获取消息列表容器
        const messagesContainer = ref(null)
        
        // 修改模板中的消息列表 div，添加 ref
        <div class="messages" ref="messagesContainer">
          <div v-for="(message, index) in messages" :key="index"
            :class="['message-item', message.role === 'user'  ? 'self' : 'other']">
            <el-avatar :size="32" :src="message.role === 'user' ? userAvatar : avatar" />
            <div class="message-content">
              <div class="content" v-if="message.role ==='system'"
                :class="{ 'loading-content': message.content === '' }">
                <formTable v-if="message.type === 'form'" :content="message.rawContent" />
                <span v-else v-html="message.content"></span>
                <span class="loading-indicator" v-if="sendLoading && index === messages.length - 1">
                  <span class="dot"></span>
                  <span class="dot"></span>
                  <span class="dot"></span>
                </span>
              </div>
              <document v-else-if="message.type === 'document' && message.role === 'user'" :content="message.content"
                :rawContent="message.rawContent" />
              <div v-else class="content">
                <span v-if="message.type === ''">{{ message.content }}</span>
                <span class="loading-indicator" v-if="sendLoading && index === messages.length - 1">
                  <span class="dot"></span>
                  <span class="dot"></span>
                  <span class="dot"></span>
                </span>
              </div>
              <div class="timestamp ">{{ moment(message.sortKey).format('MM-DD HH:mm') }}
                <span v-if="message.add" style="cursor: pointer;" @click="handleInput">填充</span>
              </div>
            </div>
          </div>
        </div>
      </el-scrollbar>
      <ScrollToBottom :target="scrollbar" ref="scrollToBottomRef" />
    </div>

    <Tools @read-click="readClick" @upload-file="handleUpload" @handle-capture="handleCapture" @his-records="hisRecords"
      @add-new-dialogue="addNewDialogue" @handle-current-change="handleCurrentChange"
      @handel-intelligent-filling-click="handelIntelligentFillingClick" />

    <div>
      <!-- 输入区域 -->
      <div class="input-area">
        <div v-show="isShowPage" style="border-bottom: 1px solid #F0F0F0;">
          <div class="card_list">
            <div v-for="(v, i) in pageInfoList" :key="i"
              :class="`card-content ${pageInfoList.length > 1 ? 'card_width' : ''}`">
              <img :src="v?.favIconUrl" style="width: 24px;display: block" />
              <div class="title-wrapper">
                <span class="els title-scroller">{{ v?.title }}</span>
                <span class="els url-scroller">{{ v?.url }}</span>
              </div>
              <el-icon class="closeIcon" size="16px" color="#909399" @click="deletePageInfo(i)">
                <CircleClose />
              </el-icon>
            </div>
          </div>

          <div v-show="type !== FunctionList.Intelligent_Form_filling" class="card-btn">
            <el-tooltip content="总结当前页面" placement="top">
              <el-button round @click="handleSummary">总结</el-button>
            </el-tooltip>
          </div>
        </div>

        <el-input ref="textareaRef" v-model="inputMessage" type="textarea" :rows="3" placeholder="输入消息..."
          @keyup.enter="handleAsk" />
        <div class="chat_area_op">
          <el-button
            :style="`background-color:${inputMessage.trim() ? '#4d6bfe' : '#d6dee8'};border-color:${inputMessage.trim() ? '#4d6bfe' : '#d6dee8'}`"
            v-if="inputMessage.trim() || !sendLoading" type="primary" circle @click="handleAsk"
            :disabled="!inputMessage.trim() || sendLoading">
            <svg-icon icon-class="send" color="#000000" />
          </el-button>
          <el-button style="background-color:#ffffff;border:2px solid rgb(134 143 153);" v-else circle
            @click="handleStopAsk">
            <svg-icon icon-class="stop" color="red" />
          </el-button>

        </div>
      </div>

    </div>


    <!--  历史记录  -->
    <historyComponent :msgUuid="msgUuid" ref="historyComponentRef" @currentData="(e) => handleCurrentData(e)" />
  </div>
</template>

<script setup>
import { ref, onMounted, nextTick, inject, useTemplateRef, reactive } from 'vue'
import { ElScrollbar, ElAvatar, ElInput, ElButton } from 'element-plus'
import moment from 'moment'
import fileLogo from '@/assets/svg/file.svg'
import {
  buildExcelUnderstandingPrompt,
  getSummaryPrompt,
  getFileContent,
  buildObjPrompt,
  modelFileUpload,
  controllerList
} from '@/entrypoints/sidepanel/utils/ai-service.js'
import { storeToRefs } from 'pinia'
import { ElMessage } from 'element-plus'
import { useMsg } from '@/entrypoints/sidepanel/hook/useMsg.ts'
import Tools from '@/entrypoints/sidepanel/component/tools.vue'
import historyComponent from '@/entrypoints/sidepanel/component/historyComponent.vue'
import document from '@/entrypoints/sidepanel/component/document.vue'
import pageMask from '@/entrypoints/sidepanel/component/pageMask.vue'
import ScrollToBottom from '@/entrypoints/sidepanel/component/ScrollToBottom.vue'
import formTable from '@/entrypoints/sidepanel/component/formTable.vue'
import userAvatar from '@/assets/images/user.png'
import avatar from '@/public/icon/32.png'
import { mockData, startMsg, mockData2, options, FunctionList } from '@/entrypoints/sidepanel/mock'
import { useAutoResizeTextarea } from '@/entrypoints/sidepanel/hook/useAutoResizeTextarea.ts'
import { getPageInfo, getXlsxValue, handleInput } from './utils/index.js'
import { useMsgStore } from '@/store/modules/msg.ts'
import { useUserStore } from '@/store/modules/user'
import { putChat } from "@/api/index.js";
import {debounce} from 'lodash'
const userStore = useUserStore()
import { getChatDetail } from '@/api/index.js'
// 在其他状态变量附近添加
const isLoadingMore = ref(false)
import { v4 as uuidv4 } from 'uuid';
// 滚动条引用
const scrollbar = ref(null)
const scrollToBottomRef = ref(null)
const xlsxData = ref({})
const isShowPage = ref(false)
const tareRef = useTemplateRef('textareaRef')
const drawerRef = useTemplateRef('historyComponentRef')
// 获取父组件提供的 Hook 实例
const msgStore = useMsgStore()
const { pageInfoList, messages, msgUuid, AIModel,page,hasNext } = storeToRefs(useMsgStore())
const formInfo = ref('')

const {
  taklToHtml,
  sendLoading,
  type,
  streamRes,
  getFormKeyAndValue,
  requestFlowFn,
  fetchRes
} = useMsg(scrollbar)
const inputMessage = ref('')
const pageInfo = ref('')
function handleStopAsk() {
  if (type.value === FunctionList.Intelligent_Form_filling) {
    inputMessage.value = ''
    pageInfoList.value = []
    isShowPage.value = false
    taklToHtml.value = false
    type.value = FunctionList.File_Operation
  }
  controllerList.value[0].abort()
  controllerList.value = []
  sendLoading.value = false
}

function handleScroll(a) {
  const { wrapRef } = scrollbar.value
  const { scrollHeight, clientHeight } = wrapRef
  scrollToBottomRef.value.showButton = scrollHeight - a.scrollTop - clientHeight >= 350
  
  // 检测是否滚动到顶部
  if (a.scrollTop <= 10 && hasNext.value) {
    handleScrollToTop()
  }
}
const messagesContainer = ref(null)
let innerText = ''
// 处理滚动到顶部的事件
const handleScrollToTop = debounce(async function () {
  console.log('滚动到顶部，可以加载更多历史消息')
  if (!sendLoading.value && !isLoadingMore.value) {
    isLoadingMore.value = true
    try {
      // 记录当前第一条消息的位置
      const firstMessage = messagesContainer.value?.firstElementChild.querySelector('.timestamp')
      innerText = firstMessage.innerText
      const oldHeight = firstMessage?.offsetTop || 0
      await msgStore.changePage()
      setTimeout(() => {
        if (firstMessage) {
          console.log([...messagesContainer.value.querySelectorAll('.timestamp')].find(_ => _.innerText === innerText).offsetTop, innerText, 778);
          const newHeight = [...messagesContainer.value.querySelectorAll('.timestamp')].find(_ => _.innerText === innerText).offsetTop
          const scrollOffset = newHeight - oldHeight
          scrollbar.value?.setScrollTop(scrollOffset)
        }
      }, 400);
      // 恢复滚动位置
     
    } finally {
      isLoadingMore.value = false
    }
  }
}, 500, { leading: false, trailing: true })
/**
 * @param {string} msg
 * @param {string} raw
 * @param {string} type 发送的类型 document 、text
 * **/
async function addMessage(msg, raw, type) {
  console.log(msg);
  const newMessage = reactive({
    id:+new Date(),
    type: type || '',
    rawContent: raw ?? msg,
    senderId: userStore.token,
    receiverId:-1, //大模型
    content: msg,
    sortKey: moment().valueOf(),
    role: 'user',
    conversationId:msgUuid.value,
    addToHistory: `${!taklToHtml.value}`
  })
  if (type === 'document') {
    newMessage.content = msg
    newMessage.rawContent = raw
  }
  if (!msg) return
  messages.value.push(newMessage)
  await nextTick(() => {
    scrollbar.value?.setScrollTop(99999)
  })
  return newMessage
}

const handleSummary = async () => {
  let params = []
  if (AIModel.value.file === true) {
    pageInfoList.value.forEach((pageInfo) => {
      params.push({ role: 'system', content: `fileid://${pageInfo.fileId}` })
    })
    params.push({
      role: 'user', content: `
    请根据这几个文件的文本内容综合分析总结
    要求：
    1.抽取文件的文本内容
    2.对每个文件中的文本内容进行分别总结
    3.对步骤二总结的内容进行综合总结`
    })
  } else {
    let tempStr = ''
    pageInfoList.value.forEach((pageInfo, index) => {
      pageInfo.handleText = getSummaryPrompt(pageInfo.content)
      tempStr += `第${index + 1}段：${pageInfo.handleText}\n`
    })
    params = [{
      role: 'user',
      content: `请根据以下这几个内容综合总结出结果：
      ${tempStr}要求：
      1. 用简洁清晰的语言提取所有的核心要点
      2. 保持客观中立的语气
      3. 按重要性排序
      4. 返回内容做好换行，以及展示样式
      5. 请以"以下是对该文件内容的总结："开头，然后用要点的形式列出主要内容。
      6. 对这几个内容进行综合分析及联想`
    }]
  }
  const msg = await addMessage(JSON.stringify(pageInfoList.value), JSON.stringify(params.map(_ => _.content)), 'document')
  putChat(msg)
  if (requestFlowFn) {
    isShowPage.value = false
    taklToHtml.value = false
    pageInfoList.value = []
    const res = await requestFlowFn(params)
  }
}

async function handelIntelligentFillingClick() {
  isShowPage.value = true
  taklToHtml.value = true
  const tempPageInfo = await getPageInfo()
  pageInfo.value = tempPageInfo
  pageInfoList.value = [tempPageInfo]
  inputMessage.value = '/智能填表 '
  type.value = FunctionList.Intelligent_Form_filling
}

function handleCurrentChange(e) {
  options.forEach((item) => {
    item.options.forEach((item2) => {
      if (item2.value === e) {
        msgStore.updateAIModel(item2)
      }
    })
  })
  if (AIModel.value.file === true) {
    isShowPage.value = false
    taklToHtml.value = false
    pageInfoList.value = []
  }
}

async function handleCurrentData(e) {
  drawerRef.value.drawer = false
  if (!e) {
    addNewDialogue()
    return
  }
  messages.value = []
  page.value = 1
  msgUuid.value = e
  chrome.storage.local.set({ msgUuid: msgUuid.value })
  await msgStore.initMsg()
  scrollbar.value?.setScrollTop(99999)
}

async function readClick() {
  if (type.value === FunctionList.Intelligent_Form_filling) {
    pageInfoList.value = []
    type.value = FunctionList.File_Operation
  }
  isShowPage.value = true
  taklToHtml.value = true
  if (pageInfoList.value.length >= Number(import.meta.env.VITE_MAX_FILE_NUMBER)) {
    ElMessage.warning(`最多添加${import.meta.env.VITE_MAX_FILE_NUMBER}个文件`)
    return
  }
  const tempPageInfo = await getPageInfo()
  if (AIModel.value.file === true) {
    const blob = new Blob([tempPageInfo.content.mainContent], { type: 'text/plain' })
    const file = new File([blob], tempPageInfo.title + '.txt', { type: 'text/plain' })
    tempPageInfo.fileId = await modelFileUpload(file)
  }
  pageInfoList.value.unshift(tempPageInfo)
}

function deletePageInfo(i) {
  pageInfoList.value.splice(i, 1)
  if (pageInfoList.value.length === 0) {
    isShowPage.value = false
    taklToHtml.value = false
    if (type.value === FunctionList.Intelligent_Form_filling) inputMessage.value = ''
    type.value = FunctionList.File_Operation
  }
}

function addNewDialogue() {
  if (messages.value.length === 0) {
    ElMessage.warning('已经是新对话')
    return
  }
  isShowPage.value = false
  taklToHtml.value = false
  messages.value = []
  page.value = 1
  pageInfoList.value = []
  msgUuid.value = uuidv4()
  chrome.storage.local.set({ msgUuid: msgUuid.value })
}

async function handleAsk() {
  const str = inputMessage.value.trim()
  inputMessage.value = ''
  if (sendLoading.value) return
  const msg = await addMessage(str)
  putChat(msg)
  if (type.value === FunctionList.Intelligent_Form_filling) {
    const res = await fetchRes(str)
    if (res.status === 'ok') {
      formInfo.value = res.data
      console.log(res, 55558)
    } else {
      type.value = FunctionList.File_Operation
      isShowPage.value = false
      taklToHtml.value = false
      pageInfoList.value = []
    }
  } else streamRes(taklToHtml.value)
}

function handleCapture() {
  ElMessage({
    message: '开发中...',
    grouping: true,
    showClose: true
  })
}

function hisRecords() {
  drawerRef.value.drawer = true
}
const handleUpload = async (file) => {
  if (AIModel.value.file === true) {
    const id = await modelFileUpload(file)
    pageInfoList.value.unshift({
      fileId: id,
      title: file.name,
      url: 'File',
      favIconUrl: fileLogo
    })
    isShowPage.value = true
    taklToHtml.value = true
    return
  }
  if (type.value === FunctionList.Intelligent_Form_filling) {
    const fileExtension = file.name.split('.').pop().toLowerCase()
    if (fileExtension === 'xlsx') {
      const readData = await getXlsxValue(file)
      readData[0].forEach((header, i) => {
        // if (!xlsxData.value[header]) xlsxData.value[header] = []
        xlsxData.value[header] = readData[1][i]
      })
      const msg = addMessage(`已上传文件：${file.name}`, buildExcelUnderstandingPrompt(readData[0], file?.name, formInfo.value))
      putChat(msg)
      const { rawContent, status } = await streamRes()
      if (status === 'ok') {
        let form = []
        if (rawContent.includes('json')) form = JSON.parse(rawContent.split('json')[1].split('```')[0])
        else form = JSON.parse(rawContent)
        handleInput(xlsxData.value, form)
      }
    } else {
      try {
        sendLoading.value = true
        const msg = await addMessage(`文件上传中`)
        putChat(msg)
        const data = await getFileValue(file)
        msg.content = `已上传文件：${file.name}`
        msg.rawContent = data
        const res2 = await getFormKeyAndValue(data, formInfo.value)
        xlsxData.value = res2.data
        msg.rawContent = buildObjPrompt(res2.data, formInfo.value)
        const { rawContent, status } = await streamRes()
        if (status === 'ok') {
          let form = []
          if (rawContent.includes('json')) form = JSON.parse(rawContent.split('json')[1].split('```')[0])
          else form = JSON.parse(rawContent)
          handleInput(xlsxData.value, form)
        }
      } catch (error) {
        msg.content = '文件解析出错'
      } finally {
        sendLoading.value = false
        // putChat({
        //   id: msgUuid.value,
        //   msg: msg
        // })
      }
    }
    type.value = FunctionList.File_Operation
    isShowPage.value = false
    taklToHtml.value = false
    pageInfoList.value = []
    return
  }
  if (type.value === FunctionList.File_Operation) {
    let formData = new FormData()
    formData.append('file', file)
    const res = await getFileContent(formData)
    pageInfoList.value.unshift({
      title: file.name,
      url: 'File',
      favIconUrl: fileLogo,
      content: {
        mainContent: res.data
      }
    })
    isShowPage.value = true
  }
}
async function getFileValue(file) {
  let formData = new FormData()
  formData.append('file', file)
  const res = await getFileContent(formData)
  return res.data
}
let a = null
// 组件挂载时滚动到底部
onMounted(async () => {
  msgStore.updateAIModel(options[0].options[0])
  await msgStore.initMsg()
  useAutoResizeTextarea(tareRef, inputMessage)
  chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
    if (message.type === 'TO_SIDE_PANEL_PAGE_INFO') {
      pageInfo.value = message.data
      // 转发到 content.js
      chrome.tabs.sendMessage(sender.tab.id, {
        type: 'TO_CONTENT_SCRIPT',
        data: message.data
      })
    }
    if (message.type === 'TO_SIDE_PANEL_PAGE_CHANGE') {
      pageInfo.value = message.data
    }
  })
  nextTick(() => {
    scrollbar.value?.setScrollTop(99999)
  })
})
</script>

<style lang="scss" scoped>
@use '@/entrypoints/sidepanel/css/chat.scss';

.loading-more-indicator {
  display: flex;
  align-items: center;
  justify-content: center;
  padding: 10px 0;
  color: #909399;
  font-size: 14px;
  
  .loading-spinner {
    width: 20px;
    height: 20px;
    margin-right: 8px;
    border: 2px solid #e6e6e6;
    border-top-color: #4d6bfe;
    border-radius: 50%;
    animation: spin 1s linear infinite;
  }
}

@keyframes spin {
  to {
    transform: rotate(360deg);
  }
}
</style>