Class AttachmentHandler

src/main/attachmentHandler.ts:9–200

* 附件处理器 * 管理文件附件的本地存储

 * 管理文件附件的本地存储
 */
class AttachmentHandler {
  private attachmentsDir: string

  constructor() {
    // 附件存储目录：应用数据目录/attachments
    this.attachmentsDir = path.join(app.getPath('userData'), 'attachments')
  }

  /**
   * 保存附件到本地
   * @param fileId 文件ID
   * @param fileName 文件名
   * @param base64Content base64编码的文件内容
   * @param pageId 页面ID（可选）
   * @param messageId 消息ID（可选）
   * @returns 相对路径
   */
  async saveAttachment(
    fileId: string,
    fileName: string,
    base64Content: string,
    pageId?: string,
    messageId?: string
  ): Promise<{ success: boolean; localPath?: string; error?: string }> {
    try {
      // 确定存储路径
      let targetDir: string
      if (pageId && messageId) {
        targetDir = path.join(this.attachmentsDir, pageId, messageId)
      } else {
        targetDir = path.join(this.attachmentsDir, 'temp')
      }

      // 创建目录
      await mkdir(targetDir, { recursive: true })

      // 提取文件扩展名
      const ext = path.extname(fileName)
      const filePath = path.join(targetDir, `${fileId}${ext}`)

      // 写入文件
      const buffer = Buffer.from(base64Content, 'base64')
      await writeFile(filePath, buffer)

      // 返回相对路径
      const relativePath = path.relative(this.attachmentsDir, filePath)
      return { success: true, localPath: relativePath }
    } catch (error) {
      console.error('保存附件失败:', error)
      return {
        success: false,
        error: error instanceof Error ? error.message : '保存附件失败'
      }
    }
  }

  /**
   * 读取附件内容