refactor: 重构useWenku.js

src/common/wangeditor-customer/menu/common/AIWriteModal.jsx

@@ -11,7 +11,7 @@ import {
 import { ConfigProvider, Modal, Input, Button } from 'antd'
 const { TextArea } = Input
 import './AISearchModal.less'
-import { usePaperOutline, useOutlineToPaper } from '@/hooks/useWenku'
+import { usePaper } from '@/hooks/useWenku'
 import { useCopyToClipboard } from 'react-use'
 
 export default function AIModal() {
@@ -20,7 +20,7 @@ export default function AIModal() {
   const [textIndent, setTextIndent] = useState(0)
   const prePromptRef = useRef(null)
   const messageScrollRef = useRef(null)
-  const { messages, setMessages, isLoading, query } = usePaperOutline()
+  const { messages, setMessages, isLoading, generateOutline, generatePaper } = usePaper()
   useEffect(() => {
     if (prePromptRef.current) {
       const width = prePromptRef.current.offsetWidth + 10
@@ -43,7 +43,7 @@ export default function AIModal() {
     }
   }
   const handleSearch = () => {
-    query(prePrompt + content)
+    generateOutline(prePrompt + content)
     setContent('')
   }
 
@@ -55,13 +55,11 @@ export default function AIModal() {
   }
 
   // 生成论文
-  const { query: outlineToPaper } = useOutlineToPaper()
-
   const handleGeneratePaper = async (msg) => {
     setMessages((prevMessages) => {
       return [...prevMessages, { content: '正在生成长文...', role: 'ai', tips: '预计10分钟', queryID: msg.queryID }]
     })
-    const paper = await outlineToPaper({ userQuery: msg.userQuery, queryID: msg.queryID, outline: msg.content })
+    const paper = await generatePaper({ userQuery: msg.userQuery, queryID: msg.queryID, outline: msg.content })
     setMessages((prevMessages) => {
       prevMessages.pop()
       return [...prevMessages, { content: '已为您生成初稿，请点击查看', role: 'ai', queryID: msg.queryID, paper }]
@@ -101,7 +99,7 @@ export default function AIModal() {
                   size="small"
                   icon={<UndoOutlined />}
                   disabled={isLoading}
-                  onClick={() => query(msg.userQuery)}>
+                  onClick={() => generateOutline(msg.userQuery)}>
                   换个大纲
                 </Button>
                 <Button

src/hooks/useAsyncTask.js

+import { useState, useCallback } from 'react'
+
+/**
+ * 通用异步任务处理hook
+ * @param {Function} taskFn 异步任务函数
+ * @returns {Object} { isLoading, error, execute }
+ */
+export function useAsyncTask(taskFn) {
+  const [isLoading, setIsLoading] = useState(false)
+  const [error, setError] = useState(null)
+
+  const execute = useCallback(
+    async (...args) => {
+      try {
+        setIsLoading(true)
+        setError(null)
+        return await taskFn(...args)
+      } catch (err) {
+        setError(err)
+        throw err
+      } finally {
+        setIsLoading(false)
+      }
+    },
+    [taskFn]
+  )
+
+  return { isLoading, error, execute }
+}

src/hooks/useWenku.js

 import { useState, useEffect, useRef } from 'react'
 import { aiSearch, paperOutline, outlineToPaper, download } from '@/api/wenku'
+import { useAsyncTask } from './useAsyncTask'
+
+/**
+ * @typedef {Object} Message
+ * @property {string} content
+ * @property {'user' | 'ai'} role
+ * @property {Array<{title: string, url: string}>} [searchReferList]
+ * @property {string} [userQuery]
+ * @property {Chapter[]} [chapters]
+ * @property {string} [logID]
+ * @property {string} [queryID]
+ */
+
+/**
+ * @typedef {Object} Chapter
+ * @property {string} title
+ * @property {number} level
+ * @property {string} desc
+ * @property {string} chapter
+ */
+
+/**
+ * @typedef {Object} Paper
+ * @property {string} content
+ * @property {string} [downloadLink]
+ * @property {string} [docID]
+ */
+
+/**
+ * @typedef {Object} DownloadResult
+ * @property {string} download_link
+ */
 
 // AI搜索
 export function useSearch() {
   const [messages, setMessages] = useState([])
-  const [isLoading, setIsLoading] = useState(false)
-  const query = async (userQuery) => {
-    try {
-      setIsLoading(true)
-      setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
-      const currentMessage = { content: '', searchReferList: [], role: 'ai', userQuery }
-      await aiSearch({
-        body: JSON.stringify({ userQuery }),
-        onmessage(message) {
-          try {
-            const data = JSON.parse(message.data)
-            const content = data.raw?.content || ''
-            const searchReferList = data.raw?.searchReferList || []
-            currentMessage.content += content
-            if (searchReferList.length) {
-              currentMessage.searchReferList.push(...searchReferList)
-              setMessages((prevMessages) => {
-                const lastMessage = prevMessages[prevMessages.length - 1]
-                if (lastMessage?.role === 'ai') {
-                  return [...prevMessages.slice(0, -1), currentMessage]
-                }
-                return [...prevMessages, currentMessage]
-              })
-            }
-          } catch (error) {
-            console.log(error)
+
+  const { isLoading, execute } = useAsyncTask(async (userQuery) => {
+    setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
+    const currentMessage = { content: '', searchReferList: [], role: 'ai', userQuery }
+
+    await aiSearch({
+      body: JSON.stringify({ userQuery }),
+      onmessage(message) {
+        try {
+          const data = JSON.parse(message.data)
+          const content = data.raw?.content || ''
+          const searchReferList = data.raw?.searchReferList || []
+          currentMessage.content += content
+          if (searchReferList.length) {
+            currentMessage.searchReferList.push(...searchReferList)
+            setMessages((prevMessages) => {
+              const lastMessage = prevMessages[prevMessages.length - 1]
+              if (lastMessage?.role === 'ai') {
+                return [...prevMessages.slice(0, -1), currentMessage]
+              }
+              return [...prevMessages, currentMessage]
+            })
           }
-        },
-      })
-    } catch (error) {
-      console.log(error)
-    } finally {
-      setIsLoading(false)
-    }
-  }
-  return { messages, setMessages, isLoading, query }
+        } catch (error) {
+          console.error('Error parsing message:', error)
+        }
+      },
+    })
+  })
+
+  return { messages, setMessages, isLoading, query: execute }
 }
 
 function parseInput(inputStr) {
@@ -95,76 +122,21 @@ function parseInput(inputStr) {
   return result
 }
 
-// 生成论文大纲
-export function usePaperOutline() {
+/**
+ * 论文生成全流程管理
+ * @returns {{
+ *   messages: Message[],
+ *   paper: Paper,
+ *   isLoading: boolean,
+ *   generateOutline: (userQuery: string) => Promise<void>,
+ *   generatePaper: (data: any) => Promise<Paper>,
+ *   downloadPaper: (docID: string) => Promise<string>
+ * }}
+ */
+export function usePaper() {
   const [messages, setMessages] = useState([])
-  const [isLoading, setIsLoading] = useState(false)
-  const query = async (userQuery) => {
-    try {
-      setIsLoading(true)
-      setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
-      const currentMessage = { content: '', role: 'ai', userQuery, chapters: [], logID: '', queryID: '' }
-      await paperOutline({
-        body: JSON.stringify({ userQuery }),
-        onmessage(message) {
-          try {
-            const data = JSON.parse(message.data)
-            const content = data.raw?.data || ''
-            if (data.logID) currentMessage.logID = data.logID
-            if (data.queryID) currentMessage.queryID = data.queryID
-            currentMessage.content += content
-            currentMessage.chapters = parseInput(currentMessage.content)
-            if (content) {
-              setMessages((prevMessages) => {
-                const lastMessage = prevMessages[prevMessages.length - 1]
-                if (lastMessage?.role === 'ai') {
-                  return [...prevMessages.slice(0, -1), currentMessage]
-                }
-                return [...prevMessages, currentMessage]
-              })
-            }
-          } catch (error) {
-            console.log(error)
-          }
-        },
-      })
-    } catch (error) {
-      console.log(error)
-    } finally {
-      setIsLoading(false)
-    }
-  }
-  return { messages, setMessages, isLoading, query }
-}
-
-// 大纲生成论文
-export function useOutlineToPaper() {
-  const { queryWithPolling } = useDownload()
-  const [isLoading, setIsLoading] = useState(false)
-  const [paper, setPaper] = useState('')
-  const query = async (data) => {
-    try {
-      setIsLoading(true)
-      const { raw } = await outlineToPaper(data)
-      setPaper(raw)
-      const downloadLink = await queryWithPolling({
-        docID: '280fe43ee63a580216fc700abb68a98271feac83' || raw.docID,
-      })
-      setPaper({ ...raw, downloadLink })
-      return { ...raw, downloadLink }
-    } catch (error) {
-      console.log(error)
-    } finally {
-      setIsLoading(false)
-    }
-  }
-  return { paper, setPaper, isLoading, query }
-}
-
-// 文档下载
-export function useDownload() {
+  const [paper, setPaper] = useState(null)
   const [downloadLink, setDownloadLink] = useState('')
-  const [isLoading, setIsLoading] = useState(false)
   const timerRef = useRef(null)
 
   const clearTimer = () => {
@@ -174,25 +146,58 @@ export function useDownload() {
     }
   }
 
-  const query = async (data) => {
-    try {
-      setIsLoading(true)
-      const { raw } = await download(data)
-      if (raw) {
-        setDownloadLink(raw.download_link)
-        return raw.download_link
-      }
-    } catch (error) {
-      console.error(error)
-    } finally {
-      setIsLoading(false)
+  const { isLoading: isOutlineLoading, execute: generateOutline } = useAsyncTask(async (userQuery) => {
+    setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
+    const currentMessage = { content: '', role: 'ai', userQuery, chapters: [], logID: '', queryID: '' }
+
+    await paperOutline({
+      body: JSON.stringify({ userQuery }),
+      onmessage(message) {
+        try {
+          const data = JSON.parse(message.data)
+          const content = data.raw?.data || ''
+          if (data.logID) currentMessage.logID = data.logID
+          if (data.queryID) currentMessage.queryID = data.queryID
+          currentMessage.content += content
+          currentMessage.chapters = parseInput(currentMessage.content)
+          if (content) {
+            setMessages((prevMessages) => {
+              const lastMessage = prevMessages[prevMessages.length - 1]
+              if (lastMessage?.role === 'ai') {
+                return [...prevMessages.slice(0, -1), currentMessage]
+              }
+              return [...prevMessages, currentMessage]
+            })
+          }
+        } catch (error) {
+          console.error('Error parsing message:', error)
+        }
+      },
+    })
+  })
+
+  const { isLoading: isPaperLoading, execute: generatePaper } = useAsyncTask(async (data) => {
+    const { raw } = await outlineToPaper(data)
+    setPaper(raw)
+    const downloadLink = await downloadWithPolling({
+      docID: raw.docID,
+    })
+    setPaper({ ...raw, downloadLink })
+    return { ...raw, downloadLink }
+  })
+
+  const { isLoading: isDownloadLoading, execute: downloadPaper } = useAsyncTask(async (docID) => {
+    const { raw } = await download({ docID })
+    if (raw) {
+      setDownloadLink(raw.download_link)
+      return raw.download_link
     }
-  }
+  })
 
-  const queryWithPolling = async (data, interval = 3000) => {
+  const downloadWithPolling = async (data, interval = 3000) => {
     const poll = async (resolve, reject) => {
       try {
-        const link = await query(data)
+        const link = await downloadPaper(data.docID)
         if (link) {
           resolve(link)
           return
@@ -209,11 +214,17 @@ export function useDownload() {
   }
 
   useEffect(() => {
-    // 清除定时器的逻辑
-    return () => {
-      clearTimer()
-    }
+    return () => clearTimer()
   }, [])
 
-  return { downloadLink, isLoading, query, queryWithPolling, clearTimer }
+  return {
+    messages,
+    setMessages,
+    paper,
+    downloadLink,
+    isLoading: isOutlineLoading || isPaperLoading || isDownloadLoading,
+    generateOutline,
+    generatePaper,
+    downloadPaper,
+  }
 }