feat: 支持多聊天合并导出

2026-08-17 11:37:06 +08:00 · 2026-08-04 19:26:18 +08:00
parent 0a3d930298
commit 60c501e148
18 changed files with 1926 additions and 287 deletions
@@ -114,6 +114,38 @@ test('NAV-01 NAV-02 every top-level page is unique and switchable', async () =>
  }
 })

+test('EXPORT-01 multi-chat selection stays local to export and forces HTML', async () => {
+  const fixture = await launchTestApp()
+  try {
+    const navigation = fixture.page.getByRole('navigation', { name: '一级导航' })
+    await fixture.page.getByRole('button', { name: '联系人 (1)' }).click()
+    await fixture.page.getByText('文件传输助手', { exact: true }).click()
+    await expect(fixture.page.getByText('转发多条内容', { exact: true })).toBeVisible()
+
+    await navigation.getByRole('button', { name: '导出' }).click()
+    const contactList = fixture.page.locator('.export-contact-list')
+    await expect(contactList.getByRole('button', { name: /文件传输助手/ })).toHaveAttribute(
+      'aria-pressed',
+      'true'
+    )
+    await fixture.page.getByRole('button', { name: '+ 添加聊天' }).click()
+    await contactList.getByRole('button', { name: /产品测试群/ }).click()
+
+    await expect(fixture.page.getByText('已选 2 / 5 个')).toBeVisible()
+    await expect(fixture.page.getByRole('button', { name: 'CSV' })).toBeDisabled()
+    await expect(fixture.page.getByRole('button', { name: /HTML/ })).toHaveClass(/active/)
+    await expect(fixture.page.getByText('2 个聊天 · 合并预览')).toBeVisible()
+    await expect(
+      fixture.page.locator('.export-preview-bubble').filter({ hasText: '这是一条脱敏测试消息' })
+    ).toBeVisible()
+
+    await navigation.getByRole('button', { name: '档案' }).click()
+    await expect(fixture.page.getByText('转发多条内容', { exact: true })).toBeVisible()
+  } finally {
+    await fixture.close()
+  }
+})
+
 test('ARCH-01 ARCH-02 folded chats and supported message types are represented explicitly', async () => {
  const fixture = await launchTestApp()
  try {
@@ -0,0 +1,170 @@
+import { expect, test } from '@playwright/test'
+import { execFileSync } from 'child_process'
+import { createWriteStream, mkdirSync, mkdtempSync, rmSync, writeFileSync } from 'fs'
+import { tmpdir } from 'os'
+import { dirname, join } from 'path'
+import { pathToFileURL } from 'url'
+import { ZipArchive } from 'archiver'
+import { renderExportPage } from '../../src/main/export-html-template'
+import type { Message } from '../../src/shared/types'
+
+const archiveMessage = (
+  id: string,
+  conversationId: string,
+  conversationName: string,
+  content: string,
+  createTime: number
+): Message => ({
+  id,
+  from: 'user',
+  type: '普通文本',
+  datetime: '',
+  content,
+  isSender: false,
+  name: '脱敏成员',
+  createTime,
+  exportConversationId: conversationId,
+  exportConversationName: conversationName
+})
+
+const zipDirectory = async (
+  sourceDir: string,
+  zipPath: string,
+  folderName: string
+): Promise<void> => {
+  const output = createWriteStream(zipPath)
+  const archive = new ZipArchive({ zlib: { level: 6 } })
+  await new Promise<void>((resolve, reject) => {
+    output.on('close', resolve)
+    output.on('error', reject)
+    archive.on('error', reject)
+    archive.pipe(output)
+    archive.directory(sourceDir, folderName)
+    void archive.finalize().catch(reject)
+  })
+}
+
+test('EXPORT-ARCHIVE-01 merged v2 archive is usable offline on desktop and mobile', async ({
+  page
+}, testInfo) => {
+  const fixtureRoot = mkdtempSync(join(tmpdir(), 'wxe-merged-archive-e2e-'))
+  const outputDir = join(fixtureRoot, 'source')
+  try {
+    const dataPath = join(outputDir, 'data', 'messages.js')
+    mkdirSync(dirname(dataPath), { recursive: true })
+    writeFileSync(join(outputDir, 'index.html'), renderExportPage('合并聊天档案'), 'utf8')
+    writeFileSync(
+      dataPath,
+      `window.__WECHAT_EXPORT__ = ${JSON.stringify({
+        version: 2,
+        name: '合并聊天档案',
+        exportedAt: '2026-08-04T00:00:00.000Z',
+        conversations: [
+          { id: 'alpha', name: '项目群', type: 'group', messageCount: 2 },
+          { id: 'beta', name: '文件传输助手', type: 'user', messageCount: 1 }
+        ],
+        messages: [
+          archiveMessage('alpha-1', 'alpha', '项目群', '项目群第一条', 1_764_547_200),
+          archiveMessage('beta-1', 'beta', '文件传输助手', '个人聊天消息', 1_769_904_000),
+          archiveMessage('alpha-2', 'alpha', '项目群', '项目群第二条', 1_769_990_400)
+        ]
+      })};\n`,
+      'utf8'
+    )
+    const zipPath = join(fixtureRoot, 'merged-archive.zip')
+    const extractedDir = join(fixtureRoot, 'extracted')
+    await zipDirectory(outputDir, zipPath, '合并聊天档案')
+    mkdirSync(extractedDir, { recursive: true })
+    execFileSync('unzip', ['-q', zipPath, '-d', extractedDir])
+    const offlineIndex = join(extractedDir, '合并聊天档案', 'index.html')
+
+    await page.setViewportSize({ width: 1440, height: 900 })
+    await page.goto(pathToFileURL(offlineIndex).href)
+    const conversationSelect = page.getByLabel('筛选聊天')
+    await expect(conversationSelect).toHaveValue('all')
+    await expect(conversationSelect.locator('option')).toHaveCount(3)
+    await expect(page.locator('#archive-title')).toBeHidden()
+    await expect(page.locator('.archive-heading #conversation-filter')).toBeVisible()
+    await expect(page.locator('#archive-meta')).toHaveText(/^更新于 /)
+    await expect(page.locator('#archive-meta')).not.toContainText('条消息')
+    await expect(page.locator('.message')).toHaveCount(3)
+    await expect(page.locator('.conversation-source')).toHaveCount(3)
+    expect(
+      await page.evaluate(() => document.documentElement.scrollWidth <= window.innerWidth)
+    ).toBe(true)
+    await page.screenshot({ path: testInfo.outputPath('merged-archive-1440.png'), fullPage: true })
+
+    await conversationSelect.selectOption('beta')
+    await expect(page.locator('.message')).toHaveCount(1)
+    await expect(page.locator('.conversation-source')).toHaveCount(0)
+
+    await conversationSelect.selectOption('all')
+    await page.setViewportSize({ width: 390, height: 844 })
+    await expect(page.locator('.timeline-year')).toHaveCount(2)
+    await expect(page.locator('.timeline-year').first()).toBeVisible()
+    await expect(page.locator('.timeline-year').first()).toHaveText('2025 年')
+    const positions = await page.evaluate(() => {
+      const conversations = document.querySelector('#conversation-filter')!.getBoundingClientRect()
+      const toolbar = document.querySelector('.toolbar')!.getBoundingClientRect()
+      const timeline = document.querySelector('#timeline')!.getBoundingClientRect()
+      return {
+        conversationTop: conversations.top,
+        conversationBottom: conversations.bottom,
+        toolbarTop: toolbar.top,
+        toolbarBottom: toolbar.bottom,
+        timelineTop: timeline.top,
+        documentWidth: document.documentElement.scrollWidth,
+        viewportWidth: window.innerWidth
+      }
+    })
+    expect(positions.conversationTop).toBeGreaterThanOrEqual(positions.toolbarTop)
+    expect(positions.conversationBottom).toBeLessThanOrEqual(positions.toolbarBottom)
+    expect(positions.timelineTop).toBeGreaterThanOrEqual(positions.toolbarBottom)
+    expect(positions.documentWidth).toBeLessThanOrEqual(positions.viewportWidth)
+    await expect(page.locator('.message')).toHaveCount(3)
+    await page.screenshot({ path: testInfo.outputPath('merged-archive-390.png'), fullPage: true })
+  } finally {
+    rmSync(fixtureRoot, { recursive: true, force: true })
+  }
+})
+
+test('EXPORT-ARCHIVE-02 legacy single-chat archive keeps its original layout', async ({
+  page
+}, testInfo) => {
+  const outputDir = mkdtempSync(join(tmpdir(), 'wxe-single-archive-e2e-'))
+  try {
+    const dataPath = join(outputDir, 'data', 'messages.js')
+    mkdirSync(dirname(dataPath), { recursive: true })
+    writeFileSync(join(outputDir, 'index.html'), renderExportPage('单聊天档案'), 'utf8')
+    writeFileSync(
+      dataPath,
+      `window.__WECHAT_EXPORT__ = ${JSON.stringify({
+        version: 1,
+        sourceId: 'single',
+        name: '单聊天档案',
+        exportedAt: '2026-08-04T00:00:00.000Z',
+        messages: [archiveMessage('single-1', 'single', '单聊天档案', '单聊天消息', 1_767_225_600)]
+      })};\n`,
+      'utf8'
+    )
+
+    await page.setViewportSize({ width: 1440, height: 900 })
+    await page.goto(pathToFileURL(join(outputDir, 'index.html')).href)
+    await expect(page.locator('#conversation-filter')).toBeHidden()
+    await expect(page.locator('#archive-title')).toBeVisible()
+    await expect(page.locator('#archive-title')).toHaveText('单聊天档案')
+    await expect(page.locator('#archive-meta')).toContainText('1 条消息 · 更新于 ')
+    await expect(page.locator('.archive-layout')).toHaveClass(/single-conversation/)
+    await expect(page.locator('.message')).toHaveCount(1)
+    await page.screenshot({ path: testInfo.outputPath('single-archive-1440.png'), fullPage: true })
+
+    await page.setViewportSize({ width: 390, height: 844 })
+    expect(
+      await page.evaluate(() => document.documentElement.scrollWidth <= window.innerWidth)
+    ).toBe(true)
+    await expect(page.locator('.message')).toHaveCount(1)
+    await page.screenshot({ path: testInfo.outputPath('single-archive-390.png'), fullPage: true })
+  } finally {
+    rmSync(outputDir, { recursive: true, force: true })
+  }
+})