Function useAttachment

src/renderer/src/hooks/useAttachment.ts:54–218

(pageId: string)

}

export function useAttachment(pageId: string): UseAttachmentResult {
  const { getState, addPendingAttachment, removePendingAttachment, clearPendingAttachments } =
    useChatUIStore()
  const pendingAttachments = getState(pageId).pendingAttachments

  // 从拖拽或粘贴的 File 对象添加附件
  const addAttachments = useCallback(
    async (files: FileList | File[]) => {
      const fileArray = Array.from(files)

      for (const file of fileArray) {
        // 类型校验
        if (!ALLOWED_TYPES.includes(file.type)) {
          console.warn(`不支持的文件类型: ${file.type}`)
          continue
        }

        // 大小校验
        if (file.size > MAX_SIZE) {
          console.warn(`文件过大: ${file.name} (${(file.size / 1024 / 1024).toFixed(2)}MB)`)
          continue
        }

        const fileId = generateId()

        // 读取文件为 base64
        const base64Content = await readFileAsBase64(file)

        // 保存到 temp 目录
        const result = await window.api.attachment.save({
          fileId,
          fileName: file.name,
          base64Content
        })

        if (result.success && result.localPath) {
          const attachment: FileAttachment = {
            id: fileId,
            name: file.name,
            type: file.type,
            size: file.size,
            localPath: result.localPath,
            createdAt: Date.now()
          }
          addPendingAttachment(pageId, attachment)
        } else {
          console.error('保存附件失败:', result.error)
        }
      }
    },
    [pageId, addPendingAttachment]
  )

  // 从文件选择器添加附件
  const addAttachmentsFromSelector = useCallback(async () => {
    const result = await window.api.selectFiles({
      multiple: true,
      filters: [{ name: 'Images', extensions: ['jpg', 'jpeg', 'png', 'gif', 'webp'] }]