' + esc(fullTime(message)) + '
' +
+ '
-
+ {selectionMode && (
+ ' + esc(fullTime(message)) + source + '
' +
(isSystem ? '' : avatar) + '
'
}
+ const renderConversations = () => {
+ if (conversations.length <= 1) {
+ conversationFilter.hidden = true
+ title.hidden = false
+ layout.classList.add('single-conversation')
+ return
+ }
+ const counts = new Map()
+ for (const message of allMessages) {
+ const id = message.exportConversationId || ''
+ counts.set(id, (counts.get(id) || 0) + 1)
+ }
+ const option = (id, label, total) =>
+ ''
+ title.hidden = true
+ conversationFilter.hidden = false
+ conversationSelect.innerHTML = option('all', '全部聊天', allMessages.length) +
+ conversations.map((conversation) => option(
+ conversation.id,
+ conversation.name,
+ counts.get(conversation.id) || 0
+ )).join('')
+ conversationSelect.value = activeConversation
+ }
+
const renderTimeline = () => {
if (filtered.length === 0) {
timeline.innerHTML = '' +
(isSystem ? '' : esc(sender)) + '
' + media + audio + quote +
'' + content + '
' + mediaStatus + '没有可跳转的月份
'
@@ -431,7 +556,11 @@ const renderExportScript = (name: string): string => `
}
const updateCount = () => {
const shown = Math.max(0, windowEnd - windowStart)
- count.textContent = '已显示 ' + shown + ' / 筛选 ' + filtered.length + ' / 全部 ' + allMessages.length
+ const scopeTotal = activeConversation === 'all'
+ ? allMessages.length
+ : allMessages.filter((message) => message.exportConversationId === activeConversation).length
+ const scopeLabel = activeConversation === 'all' ? '全部' : '当前聊天'
+ count.textContent = '已显示 ' + shown + ' / 筛选 ' + filtered.length + ' / ' + scopeLabel + ' ' + scopeTotal
}
const setScrollTop = (value) => {
scrollLoadSuppressed = true
@@ -472,6 +601,7 @@ const renderExportScript = (name: string): string => `
const applyFilters = () => {
const term = query.value.trim().toLowerCase()
filtered = allMessages.filter((message) =>
+ (activeConversation === 'all' || message.exportConversationId === activeConversation) &&
(activeKind === 'all' || kindOf(message) === activeKind) &&
(!term || searchText(message).includes(term))
)
@@ -540,6 +670,10 @@ const renderExportScript = (name: string): string => `
filters.querySelectorAll('[data-kind]').forEach((item) => item.classList.toggle('active', item === button))
applyFilters()
})
+ conversationSelect.addEventListener('change', () => {
+ activeConversation = conversationSelect.value
+ applyFilters()
+ })
timeline.addEventListener('click', (event) => {
const button = event.target.closest('[data-month]')
if (button) jumpToMonth(button.dataset.month)
@@ -577,10 +711,12 @@ const renderExportScript = (name: string): string => `
})
title.textContent = archive.name || ${JSON.stringify(name)}
- meta.textContent = allMessages.length.toLocaleString() + ' 条消息 · 更新于 ' +
- (archive.exportedAt
- ? new Date(archive.exportedAt).toLocaleString('zh-CN', { hour12: false })
- : '未知时间')
+ const updatedAt = archive.exportedAt
+ ? new Date(archive.exportedAt).toLocaleString('zh-CN', { hour12: false })
+ : '未知时间'
+ meta.textContent = (conversations.length > 1 ? '' : allMessages.length.toLocaleString() + ' 条消息 · ') +
+ '更新于 ' + updatedAt
+ renderConversations()
applyFilters()
})()
`
@@ -597,8 +733,11 @@ export function renderExportPage(name: string): string {
+
-
+
diff --git a/src/main/export-service.ts b/src/main/export-service.ts
index 461d85b..bfdf78c 100644
--- a/src/main/export-service.ts
+++ b/src/main/export-service.ts
@@ -1,14 +1,16 @@
import { app, BrowserWindow, shell } from 'electron'
import { createHash } from 'crypto'
-import { promises as fs } from 'fs'
+import { createWriteStream, promises as fs } from 'fs'
import { extname, join } from 'path'
import { fileURLToPath } from 'url'
+import { ZipArchive, type Archiver } from 'archiver'
import * as chat from './services/chat-service'
import type {
ExportJobProgress,
ExportMessageKind,
ExportRequest,
- ExportResult
+ ExportResult,
+ ExportTarget
} from '../shared/export'
import type { Message } from '../shared/types'
import { VoiceService } from './voice-service'
@@ -22,8 +24,18 @@ import { FileAssetService } from './file-asset-service'
import { mergeCachedSelfInfo } from './services/bootstrap-cache'
const jobs = new Set()
+const activeArchives = new Map()
const safeFilePart = (value: string): string =>
value.replace(/[\\/:*?"<>|]/g, '_').trim() || '聊天档案'
+const copyWritableExportFile = async (source: string, destination: string): Promise => {
+ try {
+ await fs.chmod(destination, 0o644)
+ } catch (error) {
+ if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error
+ }
+ await fs.copyFile(source, destination)
+ await fs.chmod(destination, 0o644)
+}
const exportStamp = (): string => {
const date = new Date()
const pad = (value: number): string => String(value).padStart(2, '0')
@@ -31,7 +43,15 @@ const exportStamp = (): string => {
}
const imageKeys = new ImageKeyConfigService()
-export interface HtmlExportArchive {
+export interface HtmlExportConversation {
+ id: string
+ name: string
+ type: 'user' | 'group'
+ avatarUrl?: string
+ messageCount: number
+}
+
+interface HtmlExportArchiveV1 {
version: 1
sourceId: string
name: string
@@ -39,20 +59,29 @@ export interface HtmlExportArchive {
messages: Message[]
}
+export interface HtmlExportArchive {
+ version: 2
+ name: string
+ exportedAt: string
+ conversations: HtmlExportConversation[]
+ messages: Message[]
+}
+
const archiveDataPrefix = 'window.__WECHAT_EXPORT__ = '
const hashPart = (value: string, length = 16): string =>
createHash('sha1').update(value).digest('hex').slice(0, length)
export const exportMessageKey = (message: Message, sourceId = ''): string => {
+ const conversationId = message.exportConversationId || sourceId
const sessionId = message.sessionId || sourceId
if (message.localId && message.createTime) {
- return `${sessionId}:local:${message.localId}:${message.createTime}`
+ return `${conversationId}:${sessionId}:local:${message.localId}:${message.createTime}`
}
- if (message.serverId) return `${sessionId}:server:${message.serverId}`
+ if (message.serverId) return `${conversationId}:${sessionId}:server:${message.serverId}`
if (message.id && message.createTime && !/^0\.\d+$/.test(message.id)) {
- return `${sessionId}:id:${message.id}:${message.createTime}`
+ return `${conversationId}:${sessionId}:id:${message.id}:${message.createTime}`
}
- return `${sessionId}:fallback:${hashPart(
+ return `${conversationId}:${sessionId}:fallback:${hashPart(
JSON.stringify([
message.createTime || 0,
message.senderId || '',
@@ -89,7 +118,8 @@ const mergeArchiveMessage = (previous: Message, current: Message): Message => {
export function mergeHtmlArchiveMessages(
previous: Message[],
current: Message[],
- sourceId = ''
+ sourceId = '',
+ conversationOrder: string[] = []
): Message[] {
const merged = new Map()
for (const message of previous) merged.set(exportMessageKey(message, sourceId), message)
@@ -101,6 +131,10 @@ export function mergeHtmlArchiveMessages(
return Array.from(merged.values()).sort((left, right) => {
const byTime = Number(left.createTime || 0) - Number(right.createTime || 0)
if (byTime !== 0) return byTime
+ const byConversation =
+ conversationOrder.indexOf(left.exportConversationId || sourceId) -
+ conversationOrder.indexOf(right.exportConversationId || sourceId)
+ if (byConversation !== 0) return byConversation
return Number(left.localId || 0) - Number(right.localId || 0)
})
}
@@ -127,7 +161,7 @@ export function normalizeHtmlArchiveSelfNames(
export async function readHtmlArchive(
outputDir: string,
- sourceId: string,
+ targets: ExportTarget[],
name: string
): Promise {
const dataPath = join(outputDir, 'data', 'messages.js')
@@ -136,7 +170,18 @@ export async function readHtmlArchive(
source = await fs.readFile(dataPath, 'utf8')
} catch (error) {
if ((error as NodeJS.ErrnoException).code === 'ENOENT') {
- return { version: 1, sourceId, name, exportedAt: new Date(0).toISOString(), messages: [] }
+ return {
+ version: 2,
+ name,
+ exportedAt: new Date(0).toISOString(),
+ conversations: targets.map((target) => ({
+ id: target.userMd5,
+ name: target.name,
+ type: target.type,
+ messageCount: 0
+ })),
+ messages: []
+ }
}
throw error
}
@@ -146,21 +191,100 @@ export async function readHtmlArchive(
.slice(assignment + 1)
.trim()
.replace(/;\s*$/, '')
- let archive: HtmlExportArchive
+ let parsed: HtmlExportArchive | HtmlExportArchiveV1
try {
- archive = JSON.parse(json) as HtmlExportArchive
+ parsed = JSON.parse(json) as HtmlExportArchive | HtmlExportArchiveV1
} catch {
throw new Error('现有 HTML 档案数据已损坏，请从 messages.js.bak 恢复或更换导出名称')
}
- if (archive.sourceId && archive.sourceId !== sourceId) {
- throw new Error('同名导出目录已属于另一个会话，请修改文件名称后重试')
+ const expectedIds = targets.map((target) => target.userMd5).sort()
+ if (parsed.version === 1) {
+ if (targets.length !== 1 || parsed.sourceId !== targets[0].userMd5) {
+ throw new Error('同名导出目录的聊天集合不同，请修改文件名称后重试')
+ }
+ return {
+ version: 2,
+ name: parsed.name || name,
+ exportedAt: parsed.exportedAt || new Date(0).toISOString(),
+ conversations: [
+ {
+ id: targets[0].userMd5,
+ name: targets[0].name,
+ type: targets[0].type,
+ messageCount: Array.isArray(parsed.messages) ? parsed.messages.length : 0
+ }
+ ],
+ messages: (Array.isArray(parsed.messages) ? parsed.messages : []).map((message) => ({
+ ...message,
+ exportConversationId: targets[0].userMd5,
+ exportConversationName: targets[0].name
+ }))
+ }
+ }
+ const actualIds = (Array.isArray(parsed.conversations) ? parsed.conversations : [])
+ .map((conversation) => conversation.id)
+ .sort()
+ if (actualIds.join('|') !== expectedIds.join('|')) {
+ throw new Error('同名导出目录的聊天集合不同，请修改文件名称后重试')
}
return {
- version: 1,
- sourceId,
- name: archive.name || name,
- exportedAt: archive.exportedAt || new Date(0).toISOString(),
- messages: Array.isArray(archive.messages) ? archive.messages : []
+ version: 2,
+ name: parsed.name || name,
+ exportedAt: parsed.exportedAt || new Date(0).toISOString(),
+ conversations: parsed.conversations,
+ messages: Array.isArray(parsed.messages) ? parsed.messages : []
+ }
+}
+
+async function writeZipArchive(
+ outputDir: string,
+ zipPath: string,
+ folderName: string,
+ jobId: string
+): Promise {
+ const temporaryPath = `${zipPath}.tmp-${process.pid}-${Date.now()}`
+ await fs.rm(temporaryPath, { force: true })
+ const output = createWriteStream(temporaryPath)
+ await new Promise((resolve, reject) => {
+ output.once('open', () => resolve())
+ output.once('error', reject)
+ })
+ const archive = new ZipArchive({ zlib: { level: 6 } })
+ activeArchives.set(jobId, archive)
+ try {
+ await new Promise((resolve, reject) => {
+ let settled = false
+ const finish = (error?: Error): void => {
+ if (settled) return
+ settled = true
+ if (error) reject(error)
+ else resolve()
+ }
+ output.on('close', () => finish())
+ output.on('error', finish)
+ archive.on('warning', (error) => {
+ if ((error as NodeJS.ErrnoException).code !== 'ENOENT') finish(error)
+ })
+ archive.on('error', finish)
+ archive.pipe(output)
+ archive.directory(outputDir, safeFilePart(folderName))
+ void archive.finalize().catch(finish)
+ })
+ if (!jobs.has(jobId)) throw new Error('已取消')
+ try {
+ await fs.rename(temporaryPath, zipPath)
+ } catch (error) {
+ if (!['EEXIST', 'EPERM'].includes((error as NodeJS.ErrnoException).code || '')) throw error
+ await fs.rm(zipPath, { force: true })
+ await fs.rename(temporaryPath, zipPath)
+ }
+ } finally {
+ activeArchives.delete(jobId)
+ if (!output.closed) {
+ output.destroy()
+ await new Promise((resolve) => output.once('close', () => resolve()))
+ }
+ await fs.rm(temporaryPath, { force: true })
}
}
@@ -298,11 +422,57 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
if (!win.isDestroyed()) win.webContents.send('export:progress', p)
}
try {
+ const targets = request.targets || []
+ if (targets.length < 1 || targets.length > 5) {
+ throw new Error('一次导出必须选择 1 到 5 个聊天')
+ }
+ if (new Set(targets.map((target) => target.userMd5)).size !== targets.length) {
+ throw new Error('导出聊天不能重复')
+ }
+ if (targets.length > 1 && request.format !== 'html') {
+ throw new Error('多聊天合并仅支持 HTML 格式')
+ }
+ const archiveName =
+ targets.length > 1 ? `${targets[0].name} 等 ${targets.length} 个聊天` : targets[0].name
+ const targetById = new Map(targets.map((target) => [target.userMd5, target]))
send({ jobId: request.jobId, phase: 'reading', processed: 0, total: 100, percent: 0 })
await new Promise((resolve) => setImmediate(resolve))
- const messages = (
- await chat.listMessagesAsync(request.userMd5, request.startTime, request.endTime)
- ).filter((message) => request.kinds.includes(kindOf(message)))
+ const messageEntries: { message: Message; targetOrder: number; messageOrder: number }[] = []
+ for (const [targetOrder, target] of targets.entries()) {
+ if (!jobs.has(request.jobId)) {
+ send({ jobId: request.jobId, phase: 'cancelled', processed: 0, percent: 5 })
+ return { success: false, error: '已取消' }
+ }
+ const targetMessages = (
+ await chat.listMessagesAsync(target.userMd5, request.startTime, request.endTime)
+ ).filter((message) => request.kinds.includes(kindOf(message)))
+ for (const [messageOrder, message] of targetMessages.entries()) {
+ messageEntries.push({
+ message: {
+ ...message,
+ exportConversationId: target.userMd5,
+ exportConversationName: target.name
+ },
+ targetOrder,
+ messageOrder
+ })
+ }
+ send({
+ jobId: request.jobId,
+ phase: 'reading',
+ processed: targetOrder + 1,
+ total: targets.length,
+ percent: Math.max(1, Math.round(((targetOrder + 1) / targets.length) * 10))
+ })
+ }
+ const messages = messageEntries
+ .sort((left, right) => {
+ const byTime = Number(left.message.createTime || 0) - Number(right.message.createTime || 0)
+ if (byTime !== 0) return byTime
+ if (left.targetOrder !== right.targetOrder) return left.targetOrder - right.targetOrder
+ return left.messageOrder - right.messageOrder
+ })
+ .map((entry) => entry.message)
const rawSelfInfo = await chat.getSelfAccountInfoAsync()
const selfInfo = rawSelfInfo ? mergeCachedSelfInfo(rawSelfInfo.accountRoot, rawSelfInfo) : null
const isUsableSelfName = (value: string | undefined): value is string => {
@@ -313,13 +483,14 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
return true
}
for (const message of messages) {
+ const target = targetById.get(message.exportConversationId || '')
message.exportMediaUrl = undefined
message.exportMediaType = undefined
message.exportMediaName = undefined
message.exportMediaError = undefined
message.voiceDataUrl = undefined
message.exportShowAvatar = request.includeAvatars !== false
- const mappedName = message.senderId ? request.nameMap?.[message.senderId] : undefined
+ const mappedName = message.senderId ? target?.nameMap?.[message.senderId] : undefined
if (mappedName && (!message.isSender || isUsableSelfName(mappedName))) {
message.name = mappedName
} else if (message.isSender && isUsableSelfName(selfInfo?.nickname)) {
@@ -357,7 +528,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
? join(outputDir, 'index.html')
: join(root, `${outputFolder}.${ext}`)
if (request.format === 'html') {
- const previousArchive = await readHtmlArchive(outputDir, request.userMd5, request.name)
+ const previousArchive = await readHtmlArchive(outputDir, targets, archiveName)
await fs.mkdir(join(outputDir, 'voices'), { recursive: true })
await fs.mkdir(join(outputDir, 'media'), { recursive: true })
await fs.mkdir(join(outputDir, 'avatars'), { recursive: true })
@@ -369,10 +540,14 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
.filter((value): value is string => Boolean(value))
)
)
+ const requestedAvatarUrls = Object.assign(
+ {},
+ ...targets.map((target) => target.avatarUrls || {})
+ ) as Record
const avatarMap =
request.includeAvatars === false
? {}
- : { ...chat.getContactAvatars(avatarUsernames), ...(request.avatarUrls || {}) }
+ : { ...(await chat.getContactAvatars(avatarUsernames)), ...requestedAvatarUrls }
const imageConfig = imageKeys.getConfig()
const imageService =
client && imageConfig.aesKey
@@ -382,13 +557,27 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
const stickerService = client ? new StickerService(client) : null
const fileService = client ? new FileAssetService(client) : null
const exportedAvatars = new Map()
+ const conversationAvatarUrls = new Map()
+ if (request.includeAvatars !== false) {
+ for (const target of targets) {
+ if (!jobs.has(request.jobId)) throw new Error('已取消')
+ if (!target.avatarUrl) continue
+ const resolved = await readAvatarAsset(target.avatarUrl)
+ if (!resolved) continue
+ const avatarName = `conversation_${hashPart(target.userMd5)}.${resolved.extension}`
+ await fs.writeFile(join(outputDir, 'avatars', avatarName), resolved.buffer)
+ conversationAvatarUrls.set(target.userMd5, `avatars/${avatarName}`)
+ }
+ }
const voiceService =
request.includeMedia && chat.getChatDb()
? new VoiceService(chat.getChatDb()!.getWcdb4Client())
: null
if (voiceService) {
for (const message of messages) {
+ if (!jobs.has(request.jobId)) throw new Error('已取消')
if (kindOf(message) !== 'voice') continue
+ const conversationId = message.exportConversationId || targets[0].userMd5
if (!message.sessionId || message.localId == null || !message.createTime) {
keepMediaError(request, message, '语音标识不完整，无法定位本地语音')
continue
@@ -410,7 +599,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
keepMediaError(request, message, reason)
continue
}
- const voiceName = `voice_${hashPart(exportMessageKey(message, request.userMd5))}.wav`
+ const voiceName = `voice_${hashPart(exportMessageKey(message, conversationId))}.wav`
const audioBuffer = Buffer.from(voice.data, 'base64')
await fs.writeFile(join(outputDir, 'voices', voiceName), audioBuffer)
message.voiceDataUrl = `voices/${voiceName}`
@@ -431,14 +620,26 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
}
}
for (const [index, message] of messages.entries()) {
+ if (!jobs.has(request.jobId)) {
+ send({
+ jobId: request.jobId,
+ phase: 'cancelled',
+ processed: index,
+ total: messages.length,
+ percent: 15 + Math.round((index / Math.max(messages.length, 1)) * 75)
+ })
+ return { success: false, error: '已取消' }
+ }
+ const conversationId = message.exportConversationId || targets[0].userMd5
message.exportShowAvatar = request.includeAvatars !== false
const avatar = (message.senderId ? avatarMap[message.senderId] : undefined) || message.img
const resolvedAvatar = avatar ? await readAvatarAsset(avatar) : null
const avatarBuffer = resolvedAvatar?.buffer || null
const avatarExtension = resolvedAvatar?.extension || 'jpg'
if (avatarBuffer) {
- const avatarKey =
- message.senderId || avatar || `message_${exportMessageKey(message, request.userMd5)}`
+ const avatarKey = `${conversationId}:${
+ message.senderId || avatar || `message_${exportMessageKey(message, conversationId)}`
+ }`
let avatarName = exportedAvatars.get(avatarKey)
if (!avatarName) {
avatarName = `avatar_${hashPart(avatarKey)}.${avatarExtension}`
@@ -472,7 +673,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
allowThumbnail: attempt.allowThumbnail,
preferThumbnail: attempt.preferThumbnail,
sessionId: message.sessionId,
- sessionMd5: request.userMd5,
+ sessionMd5: conversationId,
createTime: message.createTime
}
)
@@ -497,7 +698,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
}
const decoded = decryptedImage ? decodeDataUrl(decryptedImage.data) : null
if (decoded) {
- const name = `image_${hashPart(exportMessageKey(message, request.userMd5))}.${decoded.extension}`
+ const name = `image_${hashPart(exportMessageKey(message, conversationId))}.${decoded.extension}`
await fs.writeFile(join(outputDir, 'media', name), decoded.buffer)
message.exportMediaUrl = `media/${name}`
message.exportMediaType = 'image'
@@ -534,8 +735,8 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
} else if (extname(source).toLowerCase() !== '.mp4') {
keepMediaError(request, message, '视频格式不支持，仅支持本地 MP4 文件')
} else {
- const name = `video_${hashPart(exportMessageKey(message, request.userMd5))}.mp4`
- await fs.copyFile(source, join(outputDir, 'media', name))
+ const name = `video_${hashPart(exportMessageKey(message, conversationId))}.mp4`
+ await copyWritableExportFile(source, join(outputDir, 'media', name))
message.exportMediaUrl = `media/${name}`
message.exportMediaType = 'video'
}
@@ -549,7 +750,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
? await readAvatarAsset(stickerSource)
: null
if (decoded) {
- const name = `sticker_${hashPart(exportMessageKey(message, request.userMd5))}.${decoded.extension}`
+ const name = `sticker_${hashPart(exportMessageKey(message, conversationId))}.${decoded.extension}`
await fs.writeFile(join(outputDir, 'media', name), decoded.buffer)
message.exportMediaUrl = `media/${name}`
message.exportMediaType = 'sticker'
@@ -564,8 +765,8 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
if (!resolved.success || !resolved.filePath || !resolved.fileName) {
keepMediaError(request, message, resolved.error || '本地文件附件缺失')
} else {
- const name = `file_${hashPart(exportMessageKey(message, request.userMd5))}_${safeFilePart(resolved.fileName)}`
- await fs.copyFile(resolved.filePath, join(outputDir, 'media', name))
+ const name = `file_${hashPart(exportMessageKey(message, conversationId))}_${safeFilePart(resolved.fileName)}`
+ await copyWritableExportFile(resolved.filePath, join(outputDir, 'media', name))
message.exportMediaUrl = `media/${name}`
message.exportMediaType = 'file'
message.exportMediaName = message.contentData.title || resolved.fileName
@@ -583,26 +784,53 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
const mergedMessages = mergeHtmlArchiveMessages(
previousArchive.messages,
messages,
- request.userMd5
+ '',
+ targets.map((target) => target.userMd5)
)
+ const normalizedMessages = normalizeHtmlArchiveSelfNames(mergedMessages, selfInfo)
const archive: HtmlExportArchive = {
- version: 1,
- sourceId: request.userMd5,
- name: request.name,
+ version: 2,
+ name: archiveName,
exportedAt: new Date().toISOString(),
- messages: normalizeHtmlArchiveSelfNames(mergedMessages, selfInfo)
+ conversations: targets.map((target) => ({
+ id: target.userMd5,
+ name: target.name,
+ type: target.type,
+ avatarUrl:
+ conversationAvatarUrls.get(target.userMd5) ||
+ previousArchive.conversations.find((item) => item.id === target.userMd5)?.avatarUrl,
+ messageCount: normalizedMessages.filter(
+ (message) => message.exportConversationId === target.userMd5
+ ).length
+ })),
+ messages: normalizedMessages
}
- await fs.writeFile(outputPath, renderExportPage(request.name), 'utf8')
+ if (!jobs.has(request.jobId)) throw new Error('已取消')
+ await fs.writeFile(outputPath, renderExportPage(archiveName), 'utf8')
await writeHtmlArchive(outputDir, archive)
+ let completedPath = outputPath
+ if (request.zip) {
+ if (!jobs.has(request.jobId)) return { success: false, error: '已取消' }
+ const zipPath = join(root, `${outputFolder}.zip`)
+ send({
+ jobId: request.jobId,
+ phase: 'compressing',
+ processed: archive.messages.length,
+ total: archive.messages.length,
+ percent: 95
+ })
+ await writeZipArchive(outputDir, zipPath, outputFolder, request.jobId)
+ completedPath = zipPath
+ }
send({
jobId: request.jobId,
phase: 'completed',
processed: archive.messages.length,
total: archive.messages.length,
percent: 100,
- outputPath
+ outputPath: completedPath
})
- return { success: true, outputPath, messageCount: archive.messages.length }
+ return { success: true, outputPath: completedPath, messageCount: archive.messages.length }
} else {
send({
jobId: request.jobId,
@@ -612,7 +840,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
percent: 90
})
}
- await fs.writeFile(outputPath, render(request.format, messages, request.name), 'utf8')
+ await fs.writeFile(outputPath, render(request.format, messages, archiveName), 'utf8')
send({
jobId: request.jobId,
phase: 'completed',
@@ -624,6 +852,10 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
return { success: true, outputPath, messageCount: messages.length }
} catch (error) {
const message = error instanceof Error ? error.message : String(error)
+ if (!jobs.has(request.jobId) || message === '已取消') {
+ send({ jobId: request.jobId, phase: 'cancelled', processed: 0, error: '已取消' })
+ return { success: false, error: '已取消' }
+ }
send({ jobId: request.jobId, phase: 'failed', processed: 0, error: message })
return { success: false, error: message }
} finally {
@@ -632,6 +864,7 @@ export async function runExport(request: ExportRequest, win: BrowserWindow): Pro
}
export function cancelExport(jobId: string): void {
jobs.delete(jobId)
+ activeArchives.get(jobId)?.abort()
}
export async function revealExport(path: string): Promise {
shell.showItemInFolder(path)
diff --git a/src/renderer/src/App.tsx b/src/renderer/src/App.tsx
index 37c3b25..65449cb 100644
--- a/src/renderer/src/App.tsx
+++ b/src/renderer/src/App.tsx
@@ -212,7 +212,32 @@ function App(): React.ReactElement {
const [exportTasks, setExportTasks] = useState(() => {
try {
const stored = JSON.parse(localStorage.getItem('wxe_export_tasks') || '[]')
- return Array.isArray(stored) ? (stored as ExportTaskRecord[]).slice(0, 20) : []
+ if (!Array.isArray(stored)) return []
+ return stored.slice(0, 20).map(
+ (
+ value: Partial & {
+ contactId?: string
+ contactName?: string
+ }
+ ) => {
+ const targetIds = Array.isArray(value.targetIds)
+ ? value.targetIds
+ : value.contactId
+ ? [value.contactId]
+ : []
+ const targetNames = Array.isArray(value.targetNames)
+ ? value.targetNames
+ : value.contactName
+ ? [value.contactName]
+ : []
+ return {
+ ...value,
+ targetIds,
+ targetNames,
+ targetLabel: value.targetLabel || targetNames.join('、') || '聊天导出'
+ } as ExportTaskRecord
+ }
+ )
} catch {
return []
}
@@ -341,10 +366,14 @@ function App(): React.ReactElement {
const handleStartExport = async (
request: ExportRequest
): Promise => {
+ const targetNames = request.targets.map((target) => target.name)
+ const targetLabel =
+ targetNames.length > 1 ? `${targetNames[0]} 等 ${targetNames.length} 个聊天` : targetNames[0]
const task: ExportTaskRecord = {
jobId: request.jobId,
- contactId: request.userMd5,
- contactName: request.name,
+ targetIds: request.targets.map((target) => target.userMd5),
+ targetNames,
+ targetLabel,
format: request.format,
status: 'running',
progress: { jobId: request.jobId, phase: 'reading', processed: 0, percent: 0 },
@@ -1220,16 +1249,14 @@ function App(): React.ReactElement {
}
}
- const loadExportPreview = async (contact: Contact): Promise => {
- setSelectedContact(contact)
- selectedContactMd5Ref.current = contact.md5
+ const loadExportPreviewMessages = async (contact: Contact): Promise => {
try {
- const previewMessages = await window.api.getMessages(contact.md5, undefined, undefined, {
+ return await window.api.getMessages(contact.md5, undefined, undefined, {
limit: EXPORT_PREVIEW_LIMIT
})
- if (selectedContactMd5Ref.current === contact.md5) setMessages(previewMessages)
} catch (error) {
console.warn('[Export] preview load failed:', error)
+ return []
}
}
@@ -1322,7 +1349,6 @@ function App(): React.ReactElement {
const handlePageChange = (page: AppPage): void => {
setActivePage(page)
if (page === 'archive' && selectedContact) void handleSelectContact(selectedContact)
- if (page === 'export' && selectedContact) void loadExportPreview(selectedContact)
if (page === 'settings') setSettingsCategory('account-database')
if (page === 'report' && isGroupContact(selectedContact) && !reportSourceContact) {
setReportSourceContact(selectedContact)
@@ -1702,11 +1728,10 @@ function App(): React.ReactElement {
return (
void
onContactTypeChange: (value: 'all' | 'group' | 'user') => void
onSelectContact: (contact: Contact) => void
+ onCompleteSelection: () => void
onOpenSettings: () => void
}
@@ -20,6 +24,9 @@ export function ExportContactPanel({
contacts,
filteredContacts,
activeContact,
+ selectedContactIds,
+ selectionMode,
+ selectionLimit,
selfInfo,
dbReady,
contactFilter,
@@ -27,6 +34,7 @@ export function ExportContactPanel({
onContactFilterChange,
onContactTypeChange,
onSelectContact,
+ onCompleteSelection,
onOpenSettings
}: ExportContactPanelProps): React.ReactElement {
return (
@@ -65,15 +73,30 @@ export function ExportContactPanel({
${safe(name)}
+
@@ -616,7 +755,9 @@ export function renderExportPage(name: string): string {
正在加载聊天档案…
+
+ 已选 {selectedContactIds.length} / {selectionLimit} 个
+
+
+
+ )}
+
{filteredContacts.map((contact) => {
const name = displayName(contact)
+ const selected = selectedContactIds.includes(contact.md5)
+ const atLimit = selectionMode && !selected && selectedContactIds.length >= selectionLimit
return (
)
})}
diff --git a/src/renderer/src/components/export/ExportPreviewPanel.tsx b/src/renderer/src/components/export/ExportPreviewPanel.tsx
index 91d4be4..396146a 100644
--- a/src/renderer/src/components/export/ExportPreviewPanel.tsx
+++ b/src/renderer/src/components/export/ExportPreviewPanel.tsx
@@ -10,6 +10,7 @@ interface ExportPreviewPanelProps {
previewBytes: number
selfInfo: SelfInfo | null
progress: ExportJobProgress | null
+ selectedCount: number
jobId: string
onCancel: (jobId: string) => void
onReveal: (path: string) => void
@@ -22,6 +23,7 @@ export function ExportPreviewPanel({
previewBytes,
selfInfo,
progress,
+ selectedCount,
jobId,
onCancel,
onReveal
@@ -32,7 +34,9 @@ export function ExportPreviewPanel({
<>
-
-
导出预览
- 仅预览最近 20 条
+
+ {selectedCount > 1 ? `${selectedCount} 个聊天 · 合并预览` : '仅预览最近 20 条'}
+
最近消息
@@ -50,7 +54,7 @@ export function ExportPreviewPanel({
]
).map((message) => (
+ {selectedCount > 1 && message.exportConversationName
+ ? `${message.exportConversationName} · `
+ : ''}
{message.name || (message.isSender ? '我' : '联系人')} ·{' '}
{formatPreviewTime(message)}
@@ -108,7 +115,11 @@ export function ExportPreviewPanel({
- {progress?.phase === 'writing'
- ? `正在写入 ${progress.processed.toLocaleString()} 条消息... ${progress.percent ?? 0}%`
- : `正在读取消息... ${progress?.percent ?? 0}%`}
+ {progress?.phase === 'compressing'
+ ? `正在压缩资源包... ${progress.percent ?? 0}%`
+ : progress?.phase === 'writing'
+ ? `正在写入 ${progress.processed.toLocaleString()} 条消息... ${progress.percent ?? 0}%`
+ : `正在读取消息... ${progress?.percent ?? 0}%`}
- 准备导出
- - {progress?.phase === 'writing' ? '生成档案' : '分批读取聊天记录'} + {progress?.phase === 'compressing' + ? '压缩 ZIP' + : progress?.phase === 'writing' + ? '生成档案' + : '分批读取聊天记录'}
- 解析消息内容
- 处理媒体资源 @@ -118,9 +129,11 @@ export function ExportPreviewPanel({
- CSV 默认最快；HTML 会包含图片、引用和其他媒体，导出时间可能较长。 + {selectedContacts.length > 1 + ? '多聊天合并仅支持 HTML，会保留每条消息所属的聊天。' + : 'CSV 默认最快；HTML 会包含图片、引用和其他媒体，导出时间可能较长。'}
{format === 'html' && ( <> @@ -506,45 +595,6 @@ export function ExportWorkspace({ -导出格式
-
- {formatOrder.map((value) => (
-
- ))}
-
- {format === 'html' && (
-
-
-
-
- )}
- 保存设置
@@ -577,7 +627,7 @@ export function ExportWorkspace({
: '准备就绪'}
路径：{targetPath}
-