mirror of
https://github.com/hicccc77/WeFlow.git
synced 2026-03-27 15:07:55 +00:00
Compare commits
49 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b7852a8c07 | ||
|
|
4b9d94eb62 | ||
|
|
70481fd468 | ||
|
|
52c67f4d23 | ||
|
|
d3618f3065 | ||
|
|
29472beee8 | ||
|
|
acaac507b1 | ||
|
|
f25c23b2b3 | ||
|
|
5ab0466a87 | ||
|
|
d49c44f3be | ||
|
|
4577b4e955 | ||
|
|
dafde2eaba | ||
|
|
db4fab9130 | ||
|
|
9aee578707 | ||
|
|
6d74eb65ae | ||
|
|
6e8ae3a12b | ||
|
|
a4be7f9005 | ||
|
|
587ee630d7 | ||
|
|
6952a5f680 | ||
|
|
b263ecd45c | ||
|
|
74fc0e4e88 | ||
|
|
a873366342 | ||
|
|
c4dc266f93 | ||
|
|
96ff783bbd | ||
|
|
804a65f52b | ||
|
|
e88c859f4f | ||
|
|
c1a393eaf6 | ||
|
|
15e08dc529 | ||
|
|
e55bcaf7eb | ||
|
|
4e64c6ad6e | ||
|
|
5a15e1a1d6 | ||
|
|
ba07d47496 | ||
|
|
25325e80ee | ||
|
|
89783b4d45 | ||
|
|
d5f0094025 | ||
|
|
b4f37451be | ||
|
|
84ea378815 | ||
|
|
72d4db1f27 | ||
|
|
21ea879d97 | ||
|
|
a5baef2240 | ||
|
|
bbecf54aba | ||
|
|
5f868d193c | ||
|
|
62b035ab39 | ||
|
|
ff5ee33e08 | ||
|
|
8e28016e5e | ||
|
|
f17a18cb6d | ||
|
|
999f45e5f5 | ||
|
|
3e303fadd7 | ||
|
|
3b7590d8ce |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -61,3 +61,4 @@ wcdb/
|
|||||||
chatlab-format.md
|
chatlab-format.md
|
||||||
*.bak
|
*.bak
|
||||||
AGENTS.md
|
AGENTS.md
|
||||||
|
.claude/
|
||||||
@@ -19,6 +19,7 @@ WeFlow 是一个**完全本地**的微信**实时**聊天记录查看、分析
|
|||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/hicccc77/WeFlow/issues">
|
<a href="https://github.com/hicccc77/WeFlow/issues">
|
||||||
<img src="https://img.shields.io/github/issues/hicccc77/WeFlow?style=flat-square" alt="Issues">
|
<img src="https://img.shields.io/github/issues/hicccc77/WeFlow?style=flat-square" alt="Issues">
|
||||||
|
<img src="https://gh-down-badges.linkof.link/hicccc77/WeFlow/" alt="Downloads" />
|
||||||
</a>
|
</a>
|
||||||
<a href="https://t.me/weflow_cc">
|
<a href="https://t.me/weflow_cc">
|
||||||
<img src="https://img.shields.io/badge/Telegram%20频道-0088cc?style=flat-square&logo=telegram&logoColor=0088cc&labelColor=white" alt="Telegram">
|
<img src="https://img.shields.io/badge/Telegram%20频道-0088cc?style=flat-square&logo=telegram&logoColor=0088cc&labelColor=white" alt="Telegram">
|
||||||
@@ -35,6 +36,7 @@ WeFlow 是一个**完全本地**的微信**实时**聊天记录查看、分析
|
|||||||
## 主要功能
|
## 主要功能
|
||||||
|
|
||||||
- 本地实时查看聊天记录
|
- 本地实时查看聊天记录
|
||||||
|
- 朋友圈图片、视频、**实况**的预览和解密
|
||||||
- 统计分析与群聊画像
|
- 统计分析与群聊画像
|
||||||
- 年度报告与可视化概览
|
- 年度报告与可视化概览
|
||||||
- 导出聊天记录为 HTML 等格式
|
- 导出聊天记录为 HTML 等格式
|
||||||
@@ -86,6 +88,7 @@ npm run build
|
|||||||
## 致谢
|
## 致谢
|
||||||
|
|
||||||
- [密语 CipherTalk](https://github.com/ILoveBingLu/miyu) 为本项目提供了基础框架
|
- [密语 CipherTalk](https://github.com/ILoveBingLu/miyu) 为本项目提供了基础框架
|
||||||
|
- [WeChat-Channels-Video-File-Decryption](https://github.com/Evil0ctal/WeChat-Channels-Video-File-Decryption) 提供了视频解密相关的技术参考
|
||||||
|
|
||||||
## 支持我们
|
## 支持我们
|
||||||
|
|
||||||
|
|||||||
@@ -50,12 +50,20 @@ GET /api/v1/messages
|
|||||||
| 参数名 | 类型 | 必填 | 说明 |
|
| 参数名 | 类型 | 必填 | 说明 |
|
||||||
|--------|------|------|------|
|
|--------|------|------|------|
|
||||||
| `talker` | string | ✅ | 会话 ID(wxid 或群 ID) |
|
| `talker` | string | ✅ | 会话 ID(wxid 或群 ID) |
|
||||||
| `limit` | number | ❌ | 返回数量限制,默认 100 |
|
| `limit` | number | ❌ | 返回数量限制,默认 100,范围 `1~10000` |
|
||||||
| `offset` | number | ❌ | 偏移量,用于分页,默认 0 |
|
| `offset` | number | ❌ | 偏移量,用于分页,默认 0 |
|
||||||
| `start` | string | ❌ | 开始时间,格式 YYYYMMDD |
|
| `start` | string | ❌ | 开始时间,格式 YYYYMMDD |
|
||||||
| `end` | string | ❌ | 结束时间,格式 YYYYMMDD |
|
| `end` | string | ❌ | 结束时间,格式 YYYYMMDD |
|
||||||
|
| `keyword` | string | ❌ | 关键词过滤(基于消息显示文本) |
|
||||||
| `chatlab` | string | ❌ | 设为 `1` 则输出 ChatLab 格式 |
|
| `chatlab` | string | ❌ | 设为 `1` 则输出 ChatLab 格式 |
|
||||||
| `format` | string | ❌ | 输出格式:`json`(默认)或 `chatlab` |
|
| `format` | string | ❌ | 输出格式:`json`(默认)或 `chatlab` |
|
||||||
|
| `media` | string | ❌ | 设为 `1` 时导出媒体并返回媒体路径(兼容别名 `meiti`);`0` 时媒体返回占位符 |
|
||||||
|
| `image` | string | ❌ | 在 `media=1` 时控制图片导出,`1/0`(兼容别名 `tupian`) |
|
||||||
|
| `voice` | string | ❌ | 在 `media=1` 时控制语音导出,`1/0`(兼容别名 `vioce`) |
|
||||||
|
| `video` | string | ❌ | 在 `media=1` 时控制视频导出,`1/0` |
|
||||||
|
| `emoji` | string | ❌ | 在 `media=1` 时控制表情导出,`1/0` |
|
||||||
|
|
||||||
|
默认媒体导出目录:`%USERPROFILE%\\Documents\\WeFlow\\api-media`
|
||||||
|
|
||||||
**示例请求**
|
**示例请求**
|
||||||
|
|
||||||
@@ -68,6 +76,12 @@ GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&chatlab=1
|
|||||||
|
|
||||||
# 带时间范围查询
|
# 带时间范围查询
|
||||||
GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&start=20260101&end=20260205&limit=100
|
GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&start=20260101&end=20260205&limit=100
|
||||||
|
|
||||||
|
# 开启媒体导出(只导出图片和语音)
|
||||||
|
GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&media=1&image=1&voice=1&video=0&emoji=0
|
||||||
|
|
||||||
|
# 关键词过滤
|
||||||
|
GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&keyword=项目进度&limit=50
|
||||||
```
|
```
|
||||||
|
|
||||||
**响应(原始格式)**
|
**响应(原始格式)**
|
||||||
@@ -77,15 +91,21 @@ GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&start=20260101&end=202
|
|||||||
"talker": "wxid_xxx",
|
"talker": "wxid_xxx",
|
||||||
"count": 50,
|
"count": 50,
|
||||||
"hasMore": true,
|
"hasMore": true,
|
||||||
|
"media": {
|
||||||
|
"enabled": true,
|
||||||
|
"exportPath": "C:\\Users\\Alice\\Documents\\WeFlow\\api-media",
|
||||||
|
"count": 12
|
||||||
|
},
|
||||||
"messages": [
|
"messages": [
|
||||||
{
|
{
|
||||||
"localId": 123,
|
"localId": 123,
|
||||||
"talker": "wxid_xxx",
|
"localType": 3,
|
||||||
"type": 1,
|
"content": "[图片]",
|
||||||
"content": "消息内容",
|
|
||||||
"createTime": 1738713600000,
|
"createTime": 1738713600000,
|
||||||
"isSelf": false,
|
"senderUsername": "wxid_sender",
|
||||||
"sender": "wxid_sender"
|
"mediaType": "image",
|
||||||
|
"mediaFileName": "image_123.jpg",
|
||||||
|
"mediaPath": "C:\\Users\\Alice\\Documents\\WeFlow\\api-media\\wxid_xxx\\images\\image_123.jpg"
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
@@ -119,9 +139,15 @@ GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&start=20260101&end=202
|
|||||||
"accountName": "用户名",
|
"accountName": "用户名",
|
||||||
"timestamp": 1738713600000,
|
"timestamp": 1738713600000,
|
||||||
"type": 0,
|
"type": 0,
|
||||||
"content": "消息内容"
|
"content": "消息内容",
|
||||||
|
"mediaPath": "C:\\Users\\Alice\\Documents\\WeFlow\\api-media\\wxid_xxx\\images\\image_123.jpg"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"media": {
|
||||||
|
"enabled": true,
|
||||||
|
"exportPath": "C:\\Users\\Alice\\Documents\\WeFlow\\api-media",
|
||||||
|
"count": 12
|
||||||
}
|
}
|
||||||
]
|
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|||||||
115
electron/main.ts
115
electron/main.ts
@@ -21,7 +21,7 @@ import { videoService } from './services/videoService'
|
|||||||
import { snsService, isVideoUrl } from './services/snsService'
|
import { snsService, isVideoUrl } from './services/snsService'
|
||||||
import { contactExportService } from './services/contactExportService'
|
import { contactExportService } from './services/contactExportService'
|
||||||
import { windowsHelloService } from './services/windowsHelloService'
|
import { windowsHelloService } from './services/windowsHelloService'
|
||||||
import { llamaService } from './services/llamaService'
|
|
||||||
import { registerNotificationHandlers, showNotification } from './windows/notificationWindow'
|
import { registerNotificationHandlers, showNotification } from './windows/notificationWindow'
|
||||||
import { httpService } from './services/httpService'
|
import { httpService } from './services/httpService'
|
||||||
|
|
||||||
@@ -173,6 +173,20 @@ function createWindow(options: { autoShow?: boolean } = {}) {
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// 忽略微信 CDN 域名的证书错误(部分节点证书配置不正确)
|
||||||
|
win.webContents.on('certificate-error', (event, url, _error, _cert, callback) => {
|
||||||
|
const trusted = ['.qq.com', '.qpic.cn', '.weixin.qq.com', '.wechat.com']
|
||||||
|
try {
|
||||||
|
const host = new URL(url).hostname
|
||||||
|
if (trusted.some(d => host.endsWith(d))) {
|
||||||
|
event.preventDefault()
|
||||||
|
callback(true)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
} catch {}
|
||||||
|
callback(false)
|
||||||
|
})
|
||||||
|
|
||||||
return win
|
return win
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -799,67 +813,18 @@ function registerIpcHandlers() {
|
|||||||
return chatService.getNewMessages(sessionId, minTime, limit)
|
return chatService.getNewMessages(sessionId, minTime, limit)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:updateMessage', async (_, sessionId: string, localId: number, createTime: number, newContent: string) => {
|
||||||
|
return chatService.updateMessage(sessionId, localId, createTime, newContent)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:deleteMessage', async (_, sessionId: string, localId: number, createTime: number, dbPathHint?: string) => {
|
||||||
|
return chatService.deleteMessage(sessionId, localId, createTime, dbPathHint)
|
||||||
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getContact', async (_, username: string) => {
|
ipcMain.handle('chat:getContact', async (_, username: string) => {
|
||||||
return await chatService.getContact(username)
|
return await chatService.getContact(username)
|
||||||
})
|
})
|
||||||
|
|
||||||
// Llama AI
|
|
||||||
ipcMain.handle('llama:init', async () => {
|
|
||||||
return await llamaService.init()
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:loadModel', async (_, modelPath: string) => {
|
|
||||||
return llamaService.loadModel(modelPath)
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:createSession', async (_, systemPrompt?: string) => {
|
|
||||||
return llamaService.createSession(systemPrompt)
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:chat', async (event, message: string, options?: { thinking?: boolean }) => {
|
|
||||||
// We use a callback to stream back to the renderer
|
|
||||||
const webContents = event.sender
|
|
||||||
try {
|
|
||||||
if (!webContents) return { success: false, error: 'No sender' }
|
|
||||||
|
|
||||||
const response = await llamaService.chat(message, options, (token) => {
|
|
||||||
if (!webContents.isDestroyed()) {
|
|
||||||
webContents.send('llama:token', token)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
return { success: true, response }
|
|
||||||
} catch (e) {
|
|
||||||
return { success: false, error: String(e) }
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:downloadModel', async (event, url: string, savePath: string) => {
|
|
||||||
const webContents = event.sender
|
|
||||||
try {
|
|
||||||
await llamaService.downloadModel(url, savePath, (payload) => {
|
|
||||||
if (!webContents.isDestroyed()) {
|
|
||||||
webContents.send('llama:downloadProgress', payload)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
return { success: true }
|
|
||||||
} catch (e) {
|
|
||||||
return { success: false, error: String(e) }
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:getModelsPath', async () => {
|
|
||||||
return llamaService.getModelsPath()
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:checkFileExists', async (_, filePath: string) => {
|
|
||||||
const { existsSync } = await import('fs')
|
|
||||||
return existsSync(filePath)
|
|
||||||
})
|
|
||||||
|
|
||||||
ipcMain.handle('llama:getModelStatus', async (_, modelPath: string) => {
|
|
||||||
return llamaService.getModelStatus(modelPath)
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
ipcMain.handle('chat:getContactAvatar', async (_, username: string) => {
|
ipcMain.handle('chat:getContactAvatar', async (_, username: string) => {
|
||||||
return await chatService.getContactAvatar(username)
|
return await chatService.getContactAvatar(username)
|
||||||
@@ -929,6 +894,10 @@ function registerIpcHandlers() {
|
|||||||
return snsService.getTimeline(limit, offset, usernames, keyword, startTime, endTime)
|
return snsService.getTimeline(limit, offset, usernames, keyword, startTime, endTime)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:getSnsUsernames', async () => {
|
||||||
|
return snsService.getSnsUsernames()
|
||||||
|
})
|
||||||
|
|
||||||
ipcMain.handle('sns:debugResource', async (_, url: string) => {
|
ipcMain.handle('sns:debugResource', async (_, url: string) => {
|
||||||
return snsService.debugResource(url)
|
return snsService.debugResource(url)
|
||||||
})
|
})
|
||||||
@@ -975,6 +944,26 @@ function registerIpcHandlers() {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:exportTimeline', async (event, options: any) => {
|
||||||
|
return snsService.exportTimeline(options, (progress) => {
|
||||||
|
if (!event.sender.isDestroyed()) {
|
||||||
|
event.sender.send('sns:exportProgress', progress)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:selectExportDir', async () => {
|
||||||
|
const { dialog } = await import('electron')
|
||||||
|
const result = await dialog.showOpenDialog({
|
||||||
|
properties: ['openDirectory', 'createDirectory'],
|
||||||
|
title: '选择导出目录'
|
||||||
|
})
|
||||||
|
if (result.canceled || !result.filePaths?.[0]) {
|
||||||
|
return { canceled: true }
|
||||||
|
}
|
||||||
|
return { canceled: false, filePath: result.filePaths[0] }
|
||||||
|
})
|
||||||
|
|
||||||
// 私聊克隆
|
// 私聊克隆
|
||||||
|
|
||||||
|
|
||||||
@@ -1116,6 +1105,13 @@ function registerIpcHandlers() {
|
|||||||
return groupAnalyticsService.exportGroupMembers(chatroomId, outputPath)
|
return groupAnalyticsService.exportGroupMembers(chatroomId, outputPath)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle(
|
||||||
|
'groupAnalytics:exportGroupMemberMessages',
|
||||||
|
async (_, chatroomId: string, memberUsername: string, outputPath: string, startTime?: number, endTime?: number) => {
|
||||||
|
return groupAnalyticsService.exportGroupMemberMessages(chatroomId, memberUsername, outputPath, startTime, endTime)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
// 打开协议窗口
|
// 打开协议窗口
|
||||||
ipcMain.handle('window:openAgreementWindow', async () => {
|
ipcMain.handle('window:openAgreementWindow', async () => {
|
||||||
createAgreementWindow()
|
createAgreementWindow()
|
||||||
@@ -1350,7 +1346,8 @@ function registerIpcHandlers() {
|
|||||||
ipcMain.handle('http:status', async () => {
|
ipcMain.handle('http:status', async () => {
|
||||||
return {
|
return {
|
||||||
running: httpService.isRunning(),
|
running: httpService.isRunning(),
|
||||||
port: httpService.getPort()
|
port: httpService.getPort(),
|
||||||
|
mediaExportPath: httpService.getDefaultMediaExportPath()
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -131,6 +131,10 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
ipcRenderer.invoke('chat:getNewMessages', sessionId, minTime, limit),
|
ipcRenderer.invoke('chat:getNewMessages', sessionId, minTime, limit),
|
||||||
getContact: (username: string) => ipcRenderer.invoke('chat:getContact', username),
|
getContact: (username: string) => ipcRenderer.invoke('chat:getContact', username),
|
||||||
getContactAvatar: (username: string) => ipcRenderer.invoke('chat:getContactAvatar', username),
|
getContactAvatar: (username: string) => ipcRenderer.invoke('chat:getContactAvatar', username),
|
||||||
|
updateMessage: (sessionId: string, localId: number, createTime: number, newContent: string) =>
|
||||||
|
ipcRenderer.invoke('chat:updateMessage', sessionId, localId, createTime, newContent),
|
||||||
|
deleteMessage: (sessionId: string, localId: number, createTime: number, dbPathHint?: string) =>
|
||||||
|
ipcRenderer.invoke('chat:deleteMessage', sessionId, localId, createTime, dbPathHint),
|
||||||
resolveTransferDisplayNames: (chatroomId: string, payerUsername: string, receiverUsername: string) =>
|
resolveTransferDisplayNames: (chatroomId: string, payerUsername: string, receiverUsername: string) =>
|
||||||
ipcRenderer.invoke('chat:resolveTransferDisplayNames', chatroomId, payerUsername, receiverUsername),
|
ipcRenderer.invoke('chat:resolveTransferDisplayNames', chatroomId, payerUsername, receiverUsername),
|
||||||
getMyAvatarUrl: () => ipcRenderer.invoke('chat:getMyAvatarUrl'),
|
getMyAvatarUrl: () => ipcRenderer.invoke('chat:getMyAvatarUrl'),
|
||||||
@@ -216,7 +220,9 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
getGroupMessageRanking: (chatroomId: string, limit?: number, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupMessageRanking', chatroomId, limit, startTime, endTime),
|
getGroupMessageRanking: (chatroomId: string, limit?: number, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupMessageRanking', chatroomId, limit, startTime, endTime),
|
||||||
getGroupActiveHours: (chatroomId: string, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupActiveHours', chatroomId, startTime, endTime),
|
getGroupActiveHours: (chatroomId: string, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupActiveHours', chatroomId, startTime, endTime),
|
||||||
getGroupMediaStats: (chatroomId: string, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupMediaStats', chatroomId, startTime, endTime),
|
getGroupMediaStats: (chatroomId: string, startTime?: number, endTime?: number) => ipcRenderer.invoke('groupAnalytics:getGroupMediaStats', chatroomId, startTime, endTime),
|
||||||
exportGroupMembers: (chatroomId: string, outputPath: string) => ipcRenderer.invoke('groupAnalytics:exportGroupMembers', chatroomId, outputPath)
|
exportGroupMembers: (chatroomId: string, outputPath: string) => ipcRenderer.invoke('groupAnalytics:exportGroupMembers', chatroomId, outputPath),
|
||||||
|
exportGroupMemberMessages: (chatroomId: string, memberUsername: string, outputPath: string, startTime?: number, endTime?: number) =>
|
||||||
|
ipcRenderer.invoke('groupAnalytics:exportGroupMemberMessages', chatroomId, memberUsername, outputPath, startTime, endTime)
|
||||||
},
|
},
|
||||||
|
|
||||||
// 年度报告
|
// 年度报告
|
||||||
@@ -270,30 +276,16 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
sns: {
|
sns: {
|
||||||
getTimeline: (limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number) =>
|
getTimeline: (limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number) =>
|
||||||
ipcRenderer.invoke('sns:getTimeline', limit, offset, usernames, keyword, startTime, endTime),
|
ipcRenderer.invoke('sns:getTimeline', limit, offset, usernames, keyword, startTime, endTime),
|
||||||
|
getSnsUsernames: () => ipcRenderer.invoke('sns:getSnsUsernames'),
|
||||||
debugResource: (url: string) => ipcRenderer.invoke('sns:debugResource', url),
|
debugResource: (url: string) => ipcRenderer.invoke('sns:debugResource', url),
|
||||||
proxyImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:proxyImage', payload),
|
proxyImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:proxyImage', payload),
|
||||||
downloadImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:downloadImage', payload)
|
downloadImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:downloadImage', payload),
|
||||||
|
exportTimeline: (options: any) => ipcRenderer.invoke('sns:exportTimeline', options),
|
||||||
|
onExportProgress: (callback: (payload: any) => void) => {
|
||||||
|
ipcRenderer.on('sns:exportProgress', (_, payload) => callback(payload))
|
||||||
|
return () => ipcRenderer.removeAllListeners('sns:exportProgress')
|
||||||
},
|
},
|
||||||
|
selectExportDir: () => ipcRenderer.invoke('sns:selectExportDir')
|
||||||
// Llama AI
|
|
||||||
llama: {
|
|
||||||
loadModel: (modelPath: string) => ipcRenderer.invoke('llama:loadModel', modelPath),
|
|
||||||
createSession: (systemPrompt?: string) => ipcRenderer.invoke('llama:createSession', systemPrompt),
|
|
||||||
chat: (message: string, options?: any) => ipcRenderer.invoke('llama:chat', message, options),
|
|
||||||
downloadModel: (url: string, savePath: string) => ipcRenderer.invoke('llama:downloadModel', url, savePath),
|
|
||||||
getModelsPath: () => ipcRenderer.invoke('llama:getModelsPath'),
|
|
||||||
checkFileExists: (filePath: string) => ipcRenderer.invoke('llama:checkFileExists', filePath),
|
|
||||||
getModelStatus: (modelPath: string) => ipcRenderer.invoke('llama:getModelStatus', modelPath),
|
|
||||||
onToken: (callback: (token: string) => void) => {
|
|
||||||
const listener = (_: any, token: string) => callback(token)
|
|
||||||
ipcRenderer.on('llama:token', listener)
|
|
||||||
return () => ipcRenderer.removeListener('llama:token', listener)
|
|
||||||
},
|
|
||||||
onDownloadProgress: (callback: (payload: { downloaded: number; total: number; speed: number }) => void) => {
|
|
||||||
const listener = (_: any, payload: { downloaded: number; total: number; speed: number }) => callback(payload)
|
|
||||||
ipcRenderer.on('llama:downloadProgress', listener)
|
|
||||||
return () => ipcRenderer.removeListener('llama:downloadProgress', listener)
|
|
||||||
}
|
|
||||||
},
|
},
|
||||||
|
|
||||||
// HTTP API 服务
|
// HTTP API 服务
|
||||||
|
|||||||
@@ -50,6 +50,9 @@ export interface Message {
|
|||||||
emojiCdnUrl?: string
|
emojiCdnUrl?: string
|
||||||
emojiMd5?: string
|
emojiMd5?: string
|
||||||
emojiLocalPath?: string // 本地缓存 castle 路径
|
emojiLocalPath?: string // 本地缓存 castle 路径
|
||||||
|
emojiThumbUrl?: string
|
||||||
|
emojiEncryptUrl?: string
|
||||||
|
emojiAesKey?: string
|
||||||
// 引用消息相关
|
// 引用消息相关
|
||||||
quotedContent?: string
|
quotedContent?: string
|
||||||
quotedSender?: string
|
quotedSender?: string
|
||||||
@@ -84,6 +87,7 @@ export interface Message {
|
|||||||
datadesc: string
|
datadesc: string
|
||||||
datatitle?: string
|
datatitle?: string
|
||||||
}>
|
}>
|
||||||
|
_db_path?: string // 内部字段:记录消息所属数据库路径
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface Contact {
|
export interface Contact {
|
||||||
@@ -99,7 +103,7 @@ export interface ContactInfo {
|
|||||||
remark?: string
|
remark?: string
|
||||||
nickname?: string
|
nickname?: string
|
||||||
avatarUrl?: string
|
avatarUrl?: string
|
||||||
type: 'friend' | 'group' | 'official' | 'other'
|
type: 'friend' | 'group' | 'official' | 'former_friend' | 'other'
|
||||||
}
|
}
|
||||||
|
|
||||||
// 表情包缓存
|
// 表情包缓存
|
||||||
@@ -109,7 +113,7 @@ const emojiDownloading: Map<string, Promise<string | null>> = new Map()
|
|||||||
class ChatService {
|
class ChatService {
|
||||||
private configService: ConfigService
|
private configService: ConfigService
|
||||||
private connected = false
|
private connected = false
|
||||||
private messageCursors: Map<string, { cursor: number; fetched: number; batchSize: number; startTime?: number; endTime?: number; ascending?: boolean }> = new Map()
|
private messageCursors: Map<string, { cursor: number; fetched: number; batchSize: number; startTime?: number; endTime?: number; ascending?: boolean; bufferedMessages?: any[] }> = new Map()
|
||||||
private readonly messageBatchDefault = 50
|
private readonly messageBatchDefault = 50
|
||||||
private avatarCache: Map<string, ContactCacheEntry>
|
private avatarCache: Map<string, ContactCacheEntry>
|
||||||
private readonly avatarCacheTtlMs = 10 * 60 * 1000
|
private readonly avatarCacheTtlMs = 10 * 60 * 1000
|
||||||
@@ -269,6 +273,32 @@ class ChatService {
|
|||||||
this.connected = false
|
this.connected = false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改消息内容
|
||||||
|
*/
|
||||||
|
async updateMessage(sessionId: string, localId: number, createTime: number, newContent: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
try {
|
||||||
|
const connectResult = await this.ensureConnected()
|
||||||
|
if (!connectResult.success) return { success: false, error: connectResult.error }
|
||||||
|
return await wcdbService.updateMessage(sessionId, localId, createTime, newContent)
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除消息
|
||||||
|
*/
|
||||||
|
async deleteMessage(sessionId: string, localId: number, createTime: number, dbPathHint?: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
try {
|
||||||
|
const connectResult = await this.ensureConnected()
|
||||||
|
if (!connectResult.success) return { success: false, error: connectResult.error }
|
||||||
|
return await wcdbService.deleteMessage(sessionId, localId, createTime, dbPathHint)
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 获取会话列表(优化:先返回基础数据,不等待联系人信息加载)
|
* 获取会话列表(优化:先返回基础数据,不等待联系人信息加载)
|
||||||
*/
|
*/
|
||||||
@@ -573,7 +603,7 @@ class ChatService {
|
|||||||
// 使用execQuery直接查询加密的contact.db
|
// 使用execQuery直接查询加密的contact.db
|
||||||
// kind='contact', path=null表示使用已打开的contact.db
|
// kind='contact', path=null表示使用已打开的contact.db
|
||||||
const contactQuery = `
|
const contactQuery = `
|
||||||
SELECT username, remark, nick_name, alias, local_type
|
SELECT username, remark, nick_name, alias, local_type, flag, quan_pin
|
||||||
FROM contact
|
FROM contact
|
||||||
`
|
`
|
||||||
|
|
||||||
@@ -621,47 +651,25 @@ class ChatService {
|
|||||||
for (const row of rows) {
|
for (const row of rows) {
|
||||||
const username = row.username || ''
|
const username = row.username || ''
|
||||||
|
|
||||||
// 过滤系统账号和特殊账号 - 完全复制cipher的逻辑
|
|
||||||
if (!username) continue
|
if (!username) continue
|
||||||
if (username === 'filehelper' || username === 'fmessage' || username === 'floatbottle' ||
|
|
||||||
username === 'medianote' || username === 'newsapp' || username.startsWith('fake_') ||
|
|
||||||
username === 'weixin' || username === 'qmessage' || username === 'qqmail' ||
|
|
||||||
username === 'tmessage' || username.startsWith('wxid_') === false &&
|
|
||||||
username.includes('@') === false && username.startsWith('gh_') === false &&
|
|
||||||
/^[a-zA-Z0-9_-]+$/.test(username) === false) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// 判断类型 - 正确规则:wxid开头且有alias的是好友
|
const excludeNames = ['medianote', 'floatbottle', 'qmessage', 'qqmail', 'fmessage']
|
||||||
let type: 'friend' | 'group' | 'official' | 'other' = 'other'
|
let type: 'friend' | 'group' | 'official' | 'former_friend' | 'other' = 'other'
|
||||||
const localType = row.local_type || 0
|
const localType = this.getRowInt(row, ['local_type', 'localType', 'WCDB_CT_local_type'], 0)
|
||||||
|
const flag = Number(row.flag ?? 0)
|
||||||
|
const quanPin = this.getRowField(row, ['quan_pin', 'quanPin', 'WCDB_CT_quan_pin']) || ''
|
||||||
|
|
||||||
if (username.includes('@chatroom')) {
|
if (username.includes('@chatroom')) {
|
||||||
type = 'group'
|
type = 'group'
|
||||||
} else if (username.startsWith('gh_')) {
|
} else if (username.startsWith('gh_')) {
|
||||||
type = 'official'
|
type = 'official'
|
||||||
} else if (localType === 3 || localType === 4) {
|
} else if (/^(?!.*(gh_|@chatroom)).*$/.test(username) && localType === 1 && !excludeNames.includes(username)) {
|
||||||
type = 'official'
|
|
||||||
} else if (username.startsWith('wxid_') && row.alias) {
|
|
||||||
// wxid开头且有alias的是好友
|
|
||||||
type = 'friend'
|
|
||||||
} else if (localType === 1) {
|
|
||||||
// local_type=1 也是好友
|
|
||||||
type = 'friend'
|
|
||||||
} else if (localType === 2) {
|
|
||||||
// local_type=2 是群成员但非好友,跳过
|
|
||||||
continue
|
|
||||||
} else if (localType === 0) {
|
|
||||||
// local_type=0 可能是好友或其他,检查是否有备注或昵称
|
|
||||||
if (row.remark || row.nick_name) {
|
|
||||||
type = 'friend'
|
type = 'friend'
|
||||||
|
} else if (/^(?!.*(gh_|@chatroom)).*$/.test(username) && localType === 0 && quanPin) {
|
||||||
|
type = 'former_friend'
|
||||||
} else {
|
} else {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
// 其他未知类型,跳过
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
const displayName = row.remark || row.nick_name || row.alias || username
|
const displayName = row.remark || row.nick_name || row.alias || username
|
||||||
|
|
||||||
@@ -729,7 +737,7 @@ class ChatService {
|
|||||||
// 4. startTime/endTime 改变(视为全新查询)
|
// 4. startTime/endTime 改变(视为全新查询)
|
||||||
// 5. ascending 改变
|
// 5. ascending 改变
|
||||||
const needNewCursor = !state ||
|
const needNewCursor = !state ||
|
||||||
offset === 0 ||
|
offset !== state.fetched || // Offset mismatch -> must reset cursor
|
||||||
state.batchSize !== batchSize ||
|
state.batchSize !== batchSize ||
|
||||||
state.startTime !== startTime ||
|
state.startTime !== startTime ||
|
||||||
state.endTime !== endTime ||
|
state.endTime !== endTime ||
|
||||||
@@ -761,6 +769,7 @@ class ChatService {
|
|||||||
// 如果需要跳过消息(offset > 0),逐批获取但不返回
|
// 如果需要跳过消息(offset > 0),逐批获取但不返回
|
||||||
// 注意:仅在 offset === 0 时重建游标最安全;
|
// 注意:仅在 offset === 0 时重建游标最安全;
|
||||||
// 当 startTime/endTime 变化导致重建时,offset 应由前端重置为 0
|
// 当 startTime/endTime 变化导致重建时,offset 应由前端重置为 0
|
||||||
|
state.bufferedMessages = []
|
||||||
if (offset > 0) {
|
if (offset > 0) {
|
||||||
console.warn(`[ChatService] 新游标需跳过 ${offset} 条消息(startTime=${startTime}, endTime=${endTime})`)
|
console.warn(`[ChatService] 新游标需跳过 ${offset} 条消息(startTime=${startTime}, endTime=${endTime})`)
|
||||||
let skipped = 0
|
let skipped = 0
|
||||||
@@ -777,8 +786,22 @@ class ChatService {
|
|||||||
console.warn(`[ChatService] 跳过时数据耗尽: skipped=${skipped}/${offset}`)
|
console.warn(`[ChatService] 跳过时数据耗尽: skipped=${skipped}/${offset}`)
|
||||||
return { success: true, messages: [], hasMore: false }
|
return { success: true, messages: [], hasMore: false }
|
||||||
}
|
}
|
||||||
skipped += skipBatch.rows.length
|
|
||||||
state.fetched += skipBatch.rows.length
|
const count = skipBatch.rows.length
|
||||||
|
// Check if we overshot the offset
|
||||||
|
if (skipped + count > offset) {
|
||||||
|
const keepIndex = offset - skipped
|
||||||
|
if (keepIndex < count) {
|
||||||
|
state.bufferedMessages = skipBatch.rows.slice(keepIndex)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
skipped += count
|
||||||
|
state.fetched += count
|
||||||
|
|
||||||
|
// If satisfied offset, break
|
||||||
|
if (skipped >= offset) break;
|
||||||
|
|
||||||
if (!skipBatch.hasMore) {
|
if (!skipBatch.hasMore) {
|
||||||
console.warn(`[ChatService] 跳过后无更多数据: skipped=${skipped}/${offset}`)
|
console.warn(`[ChatService] 跳过后无更多数据: skipped=${skipped}/${offset}`)
|
||||||
return { success: true, messages: [], hasMore: false }
|
return { success: true, messages: [], hasMore: false }
|
||||||
@@ -787,13 +810,8 @@ class ChatService {
|
|||||||
if (attempts >= maxSkipAttempts) {
|
if (attempts >= maxSkipAttempts) {
|
||||||
console.error(`[ChatService] 跳过消息超过最大尝试次数: attempts=${attempts}`)
|
console.error(`[ChatService] 跳过消息超过最大尝试次数: attempts=${attempts}`)
|
||||||
}
|
}
|
||||||
console.log(`[ChatService] 跳过完成: skipped=${skipped}, fetched=${state.fetched}`)
|
console.log(`[ChatService] 跳过完成: skipped=${skipped}, fetched=${state.fetched}, buffered=${state.bufferedMessages?.length || 0}`)
|
||||||
}
|
}
|
||||||
} else if (state && offset !== state.fetched) {
|
|
||||||
// offset 与 fetched 不匹配,说明状态不一致
|
|
||||||
console.warn(`[ChatService] 游标状态不一致: offset=${offset}, fetched=${state.fetched}, 继续使用现有游标`)
|
|
||||||
// 不重新创建游标,而是继续使用现有游标
|
|
||||||
// 这样可以避免频繁重建导致的问题
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 确保 state 已初始化
|
// 确保 state 已初始化
|
||||||
@@ -803,19 +821,35 @@ class ChatService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 获取当前批次的消息
|
// 获取当前批次的消息
|
||||||
const batch = await wcdbService.fetchMessageBatch(state.cursor)
|
// Use buffered rows from skip logic if available
|
||||||
if (!batch.success) {
|
let rows: any[] = state.bufferedMessages || []
|
||||||
console.error('[ChatService] 获取消息批次失败:', batch.error)
|
state.bufferedMessages = undefined // Clear buffer after use
|
||||||
return { success: false, error: batch.error || '获取消息失败' }
|
|
||||||
|
// If buffer is not enough to fill a batch, try to fetch more
|
||||||
|
// Or if buffer is empty, fetch a batch
|
||||||
|
if (rows.length < batchSize) {
|
||||||
|
const nextBatch = await wcdbService.fetchMessageBatch(state.cursor)
|
||||||
|
if (nextBatch.success && nextBatch.rows) {
|
||||||
|
rows = rows.concat(nextBatch.rows)
|
||||||
|
state.fetched += nextBatch.rows.length
|
||||||
|
} else if (!nextBatch.success) {
|
||||||
|
console.error('[ChatService] 获取消息批次失败:', nextBatch.error)
|
||||||
|
// If we have some buffered rows, we can still return them?
|
||||||
|
// Or fail? Let's return what we have if any, otherwise fail.
|
||||||
|
if (rows.length === 0) {
|
||||||
|
return { success: false, error: nextBatch.error || '获取消息失败' }
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!batch.rows) {
|
// If we have more than limit (due to buffer + full batch), slice it
|
||||||
console.error('[ChatService] 获取消息失败: 返回数据为空')
|
if (rows.length > limit) {
|
||||||
return { success: false, error: '获取消息失败: 返回数据为空' }
|
rows = rows.slice(0, limit)
|
||||||
|
// Note: We don't adjust state.fetched here because it tracks cursor position.
|
||||||
|
// Next time offset will catch up or mismatch trigger reset.
|
||||||
}
|
}
|
||||||
|
|
||||||
const rows = batch.rows as Record<string, any>[]
|
const hasMore = rows.length > 0 // Simplified hasMore check for now, can be improved
|
||||||
const hasMore = batch.hasMore === true
|
|
||||||
|
|
||||||
const normalized = this.normalizeMessageOrder(this.mapRowsToMessages(rows))
|
const normalized = this.normalizeMessageOrder(this.mapRowsToMessages(rows))
|
||||||
|
|
||||||
@@ -1056,6 +1090,13 @@ class ChatService {
|
|||||||
return Number.isFinite(parsed) ? parsed : NaN
|
return Number.isFinite(parsed) ? parsed : NaN
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* HTTP API 复用消息解析逻辑,确保和应用内展示一致。
|
||||||
|
*/
|
||||||
|
mapRowsToMessagesForApi(rows: Record<string, any>[]): Message[] {
|
||||||
|
return this.mapRowsToMessages(rows)
|
||||||
|
}
|
||||||
|
|
||||||
private mapRowsToMessages(rows: Record<string, any>[]): Message[] {
|
private mapRowsToMessages(rows: Record<string, any>[]): Message[] {
|
||||||
const myWxid = this.configService.get('myWxid')
|
const myWxid = this.configService.get('myWxid')
|
||||||
const cleanedWxid = myWxid ? this.cleanAccountDirName(myWxid) : null
|
const cleanedWxid = myWxid ? this.cleanAccountDirName(myWxid) : null
|
||||||
@@ -1151,6 +1192,9 @@ class ChatService {
|
|||||||
const emojiInfo = this.parseEmojiInfo(content)
|
const emojiInfo = this.parseEmojiInfo(content)
|
||||||
emojiCdnUrl = emojiInfo.cdnUrl
|
emojiCdnUrl = emojiInfo.cdnUrl
|
||||||
emojiMd5 = emojiInfo.md5
|
emojiMd5 = emojiInfo.md5
|
||||||
|
cdnThumbUrl = emojiInfo.thumbUrl // 复用 cdnThumbUrl 字段或使用 emojiThumbUrl
|
||||||
|
// 注意:Message 接口定义的 emojiThumbUrl,这里我们统一一下
|
||||||
|
// 如果 Message 接口有 emojiThumbUrl,则使用它
|
||||||
} else if (localType === 3 && content) {
|
} else if (localType === 3 && content) {
|
||||||
const imageInfo = this.parseImageInfo(content)
|
const imageInfo = this.parseImageInfo(content)
|
||||||
imageMd5 = imageInfo.md5
|
imageMd5 = imageInfo.md5
|
||||||
@@ -1373,7 +1417,7 @@ class ChatService {
|
|||||||
/**
|
/**
|
||||||
* 解析表情包信息
|
* 解析表情包信息
|
||||||
*/
|
*/
|
||||||
private parseEmojiInfo(content: string): { cdnUrl?: string; md5?: string } {
|
private parseEmojiInfo(content: string): { cdnUrl?: string; md5?: string; thumbUrl?: string; encryptUrl?: string; aesKey?: string } {
|
||||||
try {
|
try {
|
||||||
// 提取 cdnurl
|
// 提取 cdnurl
|
||||||
let cdnUrl: string | undefined
|
let cdnUrl: string | undefined
|
||||||
@@ -1387,26 +1431,39 @@ class ChatService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 如果没有 cdnurl,尝试 thumburl
|
// 提取 thumburl
|
||||||
if (!cdnUrl) {
|
let thumbUrl: string | undefined
|
||||||
const thumbUrlMatch = /thumburl\s*=\s*['"]([^'"]+)['"]/i.exec(content) || /thumburl\s*=\s*([^'"]+?)(?=\s|\/|>)/i.exec(content)
|
const thumbUrlMatch = /thumburl\s*=\s*['"]([^'"]+)['"]/i.exec(content) || /thumburl\s*=\s*([^'"]+?)(?=\s|\/|>)/i.exec(content)
|
||||||
if (thumbUrlMatch) {
|
if (thumbUrlMatch) {
|
||||||
cdnUrl = thumbUrlMatch[1].replace(/&/g, '&')
|
thumbUrl = thumbUrlMatch[1].replace(/&/g, '&')
|
||||||
if (cdnUrl.includes('%')) {
|
if (thumbUrl.includes('%')) {
|
||||||
try {
|
try {
|
||||||
cdnUrl = decodeURIComponent(cdnUrl)
|
thumbUrl = decodeURIComponent(thumbUrl)
|
||||||
} catch { }
|
} catch { }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// 提取 md5
|
// 提取 md5
|
||||||
const md5Match = /md5\s*=\s*['"]([a-fA-F0-9]+)['"]/i.exec(content) || /md5\s*=\s*([a-fA-F0-9]+)/i.exec(content)
|
const md5Match = /md5\s*=\s*['"]([a-fA-F0-9]+)['"]/i.exec(content) || /md5\s*=\s*([a-fA-F0-9]+)/i.exec(content)
|
||||||
const md5 = md5Match ? md5Match[1] : undefined
|
const md5 = md5Match ? md5Match[1] : undefined
|
||||||
|
|
||||||
// 不构造假 URL,只返回真正的 cdnurl
|
// 提取 encrypturl
|
||||||
// 没有 cdnUrl 时保持静默,交由后续回退逻辑处理
|
let encryptUrl: string | undefined
|
||||||
return { cdnUrl, md5 }
|
const encryptUrlMatch = /encrypturl\s*=\s*['"]([^'"]+)['"]/i.exec(content) || /encrypturl\s*=\s*([^'"]+?)(?=\s|\/|>)/i.exec(content)
|
||||||
|
if (encryptUrlMatch) {
|
||||||
|
encryptUrl = encryptUrlMatch[1].replace(/&/g, '&')
|
||||||
|
if (encryptUrl.includes('%')) {
|
||||||
|
try {
|
||||||
|
encryptUrl = decodeURIComponent(encryptUrl)
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 提取 aeskey
|
||||||
|
const aesKeyMatch = /aeskey\s*=\s*['"]([a-zA-Z0-9]+)['"]/i.exec(content) || /aeskey\s*=\s*([a-zA-Z0-9]+)/i.exec(content)
|
||||||
|
const aesKey = aesKeyMatch ? aesKeyMatch[1] : undefined
|
||||||
|
|
||||||
|
return { cdnUrl, md5, thumbUrl, encryptUrl, aesKey }
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('[ChatService] 表情包解析失败:', e, { xml: content })
|
console.error('[ChatService] 表情包解析失败:', e, { xml: content })
|
||||||
return {}
|
return {}
|
||||||
@@ -2622,11 +2679,7 @@ class ChatService {
|
|||||||
// 检查内存缓存
|
// 检查内存缓存
|
||||||
const cached = emojiCache.get(cacheKey)
|
const cached = emojiCache.get(cacheKey)
|
||||||
if (cached && existsSync(cached)) {
|
if (cached && existsSync(cached)) {
|
||||||
// 读取文件并转为 data URL
|
return { success: true, localPath: cached }
|
||||||
const dataUrl = this.fileToDataUrl(cached)
|
|
||||||
if (dataUrl) {
|
|
||||||
return { success: true, localPath: dataUrl }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 检查是否正在下载
|
// 检查是否正在下载
|
||||||
@@ -2634,10 +2687,7 @@ class ChatService {
|
|||||||
if (downloading) {
|
if (downloading) {
|
||||||
const result = await downloading
|
const result = await downloading
|
||||||
if (result) {
|
if (result) {
|
||||||
const dataUrl = this.fileToDataUrl(result)
|
return { success: true, localPath: result }
|
||||||
if (dataUrl) {
|
|
||||||
return { success: true, localPath: dataUrl }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return { success: false, error: '下载失败' }
|
return { success: false, error: '下载失败' }
|
||||||
}
|
}
|
||||||
@@ -2654,10 +2704,7 @@ class ChatService {
|
|||||||
const filePath = join(cacheDir, `${cacheKey}${ext}`)
|
const filePath = join(cacheDir, `${cacheKey}${ext}`)
|
||||||
if (existsSync(filePath)) {
|
if (existsSync(filePath)) {
|
||||||
emojiCache.set(cacheKey, filePath)
|
emojiCache.set(cacheKey, filePath)
|
||||||
const dataUrl = this.fileToDataUrl(filePath)
|
return { success: true, localPath: filePath }
|
||||||
if (dataUrl) {
|
|
||||||
return { success: true, localPath: dataUrl }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2671,10 +2718,7 @@ class ChatService {
|
|||||||
|
|
||||||
if (localPath) {
|
if (localPath) {
|
||||||
emojiCache.set(cacheKey, localPath)
|
emojiCache.set(cacheKey, localPath)
|
||||||
const dataUrl = this.fileToDataUrl(localPath)
|
return { success: true, localPath }
|
||||||
if (dataUrl) {
|
|
||||||
return { success: true, localPath: dataUrl }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return { success: false, error: '下载失败' }
|
return { success: false, error: '下载失败' }
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
@@ -3917,6 +3961,13 @@ class ChatService {
|
|||||||
const imgInfo = this.parseImageInfo(rawContent)
|
const imgInfo = this.parseImageInfo(rawContent)
|
||||||
Object.assign(msg, imgInfo)
|
Object.assign(msg, imgInfo)
|
||||||
msg.imageDatName = this.parseImageDatNameFromRow(row)
|
msg.imageDatName = this.parseImageDatNameFromRow(row)
|
||||||
|
} else if (msg.localType === 47) { // Emoji
|
||||||
|
const emojiInfo = this.parseEmojiInfo(rawContent)
|
||||||
|
msg.emojiCdnUrl = emojiInfo.cdnUrl
|
||||||
|
msg.emojiMd5 = emojiInfo.md5
|
||||||
|
msg.emojiThumbUrl = emojiInfo.thumbUrl
|
||||||
|
msg.emojiEncryptUrl = emojiInfo.encryptUrl
|
||||||
|
msg.emojiAesKey = emojiInfo.aesKey
|
||||||
}
|
}
|
||||||
|
|
||||||
return msg
|
return msg
|
||||||
@@ -4227,6 +4278,34 @@ class ChatService {
|
|||||||
return { success: false, error: String(e) }
|
return { success: false, error: String(e) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载表情包文件(用于导出,返回文件路径)
|
||||||
|
*/
|
||||||
|
async downloadEmojiFile(msg: Message): Promise<string | null> {
|
||||||
|
if (!msg.emojiMd5) return null
|
||||||
|
let url = msg.emojiCdnUrl
|
||||||
|
|
||||||
|
// 尝试获取 URL
|
||||||
|
if (!url && msg.emojiEncryptUrl) {
|
||||||
|
console.warn('[ChatService] Emoji has only encryptUrl:', msg.emojiMd5)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!url) {
|
||||||
|
await this.fallbackEmoticon(msg)
|
||||||
|
url = msg.emojiCdnUrl
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!url) return null
|
||||||
|
|
||||||
|
// Reuse existing downloadEmoji method
|
||||||
|
const result = await this.downloadEmoji(url, msg.emojiMd5)
|
||||||
|
if (result.success && result.localPath) {
|
||||||
|
return result.localPath
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export const chatService = new ChatService()
|
export const chatService = new ChatService()
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ interface ConfigSchema {
|
|||||||
|
|
||||||
// 缓存相关
|
// 缓存相关
|
||||||
cachePath: string
|
cachePath: string
|
||||||
weixinDllPath: string
|
|
||||||
lastOpenedDb: string
|
lastOpenedDb: string
|
||||||
lastSession: string
|
lastSession: string
|
||||||
|
|
||||||
@@ -75,7 +75,7 @@ export class ConfigService {
|
|||||||
imageAesKey: '',
|
imageAesKey: '',
|
||||||
wxidConfigs: {},
|
wxidConfigs: {},
|
||||||
cachePath: '',
|
cachePath: '',
|
||||||
weixinDllPath: '',
|
|
||||||
lastOpenedDb: '',
|
lastOpenedDb: '',
|
||||||
lastSession: '',
|
lastSession: '',
|
||||||
theme: 'system',
|
theme: 'system',
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ interface ContactExportOptions {
|
|||||||
groups: boolean
|
groups: boolean
|
||||||
officials: boolean
|
officials: boolean
|
||||||
}
|
}
|
||||||
|
selectedUsernames?: string[]
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -40,6 +41,11 @@ class ContactExportService {
|
|||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
if (Array.isArray(options.selectedUsernames) && options.selectedUsernames.length > 0) {
|
||||||
|
const selectedSet = new Set(options.selectedUsernames)
|
||||||
|
contacts = contacts.filter(c => selectedSet.has(c.username))
|
||||||
|
}
|
||||||
|
|
||||||
if (contacts.length === 0) {
|
if (contacts.length === 0) {
|
||||||
return { success: false, error: '没有符合条件的联系人' }
|
return { success: false, error: '没有符合条件的联系人' }
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -186,6 +186,17 @@ body {
|
|||||||
word-break: break-word;
|
word-break: break-word;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.message-link-card {
|
||||||
|
color: #2563eb;
|
||||||
|
text-decoration: underline;
|
||||||
|
text-underline-offset: 2px;
|
||||||
|
word-break: break-all;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-link-card:hover {
|
||||||
|
color: #1d4ed8;
|
||||||
|
}
|
||||||
|
|
||||||
.inline-emoji {
|
.inline-emoji {
|
||||||
width: 22px;
|
width: 22px;
|
||||||
height: 22px;
|
height: 22px;
|
||||||
|
|||||||
@@ -186,6 +186,17 @@ body {
|
|||||||
word-break: break-word;
|
word-break: break-word;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.message-link-card {
|
||||||
|
color: #2563eb;
|
||||||
|
text-decoration: underline;
|
||||||
|
text-underline-offset: 2px;
|
||||||
|
word-break: break-all;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-link-card:hover {
|
||||||
|
color: #1d4ed8;
|
||||||
|
}
|
||||||
|
|
||||||
.inline-emoji {
|
.inline-emoji {
|
||||||
width: 22px;
|
width: 22px;
|
||||||
height: 22px;
|
height: 22px;
|
||||||
|
|||||||
@@ -70,6 +70,8 @@ const MESSAGE_TYPE_MAP: Record<number, number> = {
|
|||||||
export interface ExportOptions {
|
export interface ExportOptions {
|
||||||
format: 'chatlab' | 'chatlab-jsonl' | 'json' | 'html' | 'txt' | 'excel' | 'weclone' | 'sql'
|
format: 'chatlab' | 'chatlab-jsonl' | 'json' | 'html' | 'txt' | 'excel' | 'weclone' | 'sql'
|
||||||
dateRange?: { start: number; end: number } | null
|
dateRange?: { start: number; end: number } | null
|
||||||
|
senderUsername?: string
|
||||||
|
fileNameSuffix?: string
|
||||||
exportMedia?: boolean
|
exportMedia?: boolean
|
||||||
exportAvatars?: boolean
|
exportAvatars?: boolean
|
||||||
exportImages?: boolean
|
exportImages?: boolean
|
||||||
@@ -534,11 +536,14 @@ class ExportService {
|
|||||||
groupNicknamesMap: Map<string, string>,
|
groupNicknamesMap: Map<string, string>,
|
||||||
getContactName: (username: string) => Promise<string>
|
getContactName: (username: string) => Promise<string>
|
||||||
): Promise<string | null> {
|
): Promise<string | null> {
|
||||||
const xmlType = this.extractXmlValue(content, 'type')
|
const normalizedContent = this.normalizeAppMessageContent(content || '')
|
||||||
if (xmlType !== '2000') return null
|
if (!normalizedContent) return null
|
||||||
|
|
||||||
const payerUsername = this.extractXmlValue(content, 'payer_username')
|
const xmlType = this.extractXmlValue(normalizedContent, 'type')
|
||||||
const receiverUsername = this.extractXmlValue(content, 'receiver_username')
|
if (xmlType && xmlType !== '2000') return null
|
||||||
|
|
||||||
|
const payerUsername = this.extractXmlValue(normalizedContent, 'payer_username')
|
||||||
|
const receiverUsername = this.extractXmlValue(normalizedContent, 'receiver_username')
|
||||||
if (!payerUsername || !receiverUsername) return null
|
if (!payerUsername || !receiverUsername) return null
|
||||||
|
|
||||||
const cleanedMyWxid = myWxid ? this.cleanAccountDirName(myWxid) : ''
|
const cleanedMyWxid = myWxid ? this.cleanAccountDirName(myWxid) : ''
|
||||||
@@ -565,6 +570,52 @@ class ExportService {
|
|||||||
return `${payerName} 转账给 ${receiverName}`
|
return `${payerName} 转账给 ${receiverName}`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private isSameWxid(lhs?: string, rhs?: string): boolean {
|
||||||
|
const left = new Set(this.buildGroupNicknameIdCandidates([lhs]).map((id) => id.toLowerCase()))
|
||||||
|
if (left.size === 0) return false
|
||||||
|
const right = this.buildGroupNicknameIdCandidates([rhs]).map((id) => id.toLowerCase())
|
||||||
|
return right.some((id) => left.has(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
private getTransferPrefix(content: string, myWxid?: string, senderWxid?: string, isSend?: boolean): '[转账]' | '[转账收款]' {
|
||||||
|
const normalizedContent = this.normalizeAppMessageContent(content || '')
|
||||||
|
if (!normalizedContent) return '[转账]'
|
||||||
|
|
||||||
|
const paySubtype = this.extractXmlValue(normalizedContent, 'paysubtype')
|
||||||
|
// 转账消息在部分账号数据中 `payer_username` 可能为空,优先用 `paysubtype` 判定
|
||||||
|
// 实测:1=发起侧,3=收款侧
|
||||||
|
if (paySubtype === '3') return '[转账收款]'
|
||||||
|
if (paySubtype === '1') return '[转账]'
|
||||||
|
|
||||||
|
const payerUsername = this.extractXmlValue(normalizedContent, 'payer_username')
|
||||||
|
const receiverUsername = this.extractXmlValue(normalizedContent, 'receiver_username')
|
||||||
|
const senderIsPayer = senderWxid ? this.isSameWxid(senderWxid, payerUsername) : false
|
||||||
|
const senderIsReceiver = senderWxid ? this.isSameWxid(senderWxid, receiverUsername) : false
|
||||||
|
|
||||||
|
// 实测字段语义:sender 命中 receiver_username 为转账发起侧,命中 payer_username 为收款侧
|
||||||
|
if (senderWxid) {
|
||||||
|
if (senderIsReceiver && !senderIsPayer) return '[转账]'
|
||||||
|
if (senderIsPayer && !senderIsReceiver) return '[转账收款]'
|
||||||
|
}
|
||||||
|
|
||||||
|
// 兜底:按当前账号角色判断
|
||||||
|
if (myWxid) {
|
||||||
|
if (this.isSameWxid(myWxid, receiverUsername)) return '[转账]'
|
||||||
|
if (this.isSameWxid(myWxid, payerUsername)) return '[转账收款]'
|
||||||
|
}
|
||||||
|
|
||||||
|
return '[转账]'
|
||||||
|
}
|
||||||
|
|
||||||
|
private isTransferExportContent(content: string): boolean {
|
||||||
|
return content.startsWith('[转账]') || content.startsWith('[转账收款]')
|
||||||
|
}
|
||||||
|
|
||||||
|
private appendTransferDesc(content: string, transferDesc: string): string {
|
||||||
|
const prefix = content.startsWith('[转账收款]') ? '[转账收款]' : '[转账]'
|
||||||
|
return content.replace(prefix, `${prefix} (${transferDesc})`)
|
||||||
|
}
|
||||||
|
|
||||||
private looksLikeBase64(s: string): boolean {
|
private looksLikeBase64(s: string): boolean {
|
||||||
if (s.length % 4 !== 0) return false
|
if (s.length % 4 !== 0) return false
|
||||||
return /^[A-Za-z0-9+/=]+$/.test(s)
|
return /^[A-Za-z0-9+/=]+$/.test(s)
|
||||||
@@ -574,7 +625,15 @@ class ExportService {
|
|||||||
* 解析消息内容为可读文本
|
* 解析消息内容为可读文本
|
||||||
* 注意:语音消息在这里返回占位符,实际转文字在导出时异步处理
|
* 注意:语音消息在这里返回占位符,实际转文字在导出时异步处理
|
||||||
*/
|
*/
|
||||||
private parseMessageContent(content: string, localType: number, sessionId?: string, createTime?: number): string | null {
|
private parseMessageContent(
|
||||||
|
content: string,
|
||||||
|
localType: number,
|
||||||
|
sessionId?: string,
|
||||||
|
createTime?: number,
|
||||||
|
myWxid?: string,
|
||||||
|
senderWxid?: string,
|
||||||
|
isSend?: boolean
|
||||||
|
): string | null {
|
||||||
if (!content) return null
|
if (!content) return null
|
||||||
|
|
||||||
// 检查 XML 中的 type 标签(支持大 localType 的情况)
|
// 检查 XML 中的 type 标签(支持大 localType 的情况)
|
||||||
@@ -611,10 +670,11 @@ class ExportService {
|
|||||||
if (type === '2000') {
|
if (type === '2000') {
|
||||||
const feedesc = this.extractXmlValue(content, 'feedesc')
|
const feedesc = this.extractXmlValue(content, 'feedesc')
|
||||||
const payMemo = this.extractXmlValue(content, 'pay_memo')
|
const payMemo = this.extractXmlValue(content, 'pay_memo')
|
||||||
|
const transferPrefix = this.getTransferPrefix(content, myWxid, senderWxid, isSend)
|
||||||
if (feedesc) {
|
if (feedesc) {
|
||||||
return payMemo ? `[转账] ${feedesc} ${payMemo}` : `[转账] ${feedesc}`
|
return payMemo ? `${transferPrefix} ${feedesc} ${payMemo}` : `${transferPrefix} ${feedesc}`
|
||||||
}
|
}
|
||||||
return '[转账]'
|
return transferPrefix
|
||||||
}
|
}
|
||||||
|
|
||||||
if (type === '6') return title ? `[文件] ${title}` : '[文件]'
|
if (type === '6') return title ? `[文件] ${title}` : '[文件]'
|
||||||
@@ -650,10 +710,11 @@ class ExportService {
|
|||||||
if (xmlType === '2000') {
|
if (xmlType === '2000') {
|
||||||
const feedesc = this.extractXmlValue(content, 'feedesc')
|
const feedesc = this.extractXmlValue(content, 'feedesc')
|
||||||
const payMemo = this.extractXmlValue(content, 'pay_memo')
|
const payMemo = this.extractXmlValue(content, 'pay_memo')
|
||||||
|
const transferPrefix = this.getTransferPrefix(content, myWxid, senderWxid, isSend)
|
||||||
if (feedesc) {
|
if (feedesc) {
|
||||||
return payMemo ? `[转账] ${feedesc} ${payMemo}` : `[转账] ${feedesc}`
|
return payMemo ? `${transferPrefix} ${feedesc} ${payMemo}` : `${transferPrefix} ${feedesc}`
|
||||||
}
|
}
|
||||||
return '[转账]'
|
return transferPrefix
|
||||||
}
|
}
|
||||||
|
|
||||||
// 其他类型
|
// 其他类型
|
||||||
@@ -676,7 +737,10 @@ class ExportService {
|
|||||||
content: string,
|
content: string,
|
||||||
localType: number,
|
localType: number,
|
||||||
options: { exportVoiceAsText?: boolean },
|
options: { exportVoiceAsText?: boolean },
|
||||||
voiceTranscript?: string
|
voiceTranscript?: string,
|
||||||
|
myWxid?: string,
|
||||||
|
senderWxid?: string,
|
||||||
|
isSend?: boolean
|
||||||
): string {
|
): string {
|
||||||
const safeContent = content || ''
|
const safeContent = content || ''
|
||||||
|
|
||||||
@@ -742,8 +806,9 @@ class ExportService {
|
|||||||
if (subType === 2000 || title.includes('转账') || normalized.includes('transfer')) {
|
if (subType === 2000 || title.includes('转账') || normalized.includes('transfer')) {
|
||||||
const feedesc = this.extractXmlValue(normalized, 'feedesc')
|
const feedesc = this.extractXmlValue(normalized, 'feedesc')
|
||||||
const payMemo = this.extractXmlValue(normalized, 'pay_memo')
|
const payMemo = this.extractXmlValue(normalized, 'pay_memo')
|
||||||
|
const transferPrefix = this.getTransferPrefix(normalized, myWxid, senderWxid, isSend)
|
||||||
if (feedesc) {
|
if (feedesc) {
|
||||||
return payMemo ? `[转账]${feedesc} ${payMemo}` : `[转账]${feedesc}`
|
return payMemo ? `${transferPrefix}${feedesc} ${payMemo}` : `${transferPrefix}${feedesc}`
|
||||||
}
|
}
|
||||||
const amount = this.extractAmountFromText(
|
const amount = this.extractAmountFromText(
|
||||||
[
|
[
|
||||||
@@ -756,7 +821,7 @@ class ExportService {
|
|||||||
.filter(Boolean)
|
.filter(Boolean)
|
||||||
.join(' ')
|
.join(' ')
|
||||||
)
|
)
|
||||||
return amount ? `[转账]${amount}` : '[转账]'
|
return amount ? `${transferPrefix}${amount}` : transferPrefix
|
||||||
}
|
}
|
||||||
|
|
||||||
if (subType === 3 || normalized.includes('<musicurl') || normalized.includes('<songname')) {
|
if (subType === 3 || normalized.includes('<musicurl') || normalized.includes('<songname')) {
|
||||||
@@ -1256,7 +1321,7 @@ class ExportService {
|
|||||||
return rendered.join('')
|
return rendered.join('')
|
||||||
}
|
}
|
||||||
|
|
||||||
private formatHtmlMessageText(content: string, localType: number): string {
|
private formatHtmlMessageText(content: string, localType: number, myWxid?: string, senderWxid?: string, isSend?: boolean): string {
|
||||||
if (!content) return ''
|
if (!content) return ''
|
||||||
|
|
||||||
if (localType === 1) {
|
if (localType === 1) {
|
||||||
@@ -1264,10 +1329,59 @@ class ExportService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (localType === 34) {
|
if (localType === 34) {
|
||||||
return this.parseMessageContent(content, localType) || ''
|
return this.parseMessageContent(content, localType, undefined, undefined, myWxid, senderWxid, isSend) || ''
|
||||||
}
|
}
|
||||||
|
|
||||||
return this.formatPlainExportContent(content, localType, { exportVoiceAsText: false })
|
return this.formatPlainExportContent(content, localType, { exportVoiceAsText: false }, undefined, myWxid, senderWxid, isSend)
|
||||||
|
}
|
||||||
|
|
||||||
|
private extractHtmlLinkCard(content: string, localType: number): { title: string; url: string } | null {
|
||||||
|
if (!content) return null
|
||||||
|
|
||||||
|
const normalized = this.normalizeAppMessageContent(content)
|
||||||
|
const isAppMessage = localType === 49 || normalized.includes('<appmsg') || normalized.includes('<msg>')
|
||||||
|
if (!isAppMessage) return null
|
||||||
|
|
||||||
|
const subType = this.extractXmlValue(normalized, 'type')
|
||||||
|
if (subType && subType !== '5' && subType !== '49') return null
|
||||||
|
|
||||||
|
const url = this.normalizeHtmlLinkUrl(this.extractXmlValue(normalized, 'url'))
|
||||||
|
if (!url) return null
|
||||||
|
|
||||||
|
const title = this.extractXmlValue(normalized, 'title') || this.extractXmlValue(normalized, 'des') || url
|
||||||
|
return { title, url }
|
||||||
|
}
|
||||||
|
|
||||||
|
private normalizeHtmlLinkUrl(rawUrl: string): string {
|
||||||
|
const value = (rawUrl || '').trim()
|
||||||
|
if (!value) return ''
|
||||||
|
|
||||||
|
const parseHttpUrl = (candidate: string): string => {
|
||||||
|
try {
|
||||||
|
const parsed = new URL(candidate)
|
||||||
|
if (parsed.protocol === 'http:' || parsed.protocol === 'https:') {
|
||||||
|
return parsed.toString()
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
|
||||||
|
if (value.startsWith('//')) {
|
||||||
|
return parseHttpUrl(`https:${value}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
const direct = parseHttpUrl(value)
|
||||||
|
if (direct) return direct
|
||||||
|
|
||||||
|
const hasScheme = /^[a-zA-Z][a-zA-Z0-9+.-]*:/.test(value)
|
||||||
|
const isDomainLike = /^[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}(?:[/:?#].*)?$/.test(value)
|
||||||
|
if (!hasScheme && isDomainLike) {
|
||||||
|
return parseHttpUrl(`https://${value}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ''
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -1365,13 +1479,17 @@ class ExportService {
|
|||||||
result.localPath = thumbResult.localPath
|
result.localPath = thumbResult.localPath
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 为每条消息生成稳定且唯一的文件名前缀,避免跨日期/消息发生同名覆盖
|
||||||
|
const messageId = String(msg.localId || Date.now())
|
||||||
|
const imageKey = (imageMd5 || imageDatName || 'image').replace(/[^a-zA-Z0-9_-]/g, '')
|
||||||
|
|
||||||
// 从 data URL 或 file URL 获取实际路径
|
// 从 data URL 或 file URL 获取实际路径
|
||||||
let sourcePath = result.localPath
|
let sourcePath = result.localPath
|
||||||
if (sourcePath.startsWith('data:')) {
|
if (sourcePath.startsWith('data:')) {
|
||||||
// 是 data URL,需要保存为文件
|
// 是 data URL,需要保存为文件
|
||||||
const base64Data = sourcePath.split(',')[1]
|
const base64Data = sourcePath.split(',')[1]
|
||||||
const ext = this.getExtFromDataUrl(sourcePath)
|
const ext = this.getExtFromDataUrl(sourcePath)
|
||||||
const fileName = `${imageMd5 || imageDatName || msg.localId}${ext}`
|
const fileName = `${messageId}_${imageKey}${ext}`
|
||||||
const destPath = path.join(imagesDir, fileName)
|
const destPath = path.join(imagesDir, fileName)
|
||||||
|
|
||||||
fs.writeFileSync(destPath, Buffer.from(base64Data, 'base64'))
|
fs.writeFileSync(destPath, Buffer.from(base64Data, 'base64'))
|
||||||
@@ -1387,7 +1505,7 @@ class ExportService {
|
|||||||
// 复制文件
|
// 复制文件
|
||||||
if (!fs.existsSync(sourcePath)) return null
|
if (!fs.existsSync(sourcePath)) return null
|
||||||
const ext = path.extname(sourcePath) || '.jpg'
|
const ext = path.extname(sourcePath) || '.jpg'
|
||||||
const fileName = `${imageMd5 || imageDatName || msg.localId}${ext}`
|
const fileName = `${messageId}_${imageKey}${ext}`
|
||||||
const destPath = path.join(imagesDir, fileName)
|
const destPath = path.join(imagesDir, fileName)
|
||||||
|
|
||||||
if (!fs.existsSync(destPath)) {
|
if (!fs.existsSync(destPath)) {
|
||||||
@@ -1479,49 +1597,30 @@ class ExportService {
|
|||||||
fs.mkdirSync(emojisDir, { recursive: true })
|
fs.mkdirSync(emojisDir, { recursive: true })
|
||||||
}
|
}
|
||||||
|
|
||||||
// 使用消息对象中已提取的字段
|
// 使用 chatService 下载表情包 (利用其重试和 fallback 逻辑)
|
||||||
const emojiUrl = msg.emojiCdnUrl
|
const localPath = await chatService.downloadEmojiFile(msg)
|
||||||
const emojiMd5 = msg.emojiMd5
|
|
||||||
|
|
||||||
if (!emojiUrl && !emojiMd5) {
|
|
||||||
|
|
||||||
|
if (!localPath || !fs.existsSync(localPath)) {
|
||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 确定目标文件名
|
||||||
|
const ext = path.extname(localPath) || '.gif'
|
||||||
const key = emojiMd5 || String(msg.localId)
|
const key = msg.emojiMd5 || String(msg.localId)
|
||||||
// 根据 URL 判断扩展名
|
|
||||||
let ext = '.gif'
|
|
||||||
if (emojiUrl) {
|
|
||||||
if (emojiUrl.includes('.png')) ext = '.png'
|
|
||||||
else if (emojiUrl.includes('.jpg') || emojiUrl.includes('.jpeg')) ext = '.jpg'
|
|
||||||
}
|
|
||||||
const fileName = `${key}${ext}`
|
const fileName = `${key}${ext}`
|
||||||
const destPath = path.join(emojisDir, fileName)
|
const destPath = path.join(emojisDir, fileName)
|
||||||
|
|
||||||
// 如果已存在则跳过
|
// 复制文件到导出目录 (如果不存在)
|
||||||
if (fs.existsSync(destPath)) {
|
if (!fs.existsSync(destPath)) {
|
||||||
|
fs.copyFileSync(localPath, destPath)
|
||||||
|
}
|
||||||
|
|
||||||
return {
|
return {
|
||||||
relativePath: path.posix.join(mediaRelativePrefix, 'emojis', fileName),
|
relativePath: path.posix.join(mediaRelativePrefix, 'emojis', fileName),
|
||||||
kind: 'emoji'
|
kind: 'emoji'
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// 下载表情
|
|
||||||
if (emojiUrl) {
|
|
||||||
const downloaded = await this.downloadFile(emojiUrl, destPath)
|
|
||||||
if (downloaded) {
|
|
||||||
return {
|
|
||||||
relativePath: path.posix.join(mediaRelativePrefix, 'emojis', fileName),
|
|
||||||
kind: 'emoji'
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return null
|
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
|
console.error('ExportService: exportEmoji failed', e)
|
||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1704,7 +1803,8 @@ class ExportService {
|
|||||||
private async collectMessages(
|
private async collectMessages(
|
||||||
sessionId: string,
|
sessionId: string,
|
||||||
cleanedMyWxid: string,
|
cleanedMyWxid: string,
|
||||||
dateRange?: { start: number; end: number } | null
|
dateRange?: { start: number; end: number } | null,
|
||||||
|
senderUsernameFilter?: string
|
||||||
): Promise<{ rows: any[]; memberSet: Map<string, { member: ChatLabMember; avatarUrl?: string }>; firstTime: number | null; lastTime: number | null }> {
|
): Promise<{ rows: any[]; memberSet: Map<string, { member: ChatLabMember; avatarUrl?: string }>; firstTime: number | null; lastTime: number | null }> {
|
||||||
const rows: any[] = []
|
const rows: any[] = []
|
||||||
const memberSet = new Map<string, { member: ChatLabMember; avatarUrl?: string }>()
|
const memberSet = new Map<string, { member: ChatLabMember; avatarUrl?: string }>()
|
||||||
@@ -1765,6 +1865,10 @@ class ExportService {
|
|||||||
} else {
|
} else {
|
||||||
actualSender = isSend ? cleanedMyWxid : (senderUsername || sessionId)
|
actualSender = isSend ? cleanedMyWxid : (senderUsername || sessionId)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (senderUsernameFilter && !this.isSameWxid(actualSender, senderUsernameFilter)) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
senderSet.add(actualSender)
|
senderSet.add(actualSender)
|
||||||
|
|
||||||
// 提取媒体相关字段
|
// 提取媒体相关字段
|
||||||
@@ -2193,7 +2297,7 @@ class ExportService {
|
|||||||
phase: 'preparing'
|
phase: 'preparing'
|
||||||
})
|
})
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
const allMessages = collected.rows
|
const allMessages = collected.rows
|
||||||
|
|
||||||
// 如果没有消息,不创建文件
|
// 如果没有消息,不创建文件
|
||||||
@@ -2354,11 +2458,19 @@ class ExportService {
|
|||||||
// 使用预先转写的文字
|
// 使用预先转写的文字
|
||||||
content = voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]'
|
content = voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]'
|
||||||
} else {
|
} else {
|
||||||
content = this.parseMessageContent(msg.content, msg.localType, sessionId, msg.createTime)
|
content = this.parseMessageContent(
|
||||||
|
msg.content,
|
||||||
|
msg.localType,
|
||||||
|
sessionId,
|
||||||
|
msg.createTime,
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 转账消息:追加 "谁转账给谁" 信息
|
// 转账消息:追加 "谁转账给谁" 信息
|
||||||
if (content && content.startsWith('[转账]') && msg.content) {
|
if (content && this.isTransferExportContent(content) && msg.content) {
|
||||||
const transferDesc = await this.resolveTransferDesc(
|
const transferDesc = await this.resolveTransferDesc(
|
||||||
msg.content,
|
msg.content,
|
||||||
cleanedMyWxid,
|
cleanedMyWxid,
|
||||||
@@ -2369,7 +2481,7 @@ class ExportService {
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
if (transferDesc) {
|
if (transferDesc) {
|
||||||
content = content.replace('[转账]', `[转账] (${transferDesc})`)
|
content = this.appendTransferDesc(content, transferDesc)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2580,7 +2692,7 @@ class ExportService {
|
|||||||
phase: 'preparing'
|
phase: 'preparing'
|
||||||
})
|
})
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
|
|
||||||
// 如果没有消息,不创建文件
|
// 如果没有消息,不创建文件
|
||||||
if (collected.rows.length === 0) {
|
if (collected.rows.length === 0) {
|
||||||
@@ -2724,11 +2836,19 @@ class ExportService {
|
|||||||
} else if (mediaItem) {
|
} else if (mediaItem) {
|
||||||
content = mediaItem.relativePath
|
content = mediaItem.relativePath
|
||||||
} else {
|
} else {
|
||||||
content = this.parseMessageContent(msg.content, msg.localType)
|
content = this.parseMessageContent(
|
||||||
|
msg.content,
|
||||||
|
msg.localType,
|
||||||
|
undefined,
|
||||||
|
undefined,
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 转账消息:追加 "谁转账给谁" 信息
|
// 转账消息:追加 "谁转账给谁" 信息
|
||||||
if (content && content.startsWith('[转账]') && msg.content) {
|
if (content && this.isTransferExportContent(content) && msg.content) {
|
||||||
const transferDesc = await this.resolveTransferDesc(
|
const transferDesc = await this.resolveTransferDesc(
|
||||||
msg.content,
|
msg.content,
|
||||||
cleanedMyWxid,
|
cleanedMyWxid,
|
||||||
@@ -2742,7 +2862,7 @@ class ExportService {
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
if (transferDesc) {
|
if (transferDesc) {
|
||||||
content = content.replace('[转账]', `[转账] (${transferDesc})`)
|
content = this.appendTransferDesc(content, transferDesc)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2906,7 +3026,7 @@ class ExportService {
|
|||||||
phase: 'preparing'
|
phase: 'preparing'
|
||||||
})
|
})
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
|
|
||||||
// 如果没有消息,不创建文件
|
// 如果没有消息,不创建文件
|
||||||
if (collected.rows.length === 0) {
|
if (collected.rows.length === 0) {
|
||||||
@@ -3215,19 +3335,25 @@ class ExportService {
|
|||||||
msg.content,
|
msg.content,
|
||||||
msg.localType,
|
msg.localType,
|
||||||
options,
|
options,
|
||||||
voiceTranscriptMap.get(msg.localId)
|
voiceTranscriptMap.get(msg.localId),
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
)
|
)
|
||||||
: (mediaItem?.relativePath
|
: (mediaItem?.relativePath
|
||||||
|| this.formatPlainExportContent(
|
|| this.formatPlainExportContent(
|
||||||
msg.content,
|
msg.content,
|
||||||
msg.localType,
|
msg.localType,
|
||||||
options,
|
options,
|
||||||
voiceTranscriptMap.get(msg.localId)
|
voiceTranscriptMap.get(msg.localId),
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
))
|
))
|
||||||
|
|
||||||
// 转账消息:追加 "谁转账给谁" 信息
|
// 转账消息:追加 "谁转账给谁" 信息
|
||||||
let enrichedContentValue = contentValue
|
let enrichedContentValue = contentValue
|
||||||
if (contentValue.startsWith('[转账]') && msg.content) {
|
if (this.isTransferExportContent(contentValue) && msg.content) {
|
||||||
const transferDesc = await this.resolveTransferDesc(
|
const transferDesc = await this.resolveTransferDesc(
|
||||||
msg.content,
|
msg.content,
|
||||||
cleanedMyWxid,
|
cleanedMyWxid,
|
||||||
@@ -3241,7 +3367,7 @@ class ExportService {
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
if (transferDesc) {
|
if (transferDesc) {
|
||||||
enrichedContentValue = contentValue.replace('[转账]', `[转账] (${transferDesc})`)
|
enrichedContentValue = this.appendTransferDesc(contentValue, transferDesc)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -3387,7 +3513,7 @@ class ExportService {
|
|||||||
phase: 'preparing'
|
phase: 'preparing'
|
||||||
})
|
})
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
|
|
||||||
// 如果没有消息,不创建文件
|
// 如果没有消息,不创建文件
|
||||||
if (collected.rows.length === 0) {
|
if (collected.rows.length === 0) {
|
||||||
@@ -3526,19 +3652,25 @@ class ExportService {
|
|||||||
msg.content,
|
msg.content,
|
||||||
msg.localType,
|
msg.localType,
|
||||||
options,
|
options,
|
||||||
voiceTranscriptMap.get(msg.localId)
|
voiceTranscriptMap.get(msg.localId),
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
)
|
)
|
||||||
: (mediaItem?.relativePath
|
: (mediaItem?.relativePath
|
||||||
|| this.formatPlainExportContent(
|
|| this.formatPlainExportContent(
|
||||||
msg.content,
|
msg.content,
|
||||||
msg.localType,
|
msg.localType,
|
||||||
options,
|
options,
|
||||||
voiceTranscriptMap.get(msg.localId)
|
voiceTranscriptMap.get(msg.localId),
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
))
|
))
|
||||||
|
|
||||||
// 转账消息:追加 "谁转账给谁" 信息
|
// 转账消息:追加 "谁转账给谁" 信息
|
||||||
let enrichedContentValue = contentValue
|
let enrichedContentValue = contentValue
|
||||||
if (contentValue.startsWith('[转账]') && msg.content) {
|
if (this.isTransferExportContent(contentValue) && msg.content) {
|
||||||
const transferDesc = await this.resolveTransferDesc(
|
const transferDesc = await this.resolveTransferDesc(
|
||||||
msg.content,
|
msg.content,
|
||||||
cleanedMyWxid,
|
cleanedMyWxid,
|
||||||
@@ -3552,7 +3684,7 @@ class ExportService {
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
if (transferDesc) {
|
if (transferDesc) {
|
||||||
enrichedContentValue = contentValue.replace('[转账]', `[转账] (${transferDesc})`)
|
enrichedContentValue = this.appendTransferDesc(contentValue, transferDesc)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -3661,7 +3793,7 @@ class ExportService {
|
|||||||
phase: 'preparing'
|
phase: 'preparing'
|
||||||
})
|
})
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
if (collected.rows.length === 0) {
|
if (collected.rows.length === 0) {
|
||||||
return { success: false, error: '该会话在指定时间范围内没有消息' }
|
return { success: false, error: '该会话在指定时间范围内没有消息' }
|
||||||
}
|
}
|
||||||
@@ -3824,7 +3956,15 @@ class ExportService {
|
|||||||
|
|
||||||
const msgText = msg.localType === 34 && options.exportVoiceAsText
|
const msgText = msg.localType === 34 && options.exportVoiceAsText
|
||||||
? (voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]')
|
? (voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]')
|
||||||
: (this.parseMessageContent(msg.content, msg.localType, sessionId, msg.createTime) || '')
|
: (this.parseMessageContent(
|
||||||
|
msg.content,
|
||||||
|
msg.localType,
|
||||||
|
sessionId,
|
||||||
|
msg.createTime,
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
|
) || '')
|
||||||
const src = this.getWeCloneSource(msg, typeName, mediaItem)
|
const src = this.getWeCloneSource(msg, typeName, mediaItem)
|
||||||
|
|
||||||
const row = [
|
const row = [
|
||||||
@@ -3974,6 +4114,15 @@ class ExportService {
|
|||||||
const isGroup = sessionId.includes('@chatroom')
|
const isGroup = sessionId.includes('@chatroom')
|
||||||
const sessionInfo = await this.getContactInfo(sessionId)
|
const sessionInfo = await this.getContactInfo(sessionId)
|
||||||
const myInfo = await this.getContactInfo(cleanedMyWxid)
|
const myInfo = await this.getContactInfo(cleanedMyWxid)
|
||||||
|
const contactCache = new Map<string, { success: boolean; contact?: any; error?: string }>()
|
||||||
|
const getContactCached = async (username: string) => {
|
||||||
|
if (contactCache.has(username)) {
|
||||||
|
return contactCache.get(username)!
|
||||||
|
}
|
||||||
|
const result = await wcdbService.getContact(username)
|
||||||
|
contactCache.set(username, result)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
onProgress?.({
|
onProgress?.({
|
||||||
current: 0,
|
current: 0,
|
||||||
@@ -3986,13 +4135,31 @@ class ExportService {
|
|||||||
await this.ensureVoiceModel(onProgress)
|
await this.ensureVoiceModel(onProgress)
|
||||||
}
|
}
|
||||||
|
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
|
|
||||||
// 如果没有消息,不创建文件
|
// 如果没有消息,不创建文件
|
||||||
if (collected.rows.length === 0) {
|
if (collected.rows.length === 0) {
|
||||||
return { success: false, error: '该会话在指定时间范围内没有消息' }
|
return { success: false, error: '该会话在指定时间范围内没有消息' }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const senderUsernames = new Set<string>()
|
||||||
|
for (const msg of collected.rows) {
|
||||||
|
if (msg.senderUsername) senderUsernames.add(msg.senderUsername)
|
||||||
|
}
|
||||||
|
senderUsernames.add(sessionId)
|
||||||
|
await this.preloadContacts(senderUsernames, contactCache)
|
||||||
|
|
||||||
|
const groupNicknameCandidates = isGroup
|
||||||
|
? this.buildGroupNicknameIdCandidates([
|
||||||
|
...Array.from(senderUsernames.values()),
|
||||||
|
...collected.rows.map(msg => msg.senderUsername),
|
||||||
|
cleanedMyWxid
|
||||||
|
])
|
||||||
|
: []
|
||||||
|
const groupNicknamesMap = isGroup
|
||||||
|
? await this.getGroupNicknamesForRoom(sessionId, groupNicknameCandidates)
|
||||||
|
: new Map<string, string>()
|
||||||
|
|
||||||
if (isGroup) {
|
if (isGroup) {
|
||||||
await this.mergeGroupMembers(sessionId, collected.memberSet, options.exportAvatars === true)
|
await this.mergeGroupMembers(sessionId, collected.memberSet, options.exportAvatars === true)
|
||||||
}
|
}
|
||||||
@@ -4198,13 +4365,38 @@ class ExportService {
|
|||||||
const timeText = this.formatTimestamp(msg.createTime)
|
const timeText = this.formatTimestamp(msg.createTime)
|
||||||
const typeName = this.getMessageTypeName(msg.localType)
|
const typeName = this.getMessageTypeName(msg.localType)
|
||||||
|
|
||||||
let textContent = this.formatHtmlMessageText(msg.content, msg.localType)
|
let textContent = this.formatHtmlMessageText(
|
||||||
|
msg.content,
|
||||||
|
msg.localType,
|
||||||
|
cleanedMyWxid,
|
||||||
|
msg.senderUsername,
|
||||||
|
msg.isSend
|
||||||
|
)
|
||||||
if (msg.localType === 34 && useVoiceTranscript) {
|
if (msg.localType === 34 && useVoiceTranscript) {
|
||||||
textContent = voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]'
|
textContent = voiceTranscriptMap.get(msg.localId) || '[语音消息 - 转文字失败]'
|
||||||
}
|
}
|
||||||
if (mediaItem && (msg.localType === 3 || msg.localType === 47)) {
|
if (mediaItem && (msg.localType === 3 || msg.localType === 47)) {
|
||||||
textContent = ''
|
textContent = ''
|
||||||
}
|
}
|
||||||
|
if (this.isTransferExportContent(textContent) && msg.content) {
|
||||||
|
const transferDesc = await this.resolveTransferDesc(
|
||||||
|
msg.content,
|
||||||
|
cleanedMyWxid,
|
||||||
|
groupNicknamesMap,
|
||||||
|
async (username) => {
|
||||||
|
const c = await getContactCached(username)
|
||||||
|
if (c.success && c.contact) {
|
||||||
|
return c.contact.remark || c.contact.nickName || c.contact.alias || username
|
||||||
|
}
|
||||||
|
return username
|
||||||
|
}
|
||||||
|
)
|
||||||
|
if (transferDesc) {
|
||||||
|
textContent = this.appendTransferDesc(textContent, transferDesc)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const linkCard = this.extractHtmlLinkCard(msg.content, msg.localType)
|
||||||
|
|
||||||
let mediaHtml = ''
|
let mediaHtml = ''
|
||||||
if (mediaItem?.kind === 'image') {
|
if (mediaItem?.kind === 'image') {
|
||||||
@@ -4220,9 +4412,11 @@ class ExportService {
|
|||||||
mediaHtml = `<video class="message-media video" controls preload="metadata"${posterAttr} src="${this.escapeAttribute(encodeURI(mediaItem.relativePath))}"></video>`
|
mediaHtml = `<video class="message-media video" controls preload="metadata"${posterAttr} src="${this.escapeAttribute(encodeURI(mediaItem.relativePath))}"></video>`
|
||||||
}
|
}
|
||||||
|
|
||||||
const textHtml = textContent
|
const textHtml = linkCard
|
||||||
|
? `<div class="message-text"><a class="message-link-card" href="${this.escapeAttribute(linkCard.url)}" target="_blank" rel="noopener noreferrer">${this.renderTextWithEmoji(linkCard.title).replace(/\r?\n/g, '<br />')}</a></div>`
|
||||||
|
: (textContent
|
||||||
? `<div class="message-text">${this.renderTextWithEmoji(textContent).replace(/\r?\n/g, '<br />')}</div>`
|
? `<div class="message-text">${this.renderTextWithEmoji(textContent).replace(/\r?\n/g, '<br />')}</div>`
|
||||||
: ''
|
: '')
|
||||||
const senderNameHtml = isGroup
|
const senderNameHtml = isGroup
|
||||||
? `<div class="sender-name">${this.escapeHtml(senderName)}</div>`
|
? `<div class="sender-name">${this.escapeHtml(senderName)}</div>`
|
||||||
: ''
|
: ''
|
||||||
@@ -4413,7 +4607,7 @@ class ExportService {
|
|||||||
|
|
||||||
for (const sessionId of sessionIds) {
|
for (const sessionId of sessionIds) {
|
||||||
const sessionInfo = await this.getContactInfo(sessionId)
|
const sessionInfo = await this.getContactInfo(sessionId)
|
||||||
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange)
|
const collected = await this.collectMessages(sessionId, cleanedMyWxid, options.dateRange, options.senderUsername)
|
||||||
const msgs = collected.rows
|
const msgs = collected.rows
|
||||||
const voiceMsgs = msgs.filter(m => m.localType === 34)
|
const voiceMsgs = msgs.filter(m => m.localType === 34)
|
||||||
const mediaMsgs = msgs.filter(m => {
|
const mediaMsgs = msgs.filter(m => {
|
||||||
@@ -4512,7 +4706,10 @@ class ExportService {
|
|||||||
phase: 'exporting'
|
phase: 'exporting'
|
||||||
})
|
})
|
||||||
|
|
||||||
const safeName = sessionInfo.displayName.replace(/[<>:"\/\\|?*]/g, '_').replace(/\.+$/, '')
|
const sanitizeName = (value: string) => value.replace(/[<>:"\/\\|?*]/g, '_').replace(/\.+$/, '').trim()
|
||||||
|
const baseName = sanitizeName(sessionInfo.displayName || sessionId) || sanitizeName(sessionId) || 'session'
|
||||||
|
const suffix = sanitizeName(options.fileNameSuffix || '')
|
||||||
|
const safeName = suffix ? `${baseName}_${suffix}` : baseName
|
||||||
const useSessionFolder = sessionLayout === 'per-session'
|
const useSessionFolder = sessionLayout === 'per-session'
|
||||||
const sessionDir = useSessionFolder ? path.join(outputDir, safeName) : outputDir
|
const sessionDir = useSessionFolder ? path.join(outputDir, safeName) : outputDir
|
||||||
|
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import ExcelJS from 'exceljs'
|
|||||||
import { ConfigService } from './config'
|
import { ConfigService } from './config'
|
||||||
import { wcdbService } from './wcdbService'
|
import { wcdbService } from './wcdbService'
|
||||||
import { chatService } from './chatService'
|
import { chatService } from './chatService'
|
||||||
|
import type { Message } from './chatService'
|
||||||
|
|
||||||
export interface GroupChatInfo {
|
export interface GroupChatInfo {
|
||||||
username: string
|
username: string
|
||||||
@@ -339,6 +340,92 @@ class GroupAnalyticsService {
|
|||||||
return `${year}-${month}-${day} ${hour}:${minute}:${second}`
|
return `${year}-${month}-${day} ${hour}:${minute}:${second}`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private formatUnixTime(createTime: number): string {
|
||||||
|
if (!Number.isFinite(createTime) || createTime <= 0) return ''
|
||||||
|
const milliseconds = createTime > 1e12 ? createTime : createTime * 1000
|
||||||
|
const date = new Date(milliseconds)
|
||||||
|
if (Number.isNaN(date.getTime())) return String(createTime)
|
||||||
|
return this.formatDateTime(date)
|
||||||
|
}
|
||||||
|
|
||||||
|
private getSimpleMessageTypeName(localType: number): string {
|
||||||
|
const typeMap: Record<number, string> = {
|
||||||
|
1: '文本',
|
||||||
|
3: '图片',
|
||||||
|
34: '语音',
|
||||||
|
42: '名片',
|
||||||
|
43: '视频',
|
||||||
|
47: '表情',
|
||||||
|
48: '位置',
|
||||||
|
49: '链接/文件',
|
||||||
|
50: '通话',
|
||||||
|
10000: '系统',
|
||||||
|
266287972401: '拍一拍',
|
||||||
|
8594229559345: '红包',
|
||||||
|
8589934592049: '转账'
|
||||||
|
}
|
||||||
|
return typeMap[localType] || `类型(${localType})`
|
||||||
|
}
|
||||||
|
|
||||||
|
private normalizeIdCandidates(values: Array<string | null | undefined>): string[] {
|
||||||
|
return this.buildIdCandidates(values).map(value => value.toLowerCase())
|
||||||
|
}
|
||||||
|
|
||||||
|
private isSameAccountIdentity(left: string | null | undefined, right: string | null | undefined): boolean {
|
||||||
|
const leftCandidates = this.normalizeIdCandidates([left])
|
||||||
|
const rightCandidates = this.normalizeIdCandidates([right])
|
||||||
|
if (leftCandidates.length === 0 || rightCandidates.length === 0) return false
|
||||||
|
|
||||||
|
const rightSet = new Set(rightCandidates)
|
||||||
|
for (const leftCandidate of leftCandidates) {
|
||||||
|
if (rightSet.has(leftCandidate)) return true
|
||||||
|
for (const rightCandidate of rightCandidates) {
|
||||||
|
if (leftCandidate.startsWith(`${rightCandidate}_`) || rightCandidate.startsWith(`${leftCandidate}_`)) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
private resolveExportMessageContent(message: Message): string {
|
||||||
|
const parsed = String(message.parsedContent || '').trim()
|
||||||
|
if (parsed) return parsed
|
||||||
|
const raw = String(message.rawContent || '').trim()
|
||||||
|
if (raw) return raw
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
|
||||||
|
private async collectMessagesByMember(
|
||||||
|
chatroomId: string,
|
||||||
|
memberUsername: string,
|
||||||
|
startTime: number,
|
||||||
|
endTime: number
|
||||||
|
): Promise<{ success: boolean; data?: Message[]; error?: string }> {
|
||||||
|
const batchSize = 500
|
||||||
|
const matchedMessages: Message[] = []
|
||||||
|
let offset = 0
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
const batch = await chatService.getMessages(chatroomId, offset, batchSize, startTime, endTime, true)
|
||||||
|
if (!batch.success || !batch.messages) {
|
||||||
|
return { success: false, error: batch.error || '获取群消息失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const message of batch.messages) {
|
||||||
|
if (this.isSameAccountIdentity(memberUsername, message.senderUsername)) {
|
||||||
|
matchedMessages.push(message)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const fetchedCount = batch.messages.length
|
||||||
|
if (fetchedCount <= 0 || !batch.hasMore) break
|
||||||
|
offset += fetchedCount
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success: true, data: matchedMessages }
|
||||||
|
}
|
||||||
|
|
||||||
async getGroupChats(): Promise<{ success: boolean; data?: GroupChatInfo[]; error?: string }> {
|
async getGroupChats(): Promise<{ success: boolean; data?: GroupChatInfo[]; error?: string }> {
|
||||||
try {
|
try {
|
||||||
const conn = await this.ensureConnected()
|
const conn = await this.ensureConnected()
|
||||||
@@ -611,6 +698,181 @@ class GroupAnalyticsService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async exportGroupMemberMessages(
|
||||||
|
chatroomId: string,
|
||||||
|
memberUsername: string,
|
||||||
|
outputPath: string,
|
||||||
|
startTime?: number,
|
||||||
|
endTime?: number
|
||||||
|
): Promise<{ success: boolean; count?: number; error?: string }> {
|
||||||
|
try {
|
||||||
|
const conn = await this.ensureConnected()
|
||||||
|
if (!conn.success) return { success: false, error: conn.error }
|
||||||
|
|
||||||
|
const normalizedChatroomId = String(chatroomId || '').trim()
|
||||||
|
const normalizedMemberUsername = String(memberUsername || '').trim()
|
||||||
|
if (!normalizedChatroomId) return { success: false, error: '群聊ID不能为空' }
|
||||||
|
if (!normalizedMemberUsername) return { success: false, error: '成员ID不能为空' }
|
||||||
|
|
||||||
|
const beginTimestamp = Number.isFinite(startTime) && typeof startTime === 'number'
|
||||||
|
? Math.max(0, Math.floor(startTime))
|
||||||
|
: 0
|
||||||
|
const endTimestampValue = Number.isFinite(endTime) && typeof endTime === 'number'
|
||||||
|
? Math.max(0, Math.floor(endTime))
|
||||||
|
: 0
|
||||||
|
|
||||||
|
const exportDate = new Date()
|
||||||
|
const exportTime = this.formatDateTime(exportDate)
|
||||||
|
const exportVersion = '0.0.2'
|
||||||
|
const exportGenerator = 'WeFlow'
|
||||||
|
const exportPlatform = 'wechat'
|
||||||
|
|
||||||
|
const groupDisplay = await wcdbService.getDisplayNames([normalizedChatroomId, normalizedMemberUsername])
|
||||||
|
const groupName = groupDisplay.success && groupDisplay.map
|
||||||
|
? (groupDisplay.map[normalizedChatroomId] || normalizedChatroomId)
|
||||||
|
: normalizedChatroomId
|
||||||
|
const defaultMemberDisplayName = groupDisplay.success && groupDisplay.map
|
||||||
|
? (groupDisplay.map[normalizedMemberUsername] || normalizedMemberUsername)
|
||||||
|
: normalizedMemberUsername
|
||||||
|
|
||||||
|
let memberDisplayName = defaultMemberDisplayName
|
||||||
|
let memberAlias = ''
|
||||||
|
let memberRemark = ''
|
||||||
|
let memberGroupNickname = ''
|
||||||
|
const membersResult = await this.getGroupMembers(normalizedChatroomId)
|
||||||
|
if (membersResult.success && membersResult.data) {
|
||||||
|
const matchedMember = membersResult.data.find((item) =>
|
||||||
|
this.isSameAccountIdentity(item.username, normalizedMemberUsername)
|
||||||
|
)
|
||||||
|
if (matchedMember) {
|
||||||
|
memberDisplayName = matchedMember.displayName || defaultMemberDisplayName
|
||||||
|
memberAlias = matchedMember.alias || ''
|
||||||
|
memberRemark = matchedMember.remark || ''
|
||||||
|
memberGroupNickname = matchedMember.groupNickname || ''
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const collected = await this.collectMessagesByMember(
|
||||||
|
normalizedChatroomId,
|
||||||
|
normalizedMemberUsername,
|
||||||
|
beginTimestamp,
|
||||||
|
endTimestampValue
|
||||||
|
)
|
||||||
|
if (!collected.success || !collected.data) {
|
||||||
|
return { success: false, error: collected.error || '获取成员消息失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
const records = collected.data.map((message, index) => ({
|
||||||
|
index: index + 1,
|
||||||
|
time: this.formatUnixTime(message.createTime),
|
||||||
|
sender: message.senderUsername || '',
|
||||||
|
messageType: this.getSimpleMessageTypeName(message.localType),
|
||||||
|
content: this.resolveExportMessageContent(message)
|
||||||
|
}))
|
||||||
|
|
||||||
|
fs.mkdirSync(path.dirname(outputPath), { recursive: true })
|
||||||
|
const ext = path.extname(outputPath).toLowerCase()
|
||||||
|
if (ext === '.csv') {
|
||||||
|
const infoTitleRow = ['会话信息']
|
||||||
|
const infoRow = ['群聊ID', normalizedChatroomId, '', '群聊名称', groupName, '成员wxid', normalizedMemberUsername, '']
|
||||||
|
const memberRow = ['成员显示名', memberDisplayName, '成员备注', memberRemark, '群昵称', memberGroupNickname, '微信号', memberAlias]
|
||||||
|
const metaRow = ['导出工具', exportGenerator, '导出版本', exportVersion, '平台', exportPlatform, '导出时间', exportTime]
|
||||||
|
const header = ['序号', '时间', '发送者wxid', '消息类型', '内容']
|
||||||
|
|
||||||
|
const csvRows: string[][] = [infoTitleRow, infoRow, memberRow, metaRow, header]
|
||||||
|
for (const record of records) {
|
||||||
|
csvRows.push([String(record.index), record.time, record.sender, record.messageType, record.content])
|
||||||
|
}
|
||||||
|
|
||||||
|
const csvLines = csvRows.map((row) => row.map((cell) => this.escapeCsvValue(cell)).join(','))
|
||||||
|
const content = '\ufeff' + csvLines.join('\n')
|
||||||
|
fs.writeFileSync(outputPath, content, 'utf8')
|
||||||
|
} else {
|
||||||
|
const workbook = new ExcelJS.Workbook()
|
||||||
|
const worksheet = workbook.addWorksheet(this.sanitizeWorksheetName('成员消息记录'))
|
||||||
|
|
||||||
|
worksheet.getCell(1, 1).value = '会话信息'
|
||||||
|
worksheet.getCell(1, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getRow(1).height = 24
|
||||||
|
|
||||||
|
worksheet.getCell(2, 1).value = '群聊ID'
|
||||||
|
worksheet.getCell(2, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.mergeCells(2, 2, 2, 3)
|
||||||
|
worksheet.getCell(2, 2).value = normalizedChatroomId
|
||||||
|
|
||||||
|
worksheet.getCell(2, 4).value = '群聊名称'
|
||||||
|
worksheet.getCell(2, 4).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(2, 5).value = groupName
|
||||||
|
worksheet.getCell(2, 6).value = '成员wxid'
|
||||||
|
worksheet.getCell(2, 6).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.mergeCells(2, 7, 2, 8)
|
||||||
|
worksheet.getCell(2, 7).value = normalizedMemberUsername
|
||||||
|
|
||||||
|
worksheet.getCell(3, 1).value = '成员显示名'
|
||||||
|
worksheet.getCell(3, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(3, 2).value = memberDisplayName
|
||||||
|
worksheet.getCell(3, 3).value = '成员备注'
|
||||||
|
worksheet.getCell(3, 3).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(3, 4).value = memberRemark
|
||||||
|
worksheet.getCell(3, 5).value = '群昵称'
|
||||||
|
worksheet.getCell(3, 5).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(3, 6).value = memberGroupNickname
|
||||||
|
worksheet.getCell(3, 7).value = '微信号'
|
||||||
|
worksheet.getCell(3, 7).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(3, 8).value = memberAlias
|
||||||
|
|
||||||
|
worksheet.getCell(4, 1).value = '导出工具'
|
||||||
|
worksheet.getCell(4, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(4, 2).value = exportGenerator
|
||||||
|
worksheet.getCell(4, 3).value = '导出版本'
|
||||||
|
worksheet.getCell(4, 3).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(4, 4).value = exportVersion
|
||||||
|
worksheet.getCell(4, 5).value = '平台'
|
||||||
|
worksheet.getCell(4, 5).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(4, 6).value = exportPlatform
|
||||||
|
worksheet.getCell(4, 7).value = '导出时间'
|
||||||
|
worksheet.getCell(4, 7).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
worksheet.getCell(4, 8).value = exportTime
|
||||||
|
|
||||||
|
const headerRow = worksheet.getRow(5)
|
||||||
|
const header = ['序号', '时间', '发送者wxid', '消息类型', '内容']
|
||||||
|
header.forEach((title, index) => {
|
||||||
|
const cell = headerRow.getCell(index + 1)
|
||||||
|
cell.value = title
|
||||||
|
cell.font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
})
|
||||||
|
headerRow.height = 22
|
||||||
|
|
||||||
|
worksheet.getColumn(1).width = 10
|
||||||
|
worksheet.getColumn(2).width = 22
|
||||||
|
worksheet.getColumn(3).width = 30
|
||||||
|
worksheet.getColumn(4).width = 16
|
||||||
|
worksheet.getColumn(5).width = 90
|
||||||
|
worksheet.getColumn(6).width = 16
|
||||||
|
worksheet.getColumn(7).width = 20
|
||||||
|
worksheet.getColumn(8).width = 24
|
||||||
|
|
||||||
|
let currentRow = 6
|
||||||
|
for (const record of records) {
|
||||||
|
const row = worksheet.getRow(currentRow)
|
||||||
|
row.getCell(1).value = record.index
|
||||||
|
row.getCell(2).value = record.time
|
||||||
|
row.getCell(3).value = record.sender
|
||||||
|
row.getCell(4).value = record.messageType
|
||||||
|
row.getCell(5).value = record.content
|
||||||
|
row.alignment = { vertical: 'top', wrapText: true }
|
||||||
|
currentRow += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
await workbook.xlsx.writeFile(outputPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success: true, count: records.length }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async exportGroupMembers(chatroomId: string, outputPath: string): Promise<{ success: boolean; count?: number; error?: string }> {
|
async exportGroupMembers(chatroomId: string, outputPath: string): Promise<{ success: boolean; count?: number; error?: string }> {
|
||||||
try {
|
try {
|
||||||
const conn = await this.ensureConnected()
|
const conn = await this.ensureConnected()
|
||||||
|
|||||||
@@ -1,12 +1,15 @@
|
|||||||
/**
|
/**
|
||||||
* HTTP API 服务
|
* HTTP API 服务
|
||||||
* 提供 ChatLab 标准化格式的消息查询 API
|
* 提供 ChatLab 标准化格式的消息查询 API
|
||||||
*/
|
*/
|
||||||
import * as http from 'http'
|
import * as http from 'http'
|
||||||
|
import * as fs from 'fs'
|
||||||
|
import * as path from 'path'
|
||||||
import { URL } from 'url'
|
import { URL } from 'url'
|
||||||
import { chatService, Message } from './chatService'
|
import { chatService, Message } from './chatService'
|
||||||
import { wcdbService } from './wcdbService'
|
import { wcdbService } from './wcdbService'
|
||||||
import { ConfigService } from './config'
|
import { ConfigService } from './config'
|
||||||
|
import { videoService } from './videoService'
|
||||||
|
|
||||||
// ChatLab 格式定义
|
// ChatLab 格式定义
|
||||||
interface ChatLabHeader {
|
interface ChatLabHeader {
|
||||||
@@ -42,6 +45,7 @@ interface ChatLabMessage {
|
|||||||
content: string | null
|
content: string | null
|
||||||
platformMessageId?: string
|
platformMessageId?: string
|
||||||
replyToMessageId?: string
|
replyToMessageId?: string
|
||||||
|
mediaPath?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
interface ChatLabData {
|
interface ChatLabData {
|
||||||
@@ -51,6 +55,22 @@ interface ChatLabData {
|
|||||||
messages: ChatLabMessage[]
|
messages: ChatLabMessage[]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
interface ApiMediaOptions {
|
||||||
|
enabled: boolean
|
||||||
|
exportImages: boolean
|
||||||
|
exportVoices: boolean
|
||||||
|
exportVideos: boolean
|
||||||
|
exportEmojis: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
type MediaKind = 'image' | 'voice' | 'video' | 'emoji'
|
||||||
|
|
||||||
|
interface ApiExportedMedia {
|
||||||
|
kind: MediaKind
|
||||||
|
fileName: string
|
||||||
|
fullPath: string
|
||||||
|
}
|
||||||
|
|
||||||
// ChatLab 消息类型映射
|
// ChatLab 消息类型映射
|
||||||
const ChatLabType = {
|
const ChatLabType = {
|
||||||
TEXT: 0,
|
TEXT: 0,
|
||||||
@@ -163,6 +183,10 @@ class HttpService {
|
|||||||
return this.port
|
return this.port
|
||||||
}
|
}
|
||||||
|
|
||||||
|
getDefaultMediaExportPath(): string {
|
||||||
|
return this.getApiMediaExportPath()
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 处理 HTTP 请求
|
* 处理 HTTP 请求
|
||||||
*/
|
*/
|
||||||
@@ -213,7 +237,7 @@ class HttpService {
|
|||||||
ascending: boolean
|
ascending: boolean
|
||||||
): Promise<{ success: boolean; messages?: Message[]; hasMore?: boolean; error?: string }> {
|
): Promise<{ success: boolean; messages?: Message[]; hasMore?: boolean; error?: string }> {
|
||||||
try {
|
try {
|
||||||
// 使用固定 batch 大小(与 limit 相同或最大 500)来减少循环次数
|
// 使用固定 batch 大小(与 limit 相同或最多 500)来减少循环次数
|
||||||
const batchSize = Math.min(limit, 500)
|
const batchSize = Math.min(limit, 500)
|
||||||
const beginTimestamp = startTime > 10000000000 ? Math.floor(startTime / 1000) : startTime
|
const beginTimestamp = startTime > 10000000000 ? Math.floor(startTime / 1000) : startTime
|
||||||
const endTimestamp = endTime > 10000000000 ? Math.floor(endTime / 1000) : endTime
|
const endTimestamp = endTime > 10000000000 ? Math.floor(endTime / 1000) : endTime
|
||||||
@@ -240,7 +264,7 @@ class HttpService {
|
|||||||
let rows = batch.rows
|
let rows = batch.rows
|
||||||
hasMore = batch.hasMore === true
|
hasMore = batch.hasMore === true
|
||||||
|
|
||||||
// 处理 offset: 跳过前 N 条
|
// 处理 offset:跳过前 N 条
|
||||||
if (skipped < offset) {
|
if (skipped < offset) {
|
||||||
const remaining = offset - skipped
|
const remaining = offset - skipped
|
||||||
if (remaining >= rows.length) {
|
if (remaining >= rows.length) {
|
||||||
@@ -256,7 +280,7 @@ class HttpService {
|
|||||||
|
|
||||||
const trimmedRows = allRows.slice(0, limit)
|
const trimmedRows = allRows.slice(0, limit)
|
||||||
const finalHasMore = hasMore || allRows.length > limit
|
const finalHasMore = hasMore || allRows.length > limit
|
||||||
const messages = this.mapRowsToMessagesSimple(trimmedRows)
|
const messages = chatService.mapRowsToMessagesForApi(trimmedRows)
|
||||||
return { success: true, messages, hasMore: finalHasMore }
|
return { success: true, messages, hasMore: finalHasMore }
|
||||||
} finally {
|
} finally {
|
||||||
await wcdbService.closeMessageCursor(cursor)
|
await wcdbService.closeMessageCursor(cursor)
|
||||||
@@ -268,154 +292,134 @@ class HttpService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 简单的行数据到 Message 映射(用于 API 输出)
|
* Query param helpers.
|
||||||
*/
|
*/
|
||||||
private mapRowsToMessagesSimple(rows: Record<string, any>[]): Message[] {
|
private parseIntParam(value: string | null, defaultValue: number, min: number, max: number): number {
|
||||||
const myWxid = this.configService.get('myWxid') || ''
|
const parsed = parseInt(value || '', 10)
|
||||||
const messages: Message[] = []
|
if (!Number.isFinite(parsed)) return defaultValue
|
||||||
|
return Math.min(Math.max(parsed, min), max)
|
||||||
for (const row of rows) {
|
|
||||||
const content = this.getField(row, ['message_content', 'messageContent', 'content', 'msg_content', 'WCDB_CT_message_content']) || ''
|
|
||||||
const localType = parseInt(this.getField(row, ['local_type', 'localType', 'type', 'msg_type', 'WCDB_CT_local_type']) || '1', 10)
|
|
||||||
const isSendRaw = this.getField(row, ['computed_is_send', 'computedIsSend', 'is_send', 'isSend', 'WCDB_CT_is_send'])
|
|
||||||
const senderUsername = this.getField(row, ['sender_username', 'senderUsername', 'sender', 'WCDB_CT_sender_username']) || ''
|
|
||||||
const createTime = parseInt(this.getField(row, ['create_time', 'createTime', 'msg_create_time', 'WCDB_CT_create_time']) || '0', 10)
|
|
||||||
const localId = parseInt(this.getField(row, ['local_id', 'localId', 'WCDB_CT_local_id', 'rowid']) || '0', 10)
|
|
||||||
const serverId = this.getField(row, ['server_id', 'serverId', 'WCDB_CT_server_id']) || ''
|
|
||||||
|
|
||||||
let isSend: number
|
|
||||||
if (isSendRaw !== null && isSendRaw !== undefined) {
|
|
||||||
isSend = parseInt(isSendRaw, 10)
|
|
||||||
} else if (senderUsername && myWxid) {
|
|
||||||
isSend = senderUsername.toLowerCase() === myWxid.toLowerCase() ? 1 : 0
|
|
||||||
} else {
|
|
||||||
isSend = 0
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 解析消息内容中的特殊字段
|
private parseBooleanParam(url: URL, keys: string[], defaultValue: boolean = false): boolean {
|
||||||
let parsedContent = content
|
|
||||||
let xmlType: string | undefined
|
|
||||||
let linkTitle: string | undefined
|
|
||||||
let fileName: string | undefined
|
|
||||||
let emojiCdnUrl: string | undefined
|
|
||||||
let emojiMd5: string | undefined
|
|
||||||
let imageMd5: string | undefined
|
|
||||||
let videoMd5: string | undefined
|
|
||||||
let cardNickname: string | undefined
|
|
||||||
|
|
||||||
if (localType === 49 && content) {
|
|
||||||
// 提取 type 子标签
|
|
||||||
const typeMatch = /<type>(\d+)<\/type>/i.exec(content)
|
|
||||||
if (typeMatch) xmlType = typeMatch[1]
|
|
||||||
// 提取 title
|
|
||||||
const titleMatch = /<title>([^<]*)<\/title>/i.exec(content)
|
|
||||||
if (titleMatch) linkTitle = titleMatch[1]
|
|
||||||
// 提取文件名
|
|
||||||
const fnMatch = /<title>([^<]*)<\/title>/i.exec(content)
|
|
||||||
if (fnMatch) fileName = fnMatch[1]
|
|
||||||
}
|
|
||||||
|
|
||||||
if (localType === 47 && content) {
|
|
||||||
const cdnMatch = /cdnurl\s*=\s*"([^"]+)"/i.exec(content)
|
|
||||||
if (cdnMatch) emojiCdnUrl = cdnMatch[1]
|
|
||||||
const md5Match = /md5\s*=\s*"([^"]+)"/i.exec(content)
|
|
||||||
if (md5Match) emojiMd5 = md5Match[1]
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push({
|
|
||||||
localId,
|
|
||||||
talker: '',
|
|
||||||
localType,
|
|
||||||
createTime,
|
|
||||||
sortSeq: createTime,
|
|
||||||
content: parsedContent,
|
|
||||||
isSend,
|
|
||||||
senderUsername,
|
|
||||||
serverId: serverId ? parseInt(serverId, 10) || 0 : 0,
|
|
||||||
rawContent: content,
|
|
||||||
parsedContent: content,
|
|
||||||
emojiCdnUrl,
|
|
||||||
emojiMd5,
|
|
||||||
imageMd5,
|
|
||||||
videoMd5,
|
|
||||||
xmlType,
|
|
||||||
linkTitle,
|
|
||||||
fileName,
|
|
||||||
cardNickname
|
|
||||||
} as Message)
|
|
||||||
}
|
|
||||||
|
|
||||||
return messages
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 从行数据中获取字段值(兼容多种字段名)
|
|
||||||
*/
|
|
||||||
private getField(row: Record<string, any>, keys: string[]): string | null {
|
|
||||||
for (const key of keys) {
|
for (const key of keys) {
|
||||||
if (row[key] !== undefined && row[key] !== null) {
|
const raw = url.searchParams.get(key)
|
||||||
return String(row[key])
|
if (raw === null) continue
|
||||||
|
const normalized = raw.trim().toLowerCase()
|
||||||
|
if (['1', 'true', 'yes', 'on'].includes(normalized)) return true
|
||||||
|
if (['0', 'false', 'no', 'off'].includes(normalized)) return false
|
||||||
|
}
|
||||||
|
return defaultValue
|
||||||
|
}
|
||||||
|
|
||||||
|
private parseMediaOptions(url: URL): ApiMediaOptions {
|
||||||
|
const mediaEnabled = this.parseBooleanParam(url, ['media', 'meiti'], false)
|
||||||
|
if (!mediaEnabled) {
|
||||||
|
return {
|
||||||
|
enabled: false,
|
||||||
|
exportImages: false,
|
||||||
|
exportVoices: false,
|
||||||
|
exportVideos: false,
|
||||||
|
exportEmojis: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
enabled: true,
|
||||||
|
exportImages: this.parseBooleanParam(url, ['image', 'tupian'], true),
|
||||||
|
exportVoices: this.parseBooleanParam(url, ['voice', 'vioce'], true),
|
||||||
|
exportVideos: this.parseBooleanParam(url, ['video'], true),
|
||||||
|
exportEmojis: this.parseBooleanParam(url, ['emoji'], true)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
return null
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* 处理消息查询
|
|
||||||
* GET /api/v1/messages?talker=xxx&limit=100&start=20260101&chatlab=1
|
|
||||||
*/
|
|
||||||
private async handleMessages(url: URL, res: http.ServerResponse): Promise<void> {
|
private async handleMessages(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
const talker = url.searchParams.get('talker')
|
const talker = (url.searchParams.get('talker') || '').trim()
|
||||||
const limit = Math.min(parseInt(url.searchParams.get('limit') || '100', 10), 10000)
|
const limit = this.parseIntParam(url.searchParams.get('limit'), 100, 1, 10000)
|
||||||
const offset = parseInt(url.searchParams.get('offset') || '0', 10)
|
const offset = this.parseIntParam(url.searchParams.get('offset'), 0, 0, Number.MAX_SAFE_INTEGER)
|
||||||
|
const keyword = (url.searchParams.get('keyword') || '').trim().toLowerCase()
|
||||||
const startParam = url.searchParams.get('start')
|
const startParam = url.searchParams.get('start')
|
||||||
const endParam = url.searchParams.get('end')
|
const endParam = url.searchParams.get('end')
|
||||||
const chatlab = url.searchParams.get('chatlab') === '1'
|
const chatlab = this.parseBooleanParam(url, ['chatlab'], false)
|
||||||
const formatParam = url.searchParams.get('format')
|
const formatParam = (url.searchParams.get('format') || '').trim().toLowerCase()
|
||||||
const format = formatParam || (chatlab ? 'chatlab' : 'json')
|
const format = formatParam || (chatlab ? 'chatlab' : 'json')
|
||||||
|
const mediaOptions = this.parseMediaOptions(url)
|
||||||
|
|
||||||
if (!talker) {
|
if (!talker) {
|
||||||
this.sendError(res, 400, 'Missing required parameter: talker')
|
this.sendError(res, 400, 'Missing required parameter: talker')
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 解析时间参数 (支持 YYYYMMDD 格式)
|
if (format !== 'json' && format !== 'chatlab') {
|
||||||
|
this.sendError(res, 400, 'Invalid format, supported: json/chatlab')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
const startTime = this.parseTimeParam(startParam)
|
const startTime = this.parseTimeParam(startParam)
|
||||||
const endTime = this.parseTimeParam(endParam, true)
|
const endTime = this.parseTimeParam(endParam, true)
|
||||||
|
const queryOffset = keyword ? 0 : offset
|
||||||
|
const queryLimit = keyword ? 10000 : limit
|
||||||
|
|
||||||
// 使用批量获取方法,绕过 chatService 的单 batch 限制
|
const result = await this.fetchMessagesBatch(talker, queryOffset, queryLimit, startTime, endTime, true)
|
||||||
const result = await this.fetchMessagesBatch(talker, offset, limit, startTime, endTime, true)
|
|
||||||
if (!result.success || !result.messages) {
|
if (!result.success || !result.messages) {
|
||||||
this.sendError(res, 500, result.error || 'Failed to get messages')
|
this.sendError(res, 500, result.error || 'Failed to get messages')
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if (format === 'chatlab') {
|
let messages = result.messages
|
||||||
// 获取会话显示名
|
let hasMore = result.hasMore === true
|
||||||
|
|
||||||
|
if (keyword) {
|
||||||
|
const filtered = messages.filter((msg) => {
|
||||||
|
const content = (msg.parsedContent || msg.rawContent || '').toLowerCase()
|
||||||
|
return content.includes(keyword)
|
||||||
|
})
|
||||||
|
const endIndex = offset + limit
|
||||||
|
hasMore = filtered.length > endIndex
|
||||||
|
messages = filtered.slice(offset, endIndex)
|
||||||
|
}
|
||||||
|
|
||||||
|
const mediaMap = mediaOptions.enabled
|
||||||
|
? await this.exportMediaForMessages(messages, talker, mediaOptions)
|
||||||
|
: new Map<number, ApiExportedMedia>()
|
||||||
|
|
||||||
const displayNames = await this.getDisplayNames([talker])
|
const displayNames = await this.getDisplayNames([talker])
|
||||||
const talkerName = displayNames[talker] || talker
|
const talkerName = displayNames[talker] || talker
|
||||||
|
|
||||||
const chatLabData = await this.convertToChatLab(result.messages, talker, talkerName)
|
if (format === 'chatlab') {
|
||||||
this.sendJson(res, chatLabData)
|
const chatLabData = await this.convertToChatLab(messages, talker, talkerName, mediaMap)
|
||||||
} else {
|
this.sendJson(res, {
|
||||||
// 返回原始消息格式
|
...chatLabData,
|
||||||
|
media: {
|
||||||
|
enabled: mediaOptions.enabled,
|
||||||
|
exportPath: this.getApiMediaExportPath(),
|
||||||
|
count: mediaMap.size
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const apiMessages = messages.map((msg) => this.toApiMessage(msg, mediaMap.get(msg.localId)))
|
||||||
this.sendJson(res, {
|
this.sendJson(res, {
|
||||||
success: true,
|
success: true,
|
||||||
talker,
|
talker,
|
||||||
count: result.messages.length,
|
count: apiMessages.length,
|
||||||
hasMore: result.hasMore,
|
hasMore,
|
||||||
messages: result.messages
|
media: {
|
||||||
|
enabled: mediaOptions.enabled,
|
||||||
|
exportPath: this.getApiMediaExportPath(),
|
||||||
|
count: mediaMap.size
|
||||||
|
},
|
||||||
|
messages: apiMessages
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 处理会话列表查询
|
* 处理会话列表查询
|
||||||
* GET /api/v1/sessions?keyword=xxx&limit=100
|
* GET /api/v1/sessions?keyword=xxx&limit=100
|
||||||
*/
|
*/
|
||||||
private async handleSessions(url: URL, res: http.ServerResponse): Promise<void> {
|
private async handleSessions(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
const keyword = url.searchParams.get('keyword') || ''
|
const keyword = (url.searchParams.get('keyword') || '').trim()
|
||||||
const limit = parseInt(url.searchParams.get('limit') || '100', 10)
|
const limit = this.parseIntParam(url.searchParams.get('limit'), 100, 1, 10000)
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const sessions = await chatService.getSessions()
|
const sessions = await chatService.getSessions()
|
||||||
@@ -457,8 +461,8 @@ class HttpService {
|
|||||||
* GET /api/v1/contacts?keyword=xxx&limit=100
|
* GET /api/v1/contacts?keyword=xxx&limit=100
|
||||||
*/
|
*/
|
||||||
private async handleContacts(url: URL, res: http.ServerResponse): Promise<void> {
|
private async handleContacts(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
const keyword = url.searchParams.get('keyword') || ''
|
const keyword = (url.searchParams.get('keyword') || '').trim()
|
||||||
const limit = parseInt(url.searchParams.get('limit') || '100', 10)
|
const limit = this.parseIntParam(url.searchParams.get('limit'), 100, 1, 10000)
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const contacts = await chatService.getContacts()
|
const contacts = await chatService.getContacts()
|
||||||
@@ -490,6 +494,156 @@ class HttpService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private getApiMediaExportPath(): string {
|
||||||
|
return path.join(this.configService.getCacheBasePath(), 'api-media')
|
||||||
|
}
|
||||||
|
|
||||||
|
private sanitizeFileName(value: string, fallback: string): string {
|
||||||
|
const safe = (value || '')
|
||||||
|
.trim()
|
||||||
|
.replace(/[<>:"/\\|?*\x00-\x1f]/g, '_')
|
||||||
|
.replace(/\.+$/g, '')
|
||||||
|
return safe || fallback
|
||||||
|
}
|
||||||
|
|
||||||
|
private ensureDir(dirPath: string): void {
|
||||||
|
if (!fs.existsSync(dirPath)) {
|
||||||
|
fs.mkdirSync(dirPath, { recursive: true })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private detectImageExt(buffer: Buffer): string {
|
||||||
|
if (buffer.length >= 3 && buffer[0] === 0xff && buffer[1] === 0xd8 && buffer[2] === 0xff) return '.jpg'
|
||||||
|
if (buffer.length >= 8 && buffer.subarray(0, 8).equals(Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]))) return '.png'
|
||||||
|
if (buffer.length >= 6) {
|
||||||
|
const sig6 = buffer.subarray(0, 6).toString('ascii')
|
||||||
|
if (sig6 === 'GIF87a' || sig6 === 'GIF89a') return '.gif'
|
||||||
|
}
|
||||||
|
if (buffer.length >= 12 && buffer.subarray(0, 4).toString('ascii') === 'RIFF' && buffer.subarray(8, 12).toString('ascii') === 'WEBP') return '.webp'
|
||||||
|
if (buffer.length >= 2 && buffer[0] === 0x42 && buffer[1] === 0x4d) return '.bmp'
|
||||||
|
return '.jpg'
|
||||||
|
}
|
||||||
|
|
||||||
|
private async exportMediaForMessages(
|
||||||
|
messages: Message[],
|
||||||
|
talker: string,
|
||||||
|
options: ApiMediaOptions
|
||||||
|
): Promise<Map<number, ApiExportedMedia>> {
|
||||||
|
const mediaMap = new Map<number, ApiExportedMedia>()
|
||||||
|
if (!options.enabled || messages.length === 0) {
|
||||||
|
return mediaMap
|
||||||
|
}
|
||||||
|
|
||||||
|
const sessionDir = path.join(this.getApiMediaExportPath(), this.sanitizeFileName(talker, 'session'))
|
||||||
|
this.ensureDir(sessionDir)
|
||||||
|
|
||||||
|
for (const msg of messages) {
|
||||||
|
const exported = await this.exportMediaForMessage(msg, talker, sessionDir, options)
|
||||||
|
if (exported) {
|
||||||
|
mediaMap.set(msg.localId, exported)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return mediaMap
|
||||||
|
}
|
||||||
|
|
||||||
|
private async exportMediaForMessage(
|
||||||
|
msg: Message,
|
||||||
|
talker: string,
|
||||||
|
sessionDir: string,
|
||||||
|
options: ApiMediaOptions
|
||||||
|
): Promise<ApiExportedMedia | null> {
|
||||||
|
try {
|
||||||
|
if (msg.localType === 3 && options.exportImages) {
|
||||||
|
const result = await chatService.getImageData(talker, String(msg.localId))
|
||||||
|
if (result.success && result.data) {
|
||||||
|
const imageBuffer = Buffer.from(result.data, 'base64')
|
||||||
|
const ext = this.detectImageExt(imageBuffer)
|
||||||
|
const fileBase = this.sanitizeFileName(msg.imageMd5 || msg.imageDatName || `image_${msg.localId}`, `image_${msg.localId}`)
|
||||||
|
const fileName = `${fileBase}${ext}`
|
||||||
|
const targetDir = path.join(sessionDir, 'images')
|
||||||
|
const fullPath = path.join(targetDir, fileName)
|
||||||
|
this.ensureDir(targetDir)
|
||||||
|
if (!fs.existsSync(fullPath)) {
|
||||||
|
fs.writeFileSync(fullPath, imageBuffer)
|
||||||
|
}
|
||||||
|
return { kind: 'image', fileName, fullPath }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg.localType === 34 && options.exportVoices) {
|
||||||
|
const result = await chatService.getVoiceData(
|
||||||
|
talker,
|
||||||
|
String(msg.localId),
|
||||||
|
msg.createTime || undefined,
|
||||||
|
msg.serverId || undefined
|
||||||
|
)
|
||||||
|
if (result.success && result.data) {
|
||||||
|
const fileName = `voice_${msg.localId}.wav`
|
||||||
|
const targetDir = path.join(sessionDir, 'voices')
|
||||||
|
const fullPath = path.join(targetDir, fileName)
|
||||||
|
this.ensureDir(targetDir)
|
||||||
|
if (!fs.existsSync(fullPath)) {
|
||||||
|
fs.writeFileSync(fullPath, Buffer.from(result.data, 'base64'))
|
||||||
|
}
|
||||||
|
return { kind: 'voice', fileName, fullPath }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg.localType === 43 && options.exportVideos && msg.videoMd5) {
|
||||||
|
const info = await videoService.getVideoInfo(msg.videoMd5)
|
||||||
|
if (info.exists && info.videoUrl && fs.existsSync(info.videoUrl)) {
|
||||||
|
const ext = path.extname(info.videoUrl) || '.mp4'
|
||||||
|
const fileName = `${this.sanitizeFileName(msg.videoMd5, `video_${msg.localId}`)}${ext}`
|
||||||
|
const targetDir = path.join(sessionDir, 'videos')
|
||||||
|
const fullPath = path.join(targetDir, fileName)
|
||||||
|
this.ensureDir(targetDir)
|
||||||
|
if (!fs.existsSync(fullPath)) {
|
||||||
|
fs.copyFileSync(info.videoUrl, fullPath)
|
||||||
|
}
|
||||||
|
return { kind: 'video', fileName, fullPath }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg.localType === 47 && options.exportEmojis && msg.emojiCdnUrl) {
|
||||||
|
const result = await chatService.downloadEmoji(msg.emojiCdnUrl, msg.emojiMd5)
|
||||||
|
if (result.success && result.localPath && fs.existsSync(result.localPath)) {
|
||||||
|
const sourceExt = path.extname(result.localPath) || '.gif'
|
||||||
|
const fileName = `${this.sanitizeFileName(msg.emojiMd5 || `emoji_${msg.localId}`, `emoji_${msg.localId}`)}${sourceExt}`
|
||||||
|
const targetDir = path.join(sessionDir, 'emojis')
|
||||||
|
const fullPath = path.join(targetDir, fileName)
|
||||||
|
this.ensureDir(targetDir)
|
||||||
|
if (!fs.existsSync(fullPath)) {
|
||||||
|
fs.copyFileSync(result.localPath, fullPath)
|
||||||
|
}
|
||||||
|
return { kind: 'emoji', fileName, fullPath }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.warn('[HttpService] exportMediaForMessage failed:', e)
|
||||||
|
}
|
||||||
|
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
private toApiMessage(msg: Message, media?: ApiExportedMedia): Record<string, any> {
|
||||||
|
return {
|
||||||
|
localId: msg.localId,
|
||||||
|
serverId: msg.serverId,
|
||||||
|
localType: msg.localType,
|
||||||
|
createTime: msg.createTime,
|
||||||
|
sortSeq: msg.sortSeq,
|
||||||
|
isSend: msg.isSend,
|
||||||
|
senderUsername: msg.senderUsername,
|
||||||
|
content: this.getMessageContent(msg),
|
||||||
|
rawContent: msg.rawContent,
|
||||||
|
parsedContent: msg.parsedContent,
|
||||||
|
mediaType: media?.kind,
|
||||||
|
mediaFileName: media?.fileName,
|
||||||
|
mediaPath: media?.fullPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 解析时间参数
|
* 解析时间参数
|
||||||
* 支持 YYYYMMDD 格式,返回秒级时间戳
|
* 支持 YYYYMMDD 格式,返回秒级时间戳
|
||||||
@@ -497,7 +651,7 @@ class HttpService {
|
|||||||
private parseTimeParam(param: string | null, isEnd: boolean = false): number {
|
private parseTimeParam(param: string | null, isEnd: boolean = false): number {
|
||||||
if (!param) return 0
|
if (!param) return 0
|
||||||
|
|
||||||
// 纯数字且长度为8,视为 YYYYMMDD
|
// 纯数字且长度为 8,视为 YYYYMMDD
|
||||||
if (/^\d{8}$/.test(param)) {
|
if (/^\d{8}$/.test(param)) {
|
||||||
const year = parseInt(param.slice(0, 4), 10)
|
const year = parseInt(param.slice(0, 4), 10)
|
||||||
const month = parseInt(param.slice(4, 6), 10) - 1
|
const month = parseInt(param.slice(4, 6), 10) - 1
|
||||||
@@ -539,7 +693,12 @@ class HttpService {
|
|||||||
/**
|
/**
|
||||||
* 转换为 ChatLab 格式
|
* 转换为 ChatLab 格式
|
||||||
*/
|
*/
|
||||||
private async convertToChatLab(messages: Message[], talkerId: string, talkerName: string): Promise<ChatLabData> {
|
private async convertToChatLab(
|
||||||
|
messages: Message[],
|
||||||
|
talkerId: string,
|
||||||
|
talkerName: string,
|
||||||
|
mediaMap: Map<number, ApiExportedMedia> = new Map()
|
||||||
|
): Promise<ChatLabData> {
|
||||||
const isGroup = talkerId.endsWith('@chatroom')
|
const isGroup = talkerId.endsWith('@chatroom')
|
||||||
const myWxid = this.configService.get('myWxid') || ''
|
const myWxid = this.configService.get('myWxid') || ''
|
||||||
|
|
||||||
@@ -603,7 +762,8 @@ class HttpService {
|
|||||||
timestamp: msg.createTime,
|
timestamp: msg.createTime,
|
||||||
type: this.mapMessageType(msg.localType, msg),
|
type: this.mapMessageType(msg.localType, msg),
|
||||||
content: this.getMessageContent(msg),
|
content: this.getMessageContent(msg),
|
||||||
platformMessageId: msg.serverId ? String(msg.serverId) : undefined
|
platformMessageId: msg.serverId ? String(msg.serverId) : undefined,
|
||||||
|
mediaPath: mediaMap.get(msg.localId)?.fullPath
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -705,13 +865,13 @@ class HttpService {
|
|||||||
case 1:
|
case 1:
|
||||||
return msg.rawContent || null
|
return msg.rawContent || null
|
||||||
case 3:
|
case 3:
|
||||||
return msg.imageMd5 || '[图片]'
|
return '[图片]'
|
||||||
case 34:
|
case 34:
|
||||||
return '[语音]'
|
return '[语音]'
|
||||||
case 43:
|
case 43:
|
||||||
return msg.videoMd5 || '[视频]'
|
return '[视频]'
|
||||||
case 47:
|
case 47:
|
||||||
return msg.emojiCdnUrl || msg.emojiMd5 || '[表情]'
|
return '[表情]'
|
||||||
case 42:
|
case 42:
|
||||||
return msg.cardNickname || '[名片]'
|
return msg.cardNickname || '[名片]'
|
||||||
case 48:
|
case 48:
|
||||||
@@ -743,3 +903,4 @@ class HttpService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export const httpService = new HttpService()
|
export const httpService = new HttpService()
|
||||||
|
|
||||||
|
|||||||
@@ -99,23 +99,29 @@ export class Isaac64 {
|
|||||||
this.isaac64();
|
this.isaac64();
|
||||||
this.randcnt = 256;
|
this.randcnt = 256;
|
||||||
}
|
}
|
||||||
return this.randrsl[256 - (this.randcnt--)];
|
return this.randrsl[--this.randcnt];
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Generates a keystream of the specified size (in bytes).
|
* Generates a keystream where each 64-bit block is Big-Endian.
|
||||||
* @param size Size of the keystream in bytes (must be multiple of 8)
|
* This matches WeChat's behavior (Reverse index order + byte reversal).
|
||||||
* @returns Buffer containing the keystream
|
|
||||||
*/
|
*/
|
||||||
public generateKeystream(size: number): Buffer {
|
public generateKeystreamBE(size: number): Buffer {
|
||||||
const stream = new BigUint64Array(size / 8);
|
const buffer = Buffer.allocUnsafe(size);
|
||||||
for (let i = 0; i < stream.length; i++) {
|
const fullBlocks = Math.floor(size / 8);
|
||||||
stream[i] = this.getNext();
|
|
||||||
|
for (let i = 0; i < fullBlocks; i++) {
|
||||||
|
buffer.writeBigUInt64BE(this.getNext(), i * 8);
|
||||||
}
|
}
|
||||||
// WeChat's logic specifically reverses the entire byte array
|
|
||||||
const buffer = Buffer.from(stream.buffer);
|
const remaining = size % 8;
|
||||||
// 注意:根据 worker.html 的逻辑,它是对 Uint8Array 执行 reverse()
|
if (remaining > 0) {
|
||||||
// Array.from(wasmArray).reverse()
|
const lastK = this.getNext();
|
||||||
return buffer.reverse();
|
const temp = Buffer.allocUnsafe(8);
|
||||||
|
temp.writeBigUInt64BE(lastK, 0);
|
||||||
|
temp.copy(buffer, fullBlocks * 8, 0, remaining);
|
||||||
|
}
|
||||||
|
|
||||||
|
return buffer;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,371 +0,0 @@
|
|||||||
import fs from "fs";
|
|
||||||
import { app, BrowserWindow } from "electron";
|
|
||||||
import path from "path";
|
|
||||||
import { ConfigService } from './config';
|
|
||||||
|
|
||||||
// Define interfaces locally to avoid static import of types that might not be available or cause issues
|
|
||||||
type LlamaModel = any;
|
|
||||||
type LlamaContext = any;
|
|
||||||
type LlamaChatSession = any;
|
|
||||||
|
|
||||||
export class LlamaService {
|
|
||||||
private _model: LlamaModel | null = null;
|
|
||||||
private _context: LlamaContext | null = null;
|
|
||||||
private _sequence: any = null;
|
|
||||||
private _session: LlamaChatSession | null = null;
|
|
||||||
private _llama: any = null;
|
|
||||||
private _nodeLlamaCpp: any = null;
|
|
||||||
private configService = new ConfigService();
|
|
||||||
private _initialized = false;
|
|
||||||
|
|
||||||
constructor() {
|
|
||||||
// 延迟初始化,只在需要时初始化
|
|
||||||
}
|
|
||||||
|
|
||||||
public async init() {
|
|
||||||
if (this._initialized) return;
|
|
||||||
|
|
||||||
try {
|
|
||||||
// Dynamic import to handle ESM module in CJS context
|
|
||||||
this._nodeLlamaCpp = await import("node-llama-cpp");
|
|
||||||
this._llama = await this._nodeLlamaCpp.getLlama();
|
|
||||||
this._initialized = true;
|
|
||||||
console.log("[LlamaService] Llama initialized");
|
|
||||||
} catch (error) {
|
|
||||||
console.error("[LlamaService] Failed to initialize Llama:", error);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public async loadModel(modelPath: string) {
|
|
||||||
if (!this._llama) await this.init();
|
|
||||||
|
|
||||||
try {
|
|
||||||
console.log("[LlamaService] Loading model from:", modelPath);
|
|
||||||
if (!this._llama) {
|
|
||||||
throw new Error("Llama not initialized");
|
|
||||||
}
|
|
||||||
this._model = await this._llama.loadModel({
|
|
||||||
modelPath: modelPath,
|
|
||||||
gpuLayers: 'max', // Offload all layers to GPU if possible
|
|
||||||
useMlock: false // Disable mlock to avoid "VirtualLock" errors (common on Windows)
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!this._model) throw new Error("Failed to load model");
|
|
||||||
|
|
||||||
this._context = await this._model.createContext({
|
|
||||||
contextSize: 8192, // Balanced context size for better performance
|
|
||||||
batchSize: 2048 // Increase batch size for better prompt processing speed
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!this._context) throw new Error("Failed to create context");
|
|
||||||
|
|
||||||
this._sequence = this._context.getSequence();
|
|
||||||
|
|
||||||
const { LlamaChatSession } = this._nodeLlamaCpp;
|
|
||||||
this._session = new LlamaChatSession({
|
|
||||||
contextSequence: this._sequence
|
|
||||||
});
|
|
||||||
|
|
||||||
console.log("[LlamaService] Model loaded successfully");
|
|
||||||
return true;
|
|
||||||
} catch (error) {
|
|
||||||
console.error("[LlamaService] Failed to load model:", error);
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public async createSession(systemPrompt?: string) {
|
|
||||||
if (!this._context) throw new Error("Model not loaded");
|
|
||||||
if (!this._nodeLlamaCpp) await this.init();
|
|
||||||
|
|
||||||
const { LlamaChatSession } = this._nodeLlamaCpp;
|
|
||||||
|
|
||||||
if (!this._sequence) {
|
|
||||||
this._sequence = this._context.getSequence();
|
|
||||||
}
|
|
||||||
|
|
||||||
this._session = new LlamaChatSession({
|
|
||||||
contextSequence: this._sequence,
|
|
||||||
systemPrompt: systemPrompt
|
|
||||||
});
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
public async chat(message: string, options: { thinking?: boolean } = {}, onToken: (token: string) => void) {
|
|
||||||
if (!this._session) throw new Error("Session not initialized");
|
|
||||||
|
|
||||||
const thinking = options.thinking ?? false;
|
|
||||||
|
|
||||||
// Sampling parameters based on mode
|
|
||||||
const samplingParams = thinking ? {
|
|
||||||
temperature: 0.6,
|
|
||||||
topP: 0.95,
|
|
||||||
topK: 20,
|
|
||||||
repeatPenalty: 1.5 // PresencePenalty=1.5
|
|
||||||
} : {
|
|
||||||
temperature: 0.7,
|
|
||||||
topP: 0.8,
|
|
||||||
topK: 20,
|
|
||||||
repeatPenalty: 1.5
|
|
||||||
};
|
|
||||||
|
|
||||||
try {
|
|
||||||
const response = await this._session.prompt(message, {
|
|
||||||
...samplingParams,
|
|
||||||
onTextChunk: (chunk: string) => {
|
|
||||||
onToken(chunk);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
return response;
|
|
||||||
} catch (error) {
|
|
||||||
console.error("[LlamaService] Chat error:", error);
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public async getModelStatus(modelPath: string) {
|
|
||||||
try {
|
|
||||||
const exists = fs.existsSync(modelPath);
|
|
||||||
if (!exists) {
|
|
||||||
return { exists: false, path: modelPath };
|
|
||||||
}
|
|
||||||
const stats = fs.statSync(modelPath);
|
|
||||||
return {
|
|
||||||
exists: true,
|
|
||||||
path: modelPath,
|
|
||||||
size: stats.size
|
|
||||||
};
|
|
||||||
} catch (error) {
|
|
||||||
return { exists: false, error: String(error) };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private resolveModelDir(): string {
|
|
||||||
const configured = this.configService.get('whisperModelDir') as string | undefined;
|
|
||||||
if (configured) return configured;
|
|
||||||
return path.join(app.getPath('documents'), 'WeFlow', 'models');
|
|
||||||
}
|
|
||||||
|
|
||||||
public async downloadModel(url: string, savePath: string, onProgress: (payload: { downloaded: number; total: number; speed: number }) => void): Promise<void> {
|
|
||||||
// Ensure directory exists
|
|
||||||
const dir = path.dirname(savePath);
|
|
||||||
if (!fs.existsSync(dir)) {
|
|
||||||
fs.mkdirSync(dir, { recursive: true });
|
|
||||||
}
|
|
||||||
|
|
||||||
console.info(`[LlamaService] Multi-threaded download check for: ${savePath}`);
|
|
||||||
|
|
||||||
if (fs.existsSync(savePath)) {
|
|
||||||
fs.unlinkSync(savePath);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 1. Get total size and check range support
|
|
||||||
let probeResult;
|
|
||||||
try {
|
|
||||||
probeResult = await this.probeUrl(url);
|
|
||||||
} catch (err) {
|
|
||||||
console.warn("[LlamaService] Probe failed, falling back to single-thread.", err);
|
|
||||||
return this.downloadSingleThread(url, savePath, onProgress);
|
|
||||||
}
|
|
||||||
|
|
||||||
const { totalSize, acceptRanges, finalUrl } = probeResult;
|
|
||||||
console.log(`[LlamaService] Total size: ${totalSize}, Accept-Ranges: ${acceptRanges}`);
|
|
||||||
|
|
||||||
if (totalSize <= 0 || !acceptRanges) {
|
|
||||||
console.warn("[LlamaService] Ranges not supported or size unknown, falling back to single-thread.");
|
|
||||||
return this.downloadSingleThread(finalUrl, savePath, onProgress);
|
|
||||||
}
|
|
||||||
|
|
||||||
const threadCount = 4;
|
|
||||||
const chunkSize = Math.ceil(totalSize / threadCount);
|
|
||||||
const fd = fs.openSync(savePath, 'w');
|
|
||||||
|
|
||||||
let downloadedLength = 0;
|
|
||||||
let lastDownloadedLength = 0;
|
|
||||||
let lastTime = Date.now();
|
|
||||||
let speed = 0;
|
|
||||||
|
|
||||||
const speedInterval = setInterval(() => {
|
|
||||||
const now = Date.now();
|
|
||||||
const duration = (now - lastTime) / 1000;
|
|
||||||
if (duration > 0) {
|
|
||||||
speed = (downloadedLength - lastDownloadedLength) / duration;
|
|
||||||
lastDownloadedLength = downloadedLength;
|
|
||||||
lastTime = now;
|
|
||||||
onProgress({ downloaded: downloadedLength, total: totalSize, speed });
|
|
||||||
}
|
|
||||||
}, 1000);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const promises = [];
|
|
||||||
for (let i = 0; i < threadCount; i++) {
|
|
||||||
const start = i * chunkSize;
|
|
||||||
const end = i === threadCount - 1 ? totalSize - 1 : (i + 1) * chunkSize - 1;
|
|
||||||
|
|
||||||
promises.push(this.downloadChunk(finalUrl, fd, start, end, (bytes) => {
|
|
||||||
downloadedLength += bytes;
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(promises);
|
|
||||||
console.log("[LlamaService] Multi-threaded download complete");
|
|
||||||
|
|
||||||
// Final progress update
|
|
||||||
onProgress({ downloaded: totalSize, total: totalSize, speed: 0 });
|
|
||||||
} catch (err) {
|
|
||||||
console.error("[LlamaService] Multi-threaded download failed:", err);
|
|
||||||
throw err;
|
|
||||||
} finally {
|
|
||||||
clearInterval(speedInterval);
|
|
||||||
fs.closeSync(fd);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private async probeUrl(url: string): Promise<{ totalSize: number, acceptRanges: boolean, finalUrl: string }> {
|
|
||||||
const protocol = url.startsWith('https') ? require('https') : require('http');
|
|
||||||
const options = {
|
|
||||||
method: 'GET',
|
|
||||||
headers: {
|
|
||||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
|
||||||
'Referer': 'https://www.modelscope.cn/',
|
|
||||||
'Range': 'bytes=0-0'
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const req = protocol.get(url, options, (res: any) => {
|
|
||||||
if ([301, 302, 307, 308].includes(res.statusCode)) {
|
|
||||||
const location = res.headers.location;
|
|
||||||
const nextUrl = new URL(location, url).href;
|
|
||||||
this.probeUrl(nextUrl).then(resolve).catch(reject);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (res.statusCode !== 206 && res.statusCode !== 200) {
|
|
||||||
reject(new Error(`Probe failed: HTTP ${res.statusCode}`));
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const contentRange = res.headers['content-range'];
|
|
||||||
let totalSize = 0;
|
|
||||||
if (contentRange) {
|
|
||||||
const parts = contentRange.split('/');
|
|
||||||
totalSize = parseInt(parts[parts.length - 1], 10);
|
|
||||||
} else {
|
|
||||||
totalSize = parseInt(res.headers['content-length'] || '0', 10);
|
|
||||||
}
|
|
||||||
|
|
||||||
const acceptRanges = res.headers['accept-ranges'] === 'bytes' || !!contentRange;
|
|
||||||
resolve({ totalSize, acceptRanges, finalUrl: url });
|
|
||||||
res.destroy();
|
|
||||||
});
|
|
||||||
req.on('error', reject);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private async downloadChunk(url: string, fd: number, start: number, end: number, onData: (bytes: number) => void): Promise<void> {
|
|
||||||
const protocol = url.startsWith('https') ? require('https') : require('http');
|
|
||||||
const options = {
|
|
||||||
headers: {
|
|
||||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
|
||||||
'Referer': 'https://www.modelscope.cn/',
|
|
||||||
'Range': `bytes=${start}-${end}`
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const req = protocol.get(url, options, (res: any) => {
|
|
||||||
if (res.statusCode !== 206) {
|
|
||||||
reject(new Error(`Chunk download failed: HTTP ${res.statusCode}`));
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let currentOffset = start;
|
|
||||||
res.on('data', (chunk: Buffer) => {
|
|
||||||
try {
|
|
||||||
fs.writeSync(fd, chunk, 0, chunk.length, currentOffset);
|
|
||||||
currentOffset += chunk.length;
|
|
||||||
onData(chunk.length);
|
|
||||||
} catch (err) {
|
|
||||||
reject(err);
|
|
||||||
res.destroy();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
res.on('end', () => resolve());
|
|
||||||
res.on('error', reject);
|
|
||||||
});
|
|
||||||
req.on('error', reject);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private async downloadSingleThread(url: string, savePath: string, onProgress: (payload: { downloaded: number; total: number; speed: number }) => void): Promise<void> {
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const protocol = url.startsWith('https') ? require('https') : require('http');
|
|
||||||
const options = {
|
|
||||||
headers: {
|
|
||||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
|
||||||
'Referer': 'https://www.modelscope.cn/'
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const request = protocol.get(url, options, (response: any) => {
|
|
||||||
if ([301, 302, 307, 308].includes(response.statusCode)) {
|
|
||||||
const location = response.headers.location;
|
|
||||||
const nextUrl = new URL(location, url).href;
|
|
||||||
this.downloadSingleThread(nextUrl, savePath, onProgress).then(resolve).catch(reject);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (response.statusCode !== 200) {
|
|
||||||
reject(new Error(`Fallback download failed: HTTP ${response.statusCode}`));
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const totalLength = parseInt(response.headers['content-length'] || '0', 10);
|
|
||||||
let downloadedLength = 0;
|
|
||||||
let lastDownloadedLength = 0;
|
|
||||||
let lastTime = Date.now();
|
|
||||||
let speed = 0;
|
|
||||||
|
|
||||||
const fileStream = fs.createWriteStream(savePath);
|
|
||||||
response.pipe(fileStream);
|
|
||||||
|
|
||||||
const speedInterval = setInterval(() => {
|
|
||||||
const now = Date.now();
|
|
||||||
const duration = (now - lastTime) / 1000;
|
|
||||||
if (duration > 0) {
|
|
||||||
speed = (downloadedLength - lastDownloadedLength) / duration;
|
|
||||||
lastDownloadedLength = downloadedLength;
|
|
||||||
lastTime = now;
|
|
||||||
onProgress({ downloaded: downloadedLength, total: totalLength, speed });
|
|
||||||
}
|
|
||||||
}, 1000);
|
|
||||||
|
|
||||||
response.on('data', (chunk: any) => {
|
|
||||||
downloadedLength += chunk.length;
|
|
||||||
});
|
|
||||||
|
|
||||||
fileStream.on('finish', () => {
|
|
||||||
clearInterval(speedInterval);
|
|
||||||
fileStream.close();
|
|
||||||
resolve();
|
|
||||||
});
|
|
||||||
|
|
||||||
fileStream.on('error', (err: any) => {
|
|
||||||
clearInterval(speedInterval);
|
|
||||||
fs.unlink(savePath, () => { });
|
|
||||||
reject(err);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
request.on('error', reject);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
public getModelsPath() {
|
|
||||||
return this.resolveModelDir();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export const llamaService = new LlamaService();
|
|
||||||
@@ -38,6 +38,8 @@ export interface SnsPost {
|
|||||||
likes: string[]
|
likes: string[]
|
||||||
comments: { id: string; nickname: string; content: string; refCommentId: string; refNickname?: string }[]
|
comments: { id: string; nickname: string; content: string; refCommentId: string; refNickname?: string }[]
|
||||||
rawXml?: string
|
rawXml?: string
|
||||||
|
linkTitle?: string
|
||||||
|
linkUrl?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -145,6 +147,18 @@ class SnsService {
|
|||||||
return join(this.getSnsCacheDir(), `${hash}${ext}`)
|
return join(this.getSnsCacheDir(), `${hash}${ext}`)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 获取所有发过朋友圈的用户名列表
|
||||||
|
async getSnsUsernames(): Promise<{ success: boolean; usernames?: string[]; error?: string }> {
|
||||||
|
const result = await wcdbService.execQuery('sns', null, 'SELECT DISTINCT user_name FROM SnsTimeLine')
|
||||||
|
if (!result.success || !result.rows) {
|
||||||
|
// 尝试 userName 列名
|
||||||
|
const result2 = await wcdbService.execQuery('sns', null, 'SELECT DISTINCT userName FROM SnsTimeLine')
|
||||||
|
if (!result2.success || !result2.rows) return { success: false, error: result.error || result2.error }
|
||||||
|
return { success: true, usernames: result2.rows.map((r: any) => r.userName).filter(Boolean) }
|
||||||
|
}
|
||||||
|
return { success: true, usernames: result.rows.map((r: any) => r.user_name).filter(Boolean) }
|
||||||
|
}
|
||||||
|
|
||||||
async getTimeline(limit: number = 20, offset: number = 0, usernames?: string[], keyword?: string, startTime?: number, endTime?: number): Promise<{ success: boolean; timeline?: SnsPost[]; error?: string }> {
|
async getTimeline(limit: number = 20, offset: number = 0, usernames?: string[], keyword?: string, startTime?: number, endTime?: number): Promise<{ success: boolean; timeline?: SnsPost[]; error?: string }> {
|
||||||
const result = await wcdbService.getSnsTimeline(limit, offset, usernames, keyword, startTime, endTime)
|
const result = await wcdbService.getSnsTimeline(limit, offset, usernames, keyword, startTime, endTime)
|
||||||
|
|
||||||
@@ -266,6 +280,367 @@ class SnsService {
|
|||||||
return this.fetchAndDecryptImage(url, key)
|
return this.fetchAndDecryptImage(url, key)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 导出朋友圈动态
|
||||||
|
* 支持筛选条件(用户名、关键词)和媒体文件导出
|
||||||
|
*/
|
||||||
|
async exportTimeline(options: {
|
||||||
|
outputDir: string
|
||||||
|
format: 'json' | 'html'
|
||||||
|
usernames?: string[]
|
||||||
|
keyword?: string
|
||||||
|
exportMedia?: boolean
|
||||||
|
startTime?: number
|
||||||
|
endTime?: number
|
||||||
|
}, progressCallback?: (progress: { current: number; total: number; status: string }) => void): Promise<{ success: boolean; filePath?: string; postCount?: number; mediaCount?: number; error?: string }> {
|
||||||
|
const { outputDir, format, usernames, keyword, exportMedia = false, startTime, endTime } = options
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 确保输出目录存在
|
||||||
|
if (!existsSync(outputDir)) {
|
||||||
|
mkdirSync(outputDir, { recursive: true })
|
||||||
|
}
|
||||||
|
|
||||||
|
// 1. 分页加载全部帖子
|
||||||
|
const allPosts: SnsPost[] = []
|
||||||
|
const pageSize = 50
|
||||||
|
let endTs: number | undefined = endTime // 使用 endTime 作为分页起始上界
|
||||||
|
let hasMore = true
|
||||||
|
|
||||||
|
progressCallback?.({ current: 0, total: 0, status: '正在加载朋友圈数据...' })
|
||||||
|
|
||||||
|
while (hasMore) {
|
||||||
|
const result = await this.getTimeline(pageSize, 0, usernames, keyword, startTime, endTs)
|
||||||
|
if (result.success && result.timeline && result.timeline.length > 0) {
|
||||||
|
allPosts.push(...result.timeline)
|
||||||
|
// 下一页的 endTs 为当前最后一条帖子的时间 - 1
|
||||||
|
const lastTs = result.timeline[result.timeline.length - 1].createTime - 1
|
||||||
|
endTs = lastTs
|
||||||
|
hasMore = result.timeline.length >= pageSize
|
||||||
|
// 如果已经低于 startTime,提前终止
|
||||||
|
if (startTime && lastTs < startTime) {
|
||||||
|
hasMore = false
|
||||||
|
}
|
||||||
|
progressCallback?.({ current: allPosts.length, total: 0, status: `已加载 ${allPosts.length} 条动态...` })
|
||||||
|
} else {
|
||||||
|
hasMore = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (allPosts.length === 0) {
|
||||||
|
return { success: true, filePath: '', postCount: 0, mediaCount: 0 }
|
||||||
|
}
|
||||||
|
|
||||||
|
progressCallback?.({ current: 0, total: allPosts.length, status: `共 ${allPosts.length} 条动态,准备导出...` })
|
||||||
|
|
||||||
|
// 2. 如果需要导出媒体,创建 media 子目录并下载
|
||||||
|
let mediaCount = 0
|
||||||
|
const mediaDir = join(outputDir, 'media')
|
||||||
|
|
||||||
|
if (exportMedia) {
|
||||||
|
if (!existsSync(mediaDir)) {
|
||||||
|
mkdirSync(mediaDir, { recursive: true })
|
||||||
|
}
|
||||||
|
|
||||||
|
// 收集所有媒体下载任务
|
||||||
|
const mediaTasks: { media: SnsMedia; postId: string; mi: number }[] = []
|
||||||
|
for (const post of allPosts) {
|
||||||
|
post.media.forEach((media, mi) => mediaTasks.push({ media, postId: post.id, mi }))
|
||||||
|
}
|
||||||
|
|
||||||
|
// 并发下载(5路)
|
||||||
|
let done = 0
|
||||||
|
const concurrency = 5
|
||||||
|
const runTask = async (task: typeof mediaTasks[0]) => {
|
||||||
|
const { media, postId, mi } = task
|
||||||
|
try {
|
||||||
|
const isVideo = isVideoUrl(media.url)
|
||||||
|
const ext = isVideo ? 'mp4' : 'jpg'
|
||||||
|
const fileName = `${postId}_${mi}.${ext}`
|
||||||
|
const filePath = join(mediaDir, fileName)
|
||||||
|
|
||||||
|
if (existsSync(filePath)) {
|
||||||
|
;(media as any).localPath = `media/${fileName}`
|
||||||
|
mediaCount++
|
||||||
|
} else {
|
||||||
|
const result = await this.fetchAndDecryptImage(media.url, media.key)
|
||||||
|
if (result.success && result.data) {
|
||||||
|
await writeFile(filePath, result.data)
|
||||||
|
;(media as any).localPath = `media/${fileName}`
|
||||||
|
mediaCount++
|
||||||
|
} else if (result.success && result.cachePath) {
|
||||||
|
const cachedData = await readFile(result.cachePath)
|
||||||
|
await writeFile(filePath, cachedData)
|
||||||
|
;(media as any).localPath = `media/${fileName}`
|
||||||
|
mediaCount++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.warn(`[SnsExport] 媒体下载失败: ${task.media.url}`, e)
|
||||||
|
}
|
||||||
|
done++
|
||||||
|
progressCallback?.({ current: done, total: mediaTasks.length, status: `正在下载媒体 (${done}/${mediaTasks.length})...` })
|
||||||
|
}
|
||||||
|
|
||||||
|
// 控制并发的执行器
|
||||||
|
const queue = [...mediaTasks]
|
||||||
|
const workers = Array.from({ length: Math.min(concurrency, queue.length) }, async () => {
|
||||||
|
while (queue.length > 0) {
|
||||||
|
const task = queue.shift()!
|
||||||
|
await runTask(task)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
await Promise.all(workers)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2.5 下载头像
|
||||||
|
const avatarMap = new Map<string, string>()
|
||||||
|
if (format === 'html') {
|
||||||
|
if (!existsSync(mediaDir)) mkdirSync(mediaDir, { recursive: true })
|
||||||
|
const uniqueUsers = [...new Map(allPosts.filter(p => p.avatarUrl).map(p => [p.username, p])).values()]
|
||||||
|
let avatarDone = 0
|
||||||
|
const avatarQueue = [...uniqueUsers]
|
||||||
|
const avatarWorkers = Array.from({ length: Math.min(5, avatarQueue.length) }, async () => {
|
||||||
|
while (avatarQueue.length > 0) {
|
||||||
|
const post = avatarQueue.shift()!
|
||||||
|
try {
|
||||||
|
const fileName = `avatar_${crypto.createHash('md5').update(post.username).digest('hex').slice(0, 8)}.jpg`
|
||||||
|
const filePath = join(mediaDir, fileName)
|
||||||
|
if (existsSync(filePath)) {
|
||||||
|
avatarMap.set(post.username, `media/${fileName}`)
|
||||||
|
} else {
|
||||||
|
const result = await this.fetchAndDecryptImage(post.avatarUrl!)
|
||||||
|
if (result.success && result.data) {
|
||||||
|
await writeFile(filePath, result.data)
|
||||||
|
avatarMap.set(post.username, `media/${fileName}`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) { /* 头像下载失败不影响导出 */ }
|
||||||
|
avatarDone++
|
||||||
|
progressCallback?.({ current: avatarDone, total: uniqueUsers.length, status: `正在下载头像 (${avatarDone}/${uniqueUsers.length})...` })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
await Promise.all(avatarWorkers)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 生成输出文件
|
||||||
|
const timestamp = new Date().toISOString().replace(/[:.]/g, '-').slice(0, 19)
|
||||||
|
let outputFilePath: string
|
||||||
|
|
||||||
|
if (format === 'json') {
|
||||||
|
outputFilePath = join(outputDir, `朋友圈导出_${timestamp}.json`)
|
||||||
|
const exportData = {
|
||||||
|
exportTime: new Date().toISOString(),
|
||||||
|
totalPosts: allPosts.length,
|
||||||
|
filters: {
|
||||||
|
usernames: usernames || [],
|
||||||
|
keyword: keyword || ''
|
||||||
|
},
|
||||||
|
posts: allPosts.map(p => ({
|
||||||
|
id: p.id,
|
||||||
|
username: p.username,
|
||||||
|
nickname: p.nickname,
|
||||||
|
createTime: p.createTime,
|
||||||
|
createTimeStr: new Date(p.createTime * 1000).toLocaleString('zh-CN'),
|
||||||
|
contentDesc: p.contentDesc,
|
||||||
|
type: p.type,
|
||||||
|
media: p.media.map(m => ({
|
||||||
|
url: m.url,
|
||||||
|
thumb: m.thumb,
|
||||||
|
localPath: (m as any).localPath || undefined
|
||||||
|
})),
|
||||||
|
likes: p.likes,
|
||||||
|
comments: p.comments,
|
||||||
|
linkTitle: (p as any).linkTitle,
|
||||||
|
linkUrl: (p as any).linkUrl
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
await writeFile(outputFilePath, JSON.stringify(exportData, null, 2), 'utf-8')
|
||||||
|
} else {
|
||||||
|
// HTML 格式
|
||||||
|
outputFilePath = join(outputDir, `朋友圈导出_${timestamp}.html`)
|
||||||
|
const html = this.generateHtml(allPosts, { usernames, keyword }, avatarMap)
|
||||||
|
await writeFile(outputFilePath, html, 'utf-8')
|
||||||
|
}
|
||||||
|
|
||||||
|
progressCallback?.({ current: allPosts.length, total: allPosts.length, status: '导出完成!' })
|
||||||
|
|
||||||
|
return { success: true, filePath: outputFilePath, postCount: allPosts.length, mediaCount }
|
||||||
|
} catch (e: any) {
|
||||||
|
console.error('[SnsExport] 导出失败:', e)
|
||||||
|
return { success: false, error: e.message || String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成朋友圈 HTML 导出文件
|
||||||
|
*/
|
||||||
|
private generateHtml(posts: SnsPost[], filters: { usernames?: string[]; keyword?: string }, avatarMap?: Map<string, string>): string {
|
||||||
|
const escapeHtml = (str: string) => str
|
||||||
|
.replace(/&/g, '&')
|
||||||
|
.replace(/</g, '<')
|
||||||
|
.replace(/>/g, '>')
|
||||||
|
.replace(/"/g, '"')
|
||||||
|
.replace(/\n/g, '<br>')
|
||||||
|
|
||||||
|
const formatTime = (ts: number) => {
|
||||||
|
const d = new Date(ts * 1000)
|
||||||
|
const now = new Date()
|
||||||
|
const isCurrentYear = d.getFullYear() === now.getFullYear()
|
||||||
|
const pad = (n: number) => String(n).padStart(2, '0')
|
||||||
|
const timeStr = `${pad(d.getHours())}:${pad(d.getMinutes())}`
|
||||||
|
const m = d.getMonth() + 1, day = d.getDate()
|
||||||
|
return isCurrentYear ? `${m}月${day}日 ${timeStr}` : `${d.getFullYear()}年${m}月${day}日 ${timeStr}`
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成头像首字母
|
||||||
|
const avatarLetter = (name: string) => {
|
||||||
|
const ch = name.charAt(0)
|
||||||
|
return escapeHtml(ch || '?')
|
||||||
|
}
|
||||||
|
|
||||||
|
let filterInfo = ''
|
||||||
|
if (filters.keyword) filterInfo += `关键词: "${escapeHtml(filters.keyword)}" `
|
||||||
|
if (filters.usernames && filters.usernames.length > 0) filterInfo += `筛选用户: ${filters.usernames.length} 人`
|
||||||
|
|
||||||
|
const postsHtml = posts.map(post => {
|
||||||
|
const mediaCount = post.media.length
|
||||||
|
const gridClass = mediaCount === 1 ? 'grid-1' : mediaCount === 2 || mediaCount === 4 ? 'grid-2' : 'grid-3'
|
||||||
|
|
||||||
|
const mediaHtml = post.media.map((m, mi) => {
|
||||||
|
const localPath = (m as any).localPath
|
||||||
|
if (localPath) {
|
||||||
|
if (isVideoUrl(m.url)) {
|
||||||
|
return `<div class="mi"><video src="${escapeHtml(localPath)}" controls preload="metadata"></video></div>`
|
||||||
|
}
|
||||||
|
return `<div class="mi"><img src="${escapeHtml(localPath)}" loading="lazy" onclick="openLb(this.src)" alt=""></div>`
|
||||||
|
}
|
||||||
|
return `<div class="mi ml"><a href="${escapeHtml(m.url)}" target="_blank">查看媒体</a></div>`
|
||||||
|
}).join('')
|
||||||
|
|
||||||
|
const linkHtml = post.linkTitle && post.linkUrl
|
||||||
|
? `<a class="lk" href="${escapeHtml(post.linkUrl)}" target="_blank"><span class="lk-t">${escapeHtml(post.linkTitle)}</span><span class="lk-a">›</span></a>`
|
||||||
|
: ''
|
||||||
|
|
||||||
|
const likesHtml = post.likes.length > 0
|
||||||
|
? `<div class="interactions"><div class="likes">♥ ${post.likes.map(l => `<span>${escapeHtml(l)}</span>`).join('、')}</div></div>`
|
||||||
|
: ''
|
||||||
|
|
||||||
|
const commentsHtml = post.comments.length > 0
|
||||||
|
? `<div class="interactions${post.likes.length > 0 ? ' cmt-border' : ''}"><div class="cmts">${post.comments.map(c => {
|
||||||
|
const ref = c.refNickname ? `<span class="re">回复</span><b>${escapeHtml(c.refNickname)}</b>` : ''
|
||||||
|
return `<div class="cmt"><b>${escapeHtml(c.nickname)}</b>${ref}:${escapeHtml(c.content)}</div>`
|
||||||
|
}).join('')}</div></div>`
|
||||||
|
: ''
|
||||||
|
|
||||||
|
const avatarSrc = avatarMap?.get(post.username)
|
||||||
|
const avatarHtml = avatarSrc
|
||||||
|
? `<div class="avatar"><img src="${escapeHtml(avatarSrc)}" alt=""></div>`
|
||||||
|
: `<div class="avatar">${avatarLetter(post.nickname)}</div>`
|
||||||
|
|
||||||
|
return `<div class="post">
|
||||||
|
${avatarHtml}
|
||||||
|
<div class="body">
|
||||||
|
<div class="hd"><span class="nick">${escapeHtml(post.nickname)}</span><span class="tm">${formatTime(post.createTime)}</span></div>
|
||||||
|
${post.contentDesc ? `<div class="txt">${escapeHtml(post.contentDesc)}</div>` : ''}
|
||||||
|
${mediaHtml ? `<div class="mg ${gridClass}">${mediaHtml}</div>` : ''}
|
||||||
|
${linkHtml}
|
||||||
|
${likesHtml}
|
||||||
|
${commentsHtml}
|
||||||
|
</div></div>`
|
||||||
|
}).join('\n')
|
||||||
|
|
||||||
|
return `<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>朋友圈导出</title>
|
||||||
|
<style>
|
||||||
|
*{margin:0;padding:0;box-sizing:border-box}
|
||||||
|
body{font-family:-apple-system,BlinkMacSystemFont,"Segoe UI","PingFang SC","Hiragino Sans GB","Microsoft YaHei",sans-serif;background:var(--bg);color:var(--t1);line-height:1.6;-webkit-font-smoothing:antialiased}
|
||||||
|
:root{--bg:#F0EEE9;--card:rgba(255,255,255,.92);--t1:#3d3d3d;--t2:#666;--t3:#999;--accent:#8B7355;--border:rgba(0,0,0,.08);--bg3:rgba(0,0,0,.03)}
|
||||||
|
@media(prefers-color-scheme:dark){:root{--bg:#1a1a1a;--card:rgba(40,40,40,.85);--t1:#e0e0e0;--t2:#aaa;--t3:#777;--accent:#c4a882;--border:rgba(255,255,255,.1);--bg3:rgba(255,255,255,.06)}}
|
||||||
|
.container{max-width:800px;margin:0 auto;padding:20px 24px 60px}
|
||||||
|
|
||||||
|
/* 页面标题 */
|
||||||
|
.feed-hd{display:flex;align-items:center;justify-content:space-between;margin-bottom:8px;padding:0 4px}
|
||||||
|
.feed-hd h2{font-size:20px;font-weight:700}
|
||||||
|
.feed-hd .info{font-size:12px;color:var(--t3)}
|
||||||
|
|
||||||
|
/* 帖子卡片 - 头像+内容双列 */
|
||||||
|
.post{background:var(--card);border-radius:16px;border:1px solid var(--border);padding:20px;margin-bottom:24px;display:flex;gap:16px;box-shadow:0 2px 8px rgba(0,0,0,.02);transition:transform .2s,box-shadow .2s}
|
||||||
|
.post:hover{transform:translateY(-2px);box-shadow:0 8px 16px rgba(0,0,0,.06)}
|
||||||
|
.avatar{width:48px;height:48px;border-radius:12px;background:var(--accent);color:#fff;display:flex;align-items:center;justify-content:center;font-size:20px;font-weight:600;flex-shrink:0;overflow:hidden}
|
||||||
|
.avatar img{width:100%;height:100%;object-fit:cover}
|
||||||
|
.body{flex:1;min-width:0}
|
||||||
|
.hd{display:flex;flex-direction:column;margin-bottom:8px}
|
||||||
|
.nick{font-size:15px;font-weight:700;color:var(--accent);margin-bottom:2px}
|
||||||
|
.tm{font-size:12px;color:var(--t3)}
|
||||||
|
.txt{font-size:15px;line-height:1.6;white-space:pre-wrap;word-break:break-word;margin-bottom:12px}
|
||||||
|
|
||||||
|
/* 媒体网格 */
|
||||||
|
.mg{display:grid;gap:6px;margin-bottom:12px;max-width:320px}
|
||||||
|
.grid-1{max-width:300px}
|
||||||
|
.grid-1 .mi{border-radius:12px}
|
||||||
|
.grid-1 .mi img{aspect-ratio:auto;max-height:480px;object-fit:contain;background:var(--bg3)}
|
||||||
|
.grid-2{grid-template-columns:1fr 1fr}
|
||||||
|
.grid-3{grid-template-columns:1fr 1fr 1fr}
|
||||||
|
.mi{overflow:hidden;border-radius:12px;background:var(--bg3);position:relative;aspect-ratio:1}
|
||||||
|
.mi img{width:100%;height:100%;object-fit:cover;display:block;cursor:zoom-in;transition:opacity .2s}
|
||||||
|
.mi img:hover{opacity:.9}
|
||||||
|
.mi video{width:100%;height:100%;object-fit:cover;display:block;background:#000}
|
||||||
|
.ml{display:flex;align-items:center;justify-content:center}
|
||||||
|
.ml a{color:var(--accent);text-decoration:none;font-size:13px}
|
||||||
|
|
||||||
|
/* 链接卡片 */
|
||||||
|
.lk{display:flex;align-items:center;gap:10px;padding:10px;background:var(--bg3);border:1px solid var(--border);border-radius:12px;text-decoration:none;color:var(--t1);font-size:14px;margin-bottom:12px;transition:background .15s}
|
||||||
|
.lk:hover{background:var(--border)}
|
||||||
|
.lk-t{flex:1;overflow:hidden;text-overflow:ellipsis;white-space:nowrap;font-weight:600}
|
||||||
|
.lk-a{color:var(--t3);font-size:18px;flex-shrink:0}
|
||||||
|
|
||||||
|
/* 互动区域 */
|
||||||
|
.interactions{margin-top:12px;padding-top:12px;border-top:1px dashed var(--border);font-size:13px}
|
||||||
|
.interactions.cmt-border{border-top:none;padding-top:0;margin-top:8px}
|
||||||
|
.likes{color:var(--accent);font-weight:500;line-height:1.8}
|
||||||
|
.cmts{background:var(--bg3);border-radius:8px;padding:8px 12px;line-height:1.4}
|
||||||
|
.cmt{margin-bottom:4px;color:var(--t2)}
|
||||||
|
.cmt:last-child{margin-bottom:0}
|
||||||
|
.cmt b{color:var(--accent);font-weight:500}
|
||||||
|
.re{color:var(--t3);margin:0 4px;font-size:12px}
|
||||||
|
|
||||||
|
/* 灯箱 */
|
||||||
|
.lb{display:none;position:fixed;inset:0;background:rgba(0,0,0,.92);z-index:9999;align-items:center;justify-content:center;cursor:zoom-out}
|
||||||
|
.lb.on{display:flex}
|
||||||
|
.lb img{max-width:92vw;max-height:92vh;object-fit:contain;border-radius:4px}
|
||||||
|
|
||||||
|
/* 回到顶部 */
|
||||||
|
.btt{position:fixed;right:24px;bottom:32px;width:44px;height:44px;border-radius:50%;background:var(--card);box-shadow:0 2px 12px rgba(0,0,0,.12);border:1px solid var(--border);cursor:pointer;font-size:18px;display:none;align-items:center;justify-content:center;z-index:100;color:var(--t2)}
|
||||||
|
.btt:hover{transform:scale(1.1)}
|
||||||
|
.btt.show{display:flex}
|
||||||
|
|
||||||
|
/* 页脚 */
|
||||||
|
.ft{text-align:center;padding:32px 0 24px;font-size:12px;color:var(--t3)}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="container">
|
||||||
|
<div class="feed-hd"><h2>朋友圈</h2><span class="info">共 ${posts.length} 条${filterInfo ? ` · ${filterInfo}` : ''}</span></div>
|
||||||
|
${postsHtml}
|
||||||
|
<div class="ft">由 WeFlow 导出 · ${new Date().toLocaleString('zh-CN')}</div>
|
||||||
|
</div>
|
||||||
|
<div class="lb" id="lb" onclick="closeLb()"><img id="lbi" src=""></div>
|
||||||
|
<button class="btt" id="btt" onclick="scrollTo({top:0,behavior:'smooth'})">↑</button>
|
||||||
|
<script>
|
||||||
|
function openLb(s){document.getElementById('lbi').src=s;document.getElementById('lb').classList.add('on');document.body.style.overflow='hidden'}
|
||||||
|
function closeLb(){document.getElementById('lb').classList.remove('on');document.body.style.overflow=''}
|
||||||
|
document.addEventListener('keydown',function(e){if(e.key==='Escape')closeLb()})
|
||||||
|
window.addEventListener('scroll',function(){document.getElementById('btt').classList.toggle('show',window.scrollY>600)})
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>`
|
||||||
|
}
|
||||||
|
|
||||||
private async fetchAndDecryptImage(url: string, key?: string | number): Promise<{ success: boolean; data?: Buffer; contentType?: string; cachePath?: string; error?: string }> {
|
private async fetchAndDecryptImage(url: string, key?: string | number): Promise<{ success: boolean; data?: Buffer; contentType?: string; cachePath?: string; error?: string }> {
|
||||||
if (!url) return { success: false, error: 'url 不能为空' }
|
if (!url) return { success: false, error: 'url 不能为空' }
|
||||||
|
|
||||||
@@ -292,7 +667,6 @@ class SnsService {
|
|||||||
// 视频专用下载逻辑 (下载 -> 解密 -> 缓存)
|
// 视频专用下载逻辑 (下载 -> 解密 -> 缓存)
|
||||||
return new Promise(async (resolve) => {
|
return new Promise(async (resolve) => {
|
||||||
const tmpPath = join(require('os').tmpdir(), `sns_video_${Date.now()}_${Math.random().toString(36).slice(2)}.enc`)
|
const tmpPath = join(require('os').tmpdir(), `sns_video_${Date.now()}_${Math.random().toString(36).slice(2)}.enc`)
|
||||||
console.log(`[SnsService] 开始下载视频到临时文件: ${tmpPath}`)
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const https = require('https')
|
const https = require('https')
|
||||||
@@ -322,10 +696,8 @@ class SnsService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
res.pipe(fileStream)
|
res.pipe(fileStream)
|
||||||
|
|
||||||
fileStream.on('finish', async () => {
|
fileStream.on('finish', async () => {
|
||||||
fileStream.close()
|
fileStream.close()
|
||||||
console.log(`[SnsService] 视频下载完成,开始解密... Key: ${key}`)
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const encryptedBuffer = await readFile(tmpPath)
|
const encryptedBuffer = await readFile(tmpPath)
|
||||||
@@ -334,7 +706,6 @@ class SnsService {
|
|||||||
|
|
||||||
if (key && String(key).trim().length > 0) {
|
if (key && String(key).trim().length > 0) {
|
||||||
try {
|
try {
|
||||||
console.log(`[SnsService] 使用 WASM Isaac64 解密视频... Key: ${key}`)
|
|
||||||
const keyText = String(key).trim()
|
const keyText = String(key).trim()
|
||||||
let keystream: Buffer
|
let keystream: Buffer
|
||||||
|
|
||||||
@@ -344,9 +715,8 @@ class SnsService {
|
|||||||
keystream = await wasmService.getKeystream(keyText, 131072)
|
keystream = await wasmService.getKeystream(keyText, 131072)
|
||||||
} catch (wasmErr) {
|
} catch (wasmErr) {
|
||||||
// 打包漏带 wasm 或 wasm 初始化异常时,回退到纯 TS ISAAC64
|
// 打包漏带 wasm 或 wasm 初始化异常时,回退到纯 TS ISAAC64
|
||||||
console.warn(`[SnsService] WASM 解密不可用,回退 Isaac64: ${wasmErr}`)
|
|
||||||
const isaac = new Isaac64(keyText)
|
const isaac = new Isaac64(keyText)
|
||||||
keystream = isaac.generateKeystream(131072)
|
keystream = isaac.generateKeystreamBE(131072)
|
||||||
}
|
}
|
||||||
|
|
||||||
const decryptLen = Math.min(keystream.length, raw.length)
|
const decryptLen = Math.min(keystream.length, raw.length)
|
||||||
@@ -358,23 +728,16 @@ class SnsService {
|
|||||||
|
|
||||||
// 验证 MP4 签名 ('ftyp' at offset 4)
|
// 验证 MP4 签名 ('ftyp' at offset 4)
|
||||||
const ftyp = raw.subarray(4, 8).toString('ascii')
|
const ftyp = raw.subarray(4, 8).toString('ascii')
|
||||||
if (ftyp === 'ftyp') {
|
if (ftyp !== 'ftyp') {
|
||||||
console.log(`[SnsService] 视频解密成功: ${url}`)
|
// 可以在此处记录解密可能失败的标记,但不打印详细 hex
|
||||||
} else {
|
|
||||||
console.warn(`[SnsService] 视频解密可能失败: ${url}, 未找到 ftyp 签名: ${ftyp}`)
|
|
||||||
// 打印前 32 字节用于调试
|
|
||||||
console.warn(`[SnsService] Decrypted Header (first 32 bytes): ${raw.subarray(0, 32).toString('hex')}`)
|
|
||||||
}
|
}
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
console.error(`[SnsService] 视频解密出错: ${err}`)
|
console.error(`[SnsService] 视频解密出错: ${err}`)
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
console.warn(`[SnsService] 未提供 Key,跳过解密,直接保存`)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 写入最终缓存 (覆盖)
|
// 写入最终缓存 (覆盖)
|
||||||
await writeFile(cachePath, raw)
|
await writeFile(cachePath, raw)
|
||||||
console.log(`[SnsService] 视频已保存到缓存: ${cachePath}`)
|
|
||||||
|
|
||||||
// 删除临时文件
|
// 删除临时文件
|
||||||
try { await import('fs/promises').then(fs => fs.unlink(tmpPath)) } catch (e) { }
|
try { await import('fs/promises').then(fs => fs.unlink(tmpPath)) } catch (e) { }
|
||||||
@@ -392,6 +755,12 @@ class SnsService {
|
|||||||
resolve({ success: false, error: e.message })
|
resolve({ success: false, error: e.message })
|
||||||
})
|
})
|
||||||
|
|
||||||
|
req.setTimeout(15000, () => {
|
||||||
|
req.destroy()
|
||||||
|
fs.unlink(tmpPath, () => { })
|
||||||
|
resolve({ success: false, error: '请求超时' })
|
||||||
|
})
|
||||||
|
|
||||||
req.end()
|
req.end()
|
||||||
|
|
||||||
} catch (e: any) {
|
} catch (e: any) {
|
||||||
@@ -444,8 +813,24 @@ class SnsService {
|
|||||||
// 图片逻辑
|
// 图片逻辑
|
||||||
const shouldDecrypt = (xEnc === '1' || !!key) && key !== undefined && key !== null && String(key).trim().length > 0
|
const shouldDecrypt = (xEnc === '1' || !!key) && key !== undefined && key !== null && String(key).trim().length > 0
|
||||||
if (shouldDecrypt) {
|
if (shouldDecrypt) {
|
||||||
const decrypted = await wcdbService.decryptSnsImage(raw, String(key))
|
try {
|
||||||
decoded = Buffer.from(decrypted)
|
const keyStr = String(key).trim()
|
||||||
|
if (/^\d+$/.test(keyStr)) {
|
||||||
|
// 使用 WASM 版本的 Isaac64 解密图片
|
||||||
|
// 修正逻辑:使用带 reverse 且修正了 8字节对齐偏移的 getKeystream
|
||||||
|
const wasmService = WasmService.getInstance()
|
||||||
|
const keystream = await wasmService.getKeystream(keyStr, raw.length)
|
||||||
|
|
||||||
|
const decrypted = Buffer.allocUnsafe(raw.length)
|
||||||
|
for (let i = 0; i < raw.length; i++) {
|
||||||
|
decrypted[i] = raw[i] ^ keystream[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
decoded = decrypted
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[SnsService] TS Decrypt Error:', e)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 写入磁盘缓存
|
// 写入磁盘缓存
|
||||||
@@ -462,6 +847,10 @@ class SnsService {
|
|||||||
})
|
})
|
||||||
|
|
||||||
req.on('error', (e: any) => resolve({ success: false, error: e.message }))
|
req.on('error', (e: any) => resolve({ success: false, error: e.message }))
|
||||||
|
req.setTimeout(15000, () => {
|
||||||
|
req.destroy()
|
||||||
|
resolve({ success: false, error: '请求超时' })
|
||||||
|
})
|
||||||
req.end()
|
req.end()
|
||||||
} catch (e: any) {
|
} catch (e: any) {
|
||||||
resolve({ success: false, error: e.message })
|
resolve({ success: false, error: e.message })
|
||||||
|
|||||||
@@ -46,7 +46,6 @@ export class WasmService {
|
|||||||
const wasmPath = path.join(basePath, 'wasm_video_decode.wasm');
|
const wasmPath = path.join(basePath, 'wasm_video_decode.wasm');
|
||||||
const jsPath = path.join(basePath, 'wasm_video_decode.js');
|
const jsPath = path.join(basePath, 'wasm_video_decode.js');
|
||||||
|
|
||||||
console.log('[WasmService] Loading WASM from:', wasmPath);
|
|
||||||
|
|
||||||
if (!fs.existsSync(wasmPath) || !fs.existsSync(jsPath)) {
|
if (!fs.existsSync(wasmPath) || !fs.existsSync(jsPath)) {
|
||||||
throw new Error(`WASM files not found at ${basePath}`);
|
throw new Error(`WASM files not found at ${basePath}`);
|
||||||
@@ -88,7 +87,6 @@ export class WasmService {
|
|||||||
// Define Module
|
// Define Module
|
||||||
mockGlobal.Module = {
|
mockGlobal.Module = {
|
||||||
onRuntimeInitialized: () => {
|
onRuntimeInitialized: () => {
|
||||||
console.log("[WasmService] WASM Runtime Initialized");
|
|
||||||
this.wasmLoaded = true;
|
this.wasmLoaded = true;
|
||||||
resolve();
|
resolve();
|
||||||
},
|
},
|
||||||
@@ -133,10 +131,24 @@ export class WasmService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public async getKeystream(key: string, size: number = 131072): Promise<Buffer> {
|
public async getKeystream(key: string, size: number = 131072): Promise<Buffer> {
|
||||||
|
// ISAAC-64 uses 8-byte blocks. If size is not a multiple of 8,
|
||||||
|
// the global reverse() will cause a shift in alignment.
|
||||||
|
const alignSize = Math.ceil(size / 8) * 8;
|
||||||
|
const buffer = await this.getRawKeystream(key, alignSize);
|
||||||
|
|
||||||
|
// Reverse the entire aligned buffer
|
||||||
|
const reversed = new Uint8Array(buffer);
|
||||||
|
reversed.reverse();
|
||||||
|
|
||||||
|
// Return exactly the requested size from the beginning of the reversed stream.
|
||||||
|
// Since we reversed the 'aligned' buffer, index 0 is the last byte of the last block.
|
||||||
|
return Buffer.from(reversed).subarray(0, size);
|
||||||
|
}
|
||||||
|
|
||||||
|
public async getRawKeystream(key: string, size: number = 131072): Promise<Buffer> {
|
||||||
await this.init();
|
await this.init();
|
||||||
|
|
||||||
if (!this.module || !this.module.WxIsaac64) {
|
if (!this.module || !this.module.WxIsaac64) {
|
||||||
// Fallback check for asm.WxIsaac64 logic if needed, but debug showed it on Module
|
|
||||||
if (this.module.asm && this.module.asm.WxIsaac64) {
|
if (this.module.asm && this.module.asm.WxIsaac64) {
|
||||||
this.module.WxIsaac64 = this.module.asm.WxIsaac64;
|
this.module.WxIsaac64 = this.module.asm.WxIsaac64;
|
||||||
}
|
}
|
||||||
@@ -149,26 +161,19 @@ export class WasmService {
|
|||||||
try {
|
try {
|
||||||
this.capturedKeystream = null;
|
this.capturedKeystream = null;
|
||||||
const isaac = new this.module.WxIsaac64(key);
|
const isaac = new this.module.WxIsaac64(key);
|
||||||
isaac.generate(size); // This triggers the global.wasm_isaac_generate callback
|
isaac.generate(size);
|
||||||
|
|
||||||
// Cleanup if possible? isaac.delete()?
|
|
||||||
// In worker code: p.decryptor.delete()
|
|
||||||
if (isaac.delete) {
|
if (isaac.delete) {
|
||||||
isaac.delete();
|
isaac.delete();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this.capturedKeystream) {
|
if (this.capturedKeystream) {
|
||||||
// The worker_release.js logic does:
|
return Buffer.from(this.capturedKeystream);
|
||||||
// p.decryptor_array.set(r.reverse())
|
|
||||||
// So the actual keystream is the REVERSE of what is passed to the callback.
|
|
||||||
const reversed = new Uint8Array(this.capturedKeystream);
|
|
||||||
reversed.reverse();
|
|
||||||
return Buffer.from(reversed);
|
|
||||||
} else {
|
} else {
|
||||||
throw new Error('[WasmService] Failed to capture keystream (callback not called)');
|
throw new Error('[WasmService] Failed to capture keystream (callback not called)');
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error('[WasmService] Error generating keystream:', error);
|
console.error('[WasmService] Error generating raw keystream:', error);
|
||||||
throw error;
|
throw error;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -27,6 +27,8 @@ export class WcdbCore {
|
|||||||
private wcdbCloseAccount: any = null
|
private wcdbCloseAccount: any = null
|
||||||
private wcdbSetMyWxid: any = null
|
private wcdbSetMyWxid: any = null
|
||||||
private wcdbFreeString: any = null
|
private wcdbFreeString: any = null
|
||||||
|
private wcdbUpdateMessage: any = null
|
||||||
|
private wcdbDeleteMessage: any = null
|
||||||
private wcdbGetSessions: any = null
|
private wcdbGetSessions: any = null
|
||||||
private wcdbGetMessages: any = null
|
private wcdbGetMessages: any = null
|
||||||
private wcdbGetMessageCount: any = null
|
private wcdbGetMessageCount: any = null
|
||||||
@@ -64,9 +66,13 @@ export class WcdbCore {
|
|||||||
private wcdbVerifyUser: any = null
|
private wcdbVerifyUser: any = null
|
||||||
private wcdbStartMonitorPipe: any = null
|
private wcdbStartMonitorPipe: any = null
|
||||||
private wcdbStopMonitorPipe: any = null
|
private wcdbStopMonitorPipe: any = null
|
||||||
|
private wcdbGetMonitorPipeName: any = null
|
||||||
|
|
||||||
private monitorPipeClient: any = null
|
private monitorPipeClient: any = null
|
||||||
private wcdbDecryptSnsImage: any = null
|
private monitorCallback: ((type: string, json: string) => void) | null = null
|
||||||
|
private monitorReconnectTimer: any = null
|
||||||
|
private monitorPipePath: string = ''
|
||||||
|
|
||||||
|
|
||||||
private avatarUrlCache: Map<string, { url?: string; updatedAt: number }> = new Map()
|
private avatarUrlCache: Map<string, { url?: string; updatedAt: number }> = new Map()
|
||||||
private readonly avatarCacheTtlMs = 10 * 60 * 1000
|
private readonly avatarCacheTtlMs = 10 * 60 * 1000
|
||||||
@@ -90,23 +96,46 @@ export class WcdbCore {
|
|||||||
// 使用命名管道 IPC
|
// 使用命名管道 IPC
|
||||||
startMonitor(callback: (type: string, json: string) => void): boolean {
|
startMonitor(callback: (type: string, json: string) => void): boolean {
|
||||||
if (!this.wcdbStartMonitorPipe) {
|
if (!this.wcdbStartMonitorPipe) {
|
||||||
this.writeLog('startMonitor: wcdbStartMonitorPipe not available')
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
this.monitorCallback = callback
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const result = this.wcdbStartMonitorPipe()
|
const result = this.wcdbStartMonitorPipe()
|
||||||
if (result !== 0) {
|
if (result !== 0) {
|
||||||
this.writeLog(`startMonitor: wcdbStartMonitorPipe failed with ${result}`)
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 从 DLL 获取动态管道名(含 PID)
|
||||||
|
let pipePath = '\\\\.\\pipe\\weflow_monitor'
|
||||||
|
if (this.wcdbGetMonitorPipeName) {
|
||||||
|
try {
|
||||||
|
const namePtr = [null as any]
|
||||||
|
if (this.wcdbGetMonitorPipeName(namePtr) === 0 && namePtr[0]) {
|
||||||
|
pipePath = this.koffi.decode(namePtr[0], 'char', -1)
|
||||||
|
this.wcdbFreeString(namePtr[0])
|
||||||
|
}
|
||||||
|
} catch {}
|
||||||
|
}
|
||||||
|
|
||||||
|
this.connectMonitorPipe(pipePath)
|
||||||
|
return true
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[wcdbCore] startMonitor exception:', e)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 连接命名管道,支持断开后自动重连
|
||||||
|
private connectMonitorPipe(pipePath: string) {
|
||||||
|
this.monitorPipePath = pipePath
|
||||||
const net = require('net')
|
const net = require('net')
|
||||||
const PIPE_PATH = '\\\\.\\pipe\\weflow_monitor'
|
|
||||||
|
|
||||||
setTimeout(() => {
|
setTimeout(() => {
|
||||||
this.monitorPipeClient = net.createConnection(PIPE_PATH, () => {
|
if (!this.monitorCallback) return
|
||||||
this.writeLog('Monitor pipe connected')
|
|
||||||
|
this.monitorPipeClient = net.createConnection(this.monitorPipePath, () => {
|
||||||
})
|
})
|
||||||
|
|
||||||
let buffer = ''
|
let buffer = ''
|
||||||
@@ -118,70 +147,43 @@ export class WcdbCore {
|
|||||||
if (line.trim()) {
|
if (line.trim()) {
|
||||||
try {
|
try {
|
||||||
const parsed = JSON.parse(line)
|
const parsed = JSON.parse(line)
|
||||||
callback(parsed.action || 'update', line)
|
this.monitorCallback?.(parsed.action || 'update', line)
|
||||||
} catch {
|
} catch {
|
||||||
callback('update', line)
|
this.monitorCallback?.('update', line)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
this.monitorPipeClient.on('error', (err: Error) => {
|
this.monitorPipeClient.on('error', () => {
|
||||||
this.writeLog(`Monitor pipe error: ${err.message}`)
|
|
||||||
})
|
})
|
||||||
|
|
||||||
this.monitorPipeClient.on('close', () => {
|
this.monitorPipeClient.on('close', () => {
|
||||||
this.writeLog('Monitor pipe closed')
|
|
||||||
this.monitorPipeClient = null
|
this.monitorPipeClient = null
|
||||||
|
this.scheduleReconnect()
|
||||||
})
|
})
|
||||||
}, 100)
|
}, 100)
|
||||||
|
|
||||||
this.writeLog('Monitor started via named pipe IPC')
|
|
||||||
return true
|
|
||||||
} catch (e) {
|
|
||||||
console.error('打开数据库异常:', e)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
// 定时重连
|
||||||
* 解密朋友圈图片
|
private scheduleReconnect() {
|
||||||
*/
|
if (this.monitorReconnectTimer || !this.monitorCallback) return
|
||||||
async decryptSnsImage(encryptedData: Buffer, key: string): Promise<Buffer> {
|
this.monitorReconnectTimer = setTimeout(() => {
|
||||||
if (!this.initialized) {
|
this.monitorReconnectTimer = null
|
||||||
const initOk = await this.initialize()
|
if (this.monitorCallback && !this.monitorPipeClient) {
|
||||||
if (!initOk) return encryptedData
|
this.connectMonitorPipe(this.monitorPipePath)
|
||||||
|
}
|
||||||
|
}, 3000)
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!this.wcdbDecryptSnsImage) return encryptedData
|
|
||||||
|
|
||||||
try {
|
|
||||||
if (!this.wcdbDecryptSnsImage) {
|
|
||||||
console.error('[WCDB] wcdbDecryptSnsImage func is null')
|
|
||||||
return encryptedData
|
|
||||||
}
|
|
||||||
|
|
||||||
const outPtr = [null as any]
|
|
||||||
// Koffi pass Buffer as char* pointer
|
|
||||||
const result = this.wcdbDecryptSnsImage(encryptedData, encryptedData.length, key, outPtr)
|
|
||||||
|
|
||||||
if (result === 0 && outPtr[0]) {
|
|
||||||
const hex = this.decodeJsonPtr(outPtr[0])
|
|
||||||
if (hex) {
|
|
||||||
return Buffer.from(hex, 'hex')
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
console.error(`[WCDB] Decrypt SNS image failed with code: ${result}`)
|
|
||||||
// 主动获取 DLL 内部日志以诊断问题
|
|
||||||
await this.printLogs(true)
|
|
||||||
}
|
|
||||||
} catch (e) {
|
|
||||||
console.error('解密图片失败:', e)
|
|
||||||
}
|
|
||||||
return encryptedData
|
|
||||||
}
|
|
||||||
|
|
||||||
stopMonitor(): void {
|
stopMonitor(): void {
|
||||||
|
this.monitorCallback = null
|
||||||
|
if (this.monitorReconnectTimer) {
|
||||||
|
clearTimeout(this.monitorReconnectTimer)
|
||||||
|
this.monitorReconnectTimer = null
|
||||||
|
}
|
||||||
if (this.monitorPipeClient) {
|
if (this.monitorPipeClient) {
|
||||||
this.monitorPipeClient.destroy()
|
this.monitorPipeClient.destroy()
|
||||||
this.monitorPipeClient = null
|
this.monitorPipeClient = null
|
||||||
@@ -420,6 +422,20 @@ export class WcdbCore {
|
|||||||
this.wcdbSetMyWxid = null
|
this.wcdbSetMyWxid = null
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// wcdb_status wcdb_update_message(wcdb_handle handle, const char* session_id, int64_t local_id, int32_t create_time, const char* new_content, char** out_error)
|
||||||
|
try {
|
||||||
|
this.wcdbUpdateMessage = this.lib.func('int32 wcdb_update_message(int64 handle, const char* sessionId, int64 localId, int32 createTime, const char* newContent, _Out_ void** outError)')
|
||||||
|
} catch {
|
||||||
|
this.wcdbUpdateMessage = null
|
||||||
|
}
|
||||||
|
|
||||||
|
// wcdb_status wcdb_delete_message(wcdb_handle handle, const char* session_id, int64_t local_id, char** out_error)
|
||||||
|
try {
|
||||||
|
this.wcdbDeleteMessage = this.lib.func('int32 wcdb_delete_message(int64 handle, const char* sessionId, int64 localId, int32 createTime, const char* dbPathHint, _Out_ void** outError)')
|
||||||
|
} catch {
|
||||||
|
this.wcdbDeleteMessage = null
|
||||||
|
}
|
||||||
|
|
||||||
// void wcdb_free_string(char* ptr)
|
// void wcdb_free_string(char* ptr)
|
||||||
this.wcdbFreeString = this.lib.func('void wcdb_free_string(void* ptr)')
|
this.wcdbFreeString = this.lib.func('void wcdb_free_string(void* ptr)')
|
||||||
|
|
||||||
@@ -588,11 +604,13 @@ export class WcdbCore {
|
|||||||
try {
|
try {
|
||||||
this.wcdbStartMonitorPipe = this.lib.func('int32 wcdb_start_monitor_pipe()')
|
this.wcdbStartMonitorPipe = this.lib.func('int32 wcdb_start_monitor_pipe()')
|
||||||
this.wcdbStopMonitorPipe = this.lib.func('void wcdb_stop_monitor_pipe()')
|
this.wcdbStopMonitorPipe = this.lib.func('void wcdb_stop_monitor_pipe()')
|
||||||
|
this.wcdbGetMonitorPipeName = this.lib.func('int32 wcdb_get_monitor_pipe_name(_Out_ void** outName)')
|
||||||
this.writeLog('Monitor pipe functions loaded')
|
this.writeLog('Monitor pipe functions loaded')
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.warn('Failed to load monitor pipe functions:', e)
|
console.warn('Failed to load monitor pipe functions:', e)
|
||||||
this.wcdbStartMonitorPipe = null
|
this.wcdbStartMonitorPipe = null
|
||||||
this.wcdbStopMonitorPipe = null
|
this.wcdbStopMonitorPipe = null
|
||||||
|
this.wcdbGetMonitorPipeName = null
|
||||||
}
|
}
|
||||||
|
|
||||||
// void VerifyUser(int64_t hwnd_ptr, const char* message, char* out_result, int max_len)
|
// void VerifyUser(int64_t hwnd_ptr, const char* message, char* out_result, int max_len)
|
||||||
@@ -602,12 +620,7 @@ export class WcdbCore {
|
|||||||
this.wcdbVerifyUser = null
|
this.wcdbVerifyUser = null
|
||||||
}
|
}
|
||||||
|
|
||||||
// wcdb_status wcdb_decrypt_sns_image(const char* encrypted_data, int32_t data_len, const char* key, char** out_hex)
|
|
||||||
try {
|
|
||||||
this.wcdbDecryptSnsImage = this.lib.func('int32 wcdb_decrypt_sns_image(const char* data, int32 len, const char* key, _Out_ void** outHex)')
|
|
||||||
} catch {
|
|
||||||
this.wcdbDecryptSnsImage = null
|
|
||||||
}
|
|
||||||
|
|
||||||
// 初始化
|
// 初始化
|
||||||
const initResult = this.wcdbInit()
|
const initResult = this.wcdbInit()
|
||||||
@@ -1814,4 +1827,62 @@ export class WcdbCore {
|
|||||||
return { success: false, error: String(e) }
|
return { success: false, error: String(e) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
/**
|
||||||
|
* 修改消息内容
|
||||||
|
*/
|
||||||
|
async updateMessage(sessionId: string, localId: number, createTime: number, newContent: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
if (!this.initialized || !this.wcdbUpdateMessage) return { success: false, error: 'WCDB Not Initialized or Method Missing' }
|
||||||
|
if (!this.handle) return { success: false, error: 'Not Connected' }
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
try {
|
||||||
|
const outError = [null as any]
|
||||||
|
const result = this.wcdbUpdateMessage(this.handle, sessionId, localId, createTime, newContent, outError)
|
||||||
|
|
||||||
|
if (result !== 0) {
|
||||||
|
let errorMsg = 'Unknown Error'
|
||||||
|
if (outError[0]) {
|
||||||
|
errorMsg = this.decodeJsonPtr(outError[0]) || 'Unknown Error (Decode Failed)'
|
||||||
|
}
|
||||||
|
resolve({ success: false, error: errorMsg })
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
resolve({ success: true })
|
||||||
|
} catch (e) {
|
||||||
|
resolve({ success: false, error: String(e) })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除消息
|
||||||
|
*/
|
||||||
|
async deleteMessage(sessionId: string, localId: number, createTime: number, dbPathHint?: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
if (!this.initialized || !this.wcdbDeleteMessage) return { success: false, error: 'WCDB Not Initialized or Method Missing' }
|
||||||
|
if (!this.handle) return { success: false, error: 'Not Connected' }
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
try {
|
||||||
|
const outError = [null as any]
|
||||||
|
const result = this.wcdbDeleteMessage(this.handle, sessionId, localId, createTime || 0, dbPathHint || '', outError)
|
||||||
|
|
||||||
|
if (result !== 0) {
|
||||||
|
let errorMsg = 'Unknown Error'
|
||||||
|
if (outError[0]) {
|
||||||
|
errorMsg = this.decodeJsonPtr(outError[0]) || 'Unknown Error (Decode Failed)'
|
||||||
|
}
|
||||||
|
console.error(`[WcdbCore] deleteMessage fail: code=${result}, error=${errorMsg}`)
|
||||||
|
resolve({ success: false, error: errorMsg })
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
resolve({ success: true })
|
||||||
|
} catch (e) {
|
||||||
|
console.error(`[WcdbCore] deleteMessage exception:`, e)
|
||||||
|
resolve({ success: false, error: String(e) })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -136,7 +136,6 @@ export class WcdbService {
|
|||||||
*/
|
*/
|
||||||
setMonitor(callback: (type: string, json: string) => void): void {
|
setMonitor(callback: (type: string, json: string) => void): void {
|
||||||
this.monitorListener = callback;
|
this.monitorListener = callback;
|
||||||
// Notify worker to enable monitor
|
|
||||||
this.callWorker('setMonitor').catch(() => { });
|
this.callWorker('setMonitor').catch(() => { });
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -432,12 +431,21 @@ export class WcdbService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 解密朋友圈图片
|
* 修改消息内容
|
||||||
*/
|
*/
|
||||||
async decryptSnsImage(encryptedData: Buffer, key: string): Promise<Buffer> {
|
async updateMessage(sessionId: string, localId: number, createTime: number, newContent: string): Promise<{ success: boolean; error?: string }> {
|
||||||
return this.callWorker<Buffer>('decryptSnsImage', { encryptedData, key })
|
return this.callWorker('updateMessage', { sessionId, localId, createTime, newContent })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除消息
|
||||||
|
*/
|
||||||
|
async deleteMessage(sessionId: string, localId: number, createTime: number, dbPathHint?: string): Promise<{ success: boolean; error?: string }> {
|
||||||
|
return this.callWorker('deleteMessage', { sessionId, localId, createTime, dbPathHint })
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export const wcdbService = new WcdbService()
|
export const wcdbService = new WcdbService()
|
||||||
|
|||||||
@@ -150,9 +150,13 @@ if (parentPort) {
|
|||||||
case 'verifyUser':
|
case 'verifyUser':
|
||||||
result = await core.verifyUser(payload.message, payload.hwnd)
|
result = await core.verifyUser(payload.message, payload.hwnd)
|
||||||
break
|
break
|
||||||
case 'decryptSnsImage':
|
case 'updateMessage':
|
||||||
result = await core.decryptSnsImage(payload.encryptedData, payload.key)
|
result = await core.updateMessage(payload.sessionId, payload.localId, payload.createTime, payload.newContent)
|
||||||
break
|
break
|
||||||
|
case 'deleteMessage':
|
||||||
|
result = await core.deleteMessage(payload.sessionId, payload.localId, payload.createTime, payload.dbPathHint)
|
||||||
|
break
|
||||||
|
|
||||||
default:
|
default:
|
||||||
result = { success: false, error: `Unknown method: ${type}` }
|
result = { success: false, error: `Unknown method: ${type}` }
|
||||||
}
|
}
|
||||||
|
|||||||
1922
package-lock.json
generated
1922
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "weflow",
|
"name": "weflow",
|
||||||
"version": "1.5.4",
|
"version": "2.1.0",
|
||||||
"description": "WeFlow",
|
"description": "WeFlow",
|
||||||
"main": "dist-electron/main.js",
|
"main": "dist-electron/main.js",
|
||||||
"author": "cc",
|
"author": "cc",
|
||||||
@@ -32,7 +32,6 @@
|
|||||||
"jszip": "^3.10.1",
|
"jszip": "^3.10.1",
|
||||||
"koffi": "^2.9.0",
|
"koffi": "^2.9.0",
|
||||||
"lucide-react": "^0.562.0",
|
"lucide-react": "^0.562.0",
|
||||||
"node-llama-cpp": "^3.15.1",
|
|
||||||
"react": "^19.2.3",
|
"react": "^19.2.3",
|
||||||
"react-dom": "^19.2.3",
|
"react-dom": "^19.2.3",
|
||||||
"react-markdown": "^10.1.0",
|
"react-markdown": "^10.1.0",
|
||||||
|
|||||||
Binary file not shown.
39
src/App.tsx
39
src/App.tsx
@@ -22,10 +22,9 @@ import SnsPage from './pages/SnsPage'
|
|||||||
import ContactsPage from './pages/ContactsPage'
|
import ContactsPage from './pages/ContactsPage'
|
||||||
import ChatHistoryPage from './pages/ChatHistoryPage'
|
import ChatHistoryPage from './pages/ChatHistoryPage'
|
||||||
import NotificationWindow from './pages/NotificationWindow'
|
import NotificationWindow from './pages/NotificationWindow'
|
||||||
import AIChatPage from './pages/AIChatPage'
|
|
||||||
|
|
||||||
import { useAppStore } from './stores/appStore'
|
import { useAppStore } from './stores/appStore'
|
||||||
import { themes, useThemeStore, type ThemeId } from './stores/themeStore'
|
import { themes, useThemeStore, type ThemeId, type ThemeMode } from './stores/themeStore'
|
||||||
import * as configService from './services/config'
|
import * as configService from './services/config'
|
||||||
import { Download, X, Shield } from 'lucide-react'
|
import { Download, X, Shield } from 'lucide-react'
|
||||||
import './App.scss'
|
import './App.scss'
|
||||||
@@ -101,14 +100,27 @@ function App() {
|
|||||||
|
|
||||||
// 应用主题
|
// 应用主题
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
|
const mq = window.matchMedia('(prefers-color-scheme: dark)')
|
||||||
|
const applyMode = (mode: ThemeMode, systemDark?: boolean) => {
|
||||||
|
const effectiveMode = mode === 'system' ? (systemDark ?? mq.matches ? 'dark' : 'light') : mode
|
||||||
document.documentElement.setAttribute('data-theme', currentTheme)
|
document.documentElement.setAttribute('data-theme', currentTheme)
|
||||||
document.documentElement.setAttribute('data-mode', themeMode)
|
document.documentElement.setAttribute('data-mode', effectiveMode)
|
||||||
|
const symbolColor = effectiveMode === 'dark' ? '#ffffff' : '#1a1a1a'
|
||||||
// 更新窗口控件颜色以适配主题
|
|
||||||
const symbolColor = themeMode === 'dark' ? '#ffffff' : '#1a1a1a'
|
|
||||||
if (!isOnboardingWindow && !isNotificationWindow) {
|
if (!isOnboardingWindow && !isNotificationWindow) {
|
||||||
window.electronAPI.window.setTitleBarOverlay({ symbolColor })
|
window.electronAPI.window.setTitleBarOverlay({ symbolColor })
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
applyMode(themeMode)
|
||||||
|
|
||||||
|
// 监听系统主题变化
|
||||||
|
const handler = (e: MediaQueryListEvent) => {
|
||||||
|
if (useThemeStore.getState().themeMode === 'system') {
|
||||||
|
applyMode('system', e.matches)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mq.addEventListener('change', handler)
|
||||||
|
return () => mq.removeEventListener('change', handler)
|
||||||
}, [currentTheme, themeMode, isOnboardingWindow, isNotificationWindow])
|
}, [currentTheme, themeMode, isOnboardingWindow, isNotificationWindow])
|
||||||
|
|
||||||
// 读取已保存的主题设置
|
// 读取已保存的主题设置
|
||||||
@@ -122,7 +134,7 @@ function App() {
|
|||||||
if (savedThemeId && themes.some((theme) => theme.id === savedThemeId)) {
|
if (savedThemeId && themes.some((theme) => theme.id === savedThemeId)) {
|
||||||
setTheme(savedThemeId as ThemeId)
|
setTheme(savedThemeId as ThemeId)
|
||||||
}
|
}
|
||||||
if (savedThemeMode === 'light' || savedThemeMode === 'dark') {
|
if (savedThemeMode === 'light' || savedThemeMode === 'dark' || savedThemeMode === 'system') {
|
||||||
setThemeMode(savedThemeMode)
|
setThemeMode(savedThemeMode)
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
@@ -182,11 +194,13 @@ function App() {
|
|||||||
if (isNotificationWindow) return // Skip updates in notification window
|
if (isNotificationWindow) return // Skip updates in notification window
|
||||||
|
|
||||||
const removeUpdateListener = window.electronAPI?.app?.onUpdateAvailable?.((info: any) => {
|
const removeUpdateListener = window.electronAPI?.app?.onUpdateAvailable?.((info: any) => {
|
||||||
// 发现新版本时自动打开更新弹窗
|
// 发现新版本时保存更新信息,锁定状态下不弹窗,解锁后再显示
|
||||||
if (info) {
|
if (info) {
|
||||||
setUpdateInfo({ ...info, hasUpdate: true })
|
setUpdateInfo({ ...info, hasUpdate: true })
|
||||||
|
if (!useAppStore.getState().isLocked) {
|
||||||
setShowUpdateDialog(true)
|
setShowUpdateDialog(true)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
})
|
})
|
||||||
const removeProgressListener = window.electronAPI?.app?.onDownloadProgress?.((progress: any) => {
|
const removeProgressListener = window.electronAPI?.app?.onDownloadProgress?.((progress: any) => {
|
||||||
setDownloadProgress(progress)
|
setDownloadProgress(progress)
|
||||||
@@ -197,6 +211,13 @@ function App() {
|
|||||||
}
|
}
|
||||||
}, [setUpdateInfo, setDownloadProgress, setShowUpdateDialog, isNotificationWindow])
|
}, [setUpdateInfo, setDownloadProgress, setShowUpdateDialog, isNotificationWindow])
|
||||||
|
|
||||||
|
// 解锁后显示暂存的更新弹窗
|
||||||
|
useEffect(() => {
|
||||||
|
if (!isLocked && updateInfo?.hasUpdate && !showUpdateDialog && !isDownloading) {
|
||||||
|
setShowUpdateDialog(true)
|
||||||
|
}
|
||||||
|
}, [isLocked])
|
||||||
|
|
||||||
const handleUpdateNow = async () => {
|
const handleUpdateNow = async () => {
|
||||||
setShowUpdateDialog(false)
|
setShowUpdateDialog(false)
|
||||||
setIsDownloading(true)
|
setIsDownloading(true)
|
||||||
@@ -435,7 +456,7 @@ function App() {
|
|||||||
<Route path="/" element={<HomePage />} />
|
<Route path="/" element={<HomePage />} />
|
||||||
<Route path="/home" element={<HomePage />} />
|
<Route path="/home" element={<HomePage />} />
|
||||||
<Route path="/chat" element={<ChatPage />} />
|
<Route path="/chat" element={<ChatPage />} />
|
||||||
<Route path="/ai-chat" element={<AIChatPage />} />
|
|
||||||
<Route path="/analytics" element={<AnalyticsWelcomePage />} />
|
<Route path="/analytics" element={<AnalyticsWelcomePage />} />
|
||||||
<Route path="/analytics/view" element={<AnalyticsPage />} />
|
<Route path="/analytics/view" element={<AnalyticsPage />} />
|
||||||
<Route path="/group-analytics" element={<GroupAnalyticsPage />} />
|
<Route path="/group-analytics" element={<GroupAnalyticsPage />} />
|
||||||
|
|||||||
@@ -139,6 +139,18 @@
|
|||||||
font-size: 14px;
|
font-size: 14px;
|
||||||
font-weight: 600;
|
font-weight: 600;
|
||||||
color: var(--text-primary);
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
&.clickable {
|
||||||
|
cursor: pointer;
|
||||||
|
border-radius: 6px;
|
||||||
|
padding: 2px 8px;
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -212,4 +224,68 @@
|
|||||||
padding-top: 12px;
|
padding-top: 12px;
|
||||||
border-top: 1px solid var(--border-color);
|
border-top: 1px solid var(--border-color);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.year-month-picker {
|
||||||
|
padding: 4px 0;
|
||||||
|
|
||||||
|
.year-selector {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 12px;
|
||||||
|
|
||||||
|
.year-label {
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.nav-btn {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
width: 28px;
|
||||||
|
height: 28px;
|
||||||
|
padding: 0;
|
||||||
|
border: none;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 6px;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.month-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(3, 1fr);
|
||||||
|
gap: 6px;
|
||||||
|
|
||||||
|
.month-btn {
|
||||||
|
padding: 8px 0;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -26,6 +26,7 @@ function DateRangePicker({ startDate, endDate, onStartDateChange, onEndDateChang
|
|||||||
const [isOpen, setIsOpen] = useState(false)
|
const [isOpen, setIsOpen] = useState(false)
|
||||||
const [currentMonth, setCurrentMonth] = useState(new Date())
|
const [currentMonth, setCurrentMonth] = useState(new Date())
|
||||||
const [selectingStart, setSelectingStart] = useState(true)
|
const [selectingStart, setSelectingStart] = useState(true)
|
||||||
|
const [showYearMonthPicker, setShowYearMonthPicker] = useState(false)
|
||||||
const containerRef = useRef<HTMLDivElement>(null)
|
const containerRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
// 点击外部关闭
|
// 点击外部关闭
|
||||||
@@ -185,12 +186,38 @@ function DateRangePicker({ startDate, endDate, onStartDateChange, onEndDateChang
|
|||||||
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear(), currentMonth.getMonth() - 1))}>
|
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear(), currentMonth.getMonth() - 1))}>
|
||||||
<ChevronLeft size={16} />
|
<ChevronLeft size={16} />
|
||||||
</button>
|
</button>
|
||||||
<span className="month-year">{currentMonth.getFullYear()}年 {MONTH_NAMES[currentMonth.getMonth()]}</span>
|
<span className="month-year clickable" onClick={() => setShowYearMonthPicker(!showYearMonthPicker)}>
|
||||||
|
{currentMonth.getFullYear()}年 {MONTH_NAMES[currentMonth.getMonth()]}
|
||||||
|
</span>
|
||||||
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear(), currentMonth.getMonth() + 1))}>
|
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear(), currentMonth.getMonth() + 1))}>
|
||||||
<ChevronRight size={16} />
|
<ChevronRight size={16} />
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
{renderCalendar()}
|
{showYearMonthPicker ? (
|
||||||
|
<div className="year-month-picker">
|
||||||
|
<div className="year-selector">
|
||||||
|
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear() - 1, currentMonth.getMonth()))}>
|
||||||
|
<ChevronLeft size={14} />
|
||||||
|
</button>
|
||||||
|
<span className="year-label">{currentMonth.getFullYear()}年</span>
|
||||||
|
<button className="nav-btn" onClick={() => setCurrentMonth(new Date(currentMonth.getFullYear() + 1, currentMonth.getMonth()))}>
|
||||||
|
<ChevronRight size={14} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="month-grid">
|
||||||
|
{MONTH_NAMES.map((name, i) => (
|
||||||
|
<button
|
||||||
|
key={i}
|
||||||
|
className={`month-btn ${i === currentMonth.getMonth() ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setCurrentMonth(new Date(currentMonth.getFullYear(), i))
|
||||||
|
setShowYearMonthPicker(false)
|
||||||
|
}}
|
||||||
|
>{name}</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : renderCalendar()}
|
||||||
<div className="selection-hint">
|
<div className="selection-hint">
|
||||||
{selectingStart ? '请选择开始日期' : '请选择结束日期'}
|
{selectingStart ? '请选择开始日期' : '请选择结束日期'}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -14,7 +14,6 @@ export function GlobalSessionMonitor() {
|
|||||||
} = useChatStore()
|
} = useChatStore()
|
||||||
|
|
||||||
const sessionsRef = useRef(sessions)
|
const sessionsRef = useRef(sessions)
|
||||||
|
|
||||||
// 保持 ref 同步
|
// 保持 ref 同步
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
sessionsRef.current = sessions
|
sessionsRef.current = sessions
|
||||||
@@ -47,9 +46,10 @@ export function GlobalSessionMonitor() {
|
|||||||
return () => {
|
return () => {
|
||||||
removeListener()
|
removeListener()
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
}
|
}
|
||||||
return () => { }
|
return () => { }
|
||||||
}, []) // 空依赖数组 - 主要是静态的
|
}, [])
|
||||||
|
|
||||||
const refreshSessions = async () => {
|
const refreshSessions = async () => {
|
||||||
try {
|
try {
|
||||||
|
|||||||
@@ -75,6 +75,18 @@
|
|||||||
font-size: 15px;
|
font-size: 15px;
|
||||||
font-weight: 600;
|
font-weight: 600;
|
||||||
color: var(--text-primary);
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
&.clickable {
|
||||||
|
cursor: pointer;
|
||||||
|
border-radius: 6px;
|
||||||
|
padding: 2px 8px;
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.nav-btn {
|
.nav-btn {
|
||||||
@@ -97,6 +109,70 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.year-month-picker {
|
||||||
|
padding: 4px 0;
|
||||||
|
|
||||||
|
.year-selector {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 12px;
|
||||||
|
|
||||||
|
.year-label {
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.nav-btn {
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 8px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
border-color: var(--primary);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.month-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(3, 1fr);
|
||||||
|
gap: 6px;
|
||||||
|
|
||||||
|
.month-btn {
|
||||||
|
padding: 10px 0;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.calendar-grid {
|
.calendar-grid {
|
||||||
|
|||||||
@@ -24,6 +24,7 @@ const JumpToDateDialog: React.FC<JumpToDateDialogProps> = ({
|
|||||||
const isValidDate = (d: any) => d instanceof Date && !isNaN(d.getTime())
|
const isValidDate = (d: any) => d instanceof Date && !isNaN(d.getTime())
|
||||||
const [calendarDate, setCalendarDate] = useState(isValidDate(currentDate) ? new Date(currentDate) : new Date())
|
const [calendarDate, setCalendarDate] = useState(isValidDate(currentDate) ? new Date(currentDate) : new Date())
|
||||||
const [selectedDate, setSelectedDate] = useState(new Date(currentDate))
|
const [selectedDate, setSelectedDate] = useState(new Date(currentDate))
|
||||||
|
const [showYearMonthPicker, setShowYearMonthPicker] = useState(false)
|
||||||
|
|
||||||
if (!isOpen) return null
|
if (!isOpen) return null
|
||||||
|
|
||||||
@@ -137,7 +138,7 @@ const JumpToDateDialog: React.FC<JumpToDateDialogProps> = ({
|
|||||||
>
|
>
|
||||||
<ChevronLeft size={18} />
|
<ChevronLeft size={18} />
|
||||||
</button>
|
</button>
|
||||||
<span className="current-month">
|
<span className="current-month clickable" onClick={() => setShowYearMonthPicker(!showYearMonthPicker)}>
|
||||||
{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月
|
{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月
|
||||||
</span>
|
</span>
|
||||||
<button
|
<button
|
||||||
@@ -148,6 +149,31 @@ const JumpToDateDialog: React.FC<JumpToDateDialogProps> = ({
|
|||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{showYearMonthPicker ? (
|
||||||
|
<div className="year-month-picker">
|
||||||
|
<div className="year-selector">
|
||||||
|
<button className="nav-btn" onClick={() => setCalendarDate(new Date(calendarDate.getFullYear() - 1, calendarDate.getMonth(), 1))}>
|
||||||
|
<ChevronLeft size={16} />
|
||||||
|
</button>
|
||||||
|
<span className="year-label">{calendarDate.getFullYear()}年</span>
|
||||||
|
<button className="nav-btn" onClick={() => setCalendarDate(new Date(calendarDate.getFullYear() + 1, calendarDate.getMonth(), 1))}>
|
||||||
|
<ChevronRight size={16} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="month-grid">
|
||||||
|
{['一月','二月','三月','四月','五月','六月','七月','八月','九月','十月','十一月','十二月'].map((name, i) => (
|
||||||
|
<button
|
||||||
|
key={i}
|
||||||
|
className={`month-btn ${i === calendarDate.getMonth() ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setCalendarDate(new Date(calendarDate.getFullYear(), i, 1))
|
||||||
|
setShowYearMonthPicker(false)
|
||||||
|
}}
|
||||||
|
>{name}</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
<div className={`calendar-grid ${loadingDates ? 'loading' : ''}`}>
|
<div className={`calendar-grid ${loadingDates ? 'loading' : ''}`}>
|
||||||
{loadingDates && (
|
{loadingDates && (
|
||||||
<div className="calendar-loading">
|
<div className="calendar-loading">
|
||||||
@@ -174,6 +200,7 @@ const JumpToDateDialog: React.FC<JumpToDateDialogProps> = ({
|
|||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="quick-options">
|
<div className="quick-options">
|
||||||
|
|||||||
@@ -6,6 +6,13 @@
|
|||||||
backdrop-filter: blur(20px);
|
backdrop-filter: blur(20px);
|
||||||
-webkit-backdrop-filter: blur(20px);
|
-webkit-backdrop-filter: blur(20px);
|
||||||
border: 1px solid var(--border-light);
|
border: 1px solid var(--border-light);
|
||||||
|
|
||||||
|
// 浅色模式下使用不透明背景,避免透明窗口中通知过于透明
|
||||||
|
[data-mode="light"] &,
|
||||||
|
:not([data-mode]) & {
|
||||||
|
background: rgba(255, 255, 255, 1);
|
||||||
|
}
|
||||||
|
|
||||||
border-radius: 12px;
|
border-radius: 12px;
|
||||||
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.12);
|
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.12);
|
||||||
padding: 12px;
|
padding: 12px;
|
||||||
@@ -39,7 +46,7 @@
|
|||||||
backdrop-filter: none !important;
|
backdrop-filter: none !important;
|
||||||
-webkit-backdrop-filter: none !important;
|
-webkit-backdrop-filter: none !important;
|
||||||
|
|
||||||
// Ensure background is solid
|
// 确保背景不透明
|
||||||
background: var(--bg-secondary, #2c2c2c);
|
background: var(--bg-secondary, #2c2c2c);
|
||||||
color: var(--text-primary, #ffffff);
|
color: var(--text-primary, #ffffff);
|
||||||
|
|
||||||
|
|||||||
185
src/components/Sns/SnsFilterPanel.tsx
Normal file
185
src/components/Sns/SnsFilterPanel.tsx
Normal file
@@ -0,0 +1,185 @@
|
|||||||
|
import React, { useState } from 'react'
|
||||||
|
import { Search, Calendar, User, X, Filter, Check } from 'lucide-react'
|
||||||
|
import { Avatar } from '../Avatar'
|
||||||
|
// import JumpToDateDialog from '../JumpToDateDialog' // Assuming this is imported from parent or moved
|
||||||
|
|
||||||
|
interface Contact {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
avatarUrl?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface SnsFilterPanelProps {
|
||||||
|
searchKeyword: string
|
||||||
|
setSearchKeyword: (val: string) => void
|
||||||
|
jumpTargetDate?: Date
|
||||||
|
setJumpTargetDate: (date?: Date) => void
|
||||||
|
onOpenJumpDialog: () => void
|
||||||
|
selectedUsernames: string[]
|
||||||
|
setSelectedUsernames: (val: string[]) => void
|
||||||
|
contacts: Contact[]
|
||||||
|
contactSearch: string
|
||||||
|
setContactSearch: (val: string) => void
|
||||||
|
loading?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
export const SnsFilterPanel: React.FC<SnsFilterPanelProps> = ({
|
||||||
|
searchKeyword,
|
||||||
|
setSearchKeyword,
|
||||||
|
jumpTargetDate,
|
||||||
|
setJumpTargetDate,
|
||||||
|
onOpenJumpDialog,
|
||||||
|
selectedUsernames,
|
||||||
|
setSelectedUsernames,
|
||||||
|
contacts,
|
||||||
|
contactSearch,
|
||||||
|
setContactSearch,
|
||||||
|
loading
|
||||||
|
}) => {
|
||||||
|
|
||||||
|
const filteredContacts = contacts.filter(c =>
|
||||||
|
c.displayName.toLowerCase().includes(contactSearch.toLowerCase()) ||
|
||||||
|
c.username.toLowerCase().includes(contactSearch.toLowerCase())
|
||||||
|
)
|
||||||
|
|
||||||
|
const toggleUserSelection = (username: string) => {
|
||||||
|
if (selectedUsernames.includes(username)) {
|
||||||
|
setSelectedUsernames(selectedUsernames.filter(u => u !== username))
|
||||||
|
} else {
|
||||||
|
setJumpTargetDate(undefined) // Reset date jump when selecting user
|
||||||
|
setSelectedUsernames([...selectedUsernames, username])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const clearFilters = () => {
|
||||||
|
setSearchKeyword('')
|
||||||
|
setSelectedUsernames([])
|
||||||
|
setJumpTargetDate(undefined)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<aside className="sns-filter-panel">
|
||||||
|
<div className="filter-header">
|
||||||
|
<h3>筛选条件</h3>
|
||||||
|
{(searchKeyword || jumpTargetDate || selectedUsernames.length > 0) && (
|
||||||
|
<button className="reset-all-btn" onClick={clearFilters} title="重置所有筛选">
|
||||||
|
<RefreshCw size={14} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="filter-widgets">
|
||||||
|
{/* Search Widget */}
|
||||||
|
<div className="filter-widget search-widget">
|
||||||
|
<div className="widget-header">
|
||||||
|
<Search size={14} />
|
||||||
|
<span>关键词搜索</span>
|
||||||
|
</div>
|
||||||
|
<div className="input-group">
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="搜索动态内容..."
|
||||||
|
value={searchKeyword}
|
||||||
|
onChange={e => setSearchKeyword(e.target.value)}
|
||||||
|
/>
|
||||||
|
{searchKeyword && (
|
||||||
|
<button className="clear-input-btn" onClick={() => setSearchKeyword('')}>
|
||||||
|
<X size={14} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Date Widget */}
|
||||||
|
<div className="filter-widget date-widget">
|
||||||
|
<div className="widget-header">
|
||||||
|
<Calendar size={14} />
|
||||||
|
<span>时间跳转</span>
|
||||||
|
</div>
|
||||||
|
<button
|
||||||
|
className={`date-picker-trigger ${jumpTargetDate ? 'active' : ''}`}
|
||||||
|
onClick={onOpenJumpDialog}
|
||||||
|
>
|
||||||
|
<span className="date-text">
|
||||||
|
{jumpTargetDate
|
||||||
|
? jumpTargetDate.toLocaleDateString('zh-CN', { year: 'numeric', month: 'long', day: 'numeric' })
|
||||||
|
: '选择日期...'}
|
||||||
|
</span>
|
||||||
|
{jumpTargetDate && (
|
||||||
|
<div
|
||||||
|
className="clear-date-btn"
|
||||||
|
onClick={(e) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
setJumpTargetDate(undefined)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<X size={12} />
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Contact Widget */}
|
||||||
|
<div className="filter-widget contact-widget">
|
||||||
|
<div className="widget-header">
|
||||||
|
<User size={14} />
|
||||||
|
<span>联系人</span>
|
||||||
|
{selectedUsernames.length > 0 && (
|
||||||
|
<span className="badge">{selectedUsernames.length}</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="contact-search-bar">
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="查找好友..."
|
||||||
|
value={contactSearch}
|
||||||
|
onChange={e => setContactSearch(e.target.value)}
|
||||||
|
/>
|
||||||
|
<Search size={14} className="search-icon" />
|
||||||
|
{contactSearch && (
|
||||||
|
<X size={14} className="clear-icon" onClick={() => setContactSearch('')} />
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="contact-list-scroll">
|
||||||
|
{filteredContacts.map(contact => (
|
||||||
|
<div
|
||||||
|
key={contact.username}
|
||||||
|
className={`contact-row ${selectedUsernames.includes(contact.username) ? 'selected' : ''}`}
|
||||||
|
onClick={() => toggleUserSelection(contact.username)}
|
||||||
|
>
|
||||||
|
<Avatar src={contact.avatarUrl} name={contact.displayName} size={36} shape="rounded" />
|
||||||
|
<span className="contact-name">{contact.displayName}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
{filteredContacts.length === 0 && (
|
||||||
|
<div className="empty-state">没有找到联系人</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</aside>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
function RefreshCw({ size, className }: { size?: number, className?: string }) {
|
||||||
|
return (
|
||||||
|
<svg
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
width={size || 24}
|
||||||
|
height={size || 24}
|
||||||
|
viewBox="0 0 24 24"
|
||||||
|
fill="none"
|
||||||
|
stroke="currentColor"
|
||||||
|
strokeWidth="2"
|
||||||
|
strokeLinecap="round"
|
||||||
|
strokeLinejoin="round"
|
||||||
|
className={className}
|
||||||
|
>
|
||||||
|
<path d="M23 4v6h-6"></path>
|
||||||
|
<path d="M1 20v-6h6"></path>
|
||||||
|
<path d="M3.51 9a9 9 0 0 1 14.85-3.36L23 10M1 14l4.64 4.36A9 9 0 0 0 20.49 15"></path>
|
||||||
|
</svg>
|
||||||
|
)
|
||||||
|
}
|
||||||
357
src/components/Sns/SnsMediaGrid.tsx
Normal file
357
src/components/Sns/SnsMediaGrid.tsx
Normal file
@@ -0,0 +1,357 @@
|
|||||||
|
import React, { useState, useRef } from 'react'
|
||||||
|
import { Play, Lock, Download, ImageOff } from 'lucide-react'
|
||||||
|
import { LivePhotoIcon } from '../../components/LivePhotoIcon'
|
||||||
|
import { RefreshCw } from 'lucide-react'
|
||||||
|
|
||||||
|
interface SnsMedia {
|
||||||
|
url: string
|
||||||
|
thumb: string
|
||||||
|
md5?: string
|
||||||
|
token?: string
|
||||||
|
key?: string
|
||||||
|
encIdx?: string
|
||||||
|
livePhoto?: {
|
||||||
|
url: string
|
||||||
|
thumb: string
|
||||||
|
token?: string
|
||||||
|
key?: string
|
||||||
|
encIdx?: string
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
interface SnsMediaGridProps {
|
||||||
|
mediaList: SnsMedia[]
|
||||||
|
onPreview: (src: string, isVideo?: boolean, liveVideoPath?: string) => void
|
||||||
|
onMediaDeleted?: () => void
|
||||||
|
}
|
||||||
|
|
||||||
|
const isSnsVideoUrl = (url?: string): boolean => {
|
||||||
|
if (!url) return false
|
||||||
|
const lower = url.toLowerCase()
|
||||||
|
return (lower.includes('snsvideodownload') || lower.includes('.mp4') || lower.includes('video')) && !lower.includes('vweixinthumb')
|
||||||
|
}
|
||||||
|
|
||||||
|
const extractVideoFrame = async (videoPath: string): Promise<string> => {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const video = document.createElement('video')
|
||||||
|
video.preload = 'auto'
|
||||||
|
video.src = videoPath
|
||||||
|
video.muted = true
|
||||||
|
video.currentTime = 0 // Initial reset
|
||||||
|
// video.crossOrigin = 'anonymous' // Not needed for file:// usually
|
||||||
|
|
||||||
|
const onSeeked = () => {
|
||||||
|
try {
|
||||||
|
const canvas = document.createElement('canvas')
|
||||||
|
canvas.width = video.videoWidth
|
||||||
|
canvas.height = video.videoHeight
|
||||||
|
const ctx = canvas.getContext('2d')
|
||||||
|
if (ctx) {
|
||||||
|
ctx.drawImage(video, 0, 0, canvas.width, canvas.height)
|
||||||
|
const dataUrl = canvas.toDataURL('image/jpeg', 0.8)
|
||||||
|
resolve(dataUrl)
|
||||||
|
} else {
|
||||||
|
reject(new Error('Canvas context failed'))
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
reject(e)
|
||||||
|
} finally {
|
||||||
|
// Cleanup
|
||||||
|
video.removeEventListener('seeked', onSeeked)
|
||||||
|
video.src = ''
|
||||||
|
video.load()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
video.onloadedmetadata = () => {
|
||||||
|
if (video.duration === Infinity || isNaN(video.duration)) {
|
||||||
|
// Determine duration failed, try a fixed small offset
|
||||||
|
video.currentTime = 1
|
||||||
|
} else {
|
||||||
|
video.currentTime = Math.max(0.1, video.duration / 2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
video.onseeked = onSeeked
|
||||||
|
|
||||||
|
video.onerror = (e) => {
|
||||||
|
reject(new Error('Video load failed'))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const MediaItem = ({ media, onPreview, onMediaDeleted }: { media: SnsMedia; onPreview: (src: string, isVideo?: boolean, liveVideoPath?: string) => void; onMediaDeleted?: () => void }) => {
|
||||||
|
const [error, setError] = useState(false)
|
||||||
|
const [deleted, setDeleted] = useState(false)
|
||||||
|
const [loading, setLoading] = useState(true)
|
||||||
|
const markDeleted = () => { setDeleted(true); onMediaDeleted?.() }
|
||||||
|
const retryCount = useRef(0)
|
||||||
|
const [retryKey, setRetryKey] = useState(0)
|
||||||
|
const [thumbSrc, setThumbSrc] = useState<string>('')
|
||||||
|
const [videoPath, setVideoPath] = useState<string>('')
|
||||||
|
const [liveVideoPath, setLiveVideoPath] = useState<string>('')
|
||||||
|
const [isDecrypting, setIsDecrypting] = useState(false)
|
||||||
|
const [isGeneratingCover, setIsGeneratingCover] = useState(false)
|
||||||
|
|
||||||
|
const isVideo = isSnsVideoUrl(media.url)
|
||||||
|
const isLive = !!media.livePhoto
|
||||||
|
const targetUrl = media.thumb || media.url
|
||||||
|
|
||||||
|
// 视频重试:失败时重试最多2次,耗尽才标记删除
|
||||||
|
const videoRetryOrDelete = () => {
|
||||||
|
if (retryCount.current < 2) {
|
||||||
|
retryCount.current++
|
||||||
|
setRetryKey(k => k + 1)
|
||||||
|
} else {
|
||||||
|
markDeleted()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Simple effect to load image/decrypt
|
||||||
|
// Simple effect to load image/decrypt
|
||||||
|
React.useEffect(() => {
|
||||||
|
let cancelled = false
|
||||||
|
setLoading(true)
|
||||||
|
|
||||||
|
const load = async () => {
|
||||||
|
try {
|
||||||
|
if (!isVideo) {
|
||||||
|
// For images, we proxy to get the local path/base64
|
||||||
|
const result = await window.electronAPI.sns.proxyImage({
|
||||||
|
url: targetUrl,
|
||||||
|
key: media.key
|
||||||
|
})
|
||||||
|
if (cancelled) return
|
||||||
|
|
||||||
|
if (result.success) {
|
||||||
|
if (result.dataUrl) setThumbSrc(result.dataUrl)
|
||||||
|
else if (result.videoPath) setThumbSrc(`file://${result.videoPath.replace(/\\/g, '/')}`)
|
||||||
|
} else {
|
||||||
|
markDeleted()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pre-load live photo video if needed
|
||||||
|
if (isLive && media.livePhoto?.url) {
|
||||||
|
window.electronAPI.sns.proxyImage({
|
||||||
|
url: media.livePhoto.url,
|
||||||
|
key: media.livePhoto.key || media.key
|
||||||
|
}).then((res: any) => {
|
||||||
|
if (!cancelled && res.success && res.videoPath) {
|
||||||
|
setLiveVideoPath(`file://${res.videoPath.replace(/\\/g, '/')}`)
|
||||||
|
}
|
||||||
|
}).catch(() => { })
|
||||||
|
}
|
||||||
|
setLoading(false)
|
||||||
|
} else {
|
||||||
|
// Video logic: Decrypt -> Extract Frame
|
||||||
|
setIsGeneratingCover(true)
|
||||||
|
|
||||||
|
// First check if we already have it decryptable?
|
||||||
|
// Usually we need to call proxyImage with the video URL to decrypt it to cache
|
||||||
|
const result = await window.electronAPI.sns.proxyImage({
|
||||||
|
url: media.url,
|
||||||
|
key: media.key
|
||||||
|
})
|
||||||
|
|
||||||
|
if (cancelled) return
|
||||||
|
|
||||||
|
if (result.success && result.videoPath) {
|
||||||
|
const localPath = `file://${result.videoPath.replace(/\\/g, '/')}`
|
||||||
|
setVideoPath(localPath)
|
||||||
|
|
||||||
|
try {
|
||||||
|
const coverDataUrl = await extractVideoFrame(localPath)
|
||||||
|
if (!cancelled) setThumbSrc(coverDataUrl)
|
||||||
|
} catch (err) {
|
||||||
|
console.error('Frame extraction failed', err)
|
||||||
|
// 封面提取失败,用视频路径作为 fallback,让 <video> 标签显示
|
||||||
|
if (!cancelled) setThumbSrc(localPath)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
videoRetryOrDelete()
|
||||||
|
}
|
||||||
|
|
||||||
|
setIsGeneratingCover(false)
|
||||||
|
setLoading(false)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(e)
|
||||||
|
if (!cancelled) {
|
||||||
|
if (isVideo) {
|
||||||
|
videoRetryOrDelete()
|
||||||
|
} else {
|
||||||
|
markDeleted()
|
||||||
|
}
|
||||||
|
setLoading(false)
|
||||||
|
setIsGeneratingCover(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
load()
|
||||||
|
return () => { cancelled = true }
|
||||||
|
}, [media, isVideo, isLive, targetUrl, retryKey])
|
||||||
|
|
||||||
|
const handlePreview = async (e: React.MouseEvent) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
if (isVideo) {
|
||||||
|
// Decrypt video on demand if not already
|
||||||
|
if (!videoPath) {
|
||||||
|
setIsDecrypting(true)
|
||||||
|
try {
|
||||||
|
const res = await window.electronAPI.sns.proxyImage({
|
||||||
|
url: media.url,
|
||||||
|
key: media.key
|
||||||
|
})
|
||||||
|
if (res.success && res.videoPath) {
|
||||||
|
const local = `file://${res.videoPath.replace(/\\/g, '/')}`
|
||||||
|
setVideoPath(local)
|
||||||
|
onPreview(local, true, undefined)
|
||||||
|
} else {
|
||||||
|
alert('视频解密失败')
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(e)
|
||||||
|
} finally {
|
||||||
|
setIsDecrypting(false)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
onPreview(videoPath, true, undefined)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
onPreview(thumbSrc || targetUrl, false, liveVideoPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleDownload = async (e: React.MouseEvent) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
setLoading(true)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.sns.proxyImage({
|
||||||
|
url: media.url,
|
||||||
|
key: media.key
|
||||||
|
})
|
||||||
|
|
||||||
|
if (result.success) {
|
||||||
|
const link = document.createElement('a')
|
||||||
|
link.download = `sns_media_${Date.now()}.${isVideo ? 'mp4' : 'jpg'}`
|
||||||
|
|
||||||
|
if (result.dataUrl) {
|
||||||
|
link.href = result.dataUrl
|
||||||
|
} else if (result.videoPath) {
|
||||||
|
// For local video files, we need to fetch as blob to force download behavior
|
||||||
|
// or just use the file protocol url if the browser supports it
|
||||||
|
try {
|
||||||
|
const response = await fetch(`file://${result.videoPath}`)
|
||||||
|
const blob = await response.blob()
|
||||||
|
const url = URL.createObjectURL(blob)
|
||||||
|
link.href = url
|
||||||
|
setTimeout(() => URL.revokeObjectURL(url), 60000)
|
||||||
|
} catch (err) {
|
||||||
|
console.error('Video fetch failed, falling back to direct link', err)
|
||||||
|
link.href = `file://${result.videoPath}`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
document.body.appendChild(link)
|
||||||
|
link.click()
|
||||||
|
document.body.removeChild(link)
|
||||||
|
} else {
|
||||||
|
alert('下载失败: 无法获取资源')
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('Download error:', e)
|
||||||
|
alert('下载出错')
|
||||||
|
} finally {
|
||||||
|
setLoading(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (deleted) {
|
||||||
|
return (
|
||||||
|
<div className="sns-media-item deleted-media">
|
||||||
|
<div className="deleted-placeholder">
|
||||||
|
<ImageOff size={24} />
|
||||||
|
<span>已删除</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className={`sns-media-item ${isDecrypting ? 'decrypting' : ''}`}
|
||||||
|
onClick={handlePreview}
|
||||||
|
>
|
||||||
|
{(thumbSrc && !thumbSrc.startsWith('data:') && (thumbSrc.toLowerCase().endsWith('.mp4') || thumbSrc.includes('video'))) ? (
|
||||||
|
<video
|
||||||
|
key={thumbSrc}
|
||||||
|
src={`${thumbSrc}#t=0.1`}
|
||||||
|
className="media-image"
|
||||||
|
preload="auto"
|
||||||
|
muted
|
||||||
|
playsInline
|
||||||
|
disablePictureInPicture
|
||||||
|
disableRemotePlayback
|
||||||
|
onLoadedMetadata={(e) => {
|
||||||
|
e.currentTarget.currentTime = 0.1
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
) : thumbSrc ? (
|
||||||
|
<img
|
||||||
|
src={thumbSrc}
|
||||||
|
className="media-image"
|
||||||
|
loading="lazy"
|
||||||
|
onError={() => { if (!loading && !isVideo) markDeleted() }}
|
||||||
|
alt=""
|
||||||
|
/>
|
||||||
|
) : null}
|
||||||
|
|
||||||
|
{isGeneratingCover && (
|
||||||
|
<div className="media-decrypting-mask">
|
||||||
|
<RefreshCw className="spin" size={24} />
|
||||||
|
<span>解密中...</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isVideo && (
|
||||||
|
<div className="media-badge video">
|
||||||
|
{/* If we have a cover, show Play. If decrypting for preview, show spin. Generating cover has its own mask. */}
|
||||||
|
{isDecrypting ? <RefreshCw className="spin" size={16} /> : <Play size={16} fill="currentColor" />}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isLive && !isVideo && (
|
||||||
|
<div className="media-badge live">
|
||||||
|
<LivePhotoIcon size={16} />
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className="media-download-btn" onClick={handleDownload} title="下载">
|
||||||
|
<Download size={16} />
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export const SnsMediaGrid: React.FC<SnsMediaGridProps> = ({ mediaList, onPreview, onMediaDeleted }) => {
|
||||||
|
if (!mediaList || mediaList.length === 0) return null
|
||||||
|
|
||||||
|
const count = mediaList.length
|
||||||
|
let gridClass = ''
|
||||||
|
|
||||||
|
if (count === 1) gridClass = 'grid-1'
|
||||||
|
else if (count === 2) gridClass = 'grid-2'
|
||||||
|
else if (count === 3) gridClass = 'grid-3'
|
||||||
|
else if (count === 4) gridClass = 'grid-4' // 2x2
|
||||||
|
else if (count <= 6) gridClass = 'grid-6' // 3 cols
|
||||||
|
else gridClass = 'grid-9' // 3x3
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className={`sns-media-grid ${gridClass}`}>
|
||||||
|
{mediaList.map((media, idx) => (
|
||||||
|
<MediaItem key={idx} media={media} onPreview={onPreview} onMediaDeleted={onMediaDeleted} />
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
302
src/components/Sns/SnsPostItem.tsx
Normal file
302
src/components/Sns/SnsPostItem.tsx
Normal file
@@ -0,0 +1,302 @@
|
|||||||
|
import React, { useState, useMemo } from 'react'
|
||||||
|
import { Heart, ChevronRight, ImageIcon, Download, Code, MoreHorizontal, Trash2 } from 'lucide-react'
|
||||||
|
import { SnsPost, SnsLinkCardData } from '../../types/sns'
|
||||||
|
import { Avatar } from '../Avatar'
|
||||||
|
import { SnsMediaGrid } from './SnsMediaGrid'
|
||||||
|
import { getEmojiPath } from 'wechat-emojis'
|
||||||
|
|
||||||
|
// Helper functions (extracted from SnsPage.tsx but simplified/reused)
|
||||||
|
const LINK_XML_URL_TAGS = ['url', 'shorturl', 'weburl', 'webpageurl', 'jumpurl']
|
||||||
|
const LINK_XML_TITLE_TAGS = ['title', 'linktitle', 'webtitle']
|
||||||
|
const MEDIA_HOST_HINTS = ['mmsns.qpic.cn', 'vweixinthumb', 'snstimeline', 'snsvideodownload']
|
||||||
|
|
||||||
|
const isSnsVideoUrl = (url?: string): boolean => {
|
||||||
|
if (!url) return false
|
||||||
|
const lower = url.toLowerCase()
|
||||||
|
return (lower.includes('snsvideodownload') || lower.includes('.mp4') || lower.includes('video')) && !lower.includes('vweixinthumb')
|
||||||
|
}
|
||||||
|
|
||||||
|
const decodeHtmlEntities = (text: string): string => {
|
||||||
|
if (!text) return ''
|
||||||
|
return text
|
||||||
|
.replace(/<!\[CDATA\[([\s\S]*?)\]\]>/g, '$1')
|
||||||
|
.replace(/&/gi, '&')
|
||||||
|
.replace(/</gi, '<')
|
||||||
|
.replace(/>/gi, '>')
|
||||||
|
.replace(/"/gi, '"')
|
||||||
|
.replace(/'/gi, "'")
|
||||||
|
.trim()
|
||||||
|
}
|
||||||
|
|
||||||
|
const normalizeUrlCandidate = (raw: string): string | null => {
|
||||||
|
const value = decodeHtmlEntities(raw).replace(/[)\],.;]+$/, '').trim()
|
||||||
|
if (!value) return null
|
||||||
|
if (!/^https?:\/\//i.test(value)) return null
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
const simplifyUrlForCompare = (value: string): string => {
|
||||||
|
const normalized = value.trim().toLowerCase().replace(/^https?:\/\//, '')
|
||||||
|
const [withoutQuery] = normalized.split('?')
|
||||||
|
return withoutQuery.replace(/\/+$/, '')
|
||||||
|
}
|
||||||
|
|
||||||
|
const getXmlTagValues = (xml: string, tags: string[]): string[] => {
|
||||||
|
if (!xml) return []
|
||||||
|
const results: string[] = []
|
||||||
|
for (const tag of tags) {
|
||||||
|
const reg = new RegExp(`<${tag}>([\\s\\S]*?)<\\/${tag}>`, 'ig')
|
||||||
|
let match: RegExpExecArray | null
|
||||||
|
while ((match = reg.exec(xml)) !== null) {
|
||||||
|
if (match[1]) results.push(match[1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
|
const getUrlLikeStrings = (text: string): string[] => {
|
||||||
|
if (!text) return []
|
||||||
|
return text.match(/https?:\/\/[^\s<>"']+/gi) || []
|
||||||
|
}
|
||||||
|
|
||||||
|
const isLikelyMediaAssetUrl = (url: string): boolean => {
|
||||||
|
const lower = url.toLowerCase()
|
||||||
|
return MEDIA_HOST_HINTS.some((hint) => lower.includes(hint))
|
||||||
|
}
|
||||||
|
|
||||||
|
const buildLinkCardData = (post: SnsPost): SnsLinkCardData | null => {
|
||||||
|
// type 3 是链接类型,直接用 media[0] 的 url 和 thumb
|
||||||
|
if (post.type === 3) {
|
||||||
|
const url = post.media[0]?.url || post.linkUrl
|
||||||
|
if (!url) return null
|
||||||
|
const titleCandidates = [
|
||||||
|
post.linkTitle || '',
|
||||||
|
...getXmlTagValues(post.rawXml || '', LINK_XML_TITLE_TAGS),
|
||||||
|
post.contentDesc || ''
|
||||||
|
]
|
||||||
|
const title = titleCandidates
|
||||||
|
.map((v) => decodeHtmlEntities(v))
|
||||||
|
.find((v) => Boolean(v) && !/^https?:\/\//i.test(v))
|
||||||
|
return { url, title: title || '网页链接', thumb: post.media[0]?.thumb }
|
||||||
|
}
|
||||||
|
|
||||||
|
const hasVideoMedia = post.type === 15 || post.media.some((item) => isSnsVideoUrl(item.url))
|
||||||
|
if (hasVideoMedia) return null
|
||||||
|
|
||||||
|
const mediaValues = post.media
|
||||||
|
.flatMap((item) => [item.url, item.thumb])
|
||||||
|
.filter((value): value is string => Boolean(value))
|
||||||
|
const mediaSet = new Set(mediaValues.map((value) => simplifyUrlForCompare(value)))
|
||||||
|
|
||||||
|
const urlCandidates: string[] = [
|
||||||
|
post.linkUrl || '',
|
||||||
|
...getXmlTagValues(post.rawXml || '', LINK_XML_URL_TAGS),
|
||||||
|
...getUrlLikeStrings(post.rawXml || ''),
|
||||||
|
...getUrlLikeStrings(post.contentDesc || '')
|
||||||
|
]
|
||||||
|
|
||||||
|
const normalizedCandidates = urlCandidates
|
||||||
|
.map(normalizeUrlCandidate)
|
||||||
|
.filter((value): value is string => Boolean(value))
|
||||||
|
|
||||||
|
const dedupedCandidates: string[] = []
|
||||||
|
const seen = new Set<string>()
|
||||||
|
for (const candidate of normalizedCandidates) {
|
||||||
|
if (seen.has(candidate)) continue
|
||||||
|
seen.add(candidate)
|
||||||
|
dedupedCandidates.push(candidate)
|
||||||
|
}
|
||||||
|
|
||||||
|
const linkUrl = dedupedCandidates.find((candidate) => {
|
||||||
|
const simplified = simplifyUrlForCompare(candidate)
|
||||||
|
if (mediaSet.has(simplified)) return false
|
||||||
|
if (isLikelyMediaAssetUrl(candidate)) return false
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
if (!linkUrl) return null
|
||||||
|
|
||||||
|
const titleCandidates = [
|
||||||
|
post.linkTitle || '',
|
||||||
|
...getXmlTagValues(post.rawXml || '', LINK_XML_TITLE_TAGS),
|
||||||
|
post.contentDesc || ''
|
||||||
|
]
|
||||||
|
|
||||||
|
const title = titleCandidates
|
||||||
|
.map((value) => decodeHtmlEntities(value))
|
||||||
|
.find((value) => Boolean(value) && !/^https?:\/\//i.test(value))
|
||||||
|
|
||||||
|
return {
|
||||||
|
url: linkUrl,
|
||||||
|
title: title || '网页链接',
|
||||||
|
thumb: post.media[0]?.thumb || post.media[0]?.url
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const SnsLinkCard = ({ card }: { card: SnsLinkCardData }) => {
|
||||||
|
const [thumbFailed, setThumbFailed] = useState(false)
|
||||||
|
const hostname = useMemo(() => {
|
||||||
|
try {
|
||||||
|
return new URL(card.url).hostname.replace(/^www\./i, '')
|
||||||
|
} catch {
|
||||||
|
return card.url
|
||||||
|
}
|
||||||
|
}, [card.url])
|
||||||
|
|
||||||
|
const handleClick = async (e: React.MouseEvent<HTMLButtonElement>) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
try {
|
||||||
|
await window.electronAPI.shell.openExternal(card.url)
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[SnsLinkCard] openExternal failed:', error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<button type="button" className="post-link-card" onClick={handleClick}>
|
||||||
|
<div className="link-thumb">
|
||||||
|
{card.thumb && !thumbFailed ? (
|
||||||
|
<img
|
||||||
|
src={card.thumb}
|
||||||
|
alt=""
|
||||||
|
referrerPolicy="no-referrer"
|
||||||
|
loading="lazy"
|
||||||
|
onError={() => setThumbFailed(true)}
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<div className="link-thumb-fallback">
|
||||||
|
<ImageIcon size={18} />
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="link-meta">
|
||||||
|
<div className="link-title">{card.title}</div>
|
||||||
|
<div className="link-url">{hostname}</div>
|
||||||
|
</div>
|
||||||
|
<ChevronRight size={16} className="link-arrow" />
|
||||||
|
</button>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
interface SnsPostItemProps {
|
||||||
|
post: SnsPost
|
||||||
|
onPreview: (src: string, isVideo?: boolean, liveVideoPath?: string) => void
|
||||||
|
onDebug: (post: SnsPost) => void
|
||||||
|
}
|
||||||
|
|
||||||
|
export const SnsPostItem: React.FC<SnsPostItemProps> = ({ post, onPreview, onDebug }) => {
|
||||||
|
const [mediaDeleted, setMediaDeleted] = useState(false)
|
||||||
|
const linkCard = buildLinkCardData(post)
|
||||||
|
const hasVideoMedia = post.type === 15 || post.media.some((item) => isSnsVideoUrl(item.url))
|
||||||
|
const showLinkCard = Boolean(linkCard) && post.media.length <= 1 && !hasVideoMedia
|
||||||
|
const showMediaGrid = post.media.length > 0 && !showLinkCard
|
||||||
|
|
||||||
|
const formatTime = (ts: number) => {
|
||||||
|
const date = new Date(ts * 1000)
|
||||||
|
const isCurrentYear = date.getFullYear() === new Date().getFullYear()
|
||||||
|
|
||||||
|
return date.toLocaleString('zh-CN', {
|
||||||
|
year: isCurrentYear ? undefined : 'numeric',
|
||||||
|
month: 'short',
|
||||||
|
day: 'numeric',
|
||||||
|
hour: '2-digit',
|
||||||
|
minute: '2-digit'
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析微信表情
|
||||||
|
const renderTextWithEmoji = (text: string) => {
|
||||||
|
if (!text) return text
|
||||||
|
const parts = text.split(/\[(.*?)\]/g)
|
||||||
|
return parts.map((part, index) => {
|
||||||
|
if (index % 2 === 1) {
|
||||||
|
// @ts-ignore
|
||||||
|
const path = getEmojiPath(part as any)
|
||||||
|
if (path) {
|
||||||
|
return <img key={index} src={`${import.meta.env.BASE_URL}${path}`} alt={`[${part}]`} className="inline-emoji" style={{ width: 22, height: 22, verticalAlign: 'bottom', margin: '0 1px' }} />
|
||||||
|
}
|
||||||
|
return `[${part}]`
|
||||||
|
}
|
||||||
|
return part
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className={`sns-post-item ${mediaDeleted ? 'post-deleted' : ''}`}>
|
||||||
|
<div className="post-avatar-col">
|
||||||
|
<Avatar
|
||||||
|
src={post.avatarUrl}
|
||||||
|
name={post.nickname}
|
||||||
|
size={48}
|
||||||
|
shape="rounded"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="post-content-col">
|
||||||
|
<div className="post-header-row">
|
||||||
|
<div className="post-author-info">
|
||||||
|
<span className="author-name">{decodeHtmlEntities(post.nickname)}</span>
|
||||||
|
<span className="post-time">{formatTime(post.createTime)}</span>
|
||||||
|
</div>
|
||||||
|
<div className="post-header-actions">
|
||||||
|
{mediaDeleted && (
|
||||||
|
<span className="post-deleted-badge">
|
||||||
|
<Trash2 size={12} />
|
||||||
|
<span>已删除</span>
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
<button className="icon-btn-ghost debug-btn" onClick={(e) => {
|
||||||
|
e.stopPropagation();
|
||||||
|
onDebug(post);
|
||||||
|
}} title="查看原始数据">
|
||||||
|
<Code size={14} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{post.contentDesc && (
|
||||||
|
<div className="post-text">{renderTextWithEmoji(decodeHtmlEntities(post.contentDesc))}</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{showLinkCard && linkCard && (
|
||||||
|
<SnsLinkCard card={linkCard} />
|
||||||
|
)}
|
||||||
|
|
||||||
|
{showMediaGrid && (
|
||||||
|
<div className="post-media-container">
|
||||||
|
<SnsMediaGrid mediaList={post.media} onPreview={onPreview} onMediaDeleted={[1, 54].includes(post.type ?? 0) ? () => setMediaDeleted(true) : undefined} />
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{(post.likes.length > 0 || post.comments.length > 0) && (
|
||||||
|
<div className="post-interactions">
|
||||||
|
{post.likes.length > 0 && (
|
||||||
|
<div className="likes-block">
|
||||||
|
<Heart size={14} className="like-icon" />
|
||||||
|
<span className="likes-text">{post.likes.join('、')}</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{post.comments.length > 0 && (
|
||||||
|
<div className="comments-block">
|
||||||
|
{post.comments.map((c, idx) => (
|
||||||
|
<div key={idx} className="comment-row">
|
||||||
|
<span className="comment-user">{c.nickname}</span>
|
||||||
|
{c.refNickname && (
|
||||||
|
<>
|
||||||
|
<span className="reply-text">回复</span>
|
||||||
|
<span className="comment-user">{c.refNickname}</span>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
<span className="comment-colon">:</span>
|
||||||
|
<span className="comment-content">{renderTextWithEmoji(c.content)}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -1,552 +0,0 @@
|
|||||||
// AI 对话页面 - 简约大气风格
|
|
||||||
.ai-chat-page {
|
|
||||||
display: flex;
|
|
||||||
height: 100%;
|
|
||||||
width: 100%;
|
|
||||||
background: var(--bg-gradient);
|
|
||||||
color: var(--text-primary);
|
|
||||||
overflow: hidden;
|
|
||||||
|
|
||||||
.chat-container {
|
|
||||||
flex: 1;
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
max-width: 1200px;
|
|
||||||
margin: 0 auto;
|
|
||||||
width: 100%;
|
|
||||||
}
|
|
||||||
|
|
||||||
// ========== 顶部 Header - 已移除 ==========
|
|
||||||
// 模型选择器现已集成到输入框
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// ========== 聊天区域 ==========
|
|
||||||
.chat-main {
|
|
||||||
flex: 1;
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
background: var(--bg-secondary);
|
|
||||||
position: relative;
|
|
||||||
overflow: hidden;
|
|
||||||
|
|
||||||
// 空状态
|
|
||||||
.empty-state {
|
|
||||||
flex: 1;
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
padding: 40px;
|
|
||||||
|
|
||||||
.icon {
|
|
||||||
width: 80px;
|
|
||||||
height: 80px;
|
|
||||||
border-radius: 50%;
|
|
||||||
background: var(--primary-light);
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
margin-bottom: 24px;
|
|
||||||
|
|
||||||
svg {
|
|
||||||
width: 40px;
|
|
||||||
height: 40px;
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
h2 {
|
|
||||||
font-size: 20px;
|
|
||||||
font-weight: 600;
|
|
||||||
color: var(--text-primary);
|
|
||||||
margin: 0 0 8px;
|
|
||||||
}
|
|
||||||
|
|
||||||
p {
|
|
||||||
font-size: 14px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
margin: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 消息列表
|
|
||||||
.messages-list {
|
|
||||||
flex: 1;
|
|
||||||
overflow-y: auto;
|
|
||||||
padding: 24px 32px;
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
gap: 20px;
|
|
||||||
|
|
||||||
&::-webkit-scrollbar {
|
|
||||||
width: 6px;
|
|
||||||
}
|
|
||||||
|
|
||||||
&::-webkit-scrollbar-track {
|
|
||||||
background: transparent;
|
|
||||||
}
|
|
||||||
|
|
||||||
&::-webkit-scrollbar-thumb {
|
|
||||||
background: var(--border-color);
|
|
||||||
border-radius: 3px;
|
|
||||||
}
|
|
||||||
|
|
||||||
.message-row {
|
|
||||||
display: flex;
|
|
||||||
gap: 12px;
|
|
||||||
max-width: 80%;
|
|
||||||
animation: messageIn 0.3s ease-out;
|
|
||||||
|
|
||||||
// 用户消息
|
|
||||||
&.user {
|
|
||||||
align-self: flex-end;
|
|
||||||
flex-direction: row-reverse;
|
|
||||||
|
|
||||||
.avatar {
|
|
||||||
background: var(--primary-light);
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
.bubble {
|
|
||||||
background: var(--primary-gradient);
|
|
||||||
color: white;
|
|
||||||
border-radius: 18px 18px 4px 18px;
|
|
||||||
box-shadow: 0 2px 10px color-mix(in srgb, var(--primary) 20%, transparent);
|
|
||||||
|
|
||||||
.content {
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// AI 消息
|
|
||||||
&.ai {
|
|
||||||
align-self: flex-start;
|
|
||||||
|
|
||||||
.avatar {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-secondary);
|
|
||||||
}
|
|
||||||
|
|
||||||
.bubble {
|
|
||||||
background: var(--card-bg);
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
border-radius: 18px 18px 18px 4px;
|
|
||||||
backdrop-filter: blur(10px);
|
|
||||||
-webkit-backdrop-filter: blur(10px);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.avatar {
|
|
||||||
flex-shrink: 0;
|
|
||||||
width: 32px;
|
|
||||||
height: 32px;
|
|
||||||
border-radius: 50%;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
}
|
|
||||||
|
|
||||||
.bubble {
|
|
||||||
padding: 12px 16px;
|
|
||||||
flex: 1;
|
|
||||||
min-width: 0;
|
|
||||||
|
|
||||||
.content,
|
|
||||||
.markdown-content {
|
|
||||||
font-size: 14px;
|
|
||||||
line-height: 1.6;
|
|
||||||
color: var(--text-primary);
|
|
||||||
word-wrap: break-word;
|
|
||||||
overflow-wrap: break-word;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Markdown 样式
|
|
||||||
.markdown-content {
|
|
||||||
p {
|
|
||||||
margin: 0 0 0.8em;
|
|
||||||
|
|
||||||
&:last-child {
|
|
||||||
margin-bottom: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
h1,
|
|
||||||
h2,
|
|
||||||
h3,
|
|
||||||
h4,
|
|
||||||
h5,
|
|
||||||
h6 {
|
|
||||||
margin: 1em 0 0.5em;
|
|
||||||
font-weight: 600;
|
|
||||||
line-height: 1.3;
|
|
||||||
color: var(--text-primary);
|
|
||||||
|
|
||||||
&:first-child {
|
|
||||||
margin-top: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
h1 {
|
|
||||||
font-size: 1.5em;
|
|
||||||
}
|
|
||||||
|
|
||||||
h2 {
|
|
||||||
font-size: 1.3em;
|
|
||||||
}
|
|
||||||
|
|
||||||
h3 {
|
|
||||||
font-size: 1.1em;
|
|
||||||
}
|
|
||||||
|
|
||||||
ul,
|
|
||||||
ol {
|
|
||||||
margin: 0.5em 0;
|
|
||||||
padding-left: 1.5em;
|
|
||||||
}
|
|
||||||
|
|
||||||
li {
|
|
||||||
margin: 0.3em 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
code {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
padding: 2px 6px;
|
|
||||||
border-radius: 4px;
|
|
||||||
font-family: 'Consolas', 'Monaco', monospace;
|
|
||||||
font-size: 0.9em;
|
|
||||||
}
|
|
||||||
|
|
||||||
pre {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
padding: 12px;
|
|
||||||
border-radius: 8px;
|
|
||||||
overflow-x: auto;
|
|
||||||
margin: 0.8em 0;
|
|
||||||
|
|
||||||
code {
|
|
||||||
background: none;
|
|
||||||
padding: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
blockquote {
|
|
||||||
border-left: 3px solid var(--primary);
|
|
||||||
padding-left: 12px;
|
|
||||||
margin: 0.8em 0;
|
|
||||||
color: var(--text-secondary);
|
|
||||||
}
|
|
||||||
|
|
||||||
a {
|
|
||||||
color: var(--primary);
|
|
||||||
text-decoration: none;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
text-decoration: underline;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
strong {
|
|
||||||
font-weight: 600;
|
|
||||||
color: var(--text-primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
hr {
|
|
||||||
border: none;
|
|
||||||
border-top: 1px solid var(--border-color);
|
|
||||||
margin: 1em 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
table {
|
|
||||||
border-collapse: collapse;
|
|
||||||
width: 100%;
|
|
||||||
margin: 0.8em 0;
|
|
||||||
|
|
||||||
th,
|
|
||||||
td {
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
padding: 8px 12px;
|
|
||||||
text-align: left;
|
|
||||||
}
|
|
||||||
|
|
||||||
th {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
font-weight: 600;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.list-spacer {
|
|
||||||
height: 100px;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 输入区域
|
|
||||||
.input-area {
|
|
||||||
position: absolute;
|
|
||||||
bottom: 24px;
|
|
||||||
left: 50%;
|
|
||||||
transform: translateX(-50%);
|
|
||||||
width: calc(100% - 64px);
|
|
||||||
max-width: 800px;
|
|
||||||
z-index: 10;
|
|
||||||
|
|
||||||
.input-wrapper {
|
|
||||||
display: flex;
|
|
||||||
align-items: flex-end;
|
|
||||||
gap: 10px;
|
|
||||||
background: var(--card-bg);
|
|
||||||
backdrop-filter: blur(20px);
|
|
||||||
-webkit-backdrop-filter: blur(20px);
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
border-radius: 20px;
|
|
||||||
padding: 10px 14px;
|
|
||||||
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.08);
|
|
||||||
transition: all 0.2s ease;
|
|
||||||
|
|
||||||
&:focus-within {
|
|
||||||
border-color: var(--primary);
|
|
||||||
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1),
|
|
||||||
0 0 0 3px color-mix(in srgb, var(--primary) 15%, transparent);
|
|
||||||
}
|
|
||||||
|
|
||||||
textarea {
|
|
||||||
flex: 1;
|
|
||||||
min-height: 24px;
|
|
||||||
max-height: 120px;
|
|
||||||
padding: 8px 0;
|
|
||||||
background: transparent;
|
|
||||||
border: none;
|
|
||||||
resize: none;
|
|
||||||
color: var(--text-primary);
|
|
||||||
font-size: 14px;
|
|
||||||
font-family: inherit;
|
|
||||||
line-height: 1.5;
|
|
||||||
|
|
||||||
&:focus {
|
|
||||||
outline: none;
|
|
||||||
}
|
|
||||||
|
|
||||||
&::placeholder {
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&:disabled {
|
|
||||||
cursor: not-allowed;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.input-actions {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 8px;
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
// 模型选择器
|
|
||||||
.model-selector {
|
|
||||||
position: relative;
|
|
||||||
|
|
||||||
.model-btn {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
gap: 6px;
|
|
||||||
width: auto;
|
|
||||||
height: 36px;
|
|
||||||
padding: 6px 12px;
|
|
||||||
background: transparent;
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
border-radius: 10px;
|
|
||||||
cursor: pointer;
|
|
||||||
color: var(--text-secondary);
|
|
||||||
font-size: 12px;
|
|
||||||
font-weight: 500;
|
|
||||||
white-space: nowrap;
|
|
||||||
transition: all 0.2s ease;
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
svg {
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
&.spin {
|
|
||||||
animation: spin 1s linear infinite;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
background: var(--bg-hover);
|
|
||||||
border-color: var(--text-tertiary);
|
|
||||||
color: var(--text-primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.loaded {
|
|
||||||
background: color-mix(in srgb, var(--primary) 15%, transparent);
|
|
||||||
border-color: var(--primary);
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.loading {
|
|
||||||
opacity: 0.7;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.disabled {
|
|
||||||
opacity: 0.5;
|
|
||||||
cursor: not-allowed;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.model-dropdown {
|
|
||||||
position: absolute;
|
|
||||||
bottom: 100%;
|
|
||||||
right: 0;
|
|
||||||
margin-bottom: 8px;
|
|
||||||
background: var(--card-bg);
|
|
||||||
backdrop-filter: blur(20px);
|
|
||||||
-webkit-backdrop-filter: blur(20px);
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
border-radius: 12px;
|
|
||||||
box-shadow: 0 8px 24px rgba(0, 0, 0, 0.12);
|
|
||||||
z-index: 100;
|
|
||||||
overflow: hidden;
|
|
||||||
animation: dropdownIn 0.2s ease-out;
|
|
||||||
min-width: 140px;
|
|
||||||
|
|
||||||
.model-option {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: space-between;
|
|
||||||
padding: 10px 14px;
|
|
||||||
cursor: pointer;
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-primary);
|
|
||||||
transition: background 0.15s ease;
|
|
||||||
white-space: nowrap;
|
|
||||||
|
|
||||||
&:hover:not(.disabled) {
|
|
||||||
background: var(--bg-hover);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.active {
|
|
||||||
background: color-mix(in srgb, var(--primary) 20%, transparent);
|
|
||||||
color: var(--primary);
|
|
||||||
font-weight: 600;
|
|
||||||
|
|
||||||
.check {
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.check {
|
|
||||||
margin-left: 8px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
font-weight: 600;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.mode-toggle {
|
|
||||||
width: 36px;
|
|
||||||
height: 36px;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
background: transparent;
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
border-radius: 10px;
|
|
||||||
cursor: pointer;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
transition: all 0.2s ease;
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
background: var(--bg-hover);
|
|
||||||
color: var(--text-primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.active {
|
|
||||||
background: color-mix(in srgb, var(--primary) 15%, transparent);
|
|
||||||
border-color: var(--primary);
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&:disabled {
|
|
||||||
opacity: 0.4;
|
|
||||||
cursor: not-allowed;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.send-btn {
|
|
||||||
width: 36px;
|
|
||||||
height: 36px;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
background: var(--primary-gradient);
|
|
||||||
border: none;
|
|
||||||
border-radius: 10px;
|
|
||||||
cursor: pointer;
|
|
||||||
color: white;
|
|
||||||
transition: all 0.2s ease;
|
|
||||||
flex-shrink: 0;
|
|
||||||
box-shadow: 0 2px 8px color-mix(in srgb, var(--primary) 25%, transparent);
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
transform: scale(1.05);
|
|
||||||
box-shadow: 0 4px 12px color-mix(in srgb, var(--primary) 35%, transparent);
|
|
||||||
}
|
|
||||||
|
|
||||||
&:active:not(:disabled) {
|
|
||||||
transform: scale(0.98);
|
|
||||||
}
|
|
||||||
|
|
||||||
&:disabled {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
box-shadow: none;
|
|
||||||
cursor: not-allowed;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@keyframes messageIn {
|
|
||||||
from {
|
|
||||||
opacity: 0;
|
|
||||||
transform: translateY(8px);
|
|
||||||
}
|
|
||||||
|
|
||||||
to {
|
|
||||||
opacity: 1;
|
|
||||||
transform: translateY(0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@keyframes dropdownIn {
|
|
||||||
from {
|
|
||||||
opacity: 0;
|
|
||||||
transform: translateY(-8px);
|
|
||||||
}
|
|
||||||
|
|
||||||
to {
|
|
||||||
opacity: 1;
|
|
||||||
transform: translateY(0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@keyframes spin {
|
|
||||||
from {
|
|
||||||
transform: rotate(0deg);
|
|
||||||
}
|
|
||||||
|
|
||||||
to {
|
|
||||||
transform: rotate(360deg);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,391 +0,0 @@
|
|||||||
import { useState, useEffect, useRef, useCallback } from 'react'
|
|
||||||
import { Send, Bot, User, Cpu, ChevronDown, Loader2 } from 'lucide-react'
|
|
||||||
import { Virtuoso, VirtuosoHandle } from 'react-virtuoso'
|
|
||||||
import { engineService, PRESET_MODELS, ModelInfo } from '../services/EngineService'
|
|
||||||
import { MessageBubble } from '../components/MessageBubble'
|
|
||||||
import './AIChatPage.scss'
|
|
||||||
|
|
||||||
interface ChatMessage {
|
|
||||||
id: string;
|
|
||||||
role: 'user' | 'ai';
|
|
||||||
content: string;
|
|
||||||
timestamp: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 消息数量限制,避免内存过载
|
|
||||||
const MAX_MESSAGES = 200
|
|
||||||
|
|
||||||
export default function AIChatPage() {
|
|
||||||
const [input, setInput] = useState('')
|
|
||||||
const [messages, setMessages] = useState<ChatMessage[]>([])
|
|
||||||
const [isTyping, setIsTyping] = useState(false)
|
|
||||||
const [models, setModels] = useState<ModelInfo[]>([...PRESET_MODELS])
|
|
||||||
const [selectedModel, setSelectedModel] = useState<string | null>(null)
|
|
||||||
const [modelLoaded, setModelLoaded] = useState(false)
|
|
||||||
const [loadingModel, setLoadingModel] = useState(false)
|
|
||||||
const [isThinkingMode, setIsThinkingMode] = useState(true)
|
|
||||||
const [showModelDropdown, setShowModelDropdown] = useState(false)
|
|
||||||
|
|
||||||
const textareaRef = useRef<HTMLTextAreaElement>(null)
|
|
||||||
const virtuosoRef = useRef<VirtuosoHandle>(null)
|
|
||||||
const dropdownRef = useRef<HTMLDivElement>(null)
|
|
||||||
|
|
||||||
// 流式渲染优化:使用 ref 缓存内容,使用 RAF 批量更新
|
|
||||||
const streamingContentRef = useRef('')
|
|
||||||
const streamingMessageIdRef = useRef<string | null>(null)
|
|
||||||
const rafIdRef = useRef<number | null>(null)
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
checkModelsStatus()
|
|
||||||
|
|
||||||
// 初始化Llama服务(延迟初始化,用户进入此页面时启动)
|
|
||||||
const initLlama = async () => {
|
|
||||||
try {
|
|
||||||
await window.electronAPI.llama?.init()
|
|
||||||
console.log('[AIChatPage] Llama service initialized')
|
|
||||||
} catch (e) {
|
|
||||||
console.error('[AIChatPage] Failed to initialize Llama:', e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
initLlama()
|
|
||||||
|
|
||||||
// 清理函数:组件卸载时释放所有资源
|
|
||||||
return () => {
|
|
||||||
// 取消未完成的 RAF
|
|
||||||
if (rafIdRef.current !== null) {
|
|
||||||
cancelAnimationFrame(rafIdRef.current)
|
|
||||||
rafIdRef.current = null
|
|
||||||
}
|
|
||||||
|
|
||||||
// 清理 engine service 的回调引用
|
|
||||||
engineService.clearCallbacks()
|
|
||||||
}
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
// 监听页面卸载事件,确保资源释放
|
|
||||||
useEffect(() => {
|
|
||||||
const handleBeforeUnload = () => {
|
|
||||||
// 清理回调和监听器
|
|
||||||
engineService.dispose()
|
|
||||||
}
|
|
||||||
|
|
||||||
window.addEventListener('beforeunload', handleBeforeUnload)
|
|
||||||
return () => window.removeEventListener('beforeunload', handleBeforeUnload)
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
// 点击外部关闭下拉框
|
|
||||||
useEffect(() => {
|
|
||||||
const handleClickOutside = (event: MouseEvent) => {
|
|
||||||
if (dropdownRef.current && !dropdownRef.current.contains(event.target as Node)) {
|
|
||||||
setShowModelDropdown(false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
document.addEventListener('mousedown', handleClickOutside)
|
|
||||||
return () => document.removeEventListener('mousedown', handleClickOutside)
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
const scrollToBottom = useCallback(() => {
|
|
||||||
// 使用 virtuoso 的 scrollToIndex 方法滚动到底部
|
|
||||||
if (virtuosoRef.current && messages.length > 0) {
|
|
||||||
virtuosoRef.current.scrollToIndex({
|
|
||||||
index: messages.length - 1,
|
|
||||||
behavior: 'smooth'
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}, [messages.length])
|
|
||||||
|
|
||||||
const checkModelsStatus = async () => {
|
|
||||||
const updatedModels = await Promise.all(models.map(async (m) => {
|
|
||||||
const exists = await engineService.checkModelExists(m.path)
|
|
||||||
return { ...m, downloaded: exists }
|
|
||||||
}))
|
|
||||||
setModels(updatedModels)
|
|
||||||
|
|
||||||
// Auto-select first available model
|
|
||||||
if (!selectedModel) {
|
|
||||||
const available = updatedModels.find(m => m.downloaded)
|
|
||||||
if (available) {
|
|
||||||
setSelectedModel(available.path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 自动加载模型
|
|
||||||
const handleLoadModel = async (modelPath?: string) => {
|
|
||||||
const pathToLoad = modelPath || selectedModel
|
|
||||||
if (!pathToLoad) return false
|
|
||||||
|
|
||||||
setLoadingModel(true)
|
|
||||||
try {
|
|
||||||
await engineService.loadModel(pathToLoad)
|
|
||||||
// Initialize session with system prompt
|
|
||||||
await engineService.createSession("You are a helpful AI assistant.")
|
|
||||||
setModelLoaded(true)
|
|
||||||
return true
|
|
||||||
} catch (e) {
|
|
||||||
console.error("Load failed", e)
|
|
||||||
alert("模型加载失败: " + String(e))
|
|
||||||
return false
|
|
||||||
} finally {
|
|
||||||
setLoadingModel(false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 选择模型(如果有多个)
|
|
||||||
const handleSelectModel = (modelPath: string) => {
|
|
||||||
setSelectedModel(modelPath)
|
|
||||||
setShowModelDropdown(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 获取可用的已下载模型
|
|
||||||
const availableModels = models.filter(m => m.downloaded)
|
|
||||||
const selectedModelInfo = models.find(m => m.path === selectedModel)
|
|
||||||
|
|
||||||
// 优化的流式更新函数:使用 RAF 批量更新
|
|
||||||
const updateStreamingMessage = useCallback(() => {
|
|
||||||
if (!streamingMessageIdRef.current) return
|
|
||||||
|
|
||||||
setMessages(prev => prev.map(msg =>
|
|
||||||
msg.id === streamingMessageIdRef.current
|
|
||||||
? { ...msg, content: streamingContentRef.current }
|
|
||||||
: msg
|
|
||||||
))
|
|
||||||
|
|
||||||
rafIdRef.current = null
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
// Token 回调:使用 RAF 批量更新 UI
|
|
||||||
const handleToken = useCallback((token: string) => {
|
|
||||||
streamingContentRef.current += token
|
|
||||||
|
|
||||||
// 使用 requestAnimationFrame 批量更新,避免频繁渲染
|
|
||||||
if (rafIdRef.current === null) {
|
|
||||||
rafIdRef.current = requestAnimationFrame(updateStreamingMessage)
|
|
||||||
}
|
|
||||||
}, [updateStreamingMessage])
|
|
||||||
|
|
||||||
const handleSend = async () => {
|
|
||||||
if (!input.trim() || isTyping) return
|
|
||||||
|
|
||||||
// 如果模型未加载,先自动加载
|
|
||||||
if (!modelLoaded) {
|
|
||||||
if (!selectedModel) {
|
|
||||||
alert("请先下载模型(设置页面)")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
const loaded = await handleLoadModel()
|
|
||||||
if (!loaded) return
|
|
||||||
}
|
|
||||||
|
|
||||||
const userMsg: ChatMessage = {
|
|
||||||
id: Date.now().toString(),
|
|
||||||
role: 'user',
|
|
||||||
content: input,
|
|
||||||
timestamp: Date.now()
|
|
||||||
}
|
|
||||||
|
|
||||||
setMessages(prev => {
|
|
||||||
const newMessages = [...prev, userMsg]
|
|
||||||
// 限制消息数量,避免内存过载
|
|
||||||
return newMessages.length > MAX_MESSAGES
|
|
||||||
? newMessages.slice(-MAX_MESSAGES)
|
|
||||||
: newMessages
|
|
||||||
})
|
|
||||||
setInput('')
|
|
||||||
setIsTyping(true)
|
|
||||||
|
|
||||||
// Reset textarea height
|
|
||||||
if (textareaRef.current) {
|
|
||||||
textareaRef.current.style.height = 'auto'
|
|
||||||
}
|
|
||||||
|
|
||||||
const aiMsgId = (Date.now() + 1).toString()
|
|
||||||
streamingContentRef.current = ''
|
|
||||||
streamingMessageIdRef.current = aiMsgId
|
|
||||||
|
|
||||||
// Optimistic update for AI message start
|
|
||||||
setMessages(prev => {
|
|
||||||
const newMessages = [...prev, {
|
|
||||||
id: aiMsgId,
|
|
||||||
role: 'ai' as const,
|
|
||||||
content: '',
|
|
||||||
timestamp: Date.now()
|
|
||||||
}]
|
|
||||||
return newMessages.length > MAX_MESSAGES
|
|
||||||
? newMessages.slice(-MAX_MESSAGES)
|
|
||||||
: newMessages
|
|
||||||
})
|
|
||||||
|
|
||||||
// Append thinking command based on mode
|
|
||||||
const msgWithSuffix = input + (isThinkingMode ? " /think" : " /no_think")
|
|
||||||
|
|
||||||
try {
|
|
||||||
await engineService.chat(msgWithSuffix, handleToken, { thinking: isThinkingMode })
|
|
||||||
} catch (e) {
|
|
||||||
console.error("Chat failed", e)
|
|
||||||
setMessages(prev => [...prev, {
|
|
||||||
id: Date.now().toString(),
|
|
||||||
role: 'ai',
|
|
||||||
content: "❌ Error: Failed to get response from AI.",
|
|
||||||
timestamp: Date.now()
|
|
||||||
}])
|
|
||||||
} finally {
|
|
||||||
setIsTyping(false)
|
|
||||||
streamingMessageIdRef.current = null
|
|
||||||
|
|
||||||
// 确保最终状态同步
|
|
||||||
if (rafIdRef.current !== null) {
|
|
||||||
cancelAnimationFrame(rafIdRef.current)
|
|
||||||
updateStreamingMessage()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 渲染模型选择按钮(集成在输入框作为下拉项)
|
|
||||||
const renderModelSelector = () => {
|
|
||||||
// 没有可用模型
|
|
||||||
if (availableModels.length === 0) {
|
|
||||||
return (
|
|
||||||
<button
|
|
||||||
className="model-btn disabled"
|
|
||||||
title="请先在设置页面下载模型"
|
|
||||||
>
|
|
||||||
<Bot size={16} />
|
|
||||||
<span>无模型</span>
|
|
||||||
</button>
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 只有一个模型,直接显示
|
|
||||||
if (availableModels.length === 1) {
|
|
||||||
return (
|
|
||||||
<button
|
|
||||||
className={`model-btn ${modelLoaded ? 'loaded' : ''} ${loadingModel ? 'loading' : ''}`}
|
|
||||||
title={modelLoaded ? "模型已就绪" : "发送消息时自动加载"}
|
|
||||||
>
|
|
||||||
{loadingModel ? (
|
|
||||||
<Loader2 size={16} className="spin" />
|
|
||||||
) : (
|
|
||||||
<Bot size={16} />
|
|
||||||
)}
|
|
||||||
<span>{loadingModel ? '加载中' : selectedModelInfo?.name || '模型'}</span>
|
|
||||||
</button>
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 多个模型,显示下拉选择
|
|
||||||
return (
|
|
||||||
<div className="model-selector" ref={dropdownRef}>
|
|
||||||
<button
|
|
||||||
className={`model-btn ${modelLoaded ? 'loaded' : ''} ${loadingModel ? 'loading' : ''}`}
|
|
||||||
onClick={() => !loadingModel && setShowModelDropdown(!showModelDropdown)}
|
|
||||||
title="点击选择模型"
|
|
||||||
>
|
|
||||||
{loadingModel ? (
|
|
||||||
<Loader2 size={16} className="spin" />
|
|
||||||
) : (
|
|
||||||
<Bot size={16} />
|
|
||||||
)}
|
|
||||||
<span>{loadingModel ? '加载中' : selectedModelInfo?.name || '选择模型'}</span>
|
|
||||||
<ChevronDown size={13} className={showModelDropdown ? 'rotate' : ''} />
|
|
||||||
</button>
|
|
||||||
|
|
||||||
{showModelDropdown && (
|
|
||||||
<div className="model-dropdown">
|
|
||||||
{availableModels.map(model => (
|
|
||||||
<div
|
|
||||||
key={model.path}
|
|
||||||
className={`model-option ${selectedModel === model.path ? 'active' : ''}`}
|
|
||||||
onClick={() => handleSelectModel(model.path)}
|
|
||||||
>
|
|
||||||
<span>{model.name}</span>
|
|
||||||
{selectedModel === model.path && (
|
|
||||||
<span className="check">✓</span>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="ai-chat-page">
|
|
||||||
<div className="chat-main">
|
|
||||||
{messages.length === 0 ? (
|
|
||||||
<div className="empty-state">
|
|
||||||
<div className="icon">
|
|
||||||
<Bot size={40} />
|
|
||||||
</div>
|
|
||||||
<h2>AI 为你服务</h2>
|
|
||||||
<p>
|
|
||||||
{availableModels.length === 0
|
|
||||||
? "请先在设置页面下载模型"
|
|
||||||
: "输入消息开始对话,模型将自动加载"
|
|
||||||
}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<Virtuoso
|
|
||||||
ref={virtuosoRef}
|
|
||||||
data={messages}
|
|
||||||
className="messages-list"
|
|
||||||
initialTopMostItemIndex={messages.length - 1}
|
|
||||||
followOutput="smooth"
|
|
||||||
itemContent={(index, message) => (
|
|
||||||
<MessageBubble key={message.id} message={message} />
|
|
||||||
)}
|
|
||||||
components={{
|
|
||||||
Footer: () => <div className="list-spacer" />
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<div className="input-area">
|
|
||||||
<div className="input-wrapper">
|
|
||||||
<textarea
|
|
||||||
ref={textareaRef}
|
|
||||||
value={input}
|
|
||||||
onChange={e => {
|
|
||||||
setInput(e.target.value)
|
|
||||||
e.target.style.height = 'auto'
|
|
||||||
e.target.style.height = `${Math.min(e.target.scrollHeight, 120)}px`
|
|
||||||
}}
|
|
||||||
onKeyDown={e => {
|
|
||||||
if (e.key === 'Enter' && !e.shiftKey) {
|
|
||||||
e.preventDefault()
|
|
||||||
handleSend()
|
|
||||||
// Reset height after send
|
|
||||||
if (textareaRef.current) textareaRef.current.style.height = 'auto'
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
placeholder={availableModels.length === 0 ? "请先下载模型..." : "输入消息..."}
|
|
||||||
disabled={availableModels.length === 0 || loadingModel}
|
|
||||||
rows={1}
|
|
||||||
/>
|
|
||||||
<div className="input-actions">
|
|
||||||
{renderModelSelector()}
|
|
||||||
<button
|
|
||||||
className={`mode-toggle ${isThinkingMode ? 'active' : ''}`}
|
|
||||||
onClick={() => setIsThinkingMode(!isThinkingMode)}
|
|
||||||
title={isThinkingMode ? "深度思考模式已开启" : "深度思考模式已关闭"}
|
|
||||||
disabled={availableModels.length === 0}
|
|
||||||
>
|
|
||||||
<Cpu size={18} />
|
|
||||||
</button>
|
|
||||||
<button
|
|
||||||
className="send-btn"
|
|
||||||
onClick={handleSend}
|
|
||||||
disabled={!input.trim() || availableModels.length === 0 || isTyping || loadingModel}
|
|
||||||
>
|
|
||||||
<Send size={18} />
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
}
|
|
||||||
@@ -45,6 +45,12 @@
|
|||||||
font-weight: 600;
|
font-weight: 600;
|
||||||
color: var(--primary);
|
color: var(--primary);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.error-actions {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.page-header {
|
.page-header {
|
||||||
@@ -482,11 +488,41 @@
|
|||||||
margin-top: 16px;
|
margin-top: 16px;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.exclude-footer-left {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
.exclude-count {
|
.exclude-count {
|
||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
color: var(--text-tertiary);
|
color: var(--text-tertiary);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.btn-text {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 4px;
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
padding: 4px 8px;
|
||||||
|
border-radius: 6px;
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
color: var(--primary);
|
||||||
|
background: var(--primary-light);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.exclude-actions {
|
.exclude-actions {
|
||||||
display: flex;
|
display: flex;
|
||||||
gap: 8px;
|
gap: 8px;
|
||||||
|
|||||||
@@ -108,6 +108,7 @@ function AnalyticsPage() {
|
|||||||
}, [loadExcludedUsernames])
|
}, [loadExcludedUsernames])
|
||||||
|
|
||||||
const handleRefresh = () => loadData(true)
|
const handleRefresh = () => loadData(true)
|
||||||
|
const isNoSessionError = error?.includes('未找到消息会话') ?? false
|
||||||
|
|
||||||
const loadExcludeCandidates = useCallback(async () => {
|
const loadExcludeCandidates = useCallback(async () => {
|
||||||
setExcludeLoading(true)
|
setExcludeLoading(true)
|
||||||
@@ -146,6 +147,17 @@ function AnalyticsPage() {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const toggleInvertSelection = () => {
|
||||||
|
setDraftExcluded((prev) => {
|
||||||
|
const allUsernames = new Set(excludeCandidates.map(c => normalizeUsername(c.username)))
|
||||||
|
const inverted = new Set<string>()
|
||||||
|
for (const u of allUsernames) {
|
||||||
|
if (!prev.has(u)) inverted.add(u)
|
||||||
|
}
|
||||||
|
return inverted
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
const handleApplyExcluded = async () => {
|
const handleApplyExcluded = async () => {
|
||||||
const payload = Array.from(draftExcluded)
|
const payload = Array.from(draftExcluded)
|
||||||
setIsExcludeDialogOpen(false)
|
setIsExcludeDialogOpen(false)
|
||||||
@@ -164,6 +176,23 @@ function AnalyticsPage() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const handleResetExcluded = async () => {
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.analytics.setExcludedUsernames([])
|
||||||
|
if (!result.success) {
|
||||||
|
setError(result.error || '重置排除好友失败')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
setExcludedUsernames(new Set())
|
||||||
|
setDraftExcluded(new Set())
|
||||||
|
clearCache()
|
||||||
|
await window.electronAPI.cache.clearAnalytics()
|
||||||
|
await loadData(true)
|
||||||
|
} catch (e) {
|
||||||
|
setError(`重置排除好友失败: ${String(e)}`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const visibleExcludeCandidates = excludeCandidates
|
const visibleExcludeCandidates = excludeCandidates
|
||||||
.filter((candidate) => {
|
.filter((candidate) => {
|
||||||
const query = excludeQuery.trim().toLowerCase()
|
const query = excludeQuery.trim().toLowerCase()
|
||||||
@@ -344,6 +373,22 @@ function AnalyticsPage() {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (error && !isLoaded && isNoSessionError && excludedUsernames.size > 0) {
|
||||||
|
return (
|
||||||
|
<div className="error-container">
|
||||||
|
<p>{error}</p>
|
||||||
|
<div className="error-actions">
|
||||||
|
<button className="btn btn-secondary" onClick={handleResetExcluded}>
|
||||||
|
重置排除好友
|
||||||
|
</button>
|
||||||
|
<button className="btn btn-primary" onClick={() => loadData(true)}>
|
||||||
|
重试
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
if (error && !isLoaded) {
|
if (error && !isLoaded) {
|
||||||
return (<div className="error-container"><p>{error}</p><button className="btn btn-primary" onClick={() => loadData(true)}>重试</button></div>)
|
return (<div className="error-container"><p>{error}</p><button className="btn btn-primary" onClick={() => loadData(true)}>重试</button></div>)
|
||||||
}
|
}
|
||||||
@@ -493,7 +538,12 @@ function AnalyticsPage() {
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
<div className="exclude-modal-footer">
|
<div className="exclude-modal-footer">
|
||||||
|
<div className="exclude-footer-left">
|
||||||
<span className="exclude-count">已排除 {draftExcluded.size} 人</span>
|
<span className="exclude-count">已排除 {draftExcluded.size} 人</span>
|
||||||
|
<button className="btn btn-text" onClick={toggleInvertSelection} disabled={excludeLoading}>
|
||||||
|
反选
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
<div className="exclude-actions">
|
<div className="exclude-actions">
|
||||||
<button className="btn btn-secondary" onClick={() => setIsExcludeDialogOpen(false)}>
|
<button className="btn btn-secondary" onClick={() => setIsExcludeDialogOpen(false)}>
|
||||||
取消
|
取消
|
||||||
|
|||||||
@@ -1,8 +1,214 @@
|
|||||||
.chat-page {
|
.chat-page {
|
||||||
|
position: relative;
|
||||||
display: flex;
|
display: flex;
|
||||||
height: 100%;
|
height: 100%;
|
||||||
gap: 16px;
|
gap: 16px;
|
||||||
|
|
||||||
|
// 批量删除进度遮罩
|
||||||
|
.delete-progress-overlay {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
bottom: 0;
|
||||||
|
z-index: 9999;
|
||||||
|
background: rgba(0, 0, 0, 0.4);
|
||||||
|
backdrop-filter: blur(8px);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
animation: fadeIn 0.3s ease;
|
||||||
|
|
||||||
|
.delete-progress-card {
|
||||||
|
width: 400px;
|
||||||
|
padding: 24px;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 16px;
|
||||||
|
box-shadow: 0 10px 40px rgba(0, 0, 0, 0.3);
|
||||||
|
text-align: center;
|
||||||
|
|
||||||
|
.progress-header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 18px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.count {
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
font-weight: 600;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
padding: 2px 8px;
|
||||||
|
border-radius: 4px;
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-bar-container {
|
||||||
|
height: 10px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 5px;
|
||||||
|
overflow: hidden;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
|
||||||
|
.progress-bar-fill {
|
||||||
|
height: 100%;
|
||||||
|
background: linear-gradient(90deg, var(--primary), var(--primary-light));
|
||||||
|
transition: width 0.3s cubic-bezier(0.4, 0, 0.2, 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-footer {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 16px;
|
||||||
|
|
||||||
|
p {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.cancel-delete-btn {
|
||||||
|
padding: 10px;
|
||||||
|
border-radius: 8px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
font-size: 14px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover:not(:disabled) {
|
||||||
|
background: var(--danger-light);
|
||||||
|
color: var(--danger);
|
||||||
|
border-color: var(--danger);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.6;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 自定义删除确认对话框
|
||||||
|
.delete-confirm-overlay {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
bottom: 0;
|
||||||
|
z-index: 10000;
|
||||||
|
background: rgba(0, 0, 0, 0.3);
|
||||||
|
backdrop-filter: blur(4px);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
animation: fadeIn 0.2s ease;
|
||||||
|
|
||||||
|
.delete-confirm-card {
|
||||||
|
width: 360px;
|
||||||
|
padding: 32px 24px 24px;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 20px;
|
||||||
|
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.4);
|
||||||
|
text-align: center;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
|
||||||
|
.confirm-icon {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
padding: 16px;
|
||||||
|
background: var(--danger-light);
|
||||||
|
border-radius: 50%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.confirm-content {
|
||||||
|
margin-bottom: 32px;
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
margin: 0 0 12px;
|
||||||
|
font-size: 20px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
p {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.confirm-actions {
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
width: 100%;
|
||||||
|
|
||||||
|
button {
|
||||||
|
flex: 1;
|
||||||
|
padding: 12px;
|
||||||
|
border-radius: 12px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary {
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-danger-filled {
|
||||||
|
background: var(--danger);
|
||||||
|
border: none;
|
||||||
|
color: white;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: #e54d45; // Darker red
|
||||||
|
transform: translateY(-1px);
|
||||||
|
box-shadow: 0 4px 12px rgba(229, 77, 69, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 独立窗口模式 - EchoTrace 特色风格(使用主题变量)
|
// 独立窗口模式 - EchoTrace 特色风格(使用主题变量)
|
||||||
&.standalone {
|
&.standalone {
|
||||||
height: 100vh;
|
height: 100vh;
|
||||||
@@ -1082,6 +1288,21 @@
|
|||||||
z-index: 2;
|
z-index: 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.empty-chat-inline {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 60px 0;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 14px;
|
||||||
|
|
||||||
|
svg {
|
||||||
|
opacity: 0.4;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.message-list * {
|
.message-list * {
|
||||||
-webkit-app-region: no-drag !important;
|
-webkit-app-region: no-drag !important;
|
||||||
}
|
}
|
||||||
@@ -2511,6 +2732,7 @@
|
|||||||
|
|
||||||
// 发送消息中的特殊消息类型适配(除了文件和转账)
|
// 发送消息中的特殊消息类型适配(除了文件和转账)
|
||||||
.message-bubble.sent {
|
.message-bubble.sent {
|
||||||
|
|
||||||
.card-message,
|
.card-message,
|
||||||
.chat-record-message,
|
.chat-record-message,
|
||||||
.miniapp-message {
|
.miniapp-message {
|
||||||
@@ -2616,6 +2838,7 @@
|
|||||||
&:hover:not(:disabled) {
|
&:hover:not(:disabled) {
|
||||||
color: var(--primary-color);
|
color: var(--primary-color);
|
||||||
}
|
}
|
||||||
|
|
||||||
&.transcribing {
|
&.transcribing {
|
||||||
color: var(--primary-color);
|
color: var(--primary-color);
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
@@ -2637,7 +2860,9 @@
|
|||||||
padding: 1.5rem;
|
padding: 1.5rem;
|
||||||
border-bottom: 1px solid var(--border-color);
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
svg { color: var(--primary-color); }
|
svg {
|
||||||
|
color: var(--primary-color);
|
||||||
|
}
|
||||||
|
|
||||||
h3 {
|
h3 {
|
||||||
margin: 0;
|
margin: 0;
|
||||||
@@ -2697,7 +2922,10 @@
|
|||||||
|
|
||||||
li {
|
li {
|
||||||
border-bottom: 1px solid var(--border-color);
|
border-bottom: 1px solid var(--border-color);
|
||||||
&:last-child { border-bottom: none; }
|
|
||||||
|
&:last-child {
|
||||||
|
border-bottom: none;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.batch-date-row {
|
.batch-date-row {
|
||||||
@@ -2708,7 +2936,9 @@
|
|||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
transition: background 0.15s;
|
transition: background 0.15s;
|
||||||
|
|
||||||
&:hover { background: var(--bg-hover); }
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
input[type="checkbox"] {
|
input[type="checkbox"] {
|
||||||
accent-color: var(--primary-color);
|
accent-color: var(--primary-color);
|
||||||
@@ -2806,13 +3036,185 @@
|
|||||||
&.btn-secondary {
|
&.btn-secondary {
|
||||||
background: var(--bg-tertiary);
|
background: var(--bg-tertiary);
|
||||||
color: var(--text-primary);
|
color: var(--text-primary);
|
||||||
&:hover { background: var(--border-color); }
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--border-color);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
&.btn-primary, &.batch-transcribe-start-btn {
|
&.btn-primary,
|
||||||
|
&.batch-transcribe-start-btn {
|
||||||
background: var(--primary-color);
|
background: var(--primary-color);
|
||||||
color: white;
|
color: white;
|
||||||
&:hover { opacity: 0.9; }
|
|
||||||
|
&:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Context Menu
|
||||||
|
.context-menu-overlay {
|
||||||
|
background: transparent;
|
||||||
|
}
|
||||||
|
|
||||||
|
.context-menu {
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 8px;
|
||||||
|
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
|
||||||
|
padding: 4px;
|
||||||
|
min-width: 140px;
|
||||||
|
backdrop-filter: blur(10px);
|
||||||
|
animation: fadeIn 0.1s ease-out;
|
||||||
|
|
||||||
|
.menu-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 8px 12px;
|
||||||
|
cursor: pointer;
|
||||||
|
border-radius: 4px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 13px;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.delete {
|
||||||
|
color: var(--danger);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(220, 53, 69, 0.1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
svg {
|
||||||
|
opacity: 0.7;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Modal Overlay
|
||||||
|
.modal-overlay {
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
bottom: 0;
|
||||||
|
background: rgba(0, 0, 0, 0.6);
|
||||||
|
backdrop-filter: blur(4px);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
z-index: 2000;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edit Message Modal
|
||||||
|
.edit-message-modal {
|
||||||
|
width: 500px;
|
||||||
|
max-width: 90vw;
|
||||||
|
max-height: 85vh;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border-radius: 12px;
|
||||||
|
box-shadow: 0 8px 30px rgba(0, 0, 0, 0.2);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
overflow: hidden;
|
||||||
|
|
||||||
|
.modal-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 16px 20px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.close-btn {
|
||||||
|
width: 28px;
|
||||||
|
height: 28px;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
border-radius: 6px;
|
||||||
|
cursor: pointer;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.edit-message-textarea {
|
||||||
|
flex: 1;
|
||||||
|
border: none;
|
||||||
|
padding: 16px 20px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 14px;
|
||||||
|
line-height: 1.6;
|
||||||
|
resize: none;
|
||||||
|
outline: none;
|
||||||
|
min-height: 120px;
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
background: var(--bg-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 16px 20px;
|
||||||
|
border-top: 1px solid var(--border-color);
|
||||||
|
background: var(--card-bg);
|
||||||
|
|
||||||
|
button {
|
||||||
|
padding: 8px 16px;
|
||||||
|
border-radius: 6px;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border: 1px solid transparent;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary {
|
||||||
|
background: var(--primary);
|
||||||
|
border: 1px solid transparent;
|
||||||
|
color: #fff;
|
||||||
|
box-shadow: 0 2px 8px var(--primary-light);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--primary-hover);
|
||||||
|
box-shadow: 0 4px 12px var(--primary-light);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import React, { useState, useEffect, useRef, useCallback, useMemo } from 'react'
|
import React, { useState, useEffect, useRef, useCallback, useMemo } from 'react'
|
||||||
import { Search, MessageSquare, AlertCircle, Loader2, RefreshCw, X, ChevronDown, Info, Calendar, Database, Hash, Play, Pause, Image as ImageIcon, Link, Mic, CheckCircle, Copy, Check, Download, BarChart3 } from 'lucide-react'
|
import { Search, MessageSquare, AlertCircle, Loader2, RefreshCw, X, ChevronDown, Info, Calendar, Database, Hash, Play, Pause, Image as ImageIcon, Link, Mic, CheckCircle, Copy, Check, CheckSquare, Download, BarChart3, Edit2, Trash2 } from 'lucide-react'
|
||||||
import { useNavigate } from 'react-router-dom'
|
import { useNavigate } from 'react-router-dom'
|
||||||
import { createPortal } from 'react-dom'
|
import { createPortal } from 'react-dom'
|
||||||
import { useChatStore } from '../stores/chatStore'
|
import { useChatStore } from '../stores/chatStore'
|
||||||
@@ -18,6 +18,102 @@ const SYSTEM_MESSAGE_TYPES = [
|
|||||||
266287972401, // 拍一拍
|
266287972401, // 拍一拍
|
||||||
]
|
]
|
||||||
|
|
||||||
|
interface XmlField {
|
||||||
|
key: string;
|
||||||
|
value: string;
|
||||||
|
type: 'attr' | 'node';
|
||||||
|
tagName?: string;
|
||||||
|
path: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尝试解析 XML 为可编辑字段
|
||||||
|
function parseXmlToFields(xml: string): XmlField[] {
|
||||||
|
const fields: XmlField[] = []
|
||||||
|
if (!xml || !xml.includes('<')) return []
|
||||||
|
try {
|
||||||
|
const parser = new DOMParser()
|
||||||
|
// 包装一下确保是单一根节点
|
||||||
|
const wrappedXml = xml.trim().startsWith('<?xml') ? xml : `<root>${xml}</root>`
|
||||||
|
const doc = parser.parseFromString(wrappedXml, 'text/xml')
|
||||||
|
const errorNode = doc.querySelector('parsererror')
|
||||||
|
if (errorNode) return []
|
||||||
|
|
||||||
|
const walk = (node: Node, path: string = '') => {
|
||||||
|
if (node.nodeType === Node.ELEMENT_NODE) {
|
||||||
|
const element = node as Element
|
||||||
|
if (element.tagName === 'root') {
|
||||||
|
node.childNodes.forEach((child, index) => walk(child, path))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const currentPath = path ? `${path} > ${element.tagName}` : element.tagName
|
||||||
|
|
||||||
|
for (let i = 0; i < element.attributes.length; i++) {
|
||||||
|
const attr = element.attributes[i]
|
||||||
|
fields.push({
|
||||||
|
key: attr.name,
|
||||||
|
value: attr.value,
|
||||||
|
type: 'attr',
|
||||||
|
tagName: element.tagName,
|
||||||
|
path: `${currentPath}[@${attr.name}]`
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if (element.childNodes.length === 1 && element.childNodes[0].nodeType === Node.TEXT_NODE) {
|
||||||
|
const text = element.textContent?.trim() || ''
|
||||||
|
if (text) {
|
||||||
|
fields.push({
|
||||||
|
key: element.tagName,
|
||||||
|
value: text,
|
||||||
|
type: 'node',
|
||||||
|
path: currentPath
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
node.childNodes.forEach((child, index) => walk(child, `${currentPath}[${index}]`))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
doc.childNodes.forEach((node, index) => walk(node, ''))
|
||||||
|
} catch (e) {
|
||||||
|
console.warn('[XML Parse] Failed:', e)
|
||||||
|
}
|
||||||
|
return fields
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将编辑后的字段同步回 XML
|
||||||
|
function updateXmlWithFields(xml: string, fields: XmlField[]): string {
|
||||||
|
try {
|
||||||
|
const parser = new DOMParser()
|
||||||
|
const wrappedXml = xml.trim().startsWith('<?xml') ? xml : `<root>${xml}</root>`
|
||||||
|
const doc = parser.parseFromString(wrappedXml, 'text/xml')
|
||||||
|
const errorNode = doc.querySelector('parsererror')
|
||||||
|
if (errorNode) return xml
|
||||||
|
|
||||||
|
fields.forEach(f => {
|
||||||
|
if (f.type === 'attr') {
|
||||||
|
const elements = doc.getElementsByTagName(f.tagName!)
|
||||||
|
if (elements.length > 0) {
|
||||||
|
elements[0].setAttribute(f.key, f.value)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
const elements = doc.getElementsByTagName(f.key)
|
||||||
|
if (elements.length > 0 && (elements[0].childNodes.length <= 1)) {
|
||||||
|
elements[0].textContent = f.value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
let result = new XMLSerializer().serializeToString(doc)
|
||||||
|
if (!xml.trim().startsWith('<?xml')) {
|
||||||
|
result = result.replace('<root>', '').replace('</root>', '').replace('<root/>', '')
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
} catch (e) {
|
||||||
|
return xml
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 判断是否为系统消息
|
// 判断是否为系统消息
|
||||||
function isSystemMessage(localType: number): boolean {
|
function isSystemMessage(localType: number): boolean {
|
||||||
return SYSTEM_MESSAGE_TYPES.includes(localType)
|
return SYSTEM_MESSAGE_TYPES.includes(localType)
|
||||||
@@ -32,6 +128,12 @@ function formatFileSize(bytes: number): string {
|
|||||||
return Math.round(bytes / Math.pow(k, i) * 100) / 100 + ' ' + sizes[i]
|
return Math.round(bytes / Math.pow(k, i) * 100) / 100 + ' ' + sizes[i]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 清理消息内容的辅助函数
|
||||||
|
function cleanMessageContent(content: string): string {
|
||||||
|
if (!content) return ''
|
||||||
|
return content.trim()
|
||||||
|
}
|
||||||
|
|
||||||
interface ChatPageProps {
|
interface ChatPageProps {
|
||||||
// 保留接口以备将来扩展
|
// 保留接口以备将来扩展
|
||||||
}
|
}
|
||||||
@@ -179,9 +281,23 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
const [highlightedMessageKeys, setHighlightedMessageKeys] = useState<string[]>([])
|
const [highlightedMessageKeys, setHighlightedMessageKeys] = useState<string[]>([])
|
||||||
const [isRefreshingSessions, setIsRefreshingSessions] = useState(false)
|
const [isRefreshingSessions, setIsRefreshingSessions] = useState(false)
|
||||||
const [hasInitialMessages, setHasInitialMessages] = useState(false)
|
const [hasInitialMessages, setHasInitialMessages] = useState(false)
|
||||||
|
const [noMessageTable, setNoMessageTable] = useState(false)
|
||||||
|
const [fallbackDisplayName, setFallbackDisplayName] = useState<string | null>(null)
|
||||||
const [showVoiceTranscribeDialog, setShowVoiceTranscribeDialog] = useState(false)
|
const [showVoiceTranscribeDialog, setShowVoiceTranscribeDialog] = useState(false)
|
||||||
const [pendingVoiceTranscriptRequest, setPendingVoiceTranscriptRequest] = useState<{ sessionId: string; messageId: string } | null>(null)
|
const [pendingVoiceTranscriptRequest, setPendingVoiceTranscriptRequest] = useState<{ sessionId: string; messageId: string } | null>(null)
|
||||||
|
|
||||||
|
// 消息右键菜单
|
||||||
|
const [contextMenu, setContextMenu] = useState<{ x: number, y: number, message: Message } | null>(null)
|
||||||
|
const [editingMessage, setEditingMessage] = useState<{ message: Message, content: string } | null>(null)
|
||||||
|
|
||||||
|
// 多选模式
|
||||||
|
const [isSelectionMode, setIsSelectionMode] = useState(false)
|
||||||
|
const [selectedMessages, setSelectedMessages] = useState<Set<number>>(new Set())
|
||||||
|
|
||||||
|
// 编辑消息额外状态
|
||||||
|
const [editMode, setEditMode] = useState<'raw' | 'fields'>('raw')
|
||||||
|
const [tempFields, setTempFields] = useState<XmlField[]>([])
|
||||||
|
|
||||||
// 批量语音转文字相关状态(进度/结果 由全局 store 管理)
|
// 批量语音转文字相关状态(进度/结果 由全局 store 管理)
|
||||||
const { isBatchTranscribing, progress: batchTranscribeProgress, showToast: showBatchProgress, startTranscribe, updateProgress, finishTranscribe, setShowToast: setShowBatchProgress } = useBatchTranscribeStore()
|
const { isBatchTranscribing, progress: batchTranscribeProgress, showToast: showBatchProgress, startTranscribe, updateProgress, finishTranscribe, setShowToast: setShowBatchProgress } = useBatchTranscribeStore()
|
||||||
const [showBatchConfirm, setShowBatchConfirm] = useState(false)
|
const [showBatchConfirm, setShowBatchConfirm] = useState(false)
|
||||||
@@ -190,6 +306,19 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
const [batchVoiceDates, setBatchVoiceDates] = useState<string[]>([])
|
const [batchVoiceDates, setBatchVoiceDates] = useState<string[]>([])
|
||||||
const [batchSelectedDates, setBatchSelectedDates] = useState<Set<string>>(new Set())
|
const [batchSelectedDates, setBatchSelectedDates] = useState<Set<string>>(new Set())
|
||||||
|
|
||||||
|
// 批量删除相关状态
|
||||||
|
const [isDeleting, setIsDeleting] = useState(false)
|
||||||
|
const [deleteProgress, setDeleteProgress] = useState({ current: 0, total: 0 })
|
||||||
|
const [cancelDeleteRequested, setCancelDeleteRequested] = useState(false)
|
||||||
|
|
||||||
|
// 自定义删除确认对话框
|
||||||
|
const [deleteConfirm, setDeleteConfirm] = useState<{
|
||||||
|
show: boolean;
|
||||||
|
mode: 'single' | 'batch';
|
||||||
|
message?: Message;
|
||||||
|
count?: number;
|
||||||
|
}>({ show: false, mode: 'single' })
|
||||||
|
|
||||||
// 联系人信息加载控制
|
// 联系人信息加载控制
|
||||||
const isEnrichingRef = useRef(false)
|
const isEnrichingRef = useRef(false)
|
||||||
const enrichCancelledRef = useRef(false)
|
const enrichCancelledRef = useRef(false)
|
||||||
@@ -730,6 +859,10 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
if (result.success && result.messages) {
|
if (result.success && result.messages) {
|
||||||
if (offset === 0) {
|
if (offset === 0) {
|
||||||
setMessages(result.messages)
|
setMessages(result.messages)
|
||||||
|
if (result.messages.length === 0) {
|
||||||
|
setNoMessageTable(true)
|
||||||
|
setHasMoreMessages(false)
|
||||||
|
}
|
||||||
|
|
||||||
// 预取发送者信息:在关闭加载遮罩前处理
|
// 预取发送者信息:在关闭加载遮罩前处理
|
||||||
const unreadCount = session?.unreadCount ?? 0
|
const unreadCount = session?.unreadCount ?? 0
|
||||||
@@ -802,7 +935,7 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
}
|
}
|
||||||
setCurrentOffset(offset + result.messages.length)
|
setCurrentOffset(offset + result.messages.length)
|
||||||
} else if (!result.success) {
|
} else if (!result.success) {
|
||||||
setConnectionError(result.error || '加载消息失败')
|
setNoMessageTable(true)
|
||||||
setHasMoreMessages(false)
|
setHasMoreMessages(false)
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
@@ -1120,6 +1253,7 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (currentSessionId !== prevSessionRef.current) {
|
if (currentSessionId !== prevSessionRef.current) {
|
||||||
prevSessionRef.current = currentSessionId
|
prevSessionRef.current = currentSessionId
|
||||||
|
setNoMessageTable(false)
|
||||||
if (initialRevealTimerRef.current !== null) {
|
if (initialRevealTimerRef.current !== null) {
|
||||||
window.clearTimeout(initialRevealTimerRef.current)
|
window.clearTimeout(initialRevealTimerRef.current)
|
||||||
initialRevealTimerRef.current = null
|
initialRevealTimerRef.current = null
|
||||||
@@ -1133,10 +1267,11 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
}, [currentSessionId, messages.length, isLoadingMessages])
|
}, [currentSessionId, messages.length, isLoadingMessages])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (currentSessionId && messages.length === 0 && !isLoadingMessages && !isLoadingMore) {
|
if (currentSessionId && messages.length === 0 && !isLoadingMessages && !isLoadingMore && !noMessageTable) {
|
||||||
|
setHasInitialMessages(false)
|
||||||
loadMessages(currentSessionId, 0)
|
loadMessages(currentSessionId, 0)
|
||||||
}
|
}
|
||||||
}, [currentSessionId, messages.length, isLoadingMessages, isLoadingMore])
|
}, [currentSessionId, messages.length, isLoadingMessages, isLoadingMore, noMessageTable])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
return () => {
|
return () => {
|
||||||
@@ -1200,8 +1335,35 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
return `${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`
|
return `${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`
|
||||||
}, [])
|
}, [])
|
||||||
|
|
||||||
// 获取当前会话信息
|
// 获取当前会话信息(从通讯录跳转时可能不在 sessions 列表中,构造 fallback)
|
||||||
const currentSession = Array.isArray(sessions) ? sessions.find(s => s.username === currentSessionId) : undefined
|
const currentSession = (() => {
|
||||||
|
const found = Array.isArray(sessions) ? sessions.find(s => s.username === currentSessionId) : undefined
|
||||||
|
if (found || !currentSessionId) return found
|
||||||
|
return {
|
||||||
|
username: currentSessionId,
|
||||||
|
type: 0,
|
||||||
|
unreadCount: 0,
|
||||||
|
summary: '',
|
||||||
|
sortTimestamp: 0,
|
||||||
|
lastTimestamp: 0,
|
||||||
|
lastMsgType: 0,
|
||||||
|
displayName: fallbackDisplayName || currentSessionId,
|
||||||
|
} as ChatSession
|
||||||
|
})()
|
||||||
|
|
||||||
|
// 从通讯录跳转时,会话不在列表中,主动加载联系人显示名称
|
||||||
|
useEffect(() => {
|
||||||
|
if (!currentSessionId) return
|
||||||
|
const found = Array.isArray(sessions) ? sessions.find(s => s.username === currentSessionId) : undefined
|
||||||
|
if (found) {
|
||||||
|
setFallbackDisplayName(null)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
loadContactInfoBatch([currentSessionId]).then(() => {
|
||||||
|
const cached = senderAvatarCache.get(currentSessionId)
|
||||||
|
if (cached?.displayName) setFallbackDisplayName(cached.displayName)
|
||||||
|
})
|
||||||
|
}, [currentSessionId, sessions])
|
||||||
|
|
||||||
// 判断是否为群聊
|
// 判断是否为群聊
|
||||||
const isGroupChat = (username: string) => username.includes('@chatroom')
|
const isGroupChat = (username: string) => username.includes('@chatroom')
|
||||||
@@ -1394,13 +1556,302 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
const selectAllBatchDates = useCallback(() => setBatchSelectedDates(new Set(batchVoiceDates)), [batchVoiceDates])
|
const selectAllBatchDates = useCallback(() => setBatchSelectedDates(new Set(batchVoiceDates)), [batchVoiceDates])
|
||||||
const clearAllBatchDates = useCallback(() => setBatchSelectedDates(new Set()), [])
|
const clearAllBatchDates = useCallback(() => setBatchSelectedDates(new Set()), [])
|
||||||
|
|
||||||
|
const lastSelectedIdRef = useRef<number | null>(null)
|
||||||
|
|
||||||
|
const handleToggleSelection = useCallback((localId: number, isShiftKey: boolean = false) => {
|
||||||
|
setSelectedMessages(prev => {
|
||||||
|
const next = new Set(prev)
|
||||||
|
|
||||||
|
// Range selection with Shift key
|
||||||
|
if (isShiftKey && lastSelectedIdRef.current !== null && lastSelectedIdRef.current !== localId) {
|
||||||
|
const currentMsgs = useChatStore.getState().messages
|
||||||
|
const idx1 = currentMsgs.findIndex(m => m.localId === lastSelectedIdRef.current)
|
||||||
|
const idx2 = currentMsgs.findIndex(m => m.localId === localId)
|
||||||
|
|
||||||
|
if (idx1 !== -1 && idx2 !== -1) {
|
||||||
|
const start = Math.min(idx1, idx2)
|
||||||
|
const end = Math.max(idx1, idx2)
|
||||||
|
for (let i = start; i <= end; i++) {
|
||||||
|
next.add(currentMsgs[i].localId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Normal toggle
|
||||||
|
if (next.has(localId)) {
|
||||||
|
next.delete(localId)
|
||||||
|
lastSelectedIdRef.current = null // Reset last selection on uncheck? Or keep? Usually keep last interaction.
|
||||||
|
} else {
|
||||||
|
next.add(localId)
|
||||||
|
lastSelectedIdRef.current = localId
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}, [])
|
||||||
|
|
||||||
const formatBatchDateLabel = useCallback((dateStr: string) => {
|
const formatBatchDateLabel = useCallback((dateStr: string) => {
|
||||||
const [y, m, d] = dateStr.split('-').map(Number)
|
const [y, m, d] = dateStr.split('-').map(Number)
|
||||||
return `${y}年${m}月${d}日`
|
return `${y}年${m}月${d}日`
|
||||||
}, [])
|
}, [])
|
||||||
|
|
||||||
|
// 消息右键菜单处理
|
||||||
|
const handleContextMenu = useCallback((e: React.MouseEvent, message: Message) => {
|
||||||
|
e.preventDefault()
|
||||||
|
setContextMenu({
|
||||||
|
x: e.clientX,
|
||||||
|
y: e.clientY,
|
||||||
|
message
|
||||||
|
})
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
// 关闭右键菜单
|
||||||
|
useEffect(() => {
|
||||||
|
const handleClick = () => {
|
||||||
|
setContextMenu(null)
|
||||||
|
}
|
||||||
|
window.addEventListener('click', handleClick)
|
||||||
|
return () => {
|
||||||
|
window.removeEventListener('click', handleClick)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
// 删除消息 - 触发确认弹窗
|
||||||
|
const handleDelete = useCallback((target: { message: Message } | null = null) => {
|
||||||
|
const msg = target?.message || contextMenu?.message
|
||||||
|
if (!currentSessionId || !msg) return
|
||||||
|
|
||||||
|
setDeleteConfirm({
|
||||||
|
show: true,
|
||||||
|
mode: 'single',
|
||||||
|
message: msg
|
||||||
|
})
|
||||||
|
setContextMenu(null)
|
||||||
|
}, [contextMenu, currentSessionId])
|
||||||
|
|
||||||
|
// 执行单条删除动作
|
||||||
|
const performSingleDelete = async (msg: Message) => {
|
||||||
|
try {
|
||||||
|
const dbPathHint = (msg as any)._db_path
|
||||||
|
const result = await (window as any).electronAPI.chat.deleteMessage(currentSessionId, msg.localId, msg.createTime, dbPathHint)
|
||||||
|
if (result.success) {
|
||||||
|
const currentMessages = useChatStore.getState().messages
|
||||||
|
const newMessages = currentMessages.filter(m => m.localId !== msg.localId)
|
||||||
|
useChatStore.getState().setMessages(newMessages)
|
||||||
|
} else {
|
||||||
|
alert('删除失败: ' + (result.error || '原因未知'))
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(e)
|
||||||
|
alert('删除异常: ' + String(e))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 修改消息
|
||||||
|
const handleEditMessage = useCallback(() => {
|
||||||
|
if (contextMenu) {
|
||||||
|
// 允许编辑所有类型的消息
|
||||||
|
// 如果是文本消息(1),使用 parsedContent
|
||||||
|
// 如果是其他类型(如系统消息 10000),使用 rawContent 或 content 作为 XML 源码编辑
|
||||||
|
const isText = contextMenu.message.localType === 1
|
||||||
|
const rawXml = contextMenu.message.content || (contextMenu.message as any).rawContent || contextMenu.message.parsedContent || ''
|
||||||
|
|
||||||
|
const contentToEdit = isText
|
||||||
|
? cleanMessageContent(contextMenu.message.parsedContent)
|
||||||
|
: rawXml
|
||||||
|
|
||||||
|
if (!isText) {
|
||||||
|
const fields = parseXmlToFields(rawXml)
|
||||||
|
setTempFields(fields)
|
||||||
|
setEditMode(fields.length > 0 ? 'fields' : 'raw')
|
||||||
|
} else {
|
||||||
|
setEditMode('raw')
|
||||||
|
setTempFields([])
|
||||||
|
}
|
||||||
|
|
||||||
|
setEditingMessage({
|
||||||
|
message: contextMenu.message,
|
||||||
|
content: contentToEdit
|
||||||
|
})
|
||||||
|
setContextMenu(null)
|
||||||
|
}
|
||||||
|
}, [contextMenu])
|
||||||
|
|
||||||
|
// 确认修改消息
|
||||||
|
const handleSaveEdit = useCallback(async () => {
|
||||||
|
if (editingMessage && currentSessionId) {
|
||||||
|
let finalContent = editingMessage.content
|
||||||
|
|
||||||
|
// 如果是字段编辑模式,先同步回 XML
|
||||||
|
if (editMode === 'fields' && tempFields.length > 0) {
|
||||||
|
finalContent = updateXmlWithFields(editingMessage.content, tempFields)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!finalContent.trim()) {
|
||||||
|
handleDelete({ message: editingMessage.message })
|
||||||
|
setEditingMessage(null)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await (window as any).electronAPI.chat.updateMessage(currentSessionId, editingMessage.message.localId, editingMessage.message.createTime, finalContent)
|
||||||
|
if (result.success) {
|
||||||
|
const currentMessages = useChatStore.getState().messages
|
||||||
|
const newMessages = currentMessages.map(m => {
|
||||||
|
if (m.localId === editingMessage.message.localId) {
|
||||||
|
return { ...m, parsedContent: finalContent, content: finalContent, rawContent: finalContent }
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
})
|
||||||
|
useChatStore.getState().setMessages(newMessages)
|
||||||
|
setEditingMessage(null)
|
||||||
|
} else {
|
||||||
|
alert('修改失败: ' + result.error)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
alert('修改异常: ' + String(e))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}, [editingMessage, currentSessionId, editMode, tempFields, handleDelete])
|
||||||
|
|
||||||
|
// 用于在异步循环中获取最新的取消状态
|
||||||
|
const cancelDeleteRef = useRef(false)
|
||||||
|
|
||||||
|
const handleBatchDelete = () => {
|
||||||
|
if (selectedMessages.size === 0) {
|
||||||
|
alert('请先选择要删除的消息')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if (!currentSessionId) return
|
||||||
|
|
||||||
|
setDeleteConfirm({
|
||||||
|
show: true,
|
||||||
|
mode: 'batch',
|
||||||
|
count: selectedMessages.size
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const performBatchDelete = async () => {
|
||||||
|
setIsDeleting(true)
|
||||||
|
setDeleteProgress({ current: 0, total: selectedMessages.size })
|
||||||
|
setCancelDeleteRequested(false)
|
||||||
|
cancelDeleteRef.current = false
|
||||||
|
|
||||||
|
try {
|
||||||
|
const currentMessages = useChatStore.getState().messages
|
||||||
|
const selectedIds = Array.from(selectedMessages)
|
||||||
|
const deletedIds = new Set<number>()
|
||||||
|
|
||||||
|
for (let i = 0; i < selectedIds.length; i++) {
|
||||||
|
if (cancelDeleteRef.current) break
|
||||||
|
|
||||||
|
const id = selectedIds[i]
|
||||||
|
const msgObj = currentMessages.find(m => m.localId === id)
|
||||||
|
const dbPathHint = (msgObj as any)?._db_path
|
||||||
|
const createTime = msgObj?.createTime || 0
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await (window as any).electronAPI.chat.deleteMessage(currentSessionId, id, createTime, dbPathHint)
|
||||||
|
if (result.success) {
|
||||||
|
deletedIds.add(id)
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error(`删除消息 ${id} 失败:`, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
setDeleteProgress({ current: i + 1, total: selectedIds.length })
|
||||||
|
}
|
||||||
|
|
||||||
|
const finalMessages = useChatStore.getState().messages.filter(m => !deletedIds.has(m.localId))
|
||||||
|
useChatStore.getState().setMessages(finalMessages)
|
||||||
|
|
||||||
|
setIsSelectionMode(false)
|
||||||
|
setSelectedMessages(new Set())
|
||||||
|
|
||||||
|
if (cancelDeleteRef.current) {
|
||||||
|
alert(`操作已中止。已删除 ${deletedIds.size} 条,剩余记录保留。`)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
alert('批量删除出现错误: ' + String(e))
|
||||||
|
console.error(e)
|
||||||
|
} finally {
|
||||||
|
setIsDeleting(false)
|
||||||
|
setCancelDeleteRequested(false)
|
||||||
|
cancelDeleteRef.current = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className={`chat-page ${isResizing ? 'resizing' : ''}`}>
|
<div className={`chat-page ${isResizing ? 'resizing' : ''}`}>
|
||||||
|
{/* 自定义删除确认对话框 */}
|
||||||
|
{deleteConfirm.show && (
|
||||||
|
<div className="delete-confirm-overlay">
|
||||||
|
<div className="delete-confirm-card">
|
||||||
|
<div className="confirm-icon">
|
||||||
|
<Trash2 size={32} color="var(--danger)" />
|
||||||
|
</div>
|
||||||
|
<div className="confirm-content">
|
||||||
|
<h3>确认删除</h3>
|
||||||
|
<p>
|
||||||
|
{deleteConfirm.mode === 'single'
|
||||||
|
? '确定要删除这条消息吗?此操作不可恢复。'
|
||||||
|
: `确定要删除选中的 ${deleteConfirm.count} 条消息吗?`}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
<div className="confirm-actions">
|
||||||
|
<button
|
||||||
|
className="btn-secondary"
|
||||||
|
onClick={() => setDeleteConfirm({ ...deleteConfirm, show: false })}
|
||||||
|
>
|
||||||
|
取消
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
className="btn-danger-filled"
|
||||||
|
onClick={() => {
|
||||||
|
setDeleteConfirm({ ...deleteConfirm, show: false });
|
||||||
|
if (deleteConfirm.mode === 'single' && deleteConfirm.message) {
|
||||||
|
performSingleDelete(deleteConfirm.message);
|
||||||
|
} else if (deleteConfirm.mode === 'batch') {
|
||||||
|
performBatchDelete();
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
确定删除
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 批量删除进度遮罩 */}
|
||||||
|
{isDeleting && (
|
||||||
|
<div className="delete-progress-overlay">
|
||||||
|
<div className="delete-progress-card">
|
||||||
|
<div className="progress-header">
|
||||||
|
<h3>正在彻底删除消息...</h3>
|
||||||
|
<span className="count">{deleteProgress.current} / {deleteProgress.total}</span>
|
||||||
|
</div>
|
||||||
|
<div className="progress-bar-container">
|
||||||
|
<div
|
||||||
|
className="progress-bar-fill"
|
||||||
|
style={{ width: `${(deleteProgress.current / deleteProgress.total) * 100}%` }}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div className="progress-footer">
|
||||||
|
<p>请勿关闭应用或切换会话,确保所有副本都被清理。</p>
|
||||||
|
<button
|
||||||
|
className="cancel-delete-btn"
|
||||||
|
onClick={() => {
|
||||||
|
setCancelDeleteRequested(true)
|
||||||
|
cancelDeleteRef.current = true
|
||||||
|
}}
|
||||||
|
disabled={cancelDeleteRequested}
|
||||||
|
>
|
||||||
|
{cancelDeleteRequested ? '正在停止...' : '中止删除'}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
{/* 左侧会话列表 */}
|
{/* 左侧会话列表 */}
|
||||||
<div
|
<div
|
||||||
className="session-sidebar"
|
className="session-sidebar"
|
||||||
@@ -1632,6 +2083,13 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{!isLoadingMessages && messages.length === 0 && !hasMoreMessages && (
|
||||||
|
<div className="empty-chat-inline">
|
||||||
|
<MessageSquare size={32} />
|
||||||
|
<span>该联系人没有聊天记录</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{messages.map((msg, index) => {
|
{messages.map((msg, index) => {
|
||||||
const prevMsg = index > 0 ? messages[index - 1] : undefined
|
const prevMsg = index > 0 ? messages[index - 1] : undefined
|
||||||
const showDateDivider = shouldShowDateDivider(msg, prevMsg)
|
const showDateDivider = shouldShowDateDivider(msg, prevMsg)
|
||||||
@@ -1659,6 +2117,10 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
myAvatarUrl={myAvatarUrl}
|
myAvatarUrl={myAvatarUrl}
|
||||||
isGroupChat={isGroupChat(currentSession.username)}
|
isGroupChat={isGroupChat(currentSession.username)}
|
||||||
onRequireModelDownload={handleRequireModelDownload}
|
onRequireModelDownload={handleRequireModelDownload}
|
||||||
|
onContextMenu={handleContextMenu}
|
||||||
|
isSelectionMode={isSelectionMode}
|
||||||
|
isSelected={selectedMessages.has(msg.localId)}
|
||||||
|
onToggleSelection={handleToggleSelection}
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
@@ -1897,6 +2359,187 @@ function ChatPage(_props: ChatPageProps) {
|
|||||||
</div>,
|
</div>,
|
||||||
document.body
|
document.body
|
||||||
)}
|
)}
|
||||||
|
{/* 消息右键菜单 */}
|
||||||
|
{contextMenu && createPortal(
|
||||||
|
<>
|
||||||
|
<div className="context-menu-overlay" onClick={() => setContextMenu(null)}
|
||||||
|
style={{ position: 'fixed', top: 0, left: 0, right: 0, bottom: 0, zIndex: 9998 }} />
|
||||||
|
<div
|
||||||
|
className="context-menu"
|
||||||
|
style={{
|
||||||
|
position: 'fixed',
|
||||||
|
top: contextMenu.y,
|
||||||
|
left: contextMenu.x,
|
||||||
|
zIndex: 9999
|
||||||
|
}}
|
||||||
|
onClick={(e) => e.stopPropagation()}
|
||||||
|
>
|
||||||
|
<div className="menu-item" onClick={handleEditMessage}>
|
||||||
|
<Edit2 size={16} />
|
||||||
|
<span>{contextMenu.message.localType === 1 ? '修改消息' : '编辑源码'}</span>
|
||||||
|
</div>
|
||||||
|
<div className="menu-item" onClick={() => {
|
||||||
|
setIsSelectionMode(true)
|
||||||
|
setSelectedMessages(new Set([contextMenu.message.localId]))
|
||||||
|
setContextMenu(null)
|
||||||
|
}}>
|
||||||
|
<CheckSquare size={16} />
|
||||||
|
<span>多选</span>
|
||||||
|
</div>
|
||||||
|
<div className="menu-item delete" onClick={(e) => { e.stopPropagation(); handleDelete() }}>
|
||||||
|
<Trash2 size={16} />
|
||||||
|
<span>删除消息</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</>,
|
||||||
|
document.body
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 修改消息弹窗 */}
|
||||||
|
{editingMessage && createPortal(
|
||||||
|
<div className="modal-overlay">
|
||||||
|
<div className="modal-content edit-message-modal">
|
||||||
|
<div className="modal-header">
|
||||||
|
<h3 style={{ margin: 0 }}>{editingMessage.message.localType === 1 ? '修改消息' : '编辑消息'}</h3>
|
||||||
|
<button className="close-btn" onClick={() => setEditingMessage(null)}>
|
||||||
|
<X size={16} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div style={{ flex: 1, overflowY: 'auto', display: 'flex', flexDirection: 'column' }}>
|
||||||
|
{editMode === 'raw' ? (
|
||||||
|
<textarea
|
||||||
|
className="edit-message-textarea"
|
||||||
|
style={{ fontFamily: 'inherit', width: '100%', boxSizing: 'border-box' }}
|
||||||
|
value={editingMessage.content}
|
||||||
|
onChange={(e) => setEditingMessage({ ...editingMessage, content: e.target.value })}
|
||||||
|
rows={editingMessage.message.localType === 1 ? 8 : 15}
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<div style={{ padding: '16px 20px', display: 'flex', flexDirection: 'column', gap: '14px' }}>
|
||||||
|
{tempFields.map((field, idx) => (
|
||||||
|
<div key={idx} style={{ display: 'flex', flexDirection: 'column', gap: '6px' }}>
|
||||||
|
<div style={{ display: 'flex', alignItems: 'center', justifyContent: 'space-between' }}>
|
||||||
|
<span style={{ fontSize: '12px', color: 'var(--text-secondary)', fontWeight: 500 }}>
|
||||||
|
{field.tagName ? field.tagName : '节点'}: <span style={{ color: 'var(--primary)' }}>{field.key}</span>
|
||||||
|
</span>
|
||||||
|
<span style={{ fontSize: '10px', color: 'var(--text-tertiary)', opacity: 0.6 }}>
|
||||||
|
{field.type === 'attr' ? '属性' : '文本内容'}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
value={field.value}
|
||||||
|
onChange={(e) => {
|
||||||
|
const newFields = [...tempFields]
|
||||||
|
newFields[idx].value = e.target.value
|
||||||
|
setTempFields(newFields)
|
||||||
|
}}
|
||||||
|
style={{
|
||||||
|
background: 'var(--bg-tertiary)',
|
||||||
|
border: '1px solid var(--border-color)',
|
||||||
|
borderRadius: '6px',
|
||||||
|
padding: '10px 12px',
|
||||||
|
color: 'var(--text-primary)',
|
||||||
|
fontSize: '13px',
|
||||||
|
outline: 'none',
|
||||||
|
width: '100%',
|
||||||
|
boxSizing: 'border-box'
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="modal-actions" style={{ justifyContent: 'space-between' }}>
|
||||||
|
<div>
|
||||||
|
{editingMessage.message.localType !== 1 && tempFields.length > 0 && (
|
||||||
|
<button
|
||||||
|
onClick={() => setEditMode(editMode === 'raw' ? 'fields' : 'raw')}
|
||||||
|
style={{
|
||||||
|
padding: '6px 12px',
|
||||||
|
fontSize: '12px',
|
||||||
|
borderRadius: '6px',
|
||||||
|
border: '1px solid var(--border-color)',
|
||||||
|
background: editMode === 'fields' ? 'var(--primary)' : 'transparent',
|
||||||
|
color: editMode === 'fields' ? '#fff' : 'var(--text-secondary)',
|
||||||
|
cursor: 'pointer',
|
||||||
|
transition: 'all 0.2s',
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{editMode === 'raw' ? '可视化编辑' : '源码编辑'}
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div style={{ display: 'flex', gap: '12px' }}>
|
||||||
|
<button className="btn-secondary" onClick={() => setEditingMessage(null)}>取消</button>
|
||||||
|
<button className="btn-primary" onClick={handleSaveEdit}>保存</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>,
|
||||||
|
document.body
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 底部多选操作栏 */}
|
||||||
|
{isSelectionMode && (
|
||||||
|
<div style={{
|
||||||
|
position: 'absolute',
|
||||||
|
bottom: 24,
|
||||||
|
left: '50%',
|
||||||
|
transform: 'translateX(-50%)',
|
||||||
|
backgroundColor: 'var(--bg-secondary)', // Use system background
|
||||||
|
color: 'var(--text-primary)',
|
||||||
|
boxShadow: '0 8px 24px rgba(0,0,0,0.2)',
|
||||||
|
borderRadius: '12px',
|
||||||
|
padding: '12px 24px',
|
||||||
|
display: 'flex',
|
||||||
|
gap: '20px',
|
||||||
|
zIndex: 1000,
|
||||||
|
alignItems: 'center',
|
||||||
|
border: '1px solid var(--border-color)', // Subtle border
|
||||||
|
backdropFilter: 'blur(10px)'
|
||||||
|
}}>
|
||||||
|
<span style={{ fontSize: '14px', fontWeight: 500 }}>已选 {selectedMessages.size} 条</span>
|
||||||
|
<div style={{ width: '1px', height: '16px', background: 'var(--border-color)' }}></div>
|
||||||
|
<button
|
||||||
|
className="btn-danger"
|
||||||
|
onClick={handleBatchDelete}
|
||||||
|
style={{
|
||||||
|
padding: '6px 16px',
|
||||||
|
borderRadius: '6px',
|
||||||
|
border: 'none',
|
||||||
|
backgroundColor: '#fa5151',
|
||||||
|
color: 'white',
|
||||||
|
cursor: 'pointer',
|
||||||
|
fontSize: '13px',
|
||||||
|
fontWeight: 500
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
删除
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
className="btn-secondary"
|
||||||
|
onClick={() => {
|
||||||
|
setIsSelectionMode(false)
|
||||||
|
setSelectedMessages(new Set())
|
||||||
|
}}
|
||||||
|
style={{
|
||||||
|
padding: '6px 16px',
|
||||||
|
borderRadius: '6px',
|
||||||
|
border: '1px solid var(--border-color)',
|
||||||
|
backgroundColor: 'transparent',
|
||||||
|
color: 'var(--text-primary)',
|
||||||
|
cursor: 'pointer',
|
||||||
|
fontSize: '13px'
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
取消
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -1932,13 +2575,28 @@ const senderAvatarCache = new Map<string, { avatarUrl?: string; displayName?: st
|
|||||||
const senderAvatarLoading = new Map<string, Promise<{ avatarUrl?: string; displayName?: string } | null>>()
|
const senderAvatarLoading = new Map<string, Promise<{ avatarUrl?: string; displayName?: string } | null>>()
|
||||||
|
|
||||||
// 消息气泡组件
|
// 消息气泡组件
|
||||||
function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, onRequireModelDownload }: {
|
function MessageBubble({
|
||||||
|
message,
|
||||||
|
session,
|
||||||
|
showTime,
|
||||||
|
myAvatarUrl,
|
||||||
|
isGroupChat,
|
||||||
|
onRequireModelDownload,
|
||||||
|
onContextMenu,
|
||||||
|
isSelectionMode,
|
||||||
|
isSelected,
|
||||||
|
onToggleSelection
|
||||||
|
}: {
|
||||||
message: Message;
|
message: Message;
|
||||||
session: ChatSession;
|
session: ChatSession;
|
||||||
showTime?: boolean;
|
showTime?: boolean;
|
||||||
myAvatarUrl?: string;
|
myAvatarUrl?: string;
|
||||||
isGroupChat?: boolean;
|
isGroupChat?: boolean;
|
||||||
onRequireModelDownload?: (sessionId: string, messageId: string) => void;
|
onRequireModelDownload?: (sessionId: string, messageId: string) => void;
|
||||||
|
onContextMenu?: (e: React.MouseEvent, message: Message) => void;
|
||||||
|
isSelectionMode?: boolean;
|
||||||
|
isSelected?: boolean;
|
||||||
|
onToggleSelection?: (localId: number, isShiftKey?: boolean) => void;
|
||||||
}) {
|
}) {
|
||||||
const isSystem = isSystemMessage(message.localType)
|
const isSystem = isSystemMessage(message.localType)
|
||||||
const isEmoji = message.localType === 47
|
const isEmoji = message.localType === 47
|
||||||
@@ -1953,6 +2611,8 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, o
|
|||||||
const [senderName, setSenderName] = useState<string | undefined>(undefined)
|
const [senderName, setSenderName] = useState<string | undefined>(undefined)
|
||||||
const [emojiError, setEmojiError] = useState(false)
|
const [emojiError, setEmojiError] = useState(false)
|
||||||
const [emojiLoading, setEmojiLoading] = useState(false)
|
const [emojiLoading, setEmojiLoading] = useState(false)
|
||||||
|
|
||||||
|
// State variables...
|
||||||
const [imageError, setImageError] = useState(false)
|
const [imageError, setImageError] = useState(false)
|
||||||
const [imageLoading, setImageLoading] = useState(false)
|
const [imageLoading, setImageLoading] = useState(false)
|
||||||
const [imageHasUpdate, setImageHasUpdate] = useState(false)
|
const [imageHasUpdate, setImageHasUpdate] = useState(false)
|
||||||
@@ -2643,9 +3303,39 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, o
|
|||||||
void requestVoiceTranscript()
|
void requestVoiceTranscript()
|
||||||
}, [autoTranscribeEnabled, isVoice, voiceDataUrl, voiceTranscript, voiceTranscriptError, voiceTranscriptLoading, requestVoiceTranscript])
|
}, [autoTranscribeEnabled, isVoice, voiceDataUrl, voiceTranscript, voiceTranscriptError, voiceTranscriptLoading, requestVoiceTranscript])
|
||||||
|
|
||||||
|
// Selection mode handling removed from here to allow normal rendering
|
||||||
|
// We will wrap the output instead
|
||||||
|
|
||||||
|
// Regular rendering logic...
|
||||||
if (isSystem) {
|
if (isSystem) {
|
||||||
return (
|
return (
|
||||||
<div className="message-bubble system">
|
<div
|
||||||
|
className={`message-bubble system ${isSelectionMode ? 'selectable' : ''}`}
|
||||||
|
onContextMenu={(e) => onContextMenu?.(e, message)}
|
||||||
|
style={{ cursor: isSelectionMode ? 'pointer' : 'default', display: 'flex', alignItems: 'center', justifyContent: 'center', gap: '8px' }}
|
||||||
|
onClick={(e) => {
|
||||||
|
if (isSelectionMode) {
|
||||||
|
e.stopPropagation()
|
||||||
|
onToggleSelection?.(message.localId, e.shiftKey)
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{isSelectionMode && (
|
||||||
|
<div className={`checkbox ${isSelected ? 'checked' : ''}`} style={{
|
||||||
|
width: '20px',
|
||||||
|
height: '20px',
|
||||||
|
borderRadius: '4px',
|
||||||
|
border: isSelected ? 'none' : '2px solid rgba(128,128,128,0.5)',
|
||||||
|
backgroundColor: isSelected ? 'var(--primary)' : 'transparent',
|
||||||
|
display: 'flex',
|
||||||
|
alignItems: 'center',
|
||||||
|
justifyContent: 'center',
|
||||||
|
color: 'white',
|
||||||
|
flexShrink: 0
|
||||||
|
}}>
|
||||||
|
{isSelected && <Check size={14} strokeWidth={3} />}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
<div className="bubble-content">{message.parsedContent}</div>
|
<div className="bubble-content">{message.parsedContent}</div>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
@@ -3344,16 +4034,50 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, o
|
|||||||
<span>{formatTime(message.createTime)}</span>
|
<span>{formatTime(message.createTime)}</span>
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
<div className={`message-bubble ${bubbleClass} ${isEmoji && message.emojiCdnUrl && !emojiError ? 'emoji' : ''} ${isImage ? 'image' : ''} ${isVoice ? 'voice' : ''}`}>
|
<div
|
||||||
|
className={`message-wrapper-with-selection ${isSelectionMode ? 'selectable' : ''}`}
|
||||||
|
style={{
|
||||||
|
display: 'flex',
|
||||||
|
alignItems: 'flex-start',
|
||||||
|
width: '100%',
|
||||||
|
justifyContent: isSent ? 'flex-end' : 'flex-start',
|
||||||
|
cursor: isSelectionMode ? 'pointer' : 'default'
|
||||||
|
}}
|
||||||
|
onClick={(e) => {
|
||||||
|
if (isSelectionMode) {
|
||||||
|
e.stopPropagation()
|
||||||
|
onToggleSelection?.(message.localId, e.shiftKey)
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{isSelectionMode && !isSent && (
|
||||||
|
<div className={`checkbox ${isSelected ? 'checked' : ''}`} style={{
|
||||||
|
width: '20px',
|
||||||
|
height: '20px',
|
||||||
|
borderRadius: '4px',
|
||||||
|
border: isSelected ? 'none' : '2px solid rgba(128,128,128,0.5)',
|
||||||
|
backgroundColor: isSelected ? 'var(--primary)' : 'transparent',
|
||||||
|
display: 'flex',
|
||||||
|
alignItems: 'center',
|
||||||
|
justifyContent: 'center',
|
||||||
|
color: 'white',
|
||||||
|
marginRight: '12px',
|
||||||
|
marginTop: '10px', // Align with avatar top
|
||||||
|
flexShrink: 0
|
||||||
|
}}>
|
||||||
|
{isSelected && <Check size={14} strokeWidth={3} />}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className={`message-bubble ${bubbleClass} ${isEmoji && message.emojiCdnUrl && !emojiError ? 'emoji' : ''} ${isImage ? 'image' : ''} ${isVoice ? 'voice' : ''}`}
|
||||||
|
onContextMenu={(e) => onContextMenu?.(e, message)}
|
||||||
|
>
|
||||||
<div className="bubble-avatar">
|
<div className="bubble-avatar">
|
||||||
<Avatar
|
<Avatar
|
||||||
src={avatarUrl}
|
src={avatarUrl}
|
||||||
name={!isSent ? (isGroupChat ? (senderName || message.senderUsername || '?') : (session.displayName || session.username)) : '我'}
|
name={!isSent ? (isGroupChat ? (senderName || message.senderUsername || '?') : (session.displayName || session.username)) : '我'}
|
||||||
size={36}
|
size={36}
|
||||||
className="bubble-avatar"
|
className="bubble-avatar"
|
||||||
// If it's sent by me (isSent), we might not want 'group' class even if it's a group chat.
|
|
||||||
// But 'group' class mainly handles default avatar icon.
|
|
||||||
// Let's rely on standard Avatar behavior.
|
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
<div className="bubble-body">
|
<div className="bubble-body">
|
||||||
@@ -3366,6 +4090,26 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, o
|
|||||||
{renderContent()}
|
{renderContent()}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{isSelectionMode && isSent && (
|
||||||
|
<div className={`checkbox ${isSelected ? 'checked' : ''}`} style={{
|
||||||
|
width: '20px',
|
||||||
|
height: '20px',
|
||||||
|
borderRadius: '4px',
|
||||||
|
border: isSelected ? 'none' : '2px solid rgba(128,128,128,0.5)',
|
||||||
|
backgroundColor: isSelected ? 'var(--primary)' : 'transparent',
|
||||||
|
display: 'flex',
|
||||||
|
alignItems: 'center',
|
||||||
|
justifyContent: 'center',
|
||||||
|
color: 'white',
|
||||||
|
marginLeft: '12px',
|
||||||
|
marginTop: '10px',
|
||||||
|
flexShrink: 0
|
||||||
|
}}>
|
||||||
|
{isSelected && <Check size={14} strokeWidth={3} />}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
</>
|
</>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,8 +7,8 @@
|
|||||||
|
|
||||||
// 左侧联系人面板
|
// 左侧联系人面板
|
||||||
.contacts-panel {
|
.contacts-panel {
|
||||||
width: 380px;
|
width: 350px;
|
||||||
min-width: 380px;
|
min-width: 350px;
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
border-right: 1px solid var(--border-color);
|
border-right: 1px solid var(--border-color);
|
||||||
@@ -55,6 +55,11 @@
|
|||||||
.spin {
|
.spin {
|
||||||
animation: contactsSpin 1s linear infinite;
|
animation: contactsSpin 1s linear infinite;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
&.export-mode-btn.active {
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,11 +115,11 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
.type-filters {
|
.type-filters {
|
||||||
display: flex;
|
display: grid;
|
||||||
|
grid-template-columns: 1fr 1fr;
|
||||||
gap: 8px;
|
gap: 8px;
|
||||||
padding: 0 20px 16px;
|
padding: 0 20px 16px;
|
||||||
flex-wrap: nowrap;
|
max-width: 300px;
|
||||||
overflow-x: auto;
|
|
||||||
|
|
||||||
&::-webkit-scrollbar {
|
&::-webkit-scrollbar {
|
||||||
display: none;
|
display: none;
|
||||||
@@ -174,6 +179,24 @@
|
|||||||
color: var(--text-secondary);
|
color: var(--text-secondary);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.selection-toolbar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 0 20px 12px;
|
||||||
|
|
||||||
|
.checkbox-item {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.selection-count {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.loading-state,
|
.loading-state,
|
||||||
.empty-state {
|
.empty-state {
|
||||||
flex: 1;
|
flex: 1;
|
||||||
@@ -213,12 +236,35 @@
|
|||||||
padding: 12px;
|
padding: 12px;
|
||||||
border-radius: 10px;
|
border-radius: 10px;
|
||||||
transition: all 0.2s;
|
transition: all 0.2s;
|
||||||
|
cursor: pointer;
|
||||||
margin-bottom: 4px;
|
margin-bottom: 4px;
|
||||||
|
|
||||||
&:hover {
|
&:hover {
|
||||||
background: var(--bg-hover);
|
background: var(--bg-hover);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
&.selected {
|
||||||
|
background: color-mix(in srgb, var(--primary) 12%, transparent);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-select {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
flex-shrink: 0;
|
||||||
|
|
||||||
|
input[type="checkbox"] {
|
||||||
|
width: 16px;
|
||||||
|
height: 16px;
|
||||||
|
cursor: pointer;
|
||||||
|
accent-color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.contact-avatar {
|
.contact-avatar {
|
||||||
width: 44px;
|
width: 44px;
|
||||||
height: 44px;
|
height: 44px;
|
||||||
@@ -297,6 +343,94 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 右侧详情面板内的联系人资料
|
||||||
|
.detail-profile {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding-bottom: 24px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
margin-bottom: 20px;
|
||||||
|
|
||||||
|
.detail-avatar {
|
||||||
|
width: 80px;
|
||||||
|
height: 80px;
|
||||||
|
border-radius: 16px;
|
||||||
|
background: linear-gradient(135deg, var(--primary), var(--primary-hover));
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
overflow: hidden;
|
||||||
|
|
||||||
|
img { width: 100%; height: 100%; object-fit: cover; }
|
||||||
|
span { color: #fff; font-size: 28px; font-weight: 600; }
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-name {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-info-list {
|
||||||
|
margin-bottom: 24px;
|
||||||
|
|
||||||
|
.detail-row {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 10px 0;
|
||||||
|
font-size: 13px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
|
&:last-child { border-bottom: none; }
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-label {
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
min-width: 48px;
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-value {
|
||||||
|
color: var(--text-primary);
|
||||||
|
word-break: break-all;
|
||||||
|
user-select: text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.goto-chat-btn {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 12px;
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
border: none;
|
||||||
|
border-radius: 10px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover { background: var(--primary-hover); }
|
||||||
|
}
|
||||||
|
|
||||||
|
.empty-detail {
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
|
||||||
// 右侧设置面板
|
// 右侧设置面板
|
||||||
.settings-panel {
|
.settings-panel {
|
||||||
flex: 1;
|
flex: 1;
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
import { useState, useEffect, useCallback, useRef } from 'react'
|
import { useState, useEffect, useCallback, useRef } from 'react'
|
||||||
import { Search, RefreshCw, X, User, Users, MessageSquare, Loader2, FolderOpen, Download, ChevronDown } from 'lucide-react'
|
import { useNavigate } from 'react-router-dom'
|
||||||
|
import { Search, RefreshCw, X, User, Users, MessageSquare, Loader2, FolderOpen, Download, ChevronDown, MessageCircle, UserX } from 'lucide-react'
|
||||||
|
import { useChatStore } from '../stores/chatStore'
|
||||||
import './ContactsPage.scss'
|
import './ContactsPage.scss'
|
||||||
|
|
||||||
interface ContactInfo {
|
interface ContactInfo {
|
||||||
@@ -8,20 +10,28 @@ interface ContactInfo {
|
|||||||
remark?: string
|
remark?: string
|
||||||
nickname?: string
|
nickname?: string
|
||||||
avatarUrl?: string
|
avatarUrl?: string
|
||||||
type: 'friend' | 'group' | 'official' | 'other'
|
type: 'friend' | 'group' | 'official' | 'former_friend' | 'other'
|
||||||
}
|
}
|
||||||
|
|
||||||
function ContactsPage() {
|
function ContactsPage() {
|
||||||
const [contacts, setContacts] = useState<ContactInfo[]>([])
|
const [contacts, setContacts] = useState<ContactInfo[]>([])
|
||||||
const [filteredContacts, setFilteredContacts] = useState<ContactInfo[]>([])
|
const [filteredContacts, setFilteredContacts] = useState<ContactInfo[]>([])
|
||||||
|
const [selectedUsernames, setSelectedUsernames] = useState<Set<string>>(new Set())
|
||||||
const [isLoading, setIsLoading] = useState(true)
|
const [isLoading, setIsLoading] = useState(true)
|
||||||
const [searchKeyword, setSearchKeyword] = useState('')
|
const [searchKeyword, setSearchKeyword] = useState('')
|
||||||
const [contactTypes, setContactTypes] = useState({
|
const [contactTypes, setContactTypes] = useState({
|
||||||
friends: true,
|
friends: true,
|
||||||
groups: true,
|
groups: false,
|
||||||
officials: true
|
officials: false,
|
||||||
|
deletedFriends: false
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// 导出模式与查看详情
|
||||||
|
const [exportMode, setExportMode] = useState(false)
|
||||||
|
const [selectedContact, setSelectedContact] = useState<ContactInfo | null>(null)
|
||||||
|
const navigate = useNavigate()
|
||||||
|
const { setCurrentSession } = useChatStore()
|
||||||
|
|
||||||
// 导出相关状态
|
// 导出相关状态
|
||||||
const [exportFormat, setExportFormat] = useState<'json' | 'csv' | 'vcf'>('json')
|
const [exportFormat, setExportFormat] = useState<'json' | 'csv' | 'vcf'>('json')
|
||||||
const [exportAvatars, setExportAvatars] = useState(true)
|
const [exportAvatars, setExportAvatars] = useState(true)
|
||||||
@@ -62,6 +72,7 @@ function ContactsPage() {
|
|||||||
|
|
||||||
setContacts(contactsResult.contacts)
|
setContacts(contactsResult.contacts)
|
||||||
setFilteredContacts(contactsResult.contacts)
|
setFilteredContacts(contactsResult.contacts)
|
||||||
|
setSelectedUsernames(new Set())
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('加载通讯录失败:', e)
|
console.error('加载通讯录失败:', e)
|
||||||
@@ -83,6 +94,7 @@ function ContactsPage() {
|
|||||||
if (c.type === 'friend' && !contactTypes.friends) return false
|
if (c.type === 'friend' && !contactTypes.friends) return false
|
||||||
if (c.type === 'group' && !contactTypes.groups) return false
|
if (c.type === 'group' && !contactTypes.groups) return false
|
||||||
if (c.type === 'official' && !contactTypes.officials) return false
|
if (c.type === 'official' && !contactTypes.officials) return false
|
||||||
|
if (c.type === 'former_friend' && !contactTypes.deletedFriends) return false
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -111,6 +123,37 @@ function ContactsPage() {
|
|||||||
return () => document.removeEventListener('mousedown', handleClickOutside)
|
return () => document.removeEventListener('mousedown', handleClickOutside)
|
||||||
}, [showFormatSelect])
|
}, [showFormatSelect])
|
||||||
|
|
||||||
|
const selectedInFilteredCount = filteredContacts.reduce((count, contact) => {
|
||||||
|
return selectedUsernames.has(contact.username) ? count + 1 : count
|
||||||
|
}, 0)
|
||||||
|
const allFilteredSelected = filteredContacts.length > 0 && selectedInFilteredCount === filteredContacts.length
|
||||||
|
|
||||||
|
const toggleContactSelected = (username: string, checked: boolean) => {
|
||||||
|
setSelectedUsernames(prev => {
|
||||||
|
const next = new Set(prev)
|
||||||
|
if (checked) {
|
||||||
|
next.add(username)
|
||||||
|
} else {
|
||||||
|
next.delete(username)
|
||||||
|
}
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const toggleAllFilteredSelected = (checked: boolean) => {
|
||||||
|
setSelectedUsernames(prev => {
|
||||||
|
const next = new Set(prev)
|
||||||
|
filteredContacts.forEach(contact => {
|
||||||
|
if (checked) {
|
||||||
|
next.add(contact.username)
|
||||||
|
} else {
|
||||||
|
next.delete(contact.username)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
const getAvatarLetter = (name: string) => {
|
const getAvatarLetter = (name: string) => {
|
||||||
if (!name) return '?'
|
if (!name) return '?'
|
||||||
return [...name][0] || '?'
|
return [...name][0] || '?'
|
||||||
@@ -121,6 +164,7 @@ function ContactsPage() {
|
|||||||
case 'friend': return <User size={14} />
|
case 'friend': return <User size={14} />
|
||||||
case 'group': return <Users size={14} />
|
case 'group': return <Users size={14} />
|
||||||
case 'official': return <MessageSquare size={14} />
|
case 'official': return <MessageSquare size={14} />
|
||||||
|
case 'former_friend': return <UserX size={14} />
|
||||||
default: return <User size={14} />
|
default: return <User size={14} />
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -130,6 +174,7 @@ function ContactsPage() {
|
|||||||
case 'friend': return '好友'
|
case 'friend': return '好友'
|
||||||
case 'group': return '群聊'
|
case 'group': return '群聊'
|
||||||
case 'official': return '公众号'
|
case 'official': return '公众号'
|
||||||
|
case 'former_friend': return '曾经的好友'
|
||||||
default: return '其他'
|
default: return '其他'
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -154,6 +199,10 @@ function ContactsPage() {
|
|||||||
alert('请先选择导出位置')
|
alert('请先选择导出位置')
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
if (selectedUsernames.size === 0) {
|
||||||
|
alert('请至少选择一个联系人')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
setIsExporting(true)
|
setIsExporting(true)
|
||||||
try {
|
try {
|
||||||
@@ -164,7 +213,8 @@ function ContactsPage() {
|
|||||||
friends: contactTypes.friends,
|
friends: contactTypes.friends,
|
||||||
groups: contactTypes.groups,
|
groups: contactTypes.groups,
|
||||||
officials: contactTypes.officials
|
officials: contactTypes.officials
|
||||||
}
|
},
|
||||||
|
selectedUsernames: Array.from(selectedUsernames)
|
||||||
}
|
}
|
||||||
|
|
||||||
const result = await window.electronAPI.export.exportContacts(exportFolder, exportOptions)
|
const result = await window.electronAPI.export.exportContacts(exportFolder, exportOptions)
|
||||||
@@ -198,10 +248,19 @@ function ContactsPage() {
|
|||||||
<div className="contacts-panel">
|
<div className="contacts-panel">
|
||||||
<div className="panel-header">
|
<div className="panel-header">
|
||||||
<h2>通讯录</h2>
|
<h2>通讯录</h2>
|
||||||
|
<div style={{ display: 'flex', gap: 8, alignItems: 'center' }}>
|
||||||
|
<button
|
||||||
|
className={`icon-btn export-mode-btn ${exportMode ? 'active' : ''}`}
|
||||||
|
onClick={() => { setExportMode(!exportMode); setSelectedContact(null) }}
|
||||||
|
title={exportMode ? '退出导出模式' : '进入导出模式'}
|
||||||
|
>
|
||||||
|
<Download size={18} />
|
||||||
|
</button>
|
||||||
<button className="icon-btn" onClick={loadContacts} disabled={isLoading}>
|
<button className="icon-btn" onClick={loadContacts} disabled={isLoading}>
|
||||||
<RefreshCw size={18} className={isLoading ? 'spin' : ''} />
|
<RefreshCw size={18} className={isLoading ? 'spin' : ''} />
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
<div className="search-bar">
|
<div className="search-bar">
|
||||||
<Search size={16} />
|
<Search size={16} />
|
||||||
@@ -220,31 +279,20 @@ function ContactsPage() {
|
|||||||
|
|
||||||
<div className="type-filters">
|
<div className="type-filters">
|
||||||
<label className={`filter-chip ${contactTypes.friends ? 'active' : ''}`}>
|
<label className={`filter-chip ${contactTypes.friends ? 'active' : ''}`}>
|
||||||
<input
|
<input type="checkbox" checked={contactTypes.friends} onChange={e => setContactTypes({ ...contactTypes, friends: e.target.checked })} />
|
||||||
type="checkbox"
|
<User size={16} /><span>好友</span>
|
||||||
checked={contactTypes.friends}
|
|
||||||
onChange={e => setContactTypes({ ...contactTypes, friends: e.target.checked })}
|
|
||||||
/>
|
|
||||||
<User size={16} />
|
|
||||||
<span>好友</span>
|
|
||||||
</label>
|
</label>
|
||||||
<label className={`filter-chip ${contactTypes.groups ? 'active' : ''}`}>
|
<label className={`filter-chip ${contactTypes.groups ? 'active' : ''}`}>
|
||||||
<input
|
<input type="checkbox" checked={contactTypes.groups} onChange={e => setContactTypes({ ...contactTypes, groups: e.target.checked })} />
|
||||||
type="checkbox"
|
<Users size={16} /><span>群聊</span>
|
||||||
checked={contactTypes.groups}
|
|
||||||
onChange={e => setContactTypes({ ...contactTypes, groups: e.target.checked })}
|
|
||||||
/>
|
|
||||||
<Users size={16} />
|
|
||||||
<span>群聊</span>
|
|
||||||
</label>
|
</label>
|
||||||
<label className={`filter-chip ${contactTypes.officials ? 'active' : ''}`}>
|
<label className={`filter-chip ${contactTypes.officials ? 'active' : ''}`}>
|
||||||
<input
|
<input type="checkbox" checked={contactTypes.officials} onChange={e => setContactTypes({ ...contactTypes, officials: e.target.checked })} />
|
||||||
type="checkbox"
|
<MessageSquare size={16} /><span>公众号</span>
|
||||||
checked={contactTypes.officials}
|
</label>
|
||||||
onChange={e => setContactTypes({ ...contactTypes, officials: e.target.checked })}
|
<label className={`filter-chip ${contactTypes.deletedFriends ? 'active' : ''}`}>
|
||||||
/>
|
<input type="checkbox" checked={contactTypes.deletedFriends} onChange={e => setContactTypes({ ...contactTypes, deletedFriends: e.target.checked })} />
|
||||||
<MessageSquare size={16} />
|
<UserX size={16} /><span>曾经的好友</span>
|
||||||
<span>公众号</span>
|
|
||||||
</label>
|
</label>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -252,6 +300,21 @@ function ContactsPage() {
|
|||||||
共 {filteredContacts.length} 个联系人
|
共 {filteredContacts.length} 个联系人
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{exportMode && (
|
||||||
|
<div className="selection-toolbar">
|
||||||
|
<label className="checkbox-item">
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={allFilteredSelected}
|
||||||
|
onChange={e => toggleAllFilteredSelected(e.target.checked)}
|
||||||
|
disabled={filteredContacts.length === 0}
|
||||||
|
/>
|
||||||
|
<span>全选当前筛选结果</span>
|
||||||
|
</label>
|
||||||
|
<span className="selection-count">已选 {selectedUsernames.size}(当前筛选 {selectedInFilteredCount} / {filteredContacts.length})</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{isLoading ? (
|
{isLoading ? (
|
||||||
<div className="loading-state">
|
<div className="loading-state">
|
||||||
<Loader2 size={32} className="spin" />
|
<Loader2 size={32} className="spin" />
|
||||||
@@ -263,8 +326,30 @@ function ContactsPage() {
|
|||||||
</div>
|
</div>
|
||||||
) : (
|
) : (
|
||||||
<div className="contacts-list">
|
<div className="contacts-list">
|
||||||
{filteredContacts.map(contact => (
|
{filteredContacts.map(contact => {
|
||||||
<div key={contact.username} className="contact-item">
|
const isChecked = selectedUsernames.has(contact.username)
|
||||||
|
const isActive = !exportMode && selectedContact?.username === contact.username
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
key={contact.username}
|
||||||
|
className={`contact-item ${exportMode && isChecked ? 'selected' : ''} ${isActive ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
if (exportMode) {
|
||||||
|
toggleContactSelected(contact.username, !isChecked)
|
||||||
|
} else {
|
||||||
|
setSelectedContact(isActive ? null : contact)
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{exportMode && (
|
||||||
|
<label className="contact-select" onClick={e => e.stopPropagation()}>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={isChecked}
|
||||||
|
onChange={e => toggleContactSelected(contact.username, e.target.checked)}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
)}
|
||||||
<div className="contact-avatar">
|
<div className="contact-avatar">
|
||||||
{contact.avatarUrl ? (
|
{contact.avatarUrl ? (
|
||||||
<img src={contact.avatarUrl} alt="" />
|
<img src={contact.avatarUrl} alt="" />
|
||||||
@@ -283,12 +368,14 @@ function ContactsPage() {
|
|||||||
<span>{getContactTypeName(contact.type)}</span>
|
<span>{getContactTypeName(contact.type)}</span>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
))}
|
)
|
||||||
|
})}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* 右侧:导出设置 */}
|
{/* 右侧面板 */}
|
||||||
|
{exportMode ? (
|
||||||
<div className="settings-panel">
|
<div className="settings-panel">
|
||||||
<div className="panel-header">
|
<div className="panel-header">
|
||||||
<h2>导出设置</h2>
|
<h2>导出设置</h2>
|
||||||
@@ -330,11 +417,7 @@ function ContactsPage() {
|
|||||||
<div className="setting-section">
|
<div className="setting-section">
|
||||||
<h3>导出选项</h3>
|
<h3>导出选项</h3>
|
||||||
<label className="checkbox-item">
|
<label className="checkbox-item">
|
||||||
<input
|
<input type="checkbox" checked={exportAvatars} onChange={e => setExportAvatars(e.target.checked)} />
|
||||||
type="checkbox"
|
|
||||||
checked={exportAvatars}
|
|
||||||
onChange={e => setExportAvatars(e.target.checked)}
|
|
||||||
/>
|
|
||||||
<span>导出头像</span>
|
<span>导出头像</span>
|
||||||
</label>
|
</label>
|
||||||
</div>
|
</div>
|
||||||
@@ -356,22 +439,64 @@ function ContactsPage() {
|
|||||||
<button
|
<button
|
||||||
className="export-btn"
|
className="export-btn"
|
||||||
onClick={startExport}
|
onClick={startExport}
|
||||||
disabled={!exportFolder || isExporting}
|
disabled={!exportFolder || isExporting || selectedUsernames.size === 0}
|
||||||
>
|
>
|
||||||
{isExporting ? (
|
{isExporting ? (
|
||||||
<>
|
<><Loader2 size={18} className="spin" /><span>导出中...</span></>
|
||||||
<Loader2 size={18} className="spin" />
|
|
||||||
<span>导出中...</span>
|
|
||||||
</>
|
|
||||||
) : (
|
) : (
|
||||||
<>
|
<><Download size={18} /><span>开始导出</span></>
|
||||||
<Download size={18} />
|
|
||||||
<span>开始导出</span>
|
|
||||||
</>
|
|
||||||
)}
|
)}
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
) : selectedContact ? (
|
||||||
|
<div className="settings-panel">
|
||||||
|
<div className="panel-header">
|
||||||
|
<h2>联系人详情</h2>
|
||||||
|
</div>
|
||||||
|
<div className="settings-content">
|
||||||
|
<div className="detail-profile">
|
||||||
|
<div className="detail-avatar">
|
||||||
|
{selectedContact.avatarUrl ? (
|
||||||
|
<img src={selectedContact.avatarUrl} alt="" />
|
||||||
|
) : (
|
||||||
|
<span>{getAvatarLetter(selectedContact.displayName)}</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="detail-name">{selectedContact.displayName}</div>
|
||||||
|
<div className={`contact-type ${selectedContact.type}`}>
|
||||||
|
{getContactTypeIcon(selectedContact.type)}
|
||||||
|
<span>{getContactTypeName(selectedContact.type)}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="detail-info-list">
|
||||||
|
<div className="detail-row"><span className="detail-label">用户名</span><span className="detail-value">{selectedContact.username}</span></div>
|
||||||
|
<div className="detail-row"><span className="detail-label">昵称</span><span className="detail-value">{selectedContact.nickname || selectedContact.displayName}</span></div>
|
||||||
|
{selectedContact.remark && <div className="detail-row"><span className="detail-label">备注</span><span className="detail-value">{selectedContact.remark}</span></div>}
|
||||||
|
<div className="detail-row"><span className="detail-label">类型</span><span className="detail-value">{getContactTypeName(selectedContact.type)}</span></div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button
|
||||||
|
className="goto-chat-btn"
|
||||||
|
onClick={() => {
|
||||||
|
setCurrentSession(selectedContact.username)
|
||||||
|
navigate('/chat')
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<MessageCircle size={18} />
|
||||||
|
<span>查看聊天记录</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="settings-panel">
|
||||||
|
<div className="empty-detail">
|
||||||
|
<User size={48} />
|
||||||
|
<span>点击左侧联系人查看详情</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -955,6 +955,18 @@
|
|||||||
font-size: 15px;
|
font-size: 15px;
|
||||||
font-weight: 600;
|
font-weight: 600;
|
||||||
color: var(--text-primary);
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
&.clickable {
|
||||||
|
cursor: pointer;
|
||||||
|
border-radius: 6px;
|
||||||
|
padding: 2px 8px;
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1015,6 +1027,70 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.year-month-picker {
|
||||||
|
padding: 4px 0;
|
||||||
|
|
||||||
|
.year-selector {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 12px;
|
||||||
|
|
||||||
|
.year-label {
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.calendar-nav-btn {
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
border-color: var(--primary);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.month-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(3, 1fr);
|
||||||
|
gap: 6px;
|
||||||
|
|
||||||
|
.month-btn {
|
||||||
|
padding: 10px 0;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.date-picker-actions {
|
.date-picker-actions {
|
||||||
display: flex;
|
display: flex;
|
||||||
gap: 12px;
|
gap: 12px;
|
||||||
|
|||||||
@@ -53,6 +53,7 @@ function ExportPage() {
|
|||||||
const [showDatePicker, setShowDatePicker] = useState(false)
|
const [showDatePicker, setShowDatePicker] = useState(false)
|
||||||
const [calendarDate, setCalendarDate] = useState(new Date())
|
const [calendarDate, setCalendarDate] = useState(new Date())
|
||||||
const [selectingStart, setSelectingStart] = useState(true)
|
const [selectingStart, setSelectingStart] = useState(true)
|
||||||
|
const [showYearMonthPicker, setShowYearMonthPicker] = useState(false)
|
||||||
const [showMediaLayoutPrompt, setShowMediaLayoutPrompt] = useState(false)
|
const [showMediaLayoutPrompt, setShowMediaLayoutPrompt] = useState(false)
|
||||||
const [showDisplayNameSelect, setShowDisplayNameSelect] = useState(false)
|
const [showDisplayNameSelect, setShowDisplayNameSelect] = useState(false)
|
||||||
const [showPreExportDialog, setShowPreExportDialog] = useState(false)
|
const [showPreExportDialog, setShowPreExportDialog] = useState(false)
|
||||||
@@ -66,6 +67,7 @@ function ExportPage() {
|
|||||||
const [elapsedSeconds, setElapsedSeconds] = useState(0)
|
const [elapsedSeconds, setElapsedSeconds] = useState(0)
|
||||||
const displayNameDropdownRef = useRef<HTMLDivElement>(null)
|
const displayNameDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
const preselectAppliedRef = useRef(false)
|
const preselectAppliedRef = useRef(false)
|
||||||
|
const statsRequestIdRef = useRef(0)
|
||||||
|
|
||||||
const preselectSessionIds = useMemo(() => {
|
const preselectSessionIds = useMemo(() => {
|
||||||
const state = location.state as { preselectSessionIds?: unknown; preselectSessionId?: unknown } | null
|
const state = location.state as { preselectSessionIds?: unknown; preselectSessionId?: unknown } | null
|
||||||
@@ -382,7 +384,9 @@ function ExportPage() {
|
|||||||
if (selectedSessions.size === 0 || !exportFolder) return
|
if (selectedSessions.size === 0 || !exportFolder) return
|
||||||
|
|
||||||
// 先获取预估统计
|
// 先获取预估统计
|
||||||
|
const requestId = ++statsRequestIdRef.current
|
||||||
setIsLoadingStats(true)
|
setIsLoadingStats(true)
|
||||||
|
setPreExportStats(null)
|
||||||
setShowPreExportDialog(true)
|
setShowPreExportDialog(true)
|
||||||
try {
|
try {
|
||||||
const sessionList = Array.from(selectedSessions)
|
const sessionList = Array.from(selectedSessions)
|
||||||
@@ -400,16 +404,21 @@ function ExportPage() {
|
|||||||
} : null
|
} : null
|
||||||
}
|
}
|
||||||
const stats = await window.electronAPI.export.getExportStats(sessionList, exportOptions)
|
const stats = await window.electronAPI.export.getExportStats(sessionList, exportOptions)
|
||||||
|
if (statsRequestIdRef.current !== requestId) return
|
||||||
setPreExportStats(stats)
|
setPreExportStats(stats)
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('获取导出统计失败:', e)
|
console.error('获取导出统计失败:', e)
|
||||||
|
if (statsRequestIdRef.current !== requestId) return
|
||||||
setPreExportStats(null)
|
setPreExportStats(null)
|
||||||
} finally {
|
} finally {
|
||||||
|
if (statsRequestIdRef.current !== requestId) return
|
||||||
setIsLoadingStats(false)
|
setIsLoadingStats(false)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const confirmExport = () => {
|
const confirmExport = () => {
|
||||||
|
statsRequestIdRef.current++
|
||||||
|
setIsLoadingStats(false)
|
||||||
setShowPreExportDialog(false)
|
setShowPreExportDialog(false)
|
||||||
setPreExportStats(null)
|
setPreExportStats(null)
|
||||||
|
|
||||||
@@ -911,7 +920,7 @@ function ExportPage() {
|
|||||||
{isLoadingStats ? (
|
{isLoadingStats ? (
|
||||||
<div style={{ display: 'flex', alignItems: 'center', gap: 8, padding: '24px 0', justifyContent: 'center' }}>
|
<div style={{ display: 'flex', alignItems: 'center', gap: 8, padding: '24px 0', justifyContent: 'center' }}>
|
||||||
<Loader2 size={20} className="spin" />
|
<Loader2 size={20} className="spin" />
|
||||||
<span style={{ fontSize: 14, color: 'var(--text-secondary)' }}>正在统计消息...</span>
|
<span style={{ fontSize: 14, color: 'var(--text-secondary)' }}>正在统计消息,可直接点击“直接导出”跳过等待</span>
|
||||||
</div>
|
</div>
|
||||||
) : preExportStats ? (
|
) : preExportStats ? (
|
||||||
<div style={{ padding: '12px 0' }}>
|
<div style={{ padding: '12px 0' }}>
|
||||||
@@ -957,11 +966,11 @@ function ExportPage() {
|
|||||||
<p style={{ fontSize: 14, color: 'var(--text-secondary)', padding: '16px 0' }}>统计信息获取失败,仍可继续导出</p>
|
<p style={{ fontSize: 14, color: 'var(--text-secondary)', padding: '16px 0' }}>统计信息获取失败,仍可继续导出</p>
|
||||||
)}
|
)}
|
||||||
<div className="layout-actions" style={{ display: 'flex', gap: 8, justifyContent: 'flex-end', marginTop: 8 }}>
|
<div className="layout-actions" style={{ display: 'flex', gap: 8, justifyContent: 'flex-end', marginTop: 8 }}>
|
||||||
<button className="layout-cancel-btn" onClick={() => { setShowPreExportDialog(false); setPreExportStats(null) }}>
|
<button className="layout-cancel-btn" onClick={() => { statsRequestIdRef.current++; setIsLoadingStats(false); setShowPreExportDialog(false); setPreExportStats(null) }}>
|
||||||
取消
|
取消
|
||||||
</button>
|
</button>
|
||||||
<button className="layout-option-btn primary" onClick={confirmExport} disabled={isLoadingStats}>
|
<button className="layout-option-btn primary" onClick={confirmExport}>
|
||||||
<span className="layout-title">开始导出</span>
|
<span className="layout-title">{isLoadingStats ? '直接导出' : '开始导出'}</span>
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -1039,7 +1048,7 @@ function ExportPage() {
|
|||||||
|
|
||||||
{/* 日期选择弹窗 */}
|
{/* 日期选择弹窗 */}
|
||||||
{showDatePicker && (
|
{showDatePicker && (
|
||||||
<div className="export-overlay" onClick={() => setShowDatePicker(false)}>
|
<div className="export-overlay" onClick={() => { setShowDatePicker(false); setShowYearMonthPicker(false) }}>
|
||||||
<div className="date-picker-modal" onClick={e => e.stopPropagation()}>
|
<div className="date-picker-modal" onClick={e => e.stopPropagation()}>
|
||||||
<h3>选择时间范围</h3>
|
<h3>选择时间范围</h3>
|
||||||
<p style={{ fontSize: '13px', color: 'var(--text-secondary)', margin: '8px 0 16px 0' }}>
|
<p style={{ fontSize: '13px', color: 'var(--text-secondary)', margin: '8px 0 16px 0' }}>
|
||||||
@@ -1114,7 +1123,7 @@ function ExportPage() {
|
|||||||
>
|
>
|
||||||
<ChevronLeft size={18} />
|
<ChevronLeft size={18} />
|
||||||
</button>
|
</button>
|
||||||
<span className="calendar-month">
|
<span className="calendar-month clickable" onClick={() => setShowYearMonthPicker(!showYearMonthPicker)}>
|
||||||
{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月
|
{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月
|
||||||
</span>
|
</span>
|
||||||
<button
|
<button
|
||||||
@@ -1124,6 +1133,32 @@ function ExportPage() {
|
|||||||
<ChevronRight size={18} />
|
<ChevronRight size={18} />
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
{showYearMonthPicker ? (
|
||||||
|
<div className="year-month-picker">
|
||||||
|
<div className="year-selector">
|
||||||
|
<button className="calendar-nav-btn" onClick={() => setCalendarDate(new Date(calendarDate.getFullYear() - 1, calendarDate.getMonth(), 1))}>
|
||||||
|
<ChevronLeft size={16} />
|
||||||
|
</button>
|
||||||
|
<span className="year-label">{calendarDate.getFullYear()}年</span>
|
||||||
|
<button className="calendar-nav-btn" onClick={() => setCalendarDate(new Date(calendarDate.getFullYear() + 1, calendarDate.getMonth(), 1))}>
|
||||||
|
<ChevronRight size={16} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="month-grid">
|
||||||
|
{['一月','二月','三月','四月','五月','六月','七月','八月','九月','十月','十一月','十二月'].map((name, i) => (
|
||||||
|
<button
|
||||||
|
key={i}
|
||||||
|
className={`month-btn ${i === calendarDate.getMonth() ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setCalendarDate(new Date(calendarDate.getFullYear(), i, 1))
|
||||||
|
setShowYearMonthPicker(false)
|
||||||
|
}}
|
||||||
|
>{name}</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
<div className="calendar-weekdays">
|
<div className="calendar-weekdays">
|
||||||
{['日', '一', '二', '三', '四', '五', '六'].map(day => (
|
{['日', '一', '二', '三', '四', '五', '六'].map(day => (
|
||||||
<div key={day} className="calendar-weekday">{day}</div>
|
<div key={day} className="calendar-weekday">{day}</div>
|
||||||
@@ -1155,12 +1190,14 @@ function ExportPage() {
|
|||||||
)
|
)
|
||||||
})}
|
})}
|
||||||
</div>
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
<div className="date-picker-actions">
|
<div className="date-picker-actions">
|
||||||
<button className="cancel-btn" onClick={() => setShowDatePicker(false)}>
|
<button className="cancel-btn" onClick={() => { setShowDatePicker(false); setShowYearMonthPicker(false) }}>
|
||||||
取消
|
取消
|
||||||
</button>
|
</button>
|
||||||
<button className="confirm-btn" onClick={() => setShowDatePicker(false)}>
|
<button className="confirm-btn" onClick={() => { setShowDatePicker(false); setShowYearMonthPicker(false) }}>
|
||||||
确定
|
确定
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -777,6 +777,344 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.member-export-panel {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 16px;
|
||||||
|
min-height: 0;
|
||||||
|
|
||||||
|
.member-export-empty {
|
||||||
|
padding: 20px;
|
||||||
|
border-radius: 12px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
text-align: center;
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-grid {
|
||||||
|
display: grid;
|
||||||
|
gap: 12px;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(220px, 1fr));
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-field {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
position: relative;
|
||||||
|
|
||||||
|
> span {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-trigger {
|
||||||
|
width: 100%;
|
||||||
|
padding: 10px 12px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 9999px;
|
||||||
|
background: var(--bg-primary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
border-color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.open {
|
||||||
|
border-color: var(--primary);
|
||||||
|
box-shadow: 0 0 0 3px color-mix(in srgb, var(--primary) 15%, transparent);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-value {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
text-align: left;
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-dropdown {
|
||||||
|
position: absolute;
|
||||||
|
top: calc(100% + 6px);
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
background: color-mix(in srgb, var(--bg-primary) 85%, var(--bg-secondary));
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 6px;
|
||||||
|
box-shadow: var(--shadow-md);
|
||||||
|
z-index: 30;
|
||||||
|
max-height: 280px;
|
||||||
|
overflow-y: auto;
|
||||||
|
backdrop-filter: blur(14px);
|
||||||
|
-webkit-backdrop-filter: blur(14px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-option {
|
||||||
|
width: 100%;
|
||||||
|
text-align: left;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 4px;
|
||||||
|
padding: 10px 12px;
|
||||||
|
border: none;
|
||||||
|
border-radius: 10px;
|
||||||
|
background: transparent;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.15s;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 13px;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: color-mix(in srgb, var(--primary) 12%, transparent);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.option-label {
|
||||||
|
font-weight: 500;
|
||||||
|
}
|
||||||
|
|
||||||
|
.option-desc {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
line-height: 1.4;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-trigger-value {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
min-width: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-dropdown {
|
||||||
|
padding: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-search {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 7px 9px;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
input {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
outline: none;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-options {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-empty {
|
||||||
|
padding: 10px 8px;
|
||||||
|
text-align: center;
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-select-option {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 28px 1fr;
|
||||||
|
gap: 8px;
|
||||||
|
align-items: center;
|
||||||
|
padding: 8px 10px;
|
||||||
|
|
||||||
|
.member-option-main {
|
||||||
|
display: block;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-option-meta {
|
||||||
|
grid-column: 2 / 3;
|
||||||
|
font-size: 11px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
.member-option-main,
|
||||||
|
.member-option-meta {
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-folder {
|
||||||
|
grid-column: 1 / -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-folder-row {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
|
||||||
|
input {
|
||||||
|
flex: 1;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 10px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 13px;
|
||||||
|
padding: 8px 10px;
|
||||||
|
outline: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
button {
|
||||||
|
border: none;
|
||||||
|
border-radius: 10px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
padding: 0 12px;
|
||||||
|
font-size: 12px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.15s;
|
||||||
|
white-space: nowrap;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-options {
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 14px;
|
||||||
|
background: rgba(255, 255, 255, 0.05);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-chip-group {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.chip-group-label {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-chip-list {
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.export-filter-chip {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 8px 14px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 10px;
|
||||||
|
cursor: pointer;
|
||||||
|
user-select: none;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
white-space: nowrap;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
border-color: var(--text-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--primary-light);
|
||||||
|
border-color: var(--primary);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.disabled,
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.45;
|
||||||
|
cursor: not-allowed;
|
||||||
|
transform: none;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border-color: var(--border-color);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.member-export-start-btn {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
border: none;
|
||||||
|
border-radius: 10px;
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
padding: 10px 16px;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.rankings-list {
|
.rankings-list {
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
import { useState, useEffect, useRef, useCallback, useMemo } from 'react'
|
import { useState, useEffect, useRef, useCallback, useMemo } from 'react'
|
||||||
import { useLocation } from 'react-router-dom'
|
import { useLocation } from 'react-router-dom'
|
||||||
import { Users, BarChart3, Clock, Image, Loader2, RefreshCw, User, Medal, Search, X, ChevronLeft, Copy, Check, Download } from 'lucide-react'
|
import { Users, BarChart3, Clock, Image, Loader2, RefreshCw, Medal, Search, X, ChevronLeft, Copy, Check, Download, ChevronDown } from 'lucide-react'
|
||||||
import { Avatar } from '../components/Avatar'
|
import { Avatar } from '../components/Avatar'
|
||||||
import ReactECharts from 'echarts-for-react'
|
import ReactECharts from 'echarts-for-react'
|
||||||
import DateRangePicker from '../components/DateRangePicker'
|
import DateRangePicker from '../components/DateRangePicker'
|
||||||
|
import * as configService from '../services/config'
|
||||||
import './GroupAnalyticsPage.scss'
|
import './GroupAnalyticsPage.scss'
|
||||||
|
|
||||||
interface GroupChatInfo {
|
interface GroupChatInfo {
|
||||||
@@ -28,7 +29,26 @@ interface GroupMessageRank {
|
|||||||
messageCount: number
|
messageCount: number
|
||||||
}
|
}
|
||||||
|
|
||||||
type AnalysisFunction = 'members' | 'ranking' | 'activeHours' | 'mediaStats'
|
type AnalysisFunction = 'members' | 'memberExport' | 'ranking' | 'activeHours' | 'mediaStats'
|
||||||
|
type MemberExportFormat = 'chatlab' | 'chatlab-jsonl' | 'json' | 'html' | 'txt' | 'excel' | 'weclone'
|
||||||
|
|
||||||
|
interface MemberMessageExportOptions {
|
||||||
|
format: MemberExportFormat
|
||||||
|
exportAvatars: boolean
|
||||||
|
exportMedia: boolean
|
||||||
|
exportImages: boolean
|
||||||
|
exportVoices: boolean
|
||||||
|
exportVideos: boolean
|
||||||
|
exportEmojis: boolean
|
||||||
|
exportVoiceAsText: boolean
|
||||||
|
displayNamePreference: 'group-nickname' | 'remark' | 'nickname'
|
||||||
|
}
|
||||||
|
|
||||||
|
interface MemberExportFormatOption {
|
||||||
|
value: MemberExportFormat
|
||||||
|
label: string
|
||||||
|
desc: string
|
||||||
|
}
|
||||||
|
|
||||||
function GroupAnalyticsPage() {
|
function GroupAnalyticsPage() {
|
||||||
const location = useLocation()
|
const location = useLocation()
|
||||||
@@ -46,10 +66,31 @@ function GroupAnalyticsPage() {
|
|||||||
const [mediaStats, setMediaStats] = useState<{ typeCounts: Array<{ type: number; name: string; count: number }>; total: number } | null>(null)
|
const [mediaStats, setMediaStats] = useState<{ typeCounts: Array<{ type: number; name: string; count: number }>; total: number } | null>(null)
|
||||||
const [functionLoading, setFunctionLoading] = useState(false)
|
const [functionLoading, setFunctionLoading] = useState(false)
|
||||||
const [isExportingMembers, setIsExportingMembers] = useState(false)
|
const [isExportingMembers, setIsExportingMembers] = useState(false)
|
||||||
|
const [isExportingMemberMessages, setIsExportingMemberMessages] = useState(false)
|
||||||
|
const [selectedExportMemberUsername, setSelectedExportMemberUsername] = useState('')
|
||||||
|
const [exportFolder, setExportFolder] = useState('')
|
||||||
|
const [memberExportOptions, setMemberExportOptions] = useState<MemberMessageExportOptions>({
|
||||||
|
format: 'excel',
|
||||||
|
exportAvatars: true,
|
||||||
|
exportMedia: false,
|
||||||
|
exportImages: true,
|
||||||
|
exportVoices: true,
|
||||||
|
exportVideos: true,
|
||||||
|
exportEmojis: true,
|
||||||
|
exportVoiceAsText: false,
|
||||||
|
displayNamePreference: 'remark'
|
||||||
|
})
|
||||||
|
|
||||||
// 成员详情弹框
|
// 成员详情弹框
|
||||||
const [selectedMember, setSelectedMember] = useState<GroupMember | null>(null)
|
const [selectedMember, setSelectedMember] = useState<GroupMember | null>(null)
|
||||||
const [copiedField, setCopiedField] = useState<string | null>(null)
|
const [copiedField, setCopiedField] = useState<string | null>(null)
|
||||||
|
const [showMemberSelect, setShowMemberSelect] = useState(false)
|
||||||
|
const [showFormatSelect, setShowFormatSelect] = useState(false)
|
||||||
|
const [showDisplayNameSelect, setShowDisplayNameSelect] = useState(false)
|
||||||
|
const [memberSearchKeyword, setMemberSearchKeyword] = useState('')
|
||||||
|
const memberSelectDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
|
const formatDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
|
const displayNameDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
// 时间范围
|
// 时间范围
|
||||||
const [startDate, setStartDate] = useState<string>('')
|
const [startDate, setStartDate] = useState<string>('')
|
||||||
@@ -74,9 +115,84 @@ function GroupAnalyticsPage() {
|
|||||||
.filter(Boolean)
|
.filter(Boolean)
|
||||||
}, [location.state])
|
}, [location.state])
|
||||||
|
|
||||||
|
const memberExportFormatOptions = useMemo<MemberExportFormatOption[]>(() => ([
|
||||||
|
{ value: 'excel', label: 'Excel', desc: '电子表格,适合统计分析' },
|
||||||
|
{ value: 'txt', label: 'TXT', desc: '纯文本,通用格式' },
|
||||||
|
{ value: 'json', label: 'JSON', desc: '详细格式,包含完整消息信息' },
|
||||||
|
{ value: 'chatlab', label: 'ChatLab', desc: '标准格式,支持其他软件导入' },
|
||||||
|
{ value: 'chatlab-jsonl', label: 'ChatLab JSONL', desc: '流式格式,适合大量消息' },
|
||||||
|
{ value: 'html', label: 'HTML', desc: '网页格式,可直接浏览' },
|
||||||
|
{ value: 'weclone', label: 'WeClone CSV', desc: 'WeClone 兼容字段格式(CSV)' }
|
||||||
|
]), [])
|
||||||
|
const displayNameOptions = useMemo<Array<{
|
||||||
|
value: MemberMessageExportOptions['displayNamePreference']
|
||||||
|
label: string
|
||||||
|
desc: string
|
||||||
|
}>>(() => ([
|
||||||
|
{ value: 'group-nickname', label: '群昵称优先', desc: '仅群聊有效,私聊显示备注/昵称' },
|
||||||
|
{ value: 'remark', label: '备注优先', desc: '有备注显示备注,否则显示昵称' },
|
||||||
|
{ value: 'nickname', label: '微信昵称', desc: '始终显示微信昵称' }
|
||||||
|
]), [])
|
||||||
|
const selectedExportMember = useMemo(
|
||||||
|
() => members.find(member => member.username === selectedExportMemberUsername) || null,
|
||||||
|
[members, selectedExportMemberUsername]
|
||||||
|
)
|
||||||
|
const selectedFormatOption = useMemo(
|
||||||
|
() => memberExportFormatOptions.find(option => option.value === memberExportOptions.format) || memberExportFormatOptions[0],
|
||||||
|
[memberExportFormatOptions, memberExportOptions.format]
|
||||||
|
)
|
||||||
|
const selectedDisplayNameOption = useMemo(
|
||||||
|
() => displayNameOptions.find(option => option.value === memberExportOptions.displayNamePreference) || displayNameOptions[0],
|
||||||
|
[displayNameOptions, memberExportOptions.displayNamePreference]
|
||||||
|
)
|
||||||
|
const filteredMemberOptions = useMemo(() => {
|
||||||
|
const keyword = memberSearchKeyword.trim().toLowerCase()
|
||||||
|
if (!keyword) return members
|
||||||
|
return members.filter(member => {
|
||||||
|
const fields = [
|
||||||
|
member.username,
|
||||||
|
member.displayName,
|
||||||
|
member.nickname,
|
||||||
|
member.remark,
|
||||||
|
member.alias
|
||||||
|
]
|
||||||
|
return fields.some(field => String(field || '').toLowerCase().includes(keyword))
|
||||||
|
})
|
||||||
|
}, [memberSearchKeyword, members])
|
||||||
|
|
||||||
|
const loadExportPath = useCallback(async () => {
|
||||||
|
try {
|
||||||
|
const savedPath = await configService.getExportPath()
|
||||||
|
if (savedPath) {
|
||||||
|
setExportFolder(savedPath)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
const downloadsPath = await window.electronAPI.app.getDownloadsPath()
|
||||||
|
setExportFolder(downloadsPath)
|
||||||
|
} catch (e) {
|
||||||
|
console.error('加载导出路径失败:', e)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const loadGroups = useCallback(async () => {
|
||||||
|
setIsLoading(true)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.groupAnalytics.getGroupChats()
|
||||||
|
if (result.success && result.data) {
|
||||||
|
setGroups(result.data)
|
||||||
|
setFilteredGroups(result.data)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(e)
|
||||||
|
} finally {
|
||||||
|
setIsLoading(false)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
loadGroups()
|
loadGroups()
|
||||||
}, [])
|
loadExportPath()
|
||||||
|
}, [loadGroups, loadExportPath])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
preselectAppliedRef.current = false
|
preselectAppliedRef.current = false
|
||||||
@@ -90,6 +206,34 @@ function GroupAnalyticsPage() {
|
|||||||
}
|
}
|
||||||
}, [searchQuery, groups])
|
}, [searchQuery, groups])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (members.length === 0) {
|
||||||
|
setSelectedExportMemberUsername('')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
const exists = members.some(member => member.username === selectedExportMemberUsername)
|
||||||
|
if (!exists) {
|
||||||
|
setSelectedExportMemberUsername(members[0].username)
|
||||||
|
}
|
||||||
|
}, [members, selectedExportMemberUsername])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const handleClickOutside = (event: MouseEvent) => {
|
||||||
|
const target = event.target as Node
|
||||||
|
if (showMemberSelect && memberSelectDropdownRef.current && !memberSelectDropdownRef.current.contains(target)) {
|
||||||
|
setShowMemberSelect(false)
|
||||||
|
}
|
||||||
|
if (showFormatSelect && formatDropdownRef.current && !formatDropdownRef.current.contains(target)) {
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
}
|
||||||
|
if (showDisplayNameSelect && displayNameDropdownRef.current && !displayNameDropdownRef.current.contains(target)) {
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
document.addEventListener('mousedown', handleClickOutside)
|
||||||
|
return () => document.removeEventListener('mousedown', handleClickOutside)
|
||||||
|
}, [showDisplayNameSelect, showFormatSelect, showMemberSelect])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (preselectAppliedRef.current) return
|
if (preselectAppliedRef.current) return
|
||||||
if (groups.length === 0 || preselectGroupIds.length === 0) return
|
if (groups.length === 0 || preselectGroupIds.length === 0) return
|
||||||
@@ -125,27 +269,12 @@ function GroupAnalyticsPage() {
|
|||||||
|
|
||||||
// 日期范围变化时自动刷新
|
// 日期范围变化时自动刷新
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (dateRangeReady && selectedGroup && selectedFunction && selectedFunction !== 'members') {
|
if (dateRangeReady && selectedGroup && selectedFunction && selectedFunction !== 'members' && selectedFunction !== 'memberExport') {
|
||||||
setDateRangeReady(false)
|
setDateRangeReady(false)
|
||||||
loadFunctionData(selectedFunction)
|
loadFunctionData(selectedFunction)
|
||||||
}
|
}
|
||||||
}, [dateRangeReady])
|
}, [dateRangeReady])
|
||||||
|
|
||||||
const loadGroups = useCallback(async () => {
|
|
||||||
setIsLoading(true)
|
|
||||||
try {
|
|
||||||
const result = await window.electronAPI.groupAnalytics.getGroupChats()
|
|
||||||
if (result.success && result.data) {
|
|
||||||
setGroups(result.data)
|
|
||||||
setFilteredGroups(result.data)
|
|
||||||
}
|
|
||||||
} catch (e) {
|
|
||||||
console.error(e)
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false)
|
|
||||||
}
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const handleChange = () => {
|
const handleChange = () => {
|
||||||
setGroups([])
|
setGroups([])
|
||||||
@@ -157,15 +286,21 @@ function GroupAnalyticsPage() {
|
|||||||
setActiveHours({})
|
setActiveHours({})
|
||||||
setMediaStats(null)
|
setMediaStats(null)
|
||||||
void loadGroups()
|
void loadGroups()
|
||||||
|
void loadExportPath()
|
||||||
}
|
}
|
||||||
window.addEventListener('wxid-changed', handleChange as EventListener)
|
window.addEventListener('wxid-changed', handleChange as EventListener)
|
||||||
return () => window.removeEventListener('wxid-changed', handleChange as EventListener)
|
return () => window.removeEventListener('wxid-changed', handleChange as EventListener)
|
||||||
}, [loadGroups])
|
}, [loadExportPath, loadGroups])
|
||||||
|
|
||||||
const handleGroupSelect = (group: GroupChatInfo) => {
|
const handleGroupSelect = (group: GroupChatInfo) => {
|
||||||
if (selectedGroup?.username !== group.username) {
|
if (selectedGroup?.username !== group.username) {
|
||||||
setSelectedGroup(group)
|
setSelectedGroup(group)
|
||||||
setSelectedFunction(null)
|
setSelectedFunction(null)
|
||||||
|
setSelectedExportMemberUsername('')
|
||||||
|
setMemberSearchKeyword('')
|
||||||
|
setShowMemberSelect(false)
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -191,6 +326,11 @@ function GroupAnalyticsPage() {
|
|||||||
if (result.success && result.data) setMembers(result.data)
|
if (result.success && result.data) setMembers(result.data)
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
case 'memberExport': {
|
||||||
|
const result = await window.electronAPI.groupAnalytics.getGroupMembers(selectedGroup.username)
|
||||||
|
if (result.success && result.data) setMembers(result.data)
|
||||||
|
break
|
||||||
|
}
|
||||||
case 'ranking': {
|
case 'ranking': {
|
||||||
const result = await window.electronAPI.groupAnalytics.getGroupMessageRanking(selectedGroup.username, 20, startTime, endTime)
|
const result = await window.electronAPI.groupAnalytics.getGroupMessageRanking(selectedGroup.username, 20, startTime, endTime)
|
||||||
if (result.success && result.data) setRankings(result.data)
|
if (result.success && result.data) setRankings(result.data)
|
||||||
@@ -286,6 +426,7 @@ function GroupAnalyticsPage() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const handleDateRangeComplete = () => {
|
const handleDateRangeComplete = () => {
|
||||||
|
if (selectedFunction === 'memberExport') return
|
||||||
setDateRangeReady(true)
|
setDateRangeReady(true)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -323,6 +464,86 @@ function GroupAnalyticsPage() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const handleMemberExportFormatChange = (format: MemberExportFormat) => {
|
||||||
|
setMemberExportOptions(prev => {
|
||||||
|
const next = { ...prev, format }
|
||||||
|
if (format === 'html') {
|
||||||
|
return {
|
||||||
|
...next,
|
||||||
|
exportMedia: true,
|
||||||
|
exportImages: true,
|
||||||
|
exportVoices: true,
|
||||||
|
exportVideos: true,
|
||||||
|
exportEmojis: true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleChooseExportFolder = async () => {
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.dialog.openDirectory({
|
||||||
|
title: '选择导出目录'
|
||||||
|
})
|
||||||
|
if (!result.canceled && result.filePaths.length > 0) {
|
||||||
|
setExportFolder(result.filePaths[0])
|
||||||
|
await configService.setExportPath(result.filePaths[0])
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('选择导出目录失败:', e)
|
||||||
|
alert(`选择导出目录失败:${String(e)}`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleExportMemberMessages = async () => {
|
||||||
|
if (!selectedGroup || !selectedExportMemberUsername || !exportFolder || isExportingMemberMessages) return
|
||||||
|
const member = members.find(item => item.username === selectedExportMemberUsername)
|
||||||
|
if (!member) {
|
||||||
|
alert('请先选择成员')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
setIsExportingMemberMessages(true)
|
||||||
|
try {
|
||||||
|
const hasDateRange = Boolean(startDate && endDate)
|
||||||
|
const result = await window.electronAPI.export.exportSessions(
|
||||||
|
[selectedGroup.username],
|
||||||
|
exportFolder,
|
||||||
|
{
|
||||||
|
format: memberExportOptions.format,
|
||||||
|
dateRange: hasDateRange
|
||||||
|
? {
|
||||||
|
start: Math.floor(new Date(startDate).getTime() / 1000),
|
||||||
|
end: Math.floor(new Date(`${endDate}T23:59:59`).getTime() / 1000)
|
||||||
|
}
|
||||||
|
: null,
|
||||||
|
exportAvatars: memberExportOptions.exportAvatars,
|
||||||
|
exportMedia: memberExportOptions.exportMedia,
|
||||||
|
exportImages: memberExportOptions.exportMedia && memberExportOptions.exportImages,
|
||||||
|
exportVoices: memberExportOptions.exportMedia && memberExportOptions.exportVoices,
|
||||||
|
exportVideos: memberExportOptions.exportMedia && memberExportOptions.exportVideos,
|
||||||
|
exportEmojis: memberExportOptions.exportMedia && memberExportOptions.exportEmojis,
|
||||||
|
exportVoiceAsText: memberExportOptions.exportVoiceAsText,
|
||||||
|
sessionLayout: memberExportOptions.exportMedia ? 'per-session' : 'shared',
|
||||||
|
displayNamePreference: memberExportOptions.displayNamePreference,
|
||||||
|
senderUsername: member.username,
|
||||||
|
fileNameSuffix: sanitizeFileName(member.displayName || member.username)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
if (result.success && (result.successCount ?? 0) > 0) {
|
||||||
|
alert(`导出成功:${member.displayName || member.username}`)
|
||||||
|
} else {
|
||||||
|
alert(`导出失败:${result.error || '未知错误'}`)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('导出成员消息失败:', e)
|
||||||
|
alert(`导出失败:${String(e)}`)
|
||||||
|
} finally {
|
||||||
|
setIsExportingMemberMessages(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const handleCopy = async (text: string, field: string) => {
|
const handleCopy = async (text: string, field: string) => {
|
||||||
try {
|
try {
|
||||||
await navigator.clipboard.writeText(text)
|
await navigator.clipboard.writeText(text)
|
||||||
@@ -479,6 +700,10 @@ function GroupAnalyticsPage() {
|
|||||||
<Users size={32} />
|
<Users size={32} />
|
||||||
<span>群成员查看</span>
|
<span>群成员查看</span>
|
||||||
</div>
|
</div>
|
||||||
|
<div className="function-card" onClick={() => handleFunctionSelect('memberExport')}>
|
||||||
|
<Download size={32} />
|
||||||
|
<span>成员消息导出</span>
|
||||||
|
</div>
|
||||||
<div className="function-card" onClick={() => handleFunctionSelect('ranking')}>
|
<div className="function-card" onClick={() => handleFunctionSelect('ranking')}>
|
||||||
<BarChart3 size={32} />
|
<BarChart3 size={32} />
|
||||||
<span>群聊发言排行</span>
|
<span>群聊发言排行</span>
|
||||||
@@ -499,6 +724,7 @@ function GroupAnalyticsPage() {
|
|||||||
const getFunctionTitle = () => {
|
const getFunctionTitle = () => {
|
||||||
switch (selectedFunction) {
|
switch (selectedFunction) {
|
||||||
case 'members': return '群成员查看'
|
case 'members': return '群成员查看'
|
||||||
|
case 'memberExport': return '成员消息导出'
|
||||||
case 'ranking': return '群聊发言排行'
|
case 'ranking': return '群聊发言排行'
|
||||||
case 'activeHours': return '群聊活跃时段'
|
case 'activeHours': return '群聊活跃时段'
|
||||||
case 'mediaStats': return '媒体内容统计'
|
case 'mediaStats': return '媒体内容统计'
|
||||||
@@ -554,6 +780,234 @@ function GroupAnalyticsPage() {
|
|||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
{selectedFunction === 'memberExport' && (
|
||||||
|
<div className="member-export-panel">
|
||||||
|
{members.length === 0 ? (
|
||||||
|
<div className="member-export-empty">暂无群成员数据,请先刷新。</div>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<div className="member-export-grid">
|
||||||
|
<div className="member-export-field" ref={memberSelectDropdownRef}>
|
||||||
|
<span>导出成员</span>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`select-trigger ${showMemberSelect ? 'open' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setShowMemberSelect(prev => !prev)
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<div className="member-select-trigger-value">
|
||||||
|
<Avatar
|
||||||
|
src={selectedExportMember?.avatarUrl}
|
||||||
|
name={selectedExportMember?.displayName || selectedExportMember?.username || '?'}
|
||||||
|
size={24}
|
||||||
|
/>
|
||||||
|
<span className="select-value">{selectedExportMember?.displayName || selectedExportMember?.username || '请选择成员'}</span>
|
||||||
|
</div>
|
||||||
|
<ChevronDown size={16} />
|
||||||
|
</button>
|
||||||
|
{showMemberSelect && (
|
||||||
|
<div className="select-dropdown member-select-dropdown">
|
||||||
|
<div className="member-select-search">
|
||||||
|
<Search size={14} />
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
value={memberSearchKeyword}
|
||||||
|
onChange={e => setMemberSearchKeyword(e.target.value)}
|
||||||
|
placeholder="搜索 wxid / 昵称 / 备注 / 微信号"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div className="member-select-options">
|
||||||
|
{filteredMemberOptions.length === 0 ? (
|
||||||
|
<div className="member-select-empty">无匹配成员</div>
|
||||||
|
) : (
|
||||||
|
filteredMemberOptions.map(member => (
|
||||||
|
<button
|
||||||
|
key={member.username}
|
||||||
|
type="button"
|
||||||
|
className={`select-option member-select-option ${selectedExportMemberUsername === member.username ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setSelectedExportMemberUsername(member.username)
|
||||||
|
setShowMemberSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Avatar src={member.avatarUrl} name={member.displayName} size={28} />
|
||||||
|
<span className="member-option-main">{member.displayName || member.username}</span>
|
||||||
|
<span className="member-option-meta">
|
||||||
|
wxid: {member.username}
|
||||||
|
{member.alias ? ` · 微信号: ${member.alias}` : ''}
|
||||||
|
{member.remark ? ` · 备注: ${member.remark}` : ''}
|
||||||
|
{member.nickname ? ` · 昵称: ${member.nickname}` : ''}
|
||||||
|
</span>
|
||||||
|
</button>
|
||||||
|
))
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="member-export-field" ref={formatDropdownRef}>
|
||||||
|
<span>导出格式</span>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`select-trigger ${showFormatSelect ? 'open' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setShowFormatSelect(prev => !prev)
|
||||||
|
setShowMemberSelect(false)
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="select-value">{selectedFormatOption.label}</span>
|
||||||
|
<ChevronDown size={16} />
|
||||||
|
</button>
|
||||||
|
{showFormatSelect && (
|
||||||
|
<div className="select-dropdown">
|
||||||
|
{memberExportFormatOptions.map(option => (
|
||||||
|
<button
|
||||||
|
key={option.value}
|
||||||
|
type="button"
|
||||||
|
className={`select-option ${memberExportOptions.format === option.value ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
handleMemberExportFormatChange(option.value)
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="option-label">{option.label}</span>
|
||||||
|
<span className="option-desc">{option.desc}</span>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="member-export-field member-export-folder">
|
||||||
|
<span>导出目录</span>
|
||||||
|
<div className="member-export-folder-row">
|
||||||
|
<input value={exportFolder} readOnly placeholder="请选择导出目录" />
|
||||||
|
<button type="button" onClick={handleChooseExportFolder}>
|
||||||
|
选择目录
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="member-export-options">
|
||||||
|
<div className="member-export-chip-group">
|
||||||
|
<span className="chip-group-label">媒体导出</span>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportMedia ? 'active' : ''}`}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportMedia: !prev.exportMedia }))}
|
||||||
|
>
|
||||||
|
导出媒体文件
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="member-export-chip-group">
|
||||||
|
<span className="chip-group-label">媒体类型</span>
|
||||||
|
<div className="member-export-chip-list">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportImages ? 'active' : ''} ${!memberExportOptions.exportMedia ? 'disabled' : ''}`}
|
||||||
|
disabled={!memberExportOptions.exportMedia}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportImages: !prev.exportImages }))}
|
||||||
|
>
|
||||||
|
图片
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportVoices ? 'active' : ''} ${!memberExportOptions.exportMedia ? 'disabled' : ''}`}
|
||||||
|
disabled={!memberExportOptions.exportMedia}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportVoices: !prev.exportVoices }))}
|
||||||
|
>
|
||||||
|
语音
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportVideos ? 'active' : ''} ${!memberExportOptions.exportMedia ? 'disabled' : ''}`}
|
||||||
|
disabled={!memberExportOptions.exportMedia}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportVideos: !prev.exportVideos }))}
|
||||||
|
>
|
||||||
|
视频
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportEmojis ? 'active' : ''} ${!memberExportOptions.exportMedia ? 'disabled' : ''}`}
|
||||||
|
disabled={!memberExportOptions.exportMedia}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportEmojis: !prev.exportEmojis }))}
|
||||||
|
>
|
||||||
|
表情
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="member-export-chip-group">
|
||||||
|
<span className="chip-group-label">附加选项</span>
|
||||||
|
<div className="member-export-chip-list">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportVoiceAsText ? 'active' : ''}`}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportVoiceAsText: !prev.exportVoiceAsText }))}
|
||||||
|
>
|
||||||
|
语音转文字
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`export-filter-chip ${memberExportOptions.exportAvatars ? 'active' : ''}`}
|
||||||
|
onClick={() => setMemberExportOptions(prev => ({ ...prev, exportAvatars: !prev.exportAvatars }))}
|
||||||
|
>
|
||||||
|
导出头像
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="member-export-field" ref={displayNameDropdownRef}>
|
||||||
|
<span>显示名称规则</span>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`select-trigger ${showDisplayNameSelect ? 'open' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setShowDisplayNameSelect(prev => !prev)
|
||||||
|
setShowMemberSelect(false)
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="select-value">{selectedDisplayNameOption.label}</span>
|
||||||
|
<ChevronDown size={16} />
|
||||||
|
</button>
|
||||||
|
{showDisplayNameSelect && (
|
||||||
|
<div className="select-dropdown">
|
||||||
|
{displayNameOptions.map(option => (
|
||||||
|
<button
|
||||||
|
key={option.value}
|
||||||
|
type="button"
|
||||||
|
className={`select-option ${memberExportOptions.displayNamePreference === option.value ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setMemberExportOptions(prev => ({ ...prev, displayNamePreference: option.value }))
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="option-label">{option.label}</span>
|
||||||
|
<span className="option-desc">{option.desc}</span>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="member-export-actions">
|
||||||
|
<button
|
||||||
|
className="member-export-start-btn"
|
||||||
|
onClick={handleExportMemberMessages}
|
||||||
|
disabled={isExportingMemberMessages || !selectedExportMemberUsername || !exportFolder}
|
||||||
|
>
|
||||||
|
{isExportingMemberMessages ? <Loader2 size={16} className="spin" /> : <Download size={16} />}
|
||||||
|
<span>{isExportingMemberMessages ? '导出中...' : '开始导出'}</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
{selectedFunction === 'ranking' && (
|
{selectedFunction === 'ranking' && (
|
||||||
<div className="rankings-list">
|
<div className="rankings-list">
|
||||||
{rankings.map((item, index) => (
|
{rankings.map((item, index) => (
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
import { useEffect, useState, useRef } from 'react'
|
import { useEffect, useState, useRef } from 'react'
|
||||||
import { NotificationToast, type NotificationData } from '../components/NotificationToast'
|
import { NotificationToast, type NotificationData } from '../components/NotificationToast'
|
||||||
|
import { useThemeStore } from '../stores/themeStore'
|
||||||
import '../components/NotificationToast.scss'
|
import '../components/NotificationToast.scss'
|
||||||
import './NotificationWindow.scss'
|
import './NotificationWindow.scss'
|
||||||
|
|
||||||
export default function NotificationWindow() {
|
export default function NotificationWindow() {
|
||||||
|
const { currentTheme, themeMode } = useThemeStore()
|
||||||
const [notification, setNotification] = useState<NotificationData | null>(null)
|
const [notification, setNotification] = useState<NotificationData | null>(null)
|
||||||
const [prevNotification, setPrevNotification] = useState<NotificationData | null>(null)
|
const [prevNotification, setPrevNotification] = useState<NotificationData | null>(null)
|
||||||
|
|
||||||
@@ -17,6 +19,12 @@ export default function NotificationWindow() {
|
|||||||
|
|
||||||
const notificationRef = useRef<NotificationData | null>(null)
|
const notificationRef = useRef<NotificationData | null>(null)
|
||||||
|
|
||||||
|
// 应用主题到通知窗口
|
||||||
|
useEffect(() => {
|
||||||
|
document.documentElement.setAttribute('data-theme', currentTheme)
|
||||||
|
document.documentElement.setAttribute('data-mode', themeMode)
|
||||||
|
}, [currentTheme, themeMode])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
notificationRef.current = notification
|
notificationRef.current = notification
|
||||||
}, [notification])
|
}, [notification])
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import { dialog } from '../services/ipc'
|
|||||||
import * as configService from '../services/config'
|
import * as configService from '../services/config'
|
||||||
import {
|
import {
|
||||||
Eye, EyeOff, FolderSearch, FolderOpen, Search, Copy,
|
Eye, EyeOff, FolderSearch, FolderOpen, Search, Copy,
|
||||||
RotateCcw, Trash2, Plug, Check, Sun, Moon,
|
RotateCcw, Trash2, Plug, Check, Sun, Moon, Monitor,
|
||||||
Palette, Database, Download, HardDrive, Info, RefreshCw, ChevronDown, Mic,
|
Palette, Database, Download, HardDrive, Info, RefreshCw, ChevronDown, Mic,
|
||||||
ShieldCheck, Fingerprint, Lock, KeyRound, Bell, Globe, BarChart2
|
ShieldCheck, Fingerprint, Lock, KeyRound, Bell, Globe, BarChart2
|
||||||
} from 'lucide-react'
|
} from 'lucide-react'
|
||||||
@@ -55,6 +55,14 @@ function SettingsPage() {
|
|||||||
|
|
||||||
const resetChatStore = useChatStore((state) => state.reset)
|
const resetChatStore = useChatStore((state) => state.reset)
|
||||||
const { currentTheme, themeMode, setTheme, setThemeMode } = useThemeStore()
|
const { currentTheme, themeMode, setTheme, setThemeMode } = useThemeStore()
|
||||||
|
const [systemDark, setSystemDark] = useState(() => window.matchMedia('(prefers-color-scheme: dark)').matches)
|
||||||
|
useEffect(() => {
|
||||||
|
const mq = window.matchMedia('(prefers-color-scheme: dark)')
|
||||||
|
const handler = (e: MediaQueryListEvent) => setSystemDark(e.matches)
|
||||||
|
mq.addEventListener('change', handler)
|
||||||
|
return () => mq.removeEventListener('change', handler)
|
||||||
|
}, [])
|
||||||
|
const effectiveMode = themeMode === 'system' ? (systemDark ? 'dark' : 'light') : themeMode
|
||||||
const clearAnalyticsStoreCache = useAnalyticsStore((state) => state.clearCache)
|
const clearAnalyticsStoreCache = useAnalyticsStore((state) => state.clearCache)
|
||||||
|
|
||||||
const [activeTab, setActiveTab] = useState<SettingsTab>('appearance')
|
const [activeTab, setActiveTab] = useState<SettingsTab>('appearance')
|
||||||
@@ -74,7 +82,7 @@ function SettingsPage() {
|
|||||||
const exportExcelColumnsDropdownRef = useRef<HTMLDivElement>(null)
|
const exportExcelColumnsDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
const exportConcurrencyDropdownRef = useRef<HTMLDivElement>(null)
|
const exportConcurrencyDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
const [cachePath, setCachePath] = useState('')
|
const [cachePath, setCachePath] = useState('')
|
||||||
const [weixinDllPath, setWeixinDllPath] = useState('')
|
|
||||||
const [logEnabled, setLogEnabled] = useState(false)
|
const [logEnabled, setLogEnabled] = useState(false)
|
||||||
const [whisperModelName, setWhisperModelName] = useState('base')
|
const [whisperModelName, setWhisperModelName] = useState('base')
|
||||||
const [whisperModelDir, setWhisperModelDir] = useState('')
|
const [whisperModelDir, setWhisperModelDir] = useState('')
|
||||||
@@ -82,10 +90,6 @@ function SettingsPage() {
|
|||||||
const [whisperDownloadProgress, setWhisperDownloadProgress] = useState(0)
|
const [whisperDownloadProgress, setWhisperDownloadProgress] = useState(0)
|
||||||
const [whisperProgressData, setWhisperProgressData] = useState<{ downloaded: number; total: number; speed: number }>({ downloaded: 0, total: 0, speed: 0 })
|
const [whisperProgressData, setWhisperProgressData] = useState<{ downloaded: number; total: number; speed: number }>({ downloaded: 0, total: 0, speed: 0 })
|
||||||
const [whisperModelStatus, setWhisperModelStatus] = useState<{ exists: boolean; modelPath?: string; tokensPath?: string } | null>(null)
|
const [whisperModelStatus, setWhisperModelStatus] = useState<{ exists: boolean; modelPath?: string; tokensPath?: string } | null>(null)
|
||||||
const [llamaModelStatus, setLlamaModelStatus] = useState<{ exists: boolean; path?: string; size?: number } | null>(null)
|
|
||||||
const [isLlamaDownloading, setIsLlamaDownloading] = useState(false)
|
|
||||||
const [llamaDownloadProgress, setLlamaDownloadProgress] = useState(0)
|
|
||||||
const [llamaProgressData, setLlamaProgressData] = useState<{ downloaded: number; total: number; speed: number }>({ downloaded: 0, total: 0, speed: 0 })
|
|
||||||
|
|
||||||
const formatBytes = (bytes: number) => {
|
const formatBytes = (bytes: number) => {
|
||||||
if (bytes === 0) return '0 B';
|
if (bytes === 0) return '0 B';
|
||||||
@@ -148,6 +152,7 @@ function SettingsPage() {
|
|||||||
const [httpApiEnabled, setHttpApiEnabled] = useState(false)
|
const [httpApiEnabled, setHttpApiEnabled] = useState(false)
|
||||||
const [httpApiPort, setHttpApiPort] = useState(5031)
|
const [httpApiPort, setHttpApiPort] = useState(5031)
|
||||||
const [httpApiRunning, setHttpApiRunning] = useState(false)
|
const [httpApiRunning, setHttpApiRunning] = useState(false)
|
||||||
|
const [httpApiMediaExportPath, setHttpApiMediaExportPath] = useState('')
|
||||||
const [isTogglingApi, setIsTogglingApi] = useState(false)
|
const [isTogglingApi, setIsTogglingApi] = useState(false)
|
||||||
const [showApiWarning, setShowApiWarning] = useState(false)
|
const [showApiWarning, setShowApiWarning] = useState(false)
|
||||||
|
|
||||||
@@ -169,6 +174,9 @@ function SettingsPage() {
|
|||||||
if (status.port) {
|
if (status.port) {
|
||||||
setHttpApiPort(status.port)
|
setHttpApiPort(status.port)
|
||||||
}
|
}
|
||||||
|
if (status.mediaExportPath) {
|
||||||
|
setHttpApiMediaExportPath(status.mediaExportPath)
|
||||||
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('检查 API 状态失败:', e)
|
console.error('检查 API 状态失败:', e)
|
||||||
}
|
}
|
||||||
@@ -250,7 +258,7 @@ function SettingsPage() {
|
|||||||
const savedPath = await configService.getDbPath()
|
const savedPath = await configService.getDbPath()
|
||||||
const savedWxid = await configService.getMyWxid()
|
const savedWxid = await configService.getMyWxid()
|
||||||
const savedCachePath = await configService.getCachePath()
|
const savedCachePath = await configService.getCachePath()
|
||||||
const savedWeixinDllPath = await configService.getWeixinDllPath()
|
|
||||||
const savedExportPath = await configService.getExportPath()
|
const savedExportPath = await configService.getExportPath()
|
||||||
const savedLogEnabled = await configService.getLogEnabled()
|
const savedLogEnabled = await configService.getLogEnabled()
|
||||||
const savedImageXorKey = await configService.getImageXorKey()
|
const savedImageXorKey = await configService.getImageXorKey()
|
||||||
@@ -279,7 +287,7 @@ function SettingsPage() {
|
|||||||
if (savedPath) setDbPath(savedPath)
|
if (savedPath) setDbPath(savedPath)
|
||||||
if (savedWxid) setWxid(savedWxid)
|
if (savedWxid) setWxid(savedWxid)
|
||||||
if (savedCachePath) setCachePath(savedCachePath)
|
if (savedCachePath) setCachePath(savedCachePath)
|
||||||
if (savedWeixinDllPath) setWeixinDllPath(savedWeixinDllPath)
|
|
||||||
|
|
||||||
const wxidConfig = savedWxid ? await configService.getWxidConfig(savedWxid) : null
|
const wxidConfig = savedWxid ? await configService.getWxidConfig(savedWxid) : null
|
||||||
const decryptKeyToUse = wxidConfig?.decryptKey ?? savedKey ?? ''
|
const decryptKeyToUse = wxidConfig?.decryptKey ?? savedKey ?? ''
|
||||||
@@ -324,8 +332,7 @@ function SettingsPage() {
|
|||||||
|
|
||||||
if (savedWhisperModelDir) setWhisperModelDir(savedWhisperModelDir)
|
if (savedWhisperModelDir) setWhisperModelDir(savedWhisperModelDir)
|
||||||
|
|
||||||
// Load Llama status after config
|
|
||||||
void checkLlamaModelStatus()
|
|
||||||
} catch (e: any) {
|
} catch (e: any) {
|
||||||
console.error('加载配置失败:', e)
|
console.error('加载配置失败:', e)
|
||||||
}
|
}
|
||||||
@@ -616,29 +623,7 @@ function SettingsPage() {
|
|||||||
await applyWxidSelection(selectedWxid)
|
await applyWxidSelection(selectedWxid)
|
||||||
}
|
}
|
||||||
|
|
||||||
const handleSelectWeixinDllPath = async () => {
|
|
||||||
try {
|
|
||||||
const result = await dialog.openFile({
|
|
||||||
title: '选择 Weixin.dll 文件',
|
|
||||||
properties: ['openFile'],
|
|
||||||
filters: [{ name: 'DLL', extensions: ['dll'] }]
|
|
||||||
})
|
|
||||||
if (!result.canceled && result.filePaths.length > 0) {
|
|
||||||
const selectedPath = result.filePaths[0]
|
|
||||||
setWeixinDllPath(selectedPath)
|
|
||||||
await configService.setWeixinDllPath(selectedPath)
|
|
||||||
showMessage('已选择 Weixin.dll 路径', true)
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
showMessage('选择 Weixin.dll 失败', false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const handleResetWeixinDllPath = async () => {
|
|
||||||
setWeixinDllPath('')
|
|
||||||
await configService.setWeixinDllPath('')
|
|
||||||
showMessage('已清空 Weixin.dll 路径', true)
|
|
||||||
}
|
|
||||||
const handleSelectCachePath = async () => {
|
const handleSelectCachePath = async () => {
|
||||||
try {
|
try {
|
||||||
const result = await dialog.openFile({ title: '选择缓存目录', properties: ['openDirectory'] })
|
const result = await dialog.openFile({ title: '选择缓存目录', properties: ['openDirectory'] })
|
||||||
@@ -663,7 +648,6 @@ function SettingsPage() {
|
|||||||
setWhisperModelDir(dir)
|
setWhisperModelDir(dir)
|
||||||
await configService.setWhisperModelDir(dir)
|
await configService.setWhisperModelDir(dir)
|
||||||
showMessage('已选择 Whisper 模型目录', true)
|
showMessage('已选择 Whisper 模型目录', true)
|
||||||
await checkLlamaModelStatus()
|
|
||||||
}
|
}
|
||||||
} catch (e: any) {
|
} catch (e: any) {
|
||||||
showMessage('选择目录失败', false)
|
showMessage('选择目录失败', false)
|
||||||
@@ -699,68 +683,6 @@ function SettingsPage() {
|
|||||||
const handleResetWhisperModelDir = async () => {
|
const handleResetWhisperModelDir = async () => {
|
||||||
setWhisperModelDir('')
|
setWhisperModelDir('')
|
||||||
await configService.setWhisperModelDir('')
|
await configService.setWhisperModelDir('')
|
||||||
await checkLlamaModelStatus()
|
|
||||||
}
|
|
||||||
|
|
||||||
const checkLlamaModelStatus = async () => {
|
|
||||||
try {
|
|
||||||
// @ts-ignore
|
|
||||||
const modelsPath = await window.electronAPI.llama?.getModelsPath()
|
|
||||||
if (!modelsPath) return
|
|
||||||
const modelName = "Qwen3-4B-Q4_K_M.gguf" // Hardcoded preset for now
|
|
||||||
const fullPath = `${modelsPath}\\${modelName}`
|
|
||||||
// @ts-ignore
|
|
||||||
const status = await window.electronAPI.llama?.getModelStatus(fullPath)
|
|
||||||
if (status) {
|
|
||||||
setLlamaModelStatus({
|
|
||||||
exists: status.exists,
|
|
||||||
path: status.path,
|
|
||||||
size: status.size
|
|
||||||
})
|
|
||||||
}
|
|
||||||
} catch (e) {
|
|
||||||
console.error("Check llama model status failed", e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
const handleLlamaProgress = (payload: { downloaded: number; total: number; speed: number }) => {
|
|
||||||
setLlamaProgressData(payload)
|
|
||||||
if (payload.total > 0) {
|
|
||||||
setLlamaDownloadProgress((payload.downloaded / payload.total) * 100)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// @ts-ignore
|
|
||||||
const removeListener = window.electronAPI.llama?.onDownloadProgress(handleLlamaProgress)
|
|
||||||
return () => {
|
|
||||||
if (typeof removeListener === 'function') removeListener()
|
|
||||||
}
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
const handleDownloadLlamaModel = async () => {
|
|
||||||
if (isLlamaDownloading) return
|
|
||||||
setIsLlamaDownloading(true)
|
|
||||||
setLlamaDownloadProgress(0)
|
|
||||||
try {
|
|
||||||
const modelUrl = "https://www.modelscope.cn/models/Qwen/Qwen3-4B-GGUF/resolve/master/Qwen3-4B-Q4_K_M.gguf"
|
|
||||||
// @ts-ignore
|
|
||||||
const modelsPath = await window.electronAPI.llama?.getModelsPath()
|
|
||||||
const modelName = "Qwen3-4B-Q4_K_M.gguf"
|
|
||||||
const fullPath = `${modelsPath}\\${modelName}`
|
|
||||||
|
|
||||||
// @ts-ignore
|
|
||||||
const result = await window.electronAPI.llama?.downloadModel(modelUrl, fullPath)
|
|
||||||
if (result?.success) {
|
|
||||||
showMessage('Qwen3 模型下载完成', true)
|
|
||||||
await checkLlamaModelStatus()
|
|
||||||
} else {
|
|
||||||
showMessage(`模型下载失败: ${result?.error || '未知错误'}`, false)
|
|
||||||
}
|
|
||||||
} catch (e: any) {
|
|
||||||
showMessage(`模型下载失败: ${e}`, false)
|
|
||||||
} finally {
|
|
||||||
setIsLlamaDownloading(false)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const handleAutoGetDbKey = async () => {
|
const handleAutoGetDbKey = async () => {
|
||||||
@@ -1011,11 +933,14 @@ function SettingsPage() {
|
|||||||
<button className={`mode-btn ${themeMode === 'dark' ? 'active' : ''}`} onClick={() => setThemeMode('dark')}>
|
<button className={`mode-btn ${themeMode === 'dark' ? 'active' : ''}`} onClick={() => setThemeMode('dark')}>
|
||||||
<Moon size={16} /> 深色
|
<Moon size={16} /> 深色
|
||||||
</button>
|
</button>
|
||||||
|
<button className={`mode-btn ${themeMode === 'system' ? 'active' : ''}`} onClick={() => setThemeMode('system')}>
|
||||||
|
<Monitor size={16} /> 跟随系统
|
||||||
|
</button>
|
||||||
</div>
|
</div>
|
||||||
<div className="theme-grid">
|
<div className="theme-grid">
|
||||||
{themes.map((theme) => (
|
{themes.map((theme) => (
|
||||||
<div key={theme.id} className={`theme-card ${currentTheme === theme.id ? 'active' : ''}`} onClick={() => setTheme(theme.id)}>
|
<div key={theme.id} className={`theme-card ${currentTheme === theme.id ? 'active' : ''}`} onClick={() => setTheme(theme.id)}>
|
||||||
<div className="theme-preview" style={{ background: themeMode === 'dark' ? 'linear-gradient(135deg, #1a1a1a 0%, #2a2a2a 100%)' : `linear-gradient(135deg, ${theme.bgColor} 0%, ${theme.bgColor}dd 100%)` }}>
|
<div className="theme-preview" style={{ background: effectiveMode === 'dark' ? 'linear-gradient(135deg, #1a1a1a 0%, #2a2a2a 100%)' : `linear-gradient(135deg, ${theme.bgColor} 0%, ${theme.bgColor}dd 100%)` }}>
|
||||||
<div className="theme-accent" style={{ background: theme.primaryColor }} />
|
<div className="theme-accent" style={{ background: theme.primaryColor }} />
|
||||||
</div>
|
</div>
|
||||||
<div className="theme-info">
|
<div className="theme-info">
|
||||||
@@ -1332,28 +1257,7 @@ function SettingsPage() {
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="form-group">
|
|
||||||
<label>Weixin.dll 路径 <span className="optional">(可选)</span></label>
|
|
||||||
<span className="form-hint">用于朋友圈在线图片原生解密,优先使用这里配置的 DLL</span>
|
|
||||||
<input
|
|
||||||
type="text"
|
|
||||||
placeholder="例如: D:\weixindata\Weixin\Weixin.dll"
|
|
||||||
value={weixinDllPath}
|
|
||||||
onChange={(e) => {
|
|
||||||
const value = e.target.value
|
|
||||||
setWeixinDllPath(value)
|
|
||||||
scheduleConfigSave('weixinDllPath', () => configService.setWeixinDllPath(value))
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
<div className="btn-row">
|
|
||||||
<button className="btn btn-secondary" onClick={handleSelectWeixinDllPath}>
|
|
||||||
<FolderOpen size={16} /> 浏览选择
|
|
||||||
</button>
|
|
||||||
<button className="btn btn-secondary" onClick={handleResetWeixinDllPath}>
|
|
||||||
清空
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="form-group">
|
<div className="form-group">
|
||||||
<label>账号 wxid</label>
|
<label>账号 wxid</label>
|
||||||
@@ -1480,7 +1384,7 @@ function SettingsPage() {
|
|||||||
<div className="tab-content">
|
<div className="tab-content">
|
||||||
<div className="form-group">
|
<div className="form-group">
|
||||||
<label>模型管理</label>
|
<label>模型管理</label>
|
||||||
<span className="form-hint">管理语音识别和 AI 对话模型</span>
|
<span className="form-hint">管理语音识别模型</span>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="form-group">
|
<div className="form-group">
|
||||||
@@ -1550,50 +1454,6 @@ function SettingsPage() {
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="form-group">
|
|
||||||
<label>AI 对话模型 (Llama)</label>
|
|
||||||
<span className="form-hint">用于 AI 助手对话功能</span>
|
|
||||||
<div className="setting-control vertical has-border">
|
|
||||||
<div className="model-status-card">
|
|
||||||
<div className="model-info">
|
|
||||||
<div className="model-name">Qwen3 4B (Preset) (~2.6GB)</div>
|
|
||||||
<div className="model-path">
|
|
||||||
{llamaModelStatus?.exists ? (
|
|
||||||
<span className="status-indicator success"><Check size={14} /> 已安装</span>
|
|
||||||
) : (
|
|
||||||
<span className="status-indicator warning">未安装</span>
|
|
||||||
)}
|
|
||||||
{llamaModelStatus?.path && <div className="path-text" title={llamaModelStatus.path}>{llamaModelStatus.path}</div>}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div className="model-actions">
|
|
||||||
{!llamaModelStatus?.exists && !isLlamaDownloading && (
|
|
||||||
<button
|
|
||||||
className="btn-download"
|
|
||||||
onClick={handleDownloadLlamaModel}
|
|
||||||
>
|
|
||||||
<Download size={16} /> 下载模型
|
|
||||||
</button>
|
|
||||||
)}
|
|
||||||
{isLlamaDownloading && (
|
|
||||||
<div className="download-status">
|
|
||||||
<div className="status-header">
|
|
||||||
<span className="percent">{Math.floor(llamaDownloadProgress)}%</span>
|
|
||||||
<span className="metrics">
|
|
||||||
{formatBytes(llamaProgressData.downloaded)} / {formatBytes(llamaProgressData.total)}
|
|
||||||
<span className="speed">({formatBytes(llamaProgressData.speed)}/s)</span>
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
<div className="progress-bar-mini">
|
|
||||||
<div className="fill" style={{ width: `${llamaDownloadProgress}%` }}></div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="form-group">
|
<div className="form-group">
|
||||||
<label>自动转文字</label>
|
<label>自动转文字</label>
|
||||||
<span className="form-hint">收到语音消息时自动转换为文字</span>
|
<span className="form-hint">收到语音消息时自动转换为文字</span>
|
||||||
@@ -2021,6 +1881,17 @@ function SettingsPage() {
|
|||||||
)}
|
)}
|
||||||
|
|
||||||
{/* API 安全警告弹窗 */}
|
{/* API 安全警告弹窗 */}
|
||||||
|
<div className="form-group">
|
||||||
|
<label>默认媒体导出目录</label>
|
||||||
|
<span className="form-hint">`/api/v1/messages` 在开启 `media=1` 时会把媒体保存到这里</span>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
className="field-input"
|
||||||
|
value={httpApiMediaExportPath || '未获取到目录'}
|
||||||
|
readOnly
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
{showApiWarning && (
|
{showApiWarning && (
|
||||||
<div className="modal-overlay" onClick={() => setShowApiWarning(false)}>
|
<div className="modal-overlay" onClick={() => setShowApiWarning(false)}>
|
||||||
<div className="api-warning-modal" onClick={(e) => e.stopPropagation()}>
|
<div className="api-warning-modal" onClick={(e) => e.stopPropagation()}>
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -1,108 +0,0 @@
|
|||||||
|
|
||||||
export interface ModelInfo {
|
|
||||||
name: string;
|
|
||||||
path: string;
|
|
||||||
downloadUrl?: string; // If it's a known preset
|
|
||||||
size?: number;
|
|
||||||
downloaded: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
export const PRESET_MODELS: ModelInfo[] = [
|
|
||||||
{
|
|
||||||
name: "Qwen3 4B (Preset)",
|
|
||||||
path: "Qwen3-4B-Q4_K_M.gguf",
|
|
||||||
downloadUrl: "https://www.modelscope.cn/models/Qwen/Qwen3-4B-GGUF/resolve/master/Qwen3-4B-Q4_K_M.gguf",
|
|
||||||
downloaded: false
|
|
||||||
}
|
|
||||||
];
|
|
||||||
|
|
||||||
class EngineService {
|
|
||||||
private onTokenCallback: ((token: string) => void) | null = null;
|
|
||||||
private onProgressCallback: ((percent: number) => void) | null = null;
|
|
||||||
private _removeTokenListener: (() => void) | null = null;
|
|
||||||
private _removeProgressListener: (() => void) | null = null;
|
|
||||||
|
|
||||||
constructor() {
|
|
||||||
// Initialize listeners
|
|
||||||
this._removeTokenListener = window.electronAPI.llama.onToken((token: string) => {
|
|
||||||
if (this.onTokenCallback) {
|
|
||||||
this.onTokenCallback(token);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
this._removeProgressListener = window.electronAPI.llama.onDownloadProgress((percent: number) => {
|
|
||||||
if (this.onProgressCallback) {
|
|
||||||
this.onProgressCallback(percent);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
public async checkModelExists(filename: string): Promise<boolean> {
|
|
||||||
const modelsPath = await window.electronAPI.llama.getModelsPath();
|
|
||||||
const fullPath = `${modelsPath}\\${filename}`; // Windows path separator
|
|
||||||
// We might need to handle path separator properly or let main process handle it
|
|
||||||
// Updated preload to take full path or handling in main?
|
|
||||||
// Let's rely on main process exposing join or just checking relative to models dir if implemented
|
|
||||||
// Actually main process `checkFileExists` takes a path.
|
|
||||||
// Let's assume we construct path here or Main helps.
|
|
||||||
// Better: getModelsPath returns the directory.
|
|
||||||
return await window.electronAPI.llama.checkFileExists(fullPath);
|
|
||||||
}
|
|
||||||
|
|
||||||
public async getModelsPath(): Promise<string> {
|
|
||||||
return await window.electronAPI.llama.getModelsPath();
|
|
||||||
}
|
|
||||||
|
|
||||||
public async loadModel(filename: string) {
|
|
||||||
const modelsPath = await this.getModelsPath();
|
|
||||||
const fullPath = `${modelsPath}\\${filename}`;
|
|
||||||
console.log("Loading model:", fullPath);
|
|
||||||
return await window.electronAPI.llama.loadModel(fullPath);
|
|
||||||
}
|
|
||||||
|
|
||||||
public async createSession(systemPrompt?: string) {
|
|
||||||
return await window.electronAPI.llama.createSession(systemPrompt);
|
|
||||||
}
|
|
||||||
|
|
||||||
public async chat(message: string, onToken: (token: string) => void, options?: { thinking?: boolean }) {
|
|
||||||
this.onTokenCallback = onToken;
|
|
||||||
return await window.electronAPI.llama.chat(message, options);
|
|
||||||
}
|
|
||||||
|
|
||||||
public async downloadModel(url: string, filename: string, onProgress: (percent: number) => void) {
|
|
||||||
const modelsPath = await this.getModelsPath();
|
|
||||||
const fullPath = `${modelsPath}\\${filename}`;
|
|
||||||
this.onProgressCallback = onProgress;
|
|
||||||
return await window.electronAPI.llama.downloadModel(url, fullPath);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 清除当前的回调函数引用
|
|
||||||
* 用于避免内存泄漏
|
|
||||||
*/
|
|
||||||
public clearCallbacks() {
|
|
||||||
this.onTokenCallback = null;
|
|
||||||
this.onProgressCallback = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 释放所有资源
|
|
||||||
* 包括事件监听器和回调引用
|
|
||||||
*/
|
|
||||||
public dispose() {
|
|
||||||
// 清除回调
|
|
||||||
this.clearCallbacks();
|
|
||||||
|
|
||||||
// 移除事件监听器
|
|
||||||
if (this._removeTokenListener) {
|
|
||||||
this._removeTokenListener();
|
|
||||||
this._removeTokenListener = null;
|
|
||||||
}
|
|
||||||
if (this._removeProgressListener) {
|
|
||||||
this._removeProgressListener();
|
|
||||||
this._removeProgressListener = null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export const engineService = new EngineService();
|
|
||||||
@@ -12,7 +12,7 @@ export const CONFIG_KEYS = {
|
|||||||
LAST_SESSION: 'lastSession',
|
LAST_SESSION: 'lastSession',
|
||||||
WINDOW_BOUNDS: 'windowBounds',
|
WINDOW_BOUNDS: 'windowBounds',
|
||||||
CACHE_PATH: 'cachePath',
|
CACHE_PATH: 'cachePath',
|
||||||
WEIXIN_DLL_PATH: 'weixinDllPath',
|
|
||||||
EXPORT_PATH: 'exportPath',
|
EXPORT_PATH: 'exportPath',
|
||||||
AGREEMENT_ACCEPTED: 'agreementAccepted',
|
AGREEMENT_ACCEPTED: 'agreementAccepted',
|
||||||
LOG_ENABLED: 'logEnabled',
|
LOG_ENABLED: 'logEnabled',
|
||||||
@@ -118,13 +118,13 @@ export async function setWxidConfig(wxid: string, configValue: WxidConfig): Prom
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 获取主题
|
// 获取主题
|
||||||
export async function getTheme(): Promise<'light' | 'dark'> {
|
export async function getTheme(): Promise<'light' | 'dark' | 'system'> {
|
||||||
const value = await config.get(CONFIG_KEYS.THEME)
|
const value = await config.get(CONFIG_KEYS.THEME)
|
||||||
return (value as 'light' | 'dark') || 'light'
|
return (value as 'light' | 'dark' | 'system') || 'light'
|
||||||
}
|
}
|
||||||
|
|
||||||
// 设置主题
|
// 设置主题
|
||||||
export async function setTheme(theme: 'light' | 'dark'): Promise<void> {
|
export async function setTheme(theme: 'light' | 'dark' | 'system'): Promise<void> {
|
||||||
await config.set(CONFIG_KEYS.THEME, theme)
|
await config.set(CONFIG_KEYS.THEME, theme)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -163,16 +163,7 @@ export async function setCachePath(path: string): Promise<void> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// 获取 Weixin.dll 路径
|
|
||||||
export async function getWeixinDllPath(): Promise<string | null> {
|
|
||||||
const value = await config.get(CONFIG_KEYS.WEIXIN_DLL_PATH)
|
|
||||||
return value as string | null
|
|
||||||
}
|
|
||||||
|
|
||||||
// 设置 Weixin.dll 路径
|
|
||||||
export async function setWeixinDllPath(path: string): Promise<void> {
|
|
||||||
await config.set(CONFIG_KEYS.WEIXIN_DLL_PATH, path)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 获取导出路径
|
// 获取导出路径
|
||||||
export async function getExportPath(): Promise<string | null> {
|
export async function getExportPath(): Promise<string | null> {
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ import { create } from 'zustand'
|
|||||||
import { persist } from 'zustand/middleware'
|
import { persist } from 'zustand/middleware'
|
||||||
|
|
||||||
export type ThemeId = 'cloud-dancer' | 'corundum-blue' | 'kiwi-green' | 'spicy-red' | 'teal-water'
|
export type ThemeId = 'cloud-dancer' | 'corundum-blue' | 'kiwi-green' | 'spicy-red' | 'teal-water'
|
||||||
export type ThemeMode = 'light' | 'dark'
|
export type ThemeMode = 'light' | 'dark' | 'system'
|
||||||
|
|
||||||
export interface ThemeInfo {
|
export interface ThemeInfo {
|
||||||
id: ThemeId
|
id: ThemeId
|
||||||
|
|||||||
47
src/types/electron.d.ts
vendored
47
src/types/electron.d.ts
vendored
@@ -85,6 +85,8 @@ export interface ElectronAPI {
|
|||||||
}>
|
}>
|
||||||
getContact: (username: string) => Promise<Contact | null>
|
getContact: (username: string) => Promise<Contact | null>
|
||||||
getContactAvatar: (username: string) => Promise<{ avatarUrl?: string; displayName?: string } | null>
|
getContactAvatar: (username: string) => Promise<{ avatarUrl?: string; displayName?: string } | null>
|
||||||
|
updateMessage: (sessionId: string, localId: number, createTime: number, newContent: string) => Promise<{ success: boolean; error?: string }>
|
||||||
|
deleteMessage: (sessionId: string, localId: number, createTime: number, dbPathHint?: string) => Promise<{ success: boolean; error?: string }>
|
||||||
resolveTransferDisplayNames: (chatroomId: string, payerUsername: string, receiverUsername: string) => Promise<{ payerName: string; receiverName: string }>
|
resolveTransferDisplayNames: (chatroomId: string, payerUsername: string, receiverUsername: string) => Promise<{ payerName: string; receiverName: string }>
|
||||||
getContacts: () => Promise<{
|
getContacts: () => Promise<{
|
||||||
success: boolean
|
success: boolean
|
||||||
@@ -273,6 +275,17 @@ export interface ElectronAPI {
|
|||||||
count?: number
|
count?: number
|
||||||
error?: string
|
error?: string
|
||||||
}>
|
}>
|
||||||
|
exportGroupMemberMessages: (
|
||||||
|
chatroomId: string,
|
||||||
|
memberUsername: string,
|
||||||
|
outputPath: string,
|
||||||
|
startTime?: number,
|
||||||
|
endTime?: number
|
||||||
|
) => Promise<{
|
||||||
|
success: boolean
|
||||||
|
count?: number
|
||||||
|
error?: string
|
||||||
|
}>
|
||||||
}
|
}
|
||||||
annualReport: {
|
annualReport: {
|
||||||
getAvailableYears: () => Promise<{
|
getAvailableYears: () => Promise<{
|
||||||
@@ -431,7 +444,7 @@ export interface ElectronAPI {
|
|||||||
success: boolean
|
success: boolean
|
||||||
error?: string
|
error?: string
|
||||||
}>
|
}>
|
||||||
exportContacts: (outputDir: string, options: { format: 'json' | 'csv' | 'vcf'; exportAvatars: boolean; contactTypes: { friends: boolean; groups: boolean; officials: boolean } }) => Promise<{
|
exportContacts: (outputDir: string, options: { format: 'json' | 'csv' | 'vcf'; exportAvatars: boolean; contactTypes: { friends: boolean; groups: boolean; officials: boolean }; selectedUsernames?: string[] }) => Promise<{
|
||||||
success: boolean
|
success: boolean
|
||||||
successCount?: number
|
successCount?: number
|
||||||
error?: string
|
error?: string
|
||||||
@@ -478,27 +491,37 @@ export interface ElectronAPI {
|
|||||||
}>
|
}>
|
||||||
debugResource: (url: string) => Promise<{ success: boolean; status?: number; headers?: any; error?: string }>
|
debugResource: (url: string) => Promise<{ success: boolean; status?: number; headers?: any; error?: string }>
|
||||||
proxyImage: (payload: { url: string; key?: string | number }) => Promise<{ success: boolean; dataUrl?: string; error?: string }>
|
proxyImage: (payload: { url: string; key?: string | number }) => Promise<{ success: boolean; dataUrl?: string; error?: string }>
|
||||||
|
downloadImage: (payload: { url: string; key?: string | number }) => Promise<{ success: boolean; data?: any; contentType?: string; error?: string }>
|
||||||
|
exportTimeline: (options: {
|
||||||
|
outputDir: string
|
||||||
|
format: 'json' | 'html'
|
||||||
|
usernames?: string[]
|
||||||
|
keyword?: string
|
||||||
|
exportMedia?: boolean
|
||||||
|
startTime?: number
|
||||||
|
endTime?: number
|
||||||
|
}) => Promise<{ success: boolean; filePath?: string; postCount?: number; mediaCount?: number; error?: string }>
|
||||||
|
onExportProgress: (callback: (payload: { current: number; total: number; status: string }) => void) => () => void
|
||||||
|
selectExportDir: () => Promise<{ canceled: boolean; filePath?: string }>
|
||||||
|
getSnsUsernames: () => Promise<{ success: boolean; usernames?: string[]; error?: string }>
|
||||||
}
|
}
|
||||||
llama: {
|
http: {
|
||||||
loadModel: (modelPath: string) => Promise<boolean>
|
start: (port?: number) => Promise<{ success: boolean; port?: number; error?: string }>
|
||||||
createSession: (systemPrompt?: string) => Promise<boolean>
|
stop: () => Promise<{ success: boolean }>
|
||||||
chat: (message: string) => Promise<{ success: boolean; response?: any; error?: string }>
|
status: () => Promise<{ running: boolean; port: number; mediaExportPath: string }>
|
||||||
downloadModel: (url: string, savePath: string) => Promise<void>
|
|
||||||
getModelsPath: () => Promise<string>
|
|
||||||
checkFileExists: (filePath: string) => Promise<boolean>
|
|
||||||
getModelStatus: (modelPath: string) => Promise<{ exists: boolean; path?: string; size?: number; error?: string }>
|
|
||||||
onToken: (callback: (token: string) => void) => () => void
|
|
||||||
onDownloadProgress: (callback: (payload: { downloaded: number; total: number; speed: number }) => void) => () => void
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface ExportOptions {
|
export interface ExportOptions {
|
||||||
format: 'chatlab' | 'chatlab-jsonl' | 'json' | 'html' | 'txt' | 'excel' | 'sql'
|
format: 'chatlab' | 'chatlab-jsonl' | 'json' | 'html' | 'txt' | 'excel' | 'weclone' | 'sql'
|
||||||
dateRange?: { start: number; end: number } | null
|
dateRange?: { start: number; end: number } | null
|
||||||
|
senderUsername?: string
|
||||||
|
fileNameSuffix?: string
|
||||||
exportMedia?: boolean
|
exportMedia?: boolean
|
||||||
exportAvatars?: boolean
|
exportAvatars?: boolean
|
||||||
exportImages?: boolean
|
exportImages?: boolean
|
||||||
exportVoices?: boolean
|
exportVoices?: boolean
|
||||||
|
exportVideos?: boolean
|
||||||
exportEmojis?: boolean
|
exportEmojis?: boolean
|
||||||
exportVoiceAsText?: boolean
|
exportVoiceAsText?: boolean
|
||||||
excelCompactColumns?: boolean
|
excelCompactColumns?: boolean
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ export interface ContactInfo {
|
|||||||
remark?: string
|
remark?: string
|
||||||
nickname?: string
|
nickname?: string
|
||||||
avatarUrl?: string
|
avatarUrl?: string
|
||||||
type: 'friend' | 'group' | 'official' | 'other'
|
type: 'friend' | 'group' | 'official' | 'former_friend' | 'other'
|
||||||
}
|
}
|
||||||
|
|
||||||
// 消息
|
// 消息
|
||||||
|
|||||||
47
src/types/sns.ts
Normal file
47
src/types/sns.ts
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
export interface SnsLivePhoto {
|
||||||
|
url: string
|
||||||
|
thumb: string
|
||||||
|
token?: string
|
||||||
|
key?: string
|
||||||
|
encIdx?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnsMedia {
|
||||||
|
url: string
|
||||||
|
thumb: string
|
||||||
|
md5?: string
|
||||||
|
token?: string
|
||||||
|
key?: string
|
||||||
|
encIdx?: string
|
||||||
|
livePhoto?: SnsLivePhoto
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnsComment {
|
||||||
|
id: string
|
||||||
|
nickname: string
|
||||||
|
content: string
|
||||||
|
refCommentId: string
|
||||||
|
refNickname?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnsPost {
|
||||||
|
id: string
|
||||||
|
username: string
|
||||||
|
nickname: string
|
||||||
|
avatarUrl?: string
|
||||||
|
createTime: number
|
||||||
|
contentDesc: string
|
||||||
|
type?: number
|
||||||
|
media: SnsMedia[]
|
||||||
|
likes: string[]
|
||||||
|
comments: SnsComment[]
|
||||||
|
rawXml?: string
|
||||||
|
linkTitle?: string
|
||||||
|
linkUrl?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SnsLinkCardData {
|
||||||
|
title: string
|
||||||
|
url: string
|
||||||
|
thumb?: string
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user