mirror of
https://github.com/hicccc77/WeFlow.git
synced 2026-03-25 15:25:50 +00:00
Compare commits
299 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7c9d0a39c3 | ||
|
|
a5777027b1 | ||
|
|
c3e911e6fa | ||
|
|
4d03110df2 | ||
|
|
8cb640f565 | ||
|
|
494bd4f539 | ||
|
|
38169691cd | ||
|
|
bd995bc736 | ||
|
|
6e05e74d5e | ||
|
|
d3a1db4efe | ||
|
|
a19f2a57c3 | ||
|
|
666a53f6ba | ||
|
|
b156a08f0d | ||
|
|
9c76aa2189 | ||
|
|
a54c95b6ac | ||
|
|
9cb0ada1b7 | ||
|
|
54378a132f | ||
|
|
4d1632a9b9 | ||
|
|
1eab835458 | ||
|
|
fcbc7fead8 | ||
|
|
ec783e4ccc | ||
|
|
b6f97b102c | ||
|
|
e4ce9a3bd7 | ||
|
|
64d5e721af | ||
|
|
d7419669d6 | ||
|
|
ff2f6799c8 | ||
|
|
2d573896f9 | ||
|
|
ab15190c44 | ||
|
|
551995df68 | ||
|
|
8483babd10 | ||
|
|
79648cd9d5 | ||
|
|
04d690dcf1 | ||
|
|
0b308803bf | ||
|
|
419d5aace3 | ||
|
|
84005f2d43 | ||
|
|
a166079084 | ||
|
|
a70d8fe6c8 | ||
|
|
34cd337146 | ||
|
|
c9216aabad | ||
|
|
79d6aef480 | ||
|
|
8134d62056 | ||
|
|
8664ebf6f5 | ||
|
|
7b832ac2ef | ||
|
|
5934fc33ce | ||
|
|
b6d10f79de | ||
|
|
f90822694f | ||
|
|
123a088a39 | ||
|
|
9283594dd0 | ||
|
|
638246e74d | ||
|
|
f506407f67 | ||
|
|
216f201327 | ||
|
|
a557f2ada3 | ||
|
|
e15e4cc3c8 | ||
|
|
2555c46b6d | ||
|
|
fdfd59fbdf | ||
|
|
0e1c3f9364 | ||
|
|
f9bb18d97f | ||
|
|
b7339b6a35 | ||
|
|
26abc30695 | ||
|
|
1f0f824b01 | ||
|
|
cb37f534ac | ||
|
|
50903b35cf | ||
|
|
c07ef66324 | ||
|
|
6bc802e77b | ||
|
|
898c86c23f | ||
|
|
7612353389 | ||
|
|
8b37f20b0f | ||
|
|
0054509ef2 | ||
|
|
e0f22f58c8 | ||
|
|
6f41cb34ed | ||
|
|
ddbb0c3b26 | ||
|
|
f40f885af3 | ||
|
|
5413d7e2c8 | ||
|
|
53f0e299e0 | ||
|
|
65365107f5 | ||
|
|
cffeeb26ec | ||
|
|
26d4751e80 | ||
|
|
b8120a5119 | ||
|
|
68a13cefc3 | ||
|
|
cd4b8f3702 | ||
|
|
c5956ba203 | ||
|
|
f456357e01 | ||
|
|
4ef821f45f | ||
|
|
912c78e9e9 | ||
|
|
bfcd154a25 | ||
|
|
a1c8ba48b0 | ||
|
|
f93369489d | ||
|
|
014f57f152 | ||
|
|
3f1eb58af4 | ||
|
|
97f0077e95 | ||
|
|
3d9b1b0f8c | ||
|
|
cf292ca9e2 | ||
|
|
97f14030de | ||
|
|
2cfe0d8ee8 | ||
|
|
a760f45823 | ||
|
|
baa949a301 | ||
|
|
c29bbab25f | ||
|
|
29981e1232 | ||
|
|
2d043cd929 | ||
|
|
d6dca0e5f7 | ||
|
|
d47166e6f9 | ||
|
|
6e3bb9e361 | ||
|
|
b8dbc3caf1 | ||
|
|
c1145c8f89 | ||
|
|
0cba8e6d89 | ||
|
|
f6f468dff3 | ||
|
|
04fc5f9104 | ||
|
|
3c9ab6763c | ||
|
|
f360333ab4 | ||
|
|
834aa6eecb | ||
|
|
2400cc8b55 | ||
|
|
e4ed7faca9 | ||
|
|
8012aa49ee | ||
|
|
7225358b91 | ||
|
|
39688e8e0c | ||
|
|
592ca6128f | ||
|
|
7cd27d8905 | ||
|
|
bca387c54b | ||
|
|
e7e4ffd53f | ||
|
|
04e0bf6b29 | ||
|
|
dadd9d799c | ||
|
|
b3aaea16f2 | ||
|
|
f3994a1a72 | ||
|
|
26fbfd2c98 | ||
|
|
3c51dee9a6 | ||
|
|
b9fa0cc215 | ||
|
|
21f748a2dc | ||
|
|
87fe130791 | ||
|
|
ff1bc279f2 | ||
|
|
77689ec528 | ||
|
|
5ea0b65905 | ||
|
|
eac6b053ee | ||
|
|
d52abfddbf | ||
|
|
8f2e403837 | ||
|
|
17c9436c30 | ||
|
|
9969c073e5 | ||
|
|
dc83297854 | ||
|
|
b6c9f2b32b | ||
|
|
e63f901478 | ||
|
|
893cdb4d92 | ||
|
|
d99ec05e81 | ||
|
|
c8f726eddc | ||
|
|
4e57a30c90 | ||
|
|
0a88275669 | ||
|
|
2a45cf1276 | ||
|
|
d63f1e0d79 | ||
|
|
f55507cd99 | ||
|
|
836b0f9df4 | ||
|
|
b09068f1f7 | ||
|
|
714a9400d5 | ||
|
|
13dd2fca21 | ||
|
|
5d1f834b61 | ||
|
|
3ca86224eb | ||
|
|
f10e974f36 | ||
|
|
76c40e4118 | ||
|
|
5307f55840 | ||
|
|
3405f26d10 | ||
|
|
85d82bfd09 | ||
|
|
e557ee224e | ||
|
|
88544c4a5d | ||
|
|
b66fc32068 | ||
|
|
7ac3c281a3 | ||
|
|
28616493ce | ||
|
|
d68e4fe880 | ||
|
|
1fd676d63e | ||
|
|
9f31ac0529 | ||
|
|
3c32ad5ca8 | ||
|
|
879d84b597 | ||
|
|
ab3551fb91 | ||
|
|
b9d1ea316f | ||
|
|
7762bd37c9 | ||
|
|
2e61902556 | ||
|
|
9e8072c337 | ||
|
|
827e77c9a3 | ||
|
|
3956989b67 | ||
|
|
33d7c243a7 | ||
|
|
a215886015 | ||
|
|
1d9e8aded0 | ||
|
|
b7e31c9cff | ||
|
|
4e9c81a93d | ||
|
|
9181ac5d34 | ||
|
|
3a10aeb23e | ||
|
|
178f9c4fdc | ||
|
|
4d647a9467 | ||
|
|
16cbc6adb1 | ||
|
|
7afb872bff | ||
|
|
7df6182e70 | ||
|
|
40efb04a36 | ||
|
|
1f03d35253 | ||
|
|
3efaed488a | ||
|
|
decdbf95f7 | ||
|
|
cccc712814 | ||
|
|
135f4819fb | ||
|
|
388923257b | ||
|
|
6918e359e8 | ||
|
|
d5b33c7e77 | ||
|
|
d37f53e120 | ||
|
|
26478217e7 | ||
|
|
a100f4ef97 | ||
|
|
91b746dc59 | ||
|
|
1817a847de | ||
|
|
7e99feae1e | ||
|
|
2977c45365 | ||
|
|
3b363a3efa | ||
|
|
e2b0bd44d9 | ||
|
|
cc26860504 | ||
|
|
54f3e0481f | ||
|
|
a61371c8ad | ||
|
|
fd6d5e4296 | ||
|
|
514a617c55 | ||
|
|
b47007ea0c | ||
|
|
6436c39c90 | ||
|
|
eb2f90e605 | ||
|
|
bdbb85175a | ||
|
|
a5e1bfe49a | ||
|
|
b3adb54651 | ||
|
|
07e7bce6a9 | ||
|
|
baa90242a6 | ||
|
|
787db0cec2 | ||
|
|
6359118132 | ||
|
|
49614bf6d8 | ||
|
|
0901e08c5c | ||
|
|
503a77c7cf | ||
|
|
0e3ab8e4d6 | ||
|
|
4452e4921c | ||
|
|
97c1aa582d | ||
|
|
076c008329 | ||
|
|
21d785dd3c | ||
|
|
348f6c81bf | ||
|
|
d5a2e2bb62 | ||
|
|
2b51e0659e | ||
|
|
3efca5e60c | ||
|
|
2f7b917f1c | ||
|
|
8623f86505 | ||
|
|
dc74641c19 | ||
|
|
db7817cc22 | ||
|
|
ada0f68182 | ||
|
|
fe806895f0 | ||
|
|
da137d0a8f | ||
|
|
93ebc3bce3 | ||
|
|
9f6e9eb9bc | ||
|
|
996b133a4f | ||
|
|
dd2602ea35 | ||
|
|
e5cf71b7c5 | ||
|
|
f2e4e21010 | ||
|
|
240514f1e5 | ||
|
|
d4c7e86e05 | ||
|
|
2876c7a539 | ||
|
|
32cdbece2c | ||
|
|
6e7e994cc6 | ||
|
|
d95040ffaf | ||
|
|
129dfbe1b6 | ||
|
|
f8afce6bfa | ||
|
|
0423f23b9c | ||
|
|
e3655631bb | ||
|
|
945802f772 | ||
|
|
be4d9b510d | ||
|
|
0853e049c8 | ||
|
|
dc12df0fcf | ||
|
|
82ba0344b9 | ||
|
|
e8babd48b6 | ||
|
|
7c0ed66dad | ||
|
|
9402483d87 | ||
|
|
650de55202 | ||
|
|
af99ab2029 | ||
|
|
87a2675236 | ||
|
|
25f1256baa | ||
|
|
f83a37e714 | ||
|
|
1aecfb369b | ||
|
|
436b090e26 | ||
|
|
c0f2620542 | ||
|
|
72e2d82158 | ||
|
|
095c8f0db6 | ||
|
|
afa3e089b1 | ||
|
|
11969ea2d4 | ||
|
|
6707be2200 | ||
|
|
f97e102dbd | ||
|
|
3637864f9a | ||
|
|
6eabd707f8 | ||
|
|
b96a47fe29 | ||
|
|
b7eb19aad6 | ||
|
|
2e41a03c96 | ||
|
|
3151f79ee7 | ||
|
|
e7c93ea2f7 | ||
|
|
f09ab1bbcc | ||
|
|
e6a0726b8d | ||
|
|
cada002587 | ||
|
|
38e87b8cbf | ||
|
|
bd94ba7b1a | ||
|
|
756ee03aa0 | ||
|
|
76aa875085 | ||
|
|
16fa8510e6 | ||
|
|
b587e6bd6f | ||
|
|
13cc3751b5 | ||
|
|
ba65c5f3ad | ||
|
|
cfd7635323 | ||
|
|
895249940c | ||
|
|
6b85d8a5f1 | ||
|
|
5c1773efac |
57
.github/workflows/release.yml
vendored
57
.github/workflows/release.yml
vendored
@@ -21,48 +21,41 @@ jobs:
|
|||||||
- name: Install Node.js
|
- name: Install Node.js
|
||||||
uses: actions/setup-node@v4
|
uses: actions/setup-node@v4
|
||||||
with:
|
with:
|
||||||
node-version: 20
|
node-version: 22.12
|
||||||
cache: 'npm'
|
cache: 'npm'
|
||||||
|
|
||||||
- name: Install Dependencies
|
- name: Install Dependencies
|
||||||
run: npm install
|
run: npm ci
|
||||||
|
|
||||||
|
- name: Sync version with tag
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
VERSION=${GITHUB_REF_NAME#v}
|
||||||
|
echo "Syncing package.json version to $VERSION"
|
||||||
|
npm version $VERSION --no-git-tag-version --allow-same-version
|
||||||
|
|
||||||
- name: Build Frontend & Type Check
|
- name: Build Frontend & Type Check
|
||||||
run: |
|
run: |
|
||||||
npx tsc
|
npx tsc
|
||||||
npx vite build
|
npx vite build
|
||||||
|
|
||||||
- name: Build Changelog
|
|
||||||
id: build_changelog
|
|
||||||
uses: mikepenz/release-changelog-builder-action@v4
|
|
||||||
with:
|
|
||||||
outputFile: "release-notes.md"
|
|
||||||
configurationJson: |
|
|
||||||
{
|
|
||||||
"template": "# v${{ github.ref_name }} 版本发布\n\n{{CHANGELOG}}\n\n---\n> 此更新由系统自动构建",
|
|
||||||
"categories": [
|
|
||||||
{
|
|
||||||
"title": "## 新功能",
|
|
||||||
"filter": { "pattern": "^feat:.*", "flags": "i" }
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"title": "## 修复",
|
|
||||||
"filter": { "pattern": "^fix:.*", "flags": "i" }
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"title": "## 性能与维护",
|
|
||||||
"filter": { "pattern": "^(chore|docs|perf|refactor):.*", "flags": "i" }
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"ignore_labels": [],
|
|
||||||
"commitMode": true,
|
|
||||||
"empty_summary": "## 更新详情\n- 常规代码优化与维护"
|
|
||||||
}
|
|
||||||
env:
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
|
|
||||||
- name: Package and Publish
|
- name: Package and Publish
|
||||||
env:
|
env:
|
||||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
run: |
|
run: |
|
||||||
npx electron-builder --publish always "-c.releaseInfo.releaseNotesFile=release-notes.md"
|
npx electron-builder --publish always
|
||||||
|
|
||||||
|
- name: Update Release Notes
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
cat <<EOF > release_notes.md
|
||||||
|
## 更新日志
|
||||||
|
修复了一些已知问题
|
||||||
|
|
||||||
|
## 查看更多日志/获取最新动态
|
||||||
|
[点击加入 Telegram 频道](https://t.me/weflow_cc)
|
||||||
|
EOF
|
||||||
|
|
||||||
|
gh release edit "$GITHUB_REF_NAME" --notes-file release_notes.md
|
||||||
|
|||||||
9
.gitignore
vendored
9
.gitignore
vendored
@@ -13,6 +13,7 @@ dist
|
|||||||
dist-electron
|
dist-electron
|
||||||
dist-ssr
|
dist-ssr
|
||||||
*.local
|
*.local
|
||||||
|
test/
|
||||||
|
|
||||||
# Editor directories and files
|
# Editor directories and files
|
||||||
.vscode/*
|
.vscode/*
|
||||||
@@ -42,6 +43,10 @@ release
|
|||||||
# OS
|
# OS
|
||||||
Thumbs.db
|
Thumbs.db
|
||||||
|
|
||||||
|
# Electron dev cache
|
||||||
|
.electron/
|
||||||
|
.cache/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# 忽略 Visual Studio 临时文件夹
|
# 忽略 Visual Studio 临时文件夹
|
||||||
@@ -51,3 +56,7 @@ Thumbs.db
|
|||||||
*.aps
|
*.aps
|
||||||
|
|
||||||
wcdb/
|
wcdb/
|
||||||
|
*info
|
||||||
|
概述.md
|
||||||
|
chatlab-format.md
|
||||||
|
*.bak
|
||||||
64
README.md
64
README.md
@@ -20,21 +20,41 @@ WeFlow 是一个**完全本地**的微信**实时**聊天记录查看、分析
|
|||||||
<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">
|
||||||
</a>
|
</a>
|
||||||
<a href="https://github.com/hicccc77/WeFlow/blob/main/LICENSE">
|
<a href="https://t.me/weflow_cc">
|
||||||
<img src="https://img.shields.io/github/license/hicccc77/WeFlow?style=flat-square" alt="License">
|
<img src="https://img.shields.io/badge/Telegram%20频道-0088cc?style=flat-square&logo=telegram&logoColor=0088cc&labelColor=white" alt="Telegram">
|
||||||
</a>
|
</a>
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||
|
|
||||||
> [!TIP]
|
> [!TIP]
|
||||||
> 如果导出聊天记录后,想深入分析聊天内容可以试试 [ChatLab](https://chatlab.fun/)
|
> 如果导出聊天记录后,想深入分析聊天内容可以试试 [ChatLab](https://chatlab.fun/)
|
||||||
|
|
||||||
|
> [!NOTE]
|
||||||
|
> 仅支持微信 **4.0 及以上**版本,确保你的微信版本符合要求
|
||||||
|
|
||||||
## 主要功能
|
## 主要功能
|
||||||
|
|
||||||
- 本地实时查看聊天记录
|
- 本地实时查看聊天记录
|
||||||
- 统计分析与群聊画像
|
- 统计分析与群聊画像
|
||||||
- 年度报告与可视化概览
|
- 年度报告与可视化概览
|
||||||
- 导出聊天记录为 HTML 等格式
|
- 导出聊天记录为 HTML 等格式
|
||||||
- 本地解密与数据库管理
|
- HTTP API 接口(供开发者集成)
|
||||||
|
|
||||||
|
|
||||||
|
## HTTP API
|
||||||
|
|
||||||
|
> [!WARNING]
|
||||||
|
> 此功能目前处于早期阶段,接口可能会有变动,请等待后续更新完善。
|
||||||
|
|
||||||
|
WeFlow 提供本地 HTTP API 服务,支持通过接口查询消息数据,可用于与其他工具集成或二次开发。
|
||||||
|
|
||||||
|
- **启用方式**:设置 → API 服务 → 启动服务
|
||||||
|
- **默认端口**:5031
|
||||||
|
- **访问地址**:`http://127.0.0.1:5031`
|
||||||
|
- **支持格式**:原始 JSON 或 [ChatLab](https://chatlab.fun/) 标准格式
|
||||||
|
|
||||||
|
📖 完整接口文档:[点击查看](docs/HTTP-API.md)
|
||||||
|
|
||||||
|
|
||||||
## 快速开始
|
## 快速开始
|
||||||
|
|
||||||
@@ -61,38 +81,18 @@ npm run build
|
|||||||
|
|
||||||
打包产物在 `release` 目录下。
|
打包产物在 `release` 目录下。
|
||||||
|
|
||||||
## 技术栈
|
|
||||||
|
|
||||||
- **前端**: React 19 + TypeScript + Zustand
|
|
||||||
- **桌面**: Electron 39
|
|
||||||
- **构建**: Vite + electron-builder
|
|
||||||
- **数据库**: better-sqlite3 + WCDB DLL
|
|
||||||
- **样式**: SCSS + CSS Variables
|
|
||||||
|
|
||||||
## 项目结构
|
|
||||||
|
|
||||||
```
|
|
||||||
WeFlow/
|
|
||||||
├── electron/ # Electron 主进程
|
|
||||||
│ ├── main.ts # 主进程入口
|
|
||||||
│ ├── preload.ts # 预加载脚本
|
|
||||||
│ └── services/ # 后端服务
|
|
||||||
│ ├── chatService.ts # 聊天数据服务
|
|
||||||
│ ├── wcdbService.ts # 数据库服务
|
|
||||||
│ └── ...
|
|
||||||
├── src/ # React 前端
|
|
||||||
│ ├── components/ # 通用组件
|
|
||||||
│ ├── pages/ # 页面组件
|
|
||||||
│ ├── stores/ # Zustand 状态管理
|
|
||||||
│ ├── services/ # 前端服务
|
|
||||||
│ └── types/ # TypeScript 类型定义
|
|
||||||
├── public/ # 静态资源
|
|
||||||
└── resources/ # 打包资源
|
|
||||||
```
|
|
||||||
|
|
||||||
## 致谢
|
## 致谢
|
||||||
|
|
||||||
- [miyu](https://github.com/ILoveBingLu/miyu) 为本项目提供了基础框架
|
- [密语 CipherTalk](https://github.com/ILoveBingLu/miyu) 为本项目提供了基础框架
|
||||||
|
|
||||||
|
## 支持我们
|
||||||
|
|
||||||
|
如果 WeFlow 确实帮到了你,可以考虑请我们喝杯咖啡:
|
||||||
|
|
||||||
|
|
||||||
|
> TRC20 **Address:** `TZCtAw8CaeARWZBfvjidCnTcfnAtf6nvS6`
|
||||||
|
|
||||||
|
|
||||||
## Star History
|
## Star History
|
||||||
@@ -111,6 +111,4 @@ WeFlow/
|
|||||||
|
|
||||||
**请负责任地使用本工具,遵守相关法律法规**
|
**请负责任地使用本工具,遵守相关法律法规**
|
||||||
|
|
||||||
我们总是在向前走,却很少有机会回头看看
|
|
||||||
|
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
312
docs/HTTP-API.md
Normal file
312
docs/HTTP-API.md
Normal file
@@ -0,0 +1,312 @@
|
|||||||
|
# WeFlow HTTP API 接口文档
|
||||||
|
|
||||||
|
WeFlow 提供 HTTP API 服务,支持通过 HTTP 接口查询消息数据,支持 [ChatLab](https://github.com/nichuanfang/chatlab-format) 标准化格式输出。
|
||||||
|
|
||||||
|
## 启用 API 服务
|
||||||
|
|
||||||
|
在设置页面 → API 服务 → 点击「启动服务」按钮。
|
||||||
|
|
||||||
|
默认端口:`5031`
|
||||||
|
|
||||||
|
## 基础地址
|
||||||
|
|
||||||
|
```
|
||||||
|
http://127.0.0.1:5031
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 接口列表
|
||||||
|
|
||||||
|
### 1. 健康检查
|
||||||
|
|
||||||
|
检查 API 服务是否正常运行。
|
||||||
|
|
||||||
|
**请求**
|
||||||
|
```
|
||||||
|
GET /health
|
||||||
|
```
|
||||||
|
|
||||||
|
**响应**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"status": "ok"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 2. 获取消息列表
|
||||||
|
|
||||||
|
获取指定会话的消息,支持 ChatLab 格式输出。
|
||||||
|
|
||||||
|
**请求**
|
||||||
|
```
|
||||||
|
GET /api/v1/messages
|
||||||
|
```
|
||||||
|
|
||||||
|
**参数**
|
||||||
|
|
||||||
|
| 参数名 | 类型 | 必填 | 说明 |
|
||||||
|
|--------|------|------|------|
|
||||||
|
| `talker` | string | ✅ | 会话 ID(wxid 或群 ID) |
|
||||||
|
| `limit` | number | ❌ | 返回数量限制,默认 100 |
|
||||||
|
| `offset` | number | ❌ | 偏移量,用于分页,默认 0 |
|
||||||
|
| `start` | string | ❌ | 开始时间,格式 YYYYMMDD |
|
||||||
|
| `end` | string | ❌ | 结束时间,格式 YYYYMMDD |
|
||||||
|
| `chatlab` | string | ❌ | 设为 `1` 则输出 ChatLab 格式 |
|
||||||
|
| `format` | string | ❌ | 输出格式:`json`(默认)或 `chatlab` |
|
||||||
|
|
||||||
|
**示例请求**
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 获取消息(原始格式)
|
||||||
|
GET http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&limit=50
|
||||||
|
|
||||||
|
# 获取消息(ChatLab 格式)
|
||||||
|
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
|
||||||
|
```
|
||||||
|
|
||||||
|
**响应(原始格式)**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"success": true,
|
||||||
|
"talker": "wxid_xxx",
|
||||||
|
"count": 50,
|
||||||
|
"hasMore": true,
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"localId": 123,
|
||||||
|
"talker": "wxid_xxx",
|
||||||
|
"type": 1,
|
||||||
|
"content": "消息内容",
|
||||||
|
"createTime": 1738713600000,
|
||||||
|
"isSelf": false,
|
||||||
|
"sender": "wxid_sender"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
**响应(ChatLab 格式)**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"chatlab": {
|
||||||
|
"version": "0.0.2",
|
||||||
|
"exportedAt": 1738713600000,
|
||||||
|
"generator": "WeFlow",
|
||||||
|
"description": "Exported from WeFlow"
|
||||||
|
},
|
||||||
|
"meta": {
|
||||||
|
"name": "会话名称",
|
||||||
|
"platform": "wechat",
|
||||||
|
"type": "private",
|
||||||
|
"ownerId": "wxid_me"
|
||||||
|
},
|
||||||
|
"members": [
|
||||||
|
{
|
||||||
|
"platformId": "wxid_xxx",
|
||||||
|
"accountName": "用户名",
|
||||||
|
"groupNickname": "群昵称"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"sender": "wxid_xxx",
|
||||||
|
"accountName": "用户名",
|
||||||
|
"timestamp": 1738713600000,
|
||||||
|
"type": 0,
|
||||||
|
"content": "消息内容"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 3. 获取会话列表
|
||||||
|
|
||||||
|
获取所有会话列表。
|
||||||
|
|
||||||
|
**请求**
|
||||||
|
```
|
||||||
|
GET /api/v1/sessions
|
||||||
|
```
|
||||||
|
|
||||||
|
**参数**
|
||||||
|
|
||||||
|
| 参数名 | 类型 | 必填 | 说明 |
|
||||||
|
|--------|------|------|------|
|
||||||
|
| `keyword` | string | ❌ | 搜索关键词,匹配会话名或 ID |
|
||||||
|
| `limit` | number | ❌ | 返回数量限制,默认 100 |
|
||||||
|
|
||||||
|
**示例请求**
|
||||||
|
```bash
|
||||||
|
GET http://127.0.0.1:5031/api/v1/sessions
|
||||||
|
|
||||||
|
GET http://127.0.0.1:5031/api/v1/sessions?keyword=工作群&limit=20
|
||||||
|
```
|
||||||
|
|
||||||
|
**响应**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"success": true,
|
||||||
|
"count": 50,
|
||||||
|
"total": 100,
|
||||||
|
"sessions": [
|
||||||
|
{
|
||||||
|
"username": "wxid_xxx",
|
||||||
|
"displayName": "用户名",
|
||||||
|
"lastMessage": "最后一条消息",
|
||||||
|
"lastTime": 1738713600000,
|
||||||
|
"unreadCount": 0
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 4. 获取联系人列表
|
||||||
|
|
||||||
|
获取所有联系人信息。
|
||||||
|
|
||||||
|
**请求**
|
||||||
|
```
|
||||||
|
GET /api/v1/contacts
|
||||||
|
```
|
||||||
|
|
||||||
|
**参数**
|
||||||
|
|
||||||
|
| 参数名 | 类型 | 必填 | 说明 |
|
||||||
|
|--------|------|------|------|
|
||||||
|
| `keyword` | string | ❌ | 搜索关键词 |
|
||||||
|
| `limit` | number | ❌ | 返回数量限制,默认 100 |
|
||||||
|
|
||||||
|
**示例请求**
|
||||||
|
```bash
|
||||||
|
GET http://127.0.0.1:5031/api/v1/contacts
|
||||||
|
|
||||||
|
GET http://127.0.0.1:5031/api/v1/contacts?keyword=张三
|
||||||
|
```
|
||||||
|
|
||||||
|
**响应**
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"success": true,
|
||||||
|
"count": 50,
|
||||||
|
"contacts": [
|
||||||
|
{
|
||||||
|
"userName": "wxid_xxx",
|
||||||
|
"alias": "微信号",
|
||||||
|
"nickName": "昵称",
|
||||||
|
"remark": "备注名"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ChatLab 格式说明
|
||||||
|
|
||||||
|
ChatLab 是一种标准化的聊天记录交换格式,版本 0.0.2。
|
||||||
|
|
||||||
|
### 消息类型映射
|
||||||
|
|
||||||
|
| ChatLab Type | 值 | 说明 |
|
||||||
|
|--------------|-----|------|
|
||||||
|
| TEXT | 0 | 文本消息 |
|
||||||
|
| IMAGE | 1 | 图片 |
|
||||||
|
| VOICE | 2 | 语音 |
|
||||||
|
| VIDEO | 3 | 视频 |
|
||||||
|
| FILE | 4 | 文件 |
|
||||||
|
| EMOJI | 5 | 表情 |
|
||||||
|
| LINK | 7 | 链接 |
|
||||||
|
| LOCATION | 8 | 位置 |
|
||||||
|
| RED_PACKET | 20 | 红包 |
|
||||||
|
| TRANSFER | 21 | 转账 |
|
||||||
|
| CALL | 23 | 通话 |
|
||||||
|
| SYSTEM | 80 | 系统消息 |
|
||||||
|
| RECALL | 81 | 撤回消息 |
|
||||||
|
| OTHER | 99 | 其他 |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 使用示例
|
||||||
|
|
||||||
|
### PowerShell
|
||||||
|
|
||||||
|
```powershell
|
||||||
|
# 健康检查
|
||||||
|
Invoke-RestMethod http://127.0.0.1:5031/health
|
||||||
|
|
||||||
|
# 获取会话列表
|
||||||
|
Invoke-RestMethod http://127.0.0.1:5031/api/v1/sessions
|
||||||
|
|
||||||
|
# 获取消息
|
||||||
|
Invoke-RestMethod "http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&limit=10"
|
||||||
|
|
||||||
|
# 获取 ChatLab 格式
|
||||||
|
Invoke-RestMethod "http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&chatlab=1" | ConvertTo-Json -Depth 10
|
||||||
|
```
|
||||||
|
|
||||||
|
### cURL
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 健康检查
|
||||||
|
curl http://127.0.0.1:5031/health
|
||||||
|
|
||||||
|
# 获取会话列表
|
||||||
|
curl http://127.0.0.1:5031/api/v1/sessions
|
||||||
|
|
||||||
|
# 获取消息(ChatLab 格式)
|
||||||
|
curl "http://127.0.0.1:5031/api/v1/messages?talker=wxid_xxx&chatlab=1"
|
||||||
|
```
|
||||||
|
|
||||||
|
### Python
|
||||||
|
|
||||||
|
```python
|
||||||
|
import requests
|
||||||
|
|
||||||
|
BASE_URL = "http://127.0.0.1:5031"
|
||||||
|
|
||||||
|
# 获取会话列表
|
||||||
|
sessions = requests.get(f"{BASE_URL}/api/v1/sessions").json()
|
||||||
|
print(sessions)
|
||||||
|
|
||||||
|
# 获取消息
|
||||||
|
messages = requests.get(f"{BASE_URL}/api/v1/messages", params={
|
||||||
|
"talker": "wxid_xxx",
|
||||||
|
"limit": 100,
|
||||||
|
"chatlab": 1
|
||||||
|
}).json()
|
||||||
|
print(messages)
|
||||||
|
```
|
||||||
|
|
||||||
|
### JavaScript / Node.js
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
const BASE_URL = "http://127.0.0.1:5031";
|
||||||
|
|
||||||
|
// 获取会话列表
|
||||||
|
const sessions = await fetch(`${BASE_URL}/api/v1/sessions`).then(r => r.json());
|
||||||
|
console.log(sessions);
|
||||||
|
|
||||||
|
// 获取消息(ChatLab 格式)
|
||||||
|
const messages = await fetch(`${BASE_URL}/api/v1/messages?talker=wxid_xxx&chatlab=1`)
|
||||||
|
.then(r => r.json());
|
||||||
|
console.log(messages);
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 注意事项
|
||||||
|
|
||||||
|
1. API 仅监听本地地址 `127.0.0.1`,不对外网开放
|
||||||
|
2. 需要先连接数据库才能查询数据
|
||||||
|
3. 时间参数格式为 `YYYYMMDD`(如 20260205)
|
||||||
|
4. 支持 CORS,可从浏览器前端直接调用
|
||||||
45
electron/dualReportWorker.ts
Normal file
45
electron/dualReportWorker.ts
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
import { parentPort, workerData } from 'worker_threads'
|
||||||
|
import { wcdbService } from './services/wcdbService'
|
||||||
|
import { dualReportService } from './services/dualReportService'
|
||||||
|
|
||||||
|
interface WorkerConfig {
|
||||||
|
year: number
|
||||||
|
friendUsername: string
|
||||||
|
dbPath: string
|
||||||
|
decryptKey: string
|
||||||
|
myWxid: string
|
||||||
|
resourcesPath?: string
|
||||||
|
userDataPath?: string
|
||||||
|
logEnabled?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
const config = workerData as WorkerConfig
|
||||||
|
process.env.WEFLOW_WORKER = '1'
|
||||||
|
if (config.resourcesPath) {
|
||||||
|
process.env.WCDB_RESOURCES_PATH = config.resourcesPath
|
||||||
|
}
|
||||||
|
|
||||||
|
wcdbService.setPaths(config.resourcesPath || '', config.userDataPath || '')
|
||||||
|
wcdbService.setLogEnabled(config.logEnabled === true)
|
||||||
|
|
||||||
|
async function run() {
|
||||||
|
const result = await dualReportService.generateReportWithConfig({
|
||||||
|
year: config.year,
|
||||||
|
friendUsername: config.friendUsername,
|
||||||
|
dbPath: config.dbPath,
|
||||||
|
decryptKey: config.decryptKey,
|
||||||
|
wxid: config.myWxid,
|
||||||
|
onProgress: (status: string, progress: number) => {
|
||||||
|
parentPort?.postMessage({
|
||||||
|
type: 'dualReport:progress',
|
||||||
|
data: { status, progress }
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
parentPort?.postMessage({ type: 'dualReport:result', data: result })
|
||||||
|
}
|
||||||
|
|
||||||
|
run().catch((err) => {
|
||||||
|
parentPort?.postMessage({ type: 'dualReport:error', error: String(err) })
|
||||||
|
})
|
||||||
@@ -62,6 +62,12 @@ function isThumbnailDat(fileName: string): boolean {
|
|||||||
return fileName.includes('.t.dat') || fileName.includes('_t.dat')
|
return fileName.includes('.t.dat') || fileName.includes('_t.dat')
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function isHdDat(fileName: string): boolean {
|
||||||
|
const lower = fileName.toLowerCase()
|
||||||
|
const base = lower.endsWith('.dat') ? lower.slice(0, -4) : lower
|
||||||
|
return base.endsWith('_hd') || base.endsWith('_h')
|
||||||
|
}
|
||||||
|
|
||||||
function walkForDat(
|
function walkForDat(
|
||||||
root: string,
|
root: string,
|
||||||
datName: string,
|
datName: string,
|
||||||
@@ -101,6 +107,8 @@ function walkForDat(
|
|||||||
if (!isLikelyImageDatBase(baseLower)) continue
|
if (!isLikelyImageDatBase(baseLower)) continue
|
||||||
if (!hasXVariant(baseLower)) continue
|
if (!hasXVariant(baseLower)) continue
|
||||||
if (!matchesDatName(lower, datName)) continue
|
if (!matchesDatName(lower, datName)) continue
|
||||||
|
// 排除高清图片格式 (_hd, _h)
|
||||||
|
if (isHdDat(lower)) continue
|
||||||
matchedBases.add(baseLower)
|
matchedBases.add(baseLower)
|
||||||
const isThumb = isThumbnailDat(lower)
|
const isThumb = isThumbnailDat(lower)
|
||||||
if (!allowThumbnail && isThumb) continue
|
if (!allowThumbnail && isThumb) continue
|
||||||
|
|||||||
728
electron/main.ts
728
electron/main.ts
@@ -1,6 +1,7 @@
|
|||||||
import { app, BrowserWindow, ipcMain, nativeTheme } from 'electron'
|
import './preload-env'
|
||||||
|
import { app, BrowserWindow, ipcMain, nativeTheme, session } from 'electron'
|
||||||
import { Worker } from 'worker_threads'
|
import { Worker } from 'worker_threads'
|
||||||
import { join } from 'path'
|
import { join, dirname } from 'path'
|
||||||
import { autoUpdater } from 'electron-updater'
|
import { autoUpdater } from 'electron-updater'
|
||||||
import { readFile, writeFile, mkdir } from 'fs/promises'
|
import { readFile, writeFile, mkdir } from 'fs/promises'
|
||||||
import { existsSync } from 'fs'
|
import { existsSync } from 'fs'
|
||||||
@@ -13,8 +14,17 @@ import { imagePreloadService } from './services/imagePreloadService'
|
|||||||
import { analyticsService } from './services/analyticsService'
|
import { analyticsService } from './services/analyticsService'
|
||||||
import { groupAnalyticsService } from './services/groupAnalyticsService'
|
import { groupAnalyticsService } from './services/groupAnalyticsService'
|
||||||
import { annualReportService } from './services/annualReportService'
|
import { annualReportService } from './services/annualReportService'
|
||||||
import { exportService, ExportOptions } from './services/exportService'
|
import { exportService, ExportOptions, ExportProgress } from './services/exportService'
|
||||||
import { KeyService } from './services/keyService'
|
import { KeyService } from './services/keyService'
|
||||||
|
import { voiceTranscribeService } from './services/voiceTranscribeService'
|
||||||
|
import { videoService } from './services/videoService'
|
||||||
|
import { snsService } from './services/snsService'
|
||||||
|
import { contactExportService } from './services/contactExportService'
|
||||||
|
import { windowsHelloService } from './services/windowsHelloService'
|
||||||
|
import { llamaService } from './services/llamaService'
|
||||||
|
import { registerNotificationHandlers, showNotification } from './windows/notificationWindow'
|
||||||
|
import { httpService } from './services/httpService'
|
||||||
|
|
||||||
|
|
||||||
// 配置自动更新
|
// 配置自动更新
|
||||||
autoUpdater.autoDownload = false
|
autoUpdater.autoDownload = false
|
||||||
@@ -25,6 +35,47 @@ const AUTO_UPDATE_ENABLED =
|
|||||||
process.env.AUTO_UPDATE_ENABLED === '1' ||
|
process.env.AUTO_UPDATE_ENABLED === '1' ||
|
||||||
(process.env.AUTO_UPDATE_ENABLED == null && !process.env.VITE_DEV_SERVER_URL)
|
(process.env.AUTO_UPDATE_ENABLED == null && !process.env.VITE_DEV_SERVER_URL)
|
||||||
|
|
||||||
|
// 使用白名单过滤 PATH,避免被第三方目录中的旧版 VC++ 运行库劫持。
|
||||||
|
// 仅保留系统目录(Windows/System32/SysWOW64)和应用自身目录(可执行目录、resources)。
|
||||||
|
function sanitizePathEnv() {
|
||||||
|
// 开发模式不做裁剪,避免影响本地工具链
|
||||||
|
if (process.env.VITE_DEV_SERVER_URL) return
|
||||||
|
|
||||||
|
const rawPath = process.env.PATH || process.env.Path
|
||||||
|
if (!rawPath) return
|
||||||
|
|
||||||
|
const sep = process.platform === 'win32' ? ';' : ':'
|
||||||
|
const parts = rawPath.split(sep).filter(Boolean)
|
||||||
|
|
||||||
|
const systemRoot = process.env.SystemRoot || process.env.WINDIR || ''
|
||||||
|
const safePrefixes = [
|
||||||
|
systemRoot,
|
||||||
|
systemRoot ? join(systemRoot, 'System32') : '',
|
||||||
|
systemRoot ? join(systemRoot, 'SysWOW64') : '',
|
||||||
|
dirname(process.execPath),
|
||||||
|
process.resourcesPath,
|
||||||
|
join(process.resourcesPath || '', 'resources')
|
||||||
|
].filter(Boolean)
|
||||||
|
|
||||||
|
const normalize = (p: string) => p.replace(/\\/g, '/').toLowerCase()
|
||||||
|
const isSafe = (p: string) => {
|
||||||
|
const np = normalize(p)
|
||||||
|
return safePrefixes.some((prefix) => np.startsWith(normalize(prefix)))
|
||||||
|
}
|
||||||
|
|
||||||
|
const filtered = parts.filter(isSafe)
|
||||||
|
if (filtered.length !== parts.length) {
|
||||||
|
const removed = parts.filter((p) => !isSafe(p))
|
||||||
|
console.warn('[WeFlow] 使用白名单裁剪 PATH,移除目录:', removed)
|
||||||
|
const nextPath = filtered.join(sep)
|
||||||
|
process.env.PATH = nextPath
|
||||||
|
process.env.Path = nextPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 启动时立即清理 PATH,后续创建的 worker 也能继承安全的环境
|
||||||
|
sanitizePathEnv()
|
||||||
|
|
||||||
// 单例服务
|
// 单例服务
|
||||||
let configService: ConfigService | null = null
|
let configService: ConfigService | null = null
|
||||||
|
|
||||||
@@ -91,6 +142,36 @@ function createWindow(options: { autoShow?: boolean } = {}) {
|
|||||||
win.loadFile(join(__dirname, '../dist/index.html'))
|
win.loadFile(join(__dirname, '../dist/index.html'))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Handle notification click navigation
|
||||||
|
ipcMain.on('notification-clicked', (_, sessionId) => {
|
||||||
|
if (win.isMinimized()) win.restore()
|
||||||
|
win.show()
|
||||||
|
win.focus()
|
||||||
|
win.webContents.send('navigate-to-session', sessionId)
|
||||||
|
})
|
||||||
|
|
||||||
|
// 拦截请求,修改 Referer 和 User-Agent 以通过微信 CDN 鉴权
|
||||||
|
session.defaultSession.webRequest.onBeforeSendHeaders(
|
||||||
|
{
|
||||||
|
urls: [
|
||||||
|
'*://*.qpic.cn/*',
|
||||||
|
'*://*.qlogo.cn/*',
|
||||||
|
'*://*.wechat.com/*',
|
||||||
|
'*://*.weixin.qq.com/*'
|
||||||
|
]
|
||||||
|
},
|
||||||
|
(details, callback) => {
|
||||||
|
details.requestHeaders['User-Agent'] = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x63090719) XWEB/8351"
|
||||||
|
details.requestHeaders['Accept'] = "image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8"
|
||||||
|
details.requestHeaders['Accept-Encoding'] = "gzip, deflate, br"
|
||||||
|
details.requestHeaders['Accept-Language'] = "zh-CN,zh;q=0.9"
|
||||||
|
details.requestHeaders['Referer'] = "https://servicewechat.com/"
|
||||||
|
details.requestHeaders['Connection'] = "keep-alive"
|
||||||
|
details.requestHeaders['Range'] = "bytes=0-"
|
||||||
|
callback({ cancel: false, requestHeaders: details.requestHeaders })
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
return win
|
return win
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -164,10 +245,11 @@ function createOnboardingWindow() {
|
|||||||
: join(process.resourcesPath, 'icon.ico')
|
: join(process.resourcesPath, 'icon.ico')
|
||||||
|
|
||||||
onboardingWindow = new BrowserWindow({
|
onboardingWindow = new BrowserWindow({
|
||||||
width: 1100,
|
width: 960,
|
||||||
height: 720,
|
height: 680,
|
||||||
minWidth: 900,
|
minWidth: 900,
|
||||||
minHeight: 600,
|
minHeight: 620,
|
||||||
|
resizable: false,
|
||||||
frame: false,
|
frame: false,
|
||||||
transparent: true,
|
transparent: true,
|
||||||
backgroundColor: '#00000000',
|
backgroundColor: '#00000000',
|
||||||
@@ -198,6 +280,225 @@ function createOnboardingWindow() {
|
|||||||
return onboardingWindow
|
return onboardingWindow
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建独立的视频播放窗口
|
||||||
|
* 窗口大小会根据视频比例自动调整
|
||||||
|
*/
|
||||||
|
function createVideoPlayerWindow(videoPath: string, videoWidth?: number, videoHeight?: number) {
|
||||||
|
const isDev = !!process.env.VITE_DEV_SERVER_URL
|
||||||
|
const iconPath = isDev
|
||||||
|
? join(__dirname, '../public/icon.ico')
|
||||||
|
: join(process.resourcesPath, 'icon.ico')
|
||||||
|
|
||||||
|
// 获取屏幕尺寸
|
||||||
|
const { screen } = require('electron')
|
||||||
|
const primaryDisplay = screen.getPrimaryDisplay()
|
||||||
|
const { width: screenWidth, height: screenHeight } = primaryDisplay.workAreaSize
|
||||||
|
|
||||||
|
// 计算窗口尺寸,只有标题栏 40px,控制栏悬浮
|
||||||
|
let winWidth = 854
|
||||||
|
let winHeight = 520
|
||||||
|
const titleBarHeight = 40
|
||||||
|
|
||||||
|
if (videoWidth && videoHeight && videoWidth > 0 && videoHeight > 0) {
|
||||||
|
const aspectRatio = videoWidth / videoHeight
|
||||||
|
|
||||||
|
const maxWidth = Math.floor(screenWidth * 0.85)
|
||||||
|
const maxHeight = Math.floor(screenHeight * 0.85)
|
||||||
|
|
||||||
|
if (aspectRatio >= 1) {
|
||||||
|
// 横向视频
|
||||||
|
winWidth = Math.min(videoWidth, maxWidth)
|
||||||
|
winHeight = Math.floor(winWidth / aspectRatio) + titleBarHeight
|
||||||
|
|
||||||
|
if (winHeight > maxHeight) {
|
||||||
|
winHeight = maxHeight
|
||||||
|
winWidth = Math.floor((winHeight - titleBarHeight) * aspectRatio)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 竖向视频
|
||||||
|
const videoDisplayHeight = Math.min(videoHeight, maxHeight - titleBarHeight)
|
||||||
|
winHeight = videoDisplayHeight + titleBarHeight
|
||||||
|
winWidth = Math.floor(videoDisplayHeight * aspectRatio)
|
||||||
|
|
||||||
|
if (winWidth < 300) {
|
||||||
|
winWidth = 300
|
||||||
|
winHeight = Math.floor(winWidth / aspectRatio) + titleBarHeight
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
winWidth = Math.max(winWidth, 360)
|
||||||
|
winHeight = Math.max(winHeight, 280)
|
||||||
|
}
|
||||||
|
|
||||||
|
const win = new BrowserWindow({
|
||||||
|
width: winWidth,
|
||||||
|
height: winHeight,
|
||||||
|
minWidth: 360,
|
||||||
|
minHeight: 280,
|
||||||
|
icon: iconPath,
|
||||||
|
webPreferences: {
|
||||||
|
preload: join(__dirname, 'preload.js'),
|
||||||
|
contextIsolation: true,
|
||||||
|
nodeIntegration: false,
|
||||||
|
webSecurity: false
|
||||||
|
},
|
||||||
|
titleBarStyle: 'hidden',
|
||||||
|
titleBarOverlay: {
|
||||||
|
color: '#1a1a1a',
|
||||||
|
symbolColor: '#ffffff',
|
||||||
|
height: 40
|
||||||
|
},
|
||||||
|
show: false,
|
||||||
|
backgroundColor: '#000000',
|
||||||
|
autoHideMenuBar: true
|
||||||
|
})
|
||||||
|
|
||||||
|
win.once('ready-to-show', () => {
|
||||||
|
win.show()
|
||||||
|
})
|
||||||
|
|
||||||
|
const videoParam = `videoPath=${encodeURIComponent(videoPath)}`
|
||||||
|
if (process.env.VITE_DEV_SERVER_URL) {
|
||||||
|
win.loadURL(`${process.env.VITE_DEV_SERVER_URL}#/video-player-window?${videoParam}`)
|
||||||
|
|
||||||
|
win.webContents.on('before-input-event', (event, input) => {
|
||||||
|
if (input.key === 'F12' || (input.control && input.shift && input.key === 'I')) {
|
||||||
|
if (win.webContents.isDevToolsOpened()) {
|
||||||
|
win.webContents.closeDevTools()
|
||||||
|
} else {
|
||||||
|
win.webContents.openDevTools()
|
||||||
|
}
|
||||||
|
event.preventDefault()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
win.loadFile(join(__dirname, '../dist/index.html'), {
|
||||||
|
hash: `/video-player-window?${videoParam}`
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建独立的图片查看窗口
|
||||||
|
*/
|
||||||
|
function createImageViewerWindow(imagePath: string) {
|
||||||
|
const isDev = !!process.env.VITE_DEV_SERVER_URL
|
||||||
|
const iconPath = isDev
|
||||||
|
? join(__dirname, '../public/icon.ico')
|
||||||
|
: join(process.resourcesPath, 'icon.ico')
|
||||||
|
|
||||||
|
const win = new BrowserWindow({
|
||||||
|
width: 900,
|
||||||
|
height: 700,
|
||||||
|
minWidth: 400,
|
||||||
|
minHeight: 300,
|
||||||
|
icon: iconPath,
|
||||||
|
webPreferences: {
|
||||||
|
preload: join(__dirname, 'preload.js'),
|
||||||
|
contextIsolation: true,
|
||||||
|
nodeIntegration: false,
|
||||||
|
webSecurity: false // 允许加载本地文件
|
||||||
|
},
|
||||||
|
titleBarStyle: 'hidden',
|
||||||
|
titleBarOverlay: {
|
||||||
|
color: '#00000000',
|
||||||
|
symbolColor: '#ffffff',
|
||||||
|
height: 40
|
||||||
|
},
|
||||||
|
show: false,
|
||||||
|
backgroundColor: '#000000',
|
||||||
|
autoHideMenuBar: true
|
||||||
|
})
|
||||||
|
|
||||||
|
win.once('ready-to-show', () => {
|
||||||
|
win.show()
|
||||||
|
})
|
||||||
|
|
||||||
|
const imageParam = `imagePath=${encodeURIComponent(imagePath)}`
|
||||||
|
|
||||||
|
if (process.env.VITE_DEV_SERVER_URL) {
|
||||||
|
win.loadURL(`${process.env.VITE_DEV_SERVER_URL}#/image-viewer-window?${imageParam}`)
|
||||||
|
|
||||||
|
win.webContents.on('before-input-event', (event, input) => {
|
||||||
|
if (input.key === 'F12' || (input.control && input.shift && input.key === 'I')) {
|
||||||
|
if (win.webContents.isDevToolsOpened()) {
|
||||||
|
win.webContents.closeDevTools()
|
||||||
|
} else {
|
||||||
|
win.webContents.openDevTools()
|
||||||
|
}
|
||||||
|
event.preventDefault()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
win.loadFile(join(__dirname, '../dist/index.html'), {
|
||||||
|
hash: `/image-viewer-window?${imageParam}`
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return win
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建独立的聊天记录窗口
|
||||||
|
*/
|
||||||
|
function createChatHistoryWindow(sessionId: string, messageId: number) {
|
||||||
|
const isDev = !!process.env.VITE_DEV_SERVER_URL
|
||||||
|
const iconPath = isDev
|
||||||
|
? join(__dirname, '../public/icon.ico')
|
||||||
|
: join(process.resourcesPath, 'icon.ico')
|
||||||
|
|
||||||
|
// 根据系统主题设置窗口背景色
|
||||||
|
const isDark = nativeTheme.shouldUseDarkColors
|
||||||
|
|
||||||
|
const win = new BrowserWindow({
|
||||||
|
width: 600,
|
||||||
|
height: 800,
|
||||||
|
minWidth: 400,
|
||||||
|
minHeight: 500,
|
||||||
|
icon: iconPath,
|
||||||
|
webPreferences: {
|
||||||
|
preload: join(__dirname, 'preload.js'),
|
||||||
|
contextIsolation: true,
|
||||||
|
nodeIntegration: false
|
||||||
|
},
|
||||||
|
titleBarStyle: 'hidden',
|
||||||
|
titleBarOverlay: {
|
||||||
|
color: '#00000000',
|
||||||
|
symbolColor: isDark ? '#ffffff' : '#1a1a1a',
|
||||||
|
height: 32
|
||||||
|
},
|
||||||
|
show: false,
|
||||||
|
backgroundColor: isDark ? '#1A1A1A' : '#F0F0F0',
|
||||||
|
autoHideMenuBar: true
|
||||||
|
})
|
||||||
|
|
||||||
|
win.once('ready-to-show', () => {
|
||||||
|
win.show()
|
||||||
|
})
|
||||||
|
|
||||||
|
if (process.env.VITE_DEV_SERVER_URL) {
|
||||||
|
win.loadURL(`${process.env.VITE_DEV_SERVER_URL}#/chat-history/${sessionId}/${messageId}`)
|
||||||
|
|
||||||
|
win.webContents.on('before-input-event', (event, input) => {
|
||||||
|
if (input.key === 'F12' || (input.control && input.shift && input.key === 'I')) {
|
||||||
|
if (win.webContents.isDevToolsOpened()) {
|
||||||
|
win.webContents.closeDevTools()
|
||||||
|
} else {
|
||||||
|
win.webContents.openDevTools()
|
||||||
|
}
|
||||||
|
event.preventDefault()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
win.loadFile(join(__dirname, '../dist/index.html'), {
|
||||||
|
hash: `/chat-history/${sessionId}/${messageId}`
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return win
|
||||||
|
}
|
||||||
|
|
||||||
function showMainWindow() {
|
function showMainWindow() {
|
||||||
shouldShowMain = true
|
shouldShowMain = true
|
||||||
if (mainWindowReady) {
|
if (mainWindowReady) {
|
||||||
@@ -207,6 +508,7 @@ function showMainWindow() {
|
|||||||
|
|
||||||
// 注册 IPC 处理器
|
// 注册 IPC 处理器
|
||||||
function registerIpcHandlers() {
|
function registerIpcHandlers() {
|
||||||
|
registerNotificationHandlers()
|
||||||
// 配置相关
|
// 配置相关
|
||||||
ipcMain.handle('config:get', async (_, key: string) => {
|
ipcMain.handle('config:get', async (_, key: string) => {
|
||||||
return configService?.get(key as any)
|
return configService?.get(key as any)
|
||||||
@@ -304,7 +606,7 @@ function registerIpcHandlers() {
|
|||||||
|
|
||||||
// 监听下载进度
|
// 监听下载进度
|
||||||
autoUpdater.on('download-progress', (progress) => {
|
autoUpdater.on('download-progress', (progress) => {
|
||||||
win?.webContents.send('app:downloadProgress', progress.percent)
|
win?.webContents.send('app:downloadProgress', progress)
|
||||||
})
|
})
|
||||||
|
|
||||||
// 下载完成后自动安装
|
// 下载完成后自动安装
|
||||||
@@ -320,6 +622,11 @@ function registerIpcHandlers() {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('app:ignoreUpdate', async (_, version: string) => {
|
||||||
|
configService?.set('ignoredUpdateVersion', version)
|
||||||
|
return { success: true }
|
||||||
|
})
|
||||||
|
|
||||||
// 窗口控制
|
// 窗口控制
|
||||||
ipcMain.on('window:minimize', (event) => {
|
ipcMain.on('window:minimize', (event) => {
|
||||||
BrowserWindow.fromWebContents(event.sender)?.minimize()
|
BrowserWindow.fromWebContents(event.sender)?.minimize()
|
||||||
@@ -354,6 +661,85 @@ function registerIpcHandlers() {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// 打开视频播放窗口
|
||||||
|
ipcMain.handle('window:openVideoPlayerWindow', (_, videoPath: string, videoWidth?: number, videoHeight?: number) => {
|
||||||
|
createVideoPlayerWindow(videoPath, videoWidth, videoHeight)
|
||||||
|
})
|
||||||
|
|
||||||
|
// 打开聊天记录窗口
|
||||||
|
ipcMain.handle('window:openChatHistoryWindow', (_, sessionId: string, messageId: number) => {
|
||||||
|
createChatHistoryWindow(sessionId, messageId)
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
// 根据视频尺寸调整窗口大小
|
||||||
|
ipcMain.handle('window:resizeToFitVideo', (event, videoWidth: number, videoHeight: number) => {
|
||||||
|
const win = BrowserWindow.fromWebContents(event.sender)
|
||||||
|
if (!win || !videoWidth || !videoHeight) return
|
||||||
|
|
||||||
|
const { screen } = require('electron')
|
||||||
|
const primaryDisplay = screen.getPrimaryDisplay()
|
||||||
|
const { width: screenWidth, height: screenHeight } = primaryDisplay.workAreaSize
|
||||||
|
|
||||||
|
// 只有标题栏 40px,控制栏悬浮在视频上
|
||||||
|
const titleBarHeight = 40
|
||||||
|
const aspectRatio = videoWidth / videoHeight
|
||||||
|
|
||||||
|
const maxWidth = Math.floor(screenWidth * 0.85)
|
||||||
|
const maxHeight = Math.floor(screenHeight * 0.85)
|
||||||
|
|
||||||
|
let winWidth: number
|
||||||
|
let winHeight: number
|
||||||
|
|
||||||
|
if (aspectRatio >= 1) {
|
||||||
|
// 横向视频 - 以宽度为基准
|
||||||
|
winWidth = Math.min(videoWidth, maxWidth)
|
||||||
|
winHeight = Math.floor(winWidth / aspectRatio) + titleBarHeight
|
||||||
|
|
||||||
|
if (winHeight > maxHeight) {
|
||||||
|
winHeight = maxHeight
|
||||||
|
winWidth = Math.floor((winHeight - titleBarHeight) * aspectRatio)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 竖向视频 - 以高度为基准
|
||||||
|
const videoDisplayHeight = Math.min(videoHeight, maxHeight - titleBarHeight)
|
||||||
|
winHeight = videoDisplayHeight + titleBarHeight
|
||||||
|
winWidth = Math.floor(videoDisplayHeight * aspectRatio)
|
||||||
|
|
||||||
|
// 确保宽度不会太窄
|
||||||
|
if (winWidth < 300) {
|
||||||
|
winWidth = 300
|
||||||
|
winHeight = Math.floor(winWidth / aspectRatio) + titleBarHeight
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
winWidth = Math.max(winWidth, 360)
|
||||||
|
winHeight = Math.max(winHeight, 280)
|
||||||
|
|
||||||
|
// 调整窗口大小并居中
|
||||||
|
win.setSize(winWidth, winHeight)
|
||||||
|
win.center()
|
||||||
|
})
|
||||||
|
|
||||||
|
// 视频相关
|
||||||
|
ipcMain.handle('video:getVideoInfo', async (_, videoMd5: string) => {
|
||||||
|
try {
|
||||||
|
const result = await videoService.getVideoInfo(videoMd5)
|
||||||
|
return { success: true, ...result }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e), exists: false }
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('video:parseVideoMd5', async (_, content: string) => {
|
||||||
|
try {
|
||||||
|
const md5 = videoService.parseVideoMd5(content)
|
||||||
|
return { success: true, md5 }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
// 数据库路径相关
|
// 数据库路径相关
|
||||||
ipcMain.handle('dbpath:autoDetect', async () => {
|
ipcMain.handle('dbpath:autoDetect', async () => {
|
||||||
return dbPathService.autoDetect()
|
return dbPathService.autoDetect()
|
||||||
@@ -363,6 +749,10 @@ function registerIpcHandlers() {
|
|||||||
return dbPathService.scanWxids(rootPath)
|
return dbPathService.scanWxids(rootPath)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('dbpath:scanWxidCandidates', async (_, rootPath: string) => {
|
||||||
|
return dbPathService.scanWxidCandidates(rootPath)
|
||||||
|
})
|
||||||
|
|
||||||
ipcMain.handle('dbpath:getDefault', async () => {
|
ipcMain.handle('dbpath:getDefault', async () => {
|
||||||
return dbPathService.getDefaultPath()
|
return dbPathService.getDefaultPath()
|
||||||
})
|
})
|
||||||
@@ -381,6 +771,8 @@ function registerIpcHandlers() {
|
|||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// 聊天相关
|
// 聊天相关
|
||||||
ipcMain.handle('chat:connect', async () => {
|
ipcMain.handle('chat:connect', async () => {
|
||||||
return chatService.connect()
|
return chatService.connect()
|
||||||
@@ -394,20 +786,94 @@ function registerIpcHandlers() {
|
|||||||
return chatService.enrichSessionsContactInfo(usernames)
|
return chatService.enrichSessionsContactInfo(usernames)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getMessages', async (_, sessionId: string, offset?: number, limit?: number) => {
|
ipcMain.handle('chat:getMessages', async (_, sessionId: string, offset?: number, limit?: number, startTime?: number, endTime?: number, ascending?: boolean) => {
|
||||||
return chatService.getMessages(sessionId, offset, limit)
|
return chatService.getMessages(sessionId, offset, limit, startTime, endTime, ascending)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getLatestMessages', async (_, sessionId: string, limit?: number) => {
|
ipcMain.handle('chat:getLatestMessages', async (_, sessionId: string, limit?: number) => {
|
||||||
return chatService.getLatestMessages(sessionId, limit)
|
return chatService.getLatestMessages(sessionId, limit)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getContact', async (_, username: string) => {
|
ipcMain.handle('chat:getNewMessages', async (_, sessionId: string, minTime: number, limit?: number) => {
|
||||||
return chatService.getContact(username)
|
return chatService.getNewMessages(sessionId, minTime, limit)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:getContact', async (_, username: string) => {
|
||||||
|
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 chatService.getContactAvatar(username)
|
return await chatService.getContactAvatar(username)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:resolveTransferDisplayNames', async (_, chatroomId: string, payerUsername: string, receiverUsername: string) => {
|
||||||
|
return await chatService.resolveTransferDisplayNames(chatroomId, payerUsername, receiverUsername)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:getContacts', async () => {
|
||||||
|
return await chatService.getContacts()
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:getCachedMessages', async (_, sessionId: string) => {
|
||||||
|
return chatService.getCachedSessionMessages(sessionId)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getMyAvatarUrl', async () => {
|
ipcMain.handle('chat:getMyAvatarUrl', async () => {
|
||||||
@@ -431,14 +897,45 @@ function registerIpcHandlers() {
|
|||||||
return chatService.getImageData(sessionId, msgId)
|
return chatService.getImageData(sessionId, msgId)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getVoiceData', async (_, sessionId: string, msgId: string) => {
|
ipcMain.handle('chat:getVoiceData', async (_, sessionId: string, msgId: string, createTime?: number, serverId?: string | number) => {
|
||||||
return chatService.getVoiceData(sessionId, msgId)
|
return chatService.getVoiceData(sessionId, msgId, createTime, serverId)
|
||||||
|
})
|
||||||
|
ipcMain.handle('chat:getAllVoiceMessages', async (_, sessionId: string) => {
|
||||||
|
return chatService.getAllVoiceMessages(sessionId)
|
||||||
|
})
|
||||||
|
ipcMain.handle('chat:resolveVoiceCache', async (_, sessionId: string, msgId: string) => {
|
||||||
|
return chatService.resolveVoiceCache(sessionId, msgId)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('chat:getMessageById', async (_, sessionId: string, localId: number) => {
|
ipcMain.handle('chat:getVoiceTranscript', async (event, sessionId: string, msgId: string, createTime?: number) => {
|
||||||
|
return chatService.getVoiceTranscript(sessionId, msgId, createTime, (text) => {
|
||||||
|
event.sender.send('chat:voiceTranscriptPartial', { msgId, text })
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:getMessage', async (_, sessionId: string, localId: number) => {
|
||||||
return chatService.getMessageById(sessionId, localId)
|
return chatService.getMessageById(sessionId, localId)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('chat:execQuery', async (_, kind: string, path: string | null, sql: string) => {
|
||||||
|
return chatService.execQuery(kind, path, sql)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:getTimeline', async (_, limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number) => {
|
||||||
|
return snsService.getTimeline(limit, offset, usernames, keyword, startTime, endTime)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:debugResource', async (_, url: string) => {
|
||||||
|
return snsService.debugResource(url)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('sns:proxyImage', async (_, url: string) => {
|
||||||
|
return snsService.proxyImage(url)
|
||||||
|
})
|
||||||
|
|
||||||
|
// 私聊克隆
|
||||||
|
|
||||||
|
|
||||||
ipcMain.handle('image:decrypt', async (_, payload: { sessionId?: string; imageMd5?: string; imageDatName?: string; force?: boolean }) => {
|
ipcMain.handle('image:decrypt', async (_, payload: { sessionId?: string; imageMd5?: string; imageDatName?: string; force?: boolean }) => {
|
||||||
return imageDecryptService.decryptImage(payload)
|
return imageDecryptService.decryptImage(payload)
|
||||||
})
|
})
|
||||||
@@ -450,18 +947,38 @@ function registerIpcHandlers() {
|
|||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// Windows Hello
|
||||||
|
ipcMain.handle('auth:hello', async (event, message?: string) => {
|
||||||
|
// 无论哪个窗口调用,都尝试强制附着到主窗口,确保体验一致
|
||||||
|
// 如果主窗口不存在(极其罕见),则回退到调用者窗口
|
||||||
|
const targetWin = (mainWindow && !mainWindow.isDestroyed())
|
||||||
|
? mainWindow
|
||||||
|
: (BrowserWindow.fromWebContents(event.sender) || undefined)
|
||||||
|
|
||||||
|
return windowsHelloService.verify(message, targetWin)
|
||||||
|
})
|
||||||
|
|
||||||
// 导出相关
|
// 导出相关
|
||||||
ipcMain.handle('export:exportSessions', async (_, sessionIds: string[], outputDir: string, options: ExportOptions) => {
|
ipcMain.handle('export:exportSessions', async (event, sessionIds: string[], outputDir: string, options: ExportOptions) => {
|
||||||
return exportService.exportSessions(sessionIds, outputDir, options)
|
const onProgress = (progress: ExportProgress) => {
|
||||||
|
if (!event.sender.isDestroyed()) {
|
||||||
|
event.sender.send('export:progress', progress)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return exportService.exportSessions(sessionIds, outputDir, options, onProgress)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('export:exportSession', async (_, sessionId: string, outputPath: string, options: ExportOptions) => {
|
ipcMain.handle('export:exportSession', async (_, sessionId: string, outputPath: string, options: ExportOptions) => {
|
||||||
return exportService.exportSessionToChatLab(sessionId, outputPath, options)
|
return exportService.exportSessionToChatLab(sessionId, outputPath, options)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('export:exportContacts', async (_, outputDir: string, options: any) => {
|
||||||
|
return contactExportService.exportContacts(outputDir, options)
|
||||||
|
})
|
||||||
|
|
||||||
// 数据分析相关
|
// 数据分析相关
|
||||||
ipcMain.handle('analytics:getOverallStatistics', async () => {
|
ipcMain.handle('analytics:getOverallStatistics', async (_, force?: boolean) => {
|
||||||
return analyticsService.getOverallStatistics()
|
return analyticsService.getOverallStatistics(force)
|
||||||
})
|
})
|
||||||
|
|
||||||
ipcMain.handle('analytics:getContactRankings', async (_, limit?: number) => {
|
ipcMain.handle('analytics:getContactRankings', async (_, limit?: number) => {
|
||||||
@@ -472,6 +989,62 @@ function registerIpcHandlers() {
|
|||||||
return analyticsService.getTimeDistribution()
|
return analyticsService.getTimeDistribution()
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('analytics:getExcludedUsernames', async () => {
|
||||||
|
return analyticsService.getExcludedUsernames()
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('analytics:setExcludedUsernames', async (_, usernames: string[]) => {
|
||||||
|
return analyticsService.setExcludedUsernames(usernames)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('analytics:getExcludeCandidates', async () => {
|
||||||
|
return analyticsService.getExcludeCandidates()
|
||||||
|
})
|
||||||
|
|
||||||
|
// 缓存管理
|
||||||
|
ipcMain.handle('cache:clearAnalytics', async () => {
|
||||||
|
return analyticsService.clearCache()
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('cache:clearImages', async () => {
|
||||||
|
const imageResult = await imageDecryptService.clearCache()
|
||||||
|
const emojiResult = chatService.clearCaches({ includeMessages: false, includeContacts: false, includeEmojis: true })
|
||||||
|
const errors = [imageResult, emojiResult]
|
||||||
|
.filter((result) => !result.success)
|
||||||
|
.map((result) => result.error)
|
||||||
|
.filter(Boolean) as string[]
|
||||||
|
if (errors.length > 0) {
|
||||||
|
return { success: false, error: errors.join('; ') }
|
||||||
|
}
|
||||||
|
return { success: true }
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('cache:clearAll', async () => {
|
||||||
|
const [analyticsResult, imageResult] = await Promise.all([
|
||||||
|
analyticsService.clearCache(),
|
||||||
|
imageDecryptService.clearCache()
|
||||||
|
])
|
||||||
|
const chatResult = chatService.clearCaches()
|
||||||
|
const errors = [analyticsResult, imageResult, chatResult]
|
||||||
|
.filter((result) => !result.success)
|
||||||
|
.map((result) => result.error)
|
||||||
|
.filter(Boolean) as string[]
|
||||||
|
if (errors.length > 0) {
|
||||||
|
return { success: false, error: errors.join('; ') }
|
||||||
|
}
|
||||||
|
return { success: true }
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('whisper:downloadModel', async (event) => {
|
||||||
|
return voiceTranscribeService.downloadModel((progress) => {
|
||||||
|
event.sender.send('whisper:downloadProgress', progress)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('whisper:getModelStatus', async () => {
|
||||||
|
return voiceTranscribeService.getModelStatus()
|
||||||
|
})
|
||||||
|
|
||||||
// 群聊分析相关
|
// 群聊分析相关
|
||||||
ipcMain.handle('groupAnalytics:getGroupChats', async () => {
|
ipcMain.handle('groupAnalytics:getGroupChats', async () => {
|
||||||
return groupAnalyticsService.getGroupChats()
|
return groupAnalyticsService.getGroupChats()
|
||||||
@@ -493,12 +1066,21 @@ function registerIpcHandlers() {
|
|||||||
return groupAnalyticsService.getGroupMediaStats(chatroomId, startTime, endTime)
|
return groupAnalyticsService.getGroupMediaStats(chatroomId, startTime, endTime)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('groupAnalytics:exportGroupMembers', async (_, chatroomId: string, outputPath: string) => {
|
||||||
|
return groupAnalyticsService.exportGroupMembers(chatroomId, outputPath)
|
||||||
|
})
|
||||||
|
|
||||||
// 打开协议窗口
|
// 打开协议窗口
|
||||||
ipcMain.handle('window:openAgreementWindow', async () => {
|
ipcMain.handle('window:openAgreementWindow', async () => {
|
||||||
createAgreementWindow()
|
createAgreementWindow()
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// 打开图片查看窗口
|
||||||
|
ipcMain.handle('window:openImageViewerWindow', (_, imagePath: string) => {
|
||||||
|
createImageViewerWindow(imagePath)
|
||||||
|
})
|
||||||
|
|
||||||
// 完成引导,关闭引导窗口并显示主窗口
|
// 完成引导,关闭引导窗口并显示主窗口
|
||||||
ipcMain.handle('window:completeOnboarding', async () => {
|
ipcMain.handle('window:completeOnboarding', async () => {
|
||||||
try {
|
try {
|
||||||
@@ -596,6 +1178,73 @@ function registerIpcHandlers() {
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('dualReport:generateReport', async (_, payload: { friendUsername: string; year: number }) => {
|
||||||
|
const cfg = configService || new ConfigService()
|
||||||
|
configService = cfg
|
||||||
|
|
||||||
|
const dbPath = cfg.get('dbPath')
|
||||||
|
const decryptKey = cfg.get('decryptKey')
|
||||||
|
const wxid = cfg.get('myWxid')
|
||||||
|
const logEnabled = cfg.get('logEnabled')
|
||||||
|
const friendUsername = payload?.friendUsername
|
||||||
|
const year = payload?.year ?? 0
|
||||||
|
|
||||||
|
if (!friendUsername) {
|
||||||
|
return { success: false, error: '缺少好友用户名' }
|
||||||
|
}
|
||||||
|
|
||||||
|
const resourcesPath = app.isPackaged
|
||||||
|
? join(process.resourcesPath, 'resources')
|
||||||
|
: join(app.getAppPath(), 'resources')
|
||||||
|
const userDataPath = app.getPath('userData')
|
||||||
|
|
||||||
|
const workerPath = join(__dirname, 'dualReportWorker.js')
|
||||||
|
|
||||||
|
return await new Promise((resolve) => {
|
||||||
|
const worker = new Worker(workerPath, {
|
||||||
|
workerData: { year, friendUsername, dbPath, decryptKey, myWxid: wxid, resourcesPath, userDataPath, logEnabled }
|
||||||
|
})
|
||||||
|
|
||||||
|
const cleanup = () => {
|
||||||
|
worker.removeAllListeners()
|
||||||
|
}
|
||||||
|
|
||||||
|
worker.on('message', (msg: any) => {
|
||||||
|
if (msg && msg.type === 'dualReport:progress') {
|
||||||
|
for (const win of BrowserWindow.getAllWindows()) {
|
||||||
|
if (!win.isDestroyed()) {
|
||||||
|
win.webContents.send('dualReport:progress', msg.data)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if (msg && (msg.type === 'dualReport:result' || msg.type === 'done')) {
|
||||||
|
cleanup()
|
||||||
|
void worker.terminate()
|
||||||
|
resolve(msg.data ?? msg.result)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if (msg && (msg.type === 'dualReport:error' || msg.type === 'error')) {
|
||||||
|
cleanup()
|
||||||
|
void worker.terminate()
|
||||||
|
resolve({ success: false, error: msg.error || '双人报告生成失败' })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
worker.on('error', (err) => {
|
||||||
|
cleanup()
|
||||||
|
resolve({ success: false, error: String(err) })
|
||||||
|
})
|
||||||
|
|
||||||
|
worker.on('exit', (code) => {
|
||||||
|
if (code !== 0) {
|
||||||
|
cleanup()
|
||||||
|
resolve({ success: false, error: `双人报告线程异常退出: ${code}` })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
ipcMain.handle('annualReport:exportImages', async (_, payload: { baseDir: string; folderName: string; images: Array<{ name: string; dataUrl: string }> }) => {
|
ipcMain.handle('annualReport:exportImages', async (_, payload: { baseDir: string; folderName: string; images: Array<{ name: string; dataUrl: string }> }) => {
|
||||||
try {
|
try {
|
||||||
const { baseDir, folderName, images } = payload
|
const { baseDir, folderName, images } = payload
|
||||||
@@ -641,6 +1290,23 @@ function registerIpcHandlers() {
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// HTTP API 服务
|
||||||
|
ipcMain.handle('http:start', async (_, port?: number) => {
|
||||||
|
return httpService.start(port || 5031)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('http:stop', async () => {
|
||||||
|
await httpService.stop()
|
||||||
|
return { success: true }
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('http:status', async () => {
|
||||||
|
return {
|
||||||
|
running: httpService.isRunning(),
|
||||||
|
port: httpService.getPort()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 主窗口引用
|
// 主窗口引用
|
||||||
@@ -659,7 +1325,16 @@ function checkForUpdatesOnStartup() {
|
|||||||
if (result && result.updateInfo) {
|
if (result && result.updateInfo) {
|
||||||
const currentVersion = app.getVersion()
|
const currentVersion = app.getVersion()
|
||||||
const latestVersion = result.updateInfo.version
|
const latestVersion = result.updateInfo.version
|
||||||
|
|
||||||
|
// 检查是否有新版本
|
||||||
if (latestVersion !== currentVersion && mainWindow) {
|
if (latestVersion !== currentVersion && mainWindow) {
|
||||||
|
// 检查该版本是否被用户忽略
|
||||||
|
const ignoredVersion = configService?.get('ignoredUpdateVersion')
|
||||||
|
if (ignoredVersion === latestVersion) {
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// 通知渲染进程有新版本
|
// 通知渲染进程有新版本
|
||||||
mainWindow.webContents.send('app:updateAvailable', {
|
mainWindow.webContents.send('app:updateAvailable', {
|
||||||
version: latestVersion,
|
version: latestVersion,
|
||||||
@@ -675,9 +1350,11 @@ function checkForUpdatesOnStartup() {
|
|||||||
|
|
||||||
app.whenReady().then(() => {
|
app.whenReady().then(() => {
|
||||||
configService = new ConfigService()
|
configService = new ConfigService()
|
||||||
const resourcesPath = app.isPackaged
|
const candidateResources = app.isPackaged
|
||||||
? join(process.resourcesPath, 'resources')
|
? join(process.resourcesPath, 'resources')
|
||||||
: join(app.getAppPath(), 'resources')
|
: join(app.getAppPath(), 'resources')
|
||||||
|
const fallbackResources = join(process.cwd(), 'resources')
|
||||||
|
const resourcesPath = existsSync(candidateResources) ? candidateResources : fallbackResources
|
||||||
const userDataPath = app.getPath('userData')
|
const userDataPath = app.getPath('userData')
|
||||||
wcdbService.setPaths(resourcesPath, userDataPath)
|
wcdbService.setPaths(resourcesPath, userDataPath)
|
||||||
wcdbService.setLogEnabled(configService.get('logEnabled') === true)
|
wcdbService.setLogEnabled(configService.get('logEnabled') === true)
|
||||||
@@ -690,6 +1367,17 @@ app.whenReady().then(() => {
|
|||||||
createOnboardingWindow()
|
createOnboardingWindow()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 解决朋友圈图片无法加载问题(添加 Referer)
|
||||||
|
session.defaultSession.webRequest.onBeforeSendHeaders(
|
||||||
|
{
|
||||||
|
urls: ['*://*.qpic.cn/*', '*://*.wx.qq.com/*']
|
||||||
|
},
|
||||||
|
(details, callback) => {
|
||||||
|
details.requestHeaders['Referer'] = 'https://wx.qq.com/'
|
||||||
|
callback({ requestHeaders: details.requestHeaders })
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
// 启动时检测更新
|
// 启动时检测更新
|
||||||
checkForUpdatesOnStartup()
|
checkForUpdatesOnStartup()
|
||||||
|
|
||||||
|
|||||||
24
electron/nodert.d.ts
vendored
Normal file
24
electron/nodert.d.ts
vendored
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
declare module '@nodert-win10-rs4/windows.security.credentials.ui' {
|
||||||
|
export enum UserConsentVerificationResult {
|
||||||
|
Verified = 0,
|
||||||
|
DeviceNotPresent = 1,
|
||||||
|
NotConfiguredForUser = 2,
|
||||||
|
DisabledByPolicy = 3,
|
||||||
|
DeviceBusy = 4,
|
||||||
|
RetriesExhausted = 5,
|
||||||
|
Canceled = 6
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum UserConsentVerifierAvailability {
|
||||||
|
Available = 0,
|
||||||
|
DeviceNotPresent = 1,
|
||||||
|
NotConfiguredForUser = 2,
|
||||||
|
DisabledByPolicy = 3,
|
||||||
|
DeviceBusy = 4
|
||||||
|
}
|
||||||
|
|
||||||
|
export class UserConsentVerifier {
|
||||||
|
static checkAvailabilityAsync(callback: (err: Error | null, availability: UserConsentVerifierAvailability) => void): void;
|
||||||
|
static requestVerificationAsync(message: string, callback: (err: Error | null, result: UserConsentVerificationResult) => void): void;
|
||||||
|
}
|
||||||
|
}
|
||||||
39
electron/preload-env.ts
Normal file
39
electron/preload-env.ts
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
import { join, dirname } from 'path'
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 强制将本地资源目录添加到 PATH 最前端,确保优先加载本地 DLL
|
||||||
|
* 解决系统中存在冲突版本的 DLL 导致的应用崩溃问题
|
||||||
|
*/
|
||||||
|
function enforceLocalDllPriority() {
|
||||||
|
const isDev = !!process.env.VITE_DEV_SERVER_URL
|
||||||
|
const sep = process.platform === 'win32' ? ';' : ':'
|
||||||
|
|
||||||
|
let possiblePaths: string[] = []
|
||||||
|
|
||||||
|
if (isDev) {
|
||||||
|
// 开发环境
|
||||||
|
possiblePaths.push(join(process.cwd(), 'resources'))
|
||||||
|
} else {
|
||||||
|
// 生产环境
|
||||||
|
possiblePaths.push(dirname(process.execPath))
|
||||||
|
if (process.resourcesPath) {
|
||||||
|
possiblePaths.push(process.resourcesPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const dllPaths = possiblePaths.join(sep)
|
||||||
|
|
||||||
|
if (process.env.PATH) {
|
||||||
|
process.env.PATH = dllPaths + sep + process.env.PATH
|
||||||
|
} else {
|
||||||
|
process.env.PATH = dllPaths
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
enforceLocalDllPriority()
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[WeFlow] Failed to enforce local DLL priority:', e)
|
||||||
|
}
|
||||||
@@ -9,6 +9,24 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
clear: () => ipcRenderer.invoke('config:clear')
|
clear: () => ipcRenderer.invoke('config:clear')
|
||||||
},
|
},
|
||||||
|
|
||||||
|
// 通知
|
||||||
|
notification: {
|
||||||
|
show: (data: any) => ipcRenderer.invoke('notification:show', data),
|
||||||
|
close: () => ipcRenderer.invoke('notification:close'),
|
||||||
|
click: (sessionId: string) => ipcRenderer.send('notification-clicked', sessionId),
|
||||||
|
ready: () => ipcRenderer.send('notification:ready'),
|
||||||
|
resize: (width: number, height: number) => ipcRenderer.send('notification:resize', { width, height }),
|
||||||
|
onShow: (callback: (event: any, data: any) => void) => {
|
||||||
|
ipcRenderer.on('notification:show', callback)
|
||||||
|
return () => ipcRenderer.removeAllListeners('notification:show')
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
// 认证
|
||||||
|
auth: {
|
||||||
|
hello: (message?: string) => ipcRenderer.invoke('auth:hello', message)
|
||||||
|
},
|
||||||
|
|
||||||
|
|
||||||
// 对话框
|
// 对话框
|
||||||
dialog: {
|
dialog: {
|
||||||
@@ -29,7 +47,8 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
getVersion: () => ipcRenderer.invoke('app:getVersion'),
|
getVersion: () => ipcRenderer.invoke('app:getVersion'),
|
||||||
checkForUpdates: () => ipcRenderer.invoke('app:checkForUpdates'),
|
checkForUpdates: () => ipcRenderer.invoke('app:checkForUpdates'),
|
||||||
downloadAndInstall: () => ipcRenderer.invoke('app:downloadAndInstall'),
|
downloadAndInstall: () => ipcRenderer.invoke('app:downloadAndInstall'),
|
||||||
onDownloadProgress: (callback: (progress: number) => void) => {
|
ignoreUpdate: (version: string) => ipcRenderer.invoke('app:ignoreUpdate', version),
|
||||||
|
onDownloadProgress: (callback: (progress: any) => void) => {
|
||||||
ipcRenderer.on('app:downloadProgress', (_, progress) => callback(progress))
|
ipcRenderer.on('app:downloadProgress', (_, progress) => callback(progress))
|
||||||
return () => ipcRenderer.removeAllListeners('app:downloadProgress')
|
return () => ipcRenderer.removeAllListeners('app:downloadProgress')
|
||||||
},
|
},
|
||||||
@@ -42,7 +61,8 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
// 日志
|
// 日志
|
||||||
log: {
|
log: {
|
||||||
getPath: () => ipcRenderer.invoke('log:getPath'),
|
getPath: () => ipcRenderer.invoke('log:getPath'),
|
||||||
read: () => ipcRenderer.invoke('log:read')
|
read: () => ipcRenderer.invoke('log:read'),
|
||||||
|
debug: (data: any) => ipcRenderer.send('log:debug', data)
|
||||||
},
|
},
|
||||||
|
|
||||||
// 窗口控制
|
// 窗口控制
|
||||||
@@ -53,13 +73,22 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
openAgreementWindow: () => ipcRenderer.invoke('window:openAgreementWindow'),
|
openAgreementWindow: () => ipcRenderer.invoke('window:openAgreementWindow'),
|
||||||
completeOnboarding: () => ipcRenderer.invoke('window:completeOnboarding'),
|
completeOnboarding: () => ipcRenderer.invoke('window:completeOnboarding'),
|
||||||
openOnboardingWindow: () => ipcRenderer.invoke('window:openOnboardingWindow'),
|
openOnboardingWindow: () => ipcRenderer.invoke('window:openOnboardingWindow'),
|
||||||
setTitleBarOverlay: (options: { symbolColor: string }) => ipcRenderer.send('window:setTitleBarOverlay', options)
|
setTitleBarOverlay: (options: { symbolColor: string }) => ipcRenderer.send('window:setTitleBarOverlay', options),
|
||||||
|
openVideoPlayerWindow: (videoPath: string, videoWidth?: number, videoHeight?: number) =>
|
||||||
|
ipcRenderer.invoke('window:openVideoPlayerWindow', videoPath, videoWidth, videoHeight),
|
||||||
|
resizeToFitVideo: (videoWidth: number, videoHeight: number) =>
|
||||||
|
ipcRenderer.invoke('window:resizeToFitVideo', videoWidth, videoHeight),
|
||||||
|
openImageViewerWindow: (imagePath: string) =>
|
||||||
|
ipcRenderer.invoke('window:openImageViewerWindow', imagePath),
|
||||||
|
openChatHistoryWindow: (sessionId: string, messageId: number) =>
|
||||||
|
ipcRenderer.invoke('window:openChatHistoryWindow', sessionId, messageId)
|
||||||
},
|
},
|
||||||
|
|
||||||
// 数据库路径
|
// 数据库路径
|
||||||
dbPath: {
|
dbPath: {
|
||||||
autoDetect: () => ipcRenderer.invoke('dbpath:autoDetect'),
|
autoDetect: () => ipcRenderer.invoke('dbpath:autoDetect'),
|
||||||
scanWxids: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxids', rootPath),
|
scanWxids: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxids', rootPath),
|
||||||
|
scanWxidCandidates: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxidCandidates', rootPath),
|
||||||
getDefault: () => ipcRenderer.invoke('dbpath:getDefault')
|
getDefault: () => ipcRenderer.invoke('dbpath:getDefault')
|
||||||
},
|
},
|
||||||
|
|
||||||
@@ -69,7 +98,8 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
ipcRenderer.invoke('wcdb:testConnection', dbPath, hexKey, wxid),
|
ipcRenderer.invoke('wcdb:testConnection', dbPath, hexKey, wxid),
|
||||||
open: (dbPath: string, hexKey: string, wxid: string) =>
|
open: (dbPath: string, hexKey: string, wxid: string) =>
|
||||||
ipcRenderer.invoke('wcdb:open', dbPath, hexKey, wxid),
|
ipcRenderer.invoke('wcdb:open', dbPath, hexKey, wxid),
|
||||||
close: () => ipcRenderer.invoke('wcdb:close')
|
close: () => ipcRenderer.invoke('wcdb:close'),
|
||||||
|
|
||||||
},
|
},
|
||||||
|
|
||||||
// 密钥获取
|
// 密钥获取
|
||||||
@@ -93,19 +123,44 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
getSessions: () => ipcRenderer.invoke('chat:getSessions'),
|
getSessions: () => ipcRenderer.invoke('chat:getSessions'),
|
||||||
enrichSessionsContactInfo: (usernames: string[]) =>
|
enrichSessionsContactInfo: (usernames: string[]) =>
|
||||||
ipcRenderer.invoke('chat:enrichSessionsContactInfo', usernames),
|
ipcRenderer.invoke('chat:enrichSessionsContactInfo', usernames),
|
||||||
getMessages: (sessionId: string, offset?: number, limit?: number) =>
|
getMessages: (sessionId: string, offset?: number, limit?: number, startTime?: number, endTime?: number, ascending?: boolean) =>
|
||||||
ipcRenderer.invoke('chat:getMessages', sessionId, offset, limit),
|
ipcRenderer.invoke('chat:getMessages', sessionId, offset, limit, startTime, endTime, ascending),
|
||||||
getLatestMessages: (sessionId: string, limit?: number) =>
|
getLatestMessages: (sessionId: string, limit?: number) =>
|
||||||
ipcRenderer.invoke('chat:getLatestMessages', sessionId, limit),
|
ipcRenderer.invoke('chat:getLatestMessages', sessionId, limit),
|
||||||
|
getNewMessages: (sessionId: string, minTime: number, limit?: number) =>
|
||||||
|
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),
|
||||||
|
resolveTransferDisplayNames: (chatroomId: string, payerUsername: string, receiverUsername: string) =>
|
||||||
|
ipcRenderer.invoke('chat:resolveTransferDisplayNames', chatroomId, payerUsername, receiverUsername),
|
||||||
getMyAvatarUrl: () => ipcRenderer.invoke('chat:getMyAvatarUrl'),
|
getMyAvatarUrl: () => ipcRenderer.invoke('chat:getMyAvatarUrl'),
|
||||||
downloadEmoji: (cdnUrl: string, md5?: string) => ipcRenderer.invoke('chat:downloadEmoji', cdnUrl, md5),
|
downloadEmoji: (cdnUrl: string, md5?: string) => ipcRenderer.invoke('chat:downloadEmoji', cdnUrl, md5),
|
||||||
|
getCachedMessages: (sessionId: string) => ipcRenderer.invoke('chat:getCachedMessages', sessionId),
|
||||||
close: () => ipcRenderer.invoke('chat:close'),
|
close: () => ipcRenderer.invoke('chat:close'),
|
||||||
getSessionDetail: (sessionId: string) => ipcRenderer.invoke('chat:getSessionDetail', sessionId),
|
getSessionDetail: (sessionId: string) => ipcRenderer.invoke('chat:getSessionDetail', sessionId),
|
||||||
getImageData: (sessionId: string, msgId: string) => ipcRenderer.invoke('chat:getImageData', sessionId, msgId),
|
getImageData: (sessionId: string, msgId: string) => ipcRenderer.invoke('chat:getImageData', sessionId, msgId),
|
||||||
getVoiceData: (sessionId: string, msgId: string) => ipcRenderer.invoke('chat:getVoiceData', sessionId, msgId)
|
getVoiceData: (sessionId: string, msgId: string, createTime?: number, serverId?: string | number) =>
|
||||||
|
ipcRenderer.invoke('chat:getVoiceData', sessionId, msgId, createTime, serverId),
|
||||||
|
getAllVoiceMessages: (sessionId: string) => ipcRenderer.invoke('chat:getAllVoiceMessages', sessionId),
|
||||||
|
resolveVoiceCache: (sessionId: string, msgId: string) => ipcRenderer.invoke('chat:resolveVoiceCache', sessionId, msgId),
|
||||||
|
getVoiceTranscript: (sessionId: string, msgId: string, createTime?: number) => ipcRenderer.invoke('chat:getVoiceTranscript', sessionId, msgId, createTime),
|
||||||
|
onVoiceTranscriptPartial: (callback: (payload: { msgId: string; text: string }) => void) => {
|
||||||
|
const listener = (_: any, payload: { msgId: string; text: string }) => callback(payload)
|
||||||
|
ipcRenderer.on('chat:voiceTranscriptPartial', listener)
|
||||||
|
return () => ipcRenderer.removeListener('chat:voiceTranscriptPartial', listener)
|
||||||
},
|
},
|
||||||
|
execQuery: (kind: string, path: string | null, sql: string) =>
|
||||||
|
ipcRenderer.invoke('chat:execQuery', kind, path, sql),
|
||||||
|
getContacts: () => ipcRenderer.invoke('chat:getContacts'),
|
||||||
|
getMessage: (sessionId: string, localId: number) =>
|
||||||
|
ipcRenderer.invoke('chat:getMessage', sessionId, localId),
|
||||||
|
onWcdbChange: (callback: (event: any, data: { type: string; json: string }) => void) => {
|
||||||
|
ipcRenderer.on('wcdb-change', callback)
|
||||||
|
return () => ipcRenderer.removeListener('wcdb-change', callback)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// 图片解密
|
// 图片解密
|
||||||
image: {
|
image: {
|
||||||
@@ -125,24 +180,41 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
|
// 视频
|
||||||
|
video: {
|
||||||
|
getVideoInfo: (videoMd5: string) => ipcRenderer.invoke('video:getVideoInfo', videoMd5),
|
||||||
|
parseVideoMd5: (content: string) => ipcRenderer.invoke('video:parseVideoMd5', content)
|
||||||
|
},
|
||||||
|
|
||||||
// 数据分析
|
// 数据分析
|
||||||
analytics: {
|
analytics: {
|
||||||
getOverallStatistics: () => ipcRenderer.invoke('analytics:getOverallStatistics'),
|
getOverallStatistics: (force?: boolean) => ipcRenderer.invoke('analytics:getOverallStatistics', force),
|
||||||
getContactRankings: (limit?: number) => ipcRenderer.invoke('analytics:getContactRankings', limit),
|
getContactRankings: (limit?: number) => ipcRenderer.invoke('analytics:getContactRankings', limit),
|
||||||
getTimeDistribution: () => ipcRenderer.invoke('analytics:getTimeDistribution'),
|
getTimeDistribution: () => ipcRenderer.invoke('analytics:getTimeDistribution'),
|
||||||
|
getExcludedUsernames: () => ipcRenderer.invoke('analytics:getExcludedUsernames'),
|
||||||
|
setExcludedUsernames: (usernames: string[]) => ipcRenderer.invoke('analytics:setExcludedUsernames', usernames),
|
||||||
|
getExcludeCandidates: () => ipcRenderer.invoke('analytics:getExcludeCandidates'),
|
||||||
onProgress: (callback: (payload: { status: string; progress: number }) => void) => {
|
onProgress: (callback: (payload: { status: string; progress: number }) => void) => {
|
||||||
ipcRenderer.on('analytics:progress', (_, payload) => callback(payload))
|
ipcRenderer.on('analytics:progress', (_, payload) => callback(payload))
|
||||||
return () => ipcRenderer.removeAllListeners('analytics:progress')
|
return () => ipcRenderer.removeAllListeners('analytics:progress')
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
|
// 缓存管理
|
||||||
|
cache: {
|
||||||
|
clearAnalytics: () => ipcRenderer.invoke('cache:clearAnalytics'),
|
||||||
|
clearImages: () => ipcRenderer.invoke('cache:clearImages'),
|
||||||
|
clearAll: () => ipcRenderer.invoke('cache:clearAll')
|
||||||
|
},
|
||||||
|
|
||||||
// 群聊分析
|
// 群聊分析
|
||||||
groupAnalytics: {
|
groupAnalytics: {
|
||||||
getGroupChats: () => ipcRenderer.invoke('groupAnalytics:getGroupChats'),
|
getGroupChats: () => ipcRenderer.invoke('groupAnalytics:getGroupChats'),
|
||||||
getGroupMembers: (chatroomId: string) => ipcRenderer.invoke('groupAnalytics:getGroupMembers', chatroomId),
|
getGroupMembers: (chatroomId: string) => ipcRenderer.invoke('groupAnalytics:getGroupMembers', chatroomId),
|
||||||
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)
|
||||||
},
|
},
|
||||||
|
|
||||||
// 年度报告
|
// 年度报告
|
||||||
@@ -156,12 +228,73 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
|||||||
return () => ipcRenderer.removeAllListeners('annualReport:progress')
|
return () => ipcRenderer.removeAllListeners('annualReport:progress')
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
dualReport: {
|
||||||
|
generateReport: (payload: { friendUsername: string; year: number }) =>
|
||||||
|
ipcRenderer.invoke('dualReport:generateReport', payload),
|
||||||
|
onProgress: (callback: (payload: { status: string; progress: number }) => void) => {
|
||||||
|
ipcRenderer.on('dualReport:progress', (_, payload) => callback(payload))
|
||||||
|
return () => ipcRenderer.removeAllListeners('dualReport:progress')
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
// 导出
|
// 导出
|
||||||
export: {
|
export: {
|
||||||
exportSessions: (sessionIds: string[], outputDir: string, options: any) =>
|
exportSessions: (sessionIds: string[], outputDir: string, options: any) =>
|
||||||
ipcRenderer.invoke('export:exportSessions', sessionIds, outputDir, options),
|
ipcRenderer.invoke('export:exportSessions', sessionIds, outputDir, options),
|
||||||
exportSession: (sessionId: string, outputPath: string, options: any) =>
|
exportSession: (sessionId: string, outputPath: string, options: any) =>
|
||||||
ipcRenderer.invoke('export:exportSession', sessionId, outputPath, options)
|
ipcRenderer.invoke('export:exportSession', sessionId, outputPath, options),
|
||||||
|
exportContacts: (outputDir: string, options: any) =>
|
||||||
|
ipcRenderer.invoke('export:exportContacts', outputDir, options),
|
||||||
|
onProgress: (callback: (payload: { current: number; total: number; currentSession: string; phase: string }) => void) => {
|
||||||
|
ipcRenderer.on('export:progress', (_, payload) => callback(payload))
|
||||||
|
return () => ipcRenderer.removeAllListeners('export:progress')
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
whisper: {
|
||||||
|
downloadModel: () =>
|
||||||
|
ipcRenderer.invoke('whisper:downloadModel'),
|
||||||
|
getModelStatus: () =>
|
||||||
|
ipcRenderer.invoke('whisper:getModelStatus'),
|
||||||
|
onDownloadProgress: (callback: (payload: { modelName: string; downloadedBytes: number; totalBytes?: number; percent?: number }) => void) => {
|
||||||
|
ipcRenderer.on('whisper:downloadProgress', (_, payload) => callback(payload))
|
||||||
|
return () => ipcRenderer.removeAllListeners('whisper:downloadProgress')
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
// 朋友圈
|
||||||
|
sns: {
|
||||||
|
getTimeline: (limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number) =>
|
||||||
|
ipcRenderer.invoke('sns:getTimeline', limit, offset, usernames, keyword, startTime, endTime),
|
||||||
|
debugResource: (url: string) => ipcRenderer.invoke('sns:debugResource', url),
|
||||||
|
proxyImage: (url: string) => ipcRenderer.invoke('sns:proxyImage', url)
|
||||||
|
},
|
||||||
|
|
||||||
|
// 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: {
|
||||||
|
start: (port?: number) => ipcRenderer.invoke('http:start', port),
|
||||||
|
stop: () => ipcRenderer.invoke('http:stop'),
|
||||||
|
status: () => ipcRenderer.invoke('http:status')
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -1,5 +1,9 @@
|
|||||||
import { ConfigService } from './config'
|
import { ConfigService } from './config'
|
||||||
import { wcdbService } from './wcdbService'
|
import { wcdbService } from './wcdbService'
|
||||||
|
import { join } from 'path'
|
||||||
|
import { readFile, writeFile, rm } from 'fs/promises'
|
||||||
|
import { app } from 'electron'
|
||||||
|
import { createHash } from 'crypto'
|
||||||
|
|
||||||
export interface ChatStatistics {
|
export interface ChatStatistics {
|
||||||
totalMessages: number
|
totalMessages: number
|
||||||
@@ -43,6 +47,58 @@ class AnalyticsService {
|
|||||||
this.configService = new ConfigService()
|
this.configService = new ConfigService()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private normalizeUsername(username: string): string {
|
||||||
|
return username.trim().toLowerCase()
|
||||||
|
}
|
||||||
|
|
||||||
|
private normalizeExcludedUsernames(value: unknown): string[] {
|
||||||
|
if (!Array.isArray(value)) return []
|
||||||
|
const normalized = value
|
||||||
|
.map((item) => typeof item === 'string' ? item.trim().toLowerCase() : '')
|
||||||
|
.filter((item) => item.length > 0)
|
||||||
|
return Array.from(new Set(normalized))
|
||||||
|
}
|
||||||
|
|
||||||
|
private getExcludedUsernamesList(): string[] {
|
||||||
|
return this.normalizeExcludedUsernames(this.configService.get('analyticsExcludedUsernames'))
|
||||||
|
}
|
||||||
|
|
||||||
|
private getExcludedUsernamesSet(): Set<string> {
|
||||||
|
return new Set(this.getExcludedUsernamesList())
|
||||||
|
}
|
||||||
|
|
||||||
|
private escapeSqlValue(value: string): string {
|
||||||
|
return value.replace(/'/g, "''")
|
||||||
|
}
|
||||||
|
|
||||||
|
private async getAliasMap(usernames: string[]): Promise<Record<string, string>> {
|
||||||
|
const map: Record<string, string> = {}
|
||||||
|
if (usernames.length === 0) return map
|
||||||
|
|
||||||
|
const chunkSize = 200
|
||||||
|
for (let i = 0; i < usernames.length; i += chunkSize) {
|
||||||
|
const chunk = usernames.slice(i, i + chunkSize)
|
||||||
|
const inList = chunk.map((u) => `'${this.escapeSqlValue(u)}'`).join(',')
|
||||||
|
if (!inList) continue
|
||||||
|
const sql = `
|
||||||
|
SELECT username, alias
|
||||||
|
FROM contact
|
||||||
|
WHERE username IN (${inList})
|
||||||
|
`
|
||||||
|
const result = await wcdbService.execQuery('contact', null, sql)
|
||||||
|
if (!result.success || !result.rows) continue
|
||||||
|
for (const row of result.rows as Record<string, any>[]) {
|
||||||
|
const username = row.username || ''
|
||||||
|
const alias = row.alias || ''
|
||||||
|
if (username && alias) {
|
||||||
|
map[username] = alias
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return map
|
||||||
|
}
|
||||||
|
|
||||||
private cleanAccountDirName(name: string): string {
|
private cleanAccountDirName(name: string): string {
|
||||||
const trimmed = name.trim()
|
const trimmed = name.trim()
|
||||||
if (!trimmed) return trimmed
|
if (!trimmed) return trimmed
|
||||||
@@ -51,7 +107,11 @@ class AnalyticsService {
|
|||||||
if (match) return match[1]
|
if (match) return match[1]
|
||||||
return trimmed
|
return trimmed
|
||||||
}
|
}
|
||||||
return trimmed
|
|
||||||
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
|
const cleaned = suffixMatch ? suffixMatch[1] : trimmed
|
||||||
|
|
||||||
|
return cleaned
|
||||||
}
|
}
|
||||||
|
|
||||||
private isPrivateSession(username: string, cleanedWxid: string): boolean {
|
private isPrivateSession(username: string, cleanedWxid: string): boolean {
|
||||||
@@ -94,13 +154,15 @@ class AnalyticsService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private async getPrivateSessions(
|
private async getPrivateSessions(
|
||||||
cleanedWxid: string
|
cleanedWxid: string,
|
||||||
|
excludedUsernames?: Set<string>
|
||||||
): Promise<{ usernames: string[]; numericIds: string[] }> {
|
): Promise<{ usernames: string[]; numericIds: string[] }> {
|
||||||
const sessionResult = await wcdbService.getSessions()
|
const sessionResult = await wcdbService.getSessions()
|
||||||
if (!sessionResult.success || !sessionResult.sessions) {
|
if (!sessionResult.success || !sessionResult.sessions) {
|
||||||
return { usernames: [], numericIds: [] }
|
return { usernames: [], numericIds: [] }
|
||||||
}
|
}
|
||||||
const rows = sessionResult.sessions as Record<string, any>[]
|
const rows = sessionResult.sessions as Record<string, any>[]
|
||||||
|
const excluded = excludedUsernames ?? this.getExcludedUsernamesSet()
|
||||||
|
|
||||||
const sample = rows[0]
|
const sample = rows[0]
|
||||||
void sample
|
void sample
|
||||||
@@ -121,7 +183,11 @@ class AnalyticsService {
|
|||||||
return { username, idValue }
|
return { username, idValue }
|
||||||
})
|
})
|
||||||
const usernames = sessions.map((s) => s.username)
|
const usernames = sessions.map((s) => s.username)
|
||||||
const privateSessions = sessions.filter((s) => this.isPrivateSession(s.username, cleanedWxid))
|
const privateSessions = sessions.filter((s) => {
|
||||||
|
if (!this.isPrivateSession(s.username, cleanedWxid)) return false
|
||||||
|
if (excluded.size === 0) return true
|
||||||
|
return !excluded.has(this.normalizeUsername(s.username))
|
||||||
|
})
|
||||||
const privateUsernames = privateSessions.map((s) => s.username)
|
const privateUsernames = privateSessions.map((s) => s.username)
|
||||||
const numericIds = privateSessions
|
const numericIds = privateSessions
|
||||||
.map((s) => s.idValue)
|
.map((s) => s.idValue)
|
||||||
@@ -174,11 +240,18 @@ class AnalyticsService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private buildAggregateCacheKey(sessionIds: string[], beginTimestamp: number, endTimestamp: number): string {
|
private buildAggregateCacheKey(sessionIds: string[], beginTimestamp: number, endTimestamp: number): string {
|
||||||
const sample = sessionIds.slice(0, 5).join(',')
|
if (sessionIds.length === 0) {
|
||||||
return `${beginTimestamp}-${endTimestamp}-${sessionIds.length}-${sample}`
|
return `${beginTimestamp}-${endTimestamp}-0-empty`
|
||||||
|
}
|
||||||
|
const normalized = Array.from(new Set(sessionIds.map((id) => String(id)))).sort()
|
||||||
|
const hash = createHash('sha1').update(normalized.join('|')).digest('hex').slice(0, 12)
|
||||||
|
return `${beginTimestamp}-${endTimestamp}-${normalized.length}-${hash}`
|
||||||
}
|
}
|
||||||
|
|
||||||
private async computeAggregateByCursor(sessionIds: string[], beginTimestamp = 0, endTimestamp = 0): Promise<any> {
|
private async computeAggregateByCursor(sessionIds: string[], beginTimestamp = 0, endTimestamp = 0): Promise<any> {
|
||||||
|
const wxid = this.configService.get('myWxid')
|
||||||
|
const cleanedWxid = wxid ? this.cleanAccountDirName(wxid) : ''
|
||||||
|
|
||||||
const aggregate = {
|
const aggregate = {
|
||||||
total: 0,
|
total: 0,
|
||||||
sent: 0,
|
sent: 0,
|
||||||
@@ -203,8 +276,22 @@ class AnalyticsService {
|
|||||||
if (endTimestamp > 0 && createTime > endTimestamp) return
|
if (endTimestamp > 0 && createTime > endTimestamp) return
|
||||||
|
|
||||||
const localType = parseInt(row.local_type || row.type || '1', 10)
|
const localType = parseInt(row.local_type || row.type || '1', 10)
|
||||||
const isSendRaw = row.computed_is_send ?? row.is_send ?? row.isSend ?? 0
|
const isSendRaw = row.computed_is_send ?? row.is_send ?? row.isSend
|
||||||
const isSend = String(isSendRaw) === '1' || isSendRaw === 1 || isSendRaw === true
|
let isSend = String(isSendRaw) === '1' || isSendRaw === 1 || isSendRaw === true
|
||||||
|
|
||||||
|
// 如果底层没有提供 is_send,则根据发送者用户名推断
|
||||||
|
const senderUsername = row.sender_username || row.senderUsername || row.sender
|
||||||
|
if (isSendRaw === undefined || isSendRaw === null) {
|
||||||
|
if (senderUsername && (cleanedWxid)) {
|
||||||
|
const senderLower = String(senderUsername).toLowerCase()
|
||||||
|
const myWxidLower = cleanedWxid.toLowerCase()
|
||||||
|
isSend = (
|
||||||
|
senderLower === myWxidLower ||
|
||||||
|
// 兼容非 wxid 开头的账号(如果文件夹名带后缀,如 custom_backup,而 sender 是 custom)
|
||||||
|
(myWxidLower.startsWith(senderLower + '_'))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
aggregate.total += 1
|
aggregate.total += 1
|
||||||
sessionStat.total += 1
|
sessionStat.total += 1
|
||||||
@@ -253,15 +340,31 @@ class AnalyticsService {
|
|||||||
sessionIds: string[],
|
sessionIds: string[],
|
||||||
beginTimestamp = 0,
|
beginTimestamp = 0,
|
||||||
endTimestamp = 0,
|
endTimestamp = 0,
|
||||||
window?: any
|
window?: any,
|
||||||
|
force = false
|
||||||
): Promise<{ success: boolean; data?: any; source?: string; error?: string }> {
|
): Promise<{ success: boolean; data?: any; source?: string; error?: string }> {
|
||||||
const cacheKey = this.buildAggregateCacheKey(sessionIds, beginTimestamp, endTimestamp)
|
const cacheKey = this.buildAggregateCacheKey(sessionIds, beginTimestamp, endTimestamp)
|
||||||
if (this.aggregateCache && this.aggregateCache.key === cacheKey) {
|
|
||||||
|
if (force) {
|
||||||
|
if (this.aggregateCache) this.aggregateCache = null
|
||||||
|
if (this.fallbackAggregateCache) this.fallbackAggregateCache = null
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!force && this.aggregateCache && this.aggregateCache.key === cacheKey) {
|
||||||
if (Date.now() - this.aggregateCache.updatedAt < 5 * 60 * 1000) {
|
if (Date.now() - this.aggregateCache.updatedAt < 5 * 60 * 1000) {
|
||||||
return { success: true, data: this.aggregateCache.data, source: 'cache' }
|
return { success: true, data: this.aggregateCache.data, source: 'cache' }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 尝试从文件加载缓存
|
||||||
|
if (!force) {
|
||||||
|
const fileCache = await this.loadCacheFromFile()
|
||||||
|
if (fileCache && fileCache.key === cacheKey) {
|
||||||
|
this.aggregateCache = fileCache
|
||||||
|
return { success: true, data: fileCache.data, source: 'file-cache' }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (this.aggregatePromise && this.aggregatePromise.key === cacheKey) {
|
if (this.aggregatePromise && this.aggregatePromise.key === cacheKey) {
|
||||||
return this.aggregatePromise.promise
|
return this.aggregatePromise.promise
|
||||||
}
|
}
|
||||||
@@ -291,7 +394,12 @@ class AnalyticsService {
|
|||||||
|
|
||||||
this.aggregatePromise = { key: cacheKey, promise }
|
this.aggregatePromise = { key: cacheKey, promise }
|
||||||
try {
|
try {
|
||||||
return await promise
|
const result = await promise
|
||||||
|
// 如果计算成功,同时写入此文件缓存
|
||||||
|
if (result.success && result.data && result.source !== 'cache') {
|
||||||
|
this.saveCacheToFile({ key: cacheKey, data: this.aggregateCache?.data, updatedAt: Date.now() })
|
||||||
|
}
|
||||||
|
return result
|
||||||
} finally {
|
} finally {
|
||||||
if (this.aggregatePromise && this.aggregatePromise.key === cacheKey) {
|
if (this.aggregatePromise && this.aggregatePromise.key === cacheKey) {
|
||||||
this.aggregatePromise = null
|
this.aggregatePromise = null
|
||||||
@@ -299,6 +407,25 @@ class AnalyticsService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private getCacheFilePath(): string {
|
||||||
|
return join(app.getPath('documents'), 'WeFlow', 'analytics_cache.json')
|
||||||
|
}
|
||||||
|
|
||||||
|
private async loadCacheFromFile(): Promise<{ key: string; data: any; updatedAt: number } | null> {
|
||||||
|
try {
|
||||||
|
const raw = await readFile(this.getCacheFilePath(), 'utf-8')
|
||||||
|
return JSON.parse(raw)
|
||||||
|
} catch { return null }
|
||||||
|
}
|
||||||
|
|
||||||
|
private async saveCacheToFile(data: any) {
|
||||||
|
try {
|
||||||
|
await writeFile(this.getCacheFilePath(), JSON.stringify(data))
|
||||||
|
} catch (e) {
|
||||||
|
console.error('保存统计缓存失败:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private normalizeAggregateSessions(
|
private normalizeAggregateSessions(
|
||||||
sessions: Record<string, any> | undefined,
|
sessions: Record<string, any> | undefined,
|
||||||
idMap: Record<string, string> | undefined
|
idMap: Record<string, string> | undefined
|
||||||
@@ -326,7 +453,66 @@ class AnalyticsService {
|
|||||||
void results
|
void results
|
||||||
}
|
}
|
||||||
|
|
||||||
async getOverallStatistics(): Promise<{ success: boolean; data?: ChatStatistics; error?: string }> {
|
async getExcludedUsernames(): Promise<{ success: boolean; data?: string[]; error?: string }> {
|
||||||
|
try {
|
||||||
|
return { success: true, data: this.getExcludedUsernamesList() }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async setExcludedUsernames(usernames: string[]): Promise<{ success: boolean; data?: string[]; error?: string }> {
|
||||||
|
try {
|
||||||
|
const normalized = this.normalizeExcludedUsernames(usernames)
|
||||||
|
this.configService.set('analyticsExcludedUsernames', normalized)
|
||||||
|
await this.clearCache()
|
||||||
|
return { success: true, data: normalized }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async getExcludeCandidates(): Promise<{ success: boolean; data?: Array<{ username: string; displayName: string; avatarUrl?: string; wechatId?: string }>; error?: string }> {
|
||||||
|
try {
|
||||||
|
const conn = await this.ensureConnected()
|
||||||
|
if (!conn.success || !conn.cleanedWxid) return { success: false, error: conn.error }
|
||||||
|
|
||||||
|
const excluded = this.getExcludedUsernamesSet()
|
||||||
|
const sessionInfo = await this.getPrivateSessions(conn.cleanedWxid, new Set())
|
||||||
|
|
||||||
|
const usernames = new Set<string>(sessionInfo.usernames)
|
||||||
|
for (const name of excluded) usernames.add(name)
|
||||||
|
|
||||||
|
if (usernames.size === 0) {
|
||||||
|
return { success: true, data: [] }
|
||||||
|
}
|
||||||
|
|
||||||
|
const usernameList = Array.from(usernames)
|
||||||
|
const [displayNames, avatarUrls, aliasMap] = await Promise.all([
|
||||||
|
wcdbService.getDisplayNames(usernameList),
|
||||||
|
wcdbService.getAvatarUrls(usernameList),
|
||||||
|
this.getAliasMap(usernameList)
|
||||||
|
])
|
||||||
|
|
||||||
|
const entries = usernameList.map((username) => {
|
||||||
|
const displayName = displayNames.success && displayNames.map
|
||||||
|
? (displayNames.map[username] || username)
|
||||||
|
: username
|
||||||
|
const avatarUrl = avatarUrls.success && avatarUrls.map
|
||||||
|
? avatarUrls.map[username]
|
||||||
|
: undefined
|
||||||
|
const alias = aliasMap[username]
|
||||||
|
const wechatId = alias || (!username.startsWith('wxid_') ? username : '')
|
||||||
|
return { username, displayName, avatarUrl, wechatId }
|
||||||
|
})
|
||||||
|
|
||||||
|
return { success: true, data: entries }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async getOverallStatistics(force = false): Promise<{ success: boolean; data?: ChatStatistics; error?: string }> {
|
||||||
try {
|
try {
|
||||||
const conn = await this.ensureConnected()
|
const conn = await this.ensureConnected()
|
||||||
if (!conn.success || !conn.cleanedWxid) return { success: false, error: conn.error }
|
if (!conn.success || !conn.cleanedWxid) return { success: false, error: conn.error }
|
||||||
@@ -340,7 +526,7 @@ class AnalyticsService {
|
|||||||
const win = BrowserWindow.getAllWindows()[0]
|
const win = BrowserWindow.getAllWindows()[0]
|
||||||
this.setProgress(win, '正在执行原生数据聚合...', 30)
|
this.setProgress(win, '正在执行原生数据聚合...', 30)
|
||||||
|
|
||||||
const result = await this.getAggregateWithFallback(sessionInfo.usernames, 0, 0, win)
|
const result = await this.getAggregateWithFallback(sessionInfo.usernames, 0, 0, win, force)
|
||||||
|
|
||||||
if (!result.success || !result.data) {
|
if (!result.success || !result.data) {
|
||||||
return { success: false, error: result.error || '聚合统计失败' }
|
return { success: false, error: result.error || '聚合统计失败' }
|
||||||
@@ -458,8 +644,8 @@ class AnalyticsService {
|
|||||||
|
|
||||||
const d = result.data
|
const d = result.data
|
||||||
|
|
||||||
// SQLite strftime('%w') 返回 0=Sun, 1=Mon...6=Sat
|
// SQLite strftime('%w') 返回 0=周日, 1=周一...6=周六
|
||||||
// 前端期望 1=Mon...7=Sun
|
// 前端期望 1=周一...7=周日
|
||||||
const weekdayDistribution: Record<number, number> = {}
|
const weekdayDistribution: Record<number, number> = {}
|
||||||
for (const [w, count] of Object.entries(d.weekday)) {
|
for (const [w, count] of Object.entries(d.weekday)) {
|
||||||
const sqliteW = parseInt(w, 10)
|
const sqliteW = parseInt(w, 10)
|
||||||
@@ -485,6 +671,18 @@ class AnalyticsService {
|
|||||||
return { success: false, error: String(e) }
|
return { success: false, error: String(e) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async clearCache(): Promise<{ success: boolean; error?: string }> {
|
||||||
|
this.aggregateCache = null
|
||||||
|
this.fallbackAggregateCache = null
|
||||||
|
this.aggregatePromise = null
|
||||||
|
try {
|
||||||
|
await rm(this.getCacheFilePath(), { force: true })
|
||||||
|
return { success: true }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export const analyticsService = new AnalyticsService()
|
export const analyticsService = new AnalyticsService()
|
||||||
|
|||||||
@@ -69,6 +69,20 @@ export interface AnnualReportData {
|
|||||||
phrase: string
|
phrase: string
|
||||||
count: number
|
count: number
|
||||||
}[]
|
}[]
|
||||||
|
snsStats?: {
|
||||||
|
totalPosts: number
|
||||||
|
typeCounts?: Record<string, number>
|
||||||
|
topLikers: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
topLiked: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
}
|
||||||
|
lostFriend: {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
avatarUrl?: string
|
||||||
|
earlyCount: number
|
||||||
|
lateCount: number
|
||||||
|
periodDesc: string
|
||||||
|
} | null
|
||||||
}
|
}
|
||||||
|
|
||||||
class AnnualReportService {
|
class AnnualReportService {
|
||||||
@@ -101,8 +115,9 @@ class AnnualReportService {
|
|||||||
return trimmed
|
return trimmed
|
||||||
}
|
}
|
||||||
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
if (suffixMatch) return suffixMatch[1]
|
const cleaned = suffixMatch ? suffixMatch[1] : trimmed
|
||||||
return trimmed
|
|
||||||
|
return cleaned
|
||||||
}
|
}
|
||||||
|
|
||||||
private async ensureConnectedWithConfig(
|
private async ensureConnectedWithConfig(
|
||||||
@@ -178,11 +193,15 @@ class AnnualReportService {
|
|||||||
if (!raw) return ''
|
if (!raw) return ''
|
||||||
if (typeof raw === 'string') {
|
if (typeof raw === 'string') {
|
||||||
if (raw.length === 0) return ''
|
if (raw.length === 0) return ''
|
||||||
if (this.looksLikeHex(raw)) {
|
// 只有当字符串足够长(超过16字符)且看起来像 hex 时才尝试解码
|
||||||
|
// 短字符串(如 "123456" 等纯数字)容易被误判为 hex
|
||||||
|
if (raw.length > 16 && this.looksLikeHex(raw)) {
|
||||||
const bytes = Buffer.from(raw, 'hex')
|
const bytes = Buffer.from(raw, 'hex')
|
||||||
if (bytes.length > 0) return this.decodeBinaryContent(bytes)
|
if (bytes.length > 0) return this.decodeBinaryContent(bytes)
|
||||||
}
|
}
|
||||||
if (this.looksLikeBase64(raw)) {
|
// 只有当字符串足够长(超过16字符)且看起来像 base64 时才尝试解码
|
||||||
|
// 短字符串(如 "test", "home" 等)容易被误判为 base64
|
||||||
|
if (raw.length > 16 && this.looksLikeBase64(raw)) {
|
||||||
try {
|
try {
|
||||||
const bytes = Buffer.from(raw, 'base64')
|
const bytes = Buffer.from(raw, 'base64')
|
||||||
return this.decodeBinaryContent(bytes)
|
return this.decodeBinaryContent(bytes)
|
||||||
@@ -397,8 +416,15 @@ class AnnualReportService {
|
|||||||
|
|
||||||
this.reportProgress('加载会话列表...', 15, onProgress)
|
this.reportProgress('加载会话列表...', 15, onProgress)
|
||||||
|
|
||||||
const startTime = Math.floor(new Date(year, 0, 1).getTime() / 1000)
|
const isAllTime = year <= 0
|
||||||
const endTime = Math.floor(new Date(year, 11, 31, 23, 59, 59).getTime() / 1000)
|
const reportYear = isAllTime ? 0 : year
|
||||||
|
const startTime = isAllTime ? 0 : Math.floor(new Date(year, 0, 1).getTime() / 1000)
|
||||||
|
const endTime = isAllTime ? 0 : Math.floor(new Date(year, 11, 31, 23, 59, 59).getTime() / 1000)
|
||||||
|
|
||||||
|
const now = new Date()
|
||||||
|
// 全局统计始终使用自然年范围 (Jan 1st - Now/YearEnd)
|
||||||
|
const actualStartTime = startTime
|
||||||
|
const actualEndTime = endTime
|
||||||
|
|
||||||
let totalMessages = 0
|
let totalMessages = 0
|
||||||
const contactStats = new Map<string, { sent: number; received: number }>()
|
const contactStats = new Map<string, { sent: number; received: number }>()
|
||||||
@@ -420,7 +446,7 @@ class AnnualReportService {
|
|||||||
const CONVERSATION_GAP = 3600
|
const CONVERSATION_GAP = 3600
|
||||||
|
|
||||||
this.reportProgress('统计会话消息...', 20, onProgress)
|
this.reportProgress('统计会话消息...', 20, onProgress)
|
||||||
const result = await wcdbService.getAnnualReportStats(sessionIds, startTime, endTime)
|
const result = await wcdbService.getAnnualReportStats(sessionIds, actualStartTime, actualEndTime)
|
||||||
if (!result.success || !result.data) {
|
if (!result.success || !result.data) {
|
||||||
return { success: false, error: result.error ? `基础统计失败: ${result.error}` : '基础统计失败' }
|
return { success: false, error: result.error ? `基础统计失败: ${result.error}` : '基础统计失败' }
|
||||||
}
|
}
|
||||||
@@ -474,7 +500,7 @@ class AnnualReportService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
this.reportProgress('加载扩展统计... (初始化)', 30, onProgress)
|
this.reportProgress('加载扩展统计... (初始化)', 30, onProgress)
|
||||||
const extras = await wcdbService.getAnnualReportExtras(sessionIds, startTime, endTime, peakDayBegin, peakDayEnd)
|
const extras = await wcdbService.getAnnualReportExtras(sessionIds, actualStartTime, actualEndTime, peakDayBegin, peakDayEnd)
|
||||||
if (extras.success && extras.data) {
|
if (extras.success && extras.data) {
|
||||||
this.reportProgress('加载扩展统计... (解析热力图)', 32, onProgress)
|
this.reportProgress('加载扩展统计... (解析热力图)', 32, onProgress)
|
||||||
const extrasData = extras.data as any
|
const extrasData = extras.data as any
|
||||||
@@ -554,7 +580,7 @@ class AnnualReportService {
|
|||||||
// 为保持功能完整,我们进行深度集成的轻量遍历:
|
// 为保持功能完整,我们进行深度集成的轻量遍历:
|
||||||
for (let i = 0; i < sessionIds.length; i++) {
|
for (let i = 0; i < sessionIds.length; i++) {
|
||||||
const sessionId = sessionIds[i]
|
const sessionId = sessionIds[i]
|
||||||
const cursor = await wcdbService.openMessageCursorLite(sessionId, 1000, true, startTime, endTime)
|
const cursor = await wcdbService.openMessageCursorLite(sessionId, 1000, true, actualStartTime, actualEndTime)
|
||||||
if (!cursor.success || !cursor.cursor) continue
|
if (!cursor.success || !cursor.cursor) continue
|
||||||
|
|
||||||
let lastDayIndex: number | null = null
|
let lastDayIndex: number | null = null
|
||||||
@@ -575,9 +601,22 @@ class AnnualReportService {
|
|||||||
if (!createTime) continue
|
if (!createTime) continue
|
||||||
|
|
||||||
const isSendRaw = row.computed_is_send ?? row.is_send ?? '0'
|
const isSendRaw = row.computed_is_send ?? row.is_send ?? '0'
|
||||||
const isSent = parseInt(isSendRaw, 10) === 1
|
let isSent = parseInt(isSendRaw, 10) === 1
|
||||||
const localType = parseInt(row.local_type || row.type || '1', 10)
|
const localType = parseInt(row.local_type || row.type || '1', 10)
|
||||||
|
|
||||||
|
// 兼容逻辑
|
||||||
|
if (isSendRaw === undefined || isSendRaw === null || isSendRaw === '0') {
|
||||||
|
const sender = String(row.sender_username || row.sender || row.talker || '').toLowerCase()
|
||||||
|
if (sender) {
|
||||||
|
const rawLower = rawWxid.toLowerCase()
|
||||||
|
const cleanedLower = cleanedWxid.toLowerCase()
|
||||||
|
if (sender === rawLower || sender === cleanedLower ||
|
||||||
|
rawLower.startsWith(sender + '_') || cleanedLower.startsWith(sender + '_')) {
|
||||||
|
isSent = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 响应速度 & 对话发起
|
// 响应速度 & 对话发起
|
||||||
if (!conversationStarts.has(sessionId)) {
|
if (!conversationStarts.has(sessionId)) {
|
||||||
conversationStarts.set(sessionId, { initiated: 0, received: 0 })
|
conversationStarts.set(sessionId, { initiated: 0, received: 0 })
|
||||||
@@ -689,7 +728,7 @@ class AnnualReportService {
|
|||||||
|
|
||||||
if (!streakComputedInLoop) {
|
if (!streakComputedInLoop) {
|
||||||
this.reportProgress('计算连续聊天...', 45, onProgress)
|
this.reportProgress('计算连续聊天...', 45, onProgress)
|
||||||
const streakResult = await this.computeLongestStreak(sessionIds, startTime, endTime, onProgress, 45, 75)
|
const streakResult = await this.computeLongestStreak(sessionIds, actualStartTime, actualEndTime, onProgress, 45, 75)
|
||||||
if (streakResult.days > longestStreakDays) {
|
if (streakResult.days > longestStreakDays) {
|
||||||
longestStreakDays = streakResult.days
|
longestStreakDays = streakResult.days
|
||||||
longestStreakSessionId = streakResult.sessionId
|
longestStreakSessionId = streakResult.sessionId
|
||||||
@@ -698,6 +737,42 @@ class AnnualReportService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 获取朋友圈统计
|
||||||
|
this.reportProgress('分析朋友圈数据...', 75, onProgress)
|
||||||
|
let snsStatsResult: {
|
||||||
|
totalPosts: number
|
||||||
|
typeCounts?: Record<string, number>
|
||||||
|
topLikers: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
topLiked: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
} | undefined
|
||||||
|
|
||||||
|
const snsStats = await wcdbService.getSnsAnnualStats(actualStartTime, actualEndTime)
|
||||||
|
|
||||||
|
if (snsStats.success && snsStats.data) {
|
||||||
|
const d = snsStats.data
|
||||||
|
const usersToFetch = new Set<string>()
|
||||||
|
d.topLikers?.forEach((u: any) => usersToFetch.add(u.username))
|
||||||
|
d.topLiked?.forEach((u: any) => usersToFetch.add(u.username))
|
||||||
|
|
||||||
|
const snsUserIds = Array.from(usersToFetch)
|
||||||
|
const [snsDisplayNames, snsAvatarUrls] = await Promise.all([
|
||||||
|
wcdbService.getDisplayNames(snsUserIds),
|
||||||
|
wcdbService.getAvatarUrls(snsUserIds)
|
||||||
|
])
|
||||||
|
|
||||||
|
const getSnsUserInfo = (username: string) => ({
|
||||||
|
displayName: snsDisplayNames.success && snsDisplayNames.map ? (snsDisplayNames.map[username] || username) : username,
|
||||||
|
avatarUrl: snsAvatarUrls.success && snsAvatarUrls.map ? snsAvatarUrls.map[username] : undefined
|
||||||
|
})
|
||||||
|
|
||||||
|
snsStatsResult = {
|
||||||
|
totalPosts: d.totalPosts || 0,
|
||||||
|
typeCounts: d.typeCounts,
|
||||||
|
topLikers: (d.topLikers || []).map((u: any) => ({ ...u, ...getSnsUserInfo(u.username) })),
|
||||||
|
topLiked: (d.topLiked || []).map((u: any) => ({ ...u, ...getSnsUserInfo(u.username) }))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
this.reportProgress('整理联系人信息...', 85, onProgress)
|
this.reportProgress('整理联系人信息...', 85, onProgress)
|
||||||
|
|
||||||
const contactIds = Array.from(contactStats.keys())
|
const contactIds = Array.from(contactStats.keys())
|
||||||
@@ -901,8 +976,130 @@ class AnnualReportService {
|
|||||||
.slice(0, 32)
|
.slice(0, 32)
|
||||||
.map(([phrase, count]) => ({ phrase, count }))
|
.map(([phrase, count]) => ({ phrase, count }))
|
||||||
|
|
||||||
|
// 曾经的好朋友 (Once Best Friend / Lost Friend)
|
||||||
|
let lostFriend: AnnualReportData['lostFriend'] = null
|
||||||
|
let maxEarlyCount = 80 // 最低门槛
|
||||||
|
let bestEarlyCount = 0
|
||||||
|
let bestLateCount = 0
|
||||||
|
let bestSid = ''
|
||||||
|
let bestPeriodDesc = ''
|
||||||
|
|
||||||
|
const currentMonthIndex = new Date().getMonth() + 1 // 1-12
|
||||||
|
|
||||||
|
const currentYearNum = now.getFullYear()
|
||||||
|
|
||||||
|
if (isAllTime) {
|
||||||
|
const days = Object.keys(d.daily).sort()
|
||||||
|
if (days.length >= 2) {
|
||||||
|
const firstDay = Math.floor(new Date(days[0]).getTime() / 1000)
|
||||||
|
const lastDay = Math.floor(new Date(days[days.length - 1]).getTime() / 1000)
|
||||||
|
const midPoint = Math.floor((firstDay + lastDay) / 2)
|
||||||
|
|
||||||
|
this.reportProgress('分析历史趋势 (1/2)...', 86, onProgress)
|
||||||
|
const earlyRes = await wcdbService.getAggregateStats(sessionIds, 0, midPoint)
|
||||||
|
this.reportProgress('分析历史趋势 (2/2)...', 88, onProgress)
|
||||||
|
const lateRes = await wcdbService.getAggregateStats(sessionIds, midPoint, 0)
|
||||||
|
|
||||||
|
if (earlyRes.success && lateRes.success && earlyRes.data) {
|
||||||
|
const earlyData = earlyRes.data.sessions || {}
|
||||||
|
const lateData = (lateRes.data?.sessions) || {}
|
||||||
|
for (const sid of sessionIds) {
|
||||||
|
const e = earlyData[sid] || { sent: 0, received: 0 }
|
||||||
|
const l = lateData[sid] || { sent: 0, received: 0 }
|
||||||
|
const early = (e.sent || 0) + (e.received || 0)
|
||||||
|
const late = (l.sent || 0) + (l.received || 0)
|
||||||
|
if (early > 100 && early > late * 5) {
|
||||||
|
// 选择前期消息量最多的
|
||||||
|
if (early > maxEarlyCount) {
|
||||||
|
maxEarlyCount = early
|
||||||
|
bestEarlyCount = early
|
||||||
|
bestLateCount = late
|
||||||
|
bestSid = sid
|
||||||
|
bestPeriodDesc = '这段时间以来'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if (year === currentYearNum) {
|
||||||
|
// 当前年份:独立获取过去12个月的滚动数据
|
||||||
|
this.reportProgress('分析近期好友趋势...', 86, onProgress)
|
||||||
|
// 往前数12个月的起点、中点、终点
|
||||||
|
const rollingStart = Math.floor(new Date(now.getFullYear(), now.getMonth() - 11, 1).getTime() / 1000)
|
||||||
|
const rollingMid = Math.floor(new Date(now.getFullYear(), now.getMonth() - 5, 1).getTime() / 1000)
|
||||||
|
const rollingEnd = Math.floor(now.getTime() / 1000)
|
||||||
|
|
||||||
|
const earlyRes = await wcdbService.getAggregateStats(sessionIds, rollingStart, rollingMid - 1)
|
||||||
|
const lateRes = await wcdbService.getAggregateStats(sessionIds, rollingMid, rollingEnd)
|
||||||
|
|
||||||
|
if (earlyRes.success && lateRes.success && earlyRes.data) {
|
||||||
|
const earlyData = earlyRes.data.sessions || {}
|
||||||
|
const lateData = lateRes.data?.sessions || {}
|
||||||
|
for (const sid of sessionIds) {
|
||||||
|
const e = earlyData[sid] || { sent: 0, received: 0 }
|
||||||
|
const l = lateData[sid] || { sent: 0, received: 0 }
|
||||||
|
const early = (e.sent || 0) + (e.received || 0)
|
||||||
|
const late = (l.sent || 0) + (l.received || 0)
|
||||||
|
if (early > 80 && early > late * 5) {
|
||||||
|
// 选择前期消息量最多的
|
||||||
|
if (early > maxEarlyCount) {
|
||||||
|
maxEarlyCount = early
|
||||||
|
bestEarlyCount = early
|
||||||
|
bestLateCount = late
|
||||||
|
bestSid = sid
|
||||||
|
bestPeriodDesc = '去年的这个时候'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 指定完整年份 (1-6 vs 7-12)
|
||||||
|
for (const [sid, stat] of Object.entries(d.sessions)) {
|
||||||
|
const s = stat as any
|
||||||
|
const mWeights = s.monthly || {}
|
||||||
|
let early = 0
|
||||||
|
let late = 0
|
||||||
|
for (let m = 1; m <= 6; m++) early += mWeights[m] || 0
|
||||||
|
for (let m = 7; m <= 12; m++) late += mWeights[m] || 0
|
||||||
|
|
||||||
|
if (early > 80 && early > late * 5) {
|
||||||
|
// 选择前期消息量最多的
|
||||||
|
if (early > maxEarlyCount) {
|
||||||
|
maxEarlyCount = early
|
||||||
|
bestEarlyCount = early
|
||||||
|
bestLateCount = late
|
||||||
|
bestSid = sid
|
||||||
|
bestPeriodDesc = `${year}年上半年`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bestSid) {
|
||||||
|
let info = contactInfoMap.get(bestSid)
|
||||||
|
// 如果 contactInfoMap 中没有该联系人,则单独获取
|
||||||
|
if (!info) {
|
||||||
|
const [displayNameRes, avatarUrlRes] = await Promise.all([
|
||||||
|
wcdbService.getDisplayNames([bestSid]),
|
||||||
|
wcdbService.getAvatarUrls([bestSid])
|
||||||
|
])
|
||||||
|
info = {
|
||||||
|
displayName: displayNameRes.success && displayNameRes.map ? (displayNameRes.map[bestSid] || bestSid) : bestSid,
|
||||||
|
avatarUrl: avatarUrlRes.success && avatarUrlRes.map ? avatarUrlRes.map[bestSid] : undefined
|
||||||
|
}
|
||||||
|
}
|
||||||
|
lostFriend = {
|
||||||
|
username: bestSid,
|
||||||
|
displayName: info?.displayName || bestSid,
|
||||||
|
avatarUrl: info?.avatarUrl,
|
||||||
|
earlyCount: bestEarlyCount,
|
||||||
|
lateCount: bestLateCount,
|
||||||
|
periodDesc: bestPeriodDesc
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const reportData: AnnualReportData = {
|
const reportData: AnnualReportData = {
|
||||||
year,
|
year: reportYear,
|
||||||
totalMessages,
|
totalMessages,
|
||||||
totalFriends: contactStats.size,
|
totalFriends: contactStats.size,
|
||||||
coreFriends,
|
coreFriends,
|
||||||
@@ -915,7 +1112,9 @@ class AnnualReportService {
|
|||||||
mutualFriend,
|
mutualFriend,
|
||||||
socialInitiative,
|
socialInitiative,
|
||||||
responseSpeed,
|
responseSpeed,
|
||||||
topPhrases
|
topPhrases,
|
||||||
|
snsStats: snsStatsResult,
|
||||||
|
lostFriend
|
||||||
}
|
}
|
||||||
|
|
||||||
return { success: true, data: reportData }
|
return { success: true, data: reportData }
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -8,6 +8,7 @@ interface ConfigSchema {
|
|||||||
onboardingDone: boolean
|
onboardingDone: boolean
|
||||||
imageXorKey: number
|
imageXorKey: number
|
||||||
imageAesKey: string
|
imageAesKey: string
|
||||||
|
wxidConfigs: Record<string, { decryptKey?: string; imageXorKey?: number; imageAesKey?: string; updatedAt?: number }>
|
||||||
|
|
||||||
// 缓存相关
|
// 缓存相关
|
||||||
cachePath: string
|
cachePath: string
|
||||||
@@ -19,12 +20,46 @@ interface ConfigSchema {
|
|||||||
themeId: string
|
themeId: string
|
||||||
language: string
|
language: string
|
||||||
logEnabled: boolean
|
logEnabled: boolean
|
||||||
|
llmModelPath: string
|
||||||
|
whisperModelName: string
|
||||||
|
whisperModelDir: string
|
||||||
|
whisperDownloadSource: string
|
||||||
|
autoTranscribeVoice: boolean
|
||||||
|
transcribeLanguages: string[]
|
||||||
|
exportDefaultConcurrency: number
|
||||||
|
analyticsExcludedUsernames: string[]
|
||||||
|
|
||||||
|
// 安全相关
|
||||||
|
authEnabled: boolean
|
||||||
|
authPassword: string // SHA-256 hash
|
||||||
|
authUseHello: boolean
|
||||||
|
|
||||||
|
// 更新相关
|
||||||
|
ignoredUpdateVersion: string
|
||||||
|
|
||||||
|
// 通知
|
||||||
|
notificationEnabled: boolean
|
||||||
|
notificationPosition: 'top-right' | 'top-left' | 'bottom-right' | 'bottom-left'
|
||||||
|
notificationFilterMode: 'all' | 'whitelist' | 'blacklist'
|
||||||
|
notificationFilterList: string[]
|
||||||
}
|
}
|
||||||
|
|
||||||
export class ConfigService {
|
export class ConfigService {
|
||||||
private store: Store<ConfigSchema>
|
private static instance: ConfigService
|
||||||
|
private store!: Store<ConfigSchema>
|
||||||
|
|
||||||
|
static getInstance(): ConfigService {
|
||||||
|
if (!ConfigService.instance) {
|
||||||
|
ConfigService.instance = new ConfigService()
|
||||||
|
}
|
||||||
|
return ConfigService.instance
|
||||||
|
}
|
||||||
|
|
||||||
constructor() {
|
constructor() {
|
||||||
|
if (ConfigService.instance) {
|
||||||
|
return ConfigService.instance
|
||||||
|
}
|
||||||
|
ConfigService.instance = this
|
||||||
this.store = new Store<ConfigSchema>({
|
this.store = new Store<ConfigSchema>({
|
||||||
name: 'WeFlow-config',
|
name: 'WeFlow-config',
|
||||||
defaults: {
|
defaults: {
|
||||||
@@ -34,13 +69,32 @@ export class ConfigService {
|
|||||||
onboardingDone: false,
|
onboardingDone: false,
|
||||||
imageXorKey: 0,
|
imageXorKey: 0,
|
||||||
imageAesKey: '',
|
imageAesKey: '',
|
||||||
|
wxidConfigs: {},
|
||||||
cachePath: '',
|
cachePath: '',
|
||||||
lastOpenedDb: '',
|
lastOpenedDb: '',
|
||||||
lastSession: '',
|
lastSession: '',
|
||||||
theme: 'system',
|
theme: 'system',
|
||||||
themeId: 'cloud-dancer',
|
themeId: 'cloud-dancer',
|
||||||
language: 'zh-CN',
|
language: 'zh-CN',
|
||||||
logEnabled: false
|
logEnabled: false,
|
||||||
|
llmModelPath: '',
|
||||||
|
whisperModelName: 'base',
|
||||||
|
whisperModelDir: '',
|
||||||
|
whisperDownloadSource: 'tsinghua',
|
||||||
|
autoTranscribeVoice: false,
|
||||||
|
transcribeLanguages: ['zh'],
|
||||||
|
exportDefaultConcurrency: 2,
|
||||||
|
analyticsExcludedUsernames: [],
|
||||||
|
|
||||||
|
authEnabled: false,
|
||||||
|
authPassword: '',
|
||||||
|
authUseHello: false,
|
||||||
|
|
||||||
|
ignoredUpdateVersion: '',
|
||||||
|
notificationEnabled: true,
|
||||||
|
notificationPosition: 'top-right',
|
||||||
|
notificationFilterMode: 'all',
|
||||||
|
notificationFilterList: []
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
92
electron/services/contactCacheService.ts
Normal file
92
electron/services/contactCacheService.ts
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
import { join, dirname } from 'path'
|
||||||
|
import { existsSync, mkdirSync, readFileSync, writeFileSync, rmSync } from 'fs'
|
||||||
|
import { app } from 'electron'
|
||||||
|
|
||||||
|
export interface ContactCacheEntry {
|
||||||
|
displayName?: string
|
||||||
|
avatarUrl?: string
|
||||||
|
updatedAt: number
|
||||||
|
}
|
||||||
|
|
||||||
|
export class ContactCacheService {
|
||||||
|
private readonly cacheFilePath: string
|
||||||
|
private cache: Record<string, ContactCacheEntry> = {}
|
||||||
|
|
||||||
|
constructor(cacheBasePath?: string) {
|
||||||
|
const basePath = cacheBasePath && cacheBasePath.trim().length > 0
|
||||||
|
? cacheBasePath
|
||||||
|
: join(app.getPath('documents'), 'WeFlow')
|
||||||
|
this.cacheFilePath = join(basePath, 'contacts.json')
|
||||||
|
this.ensureCacheDir()
|
||||||
|
this.loadCache()
|
||||||
|
}
|
||||||
|
|
||||||
|
private ensureCacheDir() {
|
||||||
|
const dir = dirname(this.cacheFilePath)
|
||||||
|
if (!existsSync(dir)) {
|
||||||
|
mkdirSync(dir, { recursive: true })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private loadCache() {
|
||||||
|
if (!existsSync(this.cacheFilePath)) return
|
||||||
|
try {
|
||||||
|
const raw = readFileSync(this.cacheFilePath, 'utf8')
|
||||||
|
const parsed = JSON.parse(raw)
|
||||||
|
if (parsed && typeof parsed === 'object') {
|
||||||
|
// 清除无效的头像数据(hex 格式而非正确的 base64)
|
||||||
|
for (const key of Object.keys(parsed)) {
|
||||||
|
const entry = parsed[key]
|
||||||
|
if (entry?.avatarUrl && entry.avatarUrl.includes('base64,ffd8')) {
|
||||||
|
// 这是错误的 hex 格式,清除它
|
||||||
|
entry.avatarUrl = undefined
|
||||||
|
}
|
||||||
|
}
|
||||||
|
this.cache = parsed
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('ContactCacheService: 载入缓存失败', error)
|
||||||
|
this.cache = {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
get(username: string): ContactCacheEntry | undefined {
|
||||||
|
return this.cache[username]
|
||||||
|
}
|
||||||
|
|
||||||
|
getAllEntries(): Record<string, ContactCacheEntry> {
|
||||||
|
return { ...this.cache }
|
||||||
|
}
|
||||||
|
|
||||||
|
setEntries(entries: Record<string, ContactCacheEntry>): void {
|
||||||
|
if (Object.keys(entries).length === 0) return
|
||||||
|
let changed = false
|
||||||
|
for (const [username, entry] of Object.entries(entries)) {
|
||||||
|
const existing = this.cache[username]
|
||||||
|
if (!existing || entry.updatedAt >= existing.updatedAt) {
|
||||||
|
this.cache[username] = entry
|
||||||
|
changed = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (changed) {
|
||||||
|
this.persist()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private persist() {
|
||||||
|
try {
|
||||||
|
writeFileSync(this.cacheFilePath, JSON.stringify(this.cache), 'utf8')
|
||||||
|
} catch (error) {
|
||||||
|
console.error('ContactCacheService: 保存缓存失败', error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
clear(): void {
|
||||||
|
this.cache = {}
|
||||||
|
try {
|
||||||
|
rmSync(this.cacheFilePath, { force: true })
|
||||||
|
} catch (error) {
|
||||||
|
console.error('ContactCacheService: 清理缓存失败', error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
159
electron/services/contactExportService.ts
Normal file
159
electron/services/contactExportService.ts
Normal file
@@ -0,0 +1,159 @@
|
|||||||
|
import * as fs from 'fs'
|
||||||
|
import * as path from 'path'
|
||||||
|
import { chatService } from './chatService'
|
||||||
|
|
||||||
|
interface ContactExportOptions {
|
||||||
|
format: 'json' | 'csv' | 'vcf'
|
||||||
|
exportAvatars: boolean
|
||||||
|
contactTypes: {
|
||||||
|
friends: boolean
|
||||||
|
groups: boolean
|
||||||
|
officials: boolean
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 联系人导出服务
|
||||||
|
*/
|
||||||
|
class ContactExportService {
|
||||||
|
/**
|
||||||
|
* 导出联系人
|
||||||
|
*/
|
||||||
|
async exportContacts(
|
||||||
|
outputDir: string,
|
||||||
|
options: ContactExportOptions
|
||||||
|
): Promise<{ success: boolean; successCount?: number; error?: string }> {
|
||||||
|
try {
|
||||||
|
// 获取所有联系人
|
||||||
|
const contactsResult = await chatService.getContacts()
|
||||||
|
if (!contactsResult.success || !contactsResult.contacts) {
|
||||||
|
return { success: false, error: contactsResult.error || '获取联系人失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
let contacts = contactsResult.contacts
|
||||||
|
|
||||||
|
// 根据类型过滤
|
||||||
|
contacts = contacts.filter(c => {
|
||||||
|
if (c.type === 'friend' && !options.contactTypes.friends) return false
|
||||||
|
if (c.type === 'group' && !options.contactTypes.groups) return false
|
||||||
|
if (c.type === 'official' && !options.contactTypes.officials) return false
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
if (contacts.length === 0) {
|
||||||
|
return { success: false, error: '没有符合条件的联系人' }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 确保输出目录存在
|
||||||
|
if (!fs.existsSync(outputDir)) {
|
||||||
|
fs.mkdirSync(outputDir, { recursive: true })
|
||||||
|
}
|
||||||
|
|
||||||
|
const timestamp = new Date().toISOString().replace(/[:.]/g, '-').slice(0, -5)
|
||||||
|
let outputPath: string
|
||||||
|
|
||||||
|
switch (options.format) {
|
||||||
|
case 'json':
|
||||||
|
outputPath = path.join(outputDir, `contacts_${timestamp}.json`)
|
||||||
|
await this.exportToJSON(contacts, outputPath)
|
||||||
|
break
|
||||||
|
case 'csv':
|
||||||
|
outputPath = path.join(outputDir, `contacts_${timestamp}.csv`)
|
||||||
|
await this.exportToCSV(contacts, outputPath)
|
||||||
|
break
|
||||||
|
case 'vcf':
|
||||||
|
outputPath = path.join(outputDir, `contacts_${timestamp}.vcf`)
|
||||||
|
await this.exportToVCF(contacts, outputPath)
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
return { success: false, error: '不支持的导出格式' }
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success: true, successCount: contacts.length }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 导出为JSON格式
|
||||||
|
*/
|
||||||
|
private async exportToJSON(contacts: any[], outputPath: string): Promise<void> {
|
||||||
|
const data = {
|
||||||
|
exportedAt: new Date().toISOString(),
|
||||||
|
count: contacts.length,
|
||||||
|
contacts: contacts.map(c => ({
|
||||||
|
username: c.username,
|
||||||
|
displayName: c.displayName,
|
||||||
|
remark: c.remark,
|
||||||
|
nickname: c.nickname,
|
||||||
|
type: c.type
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
fs.writeFileSync(outputPath, JSON.stringify(data, null, 2), 'utf-8')
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 导出为CSV格式
|
||||||
|
*/
|
||||||
|
private async exportToCSV(contacts: any[], outputPath: string): Promise<void> {
|
||||||
|
const headers = ['用户名', '显示名称', '备注', '昵称', '类型']
|
||||||
|
const rows = contacts.map(c => [
|
||||||
|
c.username || '',
|
||||||
|
c.displayName || '',
|
||||||
|
c.remark || '',
|
||||||
|
c.nickname || '',
|
||||||
|
this.getTypeLabel(c.type)
|
||||||
|
])
|
||||||
|
|
||||||
|
const csvContent = [
|
||||||
|
headers.join(','),
|
||||||
|
...rows.map(row => row.map(cell => `"${cell}"`).join(','))
|
||||||
|
].join('\n')
|
||||||
|
|
||||||
|
fs.writeFileSync(outputPath, '\uFEFF' + csvContent, 'utf-8') // 添加BOM以支持Excel
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 导出为VCF格式(vCard)
|
||||||
|
*/
|
||||||
|
private async exportToVCF(contacts: any[], outputPath: string): Promise<void> {
|
||||||
|
const vcards = contacts
|
||||||
|
.filter(c => c.type === 'friend') // VCF通常只用于个人联系人
|
||||||
|
.map(c => {
|
||||||
|
const lines = ['BEGIN:VCARD', 'VERSION:3.0']
|
||||||
|
|
||||||
|
// 全名
|
||||||
|
lines.push(`FN:${c.displayName || c.username}`)
|
||||||
|
|
||||||
|
// 昵称
|
||||||
|
if (c.nickname) {
|
||||||
|
lines.push(`NICKNAME:${c.nickname}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 备注
|
||||||
|
if (c.remark) {
|
||||||
|
lines.push(`NOTE:${c.remark}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 微信ID
|
||||||
|
lines.push(`X-WECHAT-ID:${c.username}`)
|
||||||
|
|
||||||
|
lines.push('END:VCARD')
|
||||||
|
return lines.join('\r\n')
|
||||||
|
})
|
||||||
|
|
||||||
|
fs.writeFileSync(outputPath, vcards.join('\r\n\r\n'), 'utf-8')
|
||||||
|
}
|
||||||
|
|
||||||
|
private getTypeLabel(type: string): string {
|
||||||
|
switch (type) {
|
||||||
|
case 'friend': return '好友'
|
||||||
|
case 'group': return '群聊'
|
||||||
|
case 'official': return '公众号'
|
||||||
|
default: return '其他'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const contactExportService = new ContactExportService()
|
||||||
@@ -18,8 +18,7 @@ export class DbPathService {
|
|||||||
|
|
||||||
// 微信4.x 数据目录
|
// 微信4.x 数据目录
|
||||||
possiblePaths.push(join(home, 'Documents', 'xwechat_files'))
|
possiblePaths.push(join(home, 'Documents', 'xwechat_files'))
|
||||||
// 旧版微信数据目录
|
|
||||||
possiblePaths.push(join(home, 'Documents', 'WeChat Files'))
|
|
||||||
|
|
||||||
for (const path of possiblePaths) {
|
for (const path of possiblePaths) {
|
||||||
if (existsSync(path)) {
|
if (existsSync(path)) {
|
||||||
@@ -119,6 +118,48 @@ export class DbPathService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 扫描目录名候选(仅包含下划线的文件夹,排除 all_users)
|
||||||
|
*/
|
||||||
|
scanWxidCandidates(rootPath: string): WxidInfo[] {
|
||||||
|
const wxids: WxidInfo[] = []
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (existsSync(rootPath)) {
|
||||||
|
const entries = readdirSync(rootPath)
|
||||||
|
for (const entry of entries) {
|
||||||
|
const entryPath = join(rootPath, entry)
|
||||||
|
let stat: ReturnType<typeof statSync>
|
||||||
|
try {
|
||||||
|
stat = statSync(entryPath)
|
||||||
|
} catch {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!stat.isDirectory()) continue
|
||||||
|
const lower = entry.toLowerCase()
|
||||||
|
if (lower === 'all_users') continue
|
||||||
|
if (!entry.includes('_')) continue
|
||||||
|
|
||||||
|
wxids.push({ wxid: entry, modifiedTime: stat.mtimeMs })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (wxids.length === 0) {
|
||||||
|
const rootName = basename(rootPath)
|
||||||
|
if (rootName.includes('_') && rootName.toLowerCase() !== 'all_users') {
|
||||||
|
const rootStat = statSync(rootPath)
|
||||||
|
wxids.push({ wxid: rootName, modifiedTime: rootStat.mtimeMs })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
|
||||||
|
return wxids.sort((a, b) => {
|
||||||
|
if (b.modifiedTime !== a.modifiedTime) return b.modifiedTime - a.modifiedTime
|
||||||
|
return a.wxid.localeCompare(b.wxid)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 扫描 wxid 列表
|
* 扫描 wxid 列表
|
||||||
*/
|
*/
|
||||||
|
|||||||
466
electron/services/dualReportService.ts
Normal file
466
electron/services/dualReportService.ts
Normal file
@@ -0,0 +1,466 @@
|
|||||||
|
import { parentPort } from 'worker_threads'
|
||||||
|
import { wcdbService } from './wcdbService'
|
||||||
|
|
||||||
|
export interface DualReportMessage {
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DualReportFirstChat {
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
senderUsername?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DualReportStats {
|
||||||
|
totalMessages: number
|
||||||
|
totalWords: number
|
||||||
|
imageCount: number
|
||||||
|
voiceCount: number
|
||||||
|
emojiCount: number
|
||||||
|
myTopEmojiMd5?: string
|
||||||
|
friendTopEmojiMd5?: string
|
||||||
|
myTopEmojiUrl?: string
|
||||||
|
friendTopEmojiUrl?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DualReportData {
|
||||||
|
year: number
|
||||||
|
selfName: string
|
||||||
|
friendUsername: string
|
||||||
|
friendName: string
|
||||||
|
firstChat: DualReportFirstChat | null
|
||||||
|
firstChatMessages?: DualReportMessage[]
|
||||||
|
yearFirstChat?: {
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
friendName: string
|
||||||
|
firstThreeMessages: DualReportMessage[]
|
||||||
|
} | null
|
||||||
|
stats: DualReportStats
|
||||||
|
topPhrases: Array<{ phrase: string; count: number }>
|
||||||
|
}
|
||||||
|
|
||||||
|
class DualReportService {
|
||||||
|
private broadcastProgress(status: string, progress: number) {
|
||||||
|
if (parentPort) {
|
||||||
|
parentPort.postMessage({
|
||||||
|
type: 'dualReport:progress',
|
||||||
|
data: { status, progress }
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private reportProgress(status: string, progress: number, onProgress?: (status: string, progress: number) => void) {
|
||||||
|
if (onProgress) {
|
||||||
|
onProgress(status, progress)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
this.broadcastProgress(status, progress)
|
||||||
|
}
|
||||||
|
|
||||||
|
private cleanAccountDirName(dirName: string): string {
|
||||||
|
const trimmed = dirName.trim()
|
||||||
|
if (!trimmed) return trimmed
|
||||||
|
if (trimmed.toLowerCase().startsWith('wxid_')) {
|
||||||
|
const match = trimmed.match(/^(wxid_[^_]+)/i)
|
||||||
|
if (match) return match[1]
|
||||||
|
return trimmed
|
||||||
|
}
|
||||||
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
|
const cleaned = suffixMatch ? suffixMatch[1] : trimmed
|
||||||
|
|
||||||
|
return cleaned
|
||||||
|
}
|
||||||
|
|
||||||
|
private async ensureConnectedWithConfig(
|
||||||
|
dbPath: string,
|
||||||
|
decryptKey: string,
|
||||||
|
wxid: string
|
||||||
|
): Promise<{ success: boolean; cleanedWxid?: string; rawWxid?: string; error?: string }> {
|
||||||
|
if (!wxid) return { success: false, error: '未配置微信ID' }
|
||||||
|
if (!dbPath) return { success: false, error: '未配置数据库路径' }
|
||||||
|
if (!decryptKey) return { success: false, error: '未配置解密密钥' }
|
||||||
|
|
||||||
|
const cleanedWxid = this.cleanAccountDirName(wxid)
|
||||||
|
const ok = await wcdbService.open(dbPath, decryptKey, cleanedWxid)
|
||||||
|
if (!ok) return { success: false, error: 'WCDB 打开失败' }
|
||||||
|
return { success: true, cleanedWxid, rawWxid: wxid }
|
||||||
|
}
|
||||||
|
|
||||||
|
private decodeMessageContent(messageContent: any, compressContent: any): string {
|
||||||
|
let content = this.decodeMaybeCompressed(compressContent)
|
||||||
|
if (!content || content.length === 0) {
|
||||||
|
content = this.decodeMaybeCompressed(messageContent)
|
||||||
|
}
|
||||||
|
return content
|
||||||
|
}
|
||||||
|
|
||||||
|
private decodeMaybeCompressed(raw: any): string {
|
||||||
|
if (!raw) return ''
|
||||||
|
if (typeof raw === 'string') {
|
||||||
|
if (raw.length === 0) return ''
|
||||||
|
// 只有当字符串足够长(超过16字符)且看起来像 hex 时才尝试解码
|
||||||
|
// 短字符串(如 "123456" 等纯数字)容易被误判为 hex
|
||||||
|
if (raw.length > 16 && this.looksLikeHex(raw)) {
|
||||||
|
const bytes = Buffer.from(raw, 'hex')
|
||||||
|
if (bytes.length > 0) return this.decodeBinaryContent(bytes)
|
||||||
|
}
|
||||||
|
// 只有当字符串足够长(超过16字符)且看起来像 base64 时才尝试解码
|
||||||
|
// 短字符串(如 "test", "home" 等)容易被误判为 base64
|
||||||
|
if (raw.length > 16 && this.looksLikeBase64(raw)) {
|
||||||
|
try {
|
||||||
|
const bytes = Buffer.from(raw, 'base64')
|
||||||
|
return this.decodeBinaryContent(bytes)
|
||||||
|
} catch {
|
||||||
|
return raw
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return raw
|
||||||
|
}
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
|
||||||
|
private decodeBinaryContent(data: Buffer): string {
|
||||||
|
if (data.length === 0) return ''
|
||||||
|
try {
|
||||||
|
if (data.length >= 4) {
|
||||||
|
const magic = data.readUInt32LE(0)
|
||||||
|
if (magic === 0xFD2FB528) {
|
||||||
|
const fzstd = require('fzstd')
|
||||||
|
const decompressed = fzstd.decompress(data)
|
||||||
|
return Buffer.from(decompressed).toString('utf-8')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const decoded = data.toString('utf-8')
|
||||||
|
const replacementCount = (decoded.match(/\uFFFD/g) || []).length
|
||||||
|
if (replacementCount < decoded.length * 0.2) {
|
||||||
|
return decoded.replace(/\uFFFD/g, '')
|
||||||
|
}
|
||||||
|
return data.toString('latin1')
|
||||||
|
} catch {
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private looksLikeHex(s: string): boolean {
|
||||||
|
if (s.length % 2 !== 0) return false
|
||||||
|
return /^[0-9a-fA-F]+$/.test(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
private looksLikeBase64(s: string): boolean {
|
||||||
|
if (s.length % 4 !== 0) return false
|
||||||
|
return /^[A-Za-z0-9+/=]+$/.test(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
private formatDateTime(milliseconds: number): string {
|
||||||
|
const dt = new Date(milliseconds)
|
||||||
|
const month = String(dt.getMonth() + 1).padStart(2, '0')
|
||||||
|
const day = String(dt.getDate()).padStart(2, '0')
|
||||||
|
const hour = String(dt.getHours()).padStart(2, '0')
|
||||||
|
const minute = String(dt.getMinutes()).padStart(2, '0')
|
||||||
|
return `${month}/${day} ${hour}:${minute}`
|
||||||
|
}
|
||||||
|
|
||||||
|
private extractEmojiUrl(content: string): string | undefined {
|
||||||
|
if (!content) return undefined
|
||||||
|
const attrMatch = /cdnurl\s*=\s*['"]([^'"]+)['"]/i.exec(content)
|
||||||
|
if (attrMatch) {
|
||||||
|
let url = attrMatch[1].replace(/&/g, '&')
|
||||||
|
try {
|
||||||
|
if (url.includes('%')) {
|
||||||
|
url = decodeURIComponent(url)
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
const tagMatch = /cdnurl[^>]*>([^<]+)/i.exec(content)
|
||||||
|
return tagMatch?.[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
private extractEmojiMd5(content: string): string | undefined {
|
||||||
|
if (!content) return undefined
|
||||||
|
const match = /md5="([^"]+)"/i.exec(content) || /<md5>([^<]+)<\/md5>/i.exec(content)
|
||||||
|
return match?.[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
private async getDisplayName(username: string, fallback: string): Promise<string> {
|
||||||
|
const result = await wcdbService.getDisplayNames([username])
|
||||||
|
if (result.success && result.map) {
|
||||||
|
return result.map[username] || fallback
|
||||||
|
}
|
||||||
|
return fallback
|
||||||
|
}
|
||||||
|
|
||||||
|
private resolveIsSent(row: any, rawWxid?: string, cleanedWxid?: string): boolean {
|
||||||
|
const isSendRaw = row.computed_is_send ?? row.is_send
|
||||||
|
if (isSendRaw !== undefined && isSendRaw !== null) {
|
||||||
|
return parseInt(isSendRaw, 10) === 1
|
||||||
|
}
|
||||||
|
const sender = String(row.sender_username || row.sender || row.talker || '').toLowerCase()
|
||||||
|
if (!sender) return false
|
||||||
|
const rawLower = rawWxid ? rawWxid.toLowerCase() : ''
|
||||||
|
const cleanedLower = cleanedWxid ? cleanedWxid.toLowerCase() : ''
|
||||||
|
return !!(
|
||||||
|
sender === rawLower ||
|
||||||
|
sender === cleanedLower ||
|
||||||
|
(rawLower && rawLower.startsWith(sender + '_')) ||
|
||||||
|
(cleanedLower && cleanedLower.startsWith(sender + '_'))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private async getFirstMessages(
|
||||||
|
sessionId: string,
|
||||||
|
limit: number,
|
||||||
|
beginTimestamp: number,
|
||||||
|
endTimestamp: number
|
||||||
|
): Promise<any[]> {
|
||||||
|
const safeBegin = Math.max(0, beginTimestamp || 0)
|
||||||
|
const safeEnd = endTimestamp && endTimestamp > 0 ? endTimestamp : Math.floor(Date.now() / 1000)
|
||||||
|
const cursorResult = await wcdbService.openMessageCursor(sessionId, Math.max(1, limit), true, safeBegin, safeEnd)
|
||||||
|
if (!cursorResult.success || !cursorResult.cursor) return []
|
||||||
|
try {
|
||||||
|
const rows: any[] = []
|
||||||
|
let hasMore = true
|
||||||
|
while (hasMore && rows.length < limit) {
|
||||||
|
const batch = await wcdbService.fetchMessageBatch(cursorResult.cursor)
|
||||||
|
if (!batch.success || !batch.rows) break
|
||||||
|
for (const row of batch.rows) {
|
||||||
|
rows.push(row)
|
||||||
|
if (rows.length >= limit) break
|
||||||
|
}
|
||||||
|
hasMore = batch.hasMore === true
|
||||||
|
}
|
||||||
|
return rows.slice(0, limit)
|
||||||
|
} finally {
|
||||||
|
await wcdbService.closeMessageCursor(cursorResult.cursor)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async generateReportWithConfig(params: {
|
||||||
|
year: number
|
||||||
|
friendUsername: string
|
||||||
|
dbPath: string
|
||||||
|
decryptKey: string
|
||||||
|
wxid: string
|
||||||
|
onProgress?: (status: string, progress: number) => void
|
||||||
|
}): Promise<{ success: boolean; data?: DualReportData; error?: string }> {
|
||||||
|
try {
|
||||||
|
const { year, friendUsername, dbPath, decryptKey, wxid, onProgress } = params
|
||||||
|
this.reportProgress('正在连接数据库...', 5, onProgress)
|
||||||
|
const conn = await this.ensureConnectedWithConfig(dbPath, decryptKey, wxid)
|
||||||
|
if (!conn.success || !conn.cleanedWxid || !conn.rawWxid) return { success: false, error: conn.error }
|
||||||
|
|
||||||
|
const cleanedWxid = conn.cleanedWxid
|
||||||
|
const rawWxid = conn.rawWxid
|
||||||
|
|
||||||
|
const reportYear = year <= 0 ? 0 : year
|
||||||
|
const isAllTime = reportYear === 0
|
||||||
|
const startTime = isAllTime ? 0 : Math.floor(new Date(reportYear, 0, 1).getTime() / 1000)
|
||||||
|
const endTime = isAllTime ? 0 : Math.floor(new Date(reportYear, 11, 31, 23, 59, 59).getTime() / 1000)
|
||||||
|
|
||||||
|
this.reportProgress('加载联系人信息...', 10, onProgress)
|
||||||
|
const friendName = await this.getDisplayName(friendUsername, friendUsername)
|
||||||
|
let myName = await this.getDisplayName(rawWxid, rawWxid)
|
||||||
|
if (myName === rawWxid && cleanedWxid && cleanedWxid !== rawWxid) {
|
||||||
|
myName = await this.getDisplayName(cleanedWxid, rawWxid)
|
||||||
|
}
|
||||||
|
|
||||||
|
this.reportProgress('获取首条聊天记录...', 15, onProgress)
|
||||||
|
const firstRows = await this.getFirstMessages(friendUsername, 3, 0, 0)
|
||||||
|
let firstChat: DualReportFirstChat | null = null
|
||||||
|
if (firstRows.length > 0) {
|
||||||
|
const row = firstRows[0]
|
||||||
|
const createTime = parseInt(row.create_time || '0', 10) * 1000
|
||||||
|
const content = this.decodeMessageContent(row.message_content, row.compress_content)
|
||||||
|
firstChat = {
|
||||||
|
createTime,
|
||||||
|
createTimeStr: this.formatDateTime(createTime),
|
||||||
|
content: String(content || ''),
|
||||||
|
isSentByMe: this.resolveIsSent(row, rawWxid, cleanedWxid),
|
||||||
|
senderUsername: row.sender_username || row.sender
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const firstChatMessages: DualReportMessage[] = firstRows.map((row) => {
|
||||||
|
const msgTime = parseInt(row.create_time || '0', 10) * 1000
|
||||||
|
const msgContent = this.decodeMessageContent(row.message_content, row.compress_content)
|
||||||
|
return {
|
||||||
|
content: String(msgContent || ''),
|
||||||
|
isSentByMe: this.resolveIsSent(row, rawWxid, cleanedWxid),
|
||||||
|
createTime: msgTime,
|
||||||
|
createTimeStr: this.formatDateTime(msgTime)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
let yearFirstChat: DualReportData['yearFirstChat'] = null
|
||||||
|
if (!isAllTime) {
|
||||||
|
this.reportProgress('获取今年首次聊天...', 20, onProgress)
|
||||||
|
const firstYearRows = await this.getFirstMessages(friendUsername, 3, startTime, endTime)
|
||||||
|
if (firstYearRows.length > 0) {
|
||||||
|
const firstRow = firstYearRows[0]
|
||||||
|
const createTime = parseInt(firstRow.create_time || '0', 10) * 1000
|
||||||
|
const firstThreeMessages: DualReportMessage[] = firstYearRows.map((row) => {
|
||||||
|
const msgTime = parseInt(row.create_time || '0', 10) * 1000
|
||||||
|
const msgContent = this.decodeMessageContent(row.message_content, row.compress_content)
|
||||||
|
return {
|
||||||
|
content: String(msgContent || ''),
|
||||||
|
isSentByMe: this.resolveIsSent(row, rawWxid, cleanedWxid),
|
||||||
|
createTime: msgTime,
|
||||||
|
createTimeStr: this.formatDateTime(msgTime)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yearFirstChat = {
|
||||||
|
createTime,
|
||||||
|
createTimeStr: this.formatDateTime(createTime),
|
||||||
|
content: String(this.decodeMessageContent(firstRow.message_content, firstRow.compress_content) || ''),
|
||||||
|
isSentByMe: this.resolveIsSent(firstRow, rawWxid, cleanedWxid),
|
||||||
|
friendName,
|
||||||
|
firstThreeMessages
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
this.reportProgress('统计聊天数据...', 30, onProgress)
|
||||||
|
const stats: DualReportStats = {
|
||||||
|
totalMessages: 0,
|
||||||
|
totalWords: 0,
|
||||||
|
imageCount: 0,
|
||||||
|
voiceCount: 0,
|
||||||
|
emojiCount: 0
|
||||||
|
}
|
||||||
|
const wordCountMap = new Map<string, number>()
|
||||||
|
const myEmojiCounts = new Map<string, number>()
|
||||||
|
const friendEmojiCounts = new Map<string, number>()
|
||||||
|
const myEmojiUrlMap = new Map<string, string>()
|
||||||
|
const friendEmojiUrlMap = new Map<string, string>()
|
||||||
|
|
||||||
|
const messageCountResult = await wcdbService.getMessageCount(friendUsername)
|
||||||
|
const totalForProgress = messageCountResult.success && messageCountResult.count
|
||||||
|
? messageCountResult.count
|
||||||
|
: 0
|
||||||
|
let processed = 0
|
||||||
|
let lastProgressAt = 0
|
||||||
|
|
||||||
|
const cursorResult = await wcdbService.openMessageCursor(friendUsername, 1000, true, startTime, endTime)
|
||||||
|
if (!cursorResult.success || !cursorResult.cursor) {
|
||||||
|
return { success: false, error: cursorResult.error || '打开消息游标失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
let hasMore = true
|
||||||
|
while (hasMore) {
|
||||||
|
const batch = await wcdbService.fetchMessageBatch(cursorResult.cursor)
|
||||||
|
if (!batch.success || !batch.rows) break
|
||||||
|
for (const row of batch.rows) {
|
||||||
|
const localType = parseInt(row.local_type || row.type || '1', 10)
|
||||||
|
const isSent = this.resolveIsSent(row, rawWxid, cleanedWxid)
|
||||||
|
stats.totalMessages += 1
|
||||||
|
|
||||||
|
if (localType === 3) stats.imageCount += 1
|
||||||
|
if (localType === 34) stats.voiceCount += 1
|
||||||
|
if (localType === 47) {
|
||||||
|
stats.emojiCount += 1
|
||||||
|
const content = this.decodeMessageContent(row.message_content, row.compress_content)
|
||||||
|
const md5 = this.extractEmojiMd5(content)
|
||||||
|
const url = this.extractEmojiUrl(content)
|
||||||
|
if (md5) {
|
||||||
|
const targetMap = isSent ? myEmojiCounts : friendEmojiCounts
|
||||||
|
targetMap.set(md5, (targetMap.get(md5) || 0) + 1)
|
||||||
|
if (url) {
|
||||||
|
const urlMap = isSent ? myEmojiUrlMap : friendEmojiUrlMap
|
||||||
|
if (!urlMap.has(md5)) urlMap.set(md5, url)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (localType === 1 || localType === 244813135921) {
|
||||||
|
const content = this.decodeMessageContent(row.message_content, row.compress_content)
|
||||||
|
const text = String(content || '').trim()
|
||||||
|
if (text.length > 0) {
|
||||||
|
stats.totalWords += text.replace(/\s+/g, '').length
|
||||||
|
const normalized = text.replace(/\s+/g, ' ').trim()
|
||||||
|
if (normalized.length >= 2 &&
|
||||||
|
normalized.length <= 50 &&
|
||||||
|
!normalized.includes('http') &&
|
||||||
|
!normalized.includes('<') &&
|
||||||
|
!normalized.startsWith('[') &&
|
||||||
|
!normalized.startsWith('<?xml')) {
|
||||||
|
wordCountMap.set(normalized, (wordCountMap.get(normalized) || 0) + 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (totalForProgress > 0) {
|
||||||
|
processed++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
hasMore = batch.hasMore === true
|
||||||
|
|
||||||
|
const now = Date.now()
|
||||||
|
if (now - lastProgressAt > 200) {
|
||||||
|
if (totalForProgress > 0) {
|
||||||
|
const ratio = Math.min(1, processed / totalForProgress)
|
||||||
|
const progress = 30 + Math.floor(ratio * 50)
|
||||||
|
this.reportProgress('统计聊天数据...', progress, onProgress)
|
||||||
|
}
|
||||||
|
lastProgressAt = now
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
await wcdbService.closeMessageCursor(cursorResult.cursor)
|
||||||
|
}
|
||||||
|
|
||||||
|
const pickTop = (map: Map<string, number>): string | undefined => {
|
||||||
|
let topKey: string | undefined
|
||||||
|
let topCount = -1
|
||||||
|
for (const [key, count] of map.entries()) {
|
||||||
|
if (count > topCount) {
|
||||||
|
topCount = count
|
||||||
|
topKey = key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return topKey
|
||||||
|
}
|
||||||
|
|
||||||
|
const myTopEmojiMd5 = pickTop(myEmojiCounts)
|
||||||
|
const friendTopEmojiMd5 = pickTop(friendEmojiCounts)
|
||||||
|
|
||||||
|
stats.myTopEmojiMd5 = myTopEmojiMd5
|
||||||
|
stats.friendTopEmojiMd5 = friendTopEmojiMd5
|
||||||
|
stats.myTopEmojiUrl = myTopEmojiMd5 ? myEmojiUrlMap.get(myTopEmojiMd5) : undefined
|
||||||
|
stats.friendTopEmojiUrl = friendTopEmojiMd5 ? friendEmojiUrlMap.get(friendTopEmojiMd5) : undefined
|
||||||
|
|
||||||
|
this.reportProgress('生成常用语词云...', 85, onProgress)
|
||||||
|
const topPhrases = Array.from(wordCountMap.entries())
|
||||||
|
.filter(([_, count]) => count >= 2)
|
||||||
|
.sort((a, b) => b[1] - a[1])
|
||||||
|
.slice(0, 50)
|
||||||
|
.map(([phrase, count]) => ({ phrase, count }))
|
||||||
|
|
||||||
|
const reportData: DualReportData = {
|
||||||
|
year: reportYear,
|
||||||
|
selfName: myName,
|
||||||
|
friendUsername,
|
||||||
|
friendName,
|
||||||
|
firstChat,
|
||||||
|
firstChatMessages,
|
||||||
|
yearFirstChat,
|
||||||
|
stats,
|
||||||
|
topPhrases
|
||||||
|
}
|
||||||
|
|
||||||
|
this.reportProgress('双人报告生成完成', 100, onProgress)
|
||||||
|
return { success: true, data: reportData }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const dualReportService = new DualReportService()
|
||||||
331
electron/services/exportHtml.css
Normal file
331
electron/services/exportHtml.css
Normal file
@@ -0,0 +1,331 @@
|
|||||||
|
:root {
|
||||||
|
color-scheme: light;
|
||||||
|
--bg: #f6f7fb;
|
||||||
|
--card: #ffffff;
|
||||||
|
--text: #1f2a37;
|
||||||
|
--muted: #6b7280;
|
||||||
|
--accent: #4f46e5;
|
||||||
|
--sent: #dbeafe;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #e5e7eb;
|
||||||
|
--shadow: 0 12px 30px rgba(15, 23, 42, 0.08);
|
||||||
|
--radius: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
margin: 0;
|
||||||
|
font-family: "PingFang SC", "Microsoft YaHei", system-ui, -apple-system, sans-serif;
|
||||||
|
background: var(--bg);
|
||||||
|
color: var(--text);
|
||||||
|
}
|
||||||
|
|
||||||
|
.page {
|
||||||
|
max-width: 1080px;
|
||||||
|
margin: 32px auto 60px;
|
||||||
|
padding: 0 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header {
|
||||||
|
background: var(--card);
|
||||||
|
border-radius: var(--radius);
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
padding: 24px;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.title {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 600;
|
||||||
|
margin: 0 0 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.meta {
|
||||||
|
color: var(--muted);
|
||||||
|
font-size: 14px;
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.controls {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(220px, 1fr));
|
||||||
|
gap: 16px;
|
||||||
|
margin-top: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control label {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--muted);
|
||||||
|
}
|
||||||
|
|
||||||
|
.control input,
|
||||||
|
.control select,
|
||||||
|
.control button {
|
||||||
|
border-radius: 12px;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
padding: 10px 12px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-family: inherit;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control button {
|
||||||
|
background: var(--accent);
|
||||||
|
color: #fff;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: transform 0.1s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control button:active {
|
||||||
|
transform: scale(0.98);
|
||||||
|
}
|
||||||
|
|
||||||
|
.stats {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--muted);
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.hidden {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-time {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--muted);
|
||||||
|
margin-bottom: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-row {
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
align-items: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.sent .message-row {
|
||||||
|
flex-direction: row-reverse;
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 12px;
|
||||||
|
background: #eef2ff;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
overflow: hidden;
|
||||||
|
flex-shrink: 0;
|
||||||
|
color: #475569;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
|
||||||
|
.bubble {
|
||||||
|
max-width: min(70%, 720px);
|
||||||
|
background: var(--received);
|
||||||
|
border-radius: 18px;
|
||||||
|
padding: 12px 14px;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
box-shadow: 0 8px 20px rgba(15, 23, 42, 0.06);
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.sent .bubble {
|
||||||
|
background: var(--sent);
|
||||||
|
border-color: transparent;
|
||||||
|
}
|
||||||
|
|
||||||
|
.sender-name {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--muted);
|
||||||
|
margin-bottom: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-content {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 8px;
|
||||||
|
font-size: 14px;
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-text {
|
||||||
|
word-break: break-word;
|
||||||
|
}
|
||||||
|
|
||||||
|
.inline-emoji {
|
||||||
|
width: 22px;
|
||||||
|
height: 22px;
|
||||||
|
vertical-align: text-bottom;
|
||||||
|
margin: 0 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media {
|
||||||
|
border-radius: 14px;
|
||||||
|
max-width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.previewable {
|
||||||
|
cursor: zoom-in;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.image,
|
||||||
|
.message-media.emoji {
|
||||||
|
max-height: 260px;
|
||||||
|
object-fit: contain;
|
||||||
|
background: #f1f5f9;
|
||||||
|
padding: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.emoji {
|
||||||
|
max-height: 160px;
|
||||||
|
width: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.video {
|
||||||
|
max-height: 360px;
|
||||||
|
background: #111827;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.audio {
|
||||||
|
width: 260px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0;
|
||||||
|
background: rgba(15, 23, 42, 0.7);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
opacity: 0;
|
||||||
|
pointer-events: none;
|
||||||
|
transition: opacity 0.2s ease;
|
||||||
|
z-index: 999;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview.active {
|
||||||
|
opacity: 1;
|
||||||
|
pointer-events: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview img {
|
||||||
|
max-width: min(90vw, 1200px);
|
||||||
|
max-height: 90vh;
|
||||||
|
border-radius: 18px;
|
||||||
|
box-shadow: 0 20px 40px rgba(0, 0, 0, 0.35);
|
||||||
|
background: #0f172a;
|
||||||
|
transition: transform 0.1s ease;
|
||||||
|
cursor: zoom-out;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="cloud-dancer"] {
|
||||||
|
--accent: #6b8cff;
|
||||||
|
--sent: #e0e7ff;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #d8e0f7;
|
||||||
|
--bg: #f6f7fb;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="corundum-blue"] {
|
||||||
|
--accent: #2563eb;
|
||||||
|
--sent: #dbeafe;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #c7d2fe;
|
||||||
|
--bg: #eef2ff;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="kiwi-green"] {
|
||||||
|
--accent: #16a34a;
|
||||||
|
--sent: #dcfce7;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #bbf7d0;
|
||||||
|
--bg: #f0fdf4;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="spicy-red"] {
|
||||||
|
--accent: #e11d48;
|
||||||
|
--sent: #ffe4e6;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #fecdd3;
|
||||||
|
--bg: #fff1f2;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="teal-water"] {
|
||||||
|
--accent: #0f766e;
|
||||||
|
--sent: #ccfbf1;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #99f6e4;
|
||||||
|
--bg: #f0fdfa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.highlight {
|
||||||
|
outline: 2px solid var(--accent);
|
||||||
|
outline-offset: 4px;
|
||||||
|
border-radius: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.empty {
|
||||||
|
text-align: center;
|
||||||
|
color: var(--muted);
|
||||||
|
padding: 40px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Virtual Scroll */
|
||||||
|
.virtual-scroll-container {
|
||||||
|
height: calc(100vh - 180px);
|
||||||
|
/* Adjust based on header height */
|
||||||
|
overflow-y: auto;
|
||||||
|
position: relative;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
border-radius: var(--radius);
|
||||||
|
background: var(--bg);
|
||||||
|
margin-top: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.virtual-scroll-spacer {
|
||||||
|
opacity: 0;
|
||||||
|
pointer-events: none;
|
||||||
|
width: 1px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.virtual-scroll-content {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-list {
|
||||||
|
/* Override message-list to be inside virtual scroll */
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
302
electron/services/exportHtmlStyles.ts
Normal file
302
electron/services/exportHtmlStyles.ts
Normal file
@@ -0,0 +1,302 @@
|
|||||||
|
export const EXPORT_HTML_STYLES = `:root {
|
||||||
|
color-scheme: light;
|
||||||
|
--bg: #f6f7fb;
|
||||||
|
--card: #ffffff;
|
||||||
|
--text: #1f2a37;
|
||||||
|
--muted: #6b7280;
|
||||||
|
--accent: #4f46e5;
|
||||||
|
--sent: #dbeafe;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #e5e7eb;
|
||||||
|
--shadow: 0 12px 30px rgba(15, 23, 42, 0.08);
|
||||||
|
--radius: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
margin: 0;
|
||||||
|
font-family: "PingFang SC", "Microsoft YaHei", system-ui, -apple-system, sans-serif;
|
||||||
|
background: var(--bg);
|
||||||
|
color: var(--text);
|
||||||
|
}
|
||||||
|
|
||||||
|
.page {
|
||||||
|
max-width: 1080px;
|
||||||
|
margin: 32px auto 60px;
|
||||||
|
padding: 0 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header {
|
||||||
|
background: var(--card);
|
||||||
|
border-radius: var(--radius);
|
||||||
|
box-shadow: var(--shadow);
|
||||||
|
padding: 24px;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.title {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 600;
|
||||||
|
margin: 0 0 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.meta {
|
||||||
|
color: var(--muted);
|
||||||
|
font-size: 14px;
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.controls {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(220px, 1fr));
|
||||||
|
gap: 16px;
|
||||||
|
margin-top: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control label {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--muted);
|
||||||
|
}
|
||||||
|
|
||||||
|
.control input,
|
||||||
|
.control select,
|
||||||
|
.control button {
|
||||||
|
border-radius: 12px;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
padding: 10px 12px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-family: inherit;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control button {
|
||||||
|
background: var(--accent);
|
||||||
|
color: #fff;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: transform 0.1s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control button:active {
|
||||||
|
transform: scale(0.98);
|
||||||
|
}
|
||||||
|
|
||||||
|
.stats {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--muted);
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.hidden {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-time {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--muted);
|
||||||
|
margin-bottom: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-row {
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
align-items: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.sent .message-row {
|
||||||
|
flex-direction: row-reverse;
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 12px;
|
||||||
|
background: #eef2ff;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
overflow: hidden;
|
||||||
|
flex-shrink: 0;
|
||||||
|
color: #475569;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
|
||||||
|
.bubble {
|
||||||
|
max-width: min(70%, 720px);
|
||||||
|
background: var(--received);
|
||||||
|
border-radius: 18px;
|
||||||
|
padding: 12px 14px;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
box-shadow: 0 8px 20px rgba(15, 23, 42, 0.06);
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.sent .bubble {
|
||||||
|
background: var(--sent);
|
||||||
|
border-color: transparent;
|
||||||
|
}
|
||||||
|
|
||||||
|
.sender-name {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--muted);
|
||||||
|
margin-bottom: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-content {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 8px;
|
||||||
|
font-size: 14px;
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-text {
|
||||||
|
word-break: break-word;
|
||||||
|
}
|
||||||
|
|
||||||
|
.inline-emoji {
|
||||||
|
width: 22px;
|
||||||
|
height: 22px;
|
||||||
|
vertical-align: text-bottom;
|
||||||
|
margin: 0 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media {
|
||||||
|
border-radius: 14px;
|
||||||
|
max-width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.previewable {
|
||||||
|
cursor: zoom-in;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.image,
|
||||||
|
.message-media.emoji {
|
||||||
|
max-height: 260px;
|
||||||
|
object-fit: contain;
|
||||||
|
background: #f1f5f9;
|
||||||
|
padding: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.emoji {
|
||||||
|
max-height: 160px;
|
||||||
|
width: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.video {
|
||||||
|
max-height: 360px;
|
||||||
|
background: #111827;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-media.audio {
|
||||||
|
width: 260px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0;
|
||||||
|
background: rgba(15, 23, 42, 0.7);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
opacity: 0;
|
||||||
|
pointer-events: none;
|
||||||
|
transition: opacity 0.2s ease;
|
||||||
|
z-index: 999;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview.active {
|
||||||
|
opacity: 1;
|
||||||
|
pointer-events: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview img {
|
||||||
|
max-width: min(90vw, 1200px);
|
||||||
|
max-height: 90vh;
|
||||||
|
border-radius: 18px;
|
||||||
|
box-shadow: 0 20px 40px rgba(0, 0, 0, 0.35);
|
||||||
|
background: #0f172a;
|
||||||
|
transition: transform 0.1s ease;
|
||||||
|
cursor: zoom-out;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="cloud-dancer"] {
|
||||||
|
--accent: #6b8cff;
|
||||||
|
--sent: #e0e7ff;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #d8e0f7;
|
||||||
|
--bg: #f6f7fb;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="corundum-blue"] {
|
||||||
|
--accent: #2563eb;
|
||||||
|
--sent: #dbeafe;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #c7d2fe;
|
||||||
|
--bg: #eef2ff;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="kiwi-green"] {
|
||||||
|
--accent: #16a34a;
|
||||||
|
--sent: #dcfce7;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #bbf7d0;
|
||||||
|
--bg: #f0fdf4;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="spicy-red"] {
|
||||||
|
--accent: #e11d48;
|
||||||
|
--sent: #ffe4e6;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #fecdd3;
|
||||||
|
--bg: #fff1f2;
|
||||||
|
}
|
||||||
|
|
||||||
|
body[data-theme="teal-water"] {
|
||||||
|
--accent: #0f766e;
|
||||||
|
--sent: #ccfbf1;
|
||||||
|
--received: #ffffff;
|
||||||
|
--border: #99f6e4;
|
||||||
|
--bg: #f0fdfa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.highlight {
|
||||||
|
outline: 2px solid var(--accent);
|
||||||
|
outline-offset: 4px;
|
||||||
|
border-radius: 18px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.empty {
|
||||||
|
text-align: center;
|
||||||
|
color: var(--muted);
|
||||||
|
padding: 40px;
|
||||||
|
}
|
||||||
|
`;
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -1,5 +1,9 @@
|
|||||||
|
import * as fs from 'fs'
|
||||||
|
import * as path from 'path'
|
||||||
|
import ExcelJS from 'exceljs'
|
||||||
import { ConfigService } from './config'
|
import { ConfigService } from './config'
|
||||||
import { wcdbService } from './wcdbService'
|
import { wcdbService } from './wcdbService'
|
||||||
|
import { chatService } from './chatService'
|
||||||
|
|
||||||
export interface GroupChatInfo {
|
export interface GroupChatInfo {
|
||||||
username: string
|
username: string
|
||||||
@@ -12,6 +16,10 @@ export interface GroupMember {
|
|||||||
username: string
|
username: string
|
||||||
displayName: string
|
displayName: string
|
||||||
avatarUrl?: string
|
avatarUrl?: string
|
||||||
|
nickname?: string
|
||||||
|
alias?: string
|
||||||
|
remark?: string
|
||||||
|
groupNickname?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface GroupMessageRank {
|
export interface GroupMessageRank {
|
||||||
@@ -41,16 +49,45 @@ class GroupAnalyticsService {
|
|||||||
this.configService = new ConfigService()
|
this.configService = new ConfigService()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 并发控制:限制同时执行的 Promise 数量
|
||||||
|
private async parallelLimit<T, R>(
|
||||||
|
items: T[],
|
||||||
|
limit: number,
|
||||||
|
fn: (item: T, index: number) => Promise<R>
|
||||||
|
): Promise<R[]> {
|
||||||
|
const results: R[] = new Array(items.length)
|
||||||
|
let currentIndex = 0
|
||||||
|
|
||||||
|
async function runNext(): Promise<void> {
|
||||||
|
while (currentIndex < items.length) {
|
||||||
|
const index = currentIndex++
|
||||||
|
results[index] = await fn(items[index], index)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const workers = Array(Math.min(limit, items.length))
|
||||||
|
.fill(null)
|
||||||
|
.map(() => runNext())
|
||||||
|
|
||||||
|
await Promise.all(workers)
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
private cleanAccountDirName(name: string): string {
|
private cleanAccountDirName(name: string): string {
|
||||||
const trimmed = name.trim()
|
const trimmed = name.trim()
|
||||||
if (!trimmed) return trimmed
|
if (!trimmed) return trimmed
|
||||||
if (trimmed.toLowerCase().startsWith('wxid_')) {
|
if (trimmed.toLowerCase().startsWith('wxid_')) {
|
||||||
const match = trimmed.match(/^(wxid_[^_]+)/i)
|
const match = trimmed.match(/^(wxid_[^_]+)/i)
|
||||||
if (match) return match[1]
|
if (match) return match[1]
|
||||||
}
|
|
||||||
return trimmed
|
return trimmed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
|
const cleaned = suffixMatch ? suffixMatch[1] : trimmed
|
||||||
|
|
||||||
|
return cleaned
|
||||||
|
}
|
||||||
|
|
||||||
private async ensureConnected(): Promise<{ success: boolean; error?: string }> {
|
private async ensureConnected(): Promise<{ success: boolean; error?: string }> {
|
||||||
const wxid = this.configService.get('myWxid')
|
const wxid = this.configService.get('myWxid')
|
||||||
const dbPath = this.configService.get('dbPath')
|
const dbPath = this.configService.get('dbPath')
|
||||||
@@ -65,6 +102,56 @@ class GroupAnalyticsService {
|
|||||||
return { success: true }
|
return { success: true }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 从 DLL 获取群成员的群昵称
|
||||||
|
*/
|
||||||
|
private async getGroupNicknamesForRoom(chatroomId: string): Promise<Map<string, string>> {
|
||||||
|
try {
|
||||||
|
const result = await wcdbService.getGroupNicknames(chatroomId)
|
||||||
|
if (result.success && result.nicknames) {
|
||||||
|
return new Map(Object.entries(result.nicknames))
|
||||||
|
}
|
||||||
|
return new Map<string, string>()
|
||||||
|
} catch (e) {
|
||||||
|
console.error('getGroupNicknamesForRoom error:', e)
|
||||||
|
return new Map<string, string>()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private escapeCsvValue(value: string): string {
|
||||||
|
if (value == null) return ''
|
||||||
|
const str = String(value)
|
||||||
|
if (/[",\n\r]/.test(str)) {
|
||||||
|
return `"${str.replace(/"/g, '""')}"`
|
||||||
|
}
|
||||||
|
return str
|
||||||
|
}
|
||||||
|
|
||||||
|
private normalizeGroupNickname(value: string, wxid: string, fallback: string): string {
|
||||||
|
const trimmed = (value || '').trim()
|
||||||
|
if (!trimmed) return fallback
|
||||||
|
if (/^["'@]+$/.test(trimmed)) return fallback
|
||||||
|
if (trimmed.toLowerCase() === (wxid || '').toLowerCase()) return fallback
|
||||||
|
return trimmed
|
||||||
|
}
|
||||||
|
|
||||||
|
private sanitizeWorksheetName(name: string): string {
|
||||||
|
const cleaned = (name || '').replace(/[*?:\\/\\[\\]]/g, '_').trim()
|
||||||
|
const limited = cleaned.slice(0, 31)
|
||||||
|
return limited || 'Sheet1'
|
||||||
|
}
|
||||||
|
|
||||||
|
private formatDateTime(date: Date): string {
|
||||||
|
const pad = (value: number) => String(value).padStart(2, '0')
|
||||||
|
const year = date.getFullYear()
|
||||||
|
const month = pad(date.getMonth() + 1)
|
||||||
|
const day = pad(date.getDate())
|
||||||
|
const hour = pad(date.getHours())
|
||||||
|
const minute = pad(date.getMinutes())
|
||||||
|
const second = pad(date.getSeconds())
|
||||||
|
return `${year}-${month}-${day} ${hour}:${minute}:${second}`
|
||||||
|
}
|
||||||
|
|
||||||
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()
|
||||||
@@ -80,23 +167,38 @@ class GroupAnalyticsService {
|
|||||||
.map((row) => row.username || row.user_name || row.userName || '')
|
.map((row) => row.username || row.user_name || row.userName || '')
|
||||||
.filter((username) => username.includes('@chatroom'))
|
.filter((username) => username.includes('@chatroom'))
|
||||||
|
|
||||||
const [displayNames, avatarUrls, memberCounts] = await Promise.all([
|
const [memberCounts, contactInfo] = await Promise.all([
|
||||||
wcdbService.getDisplayNames(groupIds),
|
wcdbService.getGroupMemberCounts(groupIds),
|
||||||
wcdbService.getAvatarUrls(groupIds),
|
chatService.enrichSessionsContactInfo(groupIds)
|
||||||
wcdbService.getGroupMemberCounts(groupIds)
|
|
||||||
])
|
])
|
||||||
|
|
||||||
|
let fallbackNames: { success: boolean; map?: Record<string, string> } | null = null
|
||||||
|
let fallbackAvatars: { success: boolean; map?: Record<string, string> } | null = null
|
||||||
|
if (!contactInfo.success || !contactInfo.contacts) {
|
||||||
|
const [displayNames, avatarUrls] = await Promise.all([
|
||||||
|
wcdbService.getDisplayNames(groupIds),
|
||||||
|
wcdbService.getAvatarUrls(groupIds)
|
||||||
|
])
|
||||||
|
fallbackNames = displayNames
|
||||||
|
fallbackAvatars = avatarUrls
|
||||||
|
}
|
||||||
|
|
||||||
const groups: GroupChatInfo[] = []
|
const groups: GroupChatInfo[] = []
|
||||||
for (const groupId of groupIds) {
|
for (const groupId of groupIds) {
|
||||||
|
const contact = contactInfo.success && contactInfo.contacts ? contactInfo.contacts[groupId] : undefined
|
||||||
|
const displayName = contact?.displayName ||
|
||||||
|
(fallbackNames && fallbackNames.success && fallbackNames.map ? (fallbackNames.map[groupId] || '') : '') ||
|
||||||
|
groupId
|
||||||
|
const avatarUrl = contact?.avatarUrl ||
|
||||||
|
(fallbackAvatars && fallbackAvatars.success && fallbackAvatars.map ? fallbackAvatars.map[groupId] : undefined)
|
||||||
|
|
||||||
groups.push({
|
groups.push({
|
||||||
username: groupId,
|
username: groupId,
|
||||||
displayName: displayNames.success && displayNames.map
|
displayName,
|
||||||
? (displayNames.map[groupId] || groupId)
|
|
||||||
: groupId,
|
|
||||||
memberCount: memberCounts.success && memberCounts.map && typeof memberCounts.map[groupId] === 'number'
|
memberCount: memberCounts.success && memberCounts.map && typeof memberCounts.map[groupId] === 'number'
|
||||||
? memberCounts.map[groupId]
|
? memberCounts.map[groupId]
|
||||||
: 0,
|
: 0,
|
||||||
avatarUrl: avatarUrls.success && avatarUrls.map ? avatarUrls.map[groupId] : undefined
|
avatarUrl
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -118,14 +220,55 @@ class GroupAnalyticsService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const members = result.members as { username: string; avatarUrl?: string }[]
|
const members = result.members as { username: string; avatarUrl?: string }[]
|
||||||
const usernames = members.map((m) => m.username)
|
const usernames = members.map((m) => m.username).filter(Boolean)
|
||||||
const displayNames = await wcdbService.getDisplayNames(usernames)
|
|
||||||
|
|
||||||
const data: GroupMember[] = members.map((m) => ({
|
const [displayNames, groupNicknames] = await Promise.all([
|
||||||
username: m.username,
|
wcdbService.getDisplayNames(usernames),
|
||||||
displayName: displayNames.success && displayNames.map ? (displayNames.map[m.username] || m.username) : m.username,
|
this.getGroupNicknamesForRoom(chatroomId)
|
||||||
|
])
|
||||||
|
|
||||||
|
const contactMap = new Map<string, { remark?: string; nickName?: string; alias?: string }>()
|
||||||
|
const concurrency = 6
|
||||||
|
await this.parallelLimit(usernames, concurrency, async (username) => {
|
||||||
|
const contactResult = await wcdbService.getContact(username)
|
||||||
|
if (contactResult.success && contactResult.contact) {
|
||||||
|
const contact = contactResult.contact as any
|
||||||
|
contactMap.set(username, {
|
||||||
|
remark: contact.remark || '',
|
||||||
|
nickName: contact.nickName || contact.nick_name || '',
|
||||||
|
alias: contact.alias || ''
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
contactMap.set(username, { remark: '', nickName: '', alias: '' })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
const myWxid = this.cleanAccountDirName(this.configService.get('myWxid') || '')
|
||||||
|
const data: GroupMember[] = members.map((m) => {
|
||||||
|
const wxid = m.username || ''
|
||||||
|
const displayName = displayNames.success && displayNames.map ? (displayNames.map[wxid] || wxid) : wxid
|
||||||
|
const contact = contactMap.get(wxid)
|
||||||
|
const nickname = contact?.nickName || ''
|
||||||
|
const remark = contact?.remark || ''
|
||||||
|
const alias = contact?.alias || ''
|
||||||
|
const rawGroupNickname = groupNicknames.get(wxid.toLowerCase()) || ''
|
||||||
|
const normalizedWxid = this.cleanAccountDirName(wxid)
|
||||||
|
const groupNickname = this.normalizeGroupNickname(
|
||||||
|
rawGroupNickname,
|
||||||
|
normalizedWxid === myWxid ? myWxid : wxid,
|
||||||
|
''
|
||||||
|
)
|
||||||
|
|
||||||
|
return {
|
||||||
|
username: wxid,
|
||||||
|
displayName,
|
||||||
|
nickname,
|
||||||
|
alias,
|
||||||
|
remark,
|
||||||
|
groupNickname,
|
||||||
avatarUrl: m.avatarUrl
|
avatarUrl: m.avatarUrl
|
||||||
}))
|
}
|
||||||
|
})
|
||||||
|
|
||||||
return { success: true, data }
|
return { success: true, data }
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
@@ -181,6 +324,8 @@ class GroupAnalyticsService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async getGroupActiveHours(chatroomId: string, startTime?: number, endTime?: number): Promise<{ success: boolean; data?: GroupActiveHours; error?: string }> {
|
async getGroupActiveHours(chatroomId: string, startTime?: number, endTime?: number): Promise<{ success: boolean; data?: GroupActiveHours; error?: string }> {
|
||||||
try {
|
try {
|
||||||
const conn = await this.ensureConnected()
|
const conn = await this.ensureConnected()
|
||||||
@@ -246,6 +391,187 @@ class GroupAnalyticsService {
|
|||||||
return { success: false, error: String(e) }
|
return { success: false, error: String(e) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async exportGroupMembers(chatroomId: string, outputPath: string): Promise<{ success: boolean; count?: number; error?: string }> {
|
||||||
|
try {
|
||||||
|
const conn = await this.ensureConnected()
|
||||||
|
if (!conn.success) return { success: false, error: conn.error }
|
||||||
|
|
||||||
|
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([chatroomId])
|
||||||
|
const groupName = groupDisplay.success && groupDisplay.map
|
||||||
|
? (groupDisplay.map[chatroomId] || chatroomId)
|
||||||
|
: chatroomId
|
||||||
|
|
||||||
|
const groupContact = await wcdbService.getContact(chatroomId)
|
||||||
|
const sessionRemark = (groupContact.success && groupContact.contact)
|
||||||
|
? (groupContact.contact.remark || '')
|
||||||
|
: ''
|
||||||
|
|
||||||
|
const membersResult = await wcdbService.getGroupMembers(chatroomId)
|
||||||
|
if (!membersResult.success || !membersResult.members) {
|
||||||
|
return { success: false, error: membersResult.error || '获取群成员失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
const members = membersResult.members as { username: string; avatarUrl?: string }[]
|
||||||
|
if (members.length === 0) {
|
||||||
|
return { success: false, error: '群成员为空' }
|
||||||
|
}
|
||||||
|
|
||||||
|
const usernames = members.map((m) => m.username).filter(Boolean)
|
||||||
|
const [displayNames, groupNicknames] = await Promise.all([
|
||||||
|
wcdbService.getDisplayNames(usernames),
|
||||||
|
this.getGroupNicknamesForRoom(chatroomId)
|
||||||
|
])
|
||||||
|
|
||||||
|
const contactMap = new Map<string, { remark?: string; nickName?: string; alias?: string }>()
|
||||||
|
const concurrency = 6
|
||||||
|
await this.parallelLimit(usernames, concurrency, async (username) => {
|
||||||
|
const result = await wcdbService.getContact(username)
|
||||||
|
if (result.success && result.contact) {
|
||||||
|
const contact = result.contact as any
|
||||||
|
contactMap.set(username, {
|
||||||
|
remark: contact.remark || '',
|
||||||
|
nickName: contact.nickName || contact.nick_name || '',
|
||||||
|
alias: contact.alias || ''
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
contactMap.set(username, { remark: '', nickName: '', alias: '' })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
const infoTitleRow = ['会话信息']
|
||||||
|
const infoRow = ['微信ID', chatroomId, '', '昵称', groupName, '备注', sessionRemark || '', '']
|
||||||
|
const metaRow = ['导出工具', exportGenerator, '导出版本', exportVersion, '平台', exportPlatform, '导出时间', exportTime]
|
||||||
|
|
||||||
|
const header = ['微信昵称', '微信备注', '群昵称', 'wxid', '微信号']
|
||||||
|
const rows: string[][] = [infoTitleRow, infoRow, metaRow, header]
|
||||||
|
const myWxid = this.cleanAccountDirName(this.configService.get('myWxid') || '')
|
||||||
|
|
||||||
|
for (const member of members) {
|
||||||
|
const wxid = member.username
|
||||||
|
const normalizedWxid = this.cleanAccountDirName(wxid || '')
|
||||||
|
const contact = contactMap.get(wxid)
|
||||||
|
const fallbackName = displayNames.success && displayNames.map ? (displayNames.map[wxid] || '') : ''
|
||||||
|
const nickName = contact?.nickName || fallbackName || ''
|
||||||
|
const remark = contact?.remark || ''
|
||||||
|
const rawGroupNickname = groupNicknames.get(wxid.toLowerCase()) || ''
|
||||||
|
const alias = contact?.alias || ''
|
||||||
|
const groupNickname = this.normalizeGroupNickname(
|
||||||
|
rawGroupNickname,
|
||||||
|
normalizedWxid === myWxid ? myWxid : wxid,
|
||||||
|
''
|
||||||
|
)
|
||||||
|
|
||||||
|
rows.push([nickName, remark, groupNickname, wxid, alias])
|
||||||
|
}
|
||||||
|
|
||||||
|
const ext = path.extname(outputPath).toLowerCase()
|
||||||
|
if (ext === '.csv') {
|
||||||
|
const csvLines = rows.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 sheet = workbook.addWorksheet(this.sanitizeWorksheetName('群成员列表'))
|
||||||
|
|
||||||
|
let currentRow = 1
|
||||||
|
const titleCell = sheet.getCell(currentRow, 1)
|
||||||
|
titleCell.value = '会话信息'
|
||||||
|
titleCell.font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
titleCell.alignment = { vertical: 'middle', horizontal: 'left' }
|
||||||
|
sheet.getRow(currentRow).height = 25
|
||||||
|
currentRow++
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 1).value = '微信ID'
|
||||||
|
sheet.getCell(currentRow, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.mergeCells(currentRow, 2, currentRow, 3)
|
||||||
|
sheet.getCell(currentRow, 2).value = chatroomId
|
||||||
|
sheet.getCell(currentRow, 2).font = { name: 'Calibri', size: 11 }
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 4).value = '昵称'
|
||||||
|
sheet.getCell(currentRow, 4).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.getCell(currentRow, 5).value = groupName
|
||||||
|
sheet.getCell(currentRow, 5).font = { name: 'Calibri', size: 11 }
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 6).value = '备注'
|
||||||
|
sheet.getCell(currentRow, 6).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.mergeCells(currentRow, 7, currentRow, 8)
|
||||||
|
sheet.getCell(currentRow, 7).value = sessionRemark
|
||||||
|
sheet.getCell(currentRow, 7).font = { name: 'Calibri', size: 11 }
|
||||||
|
|
||||||
|
sheet.getRow(currentRow).height = 20
|
||||||
|
currentRow++
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 1).value = '导出工具'
|
||||||
|
sheet.getCell(currentRow, 1).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.getCell(currentRow, 2).value = exportGenerator
|
||||||
|
sheet.getCell(currentRow, 2).font = { name: 'Calibri', size: 10 }
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 3).value = '导出版本'
|
||||||
|
sheet.getCell(currentRow, 3).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.getCell(currentRow, 4).value = exportVersion
|
||||||
|
sheet.getCell(currentRow, 4).font = { name: 'Calibri', size: 10 }
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 5).value = '平台'
|
||||||
|
sheet.getCell(currentRow, 5).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.getCell(currentRow, 6).value = exportPlatform
|
||||||
|
sheet.getCell(currentRow, 6).font = { name: 'Calibri', size: 10 }
|
||||||
|
|
||||||
|
sheet.getCell(currentRow, 7).value = '导出时间'
|
||||||
|
sheet.getCell(currentRow, 7).font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
sheet.getCell(currentRow, 8).value = exportTime
|
||||||
|
sheet.getCell(currentRow, 8).font = { name: 'Calibri', size: 10 }
|
||||||
|
|
||||||
|
sheet.getRow(currentRow).height = 20
|
||||||
|
currentRow++
|
||||||
|
|
||||||
|
const headerRow = sheet.getRow(currentRow)
|
||||||
|
headerRow.height = 22
|
||||||
|
header.forEach((text, index) => {
|
||||||
|
const cell = headerRow.getCell(index + 1)
|
||||||
|
cell.value = text
|
||||||
|
cell.font = { name: 'Calibri', bold: true, size: 11 }
|
||||||
|
})
|
||||||
|
currentRow++
|
||||||
|
|
||||||
|
sheet.getColumn(1).width = 28
|
||||||
|
sheet.getColumn(2).width = 28
|
||||||
|
sheet.getColumn(3).width = 28
|
||||||
|
sheet.getColumn(4).width = 36
|
||||||
|
sheet.getColumn(5).width = 28
|
||||||
|
sheet.getColumn(6).width = 18
|
||||||
|
sheet.getColumn(7).width = 24
|
||||||
|
sheet.getColumn(8).width = 22
|
||||||
|
|
||||||
|
for (let i = 4; i < rows.length; i++) {
|
||||||
|
const [nickName, remark, groupNickname, wxid, alias] = rows[i]
|
||||||
|
const row = sheet.getRow(currentRow)
|
||||||
|
row.getCell(1).value = nickName
|
||||||
|
row.getCell(2).value = remark
|
||||||
|
row.getCell(3).value = groupNickname
|
||||||
|
row.getCell(4).value = wxid
|
||||||
|
row.getCell(5).value = alias
|
||||||
|
row.alignment = { vertical: 'top', wrapText: true }
|
||||||
|
currentRow++
|
||||||
|
}
|
||||||
|
|
||||||
|
await workbook.xlsx.writeFile(outputPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success: true, count: members.length }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export const groupAnalyticsService = new GroupAnalyticsService()
|
export const groupAnalyticsService = new GroupAnalyticsService()
|
||||||
|
|||||||
745
electron/services/httpService.ts
Normal file
745
electron/services/httpService.ts
Normal file
@@ -0,0 +1,745 @@
|
|||||||
|
/**
|
||||||
|
* HTTP API 服务
|
||||||
|
* 提供 ChatLab 标准化格式的消息查询 API
|
||||||
|
*/
|
||||||
|
import * as http from 'http'
|
||||||
|
import { URL } from 'url'
|
||||||
|
import { chatService, Message } from './chatService'
|
||||||
|
import { wcdbService } from './wcdbService'
|
||||||
|
import { ConfigService } from './config'
|
||||||
|
|
||||||
|
// ChatLab 格式定义
|
||||||
|
interface ChatLabHeader {
|
||||||
|
version: string
|
||||||
|
exportedAt: number
|
||||||
|
generator: string
|
||||||
|
description?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ChatLabMeta {
|
||||||
|
name: string
|
||||||
|
platform: string
|
||||||
|
type: 'group' | 'private'
|
||||||
|
groupId?: string
|
||||||
|
groupAvatar?: string
|
||||||
|
ownerId?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ChatLabMember {
|
||||||
|
platformId: string
|
||||||
|
accountName: string
|
||||||
|
groupNickname?: string
|
||||||
|
aliases?: string[]
|
||||||
|
avatar?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ChatLabMessage {
|
||||||
|
sender: string
|
||||||
|
accountName: string
|
||||||
|
groupNickname?: string
|
||||||
|
timestamp: number
|
||||||
|
type: number
|
||||||
|
content: string | null
|
||||||
|
platformMessageId?: string
|
||||||
|
replyToMessageId?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ChatLabData {
|
||||||
|
chatlab: ChatLabHeader
|
||||||
|
meta: ChatLabMeta
|
||||||
|
members: ChatLabMember[]
|
||||||
|
messages: ChatLabMessage[]
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChatLab 消息类型映射
|
||||||
|
const ChatLabType = {
|
||||||
|
TEXT: 0,
|
||||||
|
IMAGE: 1,
|
||||||
|
VOICE: 2,
|
||||||
|
VIDEO: 3,
|
||||||
|
FILE: 4,
|
||||||
|
EMOJI: 5,
|
||||||
|
LINK: 7,
|
||||||
|
LOCATION: 8,
|
||||||
|
RED_PACKET: 20,
|
||||||
|
TRANSFER: 21,
|
||||||
|
POKE: 22,
|
||||||
|
CALL: 23,
|
||||||
|
SHARE: 24,
|
||||||
|
REPLY: 25,
|
||||||
|
FORWARD: 26,
|
||||||
|
CONTACT: 27,
|
||||||
|
SYSTEM: 80,
|
||||||
|
RECALL: 81,
|
||||||
|
OTHER: 99
|
||||||
|
} as const
|
||||||
|
|
||||||
|
class HttpService {
|
||||||
|
private server: http.Server | null = null
|
||||||
|
private configService: ConfigService
|
||||||
|
private port: number = 5031
|
||||||
|
private running: boolean = false
|
||||||
|
private connections: Set<import('net').Socket> = new Set()
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
this.configService = ConfigService.getInstance()
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 启动 HTTP 服务
|
||||||
|
*/
|
||||||
|
async start(port: number = 5031): Promise<{ success: boolean; port?: number; error?: string }> {
|
||||||
|
if (this.running && this.server) {
|
||||||
|
return { success: true, port: this.port }
|
||||||
|
}
|
||||||
|
|
||||||
|
this.port = port
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
this.server = http.createServer((req, res) => this.handleRequest(req, res))
|
||||||
|
|
||||||
|
// 跟踪所有连接,以便关闭时能强制断开
|
||||||
|
this.server.on('connection', (socket) => {
|
||||||
|
this.connections.add(socket)
|
||||||
|
socket.on('close', () => {
|
||||||
|
this.connections.delete(socket)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
this.server.on('error', (err: NodeJS.ErrnoException) => {
|
||||||
|
if (err.code === 'EADDRINUSE') {
|
||||||
|
console.error(`[HttpService] Port ${this.port} is already in use`)
|
||||||
|
resolve({ success: false, error: `Port ${this.port} is already in use` })
|
||||||
|
} else {
|
||||||
|
console.error('[HttpService] Server error:', err)
|
||||||
|
resolve({ success: false, error: err.message })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
this.server.listen(this.port, '127.0.0.1', () => {
|
||||||
|
this.running = true
|
||||||
|
console.log(`[HttpService] HTTP API server started on http://127.0.0.1:${this.port}`)
|
||||||
|
resolve({ success: true, port: this.port })
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 停止 HTTP 服务
|
||||||
|
*/
|
||||||
|
async stop(): Promise<void> {
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
if (this.server) {
|
||||||
|
// 强制关闭所有活动连接
|
||||||
|
for (const socket of this.connections) {
|
||||||
|
socket.destroy()
|
||||||
|
}
|
||||||
|
this.connections.clear()
|
||||||
|
|
||||||
|
this.server.close(() => {
|
||||||
|
this.running = false
|
||||||
|
this.server = null
|
||||||
|
console.log('[HttpService] HTTP API server stopped')
|
||||||
|
resolve()
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
this.running = false
|
||||||
|
resolve()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查服务是否运行
|
||||||
|
*/
|
||||||
|
isRunning(): boolean {
|
||||||
|
return this.running
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取当前端口
|
||||||
|
*/
|
||||||
|
getPort(): number {
|
||||||
|
return this.port
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理 HTTP 请求
|
||||||
|
*/
|
||||||
|
private async handleRequest(req: http.IncomingMessage, res: http.ServerResponse): Promise<void> {
|
||||||
|
// 设置 CORS 头
|
||||||
|
res.setHeader('Access-Control-Allow-Origin', '*')
|
||||||
|
res.setHeader('Access-Control-Allow-Methods', 'GET, OPTIONS')
|
||||||
|
res.setHeader('Access-Control-Allow-Headers', 'Content-Type')
|
||||||
|
|
||||||
|
if (req.method === 'OPTIONS') {
|
||||||
|
res.writeHead(204)
|
||||||
|
res.end()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const url = new URL(req.url || '/', `http://127.0.0.1:${this.port}`)
|
||||||
|
const pathname = url.pathname
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 路由处理
|
||||||
|
if (pathname === '/health' || pathname === '/api/v1/health') {
|
||||||
|
this.sendJson(res, { status: 'ok' })
|
||||||
|
} else if (pathname === '/api/v1/messages') {
|
||||||
|
await this.handleMessages(url, res)
|
||||||
|
} else if (pathname === '/api/v1/sessions') {
|
||||||
|
await this.handleSessions(url, res)
|
||||||
|
} else if (pathname === '/api/v1/contacts') {
|
||||||
|
await this.handleContacts(url, res)
|
||||||
|
} else {
|
||||||
|
this.sendError(res, 404, 'Not Found')
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[HttpService] Request error:', error)
|
||||||
|
this.sendError(res, 500, String(error))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 批量获取消息(循环游标直到满足 limit)
|
||||||
|
* 绕过 chatService 的单 batch 限制,直接操作 wcdbService 游标
|
||||||
|
*/
|
||||||
|
private async fetchMessagesBatch(
|
||||||
|
talker: string,
|
||||||
|
offset: number,
|
||||||
|
limit: number,
|
||||||
|
startTime: number,
|
||||||
|
endTime: number,
|
||||||
|
ascending: boolean
|
||||||
|
): Promise<{ success: boolean; messages?: Message[]; hasMore?: boolean; error?: string }> {
|
||||||
|
try {
|
||||||
|
// 使用固定 batch 大小(与 limit 相同或最大 500)来减少循环次数
|
||||||
|
const batchSize = Math.min(limit, 500)
|
||||||
|
const beginTimestamp = startTime > 10000000000 ? Math.floor(startTime / 1000) : startTime
|
||||||
|
const endTimestamp = endTime > 10000000000 ? Math.floor(endTime / 1000) : endTime
|
||||||
|
|
||||||
|
const cursorResult = await wcdbService.openMessageCursor(talker, batchSize, ascending, beginTimestamp, endTimestamp)
|
||||||
|
if (!cursorResult.success || !cursorResult.cursor) {
|
||||||
|
return { success: false, error: cursorResult.error || '打开消息游标失败' }
|
||||||
|
}
|
||||||
|
|
||||||
|
const cursor = cursorResult.cursor
|
||||||
|
try {
|
||||||
|
const allRows: Record<string, any>[] = []
|
||||||
|
let hasMore = true
|
||||||
|
let skipped = 0
|
||||||
|
|
||||||
|
// 循环获取消息,处理 offset 跳过 + limit 累积
|
||||||
|
while (allRows.length < limit && hasMore) {
|
||||||
|
const batch = await wcdbService.fetchMessageBatch(cursor)
|
||||||
|
if (!batch.success || !batch.rows || batch.rows.length === 0) {
|
||||||
|
hasMore = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
let rows = batch.rows
|
||||||
|
hasMore = batch.hasMore === true
|
||||||
|
|
||||||
|
// 处理 offset: 跳过前 N 条
|
||||||
|
if (skipped < offset) {
|
||||||
|
const remaining = offset - skipped
|
||||||
|
if (remaining >= rows.length) {
|
||||||
|
skipped += rows.length
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
rows = rows.slice(remaining)
|
||||||
|
skipped = offset
|
||||||
|
}
|
||||||
|
|
||||||
|
allRows.push(...rows)
|
||||||
|
}
|
||||||
|
|
||||||
|
const trimmedRows = allRows.slice(0, limit)
|
||||||
|
const finalHasMore = hasMore || allRows.length > limit
|
||||||
|
const messages = this.mapRowsToMessagesSimple(trimmedRows)
|
||||||
|
return { success: true, messages, hasMore: finalHasMore }
|
||||||
|
} finally {
|
||||||
|
await wcdbService.closeMessageCursor(cursor)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[HttpService] fetchMessagesBatch error:', e)
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 简单的行数据到 Message 映射(用于 API 输出)
|
||||||
|
*/
|
||||||
|
private mapRowsToMessagesSimple(rows: Record<string, any>[]): Message[] {
|
||||||
|
const myWxid = this.configService.get('myWxid') || ''
|
||||||
|
const messages: Message[] = []
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析消息内容中的特殊字段
|
||||||
|
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) {
|
||||||
|
if (row[key] !== undefined && row[key] !== null) {
|
||||||
|
return String(row[key])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理消息查询
|
||||||
|
* GET /api/v1/messages?talker=xxx&limit=100&start=20260101&chatlab=1
|
||||||
|
*/
|
||||||
|
private async handleMessages(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
|
const talker = url.searchParams.get('talker')
|
||||||
|
const limit = Math.min(parseInt(url.searchParams.get('limit') || '100', 10), 10000)
|
||||||
|
const offset = parseInt(url.searchParams.get('offset') || '0', 10)
|
||||||
|
const startParam = url.searchParams.get('start')
|
||||||
|
const endParam = url.searchParams.get('end')
|
||||||
|
const chatlab = url.searchParams.get('chatlab') === '1'
|
||||||
|
const formatParam = url.searchParams.get('format')
|
||||||
|
const format = formatParam || (chatlab ? 'chatlab' : 'json')
|
||||||
|
|
||||||
|
if (!talker) {
|
||||||
|
this.sendError(res, 400, 'Missing required parameter: talker')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析时间参数 (支持 YYYYMMDD 格式)
|
||||||
|
const startTime = this.parseTimeParam(startParam)
|
||||||
|
const endTime = this.parseTimeParam(endParam, true)
|
||||||
|
|
||||||
|
// 使用批量获取方法,绕过 chatService 的单 batch 限制
|
||||||
|
const result = await this.fetchMessagesBatch(talker, offset, limit, startTime, endTime, true)
|
||||||
|
if (!result.success || !result.messages) {
|
||||||
|
this.sendError(res, 500, result.error || 'Failed to get messages')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if (format === 'chatlab') {
|
||||||
|
// 获取会话显示名
|
||||||
|
const displayNames = await this.getDisplayNames([talker])
|
||||||
|
const talkerName = displayNames[talker] || talker
|
||||||
|
|
||||||
|
const chatLabData = await this.convertToChatLab(result.messages, talker, talkerName)
|
||||||
|
this.sendJson(res, chatLabData)
|
||||||
|
} else {
|
||||||
|
// 返回原始消息格式
|
||||||
|
this.sendJson(res, {
|
||||||
|
success: true,
|
||||||
|
talker,
|
||||||
|
count: result.messages.length,
|
||||||
|
hasMore: result.hasMore,
|
||||||
|
messages: result.messages
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理会话列表查询
|
||||||
|
* GET /api/v1/sessions?keyword=xxx&limit=100
|
||||||
|
*/
|
||||||
|
private async handleSessions(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
|
const keyword = url.searchParams.get('keyword') || ''
|
||||||
|
const limit = parseInt(url.searchParams.get('limit') || '100', 10)
|
||||||
|
|
||||||
|
try {
|
||||||
|
const sessions = await chatService.getSessions()
|
||||||
|
if (!sessions.success || !sessions.sessions) {
|
||||||
|
this.sendError(res, 500, sessions.error || 'Failed to get sessions')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let filteredSessions = sessions.sessions
|
||||||
|
if (keyword) {
|
||||||
|
const lowerKeyword = keyword.toLowerCase()
|
||||||
|
filteredSessions = sessions.sessions.filter(s =>
|
||||||
|
s.username.toLowerCase().includes(lowerKeyword) ||
|
||||||
|
(s.displayName && s.displayName.toLowerCase().includes(lowerKeyword))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 应用 limit
|
||||||
|
const limitedSessions = filteredSessions.slice(0, limit)
|
||||||
|
|
||||||
|
this.sendJson(res, {
|
||||||
|
success: true,
|
||||||
|
count: limitedSessions.length,
|
||||||
|
sessions: limitedSessions.map(s => ({
|
||||||
|
username: s.username,
|
||||||
|
displayName: s.displayName,
|
||||||
|
type: s.type,
|
||||||
|
lastTimestamp: s.lastTimestamp,
|
||||||
|
unreadCount: s.unreadCount
|
||||||
|
}))
|
||||||
|
})
|
||||||
|
} catch (error) {
|
||||||
|
this.sendError(res, 500, String(error))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理联系人查询
|
||||||
|
* GET /api/v1/contacts?keyword=xxx&limit=100
|
||||||
|
*/
|
||||||
|
private async handleContacts(url: URL, res: http.ServerResponse): Promise<void> {
|
||||||
|
const keyword = url.searchParams.get('keyword') || ''
|
||||||
|
const limit = parseInt(url.searchParams.get('limit') || '100', 10)
|
||||||
|
|
||||||
|
try {
|
||||||
|
const contacts = await chatService.getContacts()
|
||||||
|
if (!contacts.success || !contacts.contacts) {
|
||||||
|
this.sendError(res, 500, contacts.error || 'Failed to get contacts')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let filteredContacts = contacts.contacts
|
||||||
|
if (keyword) {
|
||||||
|
const lowerKeyword = keyword.toLowerCase()
|
||||||
|
filteredContacts = contacts.contacts.filter(c =>
|
||||||
|
c.username.toLowerCase().includes(lowerKeyword) ||
|
||||||
|
(c.nickname && c.nickname.toLowerCase().includes(lowerKeyword)) ||
|
||||||
|
(c.remark && c.remark.toLowerCase().includes(lowerKeyword)) ||
|
||||||
|
(c.displayName && c.displayName.toLowerCase().includes(lowerKeyword))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
const limited = filteredContacts.slice(0, limit)
|
||||||
|
|
||||||
|
this.sendJson(res, {
|
||||||
|
success: true,
|
||||||
|
count: limited.length,
|
||||||
|
contacts: limited
|
||||||
|
})
|
||||||
|
} catch (error) {
|
||||||
|
this.sendError(res, 500, String(error))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析时间参数
|
||||||
|
* 支持 YYYYMMDD 格式,返回秒级时间戳
|
||||||
|
*/
|
||||||
|
private parseTimeParam(param: string | null, isEnd: boolean = false): number {
|
||||||
|
if (!param) return 0
|
||||||
|
|
||||||
|
// 纯数字且长度为8,视为 YYYYMMDD
|
||||||
|
if (/^\d{8}$/.test(param)) {
|
||||||
|
const year = parseInt(param.slice(0, 4), 10)
|
||||||
|
const month = parseInt(param.slice(4, 6), 10) - 1
|
||||||
|
const day = parseInt(param.slice(6, 8), 10)
|
||||||
|
const date = new Date(year, month, day)
|
||||||
|
if (isEnd) {
|
||||||
|
// 结束时间设为当天 23:59:59
|
||||||
|
date.setHours(23, 59, 59, 999)
|
||||||
|
}
|
||||||
|
return Math.floor(date.getTime() / 1000)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 纯数字,视为时间戳
|
||||||
|
if (/^\d+$/.test(param)) {
|
||||||
|
const ts = parseInt(param, 10)
|
||||||
|
// 如果是毫秒级时间戳,转为秒级
|
||||||
|
return ts > 10000000000 ? Math.floor(ts / 1000) : ts
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取显示名称
|
||||||
|
*/
|
||||||
|
private async getDisplayNames(usernames: string[]): Promise<Record<string, string>> {
|
||||||
|
try {
|
||||||
|
const result = await wcdbService.getDisplayNames(usernames)
|
||||||
|
if (result.success && result.map) {
|
||||||
|
return result.map
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[HttpService] Failed to get display names:', e)
|
||||||
|
}
|
||||||
|
// 返回空对象,调用方会使用 username 作为备用
|
||||||
|
return {}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 转换为 ChatLab 格式
|
||||||
|
*/
|
||||||
|
private async convertToChatLab(messages: Message[], talkerId: string, talkerName: string): Promise<ChatLabData> {
|
||||||
|
const isGroup = talkerId.endsWith('@chatroom')
|
||||||
|
const myWxid = this.configService.get('myWxid') || ''
|
||||||
|
|
||||||
|
// 收集所有发送者
|
||||||
|
const senderSet = new Set<string>()
|
||||||
|
for (const msg of messages) {
|
||||||
|
if (msg.senderUsername) {
|
||||||
|
senderSet.add(msg.senderUsername)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取发送者显示名
|
||||||
|
const senderNames = await this.getDisplayNames(Array.from(senderSet))
|
||||||
|
|
||||||
|
// 获取群昵称(如果是群聊)
|
||||||
|
let groupNicknamesMap = new Map<string, string>()
|
||||||
|
if (isGroup) {
|
||||||
|
try {
|
||||||
|
const result = await wcdbService.getGroupNicknames(talkerId)
|
||||||
|
if (result.success && result.nicknames) {
|
||||||
|
groupNicknamesMap = new Map(Object.entries(result.nicknames))
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[HttpService] Failed to get group nicknames:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 构建成员列表
|
||||||
|
const memberMap = new Map<string, ChatLabMember>()
|
||||||
|
for (const msg of messages) {
|
||||||
|
const sender = msg.senderUsername || ''
|
||||||
|
if (sender && !memberMap.has(sender)) {
|
||||||
|
const displayName = senderNames[sender] || sender
|
||||||
|
const isSelf = sender === myWxid || sender.toLowerCase() === myWxid.toLowerCase()
|
||||||
|
// 获取群昵称(尝试多种方式)
|
||||||
|
const groupNickname = isGroup
|
||||||
|
? (groupNicknamesMap.get(sender) || groupNicknamesMap.get(sender.toLowerCase()) || '')
|
||||||
|
: ''
|
||||||
|
memberMap.set(sender, {
|
||||||
|
platformId: sender,
|
||||||
|
accountName: isSelf ? '我' : displayName,
|
||||||
|
groupNickname: groupNickname || undefined
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 转换消息
|
||||||
|
const chatLabMessages: ChatLabMessage[] = messages.map(msg => {
|
||||||
|
const sender = msg.senderUsername || ''
|
||||||
|
const isSelf = msg.isSend === 1 || sender === myWxid
|
||||||
|
const accountName = isSelf ? '我' : (senderNames[sender] || sender)
|
||||||
|
// 获取该发送者的群昵称
|
||||||
|
const groupNickname = isGroup
|
||||||
|
? (groupNicknamesMap.get(sender) || groupNicknamesMap.get(sender.toLowerCase()) || '')
|
||||||
|
: ''
|
||||||
|
|
||||||
|
return {
|
||||||
|
sender,
|
||||||
|
accountName,
|
||||||
|
groupNickname: groupNickname || undefined,
|
||||||
|
timestamp: msg.createTime,
|
||||||
|
type: this.mapMessageType(msg.localType, msg),
|
||||||
|
content: this.getMessageContent(msg),
|
||||||
|
platformMessageId: msg.serverId ? String(msg.serverId) : undefined
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
chatlab: {
|
||||||
|
version: '0.0.2',
|
||||||
|
exportedAt: Math.floor(Date.now() / 1000),
|
||||||
|
generator: 'WeFlow'
|
||||||
|
},
|
||||||
|
meta: {
|
||||||
|
name: talkerName,
|
||||||
|
platform: 'wechat',
|
||||||
|
type: isGroup ? 'group' : 'private',
|
||||||
|
groupId: isGroup ? talkerId : undefined,
|
||||||
|
ownerId: myWxid || undefined
|
||||||
|
},
|
||||||
|
members: Array.from(memberMap.values()),
|
||||||
|
messages: chatLabMessages
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 映射 WeChat 消息类型到 ChatLab 类型
|
||||||
|
*/
|
||||||
|
private mapMessageType(localType: number, msg: Message): number {
|
||||||
|
switch (localType) {
|
||||||
|
case 1: // 文本
|
||||||
|
return ChatLabType.TEXT
|
||||||
|
case 3: // 图片
|
||||||
|
return ChatLabType.IMAGE
|
||||||
|
case 34: // 语音
|
||||||
|
return ChatLabType.VOICE
|
||||||
|
case 43: // 视频
|
||||||
|
return ChatLabType.VIDEO
|
||||||
|
case 47: // 动画表情
|
||||||
|
return ChatLabType.EMOJI
|
||||||
|
case 48: // 位置
|
||||||
|
return ChatLabType.LOCATION
|
||||||
|
case 42: // 名片
|
||||||
|
return ChatLabType.CONTACT
|
||||||
|
case 50: // 语音/视频通话
|
||||||
|
return ChatLabType.CALL
|
||||||
|
case 10000: // 系统消息
|
||||||
|
return ChatLabType.SYSTEM
|
||||||
|
case 49: // 复合消息
|
||||||
|
return this.mapType49(msg)
|
||||||
|
case 244813135921: // 引用消息
|
||||||
|
return ChatLabType.REPLY
|
||||||
|
case 266287972401: // 拍一拍
|
||||||
|
return ChatLabType.POKE
|
||||||
|
case 8594229559345: // 红包
|
||||||
|
return ChatLabType.RED_PACKET
|
||||||
|
case 8589934592049: // 转账
|
||||||
|
return ChatLabType.TRANSFER
|
||||||
|
default:
|
||||||
|
return ChatLabType.OTHER
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 映射 Type 49 子类型
|
||||||
|
*/
|
||||||
|
private mapType49(msg: Message): number {
|
||||||
|
const xmlType = msg.xmlType
|
||||||
|
|
||||||
|
switch (xmlType) {
|
||||||
|
case '5': // 链接
|
||||||
|
case '49':
|
||||||
|
return ChatLabType.LINK
|
||||||
|
case '6': // 文件
|
||||||
|
return ChatLabType.FILE
|
||||||
|
case '19': // 聊天记录
|
||||||
|
return ChatLabType.FORWARD
|
||||||
|
case '33': // 小程序
|
||||||
|
case '36':
|
||||||
|
return ChatLabType.SHARE
|
||||||
|
case '57': // 引用消息
|
||||||
|
return ChatLabType.REPLY
|
||||||
|
case '2000': // 转账
|
||||||
|
return ChatLabType.TRANSFER
|
||||||
|
case '2001': // 红包
|
||||||
|
return ChatLabType.RED_PACKET
|
||||||
|
default:
|
||||||
|
return ChatLabType.OTHER
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取消息内容
|
||||||
|
*/
|
||||||
|
private getMessageContent(msg: Message): string | null {
|
||||||
|
// 优先使用已解析的内容
|
||||||
|
if (msg.parsedContent) {
|
||||||
|
return msg.parsedContent
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据类型返回占位符
|
||||||
|
switch (msg.localType) {
|
||||||
|
case 1:
|
||||||
|
return msg.rawContent || null
|
||||||
|
case 3:
|
||||||
|
return msg.imageMd5 || '[图片]'
|
||||||
|
case 34:
|
||||||
|
return '[语音]'
|
||||||
|
case 43:
|
||||||
|
return msg.videoMd5 || '[视频]'
|
||||||
|
case 47:
|
||||||
|
return msg.emojiCdnUrl || msg.emojiMd5 || '[表情]'
|
||||||
|
case 42:
|
||||||
|
return msg.cardNickname || '[名片]'
|
||||||
|
case 48:
|
||||||
|
return '[位置]'
|
||||||
|
case 49:
|
||||||
|
return msg.linkTitle || msg.fileName || '[消息]'
|
||||||
|
default:
|
||||||
|
return msg.rawContent || null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 发送 JSON 响应
|
||||||
|
*/
|
||||||
|
private sendJson(res: http.ServerResponse, data: any): void {
|
||||||
|
res.setHeader('Content-Type', 'application/json; charset=utf-8')
|
||||||
|
res.writeHead(200)
|
||||||
|
res.end(JSON.stringify(data, null, 2))
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 发送错误响应
|
||||||
|
*/
|
||||||
|
private sendError(res: http.ServerResponse, code: number, message: string): void {
|
||||||
|
res.setHeader('Content-Type', 'application/json; charset=utf-8')
|
||||||
|
res.writeHead(code)
|
||||||
|
res.end(JSON.stringify({ error: message }))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const httpService = new HttpService()
|
||||||
@@ -1,13 +1,45 @@
|
|||||||
import { app, BrowserWindow } from 'electron'
|
import { app, BrowserWindow } from 'electron'
|
||||||
import { basename, dirname, extname, join } from 'path'
|
import { basename, dirname, extname, join } from 'path'
|
||||||
import { pathToFileURL } from 'url'
|
import { pathToFileURL } from 'url'
|
||||||
import { existsSync, mkdirSync, readdirSync, readFileSync, statSync } from 'fs'
|
import { existsSync, mkdirSync, readdirSync, readFileSync, statSync, appendFileSync } from 'fs'
|
||||||
import { writeFile } from 'fs/promises'
|
import { writeFile, rm, readdir } from 'fs/promises'
|
||||||
import crypto from 'crypto'
|
import crypto from 'crypto'
|
||||||
import { Worker } from 'worker_threads'
|
import { Worker } from 'worker_threads'
|
||||||
import { ConfigService } from './config'
|
import { ConfigService } from './config'
|
||||||
import { wcdbService } from './wcdbService'
|
import { wcdbService } from './wcdbService'
|
||||||
|
|
||||||
|
// 获取 ffmpeg-static 的路径
|
||||||
|
function getStaticFfmpegPath(): string | null {
|
||||||
|
try {
|
||||||
|
// 优先处理打包后的路径
|
||||||
|
if (app.isPackaged) {
|
||||||
|
const resourcesPath = process.resourcesPath
|
||||||
|
const packedPath = join(resourcesPath, 'app.asar.unpacked', 'node_modules', 'ffmpeg-static', 'ffmpeg.exe')
|
||||||
|
if (existsSync(packedPath)) {
|
||||||
|
return packedPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 方法1: 直接 require ffmpeg-static(开发环境)
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-var-requires
|
||||||
|
const ffmpegStatic = require('ffmpeg-static')
|
||||||
|
|
||||||
|
if (typeof ffmpegStatic === 'string' && existsSync(ffmpegStatic)) {
|
||||||
|
return ffmpegStatic
|
||||||
|
}
|
||||||
|
|
||||||
|
// 方法2: 手动构建路径(开发环境备用)
|
||||||
|
const devPath = join(process.cwd(), 'node_modules', 'ffmpeg-static', 'ffmpeg.exe')
|
||||||
|
if (existsSync(devPath)) {
|
||||||
|
return devPath
|
||||||
|
}
|
||||||
|
|
||||||
|
return null
|
||||||
|
} catch {
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type DecryptResult = {
|
type DecryptResult = {
|
||||||
success: boolean
|
success: boolean
|
||||||
localPath?: string
|
localPath?: string
|
||||||
@@ -32,10 +64,37 @@ export class ImageDecryptService {
|
|||||||
|
|
||||||
private logInfo(message: string, meta?: Record<string, unknown>): void {
|
private logInfo(message: string, meta?: Record<string, unknown>): void {
|
||||||
if (!this.configService.get('logEnabled')) return
|
if (!this.configService.get('logEnabled')) return
|
||||||
if (meta) {
|
const timestamp = new Date().toISOString()
|
||||||
console.info(message, meta)
|
const metaStr = meta ? ` ${JSON.stringify(meta)}` : ''
|
||||||
} else {
|
const logLine = `[${timestamp}] [ImageDecrypt] ${message}${metaStr}\n`
|
||||||
console.info(message)
|
|
||||||
|
// 只写入文件,不输出到控制台
|
||||||
|
this.writeLog(logLine)
|
||||||
|
}
|
||||||
|
|
||||||
|
private logError(message: string, error?: unknown, meta?: Record<string, unknown>): void {
|
||||||
|
if (!this.configService.get('logEnabled')) return
|
||||||
|
const timestamp = new Date().toISOString()
|
||||||
|
const errorStr = error ? ` Error: ${String(error)}` : ''
|
||||||
|
const metaStr = meta ? ` ${JSON.stringify(meta)}` : ''
|
||||||
|
const logLine = `[${timestamp}] [ImageDecrypt] ERROR: ${message}${errorStr}${metaStr}\n`
|
||||||
|
|
||||||
|
// 同时输出到控制台
|
||||||
|
console.error(message, error, meta)
|
||||||
|
|
||||||
|
// 写入日志文件
|
||||||
|
this.writeLog(logLine)
|
||||||
|
}
|
||||||
|
|
||||||
|
private writeLog(line: string): void {
|
||||||
|
try {
|
||||||
|
const logDir = join(app.getPath('userData'), 'logs')
|
||||||
|
if (!existsSync(logDir)) {
|
||||||
|
mkdirSync(logDir, { recursive: true })
|
||||||
|
}
|
||||||
|
appendFileSync(join(logDir, 'wcdb.log'), line, { encoding: 'utf8' })
|
||||||
|
} catch (err) {
|
||||||
|
console.error('写入日志失败:', err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -81,6 +140,7 @@ export class ImageDecryptService {
|
|||||||
return { success: true, localPath: dataUrl || this.filePathToUrl(existing), hasUpdate }
|
return { success: true, localPath: dataUrl || this.filePathToUrl(existing), hasUpdate }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
this.logInfo('未找到缓存', { md5: payload.imageMd5, datName: payload.imageDatName })
|
||||||
return { success: false, error: '未找到缓存图片' }
|
return { success: false, error: '未找到缓存图片' }
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -120,15 +180,18 @@ export class ImageDecryptService {
|
|||||||
payload: { sessionId?: string; imageMd5?: string; imageDatName?: string; force?: boolean },
|
payload: { sessionId?: string; imageMd5?: string; imageDatName?: string; force?: boolean },
|
||||||
cacheKey: string
|
cacheKey: string
|
||||||
): Promise<DecryptResult> {
|
): Promise<DecryptResult> {
|
||||||
|
this.logInfo('开始解密图片', { md5: payload.imageMd5, datName: payload.imageDatName, force: payload.force })
|
||||||
try {
|
try {
|
||||||
const wxid = this.configService.get('myWxid')
|
const wxid = this.configService.get('myWxid')
|
||||||
const dbPath = this.configService.get('dbPath')
|
const dbPath = this.configService.get('dbPath')
|
||||||
if (!wxid || !dbPath) {
|
if (!wxid || !dbPath) {
|
||||||
|
this.logError('配置缺失', undefined, { wxid: !!wxid, dbPath: !!dbPath })
|
||||||
return { success: false, error: '未配置账号或数据库路径' }
|
return { success: false, error: '未配置账号或数据库路径' }
|
||||||
}
|
}
|
||||||
|
|
||||||
const accountDir = this.resolveAccountDir(dbPath, wxid)
|
const accountDir = this.resolveAccountDir(dbPath, wxid)
|
||||||
if (!accountDir) {
|
if (!accountDir) {
|
||||||
|
this.logError('未找到账号目录', undefined, { dbPath, wxid })
|
||||||
return { success: false, error: '未找到账号目录' }
|
return { success: false, error: '未找到账号目录' }
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -142,12 +205,16 @@ export class ImageDecryptService {
|
|||||||
|
|
||||||
// 如果要求高清图但没找到,直接返回提示
|
// 如果要求高清图但没找到,直接返回提示
|
||||||
if (!datPath && payload.force) {
|
if (!datPath && payload.force) {
|
||||||
|
this.logError('未找到高清图', undefined, { md5: payload.imageMd5, datName: payload.imageDatName })
|
||||||
return { success: false, error: '未找到高清图,请在微信中点开该图片查看后重试' }
|
return { success: false, error: '未找到高清图,请在微信中点开该图片查看后重试' }
|
||||||
}
|
}
|
||||||
if (!datPath) {
|
if (!datPath) {
|
||||||
|
this.logError('未找到DAT文件', undefined, { md5: payload.imageMd5, datName: payload.imageDatName })
|
||||||
return { success: false, error: '未找到图片文件' }
|
return { success: false, error: '未找到图片文件' }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
this.logInfo('找到DAT文件', { datPath })
|
||||||
|
|
||||||
if (!extname(datPath).toLowerCase().includes('dat')) {
|
if (!extname(datPath).toLowerCase().includes('dat')) {
|
||||||
this.cacheResolvedPaths(cacheKey, payload.imageMd5, payload.imageDatName, datPath)
|
this.cacheResolvedPaths(cacheKey, payload.imageMd5, payload.imageDatName, datPath)
|
||||||
const dataUrl = this.fileToDataUrl(datPath)
|
const dataUrl = this.fileToDataUrl(datPath)
|
||||||
@@ -160,6 +227,7 @@ export class ImageDecryptService {
|
|||||||
// 查找已缓存的解密文件
|
// 查找已缓存的解密文件
|
||||||
const existing = this.findCachedOutput(cacheKey, payload.force, payload.sessionId)
|
const existing = this.findCachedOutput(cacheKey, payload.force, payload.sessionId)
|
||||||
if (existing) {
|
if (existing) {
|
||||||
|
this.logInfo('找到已解密文件', { existing, isHd: this.isHdPath(existing) })
|
||||||
const isHd = this.isHdPath(existing)
|
const isHd = this.isHdPath(existing)
|
||||||
// 如果要求高清但找到的是缩略图,继续解密高清图
|
// 如果要求高清但找到的是缩略图,继续解密高清图
|
||||||
if (!(payload.force && !isHd)) {
|
if (!(payload.force && !isHd)) {
|
||||||
@@ -192,23 +260,45 @@ export class ImageDecryptService {
|
|||||||
const aesKeyRaw = this.configService.get('imageAesKey')
|
const aesKeyRaw = this.configService.get('imageAesKey')
|
||||||
const aesKey = this.resolveAesKey(aesKeyRaw)
|
const aesKey = this.resolveAesKey(aesKeyRaw)
|
||||||
|
|
||||||
const decrypted = await this.decryptDatAuto(datPath, xorKey, aesKey)
|
this.logInfo('开始解密DAT文件', { datPath, xorKey, hasAesKey: !!aesKey })
|
||||||
|
let decrypted = await this.decryptDatAuto(datPath, xorKey, aesKey)
|
||||||
|
|
||||||
const ext = this.detectImageExtension(decrypted) || '.jpg'
|
// 检查是否是 wxgf 格式,如果是则尝试提取真实图片数据
|
||||||
|
const wxgfResult = await this.unwrapWxgf(decrypted)
|
||||||
|
decrypted = wxgfResult.data
|
||||||
|
|
||||||
const outputPath = this.getCacheOutputPathFromDat(datPath, ext, payload.sessionId)
|
let ext = this.detectImageExtension(decrypted)
|
||||||
|
|
||||||
|
// 如果是 wxgf 格式且没检测到扩展名
|
||||||
|
if (wxgfResult.isWxgf && !ext) {
|
||||||
|
ext = '.hevc'
|
||||||
|
}
|
||||||
|
|
||||||
|
const finalExt = ext || '.jpg'
|
||||||
|
|
||||||
|
const outputPath = this.getCacheOutputPathFromDat(datPath, finalExt, payload.sessionId)
|
||||||
await writeFile(outputPath, decrypted)
|
await writeFile(outputPath, decrypted)
|
||||||
|
this.logInfo('解密成功', { outputPath, size: decrypted.length })
|
||||||
|
|
||||||
|
// 对于 hevc 格式,返回错误提示
|
||||||
|
if (finalExt === '.hevc') {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: '此图片为微信新格式(wxgf),需要安装 ffmpeg 才能显示',
|
||||||
|
isThumb: this.isThumbnailPath(datPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
const isThumb = this.isThumbnailPath(datPath)
|
const isThumb = this.isThumbnailPath(datPath)
|
||||||
this.cacheResolvedPaths(cacheKey, payload.imageMd5, payload.imageDatName, outputPath)
|
this.cacheResolvedPaths(cacheKey, payload.imageMd5, payload.imageDatName, outputPath)
|
||||||
if (!isThumb) {
|
if (!isThumb) {
|
||||||
this.clearUpdateFlags(cacheKey, payload.imageMd5, payload.imageDatName)
|
this.clearUpdateFlags(cacheKey, payload.imageMd5, payload.imageDatName)
|
||||||
}
|
}
|
||||||
const dataUrl = this.bufferToDataUrl(decrypted, ext)
|
const dataUrl = this.bufferToDataUrl(decrypted, finalExt)
|
||||||
const localPath = dataUrl || this.filePathToUrl(outputPath)
|
const localPath = dataUrl || this.filePathToUrl(outputPath)
|
||||||
this.emitCacheResolved(payload, cacheKey, localPath)
|
this.emitCacheResolved(payload, cacheKey, localPath)
|
||||||
return { success: true, localPath, isThumb }
|
return { success: true, localPath, isThumb }
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
|
this.logError('解密失败', e, { md5: payload.imageMd5, datName: payload.imageDatName })
|
||||||
return { success: false, error: String(e) }
|
return { success: false, error: String(e) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -290,9 +380,9 @@ export class ImageDecryptService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
if (suffixMatch) return suffixMatch[1]
|
const cleaned = suffixMatch ? suffixMatch[1] : trimmed
|
||||||
|
|
||||||
return trimmed
|
return cleaned
|
||||||
}
|
}
|
||||||
|
|
||||||
private async resolveDatPath(
|
private async resolveDatPath(
|
||||||
@@ -325,10 +415,16 @@ export class ImageDecryptService {
|
|||||||
if (imageDatName) this.cacheDatPath(accountDir, imageDatName, hardlinkPath)
|
if (imageDatName) this.cacheDatPath(accountDir, imageDatName, hardlinkPath)
|
||||||
return hardlinkPath
|
return hardlinkPath
|
||||||
}
|
}
|
||||||
// hardlink 找到的是缩略图,但要求高清图,直接返回 null,不再搜索
|
// hardlink 找到的是缩略图,但要求高清图
|
||||||
if (!allowThumbnail && isThumb) {
|
// 尝试在同一目录下查找高清图变体(快速查找,不遍历)
|
||||||
return null
|
const hdPath = this.findHdVariantInSameDir(hardlinkPath)
|
||||||
|
if (hdPath) {
|
||||||
|
this.cacheDatPath(accountDir, imageMd5, hdPath)
|
||||||
|
if (imageDatName) this.cacheDatPath(accountDir, imageDatName, hdPath)
|
||||||
|
return hdPath
|
||||||
}
|
}
|
||||||
|
// 没找到高清图,返回 null(不进行全局搜索)
|
||||||
|
return null
|
||||||
}
|
}
|
||||||
this.logInfo('[ImageDecrypt] hardlink miss (md5)', { imageMd5 })
|
this.logInfo('[ImageDecrypt] hardlink miss (md5)', { imageMd5 })
|
||||||
if (imageDatName && this.looksLikeMd5(imageDatName) && imageDatName !== imageMd5) {
|
if (imageDatName && this.looksLikeMd5(imageDatName) && imageDatName !== imageMd5) {
|
||||||
@@ -341,9 +437,13 @@ export class ImageDecryptService {
|
|||||||
this.cacheDatPath(accountDir, imageDatName, fallbackPath)
|
this.cacheDatPath(accountDir, imageDatName, fallbackPath)
|
||||||
return fallbackPath
|
return fallbackPath
|
||||||
}
|
}
|
||||||
if (!allowThumbnail && isThumb) {
|
// 找到缩略图但要求高清图,尝试同目录查找高清图变体
|
||||||
return null
|
const hdPath = this.findHdVariantInSameDir(fallbackPath)
|
||||||
|
if (hdPath) {
|
||||||
|
this.cacheDatPath(accountDir, imageDatName, hdPath)
|
||||||
|
return hdPath
|
||||||
}
|
}
|
||||||
|
return null
|
||||||
}
|
}
|
||||||
this.logInfo('[ImageDecrypt] hardlink miss (datName)', { imageDatName })
|
this.logInfo('[ImageDecrypt] hardlink miss (datName)', { imageDatName })
|
||||||
}
|
}
|
||||||
@@ -359,10 +459,13 @@ export class ImageDecryptService {
|
|||||||
this.cacheDatPath(accountDir, imageDatName, hardlinkPath)
|
this.cacheDatPath(accountDir, imageDatName, hardlinkPath)
|
||||||
return hardlinkPath
|
return hardlinkPath
|
||||||
}
|
}
|
||||||
// hardlink 找到的是缩略图,但要求高清图,直接返回 null
|
// hardlink 找到的是缩略图,但要求高清图
|
||||||
if (!allowThumbnail && isThumb) {
|
const hdPath = this.findHdVariantInSameDir(hardlinkPath)
|
||||||
return null
|
if (hdPath) {
|
||||||
|
this.cacheDatPath(accountDir, imageDatName, hdPath)
|
||||||
|
return hdPath
|
||||||
}
|
}
|
||||||
|
return null
|
||||||
}
|
}
|
||||||
this.logInfo('[ImageDecrypt] hardlink miss (datName)', { imageDatName })
|
this.logInfo('[ImageDecrypt] hardlink miss (datName)', { imageDatName })
|
||||||
}
|
}
|
||||||
@@ -377,6 +480,9 @@ export class ImageDecryptService {
|
|||||||
const cached = this.resolvedCache.get(imageDatName)
|
const cached = this.resolvedCache.get(imageDatName)
|
||||||
if (cached && existsSync(cached)) {
|
if (cached && existsSync(cached)) {
|
||||||
if (allowThumbnail || !this.isThumbnailPath(cached)) return cached
|
if (allowThumbnail || !this.isThumbnailPath(cached)) return cached
|
||||||
|
// 缓存的是缩略图,尝试找高清图
|
||||||
|
const hdPath = this.findHdVariantInSameDir(cached)
|
||||||
|
if (hdPath) return hdPath
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -671,6 +777,17 @@ export class ImageDecryptService {
|
|||||||
|
|
||||||
const root = join(accountDir, 'msg', 'attach')
|
const root = join(accountDir, 'msg', 'attach')
|
||||||
if (!existsSync(root)) return null
|
if (!existsSync(root)) return null
|
||||||
|
|
||||||
|
// 优化1:快速概率性查找
|
||||||
|
// 包含:1. 基于文件名的前缀猜测 (旧版)
|
||||||
|
// 2. 基于日期的最近月份扫描 (新版无索引时)
|
||||||
|
const fastHit = await this.fastProbabilisticSearch(root, datName)
|
||||||
|
if (fastHit) {
|
||||||
|
this.resolvedCache.set(key, fastHit)
|
||||||
|
return fastHit
|
||||||
|
}
|
||||||
|
|
||||||
|
// 优化2:兜底扫描 (异步非阻塞)
|
||||||
const found = await this.walkForDatInWorker(root, datName.toLowerCase(), 8, allowThumbnail, thumbOnly)
|
const found = await this.walkForDatInWorker(root, datName.toLowerCase(), 8, allowThumbnail, thumbOnly)
|
||||||
if (found) {
|
if (found) {
|
||||||
this.resolvedCache.set(key, found)
|
this.resolvedCache.set(key, found)
|
||||||
@@ -679,6 +796,134 @@ export class ImageDecryptService {
|
|||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 基于文件名的哈希特征猜测可能的路径
|
||||||
|
* 包含:1. 微信旧版结构 filename.substr(0, 2)/...
|
||||||
|
* 2. 微信新版结构 msg/attach/{hash}/{YYYY-MM}/Img/filename
|
||||||
|
*/
|
||||||
|
private async fastProbabilisticSearch(root: string, datName: string): Promise<string | null> {
|
||||||
|
const { promises: fs } = require('fs')
|
||||||
|
const { join } = require('path')
|
||||||
|
|
||||||
|
try {
|
||||||
|
// --- 策略 A: 旧版路径猜测 (msg/attach/xx/yy/...) ---
|
||||||
|
const lowerName = datName.toLowerCase()
|
||||||
|
let baseName = lowerName
|
||||||
|
if (baseName.endsWith('.dat')) {
|
||||||
|
baseName = baseName.slice(0, -4)
|
||||||
|
if (baseName.endsWith('_t') || baseName.endsWith('.t') || baseName.endsWith('_hd')) {
|
||||||
|
baseName = baseName.slice(0, -3)
|
||||||
|
} else if (baseName.endsWith('_thumb')) {
|
||||||
|
baseName = baseName.slice(0, -6)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const candidates: string[] = []
|
||||||
|
if (/^[a-f0-9]{32}$/.test(baseName)) {
|
||||||
|
const dir1 = baseName.substring(0, 2)
|
||||||
|
const dir2 = baseName.substring(2, 4)
|
||||||
|
candidates.push(
|
||||||
|
join(root, dir1, dir2, datName),
|
||||||
|
join(root, dir1, dir2, 'Img', datName),
|
||||||
|
join(root, dir1, dir2, 'mg', datName),
|
||||||
|
join(root, dir1, dir2, 'Image', datName)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const path of candidates) {
|
||||||
|
try {
|
||||||
|
await fs.access(path)
|
||||||
|
return path
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- 策略 B: 新版 Session 哈希路径猜测 ---
|
||||||
|
try {
|
||||||
|
const entries = await fs.readdir(root, { withFileTypes: true })
|
||||||
|
const sessionDirs = entries
|
||||||
|
.filter((e: any) => e.isDirectory() && e.name.length === 32 && /^[a-f0-9]+$/i.test(e.name))
|
||||||
|
.map((e: any) => e.name)
|
||||||
|
|
||||||
|
if (sessionDirs.length === 0) return null
|
||||||
|
|
||||||
|
const now = new Date()
|
||||||
|
const months: string[] = []
|
||||||
|
for (let i = 0; i < 2; i++) {
|
||||||
|
const d = new Date(now.getFullYear(), now.getMonth() - i, 1)
|
||||||
|
const mStr = `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}`
|
||||||
|
months.push(mStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
const targetNames = [datName]
|
||||||
|
if (baseName !== lowerName) {
|
||||||
|
targetNames.push(`${baseName}.dat`)
|
||||||
|
targetNames.push(`${baseName}_t.dat`)
|
||||||
|
targetNames.push(`${baseName}_thumb.dat`)
|
||||||
|
}
|
||||||
|
|
||||||
|
const batchSize = 20
|
||||||
|
for (let i = 0; i < sessionDirs.length; i += batchSize) {
|
||||||
|
const batch = sessionDirs.slice(i, i + batchSize)
|
||||||
|
const tasks = batch.map(async (sessDir: string) => {
|
||||||
|
for (const month of months) {
|
||||||
|
const subDirs = ['Img', 'Image']
|
||||||
|
for (const sub of subDirs) {
|
||||||
|
const dirPath = join(root, sessDir, month, sub)
|
||||||
|
try { await fs.access(dirPath) } catch { continue }
|
||||||
|
for (const name of targetNames) {
|
||||||
|
const p = join(dirPath, name)
|
||||||
|
try { await fs.access(p); return p } catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
})
|
||||||
|
const results = await Promise.all(tasks)
|
||||||
|
const hit = results.find(r => r !== null)
|
||||||
|
if (hit) return hit
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
|
||||||
|
} catch { }
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 在同一目录下查找高清图变体
|
||||||
|
* 缩略图: xxx_t.dat -> 高清图: xxx_h.dat 或 xxx.dat
|
||||||
|
*/
|
||||||
|
private findHdVariantInSameDir(thumbPath: string): string | null {
|
||||||
|
try {
|
||||||
|
const dir = dirname(thumbPath)
|
||||||
|
const fileName = basename(thumbPath).toLowerCase()
|
||||||
|
|
||||||
|
// 提取基础名称(去掉 _t.dat 或 .t.dat)
|
||||||
|
let baseName = fileName
|
||||||
|
if (baseName.endsWith('_t.dat')) {
|
||||||
|
baseName = baseName.slice(0, -6)
|
||||||
|
} else if (baseName.endsWith('.t.dat')) {
|
||||||
|
baseName = baseName.slice(0, -6)
|
||||||
|
} else {
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尝试查找高清图变体
|
||||||
|
const variants = [
|
||||||
|
`${baseName}_h.dat`,
|
||||||
|
`${baseName}.h.dat`,
|
||||||
|
`${baseName}.dat`
|
||||||
|
]
|
||||||
|
|
||||||
|
for (const variant of variants) {
|
||||||
|
const variantPath = join(dir, variant)
|
||||||
|
if (existsSync(variantPath)) {
|
||||||
|
return variantPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
private async searchDatFileInDir(
|
private async searchDatFileInDir(
|
||||||
dirPath: string,
|
dirPath: string,
|
||||||
datName: string,
|
datName: string,
|
||||||
@@ -809,35 +1054,63 @@ export class ImageDecryptService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private findCachedOutput(cacheKey: string, preferHd: boolean = false, sessionId?: string): string | null {
|
private findCachedOutput(cacheKey: string, preferHd: boolean = false, sessionId?: string): string | null {
|
||||||
const root = this.getCacheRoot()
|
const allRoots = this.getAllCacheRoots()
|
||||||
const normalizedKey = this.normalizeDatBase(cacheKey.toLowerCase())
|
const normalizedKey = this.normalizeDatBase(cacheKey.toLowerCase())
|
||||||
const extensions = ['.jpg', '.jpeg', '.png', '.gif', '.webp']
|
const extensions = ['.jpg', '.jpeg', '.png', '.gif', '.webp']
|
||||||
|
|
||||||
|
// 遍历所有可能的缓存根路径
|
||||||
|
for (const root of allRoots) {
|
||||||
|
// 策略1: 新目录结构 Images/{sessionId}/{YYYY-MM}/{file}_hd.jpg
|
||||||
if (sessionId) {
|
if (sessionId) {
|
||||||
const sessionDir = join(root, this.sanitizeDirName(sessionId))
|
const sessionDir = join(root, this.sanitizeDirName(sessionId))
|
||||||
if (existsSync(sessionDir)) {
|
if (existsSync(sessionDir)) {
|
||||||
try {
|
try {
|
||||||
const sessionEntries = readdirSync(sessionDir)
|
const dateDirs = readdirSync(sessionDir, { withFileTypes: true })
|
||||||
for (const entry of sessionEntries) {
|
.filter(d => d.isDirectory() && /^\d{4}-\d{2}$/.test(d.name))
|
||||||
const timeDir = join(sessionDir, entry)
|
.map(d => d.name)
|
||||||
if (!this.isDirectory(timeDir)) continue
|
.sort()
|
||||||
const hit = this.findCachedOutputInDir(timeDir, normalizedKey, extensions, preferHd)
|
.reverse() // 最新的日期优先
|
||||||
if (hit) return hit
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// ignore
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 新目录结构: Images/{normalizedKey}/{normalizedKey}_thumb.jpg 或 _hd.jpg
|
for (const dateDir of dateDirs) {
|
||||||
const imageDir = join(root, normalizedKey)
|
const imageDir = join(sessionDir, dateDir)
|
||||||
if (existsSync(imageDir)) {
|
|
||||||
const hit = this.findCachedOutputInDir(imageDir, normalizedKey, extensions, preferHd)
|
const hit = this.findCachedOutputInDir(imageDir, normalizedKey, extensions, preferHd)
|
||||||
if (hit) return hit
|
if (hit) return hit
|
||||||
}
|
}
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 兼容旧的平铺结构
|
// 策略2: 遍历所有 sessionId 目录查找(如果没有指定 sessionId)
|
||||||
|
try {
|
||||||
|
const sessionDirs = readdirSync(root, { withFileTypes: true })
|
||||||
|
.filter(d => d.isDirectory())
|
||||||
|
.map(d => d.name)
|
||||||
|
|
||||||
|
for (const session of sessionDirs) {
|
||||||
|
const sessionDir = join(root, session)
|
||||||
|
// 检查是否是日期目录结构
|
||||||
|
try {
|
||||||
|
const subDirs = readdirSync(sessionDir, { withFileTypes: true })
|
||||||
|
.filter(d => d.isDirectory() && /^\d{4}-\d{2}$/.test(d.name))
|
||||||
|
.map(d => d.name)
|
||||||
|
|
||||||
|
for (const dateDir of subDirs) {
|
||||||
|
const imageDir = join(sessionDir, dateDir)
|
||||||
|
const hit = this.findCachedOutputInDir(imageDir, normalizedKey, extensions, preferHd)
|
||||||
|
if (hit) return hit
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
|
||||||
|
// 策略3: 旧目录结构 Images/{normalizedKey}/{normalizedKey}_thumb.jpg
|
||||||
|
const oldImageDir = join(root, normalizedKey)
|
||||||
|
if (existsSync(oldImageDir)) {
|
||||||
|
const hit = this.findCachedOutputInDir(oldImageDir, normalizedKey, extensions, preferHd)
|
||||||
|
if (hit) return hit
|
||||||
|
}
|
||||||
|
|
||||||
|
// 策略4: 最旧的平铺结构 Images/{file}.jpg
|
||||||
for (const ext of extensions) {
|
for (const ext of extensions) {
|
||||||
const candidate = join(root, `${cacheKey}${ext}`)
|
const candidate = join(root, `${cacheKey}${ext}`)
|
||||||
if (existsSync(candidate)) return candidate
|
if (existsSync(candidate)) return candidate
|
||||||
@@ -846,6 +1119,7 @@ export class ImageDecryptService {
|
|||||||
const candidate = join(root, `${cacheKey}_t${ext}`)
|
const candidate = join(root, `${cacheKey}_t${ext}`)
|
||||||
if (existsSync(candidate)) return candidate
|
if (existsSync(candidate)) return candidate
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
@@ -856,6 +1130,18 @@ export class ImageDecryptService {
|
|||||||
extensions: string[],
|
extensions: string[],
|
||||||
preferHd: boolean
|
preferHd: boolean
|
||||||
): string | null {
|
): string | null {
|
||||||
|
// 先检查并删除旧的 .hevc 文件(ffmpeg 转换失败时遗留的)
|
||||||
|
const hevcThumb = join(dirPath, `${normalizedKey}_thumb.hevc`)
|
||||||
|
const hevcHd = join(dirPath, `${normalizedKey}_hd.hevc`)
|
||||||
|
try {
|
||||||
|
if (existsSync(hevcThumb)) {
|
||||||
|
require('fs').unlinkSync(hevcThumb)
|
||||||
|
}
|
||||||
|
if (existsSync(hevcHd)) {
|
||||||
|
require('fs').unlinkSync(hevcHd)
|
||||||
|
}
|
||||||
|
} catch { }
|
||||||
|
|
||||||
for (const ext of extensions) {
|
for (const ext of extensions) {
|
||||||
if (preferHd) {
|
if (preferHd) {
|
||||||
const hdPath = join(dirPath, `${normalizedKey}_hd${ext}`)
|
const hdPath = join(dirPath, `${normalizedKey}_hd${ext}`)
|
||||||
@@ -863,6 +1149,8 @@ export class ImageDecryptService {
|
|||||||
}
|
}
|
||||||
const thumbPath = join(dirPath, `${normalizedKey}_thumb${ext}`)
|
const thumbPath = join(dirPath, `${normalizedKey}_thumb${ext}`)
|
||||||
if (existsSync(thumbPath)) return thumbPath
|
if (existsSync(thumbPath)) return thumbPath
|
||||||
|
|
||||||
|
// 允许返回 _hd 格式(因为它有 _hd 变体后缀)
|
||||||
if (!preferHd) {
|
if (!preferHd) {
|
||||||
const hdPath = join(dirPath, `${normalizedKey}_hd${ext}`)
|
const hdPath = join(dirPath, `${normalizedKey}_hd${ext}`)
|
||||||
if (existsSync(hdPath)) return hdPath
|
if (existsSync(hdPath)) return hdPath
|
||||||
@@ -960,8 +1248,9 @@ export class ImageDecryptService {
|
|||||||
const lower = entry.toLowerCase()
|
const lower = entry.toLowerCase()
|
||||||
if (!lower.endsWith('.dat')) continue
|
if (!lower.endsWith('.dat')) continue
|
||||||
if (this.isThumbnailDat(lower)) continue
|
if (this.isThumbnailDat(lower)) continue
|
||||||
if (!this.hasXVariant(lower.slice(0, -4))) continue
|
|
||||||
const baseLower = lower.slice(0, -4)
|
const baseLower = lower.slice(0, -4)
|
||||||
|
// 只排除没有 _x 变体后缀的文件(允许 _hd、_h 等所有带变体的)
|
||||||
|
if (!this.hasXVariant(baseLower)) continue
|
||||||
if (this.normalizeDatBase(baseLower) !== target) continue
|
if (this.normalizeDatBase(baseLower) !== target) continue
|
||||||
return join(dirPath, entry)
|
return join(dirPath, entry)
|
||||||
}
|
}
|
||||||
@@ -973,6 +1262,7 @@ export class ImageDecryptService {
|
|||||||
if (!lower.endsWith('.dat')) return false
|
if (!lower.endsWith('.dat')) return false
|
||||||
if (this.isThumbnailDat(lower)) return false
|
if (this.isThumbnailDat(lower)) return false
|
||||||
const baseLower = lower.slice(0, -4)
|
const baseLower = lower.slice(0, -4)
|
||||||
|
// 只检查是否有 _x 变体后缀(允许 _hd、_h 等所有带变体的)
|
||||||
return this.hasXVariant(baseLower)
|
return this.hasXVariant(baseLower)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -998,15 +1288,19 @@ export class ImageDecryptService {
|
|||||||
if (this.cacheIndexed) return
|
if (this.cacheIndexed) return
|
||||||
if (this.cacheIndexing) return this.cacheIndexing
|
if (this.cacheIndexing) return this.cacheIndexing
|
||||||
this.cacheIndexing = new Promise((resolve) => {
|
this.cacheIndexing = new Promise((resolve) => {
|
||||||
const root = this.getCacheRoot()
|
// 扫描所有可能的缓存根目录
|
||||||
|
const allRoots = this.getAllCacheRoots()
|
||||||
|
this.logInfo('开始索引缓存', { roots: allRoots.length })
|
||||||
|
|
||||||
|
for (const root of allRoots) {
|
||||||
try {
|
try {
|
||||||
this.indexCacheDir(root, 2, 0)
|
this.indexCacheDir(root, 3, 0) // 增加深度到3,支持 sessionId/YYYY-MM 结构
|
||||||
} catch {
|
} catch (e) {
|
||||||
this.cacheIndexed = true
|
this.logError('索引目录失败', e, { root })
|
||||||
this.cacheIndexing = null
|
|
||||||
resolve()
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
this.logInfo('缓存索引完成', { entries: this.resolvedCache.size })
|
||||||
this.cacheIndexed = true
|
this.cacheIndexed = true
|
||||||
this.cacheIndexing = null
|
this.cacheIndexing = null
|
||||||
resolve()
|
resolve()
|
||||||
@@ -1014,6 +1308,39 @@ export class ImageDecryptService {
|
|||||||
return this.cacheIndexing
|
return this.cacheIndexing
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取所有可能的缓存根路径(用于查找已缓存的图片)
|
||||||
|
* 包含当前路径、配置路径、旧版本路径
|
||||||
|
*/
|
||||||
|
private getAllCacheRoots(): string[] {
|
||||||
|
const roots: string[] = []
|
||||||
|
const configured = this.configService.get('cachePath')
|
||||||
|
const documentsPath = app.getPath('documents')
|
||||||
|
|
||||||
|
// 主要路径(当前使用的)
|
||||||
|
const mainRoot = this.getCacheRoot()
|
||||||
|
roots.push(mainRoot)
|
||||||
|
|
||||||
|
// 如果配置了自定义路径,也检查其下的 Images
|
||||||
|
if (configured) {
|
||||||
|
roots.push(join(configured, 'Images'))
|
||||||
|
roots.push(join(configured, 'images'))
|
||||||
|
}
|
||||||
|
|
||||||
|
// 默认路径
|
||||||
|
roots.push(join(documentsPath, 'WeFlow', 'Images'))
|
||||||
|
roots.push(join(documentsPath, 'WeFlow', 'images'))
|
||||||
|
|
||||||
|
// 兼容旧路径(如果有的话)
|
||||||
|
roots.push(join(documentsPath, 'WeFlowData', 'Images'))
|
||||||
|
|
||||||
|
// 去重并过滤存在的路径
|
||||||
|
const uniqueRoots = Array.from(new Set(roots))
|
||||||
|
const existingRoots = uniqueRoots.filter(r => existsSync(r))
|
||||||
|
|
||||||
|
return existingRoots
|
||||||
|
}
|
||||||
|
|
||||||
private indexCacheDir(root: string, maxDepth: number, depth: number): void {
|
private indexCacheDir(root: string, maxDepth: number, depth: number): void {
|
||||||
let entries: string[]
|
let entries: string[]
|
||||||
try {
|
try {
|
||||||
@@ -1354,6 +1681,159 @@ export class ImageDecryptService {
|
|||||||
return mostCommonKey
|
return mostCommonKey
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解包 wxgf 格式
|
||||||
|
* wxgf 是微信的图片格式,内部使用 HEVC 编码
|
||||||
|
*/
|
||||||
|
private async unwrapWxgf(buffer: Buffer): Promise<{ data: Buffer; isWxgf: boolean }> {
|
||||||
|
// 检查是否是 wxgf 格式 (77 78 67 66 = "wxgf")
|
||||||
|
if (buffer.length < 20 ||
|
||||||
|
buffer[0] !== 0x77 || buffer[1] !== 0x78 ||
|
||||||
|
buffer[2] !== 0x67 || buffer[3] !== 0x66) {
|
||||||
|
return { data: buffer, isWxgf: false }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 先尝试搜索内嵌的传统图片签名
|
||||||
|
for (let i = 4; i < Math.min(buffer.length - 12, 4096); i++) {
|
||||||
|
if (buffer[i] === 0xff && buffer[i + 1] === 0xd8 && buffer[i + 2] === 0xff) {
|
||||||
|
return { data: buffer.subarray(i), isWxgf: false }
|
||||||
|
}
|
||||||
|
if (buffer[i] === 0x89 && buffer[i + 1] === 0x50 &&
|
||||||
|
buffer[i + 2] === 0x4e && buffer[i + 3] === 0x47) {
|
||||||
|
return { data: buffer.subarray(i), isWxgf: false }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 提取 HEVC NALU 裸流
|
||||||
|
const hevcData = this.extractHevcNalu(buffer)
|
||||||
|
if (!hevcData || hevcData.length < 100) {
|
||||||
|
return { data: buffer, isWxgf: true }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尝试用 ffmpeg 转换
|
||||||
|
try {
|
||||||
|
const jpgData = await this.convertHevcToJpg(hevcData)
|
||||||
|
if (jpgData && jpgData.length > 0) {
|
||||||
|
return { data: jpgData, isWxgf: false }
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
// ffmpeg 转换失败
|
||||||
|
}
|
||||||
|
|
||||||
|
return { data: hevcData, isWxgf: true }
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 从 wxgf 数据中提取 HEVC NALU 裸流
|
||||||
|
*/
|
||||||
|
private extractHevcNalu(buffer: Buffer): Buffer | null {
|
||||||
|
const nalUnits: Buffer[] = []
|
||||||
|
let i = 4
|
||||||
|
|
||||||
|
while (i < buffer.length - 4) {
|
||||||
|
if (buffer[i] === 0x00 && buffer[i + 1] === 0x00 &&
|
||||||
|
buffer[i + 2] === 0x00 && buffer[i + 3] === 0x01) {
|
||||||
|
let nalStart = i
|
||||||
|
let nalEnd = buffer.length
|
||||||
|
|
||||||
|
for (let j = i + 4; j < buffer.length - 3; j++) {
|
||||||
|
if (buffer[j] === 0x00 && buffer[j + 1] === 0x00) {
|
||||||
|
if (buffer[j + 2] === 0x01 ||
|
||||||
|
(buffer[j + 2] === 0x00 && j + 3 < buffer.length && buffer[j + 3] === 0x01)) {
|
||||||
|
nalEnd = j
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const nalUnit = buffer.subarray(nalStart, nalEnd)
|
||||||
|
if (nalUnit.length > 3) {
|
||||||
|
nalUnits.push(nalUnit)
|
||||||
|
}
|
||||||
|
i = nalEnd
|
||||||
|
} else {
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (nalUnits.length === 0) {
|
||||||
|
for (let j = 4; j < buffer.length - 4; j++) {
|
||||||
|
if (buffer[j] === 0x00 && buffer[j + 1] === 0x00 &&
|
||||||
|
buffer[j + 2] === 0x00 && buffer[j + 3] === 0x01) {
|
||||||
|
return buffer.subarray(j)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
return Buffer.concat(nalUnits)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取 ffmpeg 可执行文件路径
|
||||||
|
*/
|
||||||
|
private getFfmpegPath(): string {
|
||||||
|
const staticPath = getStaticFfmpegPath()
|
||||||
|
this.logInfo('ffmpeg 路径检测', { staticPath, exists: staticPath ? existsSync(staticPath) : false })
|
||||||
|
|
||||||
|
if (staticPath) {
|
||||||
|
return staticPath
|
||||||
|
}
|
||||||
|
|
||||||
|
// 回退到系统 ffmpeg
|
||||||
|
return 'ffmpeg'
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 使用 ffmpeg 将 HEVC 裸流转换为 JPG
|
||||||
|
*/
|
||||||
|
private convertHevcToJpg(hevcData: Buffer): Promise<Buffer | null> {
|
||||||
|
const ffmpeg = this.getFfmpegPath()
|
||||||
|
this.logInfo('ffmpeg 转换开始', { ffmpegPath: ffmpeg, hevcSize: hevcData.length })
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
const { spawn } = require('child_process')
|
||||||
|
const chunks: Buffer[] = []
|
||||||
|
const errChunks: Buffer[] = []
|
||||||
|
|
||||||
|
const proc = spawn(ffmpeg, [
|
||||||
|
'-hide_banner',
|
||||||
|
'-loglevel', 'error',
|
||||||
|
'-f', 'hevc',
|
||||||
|
'-i', 'pipe:0',
|
||||||
|
'-vframes', '1',
|
||||||
|
'-q:v', '3',
|
||||||
|
'-f', 'mjpeg',
|
||||||
|
'pipe:1'
|
||||||
|
], {
|
||||||
|
stdio: ['pipe', 'pipe', 'pipe'],
|
||||||
|
windowsHide: true
|
||||||
|
})
|
||||||
|
|
||||||
|
proc.stdout.on('data', (chunk: Buffer) => chunks.push(chunk))
|
||||||
|
proc.stderr.on('data', (chunk: Buffer) => errChunks.push(chunk))
|
||||||
|
|
||||||
|
proc.on('close', (code: number) => {
|
||||||
|
if (code === 0 && chunks.length > 0) {
|
||||||
|
this.logInfo('ffmpeg 转换成功', { outputSize: Buffer.concat(chunks).length })
|
||||||
|
resolve(Buffer.concat(chunks))
|
||||||
|
} else {
|
||||||
|
const errMsg = Buffer.concat(errChunks).toString()
|
||||||
|
this.logInfo('ffmpeg 转换失败', { code, error: errMsg })
|
||||||
|
resolve(null)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
proc.on('error', (err: Error) => {
|
||||||
|
this.logInfo('ffmpeg 进程错误', { error: err.message })
|
||||||
|
resolve(null)
|
||||||
|
})
|
||||||
|
|
||||||
|
proc.stdin.write(hevcData)
|
||||||
|
proc.stdin.end()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
// 保留原有的解密到文件方法(用于兼容)
|
// 保留原有的解密到文件方法(用于兼容)
|
||||||
async decryptToFile(inputPath: string, outputPath: string, xorKey: number, aesKey?: Buffer): Promise<void> {
|
async decryptToFile(inputPath: string, outputPath: string, xorKey: number, aesKey?: Buffer): Promise<void> {
|
||||||
const version = this.getDatVersion(inputPath)
|
const version = this.getDatVersion(inputPath)
|
||||||
@@ -1378,6 +1858,71 @@ export class ImageDecryptService {
|
|||||||
|
|
||||||
await writeFile(outputPath, decrypted)
|
await writeFile(outputPath, decrypted)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async clearCache(): Promise<{ success: boolean; error?: string }> {
|
||||||
|
this.resolvedCache.clear()
|
||||||
|
this.hardlinkCache.clear()
|
||||||
|
this.pending.clear()
|
||||||
|
this.updateFlags.clear()
|
||||||
|
this.cacheIndexed = false
|
||||||
|
this.cacheIndexing = null
|
||||||
|
|
||||||
|
const configured = this.configService.get('cachePath')
|
||||||
|
const root = configured
|
||||||
|
? join(configured, 'Images')
|
||||||
|
: join(app.getPath('documents'), 'WeFlow', 'Images')
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (!existsSync(root)) {
|
||||||
|
return { success: true }
|
||||||
|
}
|
||||||
|
const monthPattern = /^\d{4}-\d{2}$/
|
||||||
|
const clearFilesInDir = async (dirPath: string): Promise<void> => {
|
||||||
|
let entries: Array<{ name: string; isDirectory: () => boolean }>
|
||||||
|
try {
|
||||||
|
entries = await readdir(dirPath, { withFileTypes: true })
|
||||||
|
} catch {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for (const entry of entries) {
|
||||||
|
const fullPath = join(dirPath, entry.name)
|
||||||
|
if (entry.isDirectory()) {
|
||||||
|
await clearFilesInDir(fullPath)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
await rm(fullPath, { force: true })
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const traverse = async (dirPath: string): Promise<void> => {
|
||||||
|
let entries: Array<{ name: string; isDirectory: () => boolean }>
|
||||||
|
try {
|
||||||
|
entries = await readdir(dirPath, { withFileTypes: true })
|
||||||
|
} catch {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for (const entry of entries) {
|
||||||
|
const fullPath = join(dirPath, entry.name)
|
||||||
|
if (entry.isDirectory()) {
|
||||||
|
if (monthPattern.test(entry.name)) {
|
||||||
|
await clearFilesInDir(fullPath)
|
||||||
|
} else {
|
||||||
|
await traverse(fullPath)
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
await rm(fullPath, { force: true })
|
||||||
|
} catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
await traverse(root)
|
||||||
|
return { success: true }
|
||||||
|
} catch (e) {
|
||||||
|
return { success: false, error: String(e) }
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export const imageDecryptService = new ImageDecryptService()
|
export const imageDecryptService = new ImageDecryptService()
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
import { app } from 'electron'
|
import { app } from 'electron'
|
||||||
import { join, dirname, basename } from 'path'
|
import { join, dirname, basename } from 'path'
|
||||||
import { existsSync, readdirSync, readFileSync, statSync } from 'fs'
|
import { existsSync, readdirSync, readFileSync, statSync, copyFileSync, mkdirSync } from 'fs'
|
||||||
import { execFile, spawn } from 'child_process'
|
import { execFile, spawn } from 'child_process'
|
||||||
import { promisify } from 'util'
|
import { promisify } from 'util'
|
||||||
import crypto from 'crypto'
|
import crypto from 'crypto'
|
||||||
|
import os from 'os'
|
||||||
|
|
||||||
const execFileAsync = promisify(execFile)
|
const execFileAsync = promisify(execFile)
|
||||||
|
|
||||||
@@ -32,6 +33,7 @@ export class KeyService {
|
|||||||
private ReadProcessMemory: any = null
|
private ReadProcessMemory: any = null
|
||||||
private MEMORY_BASIC_INFORMATION: any = null
|
private MEMORY_BASIC_INFORMATION: any = null
|
||||||
private TerminateProcess: any = null
|
private TerminateProcess: any = null
|
||||||
|
private QueryFullProcessImageNameW: any = null
|
||||||
|
|
||||||
// User32
|
// User32
|
||||||
private EnumWindows: any = null
|
private EnumWindows: any = null
|
||||||
@@ -41,6 +43,7 @@ export class KeyService {
|
|||||||
private GetWindowThreadProcessId: any = null
|
private GetWindowThreadProcessId: any = null
|
||||||
private IsWindowVisible: any = null
|
private IsWindowVisible: any = null
|
||||||
private EnumChildWindows: any = null
|
private EnumChildWindows: any = null
|
||||||
|
private PostMessageW: any = null
|
||||||
private WNDENUMPROC_PTR: any = null
|
private WNDENUMPROC_PTR: any = null
|
||||||
|
|
||||||
// Advapi32
|
// Advapi32
|
||||||
@@ -55,20 +58,97 @@ export class KeyService {
|
|||||||
private readonly HKEY_LOCAL_MACHINE = 0x80000002
|
private readonly HKEY_LOCAL_MACHINE = 0x80000002
|
||||||
private readonly HKEY_CURRENT_USER = 0x80000001
|
private readonly HKEY_CURRENT_USER = 0x80000001
|
||||||
private readonly ERROR_SUCCESS = 0
|
private readonly ERROR_SUCCESS = 0
|
||||||
|
private readonly WM_CLOSE = 0x0010
|
||||||
|
|
||||||
private getDllPath(): string {
|
private getDllPath(): string {
|
||||||
const resourcesPath = app.isPackaged
|
const isPackaged = typeof app !== 'undefined' && app ? app.isPackaged : process.env.NODE_ENV === 'production'
|
||||||
? join(process.resourcesPath, 'resources')
|
|
||||||
: join(app.getAppPath(), 'resources')
|
// 候选路径列表
|
||||||
return join(resourcesPath, 'wx_key.dll')
|
const candidates: string[] = []
|
||||||
|
|
||||||
|
// 1. 显式环境变量 (最高优先级)
|
||||||
|
if (process.env.WX_KEY_DLL_PATH) {
|
||||||
|
candidates.push(process.env.WX_KEY_DLL_PATH)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isPackaged) {
|
||||||
|
// 生产环境: 通常在 resources 目录下,但也可能直接在 resources 根目录
|
||||||
|
candidates.push(join(process.resourcesPath, 'resources', 'wx_key.dll'))
|
||||||
|
candidates.push(join(process.resourcesPath, 'wx_key.dll'))
|
||||||
|
} else {
|
||||||
|
// 开发环境
|
||||||
|
const cwd = process.cwd()
|
||||||
|
candidates.push(join(cwd, 'resources', 'wx_key.dll'))
|
||||||
|
candidates.push(join(app.getAppPath(), 'resources', 'wx_key.dll'))
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查并返回第一个存在的路径
|
||||||
|
for (const path of candidates) {
|
||||||
|
if (existsSync(path)) {
|
||||||
|
return path
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果都没找到,返回最可能的路径以便报错信息有参考
|
||||||
|
return candidates[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查路径是否为 UNC 路径或网络路径
|
||||||
|
private isNetworkPath(path: string): boolean {
|
||||||
|
// UNC 路径以 \\ 开头
|
||||||
|
if (path.startsWith('\\\\')) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
// 检查是否为网络映射驱动器(简化检测:A: 表示驱动器)
|
||||||
|
// 注意:这是一个启发式检测,更准确的方式需要调用 GetDriveType Windows API
|
||||||
|
// 但对于大多数 VM 共享场景,UNC 路径检测已足够
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将 DLL 复制到本地临时目录
|
||||||
|
private localizeNetworkDll(originalPath: string): string {
|
||||||
|
try {
|
||||||
|
const tempDir = join(os.tmpdir(), 'weflow_dll_cache')
|
||||||
|
if (!existsSync(tempDir)) {
|
||||||
|
mkdirSync(tempDir, { recursive: true })
|
||||||
|
}
|
||||||
|
const localPath = join(tempDir, 'wx_key.dll')
|
||||||
|
|
||||||
|
// 检查是否已经有本地副本,如果有就使用它
|
||||||
|
if (existsSync(localPath)) {
|
||||||
|
|
||||||
|
return localPath
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
copyFileSync(originalPath, localPath)
|
||||||
|
|
||||||
|
return localPath
|
||||||
|
} catch (e) {
|
||||||
|
console.error('DLL 本地化失败:', e)
|
||||||
|
// 如果本地化失败,返回原路径
|
||||||
|
return originalPath
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private ensureLoaded(): boolean {
|
private ensureLoaded(): boolean {
|
||||||
if (this.initialized) return true
|
if (this.initialized) return true
|
||||||
|
|
||||||
|
let dllPath = ''
|
||||||
try {
|
try {
|
||||||
this.koffi = require('koffi')
|
this.koffi = require('koffi')
|
||||||
const dllPath = this.getDllPath()
|
dllPath = this.getDllPath()
|
||||||
if (!existsSync(dllPath)) return false
|
|
||||||
|
if (!existsSync(dllPath)) {
|
||||||
|
console.error(`wx_key.dll 不存在于路径: ${dllPath}`)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否为网络路径,如果是则本地化
|
||||||
|
if (this.isNetworkPath(dllPath)) {
|
||||||
|
|
||||||
|
dllPath = this.localizeNetworkDll(dllPath)
|
||||||
|
}
|
||||||
|
|
||||||
this.lib = this.koffi.load(dllPath)
|
this.lib = this.koffi.load(dllPath)
|
||||||
this.initHook = this.lib.func('bool InitializeHook(uint32 targetPid)')
|
this.initHook = this.lib.func('bool InitializeHook(uint32 targetPid)')
|
||||||
@@ -80,7 +160,14 @@ export class KeyService {
|
|||||||
this.initialized = true
|
this.initialized = true
|
||||||
return true
|
return true
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('加载 wx_key.dll 失败:', e)
|
const errorMsg = e instanceof Error ? e.message : String(e)
|
||||||
|
const errorStack = e instanceof Error ? e.stack : ''
|
||||||
|
console.error(`加载 wx_key.dll 失败`)
|
||||||
|
console.error(` 路径: ${dllPath}`)
|
||||||
|
console.error(` 错误: ${errorMsg}`)
|
||||||
|
if (errorStack) {
|
||||||
|
console.error(` 堆栈: ${errorStack}`)
|
||||||
|
}
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -110,6 +197,7 @@ export class KeyService {
|
|||||||
this.OpenProcess = this.kernel32.func('OpenProcess', 'HANDLE', ['uint32', 'bool', 'uint32'])
|
this.OpenProcess = this.kernel32.func('OpenProcess', 'HANDLE', ['uint32', 'bool', 'uint32'])
|
||||||
this.CloseHandle = this.kernel32.func('CloseHandle', 'bool', ['HANDLE'])
|
this.CloseHandle = this.kernel32.func('CloseHandle', 'bool', ['HANDLE'])
|
||||||
this.TerminateProcess = this.kernel32.func('TerminateProcess', 'bool', ['HANDLE', 'uint32'])
|
this.TerminateProcess = this.kernel32.func('TerminateProcess', 'bool', ['HANDLE', 'uint32'])
|
||||||
|
this.QueryFullProcessImageNameW = this.kernel32.func('QueryFullProcessImageNameW', 'bool', ['HANDLE', 'uint32', this.koffi.out('uint16*'), this.koffi.out('uint32*')])
|
||||||
this.VirtualQueryEx = this.kernel32.func('VirtualQueryEx', 'uint64', ['HANDLE', 'uint64', this.koffi.out(this.koffi.pointer(this.MEMORY_BASIC_INFORMATION)), 'uint64'])
|
this.VirtualQueryEx = this.kernel32.func('VirtualQueryEx', 'uint64', ['HANDLE', 'uint64', this.koffi.out(this.koffi.pointer(this.MEMORY_BASIC_INFORMATION)), 'uint64'])
|
||||||
this.ReadProcessMemory = this.kernel32.func('ReadProcessMemory', 'bool', ['HANDLE', 'uint64', 'void*', 'uint64', this.koffi.out(this.koffi.pointer('uint64'))])
|
this.ReadProcessMemory = this.kernel32.func('ReadProcessMemory', 'bool', ['HANDLE', 'uint64', 'void*', 'uint64', this.koffi.out(this.koffi.pointer('uint64'))])
|
||||||
|
|
||||||
@@ -138,6 +226,7 @@ export class KeyService {
|
|||||||
|
|
||||||
this.EnumWindows = this.user32.func('EnumWindows', 'bool', [this.WNDENUMPROC_PTR, 'intptr_t'])
|
this.EnumWindows = this.user32.func('EnumWindows', 'bool', [this.WNDENUMPROC_PTR, 'intptr_t'])
|
||||||
this.EnumChildWindows = this.user32.func('EnumChildWindows', 'bool', ['void*', this.WNDENUMPROC_PTR, 'intptr_t'])
|
this.EnumChildWindows = this.user32.func('EnumChildWindows', 'bool', ['void*', this.WNDENUMPROC_PTR, 'intptr_t'])
|
||||||
|
this.PostMessageW = this.user32.func('PostMessageW', 'bool', ['void*', 'uint32', 'uintptr_t', 'intptr_t'])
|
||||||
|
|
||||||
this.GetWindowTextW = this.user32.func('GetWindowTextW', 'int', ['void*', this.koffi.out('uint16*'), 'int'])
|
this.GetWindowTextW = this.user32.func('GetWindowTextW', 'int', ['void*', this.koffi.out('uint16*'), 'int'])
|
||||||
this.GetWindowTextLengthW = this.user32.func('GetWindowTextLengthW', 'int', ['void*'])
|
this.GetWindowTextLengthW = this.user32.func('GetWindowTextLengthW', 'int', ['void*'])
|
||||||
@@ -226,7 +315,46 @@ export class KeyService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private async getProcessExecutablePath(pid: number): Promise<string | null> {
|
||||||
|
if (!this.ensureKernel32()) return null
|
||||||
|
// 0x1000 = PROCESS_QUERY_LIMITED_INFORMATION
|
||||||
|
const hProcess = this.OpenProcess(0x1000, false, pid)
|
||||||
|
if (!hProcess) return null
|
||||||
|
|
||||||
|
try {
|
||||||
|
const sizeBuf = Buffer.alloc(4)
|
||||||
|
sizeBuf.writeUInt32LE(1024, 0)
|
||||||
|
const pathBuf = Buffer.alloc(1024 * 2)
|
||||||
|
|
||||||
|
const ret = this.QueryFullProcessImageNameW(hProcess, 0, pathBuf, sizeBuf)
|
||||||
|
if (ret) {
|
||||||
|
const len = sizeBuf.readUInt32LE(0)
|
||||||
|
return pathBuf.toString('ucs2', 0, len * 2)
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
} catch (e) {
|
||||||
|
console.error('获取进程路径失败:', e)
|
||||||
|
return null
|
||||||
|
} finally {
|
||||||
|
this.CloseHandle(hProcess)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private async findWeChatInstallPath(): Promise<string | null> {
|
private async findWeChatInstallPath(): Promise<string | null> {
|
||||||
|
// 0. 优先尝试获取正在运行的微信进程路径
|
||||||
|
try {
|
||||||
|
const pid = await this.findWeChatPid()
|
||||||
|
if (pid) {
|
||||||
|
const runPath = await this.getProcessExecutablePath(pid)
|
||||||
|
if (runPath && existsSync(runPath)) {
|
||||||
|
|
||||||
|
return runPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('尝试获取运行中微信路径失败:', e)
|
||||||
|
}
|
||||||
|
|
||||||
// 1. Registry - Uninstall Keys
|
// 1. Registry - Uninstall Keys
|
||||||
const uninstallKeys = [
|
const uninstallKeys = [
|
||||||
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall',
|
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall',
|
||||||
@@ -312,16 +440,60 @@ export class KeyService {
|
|||||||
return fallbackPid ?? null
|
return fallbackPid ?? null
|
||||||
}
|
}
|
||||||
|
|
||||||
private async killWeChatProcesses() {
|
private async waitForWeChatExit(timeoutMs = 8000): Promise<boolean> {
|
||||||
|
const start = Date.now()
|
||||||
|
while (Date.now() - start < timeoutMs) {
|
||||||
|
const weixinPid = await this.findPidByImageName('Weixin.exe')
|
||||||
|
const wechatPid = await this.findPidByImageName('WeChat.exe')
|
||||||
|
if (!weixinPid && !wechatPid) return true
|
||||||
|
await new Promise(r => setTimeout(r, 400))
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
private async closeWeChatWindows(): Promise<boolean> {
|
||||||
|
if (!this.ensureUser32()) return false
|
||||||
|
let requested = false
|
||||||
|
|
||||||
|
const enumWindowsCallback = this.koffi.register((hWnd: any, lParam: any) => {
|
||||||
|
if (!this.IsWindowVisible(hWnd)) return true
|
||||||
|
const title = this.getWindowTitle(hWnd)
|
||||||
|
const className = this.getClassName(hWnd)
|
||||||
|
const classLower = (className || '').toLowerCase()
|
||||||
|
const isWeChatWindow = this.isWeChatWindowTitle(title) || classLower.includes('wechat') || classLower.includes('weixin')
|
||||||
|
if (!isWeChatWindow) return true
|
||||||
|
|
||||||
|
requested = true
|
||||||
try {
|
try {
|
||||||
await execFileAsync('taskkill', ['/F', '/IM', 'Weixin.exe'])
|
this.PostMessageW?.(hWnd, this.WM_CLOSE, 0, 0)
|
||||||
await execFileAsync('taskkill', ['/F', '/IM', 'WeChat.exe'])
|
} catch { }
|
||||||
|
return true
|
||||||
|
}, this.WNDENUMPROC_PTR)
|
||||||
|
|
||||||
|
this.EnumWindows(enumWindowsCallback, 0)
|
||||||
|
this.koffi.unregister(enumWindowsCallback)
|
||||||
|
|
||||||
|
return requested
|
||||||
|
}
|
||||||
|
|
||||||
|
private async killWeChatProcesses(): Promise<boolean> {
|
||||||
|
const requested = await this.closeWeChatWindows()
|
||||||
|
if (requested) {
|
||||||
|
const gracefulOk = await this.waitForWeChatExit(1500)
|
||||||
|
if (gracefulOk) return true
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
await execFileAsync('taskkill', ['/F', '/T', '/IM', 'Weixin.exe'])
|
||||||
|
await execFileAsync('taskkill', ['/F', '/T', '/IM', 'WeChat.exe'])
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
// Ignore if not found
|
// Ignore if not found
|
||||||
}
|
}
|
||||||
await new Promise(r => setTimeout(r, 1000))
|
|
||||||
|
return await this.waitForWeChatExit(5000)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// --- Window Detection ---
|
// --- Window Detection ---
|
||||||
|
|
||||||
private getWindowTitle(hWnd: any): string {
|
private getWindowTitle(hWnd: any): string {
|
||||||
@@ -480,12 +652,21 @@ export class KeyService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 2. Restart WeChat
|
// 2. Restart WeChat
|
||||||
onStatus?.('正在重启微信以进行获取...', 0)
|
onStatus?.('正在关闭微信以进行获取...', 0)
|
||||||
await this.killWeChatProcesses()
|
const closed = await this.killWeChatProcesses()
|
||||||
|
if (!closed) {
|
||||||
|
const err = '无法自动关闭微信,请手动退出后重试'
|
||||||
|
onStatus?.(err, 2)
|
||||||
|
return { success: false, error: err }
|
||||||
|
}
|
||||||
|
|
||||||
// 3. Launch
|
// 3. Launch
|
||||||
onStatus?.('正在启动微信...', 0)
|
onStatus?.('正在启动微信...', 0)
|
||||||
const sub = spawn(wechatPath, { detached: true, stdio: 'ignore' })
|
const sub = spawn(wechatPath, {
|
||||||
|
detached: true,
|
||||||
|
stdio: 'ignore',
|
||||||
|
cwd: dirname(wechatPath)
|
||||||
|
})
|
||||||
sub.unref()
|
sub.unref()
|
||||||
|
|
||||||
// 4. Wait for Window & Get PID (Crucial change: discover PID from window)
|
// 4. Wait for Window & Get PID (Crucial change: discover PID from window)
|
||||||
@@ -504,6 +685,11 @@ export class KeyService {
|
|||||||
if (!ok) {
|
if (!ok) {
|
||||||
const error = this.getLastErrorMsg ? this.decodeCString(this.getLastErrorMsg()) : ''
|
const error = this.getLastErrorMsg ? this.decodeCString(this.getLastErrorMsg()) : ''
|
||||||
if (error) {
|
if (error) {
|
||||||
|
// 检测权限不足错误 (NTSTATUS 0xC0000022 = STATUS_ACCESS_DENIED)
|
||||||
|
if (error.includes('0xC0000022') || error.includes('ACCESS_DENIED') || error.includes('打开目标进程失败')) {
|
||||||
|
const friendlyError = '权限不足:无法访问微信进程。\n\n解决方法:\n1. 右键 WeFlow 图标,选择"以管理员身份运行"\n2. 关闭可能拦截的安全软件(如360、火绒等)\n3. 确保微信没有以管理员权限运行'
|
||||||
|
return { success: false, error: friendlyError }
|
||||||
|
}
|
||||||
return { success: false, error }
|
return { success: false, error }
|
||||||
}
|
}
|
||||||
const statusBuffer = Buffer.alloc(256)
|
const statusBuffer = Buffer.alloc(256)
|
||||||
@@ -695,33 +881,41 @@ export class KeyService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private getXorKey(templateFiles: string[]): number | null {
|
private getXorKey(templateFiles: string[]): number | null {
|
||||||
const counts = new Map<string, number>()
|
const counts = new Map<number, number>()
|
||||||
|
const tailSignatures = [
|
||||||
|
Buffer.from([0xFF, 0xD9]),
|
||||||
|
Buffer.from([0x49, 0x45, 0x4E, 0x44, 0xAE, 0x42, 0x60, 0x82])
|
||||||
|
]
|
||||||
for (const file of templateFiles) {
|
for (const file of templateFiles) {
|
||||||
try {
|
try {
|
||||||
const bytes = readFileSync(file)
|
const bytes = readFileSync(file)
|
||||||
if (bytes.length < 2) continue
|
for (const signature of tailSignatures) {
|
||||||
const x = bytes[bytes.length - 2]
|
if (bytes.length < signature.length) continue
|
||||||
const y = bytes[bytes.length - 1]
|
const tail = bytes.subarray(bytes.length - signature.length)
|
||||||
const key = `${x}_${y}`
|
const xorKey = tail[0] ^ signature[0]
|
||||||
counts.set(key, (counts.get(key) ?? 0) + 1)
|
let valid = true
|
||||||
|
for (let i = 1; i < signature.length; i++) {
|
||||||
|
if ((tail[i] ^ xorKey) !== signature[i]) {
|
||||||
|
valid = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (valid) {
|
||||||
|
counts.set(xorKey, (counts.get(xorKey) ?? 0) + 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
} catch { }
|
} catch { }
|
||||||
}
|
}
|
||||||
if (!counts.size) return null
|
if (!counts.size) return null
|
||||||
let mostKey = ''
|
let bestKey: number | null = null
|
||||||
let mostCount = 0
|
let bestCount = 0
|
||||||
for (const [key, count] of counts) {
|
for (const [key, count] of counts) {
|
||||||
if (count > mostCount) {
|
if (count > bestCount) {
|
||||||
mostCount = count
|
bestCount = count
|
||||||
mostKey = key
|
bestKey = key
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (!mostKey) return null
|
return bestKey
|
||||||
const [xStr, yStr] = mostKey.split('_')
|
|
||||||
const x = Number(xStr)
|
|
||||||
const y = Number(yStr)
|
|
||||||
const xorKey = x ^ 0xFF
|
|
||||||
const check = y ^ 0xD9
|
|
||||||
return xorKey === check ? xorKey : null
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private getCiphertextFromTemplate(templateFiles: string[]): Buffer | null {
|
private getCiphertextFromTemplate(templateFiles: string[]): Buffer | null {
|
||||||
@@ -744,16 +938,17 @@ export class KeyService {
|
|||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
|
|
||||||
private isAlphaNumAscii(byte: number): boolean {
|
private isAlphaNumLower(byte: number): boolean {
|
||||||
return (byte >= 0x61 && byte <= 0x7a) || (byte >= 0x41 && byte <= 0x5a) || (byte >= 0x30 && byte <= 0x39)
|
// 只匹配小写字母 a-z 和数字 0-9(AES密钥格式)
|
||||||
|
return (byte >= 0x61 && byte <= 0x7a) || (byte >= 0x30 && byte <= 0x39)
|
||||||
}
|
}
|
||||||
|
|
||||||
private isUtf16AsciiKey(buf: Buffer, start: number): boolean {
|
private isUtf16LowerKey(buf: Buffer, start: number): boolean {
|
||||||
if (start + 64 > buf.length) return false
|
if (start + 64 > buf.length) return false
|
||||||
for (let j = 0; j < 32; j++) {
|
for (let j = 0; j < 32; j++) {
|
||||||
const charByte = buf[start + j * 2]
|
const charByte = buf[start + j * 2]
|
||||||
const nullByte = buf[start + j * 2 + 1]
|
const nullByte = buf[start + j * 2 + 1]
|
||||||
if (nullByte !== 0x00 || !this.isAlphaNumAscii(charByte)) {
|
if (nullByte !== 0x00 || !this.isAlphaNumLower(charByte)) {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -766,7 +961,17 @@ export class KeyService {
|
|||||||
const decipher = crypto.createDecipheriv('aes-128-ecb', key, null)
|
const decipher = crypto.createDecipheriv('aes-128-ecb', key, null)
|
||||||
decipher.setAutoPadding(false)
|
decipher.setAutoPadding(false)
|
||||||
const decrypted = Buffer.concat([decipher.update(ciphertext), decipher.final()])
|
const decrypted = Buffer.concat([decipher.update(ciphertext), decipher.final()])
|
||||||
return decrypted[0] === 0xff && decrypted[1] === 0xd8 && decrypted[2] === 0xff
|
const isJpeg = decrypted.length >= 3 && decrypted[0] === 0xff && decrypted[1] === 0xd8 && decrypted[2] === 0xff
|
||||||
|
const isPng = decrypted.length >= 8 &&
|
||||||
|
decrypted[0] === 0x89 &&
|
||||||
|
decrypted[1] === 0x50 &&
|
||||||
|
decrypted[2] === 0x4e &&
|
||||||
|
decrypted[3] === 0x47 &&
|
||||||
|
decrypted[4] === 0x0d &&
|
||||||
|
decrypted[5] === 0x0a &&
|
||||||
|
decrypted[6] === 0x1a &&
|
||||||
|
decrypted[7] === 0x0a
|
||||||
|
return isJpeg || isPng
|
||||||
} catch {
|
} catch {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
@@ -776,8 +981,6 @@ export class KeyService {
|
|||||||
const regions: Array<[number, number]> = []
|
const regions: Array<[number, number]> = []
|
||||||
const MEM_COMMIT = 0x1000
|
const MEM_COMMIT = 0x1000
|
||||||
const MEM_PRIVATE = 0x20000
|
const MEM_PRIVATE = 0x20000
|
||||||
const MEM_MAPPED = 0x40000
|
|
||||||
const MEM_IMAGE = 0x1000000
|
|
||||||
const PAGE_NOACCESS = 0x01
|
const PAGE_NOACCESS = 0x01
|
||||||
const PAGE_GUARD = 0x100
|
const PAGE_GUARD = 0x100
|
||||||
|
|
||||||
@@ -792,11 +995,10 @@ export class KeyService {
|
|||||||
const protect = info.Protect
|
const protect = info.Protect
|
||||||
const type = info.Type
|
const type = info.Type
|
||||||
const regionSize = Number(info.RegionSize)
|
const regionSize = Number(info.RegionSize)
|
||||||
if (state === MEM_COMMIT && (protect & PAGE_NOACCESS) === 0 && (protect & PAGE_GUARD) === 0) {
|
// 只收集已提交的私有内存(大幅减少扫描区域)
|
||||||
if (type === MEM_PRIVATE || type === MEM_MAPPED || type === MEM_IMAGE) {
|
if (state === MEM_COMMIT && type === MEM_PRIVATE && (protect & PAGE_NOACCESS) === 0 && (protect & PAGE_GUARD) === 0) {
|
||||||
regions.push([Number(info.BaseAddress), regionSize])
|
regions.push([Number(info.BaseAddress), regionSize])
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
const nextAddress = address + regionSize
|
const nextAddress = address + regionSize
|
||||||
if (nextAddress <= address) break
|
if (nextAddress <= address) break
|
||||||
@@ -813,72 +1015,63 @@ export class KeyService {
|
|||||||
return buffer.subarray(0, bytesRead[0])
|
return buffer.subarray(0, bytesRead[0])
|
||||||
}
|
}
|
||||||
|
|
||||||
private async getAesKeyFromMemory(pid: number, ciphertext: Buffer): Promise<string | null> {
|
private async getAesKeyFromMemory(
|
||||||
|
pid: number,
|
||||||
|
ciphertext: Buffer,
|
||||||
|
onProgress?: (current: number, total: number, message: string) => void
|
||||||
|
): Promise<string | null> {
|
||||||
if (!this.ensureKernel32()) return null
|
if (!this.ensureKernel32()) return null
|
||||||
const hProcess = this.OpenProcess(this.PROCESS_ALL_ACCESS, false, pid)
|
const hProcess = this.OpenProcess(this.PROCESS_ALL_ACCESS, false, pid)
|
||||||
if (!hProcess) return null
|
if (!hProcess) return null
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const regions = this.getMemoryRegions(hProcess)
|
const allRegions = this.getMemoryRegions(hProcess)
|
||||||
const chunkSize = 4 * 1024 * 1024
|
const totalRegions = allRegions.length
|
||||||
const overlap = 65
|
let scannedCount = 0
|
||||||
for (const [baseAddress, regionSize] of regions) {
|
let skippedCount = 0
|
||||||
if (regionSize > 100 * 1024 * 1024) continue
|
|
||||||
let offset = 0
|
for (const [baseAddress, regionSize] of allRegions) {
|
||||||
let trailing: Buffer | null = null
|
// 跳过太大的内存区域(> 100MB)
|
||||||
while (offset < regionSize) {
|
if (regionSize > 100 * 1024 * 1024) {
|
||||||
const remaining = regionSize - offset
|
skippedCount++
|
||||||
const currentChunkSize = remaining > chunkSize ? chunkSize : remaining
|
|
||||||
const chunk = this.readProcessMemory(hProcess, baseAddress + offset, currentChunkSize)
|
|
||||||
if (!chunk || !chunk.length) {
|
|
||||||
offset += currentChunkSize
|
|
||||||
trailing = null
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
let dataToScan: Buffer
|
scannedCount++
|
||||||
if (trailing && trailing.length) {
|
if (scannedCount % 10 === 0) {
|
||||||
dataToScan = Buffer.concat([trailing, chunk])
|
onProgress?.(scannedCount, totalRegions, `正在扫描微信内存... (${scannedCount}/${totalRegions})`)
|
||||||
} else {
|
await new Promise(resolve => setImmediate(resolve))
|
||||||
dataToScan = chunk
|
|
||||||
}
|
}
|
||||||
|
|
||||||
for (let i = 0; i < dataToScan.length - 34; i++) {
|
const memory = this.readProcessMemory(hProcess, baseAddress, regionSize)
|
||||||
if (this.isAlphaNumAscii(dataToScan[i])) continue
|
if (!memory) continue
|
||||||
|
|
||||||
|
// 直接在原始字节中搜索32字节的小写字母数字序列
|
||||||
|
for (let i = 0; i < memory.length - 34; i++) {
|
||||||
|
// 检查前导字符(不是小写字母或数字)
|
||||||
|
if (this.isAlphaNumLower(memory[i])) continue
|
||||||
|
|
||||||
|
// 检查接下来32个字节是否都是小写字母或数字
|
||||||
let valid = true
|
let valid = true
|
||||||
for (let j = 1; j <= 32; j++) {
|
for (let j = 1; j <= 32; j++) {
|
||||||
if (!this.isAlphaNumAscii(dataToScan[i + j])) {
|
if (!this.isAlphaNumLower(memory[i + j])) {
|
||||||
valid = false
|
valid = false
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (valid && this.isAlphaNumAscii(dataToScan[i + 33])) {
|
if (!valid) continue
|
||||||
valid = false
|
|
||||||
|
// 检查尾部字符(不是小写字母或数字)
|
||||||
|
if (i + 33 < memory.length && this.isAlphaNumLower(memory[i + 33])) {
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
if (valid) {
|
|
||||||
const keyBytes = dataToScan.subarray(i + 1, i + 33)
|
const keyBytes = memory.subarray(i + 1, i + 33)
|
||||||
if (this.verifyKey(ciphertext, keyBytes)) {
|
if (this.verifyKey(ciphertext, keyBytes)) {
|
||||||
return keyBytes.toString('ascii')
|
return keyBytes.toString('ascii')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (let i = 0; i < dataToScan.length - 65; i++) {
|
|
||||||
if (!this.isUtf16AsciiKey(dataToScan, i)) continue
|
|
||||||
const keyBytes = Buffer.alloc(32)
|
|
||||||
for (let j = 0; j < 32; j++) {
|
|
||||||
keyBytes[j] = dataToScan[i + j * 2]
|
|
||||||
}
|
|
||||||
if (this.verifyKey(ciphertext, keyBytes)) {
|
|
||||||
return keyBytes.toString('ascii')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const start = dataToScan.length - overlap
|
|
||||||
trailing = dataToScan.subarray(start < 0 ? 0 : start)
|
|
||||||
offset += currentChunkSize
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return null
|
return null
|
||||||
} finally {
|
} finally {
|
||||||
try {
|
try {
|
||||||
@@ -915,7 +1108,9 @@ export class KeyService {
|
|||||||
if (!pid) return { success: false, error: '未检测到微信进程' }
|
if (!pid) return { success: false, error: '未检测到微信进程' }
|
||||||
|
|
||||||
onProgress?.('正在扫描内存获取 AES 密钥...')
|
onProgress?.('正在扫描内存获取 AES 密钥...')
|
||||||
const aesKey = await this.getAesKeyFromMemory(pid, ciphertext)
|
const aesKey = await this.getAesKeyFromMemory(pid, ciphertext, (current, total, msg) => {
|
||||||
|
onProgress?.(`${msg} (${current}/${total})`)
|
||||||
|
})
|
||||||
if (!aesKey) {
|
if (!aesKey) {
|
||||||
return {
|
return {
|
||||||
success: false,
|
success: false,
|
||||||
|
|||||||
371
electron/services/llamaService.ts
Normal file
371
electron/services/llamaService.ts
Normal file
@@ -0,0 +1,371 @@
|
|||||||
|
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();
|
||||||
77
electron/services/messageCacheService.ts
Normal file
77
electron/services/messageCacheService.ts
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
import { join, dirname } from 'path'
|
||||||
|
import { existsSync, mkdirSync, readFileSync, writeFileSync, rmSync } from 'fs'
|
||||||
|
import { app } from 'electron'
|
||||||
|
|
||||||
|
export interface SessionMessageCacheEntry {
|
||||||
|
updatedAt: number
|
||||||
|
messages: any[]
|
||||||
|
}
|
||||||
|
|
||||||
|
export class MessageCacheService {
|
||||||
|
private readonly cacheFilePath: string
|
||||||
|
private cache: Record<string, SessionMessageCacheEntry> = {}
|
||||||
|
private readonly sessionLimit = 150
|
||||||
|
|
||||||
|
constructor(cacheBasePath?: string) {
|
||||||
|
const basePath = cacheBasePath && cacheBasePath.trim().length > 0
|
||||||
|
? cacheBasePath
|
||||||
|
: join(app.getPath('documents'), 'WeFlow')
|
||||||
|
this.cacheFilePath = join(basePath, 'session-messages.json')
|
||||||
|
this.ensureCacheDir()
|
||||||
|
this.loadCache()
|
||||||
|
}
|
||||||
|
|
||||||
|
private ensureCacheDir() {
|
||||||
|
const dir = dirname(this.cacheFilePath)
|
||||||
|
if (!existsSync(dir)) {
|
||||||
|
mkdirSync(dir, { recursive: true })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private loadCache() {
|
||||||
|
if (!existsSync(this.cacheFilePath)) return
|
||||||
|
try {
|
||||||
|
const raw = readFileSync(this.cacheFilePath, 'utf8')
|
||||||
|
const parsed = JSON.parse(raw)
|
||||||
|
if (parsed && typeof parsed === 'object') {
|
||||||
|
this.cache = parsed
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('MessageCacheService: 载入缓存失败', error)
|
||||||
|
this.cache = {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
get(sessionId: string): SessionMessageCacheEntry | undefined {
|
||||||
|
return this.cache[sessionId]
|
||||||
|
}
|
||||||
|
|
||||||
|
set(sessionId: string, messages: any[]): void {
|
||||||
|
if (!sessionId) return
|
||||||
|
const trimmed = messages.length > this.sessionLimit
|
||||||
|
? messages.slice(-this.sessionLimit)
|
||||||
|
: messages.slice()
|
||||||
|
this.cache[sessionId] = {
|
||||||
|
updatedAt: Date.now(),
|
||||||
|
messages: trimmed
|
||||||
|
}
|
||||||
|
this.persist()
|
||||||
|
}
|
||||||
|
|
||||||
|
private persist() {
|
||||||
|
try {
|
||||||
|
writeFileSync(this.cacheFilePath, JSON.stringify(this.cache), 'utf8')
|
||||||
|
} catch (error) {
|
||||||
|
console.error('MessageCacheService: 保存缓存失败', error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
clear(): void {
|
||||||
|
this.cache = {}
|
||||||
|
try {
|
||||||
|
rmSync(this.cacheFilePath, { force: true })
|
||||||
|
} catch (error) {
|
||||||
|
console.error('MessageCacheService: 清理缓存失败', error)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
251
electron/services/snsService.ts
Normal file
251
electron/services/snsService.ts
Normal file
@@ -0,0 +1,251 @@
|
|||||||
|
import { wcdbService } from './wcdbService'
|
||||||
|
import { ConfigService } from './config'
|
||||||
|
import { ContactCacheService } from './contactCacheService'
|
||||||
|
|
||||||
|
export interface SnsLivePhoto {
|
||||||
|
url: string
|
||||||
|
thumb: string
|
||||||
|
md5?: 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 SnsPost {
|
||||||
|
id: string
|
||||||
|
username: string
|
||||||
|
nickname: string
|
||||||
|
avatarUrl?: string
|
||||||
|
createTime: number
|
||||||
|
contentDesc: string
|
||||||
|
type?: number
|
||||||
|
media: SnsMedia[]
|
||||||
|
likes: string[]
|
||||||
|
comments: { id: string; nickname: string; content: string; refCommentId: string; refNickname?: string }[]
|
||||||
|
rawXml?: string // 原始 XML 数据
|
||||||
|
}
|
||||||
|
|
||||||
|
const fixSnsUrl = (url: string, token?: string) => {
|
||||||
|
if (!url) return url;
|
||||||
|
|
||||||
|
// 1. 统一使用 https
|
||||||
|
// 2. 将 /150 (缩略图) 强制改为 /0 (原图)
|
||||||
|
let fixedUrl = url.replace('http://', 'https://').replace(/\/150($|\?)/, '/0$1');
|
||||||
|
|
||||||
|
if (!token || fixedUrl.includes('token=')) return fixedUrl;
|
||||||
|
|
||||||
|
const connector = fixedUrl.includes('?') ? '&' : '?';
|
||||||
|
return `${fixedUrl}${connector}token=${token}&idx=1`;
|
||||||
|
};
|
||||||
|
|
||||||
|
class SnsService {
|
||||||
|
private contactCache: ContactCacheService
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
const config = new ConfigService()
|
||||||
|
this.contactCache = new ContactCacheService(config.get('cachePath') as 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)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if (result.success && result.timeline) {
|
||||||
|
const enrichedTimeline = result.timeline.map((post: any, index: number) => {
|
||||||
|
const contact = this.contactCache.get(post.username)
|
||||||
|
|
||||||
|
// 修复媒体 URL
|
||||||
|
const fixedMedia = post.media.map((m: any, mIdx: number) => {
|
||||||
|
const base = {
|
||||||
|
url: fixSnsUrl(m.url, m.token),
|
||||||
|
thumb: fixSnsUrl(m.thumb, m.token),
|
||||||
|
md5: m.md5,
|
||||||
|
token: m.token,
|
||||||
|
key: m.key,
|
||||||
|
encIdx: m.encIdx || m.enc_idx, // 兼容不同命名
|
||||||
|
livePhoto: m.livePhoto ? {
|
||||||
|
...m.livePhoto,
|
||||||
|
url: fixSnsUrl(m.livePhoto.url, m.livePhoto.token),
|
||||||
|
thumb: fixSnsUrl(m.livePhoto.thumb, m.livePhoto.token),
|
||||||
|
token: m.livePhoto.token,
|
||||||
|
key: m.livePhoto.key
|
||||||
|
} : undefined
|
||||||
|
}
|
||||||
|
|
||||||
|
// [MOCK] 模拟数据:如果后端没返回 key (说明 DLL 未更新),注入一些 Mock 数据以便前端开发
|
||||||
|
if (!base.key) {
|
||||||
|
base.key = 'mock_key_for_dev'
|
||||||
|
if (!base.token) {
|
||||||
|
base.token = 'mock_token_for_dev'
|
||||||
|
base.url = fixSnsUrl(base.url, base.token)
|
||||||
|
base.thumb = fixSnsUrl(base.thumb, base.token)
|
||||||
|
}
|
||||||
|
base.encIdx = '1'
|
||||||
|
|
||||||
|
// 强制给第一个帖子的第一张图加 LivePhoto 模拟
|
||||||
|
if (index === 0 && mIdx === 0 && !base.livePhoto) {
|
||||||
|
base.livePhoto = {
|
||||||
|
url: fixSnsUrl('https://tm.sh/d4cb0.mp4', 'mock_live_token'),
|
||||||
|
thumb: base.thumb,
|
||||||
|
token: 'mock_live_token',
|
||||||
|
key: 'mock_live_key'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return base
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
...post,
|
||||||
|
avatarUrl: contact?.avatarUrl,
|
||||||
|
nickname: post.nickname || contact?.displayName || post.username,
|
||||||
|
media: fixedMedia
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
return { ...result, timeline: enrichedTimeline }
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
async debugResource(url: string): Promise<{ success: boolean; status?: number; headers?: any; error?: string }> {
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
try {
|
||||||
|
const { app, net } = require('electron')
|
||||||
|
// Remove mocking 'require' if it causes issues, but here we need 'net' or 'https'
|
||||||
|
// implementing with 'https' for reliability if 'net' is main-process only special
|
||||||
|
const https = require('https')
|
||||||
|
const urlObj = new URL(url)
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
hostname: urlObj.hostname,
|
||||||
|
path: urlObj.pathname + urlObj.search,
|
||||||
|
method: 'GET',
|
||||||
|
headers: {
|
||||||
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x63090719) XWEB/8351",
|
||||||
|
"Accept": "image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8",
|
||||||
|
"Accept-Encoding": "gzip, deflate, br",
|
||||||
|
"Accept-Language": "zh-CN,zh;q=0.9",
|
||||||
|
"Referer": "https://servicewechat.com/",
|
||||||
|
"Connection": "keep-alive",
|
||||||
|
"Range": "bytes=0-10" // Keep our range check
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = https.request(options, (res: any) => {
|
||||||
|
resolve({
|
||||||
|
success: true,
|
||||||
|
status: res.statusCode,
|
||||||
|
headers: {
|
||||||
|
'x-enc': res.headers['x-enc'],
|
||||||
|
'content-length': res.headers['content-length'],
|
||||||
|
'content-type': res.headers['content-type']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
req.destroy() // We only need headers
|
||||||
|
})
|
||||||
|
|
||||||
|
req.on('error', (e: any) => {
|
||||||
|
resolve({ success: false, error: e.message })
|
||||||
|
})
|
||||||
|
|
||||||
|
req.end()
|
||||||
|
} catch (e: any) {
|
||||||
|
resolve({ success: false, error: e.message })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
private imageCache = new Map<string, string>()
|
||||||
|
|
||||||
|
async proxyImage(url: string): Promise<{ success: boolean; dataUrl?: string; error?: string }> {
|
||||||
|
// Check cache
|
||||||
|
if (this.imageCache.has(url)) {
|
||||||
|
return { success: true, dataUrl: this.imageCache.get(url) }
|
||||||
|
}
|
||||||
|
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
try {
|
||||||
|
const https = require('https')
|
||||||
|
const zlib = require('zlib')
|
||||||
|
const urlObj = new URL(url)
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
hostname: urlObj.hostname,
|
||||||
|
path: urlObj.pathname + urlObj.search,
|
||||||
|
method: 'GET',
|
||||||
|
headers: {
|
||||||
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x63090719) XWEB/8351",
|
||||||
|
"Accept": "image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8",
|
||||||
|
"Accept-Encoding": "gzip, deflate, br",
|
||||||
|
"Accept-Language": "zh-CN,zh;q=0.9",
|
||||||
|
"Referer": "https://servicewechat.com/",
|
||||||
|
"Connection": "keep-alive"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = https.request(options, (res: any) => {
|
||||||
|
if (res.statusCode !== 200) {
|
||||||
|
resolve({ success: false, error: `HTTP ${res.statusCode}` })
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const chunks: Buffer[] = []
|
||||||
|
let stream = res
|
||||||
|
|
||||||
|
// Handle gzip compression
|
||||||
|
const encoding = res.headers['content-encoding']
|
||||||
|
if (encoding === 'gzip') {
|
||||||
|
stream = res.pipe(zlib.createGunzip())
|
||||||
|
} else if (encoding === 'deflate') {
|
||||||
|
stream = res.pipe(zlib.createInflate())
|
||||||
|
} else if (encoding === 'br') {
|
||||||
|
stream = res.pipe(zlib.createBrotliDecompress())
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.on('data', (chunk: Buffer) => chunks.push(chunk))
|
||||||
|
stream.on('end', () => {
|
||||||
|
const buffer = Buffer.concat(chunks)
|
||||||
|
const contentType = res.headers['content-type'] || 'image/jpeg'
|
||||||
|
const base64 = buffer.toString('base64')
|
||||||
|
const dataUrl = `data:${contentType};base64,${base64}`
|
||||||
|
|
||||||
|
// Cache
|
||||||
|
this.imageCache.set(url, dataUrl)
|
||||||
|
|
||||||
|
resolve({ success: true, dataUrl })
|
||||||
|
})
|
||||||
|
stream.on('error', (e: any) => {
|
||||||
|
resolve({ success: false, error: e.message })
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
req.on('error', (e: any) => {
|
||||||
|
resolve({ success: false, error: e.message })
|
||||||
|
})
|
||||||
|
|
||||||
|
req.end()
|
||||||
|
} catch (e: any) {
|
||||||
|
resolve({ success: false, error: e.message })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const snsService = new SnsService()
|
||||||
279
electron/services/videoService.ts
Normal file
279
electron/services/videoService.ts
Normal file
@@ -0,0 +1,279 @@
|
|||||||
|
import { join } from 'path'
|
||||||
|
import { existsSync, readdirSync, statSync, readFileSync } from 'fs'
|
||||||
|
import { ConfigService } from './config'
|
||||||
|
import Database from 'better-sqlite3'
|
||||||
|
import { wcdbService } from './wcdbService'
|
||||||
|
|
||||||
|
export interface VideoInfo {
|
||||||
|
videoUrl?: string // 视频文件路径(用于 readFile)
|
||||||
|
coverUrl?: string // 封面 data URL
|
||||||
|
thumbUrl?: string // 缩略图 data URL
|
||||||
|
exists: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
class VideoService {
|
||||||
|
private configService: ConfigService
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
this.configService = new ConfigService()
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取数据库根目录
|
||||||
|
*/
|
||||||
|
private getDbPath(): string {
|
||||||
|
return this.configService.get('dbPath') || ''
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取当前用户的wxid
|
||||||
|
*/
|
||||||
|
private getMyWxid(): string {
|
||||||
|
return this.configService.get('myWxid') || ''
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取缓存目录(解密后的数据库存放位置)
|
||||||
|
*/
|
||||||
|
private getCachePath(): string {
|
||||||
|
return this.configService.get('cachePath') || ''
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 清理 wxid 目录名(去掉后缀)
|
||||||
|
*/
|
||||||
|
private cleanWxid(wxid: string): string {
|
||||||
|
const trimmed = wxid.trim()
|
||||||
|
if (!trimmed) return trimmed
|
||||||
|
|
||||||
|
if (trimmed.toLowerCase().startsWith('wxid_')) {
|
||||||
|
const match = trimmed.match(/^(wxid_[^_]+)/i)
|
||||||
|
if (match) return match[1]
|
||||||
|
return trimmed
|
||||||
|
}
|
||||||
|
|
||||||
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
|
||||||
|
if (suffixMatch) return suffixMatch[1]
|
||||||
|
|
||||||
|
return trimmed
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 从 video_hardlink_info_v4 表查询视频文件名
|
||||||
|
* 优先使用 cachePath 中解密后的 hardlink.db(使用 better-sqlite3)
|
||||||
|
* 如果失败,则尝试使用 wcdbService.execQuery 查询加密的 hardlink.db
|
||||||
|
*/
|
||||||
|
private async queryVideoFileName(md5: string): Promise<string | undefined> {
|
||||||
|
const cachePath = this.getCachePath()
|
||||||
|
const dbPath = this.getDbPath()
|
||||||
|
const wxid = this.getMyWxid()
|
||||||
|
const cleanedWxid = this.cleanWxid(wxid)
|
||||||
|
|
||||||
|
if (!wxid) return undefined
|
||||||
|
|
||||||
|
// 方法1:优先在 cachePath 下查找解密后的 hardlink.db
|
||||||
|
if (cachePath) {
|
||||||
|
const cacheDbPaths = [
|
||||||
|
join(cachePath, cleanedWxid, 'hardlink.db'),
|
||||||
|
join(cachePath, wxid, 'hardlink.db'),
|
||||||
|
join(cachePath, 'hardlink.db'),
|
||||||
|
join(cachePath, 'databases', cleanedWxid, 'hardlink.db'),
|
||||||
|
join(cachePath, 'databases', wxid, 'hardlink.db')
|
||||||
|
]
|
||||||
|
|
||||||
|
for (const p of cacheDbPaths) {
|
||||||
|
if (existsSync(p)) {
|
||||||
|
try {
|
||||||
|
const db = new Database(p, { readonly: true })
|
||||||
|
const row = db.prepare(`
|
||||||
|
SELECT file_name, md5 FROM video_hardlink_info_v4
|
||||||
|
WHERE md5 = ?
|
||||||
|
LIMIT 1
|
||||||
|
`).get(md5) as { file_name: string; md5: string } | undefined
|
||||||
|
db.close()
|
||||||
|
|
||||||
|
if (row?.file_name) {
|
||||||
|
const realMd5 = row.file_name.replace(/\.[^.]+$/, '')
|
||||||
|
return realMd5
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
// 忽略错误
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 方法2:使用 wcdbService.execQuery 查询加密的 hardlink.db
|
||||||
|
if (dbPath) {
|
||||||
|
// 检查 dbPath 是否已经包含 wxid
|
||||||
|
const dbPathLower = dbPath.toLowerCase()
|
||||||
|
const wxidLower = wxid.toLowerCase()
|
||||||
|
const cleanedWxidLower = cleanedWxid.toLowerCase()
|
||||||
|
const dbPathContainsWxid = dbPathLower.includes(wxidLower) || dbPathLower.includes(cleanedWxidLower)
|
||||||
|
|
||||||
|
const encryptedDbPaths: string[] = []
|
||||||
|
if (dbPathContainsWxid) {
|
||||||
|
// dbPath 已包含 wxid,不需要再拼接
|
||||||
|
encryptedDbPaths.push(join(dbPath, 'db_storage', 'hardlink', 'hardlink.db'))
|
||||||
|
} else {
|
||||||
|
// dbPath 不包含 wxid,需要拼接
|
||||||
|
encryptedDbPaths.push(join(dbPath, wxid, 'db_storage', 'hardlink', 'hardlink.db'))
|
||||||
|
encryptedDbPaths.push(join(dbPath, cleanedWxid, 'db_storage', 'hardlink', 'hardlink.db'))
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const p of encryptedDbPaths) {
|
||||||
|
if (existsSync(p)) {
|
||||||
|
try {
|
||||||
|
const escapedMd5 = md5.replace(/'/g, "''")
|
||||||
|
|
||||||
|
// 用 md5 字段查询,获取 file_name
|
||||||
|
const sql = `SELECT file_name FROM video_hardlink_info_v4 WHERE md5 = '${escapedMd5}' LIMIT 1`
|
||||||
|
|
||||||
|
const result = await wcdbService.execQuery('media', p, sql)
|
||||||
|
|
||||||
|
if (result.success && result.rows && result.rows.length > 0) {
|
||||||
|
const row = result.rows[0]
|
||||||
|
if (row?.file_name) {
|
||||||
|
// 提取不带扩展名的文件名作为实际视频 MD5
|
||||||
|
const realMd5 = String(row.file_name).replace(/\.[^.]+$/, '')
|
||||||
|
return realMd5
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
// 忽略错误
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return undefined
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 将文件转换为 data URL
|
||||||
|
*/
|
||||||
|
private fileToDataUrl(filePath: string, mimeType: string): string | undefined {
|
||||||
|
try {
|
||||||
|
if (!existsSync(filePath)) return undefined
|
||||||
|
const buffer = readFileSync(filePath)
|
||||||
|
return `data:${mimeType};base64,${buffer.toString('base64')}`
|
||||||
|
} catch {
|
||||||
|
return undefined
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据视频MD5获取视频文件信息
|
||||||
|
* 视频存放在: {数据库根目录}/{用户wxid}/msg/video/{年月}/
|
||||||
|
* 文件命名: {md5}.mp4, {md5}.jpg, {md5}_thumb.jpg
|
||||||
|
*/
|
||||||
|
async getVideoInfo(videoMd5: string): Promise<VideoInfo> {
|
||||||
|
const dbPath = this.getDbPath()
|
||||||
|
const wxid = this.getMyWxid()
|
||||||
|
|
||||||
|
if (!dbPath || !wxid || !videoMd5) {
|
||||||
|
return { exists: false }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 先尝试从数据库查询真正的视频文件名
|
||||||
|
const realVideoMd5 = await this.queryVideoFileName(videoMd5) || videoMd5
|
||||||
|
|
||||||
|
// 检查 dbPath 是否已经包含 wxid,避免重复拼接
|
||||||
|
const dbPathLower = dbPath.toLowerCase()
|
||||||
|
const wxidLower = wxid.toLowerCase()
|
||||||
|
const cleanedWxid = this.cleanWxid(wxid)
|
||||||
|
|
||||||
|
let videoBaseDir: string
|
||||||
|
if (dbPathLower.includes(wxidLower) || dbPathLower.includes(cleanedWxid.toLowerCase())) {
|
||||||
|
// dbPath 已经包含 wxid,直接使用
|
||||||
|
videoBaseDir = join(dbPath, 'msg', 'video')
|
||||||
|
} else {
|
||||||
|
// dbPath 不包含 wxid,需要拼接
|
||||||
|
videoBaseDir = join(dbPath, wxid, 'msg', 'video')
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!existsSync(videoBaseDir)) {
|
||||||
|
return { exists: false }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 遍历年月目录查找视频文件
|
||||||
|
try {
|
||||||
|
const allDirs = readdirSync(videoBaseDir)
|
||||||
|
|
||||||
|
// 支持多种目录格式: YYYY-MM, YYYYMM, 或其他
|
||||||
|
const yearMonthDirs = allDirs
|
||||||
|
.filter(dir => {
|
||||||
|
const dirPath = join(videoBaseDir, dir)
|
||||||
|
return statSync(dirPath).isDirectory()
|
||||||
|
})
|
||||||
|
.sort((a, b) => b.localeCompare(a)) // 从最新的目录开始查找
|
||||||
|
|
||||||
|
for (const yearMonth of yearMonthDirs) {
|
||||||
|
const dirPath = join(videoBaseDir, yearMonth)
|
||||||
|
|
||||||
|
const videoPath = join(dirPath, `${realVideoMd5}.mp4`)
|
||||||
|
const coverPath = join(dirPath, `${realVideoMd5}.jpg`)
|
||||||
|
const thumbPath = join(dirPath, `${realVideoMd5}_thumb.jpg`)
|
||||||
|
|
||||||
|
// 检查视频文件是否存在
|
||||||
|
if (existsSync(videoPath)) {
|
||||||
|
return {
|
||||||
|
videoUrl: videoPath, // 返回文件路径,前端通过 readFile 读取
|
||||||
|
coverUrl: this.fileToDataUrl(coverPath, 'image/jpeg'),
|
||||||
|
thumbUrl: this.fileToDataUrl(thumbPath, 'image/jpeg'),
|
||||||
|
exists: true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
// 忽略错误
|
||||||
|
}
|
||||||
|
|
||||||
|
return { exists: false }
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据消息内容解析视频MD5
|
||||||
|
*/
|
||||||
|
parseVideoMd5(content: string): string | undefined {
|
||||||
|
|
||||||
|
// 打印前500字符看看 XML 结构
|
||||||
|
|
||||||
|
if (!content) return undefined
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 提取所有可能的 md5 值进行日志
|
||||||
|
const allMd5s: string[] = []
|
||||||
|
const md5Regex = /(?:md5|rawmd5|newmd5|originsourcemd5)\s*=\s*['"]([a-fA-F0-9]+)['"]/gi
|
||||||
|
let match
|
||||||
|
while ((match = md5Regex.exec(content)) !== null) {
|
||||||
|
allMd5s.push(`${match[0]}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 提取 md5(用于查询 hardlink.db)
|
||||||
|
// 注意:不是 rawmd5,rawmd5 是另一个值
|
||||||
|
// 格式: md5="xxx" 或 <md5>xxx</md5>
|
||||||
|
|
||||||
|
// 尝试从videomsg标签中提取md5
|
||||||
|
const videoMsgMatch = /<videomsg[^>]*\smd5\s*=\s*['"]([a-fA-F0-9]+)['"]/i.exec(content)
|
||||||
|
if (videoMsgMatch) {
|
||||||
|
return videoMsgMatch[1].toLowerCase()
|
||||||
|
}
|
||||||
|
|
||||||
|
const attrMatch = /\smd5\s*=\s*['"]([a-fA-F0-9]+)['"]/i.exec(content)
|
||||||
|
if (attrMatch) {
|
||||||
|
return attrMatch[1].toLowerCase()
|
||||||
|
}
|
||||||
|
|
||||||
|
const md5Match = /<md5>([a-fA-F0-9]+)<\/md5>/i.exec(content)
|
||||||
|
if (md5Match) {
|
||||||
|
return md5Match[1].toLowerCase()
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('[VideoService] 解析视频MD5失败:', e)
|
||||||
|
}
|
||||||
|
|
||||||
|
return undefined
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const videoService = new VideoService()
|
||||||
476
electron/services/voiceTranscribeService.ts
Normal file
476
electron/services/voiceTranscribeService.ts
Normal file
@@ -0,0 +1,476 @@
|
|||||||
|
import { app } from 'electron'
|
||||||
|
import { existsSync, mkdirSync, statSync, unlinkSync, createWriteStream, openSync, writeSync, closeSync } from 'fs'
|
||||||
|
import { join } from 'path'
|
||||||
|
import * as https from 'https'
|
||||||
|
import * as http from 'http'
|
||||||
|
import { ConfigService } from './config'
|
||||||
|
|
||||||
|
// Sherpa-onnx 类型定义
|
||||||
|
type OfflineRecognizer = any
|
||||||
|
type OfflineStream = any
|
||||||
|
|
||||||
|
type ModelInfo = {
|
||||||
|
name: string
|
||||||
|
files: {
|
||||||
|
model: string
|
||||||
|
tokens: string
|
||||||
|
}
|
||||||
|
sizeBytes: number
|
||||||
|
sizeLabel: string
|
||||||
|
}
|
||||||
|
|
||||||
|
type DownloadProgress = {
|
||||||
|
modelName: string
|
||||||
|
downloadedBytes: number
|
||||||
|
totalBytes?: number
|
||||||
|
percent?: number
|
||||||
|
speed?: number
|
||||||
|
}
|
||||||
|
|
||||||
|
const SENSEVOICE_MODEL: ModelInfo = {
|
||||||
|
name: 'SenseVoiceSmall',
|
||||||
|
files: {
|
||||||
|
model: 'model.int8.onnx',
|
||||||
|
tokens: 'tokens.txt'
|
||||||
|
},
|
||||||
|
sizeBytes: 245_000_000,
|
||||||
|
sizeLabel: '245 MB'
|
||||||
|
}
|
||||||
|
|
||||||
|
const MODEL_DOWNLOAD_URLS = {
|
||||||
|
model: 'https://modelscope.cn/models/pengzhendong/sherpa-onnx-sense-voice-zh-en-ja-ko-yue/resolve/master/model.int8.onnx',
|
||||||
|
tokens: 'https://modelscope.cn/models/pengzhendong/sherpa-onnx-sense-voice-zh-en-ja-ko-yue/resolve/master/tokens.txt'
|
||||||
|
}
|
||||||
|
|
||||||
|
export class VoiceTranscribeService {
|
||||||
|
private configService = new ConfigService()
|
||||||
|
private downloadTasks = new Map<string, Promise<{ success: boolean; path?: string; error?: string }>>()
|
||||||
|
private recognizer: OfflineRecognizer | null = null
|
||||||
|
private isInitializing = false
|
||||||
|
|
||||||
|
private resolveModelDir(): string {
|
||||||
|
const configured = this.configService.get('whisperModelDir') as string | undefined
|
||||||
|
if (configured) return configured
|
||||||
|
return join(app.getPath('documents'), 'WeFlow', 'models', 'sensevoice')
|
||||||
|
}
|
||||||
|
|
||||||
|
private resolveModelPath(fileName: string): string {
|
||||||
|
return join(this.resolveModelDir(), fileName)
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查模型状态
|
||||||
|
*/
|
||||||
|
async getModelStatus(): Promise<{
|
||||||
|
success: boolean
|
||||||
|
exists?: boolean
|
||||||
|
modelPath?: string
|
||||||
|
tokensPath?: string
|
||||||
|
sizeBytes?: number
|
||||||
|
error?: string
|
||||||
|
}> {
|
||||||
|
try {
|
||||||
|
const modelPath = this.resolveModelPath(SENSEVOICE_MODEL.files.model)
|
||||||
|
const tokensPath = this.resolveModelPath(SENSEVOICE_MODEL.files.tokens)
|
||||||
|
const modelExists = existsSync(modelPath)
|
||||||
|
const tokensExists = existsSync(tokensPath)
|
||||||
|
const exists = modelExists && tokensExists
|
||||||
|
|
||||||
|
if (!exists) {
|
||||||
|
return { success: true, exists: false, modelPath, tokensPath }
|
||||||
|
}
|
||||||
|
|
||||||
|
const modelSize = statSync(modelPath).size
|
||||||
|
const tokensSize = statSync(tokensPath).size
|
||||||
|
const totalSize = modelSize + tokensSize
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
exists: true,
|
||||||
|
modelPath,
|
||||||
|
tokensPath,
|
||||||
|
sizeBytes: totalSize
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return { success: false, error: String(error) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载模型文件
|
||||||
|
*/
|
||||||
|
async downloadModel(
|
||||||
|
onProgress?: (progress: DownloadProgress) => void
|
||||||
|
): Promise<{ success: boolean; modelPath?: string; tokensPath?: string; error?: string }> {
|
||||||
|
const cacheKey = 'sensevoice'
|
||||||
|
const pending = this.downloadTasks.get(cacheKey)
|
||||||
|
if (pending) return pending
|
||||||
|
|
||||||
|
const task = (async () => {
|
||||||
|
try {
|
||||||
|
const modelDir = this.resolveModelDir()
|
||||||
|
if (!existsSync(modelDir)) {
|
||||||
|
mkdirSync(modelDir, { recursive: true })
|
||||||
|
}
|
||||||
|
|
||||||
|
const modelPath = this.resolveModelPath(SENSEVOICE_MODEL.files.model)
|
||||||
|
const tokensPath = this.resolveModelPath(SENSEVOICE_MODEL.files.tokens)
|
||||||
|
|
||||||
|
// 初始进度
|
||||||
|
onProgress?.({
|
||||||
|
modelName: SENSEVOICE_MODEL.name,
|
||||||
|
downloadedBytes: 0,
|
||||||
|
totalBytes: SENSEVOICE_MODEL.sizeBytes,
|
||||||
|
percent: 0
|
||||||
|
})
|
||||||
|
|
||||||
|
// 下载模型文件 (80% 权重)
|
||||||
|
console.info('[VoiceTranscribe] 开始下载模型文件...')
|
||||||
|
await this.downloadToFile(
|
||||||
|
MODEL_DOWNLOAD_URLS.model,
|
||||||
|
modelPath,
|
||||||
|
'model',
|
||||||
|
(downloaded, total, speed) => {
|
||||||
|
const percent = total ? (downloaded / total) * 80 : 0
|
||||||
|
onProgress?.({
|
||||||
|
modelName: SENSEVOICE_MODEL.name,
|
||||||
|
downloadedBytes: downloaded,
|
||||||
|
totalBytes: SENSEVOICE_MODEL.sizeBytes,
|
||||||
|
percent,
|
||||||
|
speed
|
||||||
|
})
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// 下载 tokens 文件 (20% 权重)
|
||||||
|
console.info('[VoiceTranscribe] 开始下载 tokens 文件...')
|
||||||
|
await this.downloadToFile(
|
||||||
|
MODEL_DOWNLOAD_URLS.tokens,
|
||||||
|
tokensPath,
|
||||||
|
'tokens',
|
||||||
|
(downloaded, total, speed) => {
|
||||||
|
const modelSize = existsSync(modelPath) ? statSync(modelPath).size : 0
|
||||||
|
const percent = total ? 80 + (downloaded / total) * 20 : 80
|
||||||
|
onProgress?.({
|
||||||
|
modelName: SENSEVOICE_MODEL.name,
|
||||||
|
downloadedBytes: modelSize + downloaded,
|
||||||
|
totalBytes: SENSEVOICE_MODEL.sizeBytes,
|
||||||
|
percent,
|
||||||
|
speed
|
||||||
|
})
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
console.info('[VoiceTranscribe] 模型下载完成')
|
||||||
|
return { success: true, modelPath, tokensPath }
|
||||||
|
} catch (error) {
|
||||||
|
const modelPath = this.resolveModelPath(SENSEVOICE_MODEL.files.model)
|
||||||
|
const tokensPath = this.resolveModelPath(SENSEVOICE_MODEL.files.tokens)
|
||||||
|
try {
|
||||||
|
if (existsSync(modelPath)) unlinkSync(modelPath)
|
||||||
|
if (existsSync(tokensPath)) unlinkSync(tokensPath)
|
||||||
|
} catch { }
|
||||||
|
return { success: false, error: String(error) }
|
||||||
|
} finally {
|
||||||
|
this.downloadTasks.delete(cacheKey)
|
||||||
|
}
|
||||||
|
})()
|
||||||
|
|
||||||
|
this.downloadTasks.set(cacheKey, task)
|
||||||
|
return task
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 转写 WAV 音频数据
|
||||||
|
*/
|
||||||
|
async transcribeWavBuffer(
|
||||||
|
wavData: Buffer,
|
||||||
|
onPartial?: (text: string) => void,
|
||||||
|
languages?: string[]
|
||||||
|
): Promise<{ success: boolean; transcript?: string; error?: string }> {
|
||||||
|
return new Promise((resolve) => {
|
||||||
|
try {
|
||||||
|
const modelPath = this.resolveModelPath(SENSEVOICE_MODEL.files.model)
|
||||||
|
const tokensPath = this.resolveModelPath(SENSEVOICE_MODEL.files.tokens)
|
||||||
|
|
||||||
|
if (!existsSync(modelPath) || !existsSync(tokensPath)) {
|
||||||
|
resolve({ success: false, error: '模型文件不存在,请先下载模型' })
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let supportedLanguages = languages
|
||||||
|
if (!supportedLanguages || supportedLanguages.length === 0) {
|
||||||
|
supportedLanguages = this.configService.get('transcribeLanguages')
|
||||||
|
if (!supportedLanguages || supportedLanguages.length === 0) {
|
||||||
|
supportedLanguages = ['zh', 'yue']
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const { Worker } = require('worker_threads')
|
||||||
|
const workerPath = join(__dirname, 'transcribeWorker.js')
|
||||||
|
|
||||||
|
const worker = new Worker(workerPath, {
|
||||||
|
workerData: {
|
||||||
|
modelPath,
|
||||||
|
tokensPath,
|
||||||
|
wavData,
|
||||||
|
sampleRate: 16000,
|
||||||
|
languages: supportedLanguages
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
let finalTranscript = ''
|
||||||
|
|
||||||
|
worker.on('message', (msg: any) => {
|
||||||
|
if (msg.type === 'partial') {
|
||||||
|
onPartial?.(msg.text)
|
||||||
|
} else if (msg.type === 'final') {
|
||||||
|
finalTranscript = msg.text
|
||||||
|
resolve({ success: true, transcript: finalTranscript })
|
||||||
|
worker.terminate()
|
||||||
|
} else if (msg.type === 'error') {
|
||||||
|
console.error('[VoiceTranscribe] Worker 错误:', msg.error)
|
||||||
|
resolve({ success: false, error: msg.error })
|
||||||
|
worker.terminate()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
worker.on('error', (err: Error) => resolve({ success: false, error: String(err) }))
|
||||||
|
worker.on('exit', (code: number) => {
|
||||||
|
if (code !== 0) resolve({ success: false, error: `Worker exited with code ${code}` })
|
||||||
|
})
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
resolve({ success: false, error: String(error) })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载文件 (支持多线程)
|
||||||
|
*/
|
||||||
|
private async downloadToFile(
|
||||||
|
url: string,
|
||||||
|
targetPath: string,
|
||||||
|
fileName: string,
|
||||||
|
onProgress?: (downloaded: number, total?: number, speed?: number) => void
|
||||||
|
): Promise<void> {
|
||||||
|
if (existsSync(targetPath)) {
|
||||||
|
unlinkSync(targetPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
console.info(`[VoiceTranscribe] 准备下载 ${fileName}: ${url}`)
|
||||||
|
|
||||||
|
// 1. 探测支持情况
|
||||||
|
let probeResult
|
||||||
|
try {
|
||||||
|
probeResult = await this.probeUrl(url)
|
||||||
|
} catch (err) {
|
||||||
|
console.warn(`[VoiceTranscribe] ${fileName} 探测失败,使用单线程`, err)
|
||||||
|
return this.downloadSingleThread(url, targetPath, fileName, onProgress)
|
||||||
|
}
|
||||||
|
|
||||||
|
const { totalSize, acceptRanges, finalUrl } = probeResult
|
||||||
|
|
||||||
|
// 如果文件太小 (< 2MB) 或者不支持 Range,使用单线程
|
||||||
|
if (totalSize < 2 * 1024 * 1024 || !acceptRanges) {
|
||||||
|
return this.downloadSingleThread(finalUrl, targetPath, fileName, onProgress)
|
||||||
|
}
|
||||||
|
|
||||||
|
console.info(`[VoiceTranscribe] ${fileName} 开始多线程下载 (4 线程), 大小: ${(totalSize / 1024 / 1024).toFixed(2)} MB`)
|
||||||
|
|
||||||
|
const threadCount = 4
|
||||||
|
const chunkSize = Math.ceil(totalSize / threadCount)
|
||||||
|
const fd = openSync(targetPath, 'w')
|
||||||
|
|
||||||
|
let downloadedTotal = 0
|
||||||
|
let lastDownloaded = 0
|
||||||
|
let lastTime = Date.now()
|
||||||
|
let speed = 0
|
||||||
|
|
||||||
|
const speedInterval = setInterval(() => {
|
||||||
|
const now = Date.now()
|
||||||
|
const duration = (now - lastTime) / 1000
|
||||||
|
if (duration > 0) {
|
||||||
|
speed = (downloadedTotal - lastDownloaded) / duration
|
||||||
|
lastDownloaded = downloadedTotal
|
||||||
|
lastTime = now
|
||||||
|
onProgress?.(downloadedTotal, 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) => {
|
||||||
|
downloadedTotal += bytes
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
|
await Promise.all(promises)
|
||||||
|
// Final progress update
|
||||||
|
onProgress?.(totalSize, totalSize, 0)
|
||||||
|
console.info(`[VoiceTranscribe] ${fileName} 多线程下载完成`)
|
||||||
|
} catch (err) {
|
||||||
|
console.error(`[VoiceTranscribe] ${fileName} 多线程下载失败:`, err)
|
||||||
|
throw err
|
||||||
|
} finally {
|
||||||
|
clearInterval(speedInterval)
|
||||||
|
closeSync(fd)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private async probeUrl(url: string, remainingRedirects = 5): Promise<{ totalSize: number, acceptRanges: boolean, finalUrl: string }> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const protocol = url.startsWith('https') ? https : 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://modelscope.cn/',
|
||||||
|
'Range': 'bytes=0-0'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = protocol.get(url, options, (res) => {
|
||||||
|
if ([301, 302, 303, 307, 308].includes(res.statusCode || 0)) {
|
||||||
|
const location = res.headers.location
|
||||||
|
if (location && remainingRedirects > 0) {
|
||||||
|
const nextUrl = new URL(location, url).href
|
||||||
|
this.probeUrl(nextUrl, remainingRedirects - 1).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> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const protocol = url.startsWith('https') ? https : 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://modelscope.cn/',
|
||||||
|
'Range': `bytes=${start}-${end}`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = protocol.get(url, options, (res) => {
|
||||||
|
if (res.statusCode !== 206) {
|
||||||
|
reject(new Error(`Chunk download failed: HTTP ${res.statusCode}`))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let currentOffset = start
|
||||||
|
res.on('data', (chunk: Buffer) => {
|
||||||
|
try {
|
||||||
|
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, targetPath: string, fileName: string, onProgress?: (downloaded: number, total?: number, speed?: number) => void, remainingRedirects = 5): Promise<void> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const protocol = url.startsWith('https') ? https : 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://modelscope.cn/'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const request = protocol.get(url, options, (response) => {
|
||||||
|
if ([301, 302, 303, 307, 308].includes(response.statusCode || 0)) {
|
||||||
|
const location = response.headers.location
|
||||||
|
if (location && remainingRedirects > 0) {
|
||||||
|
const nextUrl = new URL(location, url).href
|
||||||
|
this.downloadSingleThread(nextUrl, targetPath, fileName, onProgress, remainingRedirects - 1).then(resolve).catch(reject)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (response.statusCode !== 200) {
|
||||||
|
reject(new Error(`Fallback download failed: HTTP ${response.statusCode}`))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const totalBytes = Number(response.headers['content-length'] || 0) || undefined
|
||||||
|
let downloadedBytes = 0
|
||||||
|
let lastDownloaded = 0
|
||||||
|
let lastTime = Date.now()
|
||||||
|
let speed = 0
|
||||||
|
|
||||||
|
const speedInterval = setInterval(() => {
|
||||||
|
const now = Date.now()
|
||||||
|
const duration = (now - lastTime) / 1000
|
||||||
|
if (duration > 0) {
|
||||||
|
speed = (downloadedBytes - lastDownloaded) / duration
|
||||||
|
lastDownloaded = downloadedBytes
|
||||||
|
lastTime = now
|
||||||
|
onProgress?.(downloadedBytes, totalBytes, speed)
|
||||||
|
}
|
||||||
|
}, 1000)
|
||||||
|
|
||||||
|
const writer = createWriteStream(targetPath)
|
||||||
|
response.on('data', (chunk) => {
|
||||||
|
downloadedBytes += chunk.length
|
||||||
|
})
|
||||||
|
|
||||||
|
writer.on('finish', () => {
|
||||||
|
clearInterval(speedInterval)
|
||||||
|
writer.close()
|
||||||
|
resolve()
|
||||||
|
})
|
||||||
|
|
||||||
|
writer.on('error', (err) => {
|
||||||
|
clearInterval(speedInterval)
|
||||||
|
reject(err)
|
||||||
|
})
|
||||||
|
response.pipe(writer)
|
||||||
|
})
|
||||||
|
request.on('error', reject)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
dispose() {
|
||||||
|
if (this.recognizer) {
|
||||||
|
this.recognizer = null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const voiceTranscribeService = new VoiceTranscribeService()
|
||||||
1713
electron/services/wcdbCore.ts
Normal file
1713
electron/services/wcdbCore.ts
Normal file
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
32
electron/services/windowsHelloService.ts
Normal file
32
electron/services/windowsHelloService.ts
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
import { wcdbService } from './wcdbService'
|
||||||
|
import { BrowserWindow } from 'electron'
|
||||||
|
|
||||||
|
export class WindowsHelloService {
|
||||||
|
private verificationPromise: Promise<{ success: boolean; error?: string }> | null = null
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 验证 Windows Hello
|
||||||
|
* @param message 提示信息
|
||||||
|
*/
|
||||||
|
async verify(message: string = '请验证您的身份以解锁 WeFlow', targetWindow?: BrowserWindow): Promise<{ success: boolean; error?: string }> {
|
||||||
|
// Prevent concurrent verification requests
|
||||||
|
if (this.verificationPromise) {
|
||||||
|
return this.verificationPromise
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取窗口句柄: 优先使用传入的窗口,否则尝试获取焦点窗口,最后兜底主窗口
|
||||||
|
const window = targetWindow || BrowserWindow.getFocusedWindow() || BrowserWindow.getAllWindows()[0]
|
||||||
|
const hwndBuffer = window?.getNativeWindowHandle()
|
||||||
|
// Convert buffer to int string for transport
|
||||||
|
const hwndStr = hwndBuffer ? BigInt('0x' + hwndBuffer.toString('hex')).toString() : undefined
|
||||||
|
|
||||||
|
this.verificationPromise = wcdbService.verifyUser(message, hwndStr)
|
||||||
|
.finally(() => {
|
||||||
|
this.verificationPromise = null
|
||||||
|
})
|
||||||
|
|
||||||
|
return this.verificationPromise
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const windowsHelloService = new WindowsHelloService()
|
||||||
166
electron/transcribeWorker.ts
Normal file
166
electron/transcribeWorker.ts
Normal file
@@ -0,0 +1,166 @@
|
|||||||
|
import { parentPort, workerData } from 'worker_threads'
|
||||||
|
|
||||||
|
interface WorkerParams {
|
||||||
|
modelPath: string
|
||||||
|
tokensPath: string
|
||||||
|
wavData: Buffer
|
||||||
|
sampleRate: number
|
||||||
|
languages?: string[]
|
||||||
|
}
|
||||||
|
|
||||||
|
// 语言标记映射
|
||||||
|
const LANGUAGE_TAGS: Record<string, string> = {
|
||||||
|
'zh': '<|zh|>',
|
||||||
|
'en': '<|en|>',
|
||||||
|
'ja': '<|ja|>',
|
||||||
|
'ko': '<|ko|>',
|
||||||
|
'yue': '<|yue|>' // 粤语
|
||||||
|
}
|
||||||
|
|
||||||
|
// 技术标签(识别语言、语速、ITN等),需要从最终文本中移除
|
||||||
|
const TECH_TAGS = [
|
||||||
|
'<|zh|>', '<|en|>', '<|ja|>', '<|ko|>', '<|yue|>',
|
||||||
|
'<|nospeech|>', '<|speech|>',
|
||||||
|
'<|itn|>', '<|wo_itn|>',
|
||||||
|
'<|NORMAL|>'
|
||||||
|
]
|
||||||
|
|
||||||
|
// 情感与事件标签映射,转换为直观的 Emoji
|
||||||
|
const RICH_TAG_MAP: Record<string, string> = {
|
||||||
|
'<|HAPPY|>': '😊',
|
||||||
|
'<|SAD|>': '😔',
|
||||||
|
'<|ANGRY|>': '😠',
|
||||||
|
'<|NEUTRAL|>': '', // 中性情感不特别标记
|
||||||
|
'<|FEARFUL|>': '😨',
|
||||||
|
'<|DISGUSTED|>': '🤢',
|
||||||
|
'<|SURPRISED|>': '😮',
|
||||||
|
'<|BGM|>': '🎵',
|
||||||
|
'<|Applause|>': '👏',
|
||||||
|
'<|Laughter|>': '😂',
|
||||||
|
'<|Cry|>': '😭',
|
||||||
|
'<|Cough|>': ' (咳嗽) ',
|
||||||
|
'<|Sneeze|>': ' (喷嚏) ',
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 富文本后处理:移除技术标签,转换识别出的情感和声音事件
|
||||||
|
*/
|
||||||
|
function richTranscribePostProcess(text: string): string {
|
||||||
|
if (!text) return ''
|
||||||
|
|
||||||
|
let processed = text
|
||||||
|
|
||||||
|
// 1. 转换情感和事件标签
|
||||||
|
for (const [tag, replacement] of Object.entries(RICH_TAG_MAP)) {
|
||||||
|
// 使用正则全局替换,不区分大小写以防不同版本差异
|
||||||
|
const escapedTag = tag.replace(/[|<>]/g, '\\$&')
|
||||||
|
processed = processed.replace(new RegExp(escapedTag, 'gi'), replacement)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 移除所有剩余的技术标签
|
||||||
|
for (const tag of TECH_TAGS) {
|
||||||
|
const escapedTag = tag.replace(/[|<>]/g, '\\$&')
|
||||||
|
processed = processed.replace(new RegExp(escapedTag, 'gi'), '')
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 清理多余空格并返回
|
||||||
|
return processed.replace(/\s+/g, ' ').trim()
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查识别结果是否在允许的语言列表中
|
||||||
|
function isLanguageAllowed(result: any, allowedLanguages: string[]): boolean {
|
||||||
|
if (!result || !result.lang) {
|
||||||
|
// 如果没有语言信息,默认允许(或从文本开头尝试提取)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果没有指定语言或语言列表为空,默认允许中文和粤语
|
||||||
|
if (!allowedLanguages || allowedLanguages.length === 0) {
|
||||||
|
allowedLanguages = ['zh', 'yue']
|
||||||
|
}
|
||||||
|
|
||||||
|
const langTag = result.lang
|
||||||
|
|
||||||
|
|
||||||
|
// 检查是否在允许的语言列表中
|
||||||
|
for (const lang of allowedLanguages) {
|
||||||
|
if (LANGUAGE_TAGS[lang] === langTag) {
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
async function run() {
|
||||||
|
if (!parentPort) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 动态加载以捕获可能的加载错误(如 C++ 运行库缺失等)
|
||||||
|
let sherpa: any;
|
||||||
|
try {
|
||||||
|
sherpa = require('sherpa-onnx-node');
|
||||||
|
} catch (requireError) {
|
||||||
|
parentPort.postMessage({ type: 'error', error: 'Failed to load speech engine: ' + String(requireError) });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { modelPath, tokensPath, wavData: rawWavData, sampleRate, languages } = workerData as WorkerParams
|
||||||
|
const wavData = Buffer.from(rawWavData);
|
||||||
|
// 确保有有效的语言列表,默认只允许中文
|
||||||
|
let allowedLanguages = languages || ['zh']
|
||||||
|
if (allowedLanguages.length === 0) {
|
||||||
|
allowedLanguages = ['zh']
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// 1. 初始化识别器 (SenseVoiceSmall)
|
||||||
|
const recognizerConfig = {
|
||||||
|
modelConfig: {
|
||||||
|
senseVoice: {
|
||||||
|
model: modelPath,
|
||||||
|
useInverseTextNormalization: 1
|
||||||
|
},
|
||||||
|
tokens: tokensPath,
|
||||||
|
numThreads: 2,
|
||||||
|
debug: 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const recognizer = new sherpa.OfflineRecognizer(recognizerConfig)
|
||||||
|
|
||||||
|
// 2. 处理音频数据 (全量识别)
|
||||||
|
const pcmData = wavData.slice(44)
|
||||||
|
const samples = new Float32Array(pcmData.length / 2)
|
||||||
|
for (let i = 0; i < samples.length; i++) {
|
||||||
|
samples[i] = pcmData.readInt16LE(i * 2) / 32768.0
|
||||||
|
}
|
||||||
|
|
||||||
|
const stream = recognizer.createStream()
|
||||||
|
stream.acceptWaveform({ sampleRate, samples })
|
||||||
|
recognizer.decode(stream)
|
||||||
|
const result = recognizer.getResult(stream)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// 3. 检查语言是否在白名单中
|
||||||
|
if (isLanguageAllowed(result, allowedLanguages)) {
|
||||||
|
const processedText = richTranscribePostProcess(result.text)
|
||||||
|
|
||||||
|
parentPort.postMessage({ type: 'final', text: processedText })
|
||||||
|
} else {
|
||||||
|
|
||||||
|
parentPort.postMessage({ type: 'final', text: '' })
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
parentPort.postMessage({ type: 'error', error: String(error) })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
run();
|
||||||
|
|
||||||
4
electron/types/sherpa-onnx-node.d.ts
vendored
Normal file
4
electron/types/sherpa-onnx-node.d.ts
vendored
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
declare module 'sherpa-onnx-node' {
|
||||||
|
const content: any;
|
||||||
|
export = content;
|
||||||
|
}
|
||||||
22
electron/types/whisper-node.d.ts
vendored
Normal file
22
electron/types/whisper-node.d.ts
vendored
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
declare module 'whisper-node' {
|
||||||
|
export type WhisperSegment = {
|
||||||
|
start: string
|
||||||
|
end: string
|
||||||
|
speech: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export type WhisperOptions = {
|
||||||
|
modelName?: string
|
||||||
|
modelPath?: string
|
||||||
|
whisperOptions?: {
|
||||||
|
language?: string
|
||||||
|
gen_file_txt?: boolean
|
||||||
|
gen_file_subtitle?: boolean
|
||||||
|
gen_file_vtt?: boolean
|
||||||
|
word_timestamps?: boolean
|
||||||
|
timestamp_size?: number
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export default function whisper(filePath: string, options?: WhisperOptions): Promise<WhisperSegment[]>
|
||||||
|
}
|
||||||
156
electron/wcdbWorker.ts
Normal file
156
electron/wcdbWorker.ts
Normal file
@@ -0,0 +1,156 @@
|
|||||||
|
import { parentPort, workerData } from 'worker_threads'
|
||||||
|
import { WcdbCore } from './services/wcdbCore'
|
||||||
|
|
||||||
|
const core = new WcdbCore()
|
||||||
|
|
||||||
|
if (parentPort) {
|
||||||
|
parentPort.on('message', async (msg) => {
|
||||||
|
const { id, type, payload } = msg
|
||||||
|
|
||||||
|
try {
|
||||||
|
let result: any
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case 'setPaths':
|
||||||
|
core.setPaths(payload.resourcesPath, payload.userDataPath)
|
||||||
|
result = { success: true }
|
||||||
|
break
|
||||||
|
case 'setLogEnabled':
|
||||||
|
core.setLogEnabled(payload.enabled)
|
||||||
|
result = { success: true }
|
||||||
|
break
|
||||||
|
case 'setMonitor':
|
||||||
|
core.setMonitor((type, json) => {
|
||||||
|
parentPort!.postMessage({
|
||||||
|
id: -1,
|
||||||
|
type: 'monitor',
|
||||||
|
payload: { type, json }
|
||||||
|
})
|
||||||
|
})
|
||||||
|
result = { success: true }
|
||||||
|
break
|
||||||
|
case 'testConnection':
|
||||||
|
result = await core.testConnection(payload.dbPath, payload.hexKey, payload.wxid)
|
||||||
|
break
|
||||||
|
case 'open':
|
||||||
|
result = await core.open(payload.dbPath, payload.hexKey, payload.wxid)
|
||||||
|
break
|
||||||
|
case 'close':
|
||||||
|
core.close()
|
||||||
|
result = { success: true }
|
||||||
|
break
|
||||||
|
case 'isConnected':
|
||||||
|
result = core.isConnected()
|
||||||
|
break
|
||||||
|
case 'getSessions':
|
||||||
|
result = await core.getSessions()
|
||||||
|
break
|
||||||
|
case 'getMessages':
|
||||||
|
result = await core.getMessages(payload.sessionId, payload.limit, payload.offset)
|
||||||
|
break
|
||||||
|
case 'getNewMessages':
|
||||||
|
result = await core.getNewMessages(payload.sessionId, payload.minTime, payload.limit)
|
||||||
|
break
|
||||||
|
case 'getMessageCount':
|
||||||
|
result = await core.getMessageCount(payload.sessionId)
|
||||||
|
break
|
||||||
|
case 'getDisplayNames':
|
||||||
|
result = await core.getDisplayNames(payload.usernames)
|
||||||
|
break
|
||||||
|
case 'getAvatarUrls':
|
||||||
|
result = await core.getAvatarUrls(payload.usernames)
|
||||||
|
break
|
||||||
|
case 'getGroupMemberCount':
|
||||||
|
result = await core.getGroupMemberCount(payload.chatroomId)
|
||||||
|
break
|
||||||
|
case 'getGroupMemberCounts':
|
||||||
|
result = await core.getGroupMemberCounts(payload.chatroomIds)
|
||||||
|
break
|
||||||
|
case 'getGroupMembers':
|
||||||
|
result = await core.getGroupMembers(payload.chatroomId)
|
||||||
|
break
|
||||||
|
case 'getGroupNicknames':
|
||||||
|
result = await core.getGroupNicknames(payload.chatroomId)
|
||||||
|
break
|
||||||
|
case 'getMessageTables':
|
||||||
|
result = await core.getMessageTables(payload.sessionId)
|
||||||
|
break
|
||||||
|
case 'getMessageTableStats':
|
||||||
|
result = await core.getMessageTableStats(payload.sessionId)
|
||||||
|
break
|
||||||
|
case 'getMessageMeta':
|
||||||
|
result = await core.getMessageMeta(payload.dbPath, payload.tableName, payload.limit, payload.offset)
|
||||||
|
break
|
||||||
|
case 'getContact':
|
||||||
|
result = await core.getContact(payload.username)
|
||||||
|
break
|
||||||
|
case 'getAggregateStats':
|
||||||
|
result = await core.getAggregateStats(payload.sessionIds, payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'getAvailableYears':
|
||||||
|
result = await core.getAvailableYears(payload.sessionIds)
|
||||||
|
break
|
||||||
|
case 'getAnnualReportStats':
|
||||||
|
result = await core.getAnnualReportStats(payload.sessionIds, payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'getAnnualReportExtras':
|
||||||
|
result = await core.getAnnualReportExtras(payload.sessionIds, payload.beginTimestamp, payload.endTimestamp, payload.peakDayBegin, payload.peakDayEnd)
|
||||||
|
break
|
||||||
|
case 'getGroupStats':
|
||||||
|
result = await core.getGroupStats(payload.chatroomId, payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'openMessageCursor':
|
||||||
|
result = await core.openMessageCursor(payload.sessionId, payload.batchSize, payload.ascending, payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'openMessageCursorLite':
|
||||||
|
result = await core.openMessageCursorLite(payload.sessionId, payload.batchSize, payload.ascending, payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'fetchMessageBatch':
|
||||||
|
result = await core.fetchMessageBatch(payload.cursor)
|
||||||
|
break
|
||||||
|
case 'closeMessageCursor':
|
||||||
|
result = await core.closeMessageCursor(payload.cursor)
|
||||||
|
break
|
||||||
|
case 'execQuery':
|
||||||
|
result = await core.execQuery(payload.kind, payload.path, payload.sql)
|
||||||
|
break
|
||||||
|
case 'getEmoticonCdnUrl':
|
||||||
|
result = await core.getEmoticonCdnUrl(payload.dbPath, payload.md5)
|
||||||
|
break
|
||||||
|
case 'listMessageDbs':
|
||||||
|
result = await core.listMessageDbs()
|
||||||
|
break
|
||||||
|
case 'listMediaDbs':
|
||||||
|
result = await core.listMediaDbs()
|
||||||
|
break
|
||||||
|
case 'getMessageById':
|
||||||
|
result = await core.getMessageById(payload.sessionId, payload.localId)
|
||||||
|
break
|
||||||
|
case 'getVoiceData':
|
||||||
|
result = await core.getVoiceData(payload.sessionId, payload.createTime, payload.candidates, payload.localId, payload.svrId)
|
||||||
|
if (!result.success) {
|
||||||
|
console.error('[wcdbWorker] getVoiceData failed:', result.error)
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case 'getSnsTimeline':
|
||||||
|
result = await core.getSnsTimeline(payload.limit, payload.offset, payload.usernames, payload.keyword, payload.startTime, payload.endTime)
|
||||||
|
break
|
||||||
|
case 'getSnsAnnualStats':
|
||||||
|
result = await core.getSnsAnnualStats(payload.beginTimestamp, payload.endTimestamp)
|
||||||
|
break
|
||||||
|
case 'getLogs':
|
||||||
|
result = await core.getLogs()
|
||||||
|
break
|
||||||
|
case 'verifyUser':
|
||||||
|
result = await core.verifyUser(payload.message, payload.hwnd)
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
result = { success: false, error: `Unknown method: ${type}` }
|
||||||
|
}
|
||||||
|
|
||||||
|
parentPort!.postMessage({ id, result })
|
||||||
|
} catch (e) {
|
||||||
|
parentPort!.postMessage({ id, error: String(e) })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
200
electron/windows/notificationWindow.ts
Normal file
200
electron/windows/notificationWindow.ts
Normal file
@@ -0,0 +1,200 @@
|
|||||||
|
import { BrowserWindow, ipcMain, screen } from 'electron'
|
||||||
|
import { join } from 'path'
|
||||||
|
import { ConfigService } from '../services/config'
|
||||||
|
|
||||||
|
let notificationWindow: BrowserWindow | null = null
|
||||||
|
let closeTimer: NodeJS.Timeout | null = null
|
||||||
|
|
||||||
|
export function createNotificationWindow() {
|
||||||
|
if (notificationWindow && !notificationWindow.isDestroyed()) {
|
||||||
|
return notificationWindow
|
||||||
|
}
|
||||||
|
|
||||||
|
const isDev = !!process.env.VITE_DEV_SERVER_URL
|
||||||
|
const iconPath = isDev
|
||||||
|
? join(__dirname, '../../public/icon.ico')
|
||||||
|
: join(process.resourcesPath, 'icon.ico')
|
||||||
|
|
||||||
|
console.log('[NotificationWindow] Creating window...')
|
||||||
|
const width = 344
|
||||||
|
const height = 114
|
||||||
|
|
||||||
|
// Update default creation size
|
||||||
|
notificationWindow = new BrowserWindow({
|
||||||
|
width: width,
|
||||||
|
height: height,
|
||||||
|
type: 'toolbar', // 有助于在某些操作系统上保持置顶
|
||||||
|
frame: false,
|
||||||
|
transparent: true,
|
||||||
|
resizable: false,
|
||||||
|
show: false,
|
||||||
|
alwaysOnTop: true,
|
||||||
|
skipTaskbar: true,
|
||||||
|
focusable: false, // 不抢占焦点
|
||||||
|
icon: iconPath,
|
||||||
|
webPreferences: {
|
||||||
|
preload: join(__dirname, 'preload.js'), // FIX: Use correct relative path (same dir in dist)
|
||||||
|
contextIsolation: true,
|
||||||
|
nodeIntegration: false,
|
||||||
|
// devTools: true // Enable DevTools
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
// notificationWindow.webContents.openDevTools({ mode: 'detach' }) // DEBUG: Force Open DevTools
|
||||||
|
notificationWindow.setIgnoreMouseEvents(true, { forward: true }) // 初始点击穿透
|
||||||
|
|
||||||
|
// 处理鼠标事件 (如果需要从渲染进程转发,但目前特定区域处理?)
|
||||||
|
// 实际上,我们希望窗口可点击。
|
||||||
|
// 我们将在显示时将忽略鼠标事件设为 false。
|
||||||
|
|
||||||
|
const loadUrl = isDev
|
||||||
|
? `${process.env.VITE_DEV_SERVER_URL}#/notification-window`
|
||||||
|
: `file://${join(__dirname, '../dist/index.html')}#/notification-window`
|
||||||
|
|
||||||
|
console.log('[NotificationWindow] Loading URL:', loadUrl)
|
||||||
|
notificationWindow.loadURL(loadUrl)
|
||||||
|
|
||||||
|
notificationWindow.on('closed', () => {
|
||||||
|
notificationWindow = null
|
||||||
|
})
|
||||||
|
|
||||||
|
return notificationWindow
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function showNotification(data: any) {
|
||||||
|
// 先检查配置
|
||||||
|
const config = ConfigService.getInstance()
|
||||||
|
const enabled = await config.get('notificationEnabled')
|
||||||
|
if (enabled === false) return // 默认为 true
|
||||||
|
|
||||||
|
// 检查会话过滤
|
||||||
|
const filterMode = config.get('notificationFilterMode') || 'all'
|
||||||
|
const filterList = config.get('notificationFilterList') || []
|
||||||
|
const sessionId = data.sessionId
|
||||||
|
|
||||||
|
if (sessionId && filterMode !== 'all' && filterList.length > 0) {
|
||||||
|
const isInList = filterList.includes(sessionId)
|
||||||
|
if (filterMode === 'whitelist' && !isInList) {
|
||||||
|
// 白名单模式:不在列表中则不显示
|
||||||
|
console.log('[NotificationWindow] Filtered by whitelist:', sessionId)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if (filterMode === 'blacklist' && isInList) {
|
||||||
|
// 黑名单模式:在列表中则不显示
|
||||||
|
console.log('[NotificationWindow] Filtered by blacklist:', sessionId)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let win = notificationWindow
|
||||||
|
if (!win || win.isDestroyed()) {
|
||||||
|
win = createNotificationWindow()
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!win) return
|
||||||
|
|
||||||
|
// 确保加载完成
|
||||||
|
if (win.webContents.isLoading()) {
|
||||||
|
win.once('ready-to-show', () => {
|
||||||
|
showAndSend(win!, data)
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
showAndSend(win, data)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let lastNotificationData: any = null
|
||||||
|
|
||||||
|
async function showAndSend(win: BrowserWindow, data: any) {
|
||||||
|
lastNotificationData = data
|
||||||
|
const config = ConfigService.getInstance()
|
||||||
|
const position = (await config.get('notificationPosition')) || 'top-right'
|
||||||
|
|
||||||
|
// 更新位置
|
||||||
|
const { width: screenWidth, height: screenHeight } = screen.getPrimaryDisplay().workAreaSize
|
||||||
|
const winWidth = 344
|
||||||
|
const winHeight = 114
|
||||||
|
const padding = 20
|
||||||
|
|
||||||
|
let x = 0
|
||||||
|
let y = 0
|
||||||
|
|
||||||
|
switch (position) {
|
||||||
|
case 'top-right':
|
||||||
|
x = screenWidth - winWidth - padding
|
||||||
|
y = padding
|
||||||
|
break
|
||||||
|
case 'bottom-right':
|
||||||
|
x = screenWidth - winWidth - padding
|
||||||
|
y = screenHeight - winHeight - padding
|
||||||
|
break
|
||||||
|
case 'top-left':
|
||||||
|
x = padding
|
||||||
|
y = padding
|
||||||
|
break
|
||||||
|
case 'bottom-left':
|
||||||
|
x = padding
|
||||||
|
y = screenHeight - winHeight - padding
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
win.setPosition(Math.floor(x), Math.floor(y))
|
||||||
|
win.setSize(winWidth, winHeight) // 确保尺寸
|
||||||
|
|
||||||
|
// 设为可交互
|
||||||
|
win.setIgnoreMouseEvents(false)
|
||||||
|
win.showInactive() // 显示但不聚焦
|
||||||
|
win.setAlwaysOnTop(true, 'screen-saver') // 最高层级
|
||||||
|
|
||||||
|
win.webContents.send('notification:show', data)
|
||||||
|
|
||||||
|
// 自动关闭计时器通常由渲染进程管理
|
||||||
|
// 渲染进程发送 'notification:close' 来隐藏窗口
|
||||||
|
}
|
||||||
|
|
||||||
|
export function registerNotificationHandlers() {
|
||||||
|
ipcMain.handle('notification:show', (_, data) => {
|
||||||
|
showNotification(data)
|
||||||
|
})
|
||||||
|
|
||||||
|
ipcMain.handle('notification:close', () => {
|
||||||
|
if (notificationWindow && !notificationWindow.isDestroyed()) {
|
||||||
|
notificationWindow.hide()
|
||||||
|
notificationWindow.setIgnoreMouseEvents(true, { forward: true })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
// Handle renderer ready event (fix race condition)
|
||||||
|
ipcMain.on('notification:ready', (event) => {
|
||||||
|
console.log('[NotificationWindow] Renderer ready, checking cached data')
|
||||||
|
if (lastNotificationData && notificationWindow && !notificationWindow.isDestroyed()) {
|
||||||
|
console.log('[NotificationWindow] Re-sending cached data')
|
||||||
|
notificationWindow.webContents.send('notification:show', lastNotificationData)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
// Handle resize request from renderer
|
||||||
|
ipcMain.on('notification:resize', (event, { width, height }) => {
|
||||||
|
if (notificationWindow && !notificationWindow.isDestroyed()) {
|
||||||
|
// Enforce max-height if needed, or trust renderer
|
||||||
|
// Ensure it doesn't go off screen bottom?
|
||||||
|
// Logic in showAndSend handles position, but we need to keep anchor point (top-right usually).
|
||||||
|
// If we resize, we should re-calculate position to keep it anchored?
|
||||||
|
// Actually, setSize changes size. If it's top-right, x/y stays same -> window grows down. That's fine for top-right.
|
||||||
|
// If bottom-right, growing down pushes it off screen.
|
||||||
|
|
||||||
|
// Simple version: just setSize. For V1 we assume Top-Right.
|
||||||
|
// But wait, the config supports bottom-right.
|
||||||
|
// We can re-call setPosition or just let it be.
|
||||||
|
// If bottom-right, y needs to prevent overflow.
|
||||||
|
|
||||||
|
// Ideally we get current config position
|
||||||
|
const bounds = notificationWindow.getBounds()
|
||||||
|
// Check if we need to adjust Y?
|
||||||
|
// For now, let's just set the size as requested.
|
||||||
|
notificationWindow.setSize(Math.round(width), Math.round(height))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
// 'notification-clicked' 在 main.ts 中处理 (导航)
|
||||||
|
}
|
||||||
@@ -2,6 +2,7 @@
|
|||||||
ManifestDPIAware true
|
ManifestDPIAware true
|
||||||
|
|
||||||
!include "WordFunc.nsh"
|
!include "WordFunc.nsh"
|
||||||
|
!include "nsDialogs.nsh"
|
||||||
|
|
||||||
!macro customInit
|
!macro customInit
|
||||||
; 设置 DPI 感知
|
; 设置 DPI 感知
|
||||||
@@ -16,3 +17,49 @@ ManifestDPIAware true
|
|||||||
StrCpy $INSTDIR "$INSTDIR\WeFlow"
|
StrCpy $INSTDIR "$INSTDIR\WeFlow"
|
||||||
${EndIf}
|
${EndIf}
|
||||||
!macroend
|
!macroend
|
||||||
|
|
||||||
|
; 安装完成后检测并安装 VC++ Redistributable
|
||||||
|
!macro customInstall
|
||||||
|
; 检查 VC++ 2015-2022 x64 是否已安装
|
||||||
|
ReadRegStr $0 HKLM "SOFTWARE\Microsoft\VisualStudio\14.0\VC\Runtimes\x64" "Installed"
|
||||||
|
${If} $0 != "1"
|
||||||
|
; 未安装,显示提示并下载
|
||||||
|
MessageBox MB_YESNO|MB_ICONQUESTION "检测到系统缺少 Visual C++ 运行库,这可能导致程序无法正常运行。$\n$\n是否立即下载并安装?(约 24MB)" IDYES downloadVC IDNO skipVC
|
||||||
|
|
||||||
|
downloadVC:
|
||||||
|
DetailPrint "正在下载 Visual C++ Redistributable..."
|
||||||
|
SetOutPath "$TEMP"
|
||||||
|
|
||||||
|
; 从微软官方下载 VC++ Redistributable x64
|
||||||
|
inetc::get /TIMEOUT=30000 /CAPTION "下载 Visual C++ 运行库" /BANNER "正在下载,请稍候..." \
|
||||||
|
"https://aka.ms/vs/17/release/vc_redist.x64.exe" "$TEMP\vc_redist.x64.exe" /END
|
||||||
|
Pop $0
|
||||||
|
|
||||||
|
${If} $0 == "OK"
|
||||||
|
DetailPrint "下载完成,正在安装..."
|
||||||
|
; 使用 ShellExecute 以管理员权限运行
|
||||||
|
ExecShell "runas" '"$TEMP\vc_redist.x64.exe"' "/install /quiet /norestart" SW_HIDE
|
||||||
|
; 等待安装完成
|
||||||
|
Sleep 5000
|
||||||
|
; 检查是否安装成功
|
||||||
|
ReadRegStr $1 HKLM "SOFTWARE\Microsoft\VisualStudio\14.0\VC\Runtimes\x64" "Installed"
|
||||||
|
${If} $1 == "1"
|
||||||
|
DetailPrint "Visual C++ Redistributable 安装成功"
|
||||||
|
MessageBox MB_OK|MB_ICONINFORMATION "Visual C++ 运行库安装成功!"
|
||||||
|
${Else}
|
||||||
|
MessageBox MB_OK|MB_ICONEXCLAMATION "Visual C++ 运行库安装失败,你可能需要手动安装。"
|
||||||
|
${EndIf}
|
||||||
|
Delete "$TEMP\vc_redist.x64.exe"
|
||||||
|
${Else}
|
||||||
|
MessageBox MB_OK|MB_ICONEXCLAMATION "下载失败:$0$\n$\n你可以稍后手动下载安装 Visual C++ Redistributable。"
|
||||||
|
${EndIf}
|
||||||
|
Goto doneVC
|
||||||
|
|
||||||
|
skipVC:
|
||||||
|
DetailPrint "用户跳过 Visual C++ Redistributable 安装"
|
||||||
|
|
||||||
|
doneVC:
|
||||||
|
${Else}
|
||||||
|
DetailPrint "Visual C++ Redistributable 已安装"
|
||||||
|
${EndIf}
|
||||||
|
!macroend
|
||||||
|
|||||||
3981
package-lock.json
generated
3981
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
47
package.json
47
package.json
@@ -1,16 +1,22 @@
|
|||||||
{
|
{
|
||||||
"name": "weflow",
|
"name": "weflow",
|
||||||
"version": "1.0.4",
|
"version": "1.5.4",
|
||||||
"description": "WeFlow - 微信聊天记录查看工具",
|
"description": "WeFlow",
|
||||||
"main": "dist-electron/main.js",
|
"main": "dist-electron/main.js",
|
||||||
"author": "cc",
|
"author": "cc",
|
||||||
|
"repository": {
|
||||||
|
"type": "git",
|
||||||
|
"url": "https://github.com/hicccc77/WeFlow"
|
||||||
|
},
|
||||||
|
"//": "二改不应改变此处的作者与应用信息",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
|
"postinstall": "echo 'No native modules to rebuild'",
|
||||||
|
"rebuild": "electron-rebuild",
|
||||||
"dev": "vite",
|
"dev": "vite",
|
||||||
"build": "tsc && vite build && electron-builder",
|
"build": "tsc && vite build && electron-builder",
|
||||||
"preview": "vite preview",
|
"preview": "vite preview",
|
||||||
"electron:dev": "vite --mode electron",
|
"electron:dev": "vite --mode electron",
|
||||||
"electron:build": "npm run build",
|
"electron:build": "npm run build"
|
||||||
"postinstall": "electron-rebuild"
|
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"better-sqlite3": "^12.5.0",
|
"better-sqlite3": "^12.5.0",
|
||||||
@@ -18,15 +24,23 @@
|
|||||||
"echarts-for-react": "^3.0.2",
|
"echarts-for-react": "^3.0.2",
|
||||||
"electron-store": "^10.0.0",
|
"electron-store": "^10.0.0",
|
||||||
"electron-updater": "^6.3.9",
|
"electron-updater": "^6.3.9",
|
||||||
|
"exceljs": "^4.4.0",
|
||||||
|
"ffmpeg-static": "^5.3.0",
|
||||||
"fzstd": "^0.1.1",
|
"fzstd": "^0.1.1",
|
||||||
"html2canvas": "^1.4.1",
|
"html2canvas": "^1.4.1",
|
||||||
"jieba-wasm": "^2.2.0",
|
"jieba-wasm": "^2.2.0",
|
||||||
"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-router-dom": "^7.1.1",
|
"react-router-dom": "^7.1.1",
|
||||||
|
"react-virtuoso": "^4.18.1",
|
||||||
|
"remark-gfm": "^4.0.1",
|
||||||
|
"sherpa-onnx-node": "^1.10.38",
|
||||||
|
"silk-wasm": "^3.7.1",
|
||||||
"wechat-emojis": "^1.0.2",
|
"wechat-emojis": "^1.0.2",
|
||||||
"zustand": "^5.0.2"
|
"zustand": "^5.0.2"
|
||||||
},
|
},
|
||||||
@@ -49,6 +63,8 @@
|
|||||||
"appId": "com.WeFlow.app",
|
"appId": "com.WeFlow.app",
|
||||||
"publish": {
|
"publish": {
|
||||||
"provider": "github",
|
"provider": "github",
|
||||||
|
"owner": "hicccc77",
|
||||||
|
"repo": "WeFlow",
|
||||||
"releaseType": "release"
|
"releaseType": "release"
|
||||||
},
|
},
|
||||||
"productName": "WeFlow",
|
"productName": "WeFlow",
|
||||||
@@ -96,6 +112,29 @@
|
|||||||
"files": [
|
"files": [
|
||||||
"dist/**/*",
|
"dist/**/*",
|
||||||
"dist-electron/**/*"
|
"dist-electron/**/*"
|
||||||
|
],
|
||||||
|
"asarUnpack": [
|
||||||
|
"node_modules/silk-wasm/**/*",
|
||||||
|
"node_modules/sherpa-onnx-node/**/*",
|
||||||
|
"node_modules/ffmpeg-static/**/*"
|
||||||
|
],
|
||||||
|
"extraFiles": [
|
||||||
|
{
|
||||||
|
"from": "resources/msvcp140.dll",
|
||||||
|
"to": "."
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"from": "resources/msvcp140_1.dll",
|
||||||
|
"to": "."
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"from": "resources/vcruntime140.dll",
|
||||||
|
"to": "."
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"from": "resources/vcruntime140_1.dll",
|
||||||
|
"to": "."
|
||||||
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
BIN
resources/SDL2.dll
Normal file
BIN
resources/SDL2.dll
Normal file
Binary file not shown.
BIN
resources/msvcp140.dll
Normal file
BIN
resources/msvcp140.dll
Normal file
Binary file not shown.
BIN
resources/msvcp140_1.dll
Normal file
BIN
resources/msvcp140_1.dll
Normal file
Binary file not shown.
Binary file not shown.
BIN
resources/vcruntime140.dll
Normal file
BIN
resources/vcruntime140.dll
Normal file
Binary file not shown.
BIN
resources/vcruntime140_1.dll
Normal file
BIN
resources/vcruntime140_1.dll
Normal file
Binary file not shown.
Binary file not shown.
230
src/App.tsx
230
src/App.tsx
@@ -7,44 +7,79 @@ import WelcomePage from './pages/WelcomePage'
|
|||||||
import HomePage from './pages/HomePage'
|
import HomePage from './pages/HomePage'
|
||||||
import ChatPage from './pages/ChatPage'
|
import ChatPage from './pages/ChatPage'
|
||||||
import AnalyticsPage from './pages/AnalyticsPage'
|
import AnalyticsPage from './pages/AnalyticsPage'
|
||||||
|
import AnalyticsWelcomePage from './pages/AnalyticsWelcomePage'
|
||||||
import AnnualReportPage from './pages/AnnualReportPage'
|
import AnnualReportPage from './pages/AnnualReportPage'
|
||||||
import AnnualReportWindow from './pages/AnnualReportWindow'
|
import AnnualReportWindow from './pages/AnnualReportWindow'
|
||||||
|
import DualReportPage from './pages/DualReportPage'
|
||||||
|
import DualReportWindow from './pages/DualReportWindow'
|
||||||
import AgreementPage from './pages/AgreementPage'
|
import AgreementPage from './pages/AgreementPage'
|
||||||
import GroupAnalyticsPage from './pages/GroupAnalyticsPage'
|
import GroupAnalyticsPage from './pages/GroupAnalyticsPage'
|
||||||
import DataManagementPage from './pages/DataManagementPage'
|
|
||||||
import SettingsPage from './pages/SettingsPage'
|
import SettingsPage from './pages/SettingsPage'
|
||||||
import ExportPage from './pages/ExportPage'
|
import ExportPage from './pages/ExportPage'
|
||||||
|
import VideoWindow from './pages/VideoWindow'
|
||||||
|
import ImageWindow from './pages/ImageWindow'
|
||||||
|
import SnsPage from './pages/SnsPage'
|
||||||
|
import ContactsPage from './pages/ContactsPage'
|
||||||
|
import ChatHistoryPage from './pages/ChatHistoryPage'
|
||||||
|
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 } 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'
|
||||||
|
|
||||||
|
import UpdateDialog from './components/UpdateDialog'
|
||||||
|
import UpdateProgressCapsule from './components/UpdateProgressCapsule'
|
||||||
|
import LockScreen from './components/LockScreen'
|
||||||
|
import { GlobalSessionMonitor } from './components/GlobalSessionMonitor'
|
||||||
|
|
||||||
function App() {
|
function App() {
|
||||||
const navigate = useNavigate()
|
const navigate = useNavigate()
|
||||||
const location = useLocation()
|
const location = useLocation()
|
||||||
const { setDbConnected } = useAppStore()
|
|
||||||
|
const {
|
||||||
|
setDbConnected,
|
||||||
|
updateInfo,
|
||||||
|
setUpdateInfo,
|
||||||
|
isDownloading,
|
||||||
|
setIsDownloading,
|
||||||
|
downloadProgress,
|
||||||
|
setDownloadProgress,
|
||||||
|
showUpdateDialog,
|
||||||
|
setShowUpdateDialog,
|
||||||
|
setUpdateError,
|
||||||
|
isLocked,
|
||||||
|
setLocked
|
||||||
|
} = useAppStore()
|
||||||
|
|
||||||
const { currentTheme, themeMode, setTheme, setThemeMode } = useThemeStore()
|
const { currentTheme, themeMode, setTheme, setThemeMode } = useThemeStore()
|
||||||
const isAgreementWindow = location.pathname === '/agreement-window'
|
const isAgreementWindow = location.pathname === '/agreement-window'
|
||||||
const isOnboardingWindow = location.pathname === '/onboarding-window'
|
const isOnboardingWindow = location.pathname === '/onboarding-window'
|
||||||
|
const isVideoPlayerWindow = location.pathname === '/video-player-window'
|
||||||
|
const isChatHistoryWindow = location.pathname.startsWith('/chat-history/')
|
||||||
|
const isNotificationWindow = location.pathname === '/notification-window'
|
||||||
const [themeHydrated, setThemeHydrated] = useState(false)
|
const [themeHydrated, setThemeHydrated] = useState(false)
|
||||||
|
|
||||||
|
// 锁定状态
|
||||||
|
// const [isLocked, setIsLocked] = useState(false) // Moved to store
|
||||||
|
const [lockAvatar, setLockAvatar] = useState<string | undefined>(
|
||||||
|
localStorage.getItem('app_lock_avatar') || undefined
|
||||||
|
)
|
||||||
|
const [lockUseHello, setLockUseHello] = useState(false)
|
||||||
|
|
||||||
// 协议同意状态
|
// 协议同意状态
|
||||||
const [showAgreement, setShowAgreement] = useState(false)
|
const [showAgreement, setShowAgreement] = useState(false)
|
||||||
const [agreementChecked, setAgreementChecked] = useState(false)
|
const [agreementChecked, setAgreementChecked] = useState(false)
|
||||||
const [agreementLoading, setAgreementLoading] = useState(true)
|
const [agreementLoading, setAgreementLoading] = useState(true)
|
||||||
|
|
||||||
// 更新提示状态
|
|
||||||
const [updateInfo, setUpdateInfo] = useState<{ version: string; releaseNotes: string } | null>(null)
|
|
||||||
const [isDownloading, setIsDownloading] = useState(false)
|
|
||||||
const [downloadProgress, setDownloadProgress] = useState(0)
|
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const root = document.documentElement
|
const root = document.documentElement
|
||||||
const body = document.body
|
const body = document.body
|
||||||
const appRoot = document.getElementById('app')
|
const appRoot = document.getElementById('app')
|
||||||
|
|
||||||
if (isOnboardingWindow) {
|
if (isOnboardingWindow || isNotificationWindow) {
|
||||||
root.style.background = 'transparent'
|
root.style.background = 'transparent'
|
||||||
body.style.background = 'transparent'
|
body.style.background = 'transparent'
|
||||||
body.style.overflow = 'hidden'
|
body.style.overflow = 'hidden'
|
||||||
@@ -70,10 +105,10 @@ function App() {
|
|||||||
|
|
||||||
// 更新窗口控件颜色以适配主题
|
// 更新窗口控件颜色以适配主题
|
||||||
const symbolColor = themeMode === 'dark' ? '#ffffff' : '#1a1a1a'
|
const symbolColor = themeMode === 'dark' ? '#ffffff' : '#1a1a1a'
|
||||||
if (!isOnboardingWindow) {
|
if (!isOnboardingWindow && !isNotificationWindow) {
|
||||||
window.electronAPI.window.setTitleBarOverlay({ symbolColor })
|
window.electronAPI.window.setTitleBarOverlay({ symbolColor })
|
||||||
}
|
}
|
||||||
}, [currentTheme, themeMode, isOnboardingWindow])
|
}, [currentTheme, themeMode, isOnboardingWindow, isNotificationWindow])
|
||||||
|
|
||||||
// 读取已保存的主题设置
|
// 读取已保存的主题设置
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -143,26 +178,48 @@ function App() {
|
|||||||
|
|
||||||
// 监听启动时的更新通知
|
// 监听启动时的更新通知
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const removeUpdateListener = window.electronAPI.app.onUpdateAvailable?.((info) => {
|
if (isNotificationWindow) return // Skip updates in notification window
|
||||||
setUpdateInfo(info)
|
|
||||||
|
const removeUpdateListener = window.electronAPI?.app?.onUpdateAvailable?.((info: any) => {
|
||||||
|
// 发现新版本时自动打开更新弹窗
|
||||||
|
if (info) {
|
||||||
|
setUpdateInfo({ ...info, hasUpdate: true })
|
||||||
|
setShowUpdateDialog(true)
|
||||||
|
}
|
||||||
})
|
})
|
||||||
const removeProgressListener = window.electronAPI.app.onDownloadProgress?.((progress) => {
|
const removeProgressListener = window.electronAPI?.app?.onDownloadProgress?.((progress: any) => {
|
||||||
setDownloadProgress(progress)
|
setDownloadProgress(progress)
|
||||||
})
|
})
|
||||||
return () => {
|
return () => {
|
||||||
removeUpdateListener?.()
|
removeUpdateListener?.()
|
||||||
removeProgressListener?.()
|
removeProgressListener?.()
|
||||||
}
|
}
|
||||||
}, [])
|
}, [setUpdateInfo, setDownloadProgress, setShowUpdateDialog, isNotificationWindow])
|
||||||
|
|
||||||
const handleUpdateNow = async () => {
|
const handleUpdateNow = async () => {
|
||||||
|
setShowUpdateDialog(false)
|
||||||
setIsDownloading(true)
|
setIsDownloading(true)
|
||||||
setDownloadProgress(0)
|
setDownloadProgress({ percent: 0 })
|
||||||
try {
|
try {
|
||||||
await window.electronAPI.app.downloadAndInstall()
|
await window.electronAPI.app.downloadAndInstall()
|
||||||
} catch (e) {
|
} catch (e: any) {
|
||||||
console.error('更新失败:', e)
|
console.error('更新失败:', e)
|
||||||
setIsDownloading(false)
|
setIsDownloading(false)
|
||||||
|
// Extract clean error message if possible
|
||||||
|
const errorMsg = e.message || String(e)
|
||||||
|
setUpdateError(errorMsg.includes('暂时禁用') ? '自动更新已暂时禁用' : errorMsg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleIgnoreUpdate = async () => {
|
||||||
|
if (!updateInfo || !updateInfo.version) return
|
||||||
|
|
||||||
|
try {
|
||||||
|
await window.electronAPI.app.ignoreUpdate(updateInfo.version)
|
||||||
|
setShowUpdateDialog(false)
|
||||||
|
setUpdateInfo(null)
|
||||||
|
} catch (e: any) {
|
||||||
|
console.error('忽略更新失败:', e)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -180,34 +237,81 @@ function App() {
|
|||||||
const decryptKey = await configService.getDecryptKey()
|
const decryptKey = await configService.getDecryptKey()
|
||||||
const wxid = await configService.getMyWxid()
|
const wxid = await configService.getMyWxid()
|
||||||
const onboardingDone = await configService.getOnboardingDone()
|
const onboardingDone = await configService.getOnboardingDone()
|
||||||
|
const wxidConfig = wxid ? await configService.getWxidConfig(wxid) : null
|
||||||
|
const effectiveDecryptKey = wxidConfig?.decryptKey || decryptKey
|
||||||
|
|
||||||
|
if (wxidConfig?.decryptKey && wxidConfig.decryptKey !== decryptKey) {
|
||||||
|
await configService.setDecryptKey(wxidConfig.decryptKey)
|
||||||
|
}
|
||||||
|
|
||||||
// 如果配置完整,自动测试连接
|
// 如果配置完整,自动测试连接
|
||||||
if (dbPath && decryptKey && wxid) {
|
if (dbPath && effectiveDecryptKey && wxid) {
|
||||||
if (!onboardingDone) {
|
if (!onboardingDone) {
|
||||||
await configService.setOnboardingDone(true)
|
await configService.setOnboardingDone(true)
|
||||||
}
|
}
|
||||||
console.log('检测到已保存的配置,正在自动连接...')
|
|
||||||
const result = await window.electronAPI.chat.connect()
|
const result = await window.electronAPI.chat.connect()
|
||||||
|
|
||||||
if (result.success) {
|
if (result.success) {
|
||||||
console.log('自动连接成功')
|
|
||||||
setDbConnected(true, dbPath)
|
setDbConnected(true, dbPath)
|
||||||
// 如果当前在欢迎页,跳转到首页
|
// 如果当前在欢迎页,跳转到首页
|
||||||
if (window.location.hash === '#/' || window.location.hash === '') {
|
if (window.location.hash === '#/' || window.location.hash === '') {
|
||||||
navigate('/home')
|
navigate('/home')
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
console.log('自动连接失败:', result.error)
|
|
||||||
|
// 如果错误信息包含 VC++ 或 DLL 相关内容,不清除配置,只提示用户
|
||||||
|
// 其他错误可能需要重新配置
|
||||||
|
const errorMsg = result.error || ''
|
||||||
|
if (errorMsg.includes('Visual C++') ||
|
||||||
|
errorMsg.includes('DLL') ||
|
||||||
|
errorMsg.includes('Worker') ||
|
||||||
|
errorMsg.includes('126') ||
|
||||||
|
errorMsg.includes('模块')) {
|
||||||
|
console.warn('检测到可能的运行时依赖问题:', errorMsg)
|
||||||
|
// 不清除配置,让用户安装 VC++ 后重试
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('自动连接出错:', e)
|
console.error('自动连接出错:', e)
|
||||||
|
// 捕获异常但不清除配置,防止循环重新引导
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
autoConnect()
|
autoConnect()
|
||||||
}, [isAgreementWindow, isOnboardingWindow, navigate, setDbConnected])
|
}, [isAgreementWindow, isOnboardingWindow, navigate, setDbConnected])
|
||||||
|
|
||||||
|
// 检查应用锁
|
||||||
|
useEffect(() => {
|
||||||
|
if (isAgreementWindow || isOnboardingWindow || isVideoPlayerWindow) return
|
||||||
|
|
||||||
|
const checkLock = async () => {
|
||||||
|
// 并行获取配置,减少等待
|
||||||
|
const [enabled, useHello] = await Promise.all([
|
||||||
|
configService.getAuthEnabled(),
|
||||||
|
configService.getAuthUseHello()
|
||||||
|
])
|
||||||
|
|
||||||
|
if (enabled) {
|
||||||
|
setLockUseHello(useHello)
|
||||||
|
setLocked(true)
|
||||||
|
// 尝试获取头像
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.chat.getMyAvatarUrl()
|
||||||
|
if (result && result.success && result.avatarUrl) {
|
||||||
|
setLockAvatar(result.avatarUrl)
|
||||||
|
localStorage.setItem('app_lock_avatar', result.avatarUrl)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('获取锁屏头像失败', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
checkLock()
|
||||||
|
}, [isAgreementWindow, isOnboardingWindow, isVideoPlayerWindow])
|
||||||
|
|
||||||
// 独立协议窗口
|
// 独立协议窗口
|
||||||
if (isAgreementWindow) {
|
if (isAgreementWindow) {
|
||||||
return <AgreementPage />
|
return <AgreementPage />
|
||||||
@@ -217,11 +321,45 @@ function App() {
|
|||||||
return <WelcomePage standalone />
|
return <WelcomePage standalone />
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 独立视频播放窗口
|
||||||
|
if (isVideoPlayerWindow) {
|
||||||
|
return <VideoWindow />
|
||||||
|
}
|
||||||
|
|
||||||
|
// 独立图片查看窗口
|
||||||
|
const isImageViewerWindow = location.pathname === '/image-viewer-window'
|
||||||
|
if (isImageViewerWindow) {
|
||||||
|
return <ImageWindow />
|
||||||
|
}
|
||||||
|
|
||||||
|
// 独立聊天记录窗口
|
||||||
|
if (isChatHistoryWindow) {
|
||||||
|
return <ChatHistoryPage />
|
||||||
|
}
|
||||||
|
|
||||||
|
// 独立通知窗口
|
||||||
|
if (isNotificationWindow) {
|
||||||
|
return <NotificationWindow />
|
||||||
|
}
|
||||||
|
|
||||||
// 主窗口 - 完整布局
|
// 主窗口 - 完整布局
|
||||||
return (
|
return (
|
||||||
<div className="app-container">
|
<div className="app-container">
|
||||||
|
{isLocked && (
|
||||||
|
<LockScreen
|
||||||
|
onUnlock={() => setLocked(false)}
|
||||||
|
avatar={lockAvatar}
|
||||||
|
useHello={lockUseHello}
|
||||||
|
/>
|
||||||
|
)}
|
||||||
<TitleBar />
|
<TitleBar />
|
||||||
|
|
||||||
|
{/* 全局悬浮进度胶囊 (处理:新版本提示、下载进度、错误提示) */}
|
||||||
|
<UpdateProgressCapsule />
|
||||||
|
|
||||||
|
{/* 全局会话监听与通知 */}
|
||||||
|
<GlobalSessionMonitor />
|
||||||
|
|
||||||
{/* 用户协议弹窗 */}
|
{/* 用户协议弹窗 */}
|
||||||
{showAgreement && !agreementLoading && (
|
{showAgreement && !agreementLoading && (
|
||||||
<div className="agreement-overlay">
|
<div className="agreement-overlay">
|
||||||
@@ -243,13 +381,13 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
<div className="agreement-text">
|
<div className="agreement-text">
|
||||||
<h4>1. 数据安全</h4>
|
<h4>1. 数据安全</h4>
|
||||||
<p>本软件所有数据处理均在本地完成,不会上传任何聊天记录、个人信息到服务器。您的数据完全由您自己掌控。</p>
|
<p>本软件所有数据处理均在本地完成,不会上传任何聊天记录、个人信息到服务器。你的数据完全由你自己掌控。</p>
|
||||||
|
|
||||||
<h4>2. 使用须知</h4>
|
<h4>2. 使用须知</h4>
|
||||||
<p>本软件仅供个人学习研究使用,请勿用于任何非法用途。使用本软件解密、查看、分析的数据应为您本人所有或已获得授权。</p>
|
<p>本软件仅供个人学习研究使用,请勿用于任何非法用途。使用本软件解密、查看、分析的数据应为你本人所有或已获得授权。</p>
|
||||||
|
|
||||||
<h4>3. 免责声明</h4>
|
<h4>3. 免责声明</h4>
|
||||||
<p>因使用本软件产生的任何直接或间接损失,开发者不承担任何责任。请确保您的使用行为符合当地法律法规。</p>
|
<p>因使用本软件产生的任何直接或间接损失,开发者不承担任何责任。请确保你的使用行为符合当地法律法规。</p>
|
||||||
|
|
||||||
<h4>4. 隐私保护</h4>
|
<h4>4. 隐私保护</h4>
|
||||||
<p>本软件不收集任何用户数据。软件更新检测仅获取版本信息,不涉及任何个人隐私。</p>
|
<p>本软件不收集任何用户数据。软件更新检测仅获取版本信息,不涉及任何个人隐私。</p>
|
||||||
@@ -273,31 +411,16 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
{/* 更新提示条 */}
|
{/* 更新提示对话框 */}
|
||||||
{updateInfo && (
|
<UpdateDialog
|
||||||
<div className="update-banner">
|
open={showUpdateDialog}
|
||||||
<span className="update-text">
|
updateInfo={updateInfo}
|
||||||
发现新版本 <strong>v{updateInfo.version}</strong>
|
onClose={() => setShowUpdateDialog(false)}
|
||||||
</span>
|
onUpdate={handleUpdateNow}
|
||||||
{isDownloading ? (
|
onIgnore={handleIgnoreUpdate}
|
||||||
<div className="update-progress">
|
isDownloading={isDownloading}
|
||||||
<div className="progress-bar">
|
progress={downloadProgress}
|
||||||
<div className="progress-fill" style={{ width: `${downloadProgress}%` }} />
|
/>
|
||||||
</div>
|
|
||||||
<span>{downloadProgress.toFixed(0)}%</span>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<>
|
|
||||||
<button className="update-btn" onClick={handleUpdateNow}>
|
|
||||||
<Download size={14} /> 立即更新
|
|
||||||
</button>
|
|
||||||
<button className="dismiss-btn" onClick={dismissUpdate}>
|
|
||||||
<X size={14} />
|
|
||||||
</button>
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<div className="main-layout">
|
<div className="main-layout">
|
||||||
<Sidebar />
|
<Sidebar />
|
||||||
@@ -307,13 +430,20 @@ 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="/analytics" element={<AnalyticsPage />} />
|
<Route path="/ai-chat" element={<AIChatPage />} />
|
||||||
|
<Route path="/analytics" element={<AnalyticsWelcomePage />} />
|
||||||
|
<Route path="/analytics/view" element={<AnalyticsPage />} />
|
||||||
<Route path="/group-analytics" element={<GroupAnalyticsPage />} />
|
<Route path="/group-analytics" element={<GroupAnalyticsPage />} />
|
||||||
<Route path="/annual-report" element={<AnnualReportPage />} />
|
<Route path="/annual-report" element={<AnnualReportPage />} />
|
||||||
<Route path="/annual-report/view" element={<AnnualReportWindow />} />
|
<Route path="/annual-report/view" element={<AnnualReportWindow />} />
|
||||||
<Route path="/data-management" element={<DataManagementPage />} />
|
<Route path="/dual-report" element={<DualReportPage />} />
|
||||||
|
<Route path="/dual-report/view" element={<DualReportWindow />} />
|
||||||
|
|
||||||
<Route path="/settings" element={<SettingsPage />} />
|
<Route path="/settings" element={<SettingsPage />} />
|
||||||
<Route path="/export" element={<ExportPage />} />
|
<Route path="/export" element={<ExportPage />} />
|
||||||
|
<Route path="/sns" element={<SnsPage />} />
|
||||||
|
<Route path="/contacts" element={<ContactsPage />} />
|
||||||
|
<Route path="/chat-history/:sessionId/:messageId" element={<ChatHistoryPage />} />
|
||||||
</Routes>
|
</Routes>
|
||||||
</RouteGuard>
|
</RouteGuard>
|
||||||
</main>
|
</main>
|
||||||
|
|||||||
73
src/components/AnimatedStreamingText.tsx
Normal file
73
src/components/AnimatedStreamingText.tsx
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
import React, { memo, useEffect, useState, useRef } from 'react'
|
||||||
|
|
||||||
|
interface AnimatedStreamingTextProps {
|
||||||
|
text: string
|
||||||
|
className?: string
|
||||||
|
loading?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
export const AnimatedStreamingText = memo(({ text, className, loading }: AnimatedStreamingTextProps) => {
|
||||||
|
const [displayedSegments, setDisplayedSegments] = useState<string[]>([])
|
||||||
|
const prevTextRef = useRef('')
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const currentText = (text || '').trim()
|
||||||
|
const prevText = prevTextRef.current
|
||||||
|
|
||||||
|
if (currentText === prevText) return
|
||||||
|
if (!currentText.startsWith(prevText) && prevText !== '') {
|
||||||
|
// 如果不是追加而是全新的文本(比如重新识别),则重置
|
||||||
|
setDisplayedSegments([currentText])
|
||||||
|
prevTextRef.current = currentText
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const newPart = currentText.slice(prevText.length)
|
||||||
|
if (newPart) {
|
||||||
|
// 将新部分作为单独的段加入,以触发动画
|
||||||
|
setDisplayedSegments(prev => [...prev, newPart])
|
||||||
|
}
|
||||||
|
prevTextRef.current = currentText
|
||||||
|
}, [text])
|
||||||
|
|
||||||
|
// 处理 loading 状态的显示
|
||||||
|
if (loading && !text) {
|
||||||
|
return <span className={className}>转写中<span className="dot-flashing">...</span></span>
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<span className={className}>
|
||||||
|
{displayedSegments.map((segment, index) => (
|
||||||
|
<span key={index} className="fade-in-text">
|
||||||
|
{segment}
|
||||||
|
</span>
|
||||||
|
))}
|
||||||
|
<style>{`
|
||||||
|
.fade-in-text {
|
||||||
|
animation: premiumFadeIn 0.8s cubic-bezier(0.16, 1, 0.3, 1) forwards;
|
||||||
|
opacity: 0;
|
||||||
|
display: inline-block;
|
||||||
|
filter: blur(4px);
|
||||||
|
}
|
||||||
|
@keyframes premiumFadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(4px) scale(0.98);
|
||||||
|
filter: blur(4px);
|
||||||
|
}
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateY(0) scale(1);
|
||||||
|
filter: blur(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.dot-flashing {
|
||||||
|
animation: blink 1s infinite;
|
||||||
|
}
|
||||||
|
@keyframes blink { 50% { opacity: 0; } }
|
||||||
|
`}</style>
|
||||||
|
</span>
|
||||||
|
)
|
||||||
|
})
|
||||||
|
|
||||||
|
AnimatedStreamingText.displayName = 'AnimatedStreamingText'
|
||||||
79
src/components/Avatar.scss
Normal file
79
src/components/Avatar.scss
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
.avatar-component {
|
||||||
|
position: relative;
|
||||||
|
display: inline-block;
|
||||||
|
overflow: hidden;
|
||||||
|
background-color: var(--bg-tertiary, #f5f5f5);
|
||||||
|
flex-shrink: 0;
|
||||||
|
border-radius: 4px;
|
||||||
|
/* Default radius */
|
||||||
|
|
||||||
|
&.circle {
|
||||||
|
border-radius: 50%;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.rounded {
|
||||||
|
border-radius: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Image styling */
|
||||||
|
img.avatar-image {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
opacity: 0;
|
||||||
|
transition: opacity 0.3s ease-in-out;
|
||||||
|
border-radius: inherit;
|
||||||
|
|
||||||
|
&.loaded {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.instant {
|
||||||
|
transition: none !important;
|
||||||
|
opacity: 1 !important;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Placeholder/Letter styling */
|
||||||
|
.avatar-placeholder {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-secondary, #666);
|
||||||
|
background-color: var(--bg-tertiary, #e0e0e0);
|
||||||
|
font-size: 1.2em;
|
||||||
|
text-transform: uppercase;
|
||||||
|
user-select: none;
|
||||||
|
border-radius: inherit;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Loading Skeleton */
|
||||||
|
.avatar-skeleton {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
background: linear-gradient(90deg,
|
||||||
|
var(--bg-tertiary, #f0f0f0) 25%,
|
||||||
|
var(--bg-secondary, #e0e0e0) 50%,
|
||||||
|
var(--bg-tertiary, #f0f0f0) 75%);
|
||||||
|
background-size: 200% 100%;
|
||||||
|
animation: shimmer 1.5s infinite;
|
||||||
|
z-index: 1;
|
||||||
|
border-radius: inherit;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes shimmer {
|
||||||
|
0% {
|
||||||
|
background-position: 200% 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
100% {
|
||||||
|
background-position: -200% 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
129
src/components/Avatar.tsx
Normal file
129
src/components/Avatar.tsx
Normal file
@@ -0,0 +1,129 @@
|
|||||||
|
import React, { useState, useEffect, useRef, useMemo } from 'react'
|
||||||
|
import { User } from 'lucide-react'
|
||||||
|
import { avatarLoadQueue } from '../utils/AvatarLoadQueue'
|
||||||
|
import './Avatar.scss'
|
||||||
|
|
||||||
|
// 全局缓存已成功加载过的头像 URL,用于控制后续是否显示动画
|
||||||
|
const loadedAvatarCache = new Set<string>()
|
||||||
|
|
||||||
|
interface AvatarProps {
|
||||||
|
src?: string
|
||||||
|
name?: string
|
||||||
|
size?: number | string
|
||||||
|
shape?: 'circle' | 'square' | 'rounded'
|
||||||
|
className?: string
|
||||||
|
lazy?: boolean
|
||||||
|
onClick?: () => void
|
||||||
|
}
|
||||||
|
|
||||||
|
export const Avatar = React.memo(function Avatar({
|
||||||
|
src,
|
||||||
|
name,
|
||||||
|
size = 48,
|
||||||
|
shape = 'rounded',
|
||||||
|
className = '',
|
||||||
|
lazy = true,
|
||||||
|
onClick
|
||||||
|
}: AvatarProps) {
|
||||||
|
// 如果 URL 已在缓存中,则直接标记为已加载,不显示骨架屏和淡入动画
|
||||||
|
const isCached = useMemo(() => src ? loadedAvatarCache.has(src) : false, [src])
|
||||||
|
const [imageLoaded, setImageLoaded] = useState(isCached)
|
||||||
|
const [imageError, setImageError] = useState(false)
|
||||||
|
const [shouldLoad, setShouldLoad] = useState(!lazy || isCached)
|
||||||
|
const [isInQueue, setIsInQueue] = useState(false)
|
||||||
|
const imgRef = useRef<HTMLImageElement>(null)
|
||||||
|
const containerRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
|
const getAvatarLetter = (): string => {
|
||||||
|
if (!name) return '?'
|
||||||
|
const chars = [...name]
|
||||||
|
return chars[0] || '?'
|
||||||
|
}
|
||||||
|
|
||||||
|
// Intersection Observer for lazy loading
|
||||||
|
useEffect(() => {
|
||||||
|
if (!lazy || shouldLoad || isInQueue || !src || !containerRef.current || isCached) return
|
||||||
|
|
||||||
|
const observer = new IntersectionObserver(
|
||||||
|
(entries) => {
|
||||||
|
entries.forEach((entry) => {
|
||||||
|
if (entry.isIntersecting && !isInQueue) {
|
||||||
|
setIsInQueue(true)
|
||||||
|
avatarLoadQueue.enqueue(src).then(() => {
|
||||||
|
setShouldLoad(true)
|
||||||
|
}).catch(() => {
|
||||||
|
// 加载失败不要立刻显示错误,让浏览器渲染去报错
|
||||||
|
setShouldLoad(true)
|
||||||
|
}).finally(() => {
|
||||||
|
setIsInQueue(false)
|
||||||
|
})
|
||||||
|
observer.disconnect()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
},
|
||||||
|
{ rootMargin: '100px' }
|
||||||
|
)
|
||||||
|
|
||||||
|
observer.observe(containerRef.current)
|
||||||
|
|
||||||
|
return () => observer.disconnect()
|
||||||
|
}, [src, lazy, shouldLoad, isInQueue, isCached])
|
||||||
|
|
||||||
|
// Reset state when src changes
|
||||||
|
useEffect(() => {
|
||||||
|
const cached = src ? loadedAvatarCache.has(src) : false
|
||||||
|
setImageLoaded(cached)
|
||||||
|
setImageError(false)
|
||||||
|
if (lazy && !cached) {
|
||||||
|
setShouldLoad(false)
|
||||||
|
setIsInQueue(false)
|
||||||
|
} else {
|
||||||
|
setShouldLoad(true)
|
||||||
|
}
|
||||||
|
}, [src, lazy])
|
||||||
|
|
||||||
|
// Check if image is already cached/loaded
|
||||||
|
useEffect(() => {
|
||||||
|
if (shouldLoad && imgRef.current?.complete && imgRef.current?.naturalWidth > 0) {
|
||||||
|
setImageLoaded(true)
|
||||||
|
}
|
||||||
|
}, [src, shouldLoad])
|
||||||
|
|
||||||
|
const style = {
|
||||||
|
width: typeof size === 'number' ? `${size}px` : size,
|
||||||
|
height: typeof size === 'number' ? `${size}px` : size,
|
||||||
|
}
|
||||||
|
|
||||||
|
const hasValidUrl = !!src && !imageError && shouldLoad
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
ref={containerRef}
|
||||||
|
className={`avatar-component ${shape} ${className}`}
|
||||||
|
style={style}
|
||||||
|
onClick={onClick}
|
||||||
|
>
|
||||||
|
{hasValidUrl ? (
|
||||||
|
<>
|
||||||
|
{!imageLoaded && <div className="avatar-skeleton" />}
|
||||||
|
<img
|
||||||
|
ref={imgRef}
|
||||||
|
src={src}
|
||||||
|
alt={name || 'avatar'}
|
||||||
|
className={`avatar-image ${imageLoaded ? 'loaded' : ''} ${isCached ? 'instant' : ''}`}
|
||||||
|
onLoad={() => {
|
||||||
|
if (src) loadedAvatarCache.add(src)
|
||||||
|
setImageLoaded(true)
|
||||||
|
}}
|
||||||
|
onError={() => setImageError(true)}
|
||||||
|
loading={lazy ? "lazy" : "eager"}
|
||||||
|
/>
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<div className="avatar-placeholder">
|
||||||
|
{name ? <span className="avatar-letter">{getAvatarLetter()}</span> : <User size="50%" />}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
})
|
||||||
271
src/components/GlobalSessionMonitor.tsx
Normal file
271
src/components/GlobalSessionMonitor.tsx
Normal file
@@ -0,0 +1,271 @@
|
|||||||
|
import { useEffect, useRef } from 'react'
|
||||||
|
import { useChatStore } from '../stores/chatStore'
|
||||||
|
import type { ChatSession } from '../types/models'
|
||||||
|
import { useNavigate } from 'react-router-dom'
|
||||||
|
|
||||||
|
export function GlobalSessionMonitor() {
|
||||||
|
const navigate = useNavigate()
|
||||||
|
const {
|
||||||
|
sessions,
|
||||||
|
setSessions,
|
||||||
|
currentSessionId,
|
||||||
|
appendMessages,
|
||||||
|
messages
|
||||||
|
} = useChatStore()
|
||||||
|
|
||||||
|
const sessionsRef = useRef(sessions)
|
||||||
|
|
||||||
|
// 保持 ref 同步
|
||||||
|
useEffect(() => {
|
||||||
|
sessionsRef.current = sessions
|
||||||
|
}, [sessions])
|
||||||
|
|
||||||
|
// 去重辅助函数:获取消息 key
|
||||||
|
const getMessageKey = (msg: any) => {
|
||||||
|
if (msg.localId && msg.localId > 0) return `l:${msg.localId}`
|
||||||
|
return `t:${msg.createTime}:${msg.sortSeq || 0}:${msg.serverId || 0}`
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理数据库变更
|
||||||
|
useEffect(() => {
|
||||||
|
const handleDbChange = (_event: any, data: { type: string; json: string }) => {
|
||||||
|
try {
|
||||||
|
const payload = JSON.parse(data.json)
|
||||||
|
const tableName = payload.table
|
||||||
|
|
||||||
|
// 只关注 Session 表
|
||||||
|
if (tableName === 'Session' || tableName === 'session') {
|
||||||
|
refreshSessions()
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('解析数据库变更失败:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (window.electronAPI.chat.onWcdbChange) {
|
||||||
|
const removeListener = window.electronAPI.chat.onWcdbChange(handleDbChange)
|
||||||
|
return () => {
|
||||||
|
removeListener()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return () => { }
|
||||||
|
}, []) // 空依赖数组 - 主要是静态的
|
||||||
|
|
||||||
|
const refreshSessions = async () => {
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.chat.getSessions()
|
||||||
|
if (result.success && result.sessions && Array.isArray(result.sessions)) {
|
||||||
|
const newSessions = result.sessions as ChatSession[]
|
||||||
|
const oldSessions = sessionsRef.current
|
||||||
|
|
||||||
|
// 1. 检测变更并通知
|
||||||
|
checkForNewMessages(oldSessions, newSessions)
|
||||||
|
|
||||||
|
// 2. 更新 store
|
||||||
|
setSessions(newSessions)
|
||||||
|
|
||||||
|
// 3. 如果在活跃会话中,增量刷新消息
|
||||||
|
const currentId = useChatStore.getState().currentSessionId
|
||||||
|
if (currentId) {
|
||||||
|
const currentSessionNew = newSessions.find(s => s.username === currentId)
|
||||||
|
const currentSessionOld = oldSessions.find(s => s.username === currentId)
|
||||||
|
|
||||||
|
if (currentSessionNew && (!currentSessionOld || currentSessionNew.lastTimestamp > currentSessionOld.lastTimestamp)) {
|
||||||
|
void handleActiveSessionRefresh(currentId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('全局会话刷新失败:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const checkForNewMessages = async (oldSessions: ChatSession[], newSessions: ChatSession[]) => {
|
||||||
|
if (!oldSessions || oldSessions.length === 0) {
|
||||||
|
console.log('[NotificationFilter] Skipping check on initial load (empty baseline)')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const oldMap = new Map(oldSessions.map(s => [s.username, s]))
|
||||||
|
|
||||||
|
for (const newSession of newSessions) {
|
||||||
|
const oldSession = oldMap.get(newSession.username)
|
||||||
|
|
||||||
|
// 条件: 新会话或时间戳更新
|
||||||
|
const isCurrentSession = newSession.username === useChatStore.getState().currentSessionId
|
||||||
|
|
||||||
|
if (!isCurrentSession && (!oldSession || newSession.lastTimestamp > oldSession.lastTimestamp)) {
|
||||||
|
// 这是新消息事件
|
||||||
|
|
||||||
|
// 1. 群聊过滤自己发送的消息
|
||||||
|
if (newSession.username.includes('@chatroom')) {
|
||||||
|
// 如果是自己发的消息,不弹通知
|
||||||
|
// 注意:lastMsgSender 需要后端支持返回
|
||||||
|
// 使用宽松比较以处理 wxid_ 前缀差异
|
||||||
|
if (newSession.lastMsgSender && newSession.selfWxid) {
|
||||||
|
const sender = newSession.lastMsgSender.replace(/^wxid_/, '');
|
||||||
|
const self = newSession.selfWxid.replace(/^wxid_/, '');
|
||||||
|
|
||||||
|
// 使用主进程日志打印,方便用户查看
|
||||||
|
const debugInfo = {
|
||||||
|
type: 'NotificationFilter',
|
||||||
|
username: newSession.username,
|
||||||
|
lastMsgSender: newSession.lastMsgSender,
|
||||||
|
selfWxid: newSession.selfWxid,
|
||||||
|
senderClean: sender,
|
||||||
|
selfClean: self,
|
||||||
|
match: sender === self
|
||||||
|
};
|
||||||
|
|
||||||
|
if (window.electronAPI.log?.debug) {
|
||||||
|
window.electronAPI.log.debug(debugInfo);
|
||||||
|
} else {
|
||||||
|
console.log('[NotificationFilter]', debugInfo);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sender === self) {
|
||||||
|
if (window.electronAPI.log?.debug) {
|
||||||
|
window.electronAPI.log.debug('[NotificationFilter] Filtered own message');
|
||||||
|
} else {
|
||||||
|
console.log('[NotificationFilter] Filtered own message');
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
const missingInfo = {
|
||||||
|
type: 'NotificationFilter Missing info',
|
||||||
|
lastMsgSender: newSession.lastMsgSender,
|
||||||
|
selfWxid: newSession.selfWxid
|
||||||
|
};
|
||||||
|
if (window.electronAPI.log?.debug) {
|
||||||
|
window.electronAPI.log.debug(missingInfo);
|
||||||
|
} else {
|
||||||
|
console.log('[NotificationFilter] Missing info:', missingInfo);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 新增:如果未读数量没有增加,说明可能是自己在其他设备回复(或者已读),不弹通知
|
||||||
|
const oldUnread = oldSession ? oldSession.unreadCount : 0
|
||||||
|
const newUnread = newSession.unreadCount
|
||||||
|
if (newUnread <= oldUnread) {
|
||||||
|
// 仅仅是状态同步(如自己在手机上发消息 or 已读),跳过通知
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
let title = newSession.displayName || newSession.username
|
||||||
|
let avatarUrl = newSession.avatarUrl
|
||||||
|
let content = newSession.summary || '[新消息]'
|
||||||
|
|
||||||
|
if (newSession.username.includes('@chatroom')) {
|
||||||
|
// 1. 群聊过滤自己发送的消息
|
||||||
|
// 辅助函数:清理 wxid 后缀 (如 _8602)
|
||||||
|
const cleanWxid = (id: string) => {
|
||||||
|
if (!id) return '';
|
||||||
|
const trimmed = id.trim();
|
||||||
|
// 仅移除末尾的 _xxxx (4位字母数字)
|
||||||
|
const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/);
|
||||||
|
return suffixMatch ? suffixMatch[1] : trimmed;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (newSession.lastMsgSender && newSession.selfWxid) {
|
||||||
|
const senderClean = cleanWxid(newSession.lastMsgSender);
|
||||||
|
const selfClean = cleanWxid(newSession.selfWxid);
|
||||||
|
const match = senderClean === selfClean;
|
||||||
|
|
||||||
|
if (match) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 群聊显示发送者名字 (放在内容中: "Name: Message")
|
||||||
|
// 标题保持为群聊名称 (title 变量)
|
||||||
|
if (newSession.lastSenderDisplayName) {
|
||||||
|
content = `${newSession.lastSenderDisplayName}: ${content}`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 修复 "Random User" 的逻辑 (缺少具体信息)
|
||||||
|
// 如果标题看起来像 wxid 或没有头像,尝试获取信息
|
||||||
|
const needsEnrichment = !newSession.displayName || !newSession.avatarUrl || newSession.displayName === newSession.username
|
||||||
|
|
||||||
|
if (needsEnrichment && newSession.username) {
|
||||||
|
try {
|
||||||
|
// 尝试丰富或获取联系人详情
|
||||||
|
const contact = await window.electronAPI.chat.getContact(newSession.username)
|
||||||
|
if (contact) {
|
||||||
|
if (contact.remark || contact.nickname) {
|
||||||
|
title = contact.remark || contact.nickname
|
||||||
|
}
|
||||||
|
if (contact.avatarUrl) {
|
||||||
|
avatarUrl = contact.avatarUrl
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 如果不在缓存/数据库中
|
||||||
|
const enrichResult = await window.electronAPI.chat.enrichSessionsContactInfo([newSession.username])
|
||||||
|
if (enrichResult.success && enrichResult.contacts) {
|
||||||
|
const enrichedContact = enrichResult.contacts[newSession.username]
|
||||||
|
if (enrichedContact) {
|
||||||
|
if (enrichedContact.displayName) {
|
||||||
|
title = enrichedContact.displayName
|
||||||
|
}
|
||||||
|
if (enrichedContact.avatarUrl) {
|
||||||
|
avatarUrl = enrichedContact.avatarUrl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 如果仍然没有有效名称,再尝试一次获取
|
||||||
|
if (title === newSession.username || title.startsWith('wxid_')) {
|
||||||
|
const retried = await window.electronAPI.chat.getContact(newSession.username)
|
||||||
|
if (retried) {
|
||||||
|
title = retried.remark || retried.nickname || title
|
||||||
|
avatarUrl = retried.avatarUrl || avatarUrl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.warn('获取通知的联系人信息失败', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 最终检查:如果标题仍是 wxid 格式,则跳过通知(避免显示乱跳用户)
|
||||||
|
// 群聊例外,因为群聊 username 包含 @chatroom
|
||||||
|
const isGroupChat = newSession.username.includes('@chatroom')
|
||||||
|
const isWxidTitle = title.startsWith('wxid_') && title === newSession.username
|
||||||
|
if (isWxidTitle && !isGroupChat) {
|
||||||
|
console.warn('[NotificationFilter] 跳过无法识别的用户通知:', newSession.username)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// 调用 IPC 以显示独立窗口通知
|
||||||
|
window.electronAPI.notification?.show({
|
||||||
|
title: title,
|
||||||
|
content: content,
|
||||||
|
avatarUrl: avatarUrl,
|
||||||
|
sessionId: newSession.username
|
||||||
|
})
|
||||||
|
|
||||||
|
// 我们不再为 Toast 设置本地状态
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleActiveSessionRefresh = async (sessionId: string) => {
|
||||||
|
// 从 ChatPage 复制/调整的逻辑,以保持集中
|
||||||
|
const state = useChatStore.getState()
|
||||||
|
const lastMsg = state.messages[state.messages.length - 1]
|
||||||
|
const minTime = lastMsg?.createTime || 0
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await (window.electronAPI.chat as any).getNewMessages(sessionId, minTime)
|
||||||
|
if (result.success && result.messages && result.messages.length > 0) {
|
||||||
|
appendMessages(result.messages, false) // 追加到末尾
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.warn('后台活跃会话刷新失败:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 此组件不再渲染 UI
|
||||||
|
return null
|
||||||
|
}
|
||||||
46
src/components/ImagePreview.scss
Normal file
46
src/components/ImagePreview.scss
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
.image-preview-overlay {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0;
|
||||||
|
background: rgba(0, 0, 0, 0.9);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
z-index: 9999;
|
||||||
|
user-select: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.preview-image {
|
||||||
|
max-width: 90vw;
|
||||||
|
max-height: 90vh;
|
||||||
|
object-fit: contain;
|
||||||
|
transition: transform 0.15s ease-out;
|
||||||
|
|
||||||
|
&.dragging {
|
||||||
|
transition: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-preview-close {
|
||||||
|
position: absolute;
|
||||||
|
bottom: 40px;
|
||||||
|
left: 50%;
|
||||||
|
transform: translateX(-50%);
|
||||||
|
width: 48px;
|
||||||
|
height: 48px;
|
||||||
|
border-radius: 50%;
|
||||||
|
border: 2px solid rgba(255, 255, 255, 0.3);
|
||||||
|
background: rgba(0, 0, 0, 0.7);
|
||||||
|
color: #fff;
|
||||||
|
cursor: pointer;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
transition: all 0.2s;
|
||||||
|
backdrop-filter: blur(10px);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(255, 255, 255, 0.2);
|
||||||
|
border-color: rgba(255, 255, 255, 0.5);
|
||||||
|
transform: translateX(-50%) scale(1.1);
|
||||||
|
}
|
||||||
|
}
|
||||||
101
src/components/ImagePreview.tsx
Normal file
101
src/components/ImagePreview.tsx
Normal file
@@ -0,0 +1,101 @@
|
|||||||
|
import React, { useState, useRef, useCallback, useEffect } from 'react'
|
||||||
|
import { X } from 'lucide-react'
|
||||||
|
import { createPortal } from 'react-dom'
|
||||||
|
import './ImagePreview.scss'
|
||||||
|
|
||||||
|
interface ImagePreviewProps {
|
||||||
|
src: string
|
||||||
|
onClose: () => void
|
||||||
|
}
|
||||||
|
|
||||||
|
export const ImagePreview: React.FC<ImagePreviewProps> = ({ src, onClose }) => {
|
||||||
|
const [scale, setScale] = useState(1)
|
||||||
|
const [position, setPosition] = useState({ x: 0, y: 0 })
|
||||||
|
const [isDragging, setIsDragging] = useState(false)
|
||||||
|
const dragStart = useRef({ x: 0, y: 0 })
|
||||||
|
const positionStart = useRef({ x: 0, y: 0 })
|
||||||
|
const containerRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
|
// 滚轮缩放
|
||||||
|
const handleWheel = useCallback((e: React.WheelEvent) => {
|
||||||
|
e.preventDefault()
|
||||||
|
const delta = e.deltaY > 0 ? 0.9 : 1.1
|
||||||
|
setScale(prev => Math.min(Math.max(prev * delta, 0.5), 5))
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
// 开始拖动
|
||||||
|
const handleMouseDown = useCallback((e: React.MouseEvent) => {
|
||||||
|
if (scale <= 1) return
|
||||||
|
e.preventDefault()
|
||||||
|
setIsDragging(true)
|
||||||
|
dragStart.current = { x: e.clientX, y: e.clientY }
|
||||||
|
positionStart.current = { ...position }
|
||||||
|
}, [scale, position])
|
||||||
|
|
||||||
|
// 拖动中
|
||||||
|
const handleMouseMove = useCallback((e: React.MouseEvent) => {
|
||||||
|
if (!isDragging) return
|
||||||
|
const dx = e.clientX - dragStart.current.x
|
||||||
|
const dy = e.clientY - dragStart.current.y
|
||||||
|
setPosition({
|
||||||
|
x: positionStart.current.x + dx,
|
||||||
|
y: positionStart.current.y + dy
|
||||||
|
})
|
||||||
|
}, [isDragging])
|
||||||
|
|
||||||
|
// 结束拖动
|
||||||
|
const handleMouseUp = useCallback(() => {
|
||||||
|
setIsDragging(false)
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
// 双击重置
|
||||||
|
const handleDoubleClick = useCallback(() => {
|
||||||
|
setScale(1)
|
||||||
|
setPosition({ x: 0, y: 0 })
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
// 点击背景关闭
|
||||||
|
const handleOverlayClick = useCallback((e: React.MouseEvent) => {
|
||||||
|
if (e.target === containerRef.current) {
|
||||||
|
onClose()
|
||||||
|
}
|
||||||
|
}, [onClose])
|
||||||
|
|
||||||
|
// ESC 关闭
|
||||||
|
useEffect(() => {
|
||||||
|
const handleKeyDown = (e: KeyboardEvent) => {
|
||||||
|
if (e.key === 'Escape') onClose()
|
||||||
|
}
|
||||||
|
window.addEventListener('keydown', handleKeyDown)
|
||||||
|
return () => window.removeEventListener('keydown', handleKeyDown)
|
||||||
|
}, [onClose])
|
||||||
|
|
||||||
|
return createPortal(
|
||||||
|
<div
|
||||||
|
ref={containerRef}
|
||||||
|
className="image-preview-overlay"
|
||||||
|
onClick={handleOverlayClick}
|
||||||
|
onMouseMove={handleMouseMove}
|
||||||
|
onMouseUp={handleMouseUp}
|
||||||
|
onMouseLeave={handleMouseUp}
|
||||||
|
>
|
||||||
|
<img
|
||||||
|
src={src}
|
||||||
|
alt="图片预览"
|
||||||
|
className={`preview-image ${isDragging ? 'dragging' : ''}`}
|
||||||
|
style={{
|
||||||
|
transform: `translate(${position.x}px, ${position.y}px) scale(${scale})`,
|
||||||
|
cursor: scale > 1 ? (isDragging ? 'grabbing' : 'grab') : 'default'
|
||||||
|
}}
|
||||||
|
onWheel={handleWheel}
|
||||||
|
onMouseDown={handleMouseDown}
|
||||||
|
onDoubleClick={handleDoubleClick}
|
||||||
|
draggable={false}
|
||||||
|
/>
|
||||||
|
<button className="image-preview-close" onClick={onClose}>
|
||||||
|
<X size={20} />
|
||||||
|
</button>
|
||||||
|
</div>,
|
||||||
|
document.body
|
||||||
|
)
|
||||||
|
}
|
||||||
238
src/components/JumpToDateDialog.scss
Normal file
238
src/components/JumpToDateDialog.scss
Normal file
@@ -0,0 +1,238 @@
|
|||||||
|
.jump-date-overlay {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0;
|
||||||
|
background: rgba(0, 0, 0, 0.4);
|
||||||
|
backdrop-filter: blur(4px);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
z-index: 2000;
|
||||||
|
animation: fadeIn 0.2s ease-out;
|
||||||
|
}
|
||||||
|
|
||||||
|
.jump-date-modal {
|
||||||
|
background: var(--card-bg);
|
||||||
|
width: 340px;
|
||||||
|
border-radius: 16px;
|
||||||
|
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.25);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
overflow: hidden;
|
||||||
|
animation: modalSlideUp 0.3s cubic-bezier(0.16, 1, 0.3, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.jump-date-header {
|
||||||
|
padding: 18px 20px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
|
.title-area {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.close-btn {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
cursor: pointer;
|
||||||
|
padding: 4px;
|
||||||
|
border-radius: 6px;
|
||||||
|
display: flex;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.calendar-view {
|
||||||
|
padding: 20px;
|
||||||
|
|
||||||
|
.calendar-nav {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
|
||||||
|
.current-month {
|
||||||
|
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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.calendar-grid {
|
||||||
|
.weekdays {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(7, 1fr);
|
||||||
|
margin-bottom: 8px;
|
||||||
|
|
||||||
|
.weekday {
|
||||||
|
text-align: center;
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
padding: 4px 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.days {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(7, 1fr);
|
||||||
|
gap: 4px;
|
||||||
|
|
||||||
|
.day-cell {
|
||||||
|
aspect-ratio: 1;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&.empty {
|
||||||
|
cursor: default;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:not(.empty):hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.selected {
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.today:not(.selected) {
|
||||||
|
color: var(--primary);
|
||||||
|
font-weight: 600;
|
||||||
|
background: var(--primary-light);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.quick-options {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 0 20px 16px;
|
||||||
|
|
||||||
|
button {
|
||||||
|
flex: 1;
|
||||||
|
padding: 8px;
|
||||||
|
font-size: 12px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 6px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--primary);
|
||||||
|
border-color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-footer {
|
||||||
|
padding: 16px 20px;
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
|
||||||
|
button {
|
||||||
|
flex: 1;
|
||||||
|
padding: 10px;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.cancel-btn {
|
||||||
|
background: transparent;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.confirm-btn {
|
||||||
|
background: var(--primary);
|
||||||
|
border: none;
|
||||||
|
color: #fff;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--primary-hover);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes modalSlideUp {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
156
src/components/JumpToDateDialog.tsx
Normal file
156
src/components/JumpToDateDialog.tsx
Normal file
@@ -0,0 +1,156 @@
|
|||||||
|
import React, { useState } from 'react'
|
||||||
|
import { X, ChevronLeft, ChevronRight, Calendar as CalendarIcon } from 'lucide-react'
|
||||||
|
import './JumpToDateDialog.scss'
|
||||||
|
|
||||||
|
interface JumpToDateDialogProps {
|
||||||
|
isOpen: boolean
|
||||||
|
onClose: () => void
|
||||||
|
onSelect: (date: Date) => void
|
||||||
|
currentDate?: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
const JumpToDateDialog: React.FC<JumpToDateDialogProps> = ({
|
||||||
|
isOpen,
|
||||||
|
onClose,
|
||||||
|
onSelect,
|
||||||
|
currentDate = new Date()
|
||||||
|
}) => {
|
||||||
|
const [calendarDate, setCalendarDate] = useState(new Date(currentDate))
|
||||||
|
const [selectedDate, setSelectedDate] = useState(new Date(currentDate))
|
||||||
|
|
||||||
|
if (!isOpen) return null
|
||||||
|
|
||||||
|
const getDaysInMonth = (date: Date) => {
|
||||||
|
const year = date.getFullYear()
|
||||||
|
const month = date.getMonth()
|
||||||
|
return new Date(year, month + 1, 0).getDate()
|
||||||
|
}
|
||||||
|
|
||||||
|
const getFirstDayOfMonth = (date: Date) => {
|
||||||
|
const year = date.getFullYear()
|
||||||
|
const month = date.getMonth()
|
||||||
|
return new Date(year, month, 1).getDay()
|
||||||
|
}
|
||||||
|
|
||||||
|
const generateCalendar = () => {
|
||||||
|
const daysInMonth = getDaysInMonth(calendarDate)
|
||||||
|
const firstDay = getFirstDayOfMonth(calendarDate)
|
||||||
|
const days: (number | null)[] = []
|
||||||
|
|
||||||
|
for (let i = 0; i < firstDay; i++) {
|
||||||
|
days.push(null)
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let i = 1; i <= daysInMonth; i++) {
|
||||||
|
days.push(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
return days
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleDateClick = (day: number) => {
|
||||||
|
const newDate = new Date(calendarDate.getFullYear(), calendarDate.getMonth(), day)
|
||||||
|
setSelectedDate(newDate)
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleConfirm = () => {
|
||||||
|
onSelect(selectedDate)
|
||||||
|
onClose()
|
||||||
|
}
|
||||||
|
|
||||||
|
const isToday = (day: number) => {
|
||||||
|
const today = new Date()
|
||||||
|
return day === today.getDate() &&
|
||||||
|
calendarDate.getMonth() === today.getMonth() &&
|
||||||
|
calendarDate.getFullYear() === today.getFullYear()
|
||||||
|
}
|
||||||
|
|
||||||
|
const isSelected = (day: number) => {
|
||||||
|
return day === selectedDate.getDate() &&
|
||||||
|
calendarDate.getMonth() === selectedDate.getMonth() &&
|
||||||
|
calendarDate.getFullYear() === selectedDate.getFullYear()
|
||||||
|
}
|
||||||
|
|
||||||
|
const weekdays = ['日', '一', '二', '三', '四', '五', '六']
|
||||||
|
const days = generateCalendar()
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="jump-date-overlay" onClick={onClose}>
|
||||||
|
<div className="jump-date-modal" onClick={e => e.stopPropagation()}>
|
||||||
|
<div className="jump-date-header">
|
||||||
|
<div className="title-area">
|
||||||
|
<CalendarIcon size={18} />
|
||||||
|
<h3>跳转到日期</h3>
|
||||||
|
</div>
|
||||||
|
<button className="close-btn" onClick={onClose}>
|
||||||
|
<X size={18} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="calendar-view">
|
||||||
|
<div className="calendar-nav">
|
||||||
|
<button
|
||||||
|
className="nav-btn"
|
||||||
|
onClick={() => setCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() - 1, 1))}
|
||||||
|
>
|
||||||
|
<ChevronLeft size={18} />
|
||||||
|
</button>
|
||||||
|
<span className="current-month">
|
||||||
|
{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月
|
||||||
|
</span>
|
||||||
|
<button
|
||||||
|
className="nav-btn"
|
||||||
|
onClick={() => setCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() + 1, 1))}
|
||||||
|
>
|
||||||
|
<ChevronRight size={18} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="calendar-grid">
|
||||||
|
<div className="weekdays">
|
||||||
|
{weekdays.map(d => <div key={d} className="weekday">{d}</div>)}
|
||||||
|
</div>
|
||||||
|
<div className="days">
|
||||||
|
{days.map((day, i) => (
|
||||||
|
<div
|
||||||
|
key={i}
|
||||||
|
className={`day-cell ${day === null ? 'empty' : ''} ${day !== null && isSelected(day) ? 'selected' : ''} ${day !== null && isToday(day) ? 'today' : ''}`}
|
||||||
|
onClick={() => day !== null && handleDateClick(day)}
|
||||||
|
>
|
||||||
|
{day}
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="quick-options">
|
||||||
|
<button onClick={() => {
|
||||||
|
const d = new Date()
|
||||||
|
setSelectedDate(d)
|
||||||
|
setCalendarDate(new Date(d))
|
||||||
|
}}>今天</button>
|
||||||
|
<button onClick={() => {
|
||||||
|
const d = new Date()
|
||||||
|
d.setDate(d.getDate() - 7)
|
||||||
|
setSelectedDate(d)
|
||||||
|
setCalendarDate(new Date(d))
|
||||||
|
}}>一周前</button>
|
||||||
|
<button onClick={() => {
|
||||||
|
const d = new Date()
|
||||||
|
d.setMonth(d.getMonth() - 1)
|
||||||
|
setSelectedDate(d)
|
||||||
|
setCalendarDate(new Date(d))
|
||||||
|
}}>一月前</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="dialog-footer">
|
||||||
|
<button className="cancel-btn" onClick={onClose}>取消</button>
|
||||||
|
<button className="confirm-btn" onClick={handleConfirm}>跳转</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default JumpToDateDialog
|
||||||
29
src/components/LivePhotoIcon.tsx
Normal file
29
src/components/LivePhotoIcon.tsx
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
import React from 'react';
|
||||||
|
|
||||||
|
interface LivePhotoIconProps {
|
||||||
|
size?: number | string;
|
||||||
|
className?: string;
|
||||||
|
style?: React.CSSProperties;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const LivePhotoIcon: React.FC<LivePhotoIconProps> = ({ size = 24, className = '', style = {} }) => {
|
||||||
|
return (
|
||||||
|
<svg
|
||||||
|
width={size}
|
||||||
|
height={size}
|
||||||
|
viewBox="0 0 24 24"
|
||||||
|
version="1.1"
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
className={className}
|
||||||
|
style={style}
|
||||||
|
>
|
||||||
|
<g stroke="none" strokeWidth="1" fill="none" fillRule="evenodd" strokeLinecap="round" strokeLinejoin="round">
|
||||||
|
<g stroke="currentColor" strokeWidth="2">
|
||||||
|
<circle fill="currentColor" stroke="none" cx="12" cy="12" r="2.5"></circle>
|
||||||
|
<circle cx="12" cy="12" r="5.5"></circle>
|
||||||
|
<circle cx="12" cy="12" r="9" strokeDasharray="1 3.7"></circle>
|
||||||
|
</g>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
);
|
||||||
|
};
|
||||||
185
src/components/LockScreen.scss
Normal file
185
src/components/LockScreen.scss
Normal file
@@ -0,0 +1,185 @@
|
|||||||
|
.lock-screen {
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100vw;
|
||||||
|
height: 100vh;
|
||||||
|
background-color: var(--bg-primary);
|
||||||
|
z-index: 9999;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
user-select: none;
|
||||||
|
-webkit-app-region: drag;
|
||||||
|
transition: all 0.5s cubic-bezier(0.22, 1, 0.36, 1);
|
||||||
|
backdrop-filter: blur(25px) saturate(180%);
|
||||||
|
background-color: var(--bg-primary);
|
||||||
|
// 让背景带一点透明度以增强毛玻璃效果
|
||||||
|
opacity: 1;
|
||||||
|
|
||||||
|
&.unlocked {
|
||||||
|
opacity: 0;
|
||||||
|
pointer-events: none;
|
||||||
|
backdrop-filter: blur(0) saturate(100%);
|
||||||
|
transform: scale(1.02);
|
||||||
|
|
||||||
|
.lock-content {
|
||||||
|
transform: translateY(-20px) scale(0.95);
|
||||||
|
filter: blur(10px);
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.lock-content {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
width: 320px;
|
||||||
|
-webkit-app-region: no-drag;
|
||||||
|
animation: fadeIn 0.5s cubic-bezier(0.4, 0, 0.2, 1);
|
||||||
|
transition: all 0.4s cubic-bezier(0.4, 0, 0.2, 1);
|
||||||
|
|
||||||
|
.lock-avatar {
|
||||||
|
width: 100px;
|
||||||
|
height: 100px;
|
||||||
|
border-radius: 50%;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1);
|
||||||
|
border: 4px solid var(--bg-total);
|
||||||
|
background-color: var(--bg-secondary);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.lock-title {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin-bottom: 32px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lock-form {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 16px;
|
||||||
|
|
||||||
|
.input-group {
|
||||||
|
position: relative;
|
||||||
|
width: 100%;
|
||||||
|
|
||||||
|
input {
|
||||||
|
width: 100%;
|
||||||
|
height: 48px;
|
||||||
|
padding: 0 16px;
|
||||||
|
padding-right: 48px;
|
||||||
|
border-radius: 12px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background-color: var(--bg-input);
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 16px;
|
||||||
|
outline: none;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
border-color: var(--primary-color);
|
||||||
|
box-shadow: 0 0 0 2px var(--primary-color-alpha);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.submit-btn {
|
||||||
|
position: absolute;
|
||||||
|
right: 8px;
|
||||||
|
top: 8px;
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
border-radius: 8px;
|
||||||
|
border: none;
|
||||||
|
background: var(--primary-color);
|
||||||
|
color: white;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: opacity 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.hello-btn {
|
||||||
|
width: 100%;
|
||||||
|
height: 48px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 8px;
|
||||||
|
border-radius: 12px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background-color: var(--bg-secondary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background-color: var(--bg-hover);
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.loading {
|
||||||
|
opacity: 0.7;
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.lock-error {
|
||||||
|
margin-top: 16px;
|
||||||
|
color: #ff4d4f;
|
||||||
|
font-size: 14px;
|
||||||
|
animation: shake 0.5s ease-in-out;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes shake {
|
||||||
|
|
||||||
|
0%,
|
||||||
|
100% {
|
||||||
|
transform: translateX(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
10%,
|
||||||
|
30%,
|
||||||
|
50%,
|
||||||
|
70%,
|
||||||
|
90% {
|
||||||
|
transform: translateX(-4px);
|
||||||
|
}
|
||||||
|
|
||||||
|
20%,
|
||||||
|
40%,
|
||||||
|
60%,
|
||||||
|
80% {
|
||||||
|
transform: translateX(4px);
|
||||||
|
}
|
||||||
|
}
|
||||||
169
src/components/LockScreen.tsx
Normal file
169
src/components/LockScreen.tsx
Normal file
@@ -0,0 +1,169 @@
|
|||||||
|
import { useState, useEffect, useRef } from 'react'
|
||||||
|
import * as configService from '../services/config'
|
||||||
|
import { ArrowRight, Fingerprint, Lock, ScanFace, ShieldCheck } from 'lucide-react'
|
||||||
|
import './LockScreen.scss'
|
||||||
|
|
||||||
|
interface LockScreenProps {
|
||||||
|
onUnlock: () => void
|
||||||
|
avatar?: string
|
||||||
|
useHello?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
async function sha256(message: string) {
|
||||||
|
const msgBuffer = new TextEncoder().encode(message)
|
||||||
|
const hashBuffer = await crypto.subtle.digest('SHA-256', msgBuffer)
|
||||||
|
const hashArray = Array.from(new Uint8Array(hashBuffer))
|
||||||
|
const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join('')
|
||||||
|
return hashHex
|
||||||
|
}
|
||||||
|
|
||||||
|
export default function LockScreen({ onUnlock, avatar, useHello = false }: LockScreenProps) {
|
||||||
|
const [password, setPassword] = useState('')
|
||||||
|
const [error, setError] = useState('')
|
||||||
|
const [isVerifying, setIsVerifying] = useState(false)
|
||||||
|
const [isUnlocked, setIsUnlocked] = useState(false)
|
||||||
|
const [showHello, setShowHello] = useState(false)
|
||||||
|
const [helloAvailable, setHelloAvailable] = useState(false)
|
||||||
|
|
||||||
|
// 用于取消 WebAuthn 请求
|
||||||
|
const abortControllerRef = useRef<AbortController | null>(null)
|
||||||
|
const inputRef = useRef<HTMLInputElement>(null)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
// 快速检查配置并启动
|
||||||
|
quickStartHello()
|
||||||
|
inputRef.current?.focus()
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
// 组件卸载时取消请求
|
||||||
|
abortControllerRef.current?.abort()
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const handleUnlock = () => {
|
||||||
|
setIsUnlocked(true)
|
||||||
|
setTimeout(() => {
|
||||||
|
onUnlock()
|
||||||
|
}, 1500)
|
||||||
|
}
|
||||||
|
|
||||||
|
const quickStartHello = async () => {
|
||||||
|
try {
|
||||||
|
// 如果父组件已经告诉我们要用 Hello,直接开始,不等待 IPC
|
||||||
|
let shouldUseHello = useHello
|
||||||
|
|
||||||
|
// 为了稳健,如果 prop 没传(虽然现在都传了),再 check 一次 config
|
||||||
|
if (!shouldUseHello) {
|
||||||
|
shouldUseHello = await configService.getAuthUseHello()
|
||||||
|
}
|
||||||
|
|
||||||
|
if (shouldUseHello) {
|
||||||
|
// 标记为可用,显示按钮
|
||||||
|
setHelloAvailable(true)
|
||||||
|
setShowHello(true)
|
||||||
|
// 立即执行验证 (0延迟)
|
||||||
|
verifyHello()
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('Quick start hello failed', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const verifyHello = async () => {
|
||||||
|
if (isVerifying || isUnlocked) return
|
||||||
|
|
||||||
|
setIsVerifying(true)
|
||||||
|
setError('')
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.auth.hello()
|
||||||
|
|
||||||
|
if (result.success) {
|
||||||
|
handleUnlock()
|
||||||
|
} else {
|
||||||
|
console.error('Hello verification failed:', result.error)
|
||||||
|
setError(result.error || '验证失败')
|
||||||
|
}
|
||||||
|
} catch (e: any) {
|
||||||
|
console.error('Hello verification error:', e)
|
||||||
|
setError(`验证失败: ${e.message || String(e)}`)
|
||||||
|
} finally {
|
||||||
|
setIsVerifying(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handlePasswordSubmit = async (e?: React.FormEvent) => {
|
||||||
|
e?.preventDefault()
|
||||||
|
if (!password || isUnlocked) return
|
||||||
|
|
||||||
|
// 如果正在进行 Hello 验证,它会自动失败或被取代,UI上不用特意取消
|
||||||
|
// 因为 native 调用是模态的或者独立的,我们只要让 JS 状态不对锁住即可
|
||||||
|
|
||||||
|
// 不再检查 isVerifying,因为我们允许打断 Hello
|
||||||
|
setIsVerifying(true)
|
||||||
|
setError('')
|
||||||
|
|
||||||
|
try {
|
||||||
|
const storedHash = await configService.getAuthPassword()
|
||||||
|
const inputHash = await sha256(password)
|
||||||
|
|
||||||
|
if (inputHash === storedHash) {
|
||||||
|
handleUnlock()
|
||||||
|
} else {
|
||||||
|
setError('密码错误')
|
||||||
|
setPassword('')
|
||||||
|
setIsVerifying(false)
|
||||||
|
// 如果密码错误,是否重新触发 Hello?
|
||||||
|
// 用户可能想重试密码,暂时不自动触发
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
setError('验证失败')
|
||||||
|
setIsVerifying(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className={`lock-screen ${isUnlocked ? 'unlocked' : ''}`}>
|
||||||
|
<div className="lock-content">
|
||||||
|
<div className="lock-avatar">
|
||||||
|
{avatar ? (
|
||||||
|
<img src={avatar} alt="User" style={{ width: '100%', height: '100%', borderRadius: '50%' }} />
|
||||||
|
) : (
|
||||||
|
<Lock size={40} />
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<h2 className="lock-title">WeFlow 已锁定</h2>
|
||||||
|
|
||||||
|
<form className="lock-form" onSubmit={handlePasswordSubmit}>
|
||||||
|
<div className="input-group">
|
||||||
|
<input
|
||||||
|
ref={inputRef}
|
||||||
|
type="password"
|
||||||
|
placeholder="输入应用密码"
|
||||||
|
value={password}
|
||||||
|
onChange={(e) => setPassword(e.target.value)}
|
||||||
|
// 移除 disabled,允许用户随时输入
|
||||||
|
/>
|
||||||
|
<button type="submit" className="submit-btn" disabled={!password}>
|
||||||
|
<ArrowRight size={18} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{showHello && (
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`hello-btn ${isVerifying ? 'loading' : ''}`}
|
||||||
|
onClick={verifyHello}
|
||||||
|
>
|
||||||
|
<Fingerprint size={20} />
|
||||||
|
{isVerifying ? '验证中...' : '使用 Windows Hello 解锁'}
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</form>
|
||||||
|
|
||||||
|
{error && <div className="lock-error">{error}</div>}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
36
src/components/MessageBubble.tsx
Normal file
36
src/components/MessageBubble.tsx
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
import React from 'react'
|
||||||
|
import { Bot, User } from 'lucide-react'
|
||||||
|
|
||||||
|
interface ChatMessage {
|
||||||
|
id: string;
|
||||||
|
role: 'user' | 'ai';
|
||||||
|
content: string;
|
||||||
|
timestamp: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface MessageBubbleProps {
|
||||||
|
message: ChatMessage;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 优化后的消息气泡组件
|
||||||
|
* 使用 React.memo 避免不必要的重新渲染
|
||||||
|
*/
|
||||||
|
export const MessageBubble = React.memo<MessageBubbleProps>(({ message }) => {
|
||||||
|
return (
|
||||||
|
<div className={`message-row ${message.role}`}>
|
||||||
|
<div className="avatar">
|
||||||
|
{message.role === 'ai' ? <Bot size={24} /> : <User size={24} />}
|
||||||
|
</div>
|
||||||
|
<div className="bubble">
|
||||||
|
<div className="content">{message.content}</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}, (prevProps, nextProps) => {
|
||||||
|
// 自定义比较函数:只有内容或ID变化时才重新渲染
|
||||||
|
return prevProps.message.content === nextProps.message.content &&
|
||||||
|
prevProps.message.id === nextProps.message.id
|
||||||
|
})
|
||||||
|
|
||||||
|
MessageBubble.displayName = 'MessageBubble'
|
||||||
200
src/components/NotificationToast.scss
Normal file
200
src/components/NotificationToast.scss
Normal file
@@ -0,0 +1,200 @@
|
|||||||
|
.notification-toast-container {
|
||||||
|
position: fixed;
|
||||||
|
z-index: 9999;
|
||||||
|
width: 320px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
backdrop-filter: blur(20px);
|
||||||
|
-webkit-backdrop-filter: blur(20px);
|
||||||
|
border: 1px solid var(--border-light);
|
||||||
|
border-radius: 12px;
|
||||||
|
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.12);
|
||||||
|
padding: 12px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
|
||||||
|
opacity: 0;
|
||||||
|
transform: scale(0.95);
|
||||||
|
pointer-events: none; // Allow clicking through when hidden
|
||||||
|
|
||||||
|
&.visible {
|
||||||
|
opacity: 1;
|
||||||
|
transform: scale(1);
|
||||||
|
pointer-events: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.static {
|
||||||
|
position: relative !important;
|
||||||
|
width: calc(100% - 4px) !important; // Leave 2px margin for anti-aliasing saftey
|
||||||
|
height: auto !important; // Fits content
|
||||||
|
min-height: 0;
|
||||||
|
top: 0 !important;
|
||||||
|
bottom: 0 !important;
|
||||||
|
left: 0 !important;
|
||||||
|
right: 0 !important;
|
||||||
|
transform: none !important;
|
||||||
|
margin: 2px !important; // 2px centered margin
|
||||||
|
border-radius: 12px !important; // Rounded corners
|
||||||
|
|
||||||
|
|
||||||
|
// Disable backdrop filter
|
||||||
|
backdrop-filter: none !important;
|
||||||
|
-webkit-backdrop-filter: none !important;
|
||||||
|
|
||||||
|
// Ensure background is solid
|
||||||
|
background: var(--bg-secondary, #2c2c2c);
|
||||||
|
color: var(--text-primary, #ffffff);
|
||||||
|
|
||||||
|
box-shadow: none !important; // NO SHADOW
|
||||||
|
border: 1px solid var(--border-light, rgba(255, 255, 255, 0.1));
|
||||||
|
|
||||||
|
display: flex;
|
||||||
|
padding: 16px;
|
||||||
|
padding-right: 32px; // Make space for close button
|
||||||
|
box-sizing: border-box;
|
||||||
|
|
||||||
|
// Force close button to be visible but transparent background
|
||||||
|
.notification-close {
|
||||||
|
opacity: 1 !important;
|
||||||
|
top: 12px;
|
||||||
|
right: 12px;
|
||||||
|
background: transparent !important; // Transparent per user request
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
color: var(--text-primary);
|
||||||
|
background: rgba(255, 255, 255, 0.1) !important; // Subtle hover effect
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-time {
|
||||||
|
top: 24px; // Match padding
|
||||||
|
right: 40px; // Left of close button (12px + 20px + 8px)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Position variants
|
||||||
|
&.bottom-right {
|
||||||
|
bottom: 24px;
|
||||||
|
right: 24px;
|
||||||
|
transform: translate(0, 20px) scale(0.95);
|
||||||
|
|
||||||
|
&.visible {
|
||||||
|
transform: translate(0, 0) scale(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.top-right {
|
||||||
|
top: 24px;
|
||||||
|
right: 24px;
|
||||||
|
transform: translate(0, -20px) scale(0.95);
|
||||||
|
|
||||||
|
&.visible {
|
||||||
|
transform: translate(0, 0) scale(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.bottom-left {
|
||||||
|
bottom: 24px;
|
||||||
|
left: 24px;
|
||||||
|
transform: translate(0, 20px) scale(0.95);
|
||||||
|
|
||||||
|
&.visible {
|
||||||
|
transform: translate(0, 0) scale(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.top-left {
|
||||||
|
top: 24px;
|
||||||
|
left: 24px;
|
||||||
|
transform: translate(0, -20px) scale(0.95);
|
||||||
|
|
||||||
|
&.visible {
|
||||||
|
transform: translate(0, 0) scale(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
box-shadow: 0 12px 48px rgba(0, 0, 0, 0.16) !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-content {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-avatar {
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-text {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
|
||||||
|
.notification-header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
margin-bottom: 4px;
|
||||||
|
|
||||||
|
.notification-title {
|
||||||
|
font-weight: 600;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
max-width: 100%; // 允许缩放
|
||||||
|
flex: 1; // 占据剩余空间
|
||||||
|
min-width: 0; // 关键:允许 flex 子项收缩到内容以下
|
||||||
|
margin-right: 60px; // Make space for absolute time + close button
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-time {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
position: absolute;
|
||||||
|
top: 16px;
|
||||||
|
right: 36px; // Left of close button (8px + 20px + 8px)
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-body {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
line-height: 1.4;
|
||||||
|
display: -webkit-box;
|
||||||
|
-webkit-line-clamp: 2;
|
||||||
|
line-clamp: 2;
|
||||||
|
-webkit-box-orient: vertical;
|
||||||
|
overflow: hidden;
|
||||||
|
word-break: break-all;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification-close {
|
||||||
|
position: absolute;
|
||||||
|
top: 8px;
|
||||||
|
right: 8px;
|
||||||
|
width: 20px;
|
||||||
|
height: 20px;
|
||||||
|
border-radius: 50%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
border: none;
|
||||||
|
background: transparent;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
cursor: pointer;
|
||||||
|
opacity: 0;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&:hover .notification-close {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
108
src/components/NotificationToast.tsx
Normal file
108
src/components/NotificationToast.tsx
Normal file
@@ -0,0 +1,108 @@
|
|||||||
|
import React, { useEffect, useState } from 'react'
|
||||||
|
import { createPortal } from 'react-dom'
|
||||||
|
import { X } from 'lucide-react'
|
||||||
|
import { Avatar } from './Avatar'
|
||||||
|
import './NotificationToast.scss'
|
||||||
|
|
||||||
|
export interface NotificationData {
|
||||||
|
id: string
|
||||||
|
sessionId: string
|
||||||
|
avatarUrl?: string
|
||||||
|
title: string
|
||||||
|
content: string
|
||||||
|
timestamp: number
|
||||||
|
}
|
||||||
|
|
||||||
|
interface NotificationToastProps {
|
||||||
|
data: NotificationData | null
|
||||||
|
onClose: () => void
|
||||||
|
onClick: (sessionId: string) => void
|
||||||
|
duration?: number
|
||||||
|
position?: 'top-right' | 'top-left' | 'bottom-right' | 'bottom-left'
|
||||||
|
isStatic?: boolean
|
||||||
|
initialVisible?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
export function NotificationToast({
|
||||||
|
data,
|
||||||
|
onClose,
|
||||||
|
onClick,
|
||||||
|
duration = 5000,
|
||||||
|
position = 'top-right',
|
||||||
|
isStatic = false,
|
||||||
|
initialVisible = false
|
||||||
|
}: NotificationToastProps) {
|
||||||
|
const [isVisible, setIsVisible] = useState(initialVisible)
|
||||||
|
const [currentData, setCurrentData] = useState<NotificationData | null>(null)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (data) {
|
||||||
|
setCurrentData(data)
|
||||||
|
setIsVisible(true)
|
||||||
|
|
||||||
|
const timer = setTimeout(() => {
|
||||||
|
setIsVisible(false)
|
||||||
|
// clean up data after animation
|
||||||
|
setTimeout(onClose, 300)
|
||||||
|
}, duration)
|
||||||
|
|
||||||
|
return () => clearTimeout(timer)
|
||||||
|
} else {
|
||||||
|
setIsVisible(false)
|
||||||
|
}
|
||||||
|
}, [data, duration, onClose])
|
||||||
|
|
||||||
|
if (!currentData) return null
|
||||||
|
|
||||||
|
const handleClose = (e: React.MouseEvent) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
setIsVisible(false)
|
||||||
|
setTimeout(onClose, 300)
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleClick = () => {
|
||||||
|
setIsVisible(false)
|
||||||
|
setTimeout(() => {
|
||||||
|
onClose()
|
||||||
|
onClick(currentData.sessionId)
|
||||||
|
}, 300)
|
||||||
|
}
|
||||||
|
|
||||||
|
const content = (
|
||||||
|
<div
|
||||||
|
className={`notification-toast-container ${position} ${isVisible ? 'visible' : ''} ${isStatic ? 'static' : ''}`}
|
||||||
|
onClick={handleClick}
|
||||||
|
>
|
||||||
|
<div className="notification-content">
|
||||||
|
<div className="notification-avatar">
|
||||||
|
<Avatar
|
||||||
|
src={currentData.avatarUrl}
|
||||||
|
name={currentData.title}
|
||||||
|
size={40}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div className="notification-text">
|
||||||
|
<div className="notification-header">
|
||||||
|
<span className="notification-title">{currentData.title}</span>
|
||||||
|
<span className="notification-time">
|
||||||
|
{new Date(currentData.timestamp * 1000).toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' })}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<div className="notification-body">
|
||||||
|
{currentData.content}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<button className="notification-close" onClick={handleClose}>
|
||||||
|
<X size={14} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
|
||||||
|
if (isStatic) {
|
||||||
|
return content
|
||||||
|
}
|
||||||
|
|
||||||
|
// Portal to document.body to ensure it's on top
|
||||||
|
return createPortal(content, document.body)
|
||||||
|
}
|
||||||
@@ -6,8 +6,7 @@ interface RouteGuardProps {
|
|||||||
children: React.ReactNode
|
children: React.ReactNode
|
||||||
}
|
}
|
||||||
|
|
||||||
// 不需要数据库连接的页面
|
const PUBLIC_ROUTES = ['/', '/home', '/settings']
|
||||||
const PUBLIC_ROUTES = ['/', '/home', '/settings', '/data-management']
|
|
||||||
|
|
||||||
function RouteGuard({ children }: RouteGuardProps) {
|
function RouteGuard({ children }: RouteGuardProps) {
|
||||||
const navigate = useNavigate()
|
const navigate = useNavigate()
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
.sidebar {
|
.sidebar {
|
||||||
width: 200px;
|
width: 220px;
|
||||||
background: var(--bg-secondary);
|
background: var(--bg-secondary);
|
||||||
border-right: 1px solid var(--border-color);
|
border-right: 1px solid var(--border-color);
|
||||||
display: flex;
|
display: flex;
|
||||||
@@ -32,14 +32,14 @@
|
|||||||
display: flex;
|
display: flex;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
gap: 4px;
|
gap: 4px;
|
||||||
padding: 0 8px;
|
padding: 0 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.nav-item {
|
.nav-item {
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
gap: 12px;
|
gap: 12px;
|
||||||
padding: 10px 16px;
|
padding: 10px 12px;
|
||||||
border-radius: 9999px;
|
border-radius: 9999px;
|
||||||
color: var(--text-secondary);
|
color: var(--text-secondary);
|
||||||
text-decoration: none;
|
text-decoration: none;
|
||||||
@@ -49,7 +49,6 @@
|
|||||||
background: transparent;
|
background: transparent;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
font-family: inherit;
|
font-family: inherit;
|
||||||
width: 100%;
|
|
||||||
|
|
||||||
&:hover {
|
&:hover {
|
||||||
background: var(--bg-tertiary);
|
background: var(--bg-tertiary);
|
||||||
@@ -77,7 +76,7 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
.sidebar-footer {
|
.sidebar-footer {
|
||||||
padding: 0 8px;
|
padding: 0 12px;
|
||||||
border-top: 1px solid var(--border-color);
|
border-top: 1px solid var(--border-color);
|
||||||
padding-top: 12px;
|
padding-top: 12px;
|
||||||
margin-top: 8px;
|
margin-top: 8px;
|
||||||
|
|||||||
@@ -1,11 +1,19 @@
|
|||||||
import { useState } from 'react'
|
import { useState, useEffect } from 'react'
|
||||||
import { NavLink, useLocation } from 'react-router-dom'
|
import { NavLink, useLocation } from 'react-router-dom'
|
||||||
import { Home, MessageSquare, BarChart3, Users, FileText, Database, Settings, ChevronLeft, ChevronRight, Download } from 'lucide-react'
|
import { Home, MessageSquare, BarChart3, Users, FileText, Database, Settings, ChevronLeft, ChevronRight, Download, Aperture, UserCircle, Lock } from 'lucide-react'
|
||||||
|
import { useAppStore } from '../stores/appStore'
|
||||||
|
import * as configService from '../services/config'
|
||||||
import './Sidebar.scss'
|
import './Sidebar.scss'
|
||||||
|
|
||||||
function Sidebar() {
|
function Sidebar() {
|
||||||
const location = useLocation()
|
const location = useLocation()
|
||||||
const [collapsed, setCollapsed] = useState(false)
|
const [collapsed, setCollapsed] = useState(false)
|
||||||
|
const [authEnabled, setAuthEnabled] = useState(false)
|
||||||
|
const setLocked = useAppStore(state => state.setLocked)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
configService.getAuthEnabled().then(setAuthEnabled)
|
||||||
|
}, [])
|
||||||
|
|
||||||
const isActive = (path: string) => {
|
const isActive = (path: string) => {
|
||||||
return location.pathname === path || location.pathname.startsWith(`${path}/`)
|
return location.pathname === path || location.pathname.startsWith(`${path}/`)
|
||||||
@@ -34,6 +42,26 @@ function Sidebar() {
|
|||||||
<span className="nav-label">聊天</span>
|
<span className="nav-label">聊天</span>
|
||||||
</NavLink>
|
</NavLink>
|
||||||
|
|
||||||
|
{/* 朋友圈 */}
|
||||||
|
<NavLink
|
||||||
|
to="/sns"
|
||||||
|
className={`nav-item ${isActive('/sns') ? 'active' : ''}`}
|
||||||
|
title={collapsed ? '朋友圈' : undefined}
|
||||||
|
>
|
||||||
|
<span className="nav-icon"><Aperture size={20} /></span>
|
||||||
|
<span className="nav-label">朋友圈</span>
|
||||||
|
</NavLink>
|
||||||
|
|
||||||
|
{/* 通讯录 */}
|
||||||
|
<NavLink
|
||||||
|
to="/contacts"
|
||||||
|
className={`nav-item ${isActive('/contacts') ? 'active' : ''}`}
|
||||||
|
title={collapsed ? '通讯录' : undefined}
|
||||||
|
>
|
||||||
|
<span className="nav-icon"><UserCircle size={20} /></span>
|
||||||
|
<span className="nav-label">通讯录</span>
|
||||||
|
</NavLink>
|
||||||
|
|
||||||
{/* 私聊分析 */}
|
{/* 私聊分析 */}
|
||||||
<NavLink
|
<NavLink
|
||||||
to="/analytics"
|
to="/analytics"
|
||||||
@@ -74,18 +102,21 @@ function Sidebar() {
|
|||||||
<span className="nav-label">导出</span>
|
<span className="nav-label">导出</span>
|
||||||
</NavLink>
|
</NavLink>
|
||||||
|
|
||||||
{/* 数据管理 */}
|
|
||||||
<NavLink
|
|
||||||
to="/data-management"
|
|
||||||
className={`nav-item ${isActive('/data-management') ? 'active' : ''}`}
|
|
||||||
title={collapsed ? '数据管理' : undefined}
|
|
||||||
>
|
|
||||||
<span className="nav-icon"><Database size={20} /></span>
|
|
||||||
<span className="nav-label">数据管理</span>
|
|
||||||
</NavLink>
|
|
||||||
</nav>
|
</nav>
|
||||||
|
|
||||||
<div className="sidebar-footer">
|
<div className="sidebar-footer">
|
||||||
|
{authEnabled && (
|
||||||
|
<button
|
||||||
|
className="nav-item"
|
||||||
|
onClick={() => setLocked(true)}
|
||||||
|
title={collapsed ? '锁定' : undefined}
|
||||||
|
>
|
||||||
|
<span className="nav-icon"><Lock size={20} /></span>
|
||||||
|
<span className="nav-label">锁定</span>
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
|
||||||
<NavLink
|
<NavLink
|
||||||
to="/settings"
|
to="/settings"
|
||||||
className={`nav-item ${isActive('/settings') ? 'active' : ''}`}
|
className={`nav-item ${isActive('/settings') ? 'active' : ''}`}
|
||||||
|
|||||||
@@ -1,10 +1,14 @@
|
|||||||
import './TitleBar.scss'
|
import './TitleBar.scss'
|
||||||
|
|
||||||
function TitleBar() {
|
interface TitleBarProps {
|
||||||
|
title?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
function TitleBar({ title }: TitleBarProps = {}) {
|
||||||
return (
|
return (
|
||||||
<div className="title-bar">
|
<div className="title-bar">
|
||||||
<img src="./logo.png" alt="WeFlow" className="title-logo" />
|
<img src="./logo.png" alt="WeFlow" className="title-logo" />
|
||||||
<span className="titles">WeFlow</span>
|
<span className="titles">{title || 'WeFlow'}</span>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
274
src/components/UpdateDialog.scss
Normal file
274
src/components/UpdateDialog.scss
Normal file
@@ -0,0 +1,274 @@
|
|||||||
|
.update-dialog-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: 9999;
|
||||||
|
animation: fadeIn 0.3s ease-out;
|
||||||
|
|
||||||
|
.update-dialog {
|
||||||
|
width: 680px;
|
||||||
|
background: #f5f5f5;
|
||||||
|
border-radius: 24px;
|
||||||
|
box-shadow: 0 20px 40px rgba(0, 0, 0, 0.2);
|
||||||
|
overflow: hidden;
|
||||||
|
position: relative;
|
||||||
|
animation: slideUp 0.3s ease-out;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
|
||||||
|
/* Top Section (White/Gradient) */
|
||||||
|
.dialog-header {
|
||||||
|
background: #ffffff;
|
||||||
|
padding: 40px 20px 30px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
text-align: center;
|
||||||
|
position: relative;
|
||||||
|
|
||||||
|
/* Subtle radial gradient effect in top left as seen in image */
|
||||||
|
&::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
top: -50px;
|
||||||
|
left: -50px;
|
||||||
|
width: 200px;
|
||||||
|
height: 200px;
|
||||||
|
background: radial-gradient(circle, rgba(255, 235, 220, 0.4) 0%, rgba(255, 255, 255, 0) 70%);
|
||||||
|
opacity: 0.8;
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.version-tag {
|
||||||
|
background: #f0eee9;
|
||||||
|
color: #8c7b6e;
|
||||||
|
padding: 4px 16px;
|
||||||
|
border-radius: 12px;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 600;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
letter-spacing: 0.5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
h2 {
|
||||||
|
font-size: 32px;
|
||||||
|
font-weight: 800;
|
||||||
|
color: #333333;
|
||||||
|
margin: 0 0 12px;
|
||||||
|
letter-spacing: -0.5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.subtitle {
|
||||||
|
font-size: 15px;
|
||||||
|
color: #999999;
|
||||||
|
font-weight: 400;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Content Section (Light Gray) */
|
||||||
|
.dialog-content {
|
||||||
|
background: #f2f2f2;
|
||||||
|
padding: 24px 40px 40px;
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
|
||||||
|
.update-notes-container {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
padding: 20px 0;
|
||||||
|
margin-bottom: 30px;
|
||||||
|
|
||||||
|
.icon-box {
|
||||||
|
background: #fbfbfb; // Beige-ish white
|
||||||
|
width: 48px;
|
||||||
|
height: 48px;
|
||||||
|
border-radius: 16px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
margin-right: 20px;
|
||||||
|
flex-shrink: 0;
|
||||||
|
color: #8c7b6e;
|
||||||
|
box-shadow: 0 4px 10px rgba(0, 0, 0, 0.03);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
opacity: 0.8;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.text-box {
|
||||||
|
flex: 1;
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 700;
|
||||||
|
color: #333333;
|
||||||
|
margin: 0 0 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
p {
|
||||||
|
font-size: 14px;
|
||||||
|
color: #666666;
|
||||||
|
line-height: 1.6;
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
ul {
|
||||||
|
margin: 8px 0 0 18px;
|
||||||
|
padding: 0;
|
||||||
|
|
||||||
|
li {
|
||||||
|
font-size: 14px;
|
||||||
|
color: #666666;
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-section {
|
||||||
|
margin-bottom: 30px;
|
||||||
|
|
||||||
|
.progress-info-row {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
font-size: 12px;
|
||||||
|
color: #888;
|
||||||
|
font-weight: 500;
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-bar-bg {
|
||||||
|
height: 6px;
|
||||||
|
background: #e0e0e0;
|
||||||
|
border-radius: 3px;
|
||||||
|
overflow: hidden;
|
||||||
|
|
||||||
|
.progress-bar-fill {
|
||||||
|
height: 100%;
|
||||||
|
background: #000000;
|
||||||
|
border-radius: 3px;
|
||||||
|
transition: width 0.3s ease;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-text {
|
||||||
|
text-align: center;
|
||||||
|
margin-top: 12px;
|
||||||
|
font-size: 13px;
|
||||||
|
color: #666;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.actions {
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 12px;
|
||||||
|
|
||||||
|
.btn-ignore {
|
||||||
|
background: transparent;
|
||||||
|
color: #666666;
|
||||||
|
border: 1px solid #d0d0d0;
|
||||||
|
padding: 16px 32px;
|
||||||
|
border-radius: 20px;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: #f5f5f5;
|
||||||
|
border-color: #999999;
|
||||||
|
color: #333333;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
transform: scale(0.98);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-update {
|
||||||
|
background: #000000;
|
||||||
|
color: #ffffff;
|
||||||
|
border: none;
|
||||||
|
padding: 16px 48px;
|
||||||
|
border-radius: 20px; // Pill shape
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: transform 0.2s, box-shadow 0.2s;
|
||||||
|
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
box-shadow: 0 6px 16px rgba(0, 0, 0, 0.2);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.7;
|
||||||
|
cursor: not-allowed;
|
||||||
|
transform: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.close-btn {
|
||||||
|
position: absolute;
|
||||||
|
top: 16px;
|
||||||
|
right: 16px;
|
||||||
|
background: rgba(0, 0, 0, 0.05);
|
||||||
|
border: none;
|
||||||
|
color: #999;
|
||||||
|
cursor: pointer;
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
border-radius: 50%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
z-index: 10;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(0, 0, 0, 0.1);
|
||||||
|
color: #333;
|
||||||
|
transform: rotate(90deg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes slideUp {
|
||||||
|
from {
|
||||||
|
transform: translateY(20px);
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
transform: translateY(0);
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
139
src/components/UpdateDialog.tsx
Normal file
139
src/components/UpdateDialog.tsx
Normal file
@@ -0,0 +1,139 @@
|
|||||||
|
import React, { useEffect, useState } from 'react'
|
||||||
|
import { Quote, X } from 'lucide-react'
|
||||||
|
import './UpdateDialog.scss'
|
||||||
|
|
||||||
|
interface UpdateInfo {
|
||||||
|
version?: string
|
||||||
|
releaseNotes?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface UpdateDialogProps {
|
||||||
|
open: boolean
|
||||||
|
updateInfo: UpdateInfo | null
|
||||||
|
onClose: () => void
|
||||||
|
onUpdate: () => void
|
||||||
|
onIgnore?: () => void
|
||||||
|
isDownloading: boolean
|
||||||
|
progress: number | {
|
||||||
|
percent: number
|
||||||
|
bytesPerSecond?: number
|
||||||
|
transferred?: number
|
||||||
|
total?: number
|
||||||
|
remaining?: number // seconds
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const UpdateDialog: React.FC<UpdateDialogProps> = ({
|
||||||
|
open,
|
||||||
|
updateInfo,
|
||||||
|
onClose,
|
||||||
|
onUpdate,
|
||||||
|
onIgnore,
|
||||||
|
isDownloading,
|
||||||
|
progress
|
||||||
|
}) => {
|
||||||
|
if (!open || !updateInfo) return null
|
||||||
|
|
||||||
|
// Safe normalize progress
|
||||||
|
const safeProgress = typeof progress === 'number' ? { percent: progress } : (progress || { percent: 0 })
|
||||||
|
const percent = safeProgress.percent || 0
|
||||||
|
const bytesPerSecond = safeProgress.bytesPerSecond
|
||||||
|
const total = safeProgress.total
|
||||||
|
const transferred = safeProgress.transferred
|
||||||
|
const remaining = safeProgress.remaining
|
||||||
|
|
||||||
|
// Format bytes
|
||||||
|
const formatBytes = (bytes: number) => {
|
||||||
|
if (!Number.isFinite(bytes) || bytes === 0) return '0 B'
|
||||||
|
const k = 1024
|
||||||
|
const sizes = ['B', 'KB', 'MB', 'GB', 'TB']
|
||||||
|
const i = Math.floor(Math.log(bytes) / Math.log(k))
|
||||||
|
const unitIndex = Math.max(0, Math.min(i, sizes.length - 1))
|
||||||
|
return parseFloat((bytes / Math.pow(k, unitIndex)).toFixed(1)) + ' ' + sizes[unitIndex]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Format speed
|
||||||
|
const formatSpeed = (bytesPerSecond: number) => {
|
||||||
|
return `${formatBytes(bytesPerSecond)}/s`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Format time
|
||||||
|
const formatTime = (seconds: number) => {
|
||||||
|
if (!Number.isFinite(seconds)) return '计算中...'
|
||||||
|
if (seconds < 60) return `${Math.ceil(seconds)} 秒`
|
||||||
|
const minutes = Math.floor(seconds / 60)
|
||||||
|
const remainingSeconds = Math.ceil(seconds % 60)
|
||||||
|
return `${minutes} 分 ${remainingSeconds} 秒`
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="update-dialog-overlay">
|
||||||
|
<div className="update-dialog">
|
||||||
|
{!isDownloading && (
|
||||||
|
<button className="close-btn" onClick={onClose}>
|
||||||
|
<X size={20} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className="dialog-header">
|
||||||
|
<div className="version-tag">
|
||||||
|
新版本 {updateInfo.version}
|
||||||
|
</div>
|
||||||
|
<h2>欢迎体验全新的 WeFlow</h2>
|
||||||
|
<div className="subtitle">我们带来了一些改进</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="dialog-content">
|
||||||
|
<div className="update-notes-container">
|
||||||
|
<div className="icon-box">
|
||||||
|
<Quote size={20} />
|
||||||
|
</div>
|
||||||
|
<div className="text-box">
|
||||||
|
<h3>优化</h3>
|
||||||
|
{updateInfo.releaseNotes ? (
|
||||||
|
<div dangerouslySetInnerHTML={{ __html: updateInfo.releaseNotes }} />
|
||||||
|
) : (
|
||||||
|
<p>修复了一些已知问题,提升了稳定性。</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{isDownloading ? (
|
||||||
|
<div className="progress-section">
|
||||||
|
<div className="progress-info-row">
|
||||||
|
<span>{bytesPerSecond ? formatSpeed(bytesPerSecond) : '下载中...'}</span>
|
||||||
|
<span>{total ? `${formatBytes(transferred || 0)} / ${formatBytes(total)}` : `${percent.toFixed(1)}%`}</span>
|
||||||
|
{remaining !== undefined && <span>剩余 {formatTime(remaining)}</span>}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="progress-bar-bg">
|
||||||
|
<div
|
||||||
|
className="progress-bar-fill"
|
||||||
|
style={{ width: `${percent}%` }}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Fallback status text if detailed info is missing */}
|
||||||
|
{(!bytesPerSecond && !total) && (
|
||||||
|
<div className="status-text">{percent.toFixed(0)}% 已下载</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="actions">
|
||||||
|
{onIgnore && (
|
||||||
|
<button className="btn-ignore" onClick={onIgnore}>
|
||||||
|
忽略本次更新
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
<button className="btn-update" onClick={onUpdate}>
|
||||||
|
开启新旅程
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default UpdateDialog
|
||||||
192
src/components/UpdateProgressCapsule.scss
Normal file
192
src/components/UpdateProgressCapsule.scss
Normal file
@@ -0,0 +1,192 @@
|
|||||||
|
.update-progress-capsule {
|
||||||
|
position: fixed;
|
||||||
|
top: 38px; // Just below title bar
|
||||||
|
left: 50%;
|
||||||
|
transform: translateX(-50%);
|
||||||
|
z-index: 9998;
|
||||||
|
cursor: pointer;
|
||||||
|
animation: capsuleSlideDown 0.4s cubic-bezier(0.16, 1, 0.3, 1);
|
||||||
|
user-select: none;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
.capsule-content {
|
||||||
|
background: rgba(255, 255, 255, 0.95);
|
||||||
|
box-shadow: 0 8px 20px rgba(0, 0, 0, 0.12);
|
||||||
|
transform: scale(1.02);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.capsule-content {
|
||||||
|
background: var(--bg-primary);
|
||||||
|
backdrop-filter: blur(12px);
|
||||||
|
-webkit-backdrop-filter: blur(12px);
|
||||||
|
padding: 8px 18px;
|
||||||
|
border-radius: 24px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
box-shadow: 0 8px 16px rgba(0, 0, 0, 0.12);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
height: 40px;
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
transition: all 0.3s cubic-bezier(0.16, 1, 0.3, 1);
|
||||||
|
|
||||||
|
.icon-wrapper {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
.download-icon {
|
||||||
|
animation: capsulePulse 2s infinite ease-in-out;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-wrapper {
|
||||||
|
display: flex;
|
||||||
|
align-items: baseline;
|
||||||
|
gap: 10px;
|
||||||
|
z-index: 1;
|
||||||
|
|
||||||
|
.percent-text {
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 700;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.speed-text {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-weight: 500;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.error-text {
|
||||||
|
font-size: 15px;
|
||||||
|
color: #ff4d4f;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
.available-text {
|
||||||
|
font-size: 15px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-bg {
|
||||||
|
position: absolute;
|
||||||
|
bottom: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
height: 3px;
|
||||||
|
background: rgba(0, 0, 0, 0.05);
|
||||||
|
|
||||||
|
.progress-fill {
|
||||||
|
height: 100%;
|
||||||
|
background: var(--primary);
|
||||||
|
transition: width 0.3s ease;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.capsule-close {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
padding: 4px;
|
||||||
|
margin-left: -4px;
|
||||||
|
margin-right: -8px;
|
||||||
|
cursor: pointer;
|
||||||
|
opacity: 0.5;
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
opacity: 1;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 50%;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// State Modifiers
|
||||||
|
&.state-available {
|
||||||
|
.capsule-content {
|
||||||
|
background: var(--primary);
|
||||||
|
border-color: rgba(255, 255, 255, 0.1);
|
||||||
|
color: white;
|
||||||
|
|
||||||
|
.icon-wrapper {
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-wrapper {
|
||||||
|
.available-text {
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.capsule-close {
|
||||||
|
color: rgba(255, 255, 255, 0.8);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(255, 255, 255, 0.1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.state-downloading {
|
||||||
|
.capsule-content {
|
||||||
|
background: var(--bg-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.state-error {
|
||||||
|
.capsule-content {
|
||||||
|
background: #fff1f0;
|
||||||
|
border-color: #ffa39e;
|
||||||
|
|
||||||
|
.icon-wrapper {
|
||||||
|
color: #ff4d4f;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-wrapper .error-text {
|
||||||
|
color: #cf1322;
|
||||||
|
}
|
||||||
|
|
||||||
|
.capsule-close {
|
||||||
|
color: #cf1322;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes capsuleSlideDown {
|
||||||
|
from {
|
||||||
|
transform: translate(-50%, -40px);
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
transform: translate(-50%, 0);
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes capsulePulse {
|
||||||
|
|
||||||
|
0%,
|
||||||
|
100% {
|
||||||
|
transform: translateY(0);
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
50% {
|
||||||
|
transform: translateY(2px);
|
||||||
|
opacity: 0.6;
|
||||||
|
}
|
||||||
|
}
|
||||||
118
src/components/UpdateProgressCapsule.tsx
Normal file
118
src/components/UpdateProgressCapsule.tsx
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
import React from 'react'
|
||||||
|
import { useAppStore } from '../stores/appStore'
|
||||||
|
import { Download, X, AlertCircle, Info } from 'lucide-react'
|
||||||
|
import './UpdateProgressCapsule.scss'
|
||||||
|
|
||||||
|
const UpdateProgressCapsule: React.FC = () => {
|
||||||
|
const {
|
||||||
|
isDownloading,
|
||||||
|
downloadProgress,
|
||||||
|
showUpdateDialog,
|
||||||
|
setShowUpdateDialog,
|
||||||
|
updateInfo,
|
||||||
|
setUpdateInfo,
|
||||||
|
updateError,
|
||||||
|
setUpdateError
|
||||||
|
} = useAppStore()
|
||||||
|
|
||||||
|
// Control visibility
|
||||||
|
// If dialog is open, we usually hide the capsule UNLESS we want it as a mini-indicator
|
||||||
|
// For now, let's hide it if the dialog is open
|
||||||
|
if (showUpdateDialog) return null
|
||||||
|
|
||||||
|
// State mapping
|
||||||
|
const hasError = !!updateError
|
||||||
|
const hasUpdate = !!updateInfo && updateInfo.hasUpdate
|
||||||
|
|
||||||
|
if (!hasError && !isDownloading && !hasUpdate) return null
|
||||||
|
|
||||||
|
// Safe normalize progress
|
||||||
|
const safeProgress = typeof downloadProgress === 'number' ? { percent: downloadProgress } : (downloadProgress || { percent: 0 })
|
||||||
|
const percent = safeProgress.percent || 0
|
||||||
|
const bytesPerSecond = safeProgress.bytesPerSecond
|
||||||
|
|
||||||
|
const formatBytes = (bytes: number) => {
|
||||||
|
if (!Number.isFinite(bytes) || bytes === 0) return '0 B'
|
||||||
|
const k = 1024
|
||||||
|
const sizes = ['B', 'KB', 'MB', 'GB']
|
||||||
|
const i = Math.floor(Math.log(bytes) / Math.log(k))
|
||||||
|
const unitIndex = Math.max(0, Math.min(i, sizes.length - 1))
|
||||||
|
return parseFloat((bytes / Math.pow(k, unitIndex)).toFixed(1)) + ' ' + sizes[unitIndex]
|
||||||
|
}
|
||||||
|
|
||||||
|
const formatSpeed = (bps: number) => {
|
||||||
|
return `${formatBytes(bps)}/s`
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleClose = (e: React.MouseEvent) => {
|
||||||
|
e.stopPropagation()
|
||||||
|
if (hasError) {
|
||||||
|
setUpdateError(null)
|
||||||
|
} else if (hasUpdate && !isDownloading) {
|
||||||
|
setUpdateInfo(null)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Determine appearance class and content
|
||||||
|
let capsuleClass = 'update-progress-capsule'
|
||||||
|
let content = null
|
||||||
|
|
||||||
|
if (hasError) {
|
||||||
|
capsuleClass += ' state-error'
|
||||||
|
content = (
|
||||||
|
<>
|
||||||
|
<div className="icon-wrapper">
|
||||||
|
<AlertCircle size={14} />
|
||||||
|
</div>
|
||||||
|
<div className="info-wrapper">
|
||||||
|
<span className="error-text">更新失败: {updateError}</span>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)
|
||||||
|
} else if (isDownloading) {
|
||||||
|
capsuleClass += ' state-downloading'
|
||||||
|
content = (
|
||||||
|
<>
|
||||||
|
<div className="icon-wrapper">
|
||||||
|
<Download size={14} className="download-icon" />
|
||||||
|
</div>
|
||||||
|
<div className="info-wrapper">
|
||||||
|
<span className="percent-text">{percent.toFixed(0)}%</span>
|
||||||
|
{bytesPerSecond > 0 && (
|
||||||
|
<span className="speed-text">{formatSpeed(bytesPerSecond)}</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="progress-bg">
|
||||||
|
<div className="progress-fill" style={{ width: `${percent}%` }} />
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)
|
||||||
|
} else if (hasUpdate) {
|
||||||
|
capsuleClass += ' state-available'
|
||||||
|
content = (
|
||||||
|
<>
|
||||||
|
<div className="icon-wrapper">
|
||||||
|
<Info size={14} />
|
||||||
|
</div>
|
||||||
|
<div className="info-wrapper">
|
||||||
|
<span className="available-text">发现新版本 v{updateInfo?.version}</span>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className={capsuleClass} onClick={() => setShowUpdateDialog(true)}>
|
||||||
|
<div className="capsule-content">
|
||||||
|
{content}
|
||||||
|
{!isDownloading && (
|
||||||
|
<button className="capsule-close" onClick={handleClose}>
|
||||||
|
<X size={12} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default UpdateProgressCapsule
|
||||||
262
src/components/VoiceTranscribeDialog.scss
Normal file
262
src/components/VoiceTranscribeDialog.scss
Normal file
@@ -0,0 +1,262 @@
|
|||||||
|
.voice-transcribe-dialog-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: 10000;
|
||||||
|
animation: fadeIn 0.2s ease-out;
|
||||||
|
}
|
||||||
|
|
||||||
|
.voice-transcribe-dialog {
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border-radius: 16px;
|
||||||
|
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.3);
|
||||||
|
width: 90%;
|
||||||
|
max-width: 480px;
|
||||||
|
animation: slideUp 0.3s ease-out;
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 20px 24px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.close-button {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
padding: 4px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
border-radius: 6px;
|
||||||
|
transition: all 0.15s ease;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-content {
|
||||||
|
padding: 24px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-section {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
text-align: center;
|
||||||
|
gap: 16px;
|
||||||
|
|
||||||
|
.info-icon {
|
||||||
|
color: var(--primary);
|
||||||
|
opacity: 0.8;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-text {
|
||||||
|
font-size: 15px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.model-info {
|
||||||
|
width: 100%;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 16px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
|
||||||
|
.model-item {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
font-size: 14px;
|
||||||
|
|
||||||
|
.label {
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.value {
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-weight: 500;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.download-section {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
gap: 16px;
|
||||||
|
padding: 20px 0;
|
||||||
|
|
||||||
|
.download-icon {
|
||||||
|
.downloading-icon {
|
||||||
|
color: var(--primary);
|
||||||
|
animation: bounce 1s ease-in-out infinite;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.download-text {
|
||||||
|
font-size: 15px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-bar {
|
||||||
|
width: 100%;
|
||||||
|
height: 6px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 3px;
|
||||||
|
overflow: hidden;
|
||||||
|
|
||||||
|
.progress-fill {
|
||||||
|
height: 100%;
|
||||||
|
background: var(--primary-gradient);
|
||||||
|
border-radius: 3px;
|
||||||
|
transition: width 0.3s ease;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.progress-text {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
margin: 0;
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
}
|
||||||
|
|
||||||
|
.download-hint {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
margin: 8px 0 0 0;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.complete-section {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
gap: 16px;
|
||||||
|
padding: 20px 0;
|
||||||
|
|
||||||
|
.complete-icon {
|
||||||
|
color: #10b981;
|
||||||
|
}
|
||||||
|
|
||||||
|
.complete-text {
|
||||||
|
font-size: 15px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.error-message {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 12px 16px;
|
||||||
|
background: rgba(239, 68, 68, 0.1);
|
||||||
|
border: 1px solid rgba(239, 68, 68, 0.3);
|
||||||
|
border-radius: 8px;
|
||||||
|
color: #ef4444;
|
||||||
|
font-size: 14px;
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-actions {
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
margin-top: 24px;
|
||||||
|
|
||||||
|
button {
|
||||||
|
flex: 1;
|
||||||
|
padding: 12px 20px;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.15s ease;
|
||||||
|
border: none;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 6px;
|
||||||
|
|
||||||
|
&.btn-secondary {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.btn-primary {
|
||||||
|
background: var(--primary);
|
||||||
|
color: white;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes slideUp {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes bounce {
|
||||||
|
|
||||||
|
0%,
|
||||||
|
100% {
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
50% {
|
||||||
|
transform: translateY(-10px);
|
||||||
|
}
|
||||||
|
}
|
||||||
160
src/components/VoiceTranscribeDialog.tsx
Normal file
160
src/components/VoiceTranscribeDialog.tsx
Normal file
@@ -0,0 +1,160 @@
|
|||||||
|
import React, { useState, useEffect } from 'react'
|
||||||
|
import { Download, X, CheckCircle, AlertCircle } from 'lucide-react'
|
||||||
|
import './VoiceTranscribeDialog.scss'
|
||||||
|
|
||||||
|
interface VoiceTranscribeDialogProps {
|
||||||
|
onClose: () => void
|
||||||
|
onDownloadComplete: () => void
|
||||||
|
}
|
||||||
|
|
||||||
|
export const VoiceTranscribeDialog: React.FC<VoiceTranscribeDialogProps> = ({
|
||||||
|
onClose,
|
||||||
|
onDownloadComplete
|
||||||
|
}) => {
|
||||||
|
const [isDownloading, setIsDownloading] = useState(false)
|
||||||
|
const [downloadProgress, setDownloadProgress] = useState(0)
|
||||||
|
const [downloadError, setDownloadError] = useState<string | null>(null)
|
||||||
|
const [isComplete, setIsComplete] = useState(false)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
// 监听下载进度
|
||||||
|
if (!window.electronAPI?.whisper?.onDownloadProgress) {
|
||||||
|
console.warn('[VoiceTranscribeDialog] whisper API 不可用')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const removeListener = window.electronAPI.whisper.onDownloadProgress((payload: { modelName: string; downloadedBytes: number; totalBytes?: number; percent?: number }) => {
|
||||||
|
if (payload.percent !== undefined) {
|
||||||
|
setDownloadProgress(payload.percent)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
removeListener?.()
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const handleDownload = async () => {
|
||||||
|
if (!window.electronAPI?.whisper?.downloadModel) {
|
||||||
|
setDownloadError('语音转文字功能不可用')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
setIsDownloading(true)
|
||||||
|
setDownloadError(null)
|
||||||
|
setDownloadProgress(0)
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.whisper.downloadModel()
|
||||||
|
|
||||||
|
if (result?.success) {
|
||||||
|
setIsComplete(true)
|
||||||
|
setDownloadProgress(100)
|
||||||
|
|
||||||
|
// 延迟关闭弹窗并触发转写
|
||||||
|
setTimeout(() => {
|
||||||
|
onDownloadComplete()
|
||||||
|
}, 1000)
|
||||||
|
} else {
|
||||||
|
setDownloadError(result?.error || '下载失败')
|
||||||
|
setIsDownloading(false)
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
setDownloadError(String(error))
|
||||||
|
setIsDownloading(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleCancel = () => {
|
||||||
|
if (!isDownloading && !isComplete) {
|
||||||
|
onClose()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="voice-transcribe-dialog-overlay" onClick={handleCancel}>
|
||||||
|
<div className="voice-transcribe-dialog" onClick={(e) => e.stopPropagation()}>
|
||||||
|
<div className="dialog-header">
|
||||||
|
<h3>语音转文字</h3>
|
||||||
|
{!isDownloading && !isComplete && (
|
||||||
|
<button className="close-button" onClick={onClose}>
|
||||||
|
<X size={20} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="dialog-content">
|
||||||
|
{!isDownloading && !isComplete && (
|
||||||
|
<>
|
||||||
|
<div className="info-section">
|
||||||
|
<AlertCircle size={48} className="info-icon" />
|
||||||
|
<p className="info-text">
|
||||||
|
首次使用语音转文字功能需要下载 AI 模型
|
||||||
|
</p>
|
||||||
|
<div className="model-info">
|
||||||
|
<div className="model-item">
|
||||||
|
<span className="label">模型名称:</span>
|
||||||
|
<span className="value">SenseVoiceSmall</span>
|
||||||
|
</div>
|
||||||
|
<div className="model-item">
|
||||||
|
<span className="label">文件大小:</span>
|
||||||
|
<span className="value">约 240 MB</span>
|
||||||
|
</div>
|
||||||
|
<div className="model-item">
|
||||||
|
<span className="label">支持语言:</span>
|
||||||
|
<span className="value">中文、粤语、英文、日文、韩文</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{downloadError && (
|
||||||
|
<div className="error-message">
|
||||||
|
<AlertCircle size={16} />
|
||||||
|
<span>{downloadError}</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className="dialog-actions">
|
||||||
|
<button className="btn-secondary" onClick={onClose}>
|
||||||
|
取消
|
||||||
|
</button>
|
||||||
|
<button className="btn-primary" onClick={handleDownload}>
|
||||||
|
<Download size={16} />
|
||||||
|
<span>立即下载</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isDownloading && !isComplete && (
|
||||||
|
<div className="download-section">
|
||||||
|
<div className="download-icon">
|
||||||
|
<Download size={48} className="downloading-icon" />
|
||||||
|
</div>
|
||||||
|
<p className="download-text">
|
||||||
|
{downloadProgress < 1 ? '正在连接服务器...' : '正在下载模型...'}
|
||||||
|
</p>
|
||||||
|
<div className="progress-bar">
|
||||||
|
<div
|
||||||
|
className="progress-fill"
|
||||||
|
style={{ width: `${downloadProgress}%` }}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<p className="progress-text">{downloadProgress.toFixed(1)}%</p>
|
||||||
|
{downloadProgress < 1 && (
|
||||||
|
<p className="download-hint">首次连接可能需要较长时间,请耐心等待</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isComplete && (
|
||||||
|
<div className="complete-section">
|
||||||
|
<CheckCircle size={48} className="complete-icon" />
|
||||||
|
<p className="complete-text">下载完成!正在转写语音...</p>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
552
src/pages/AIChatPage.scss
Normal file
552
src/pages/AIChatPage.scss
Normal file
@@ -0,0 +1,552 @@
|
|||||||
|
// 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);
|
||||||
|
}
|
||||||
|
}
|
||||||
391
src/pages/AIChatPage.tsx
Normal file
391
src/pages/AIChatPage.tsx
Normal file
@@ -0,0 +1,391 @@
|
|||||||
|
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>
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -11,7 +11,7 @@ function AgreementPage() {
|
|||||||
<h2>用户协议</h2>
|
<h2>用户协议</h2>
|
||||||
|
|
||||||
<h3>一、总则</h3>
|
<h3>一、总则</h3>
|
||||||
<p>欢迎使用WeFlow(WeFlow)软件。请在使用本软件前仔细阅读本协议。一旦您开始使用本软件,即表示您已充分理解并同意本协议的全部内容。</p>
|
<p>欢迎使用WeFlow(WeFlow)软件。请在使用本软件前仔细阅读本协议。一旦你开始使用本软件,即表示你已充分理解并同意本协议的全部内容。</p>
|
||||||
|
|
||||||
<h3>二、软件说明</h3>
|
<h3>二、软件说明</h3>
|
||||||
<p>WeFlow是一款本地化的微信聊天记录查看与分析工具,所有数据处理均在用户本地设备上完成。</p>
|
<p>WeFlow是一款本地化的微信聊天记录查看与分析工具,所有数据处理均在用户本地设备上完成。</p>
|
||||||
@@ -35,7 +35,7 @@ function AgreementPage() {
|
|||||||
<p>本软件不收集、不上传、不存储任何用户个人信息或聊天数据。所有数据处理均在本地完成。</p>
|
<p>本软件不收集、不上传、不存储任何用户个人信息或聊天数据。所有数据处理均在本地完成。</p>
|
||||||
|
|
||||||
<h3>二、数据安全</h3>
|
<h3>二、数据安全</h3>
|
||||||
<p>您的聊天记录和个人数据完全存储在您的本地设备上,本软件不会将任何数据传输至外部服务器。</p>
|
<p>你的聊天记录和个人数据完全存储在你的本地设备上,本软件不会将任何数据传输至外部服务器。</p>
|
||||||
|
|
||||||
<h3>三、网络请求</h3>
|
<h3>三、网络请求</h3>
|
||||||
<p>本软件仅在检查更新时会访问更新服务器获取版本信息,不涉及任何用户数据的传输。</p>
|
<p>本软件仅在检查更新时会访问更新服务器获取版本信息,不涉及任何用户数据的传输。</p>
|
||||||
|
|||||||
@@ -47,6 +47,24 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.page-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 12px;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header-actions {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@keyframes spin {
|
@keyframes spin {
|
||||||
from {
|
from {
|
||||||
transform: rotate(0deg);
|
transform: rotate(0deg);
|
||||||
@@ -293,3 +311,184 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 排除好友弹窗
|
||||||
|
.exclude-modal-overlay {
|
||||||
|
position: fixed;
|
||||||
|
inset: 0;
|
||||||
|
background: rgba(0, 0, 0, 0.45);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
z-index: 1000;
|
||||||
|
backdrop-filter: blur(4px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-modal {
|
||||||
|
width: 560px;
|
||||||
|
max-width: calc(100vw - 48px);
|
||||||
|
background: var(--card-bg);
|
||||||
|
border-radius: 16px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
padding: 20px;
|
||||||
|
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.2);
|
||||||
|
|
||||||
|
.exclude-modal-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-bottom: 12px;
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 16px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-close {
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
border-radius: 50%;
|
||||||
|
border: none;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.15s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-modal-search {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 8px 12px;
|
||||||
|
border-radius: 10px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background: var(--bg-primary);
|
||||||
|
margin-bottom: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
|
||||||
|
input {
|
||||||
|
flex: 1;
|
||||||
|
border: none;
|
||||||
|
outline: none;
|
||||||
|
background: transparent;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.clear-search {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
padding: 2px;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-modal-body {
|
||||||
|
max-height: 420px;
|
||||||
|
overflow: auto;
|
||||||
|
padding-right: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-loading,
|
||||||
|
.exclude-error,
|
||||||
|
.exclude-empty {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 8px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
padding: 24px 0;
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 8px 10px;
|
||||||
|
border-radius: 10px;
|
||||||
|
cursor: pointer;
|
||||||
|
border: 1px solid transparent;
|
||||||
|
transition: all 0.15s;
|
||||||
|
background: var(--bg-primary);
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
border-color: rgba(7, 193, 96, 0.4);
|
||||||
|
background: rgba(7, 193, 96, 0.08);
|
||||||
|
}
|
||||||
|
|
||||||
|
input {
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-avatar {
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-info {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
min-width: 0;
|
||||||
|
gap: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-name {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-weight: 500;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-username {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-modal-footer {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-count {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.exclude-actions {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,20 +1,52 @@
|
|||||||
import { useState, useEffect } from 'react'
|
import { useState, useEffect, useCallback } from 'react'
|
||||||
import { Users, Clock, MessageSquare, Send, Inbox, Calendar, Loader2, RefreshCw, User, Medal } from 'lucide-react'
|
import { useLocation } from 'react-router-dom'
|
||||||
|
import { Users, Clock, MessageSquare, Send, Inbox, Calendar, Loader2, RefreshCw, Medal, UserMinus, Search, X } from 'lucide-react'
|
||||||
import ReactECharts from 'echarts-for-react'
|
import ReactECharts from 'echarts-for-react'
|
||||||
import { useAnalyticsStore } from '../stores/analyticsStore'
|
import { useAnalyticsStore } from '../stores/analyticsStore'
|
||||||
import { useThemeStore } from '../stores/themeStore'
|
import { useThemeStore } from '../stores/themeStore'
|
||||||
import './AnalyticsPage.scss'
|
import './AnalyticsPage.scss'
|
||||||
import './DataManagementPage.scss'
|
import { Avatar } from '../components/Avatar'
|
||||||
|
|
||||||
|
interface ExcludeCandidate {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
avatarUrl?: string
|
||||||
|
wechatId?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
const normalizeUsername = (value: string) => value.trim().toLowerCase()
|
||||||
|
|
||||||
function AnalyticsPage() {
|
function AnalyticsPage() {
|
||||||
const [isLoading, setIsLoading] = useState(false)
|
const [isLoading, setIsLoading] = useState(false)
|
||||||
const [loadingStatus, setLoadingStatus] = useState('')
|
const [loadingStatus, setLoadingStatus] = useState('')
|
||||||
const [error, setError] = useState<string | null>(null)
|
const [error, setError] = useState<string | null>(null)
|
||||||
const [progress, setProgress] = useState(0)
|
const [progress, setProgress] = useState(0)
|
||||||
|
const [isExcludeDialogOpen, setIsExcludeDialogOpen] = useState(false)
|
||||||
|
const [excludeCandidates, setExcludeCandidates] = useState<ExcludeCandidate[]>([])
|
||||||
|
const [excludeQuery, setExcludeQuery] = useState('')
|
||||||
|
const [excludeLoading, setExcludeLoading] = useState(false)
|
||||||
|
const [excludeError, setExcludeError] = useState<string | null>(null)
|
||||||
|
const [excludedUsernames, setExcludedUsernames] = useState<Set<string>>(new Set())
|
||||||
|
const [draftExcluded, setDraftExcluded] = useState<Set<string>>(new Set())
|
||||||
|
|
||||||
const themeMode = useThemeStore((state) => state.themeMode)
|
const themeMode = useThemeStore((state) => state.themeMode)
|
||||||
const { statistics, rankings, timeDistribution, isLoaded, setStatistics, setRankings, setTimeDistribution, markLoaded } = useAnalyticsStore()
|
const { statistics, rankings, timeDistribution, isLoaded, setStatistics, setRankings, setTimeDistribution, markLoaded, clearCache } = useAnalyticsStore()
|
||||||
const loadData = async (forceRefresh = false) => {
|
|
||||||
|
const loadExcludedUsernames = useCallback(async () => {
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.analytics.getExcludedUsernames()
|
||||||
|
if (result.success && result.data) {
|
||||||
|
setExcludedUsernames(new Set(result.data.map(normalizeUsername)))
|
||||||
|
} else {
|
||||||
|
setExcludedUsernames(new Set())
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.warn('加载排除名单失败', e)
|
||||||
|
setExcludedUsernames(new Set())
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const loadData = useCallback(async (forceRefresh = false) => {
|
||||||
if (isLoaded && !forceRefresh) return
|
if (isLoaded && !forceRefresh) return
|
||||||
setIsLoading(true)
|
setIsLoading(true)
|
||||||
setError(null)
|
setError(null)
|
||||||
@@ -28,7 +60,7 @@ function AnalyticsPage() {
|
|||||||
|
|
||||||
try {
|
try {
|
||||||
setLoadingStatus('正在统计消息数据...')
|
setLoadingStatus('正在统计消息数据...')
|
||||||
const statsResult = await window.electronAPI.analytics.getOverallStatistics()
|
const statsResult = await window.electronAPI.analytics.getOverallStatistics(forceRefresh)
|
||||||
if (statsResult.success && statsResult.data) {
|
if (statsResult.success && statsResult.data) {
|
||||||
setStatistics(statsResult.data)
|
setStatistics(statsResult.data)
|
||||||
} else {
|
} else {
|
||||||
@@ -53,12 +85,100 @@ function AnalyticsPage() {
|
|||||||
setIsLoading(false)
|
setIsLoading(false)
|
||||||
if (removeListener) removeListener()
|
if (removeListener) removeListener()
|
||||||
}
|
}
|
||||||
}
|
}, [isLoaded, markLoaded, setRankings, setStatistics, setTimeDistribution])
|
||||||
|
|
||||||
useEffect(() => { loadData() }, [])
|
const location = useLocation()
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const force = location.state?.forceRefresh === true
|
||||||
|
loadData(force)
|
||||||
|
}, [location.state, loadData])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const handleChange = () => {
|
||||||
|
loadExcludedUsernames()
|
||||||
|
loadData(true)
|
||||||
|
}
|
||||||
|
window.addEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
return () => window.removeEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
}, [loadData, loadExcludedUsernames])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
loadExcludedUsernames()
|
||||||
|
}, [loadExcludedUsernames])
|
||||||
|
|
||||||
const handleRefresh = () => loadData(true)
|
const handleRefresh = () => loadData(true)
|
||||||
|
|
||||||
|
const loadExcludeCandidates = useCallback(async () => {
|
||||||
|
setExcludeLoading(true)
|
||||||
|
setExcludeError(null)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.analytics.getExcludeCandidates()
|
||||||
|
if (result.success && result.data) {
|
||||||
|
setExcludeCandidates(result.data)
|
||||||
|
} else {
|
||||||
|
setExcludeError(result.error || '加载好友列表失败')
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
setExcludeError(String(e))
|
||||||
|
} finally {
|
||||||
|
setExcludeLoading(false)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const openExcludeDialog = async () => {
|
||||||
|
setExcludeQuery('')
|
||||||
|
setDraftExcluded(new Set(excludedUsernames))
|
||||||
|
setIsExcludeDialogOpen(true)
|
||||||
|
await loadExcludeCandidates()
|
||||||
|
}
|
||||||
|
|
||||||
|
const toggleExcluded = (username: string) => {
|
||||||
|
const key = normalizeUsername(username)
|
||||||
|
setDraftExcluded((prev) => {
|
||||||
|
const next = new Set(prev)
|
||||||
|
if (next.has(key)) {
|
||||||
|
next.delete(key)
|
||||||
|
} else {
|
||||||
|
next.add(key)
|
||||||
|
}
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleApplyExcluded = async () => {
|
||||||
|
const payload = Array.from(draftExcluded)
|
||||||
|
setIsExcludeDialogOpen(false)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.analytics.setExcludedUsernames(payload)
|
||||||
|
if (!result.success) {
|
||||||
|
alert(result.error || '更新排除名单失败')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
setExcludedUsernames(new Set((result.data || payload).map(normalizeUsername)))
|
||||||
|
clearCache()
|
||||||
|
await window.electronAPI.cache.clearAnalytics()
|
||||||
|
await loadData(true)
|
||||||
|
} catch (e) {
|
||||||
|
alert(`更新排除名单失败:${String(e)}`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const visibleExcludeCandidates = excludeCandidates
|
||||||
|
.filter((candidate) => {
|
||||||
|
const query = excludeQuery.trim().toLowerCase()
|
||||||
|
if (!query) return true
|
||||||
|
const wechatId = candidate.wechatId || ''
|
||||||
|
const haystack = `${candidate.displayName} ${candidate.username} ${wechatId}`.toLowerCase()
|
||||||
|
return haystack.includes(query)
|
||||||
|
})
|
||||||
|
.sort((a, b) => {
|
||||||
|
const aSelected = draftExcluded.has(normalizeUsername(a.username))
|
||||||
|
const bSelected = draftExcluded.has(normalizeUsername(b.username))
|
||||||
|
if (aSelected !== bSelected) return aSelected ? -1 : 1
|
||||||
|
return a.displayName.localeCompare(b.displayName, 'zh')
|
||||||
|
})
|
||||||
|
|
||||||
const formatDate = (timestamp: number | null) => {
|
const formatDate = (timestamp: number | null) => {
|
||||||
if (!timestamp) return '-'
|
if (!timestamp) return '-'
|
||||||
const date = new Date(timestamp * 1000)
|
const date = new Date(timestamp * 1000)
|
||||||
@@ -233,10 +353,16 @@ function AnalyticsPage() {
|
|||||||
<>
|
<>
|
||||||
<div className="page-header">
|
<div className="page-header">
|
||||||
<h1>私聊分析</h1>
|
<h1>私聊分析</h1>
|
||||||
|
<div className="header-actions">
|
||||||
<button className="btn btn-secondary" onClick={handleRefresh} disabled={isLoading}>
|
<button className="btn btn-secondary" onClick={handleRefresh} disabled={isLoading}>
|
||||||
<RefreshCw size={16} className={isLoading ? 'spin' : ''} />
|
<RefreshCw size={16} className={isLoading ? 'spin' : ''} />
|
||||||
{isLoading ? '刷新中...' : '刷新'}
|
{isLoading ? '刷新中...' : '刷新'}
|
||||||
</button>
|
</button>
|
||||||
|
<button className="btn btn-secondary" onClick={openExcludeDialog}>
|
||||||
|
<UserMinus size={16} />
|
||||||
|
排除好友{excludedUsernames.size > 0 ? ` (${excludedUsernames.size})` : ''}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="page-scroll">
|
<div className="page-scroll">
|
||||||
<section className="page-section">
|
<section className="page-section">
|
||||||
@@ -289,7 +415,7 @@ function AnalyticsPage() {
|
|||||||
<div key={contact.username} className="ranking-item">
|
<div key={contact.username} className="ranking-item">
|
||||||
<span className={`rank ${index < 3 ? 'top' : ''}`}>{index + 1}</span>
|
<span className={`rank ${index < 3 ? 'top' : ''}`}>{index + 1}</span>
|
||||||
<div className="contact-avatar">
|
<div className="contact-avatar">
|
||||||
{contact.avatarUrl ? <img src={contact.avatarUrl} alt="" /> : <div className="avatar-placeholder"><User size={20} /></div>}
|
<Avatar src={contact.avatarUrl} name={contact.displayName} size={36} />
|
||||||
{index < 3 && <div className={`medal medal-${index + 1}`}><Medal size={10} /></div>}
|
{index < 3 && <div className={`medal medal-${index + 1}`}><Medal size={10} /></div>}
|
||||||
</div>
|
</div>
|
||||||
<div className="contact-info">
|
<div className="contact-info">
|
||||||
@@ -302,6 +428,84 @@ function AnalyticsPage() {
|
|||||||
</div>
|
</div>
|
||||||
</section>
|
</section>
|
||||||
</div>
|
</div>
|
||||||
|
{isExcludeDialogOpen && (
|
||||||
|
<div className="exclude-modal-overlay" onClick={() => setIsExcludeDialogOpen(false)}>
|
||||||
|
<div className="exclude-modal" onClick={e => e.stopPropagation()}>
|
||||||
|
<div className="exclude-modal-header">
|
||||||
|
<h3>选择不统计的好友</h3>
|
||||||
|
<button className="modal-close" onClick={() => setIsExcludeDialogOpen(false)}>
|
||||||
|
<X size={18} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="exclude-modal-search">
|
||||||
|
<Search size={16} />
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="搜索好友"
|
||||||
|
value={excludeQuery}
|
||||||
|
onChange={e => setExcludeQuery(e.target.value)}
|
||||||
|
disabled={excludeLoading}
|
||||||
|
/>
|
||||||
|
{excludeQuery && (
|
||||||
|
<button className="clear-search" onClick={() => setExcludeQuery('')}>
|
||||||
|
<X size={14} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="exclude-modal-body">
|
||||||
|
{excludeLoading && (
|
||||||
|
<div className="exclude-loading">
|
||||||
|
<Loader2 size={20} className="spin" />
|
||||||
|
<span>正在加载好友列表...</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
{!excludeLoading && excludeError && (
|
||||||
|
<div className="exclude-error">{excludeError}</div>
|
||||||
|
)}
|
||||||
|
{!excludeLoading && !excludeError && (
|
||||||
|
<div className="exclude-list">
|
||||||
|
{visibleExcludeCandidates.map((candidate) => {
|
||||||
|
const isChecked = draftExcluded.has(normalizeUsername(candidate.username))
|
||||||
|
const wechatId = candidate.wechatId?.trim() || candidate.username
|
||||||
|
return (
|
||||||
|
<label key={candidate.username} className={`exclude-item ${isChecked ? 'active' : ''}`}>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={isChecked}
|
||||||
|
onChange={() => toggleExcluded(candidate.username)}
|
||||||
|
/>
|
||||||
|
<div className="exclude-avatar">
|
||||||
|
<Avatar src={candidate.avatarUrl} name={candidate.displayName} size={32} />
|
||||||
|
</div>
|
||||||
|
<div className="exclude-info">
|
||||||
|
<span className="exclude-name">{candidate.displayName}</span>
|
||||||
|
<span className="exclude-username">{wechatId}</span>
|
||||||
|
</div>
|
||||||
|
</label>
|
||||||
|
)
|
||||||
|
})}
|
||||||
|
{visibleExcludeCandidates.length === 0 && (
|
||||||
|
<div className="exclude-empty">
|
||||||
|
{excludeQuery.trim() ? '未找到匹配好友' : '暂无可选好友'}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="exclude-modal-footer">
|
||||||
|
<span className="exclude-count">已排除 {draftExcluded.size} 人</span>
|
||||||
|
<div className="exclude-actions">
|
||||||
|
<button className="btn btn-secondary" onClick={() => setIsExcludeDialogOpen(false)}>
|
||||||
|
取消
|
||||||
|
</button>
|
||||||
|
<button className="btn btn-primary" onClick={handleApplyExcluded} disabled={excludeLoading}>
|
||||||
|
应用
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</>
|
</>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
119
src/pages/AnalyticsWelcomePage.scss
Normal file
119
src/pages/AnalyticsWelcomePage.scss
Normal file
@@ -0,0 +1,119 @@
|
|||||||
|
.analytics-welcome-container {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
height: 100%;
|
||||||
|
padding: 40px;
|
||||||
|
background: var(--bg-primary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
animation: fadeIn 0.4s ease-out;
|
||||||
|
|
||||||
|
.welcome-content {
|
||||||
|
text-align: center;
|
||||||
|
max-width: 600px;
|
||||||
|
|
||||||
|
.icon-wrapper {
|
||||||
|
width: 80px;
|
||||||
|
height: 80px;
|
||||||
|
margin: 0 auto 24px;
|
||||||
|
background: rgba(7, 193, 96, 0.1);
|
||||||
|
border-radius: 20px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: #07c160;
|
||||||
|
|
||||||
|
svg {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
font-size: 28px;
|
||||||
|
margin-bottom: 12px;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
p {
|
||||||
|
color: var(--text-secondary);
|
||||||
|
margin-bottom: 40px;
|
||||||
|
font-size: 16px;
|
||||||
|
line-height: 1.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.action-cards {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 1fr 1fr;
|
||||||
|
gap: 20px;
|
||||||
|
margin-top: 20px;
|
||||||
|
|
||||||
|
button {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
padding: 30px 20px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 12px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
text-align: center;
|
||||||
|
|
||||||
|
&:hover:not(:disabled) {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
border-color: #07c160;
|
||||||
|
box-shadow: 0 4px 12px rgba(7, 193, 96, 0.1);
|
||||||
|
|
||||||
|
.card-icon {
|
||||||
|
color: #07c160;
|
||||||
|
background: rgba(7, 193, 96, 0.1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.6;
|
||||||
|
cursor: not-allowed;
|
||||||
|
filter: grayscale(100%);
|
||||||
|
}
|
||||||
|
|
||||||
|
.card-icon {
|
||||||
|
width: 50px;
|
||||||
|
height: 50px;
|
||||||
|
border-radius: 12px;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 18px;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
span {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeIn {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(10px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
63
src/pages/AnalyticsWelcomePage.tsx
Normal file
63
src/pages/AnalyticsWelcomePage.tsx
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
import { useNavigate } from 'react-router-dom'
|
||||||
|
import { BarChart2, History, RefreshCcw } from 'lucide-react'
|
||||||
|
import { useAnalyticsStore } from '../stores/analyticsStore'
|
||||||
|
import './AnalyticsWelcomePage.scss'
|
||||||
|
|
||||||
|
function AnalyticsWelcomePage() {
|
||||||
|
const navigate = useNavigate()
|
||||||
|
// 检查是否有任何缓存数据加载或基本的存储状态表明它已准备好。
|
||||||
|
// 实际上,如果 store 没有持久化,`isLoaded` 可能会在应用刷新时重置。
|
||||||
|
// 如果用户点击“加载缓存”但缓存为空,AnalyticsPage 的逻辑(loadData 不带 force)将尝试从后端缓存加载。
|
||||||
|
// 如果后端缓存也为空,则会重新计算。
|
||||||
|
|
||||||
|
// 我们也可以检查 `lastLoadTime` 来显示“上次更新:xxx”(如果已持久化)。
|
||||||
|
const { lastLoadTime } = useAnalyticsStore()
|
||||||
|
|
||||||
|
const handleLoadCache = () => {
|
||||||
|
navigate('/analytics/view')
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleNewAnalysis = () => {
|
||||||
|
navigate('/analytics/view', { state: { forceRefresh: true } })
|
||||||
|
}
|
||||||
|
|
||||||
|
const formatLastTime = (ts: number | null) => {
|
||||||
|
if (!ts) return '无记录'
|
||||||
|
return new Date(ts).toLocaleString()
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="analytics-welcome-container">
|
||||||
|
<div className="welcome-content">
|
||||||
|
<div className="icon-wrapper">
|
||||||
|
<BarChart2 size={40} />
|
||||||
|
</div>
|
||||||
|
<h1>私聊数据分析</h1>
|
||||||
|
<p>
|
||||||
|
WeFlow 可以分析你的聊天记录,生成详细的统计报表。<br />
|
||||||
|
你可以选择加载上次的分析结果(速度快),或者开始新的分析(数据最新)。
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<div className="action-cards">
|
||||||
|
<button onClick={handleLoadCache}>
|
||||||
|
<div className="card-icon">
|
||||||
|
<History size={24} />
|
||||||
|
</div>
|
||||||
|
<h3>加载缓存</h3>
|
||||||
|
<span>查看上次分析结果<br />(上次更新: {formatLastTime(lastLoadTime)})</span>
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<button onClick={handleNewAnalysis}>
|
||||||
|
<div className="card-icon">
|
||||||
|
<RefreshCcw size={24} />
|
||||||
|
</div>
|
||||||
|
<h3>新的分析</h3>
|
||||||
|
<span>重新扫描并计算数据<br />(可能需要几分钟)</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default AnalyticsWelcomePage
|
||||||
@@ -5,6 +5,7 @@
|
|||||||
justify-content: center;
|
justify-content: center;
|
||||||
min-height: 100%;
|
min-height: 100%;
|
||||||
text-align: center;
|
text-align: center;
|
||||||
|
padding: 40px 24px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.header-icon {
|
.header-icon {
|
||||||
@@ -25,6 +26,63 @@
|
|||||||
margin: 0 0 48px;
|
margin: 0 0 48px;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.report-sections {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 32px;
|
||||||
|
width: min(760px, 100%);
|
||||||
|
}
|
||||||
|
|
||||||
|
.report-section {
|
||||||
|
width: 100%;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 20px;
|
||||||
|
padding: 28px;
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 16px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-title {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 20px;
|
||||||
|
font-weight: 700;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-desc {
|
||||||
|
margin: 8px 0 0;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-badge {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 6px 10px;
|
||||||
|
border-radius: 999px;
|
||||||
|
background: color-mix(in srgb, var(--primary) 12%, transparent);
|
||||||
|
color: var(--primary);
|
||||||
|
border: 1px solid color-mix(in srgb, var(--primary) 30%, transparent);
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: 600;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-hint {
|
||||||
|
margin: 12px 0 0;
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
.year-grid {
|
.year-grid {
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-wrap: wrap;
|
flex-wrap: wrap;
|
||||||
@@ -34,6 +92,12 @@
|
|||||||
margin-bottom: 48px;
|
margin-bottom: 48px;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.report-section .year-grid {
|
||||||
|
justify-content: flex-start;
|
||||||
|
max-width: none;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
}
|
||||||
|
|
||||||
.year-card {
|
.year-card {
|
||||||
width: 120px;
|
width: 120px;
|
||||||
height: 100px;
|
height: 100px;
|
||||||
@@ -104,6 +168,13 @@
|
|||||||
opacity: 0.6;
|
opacity: 0.6;
|
||||||
cursor: not-allowed;
|
cursor: not-allowed;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
&.secondary {
|
||||||
|
background: var(--card-bg);
|
||||||
|
color: var(--text-primary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
box-shadow: none;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.spin {
|
.spin {
|
||||||
|
|||||||
@@ -1,12 +1,15 @@
|
|||||||
import { useState, useEffect } from 'react'
|
import { useState, useEffect } from 'react'
|
||||||
import { useNavigate } from 'react-router-dom'
|
import { useNavigate } from 'react-router-dom'
|
||||||
import { Calendar, Loader2, Sparkles } from 'lucide-react'
|
import { Calendar, Loader2, Sparkles, Users } from 'lucide-react'
|
||||||
import './AnnualReportPage.scss'
|
import './AnnualReportPage.scss'
|
||||||
|
|
||||||
|
type YearOption = number | 'all'
|
||||||
|
|
||||||
function AnnualReportPage() {
|
function AnnualReportPage() {
|
||||||
const navigate = useNavigate()
|
const navigate = useNavigate()
|
||||||
const [availableYears, setAvailableYears] = useState<number[]>([])
|
const [availableYears, setAvailableYears] = useState<number[]>([])
|
||||||
const [selectedYear, setSelectedYear] = useState<number | null>(null)
|
const [selectedYear, setSelectedYear] = useState<YearOption | null>(null)
|
||||||
|
const [selectedPairYear, setSelectedPairYear] = useState<YearOption | null>(null)
|
||||||
const [isLoading, setIsLoading] = useState(true)
|
const [isLoading, setIsLoading] = useState(true)
|
||||||
const [isGenerating, setIsGenerating] = useState(false)
|
const [isGenerating, setIsGenerating] = useState(false)
|
||||||
const [loadError, setLoadError] = useState<string | null>(null)
|
const [loadError, setLoadError] = useState<string | null>(null)
|
||||||
@@ -22,7 +25,8 @@ function AnnualReportPage() {
|
|||||||
const result = await window.electronAPI.annualReport.getAvailableYears()
|
const result = await window.electronAPI.annualReport.getAvailableYears()
|
||||||
if (result.success && result.data && result.data.length > 0) {
|
if (result.success && result.data && result.data.length > 0) {
|
||||||
setAvailableYears(result.data)
|
setAvailableYears(result.data)
|
||||||
setSelectedYear(result.data[0])
|
setSelectedYear((prev) => prev ?? result.data[0])
|
||||||
|
setSelectedPairYear((prev) => prev ?? result.data[0])
|
||||||
} else if (!result.success) {
|
} else if (!result.success) {
|
||||||
setLoadError(result.error || '加载年度数据失败')
|
setLoadError(result.error || '加载年度数据失败')
|
||||||
}
|
}
|
||||||
@@ -35,10 +39,11 @@ function AnnualReportPage() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const handleGenerateReport = async () => {
|
const handleGenerateReport = async () => {
|
||||||
if (!selectedYear) return
|
if (selectedYear === null) return
|
||||||
setIsGenerating(true)
|
setIsGenerating(true)
|
||||||
try {
|
try {
|
||||||
navigate(`/annual-report/view?year=${selectedYear}`)
|
const yearParam = selectedYear === 'all' ? 0 : selectedYear
|
||||||
|
navigate(`/annual-report/view?year=${yearParam}`)
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('生成报告失败:', e)
|
console.error('生成报告失败:', e)
|
||||||
} finally {
|
} finally {
|
||||||
@@ -46,6 +51,12 @@ function AnnualReportPage() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const handleGenerateDualReport = () => {
|
||||||
|
if (selectedPairYear === null) return
|
||||||
|
const yearParam = selectedPairYear === 'all' ? 0 : selectedPairYear
|
||||||
|
navigate(`/dual-report?year=${yearParam}`)
|
||||||
|
}
|
||||||
|
|
||||||
if (isLoading) {
|
if (isLoading) {
|
||||||
return (
|
return (
|
||||||
<div className="annual-report-page">
|
<div className="annual-report-page">
|
||||||
@@ -67,21 +78,39 @@ function AnnualReportPage() {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const yearOptions: YearOption[] = availableYears.length > 0
|
||||||
|
? ['all', ...availableYears]
|
||||||
|
: []
|
||||||
|
|
||||||
|
const getYearLabel = (value: YearOption | null) => {
|
||||||
|
if (!value) return ''
|
||||||
|
return value === 'all' ? '全部时间' : `${value} 年`
|
||||||
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="annual-report-page">
|
<div className="annual-report-page">
|
||||||
<Sparkles size={32} className="header-icon" />
|
<Sparkles size={32} className="header-icon" />
|
||||||
<h1 className="page-title">年度报告</h1>
|
<h1 className="page-title">年度报告</h1>
|
||||||
<p className="page-desc">选择年份,生成你的微信聊天年度回顾</p>
|
<p className="page-desc">选择年份,回顾你在微信里的点点滴滴</p>
|
||||||
|
|
||||||
|
<div className="report-sections">
|
||||||
|
<section className="report-section">
|
||||||
|
<div className="section-header">
|
||||||
|
<div>
|
||||||
|
<h2 className="section-title">总年度报告</h2>
|
||||||
|
<p className="section-desc">包含所有会话与消息</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
<div className="year-grid">
|
<div className="year-grid">
|
||||||
{availableYears.map(year => (
|
{yearOptions.map(option => (
|
||||||
<div
|
<div
|
||||||
key={year}
|
key={option}
|
||||||
className={`year-card ${selectedYear === year ? 'selected' : ''}`}
|
className={`year-card ${option === 'all' ? 'all-time' : ''} ${selectedYear === option ? 'selected' : ''}`}
|
||||||
onClick={() => setSelectedYear(year)}
|
onClick={() => setSelectedYear(option)}
|
||||||
>
|
>
|
||||||
<span className="year-number">{year}</span>
|
<span className="year-number">{option === 'all' ? '全部' : option}</span>
|
||||||
<span className="year-label">年</span>
|
<span className="year-label">{option === 'all' ? '时间' : '年'}</span>
|
||||||
</div>
|
</div>
|
||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
@@ -99,10 +128,48 @@ function AnnualReportPage() {
|
|||||||
) : (
|
) : (
|
||||||
<>
|
<>
|
||||||
<Sparkles size={20} />
|
<Sparkles size={20} />
|
||||||
<span>生成 {selectedYear} 年度报告</span>
|
<span>生成 {getYearLabel(selectedYear)} 年度报告</span>
|
||||||
</>
|
</>
|
||||||
)}
|
)}
|
||||||
</button>
|
</button>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section className="report-section">
|
||||||
|
<div className="section-header">
|
||||||
|
<div>
|
||||||
|
<h2 className="section-title">双人年度报告</h2>
|
||||||
|
<p className="section-desc">选择一位好友,只看你们的私聊</p>
|
||||||
|
</div>
|
||||||
|
<div className="section-badge">
|
||||||
|
<Users size={16} />
|
||||||
|
<span>私聊</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="year-grid">
|
||||||
|
{yearOptions.map(option => (
|
||||||
|
<div
|
||||||
|
key={`pair-${option}`}
|
||||||
|
className={`year-card ${option === 'all' ? 'all-time' : ''} ${selectedPairYear === option ? 'selected' : ''}`}
|
||||||
|
onClick={() => setSelectedPairYear(option)}
|
||||||
|
>
|
||||||
|
<span className="year-number">{option === 'all' ? '全部' : option}</span>
|
||||||
|
<span className="year-label">{option === 'all' ? '时间' : '年'}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button
|
||||||
|
className="generate-btn secondary"
|
||||||
|
onClick={handleGenerateDualReport}
|
||||||
|
disabled={!selectedPairYear}
|
||||||
|
>
|
||||||
|
<Users size={20} />
|
||||||
|
<span>选择好友并生成报告</span>
|
||||||
|
</button>
|
||||||
|
<p className="section-hint">从聊天排行中选择好友生成双人报告</p>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1279,3 +1279,134 @@
|
|||||||
color: var(--ar-text-sub) !important;
|
color: var(--ar-text-sub) !important;
|
||||||
text-align: center;
|
text-align: center;
|
||||||
}
|
}
|
||||||
|
// 曾经的好朋友 视觉效果
|
||||||
|
.lost-friend-visual {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 32px;
|
||||||
|
margin: 64px auto 48px;
|
||||||
|
position: relative;
|
||||||
|
max-width: 480px;
|
||||||
|
|
||||||
|
.avatar-group {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
z-index: 2;
|
||||||
|
|
||||||
|
.avatar-label {
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
font-weight: 500;
|
||||||
|
opacity: 0.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.sender {
|
||||||
|
animation: fadeInRight 1s ease-out backwards;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.receiver {
|
||||||
|
animation: fadeInLeft 1s ease-out backwards;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.fading-line {
|
||||||
|
position: relative;
|
||||||
|
flex: 1;
|
||||||
|
height: 2px;
|
||||||
|
min-width: 120px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
|
||||||
|
.line-path {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
background: linear-gradient(to right,
|
||||||
|
var(--ar-primary) 0%,
|
||||||
|
rgba(var(--ar-primary-rgb), 0.4) 50%,
|
||||||
|
rgba(var(--ar-primary-rgb), 0.05) 100%);
|
||||||
|
border-radius: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.line-glow {
|
||||||
|
position: absolute;
|
||||||
|
inset: -4px 0;
|
||||||
|
background: linear-gradient(to right,
|
||||||
|
rgba(var(--ar-primary-rgb), 0.2) 0%,
|
||||||
|
transparent 100%);
|
||||||
|
filter: blur(8px);
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.flow-particle {
|
||||||
|
position: absolute;
|
||||||
|
width: 40px;
|
||||||
|
height: 2px;
|
||||||
|
background: linear-gradient(to right, transparent, var(--ar-primary), transparent);
|
||||||
|
border-radius: 2px;
|
||||||
|
opacity: 0;
|
||||||
|
animation: flowAcross 4s infinite linear;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.hero-desc.fading {
|
||||||
|
opacity: 0.7;
|
||||||
|
font-style: italic;
|
||||||
|
font-size: 16px;
|
||||||
|
margin-top: 32px;
|
||||||
|
line-height: 1.8;
|
||||||
|
letter-spacing: 0.05em;
|
||||||
|
animation: fadeIn 1.5s ease-out 0.5s backwards;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes flowAcross {
|
||||||
|
0% {
|
||||||
|
left: -20%;
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
10% {
|
||||||
|
opacity: 0.8;
|
||||||
|
}
|
||||||
|
|
||||||
|
50% {
|
||||||
|
opacity: 0.4;
|
||||||
|
}
|
||||||
|
|
||||||
|
90% {
|
||||||
|
opacity: 0.1;
|
||||||
|
}
|
||||||
|
|
||||||
|
100% {
|
||||||
|
left: 120%;
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeInRight {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateX(-20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateX(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes fadeInLeft {
|
||||||
|
from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateX(20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
opacity: 1;
|
||||||
|
transform: translateX(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { useState, useEffect, useRef } from 'react'
|
import { useState, useEffect, useRef } from 'react'
|
||||||
import { Loader2, Download, Image, Check, X } from 'lucide-react'
|
import { Loader2, Download, Image, Check, X, SlidersHorizontal } from 'lucide-react'
|
||||||
import html2canvas from 'html2canvas'
|
import html2canvas from 'html2canvas'
|
||||||
import { useThemeStore } from '../stores/themeStore'
|
import { useThemeStore } from '../stores/themeStore'
|
||||||
import './AnnualReportWindow.scss'
|
import './AnnualReportWindow.scss'
|
||||||
@@ -71,6 +71,20 @@ interface AnnualReportData {
|
|||||||
socialInitiative?: { initiatedChats: number; receivedChats: number; initiativeRate: number } | null
|
socialInitiative?: { initiatedChats: number; receivedChats: number; initiativeRate: number } | null
|
||||||
responseSpeed?: { avgResponseTime: number; fastestFriend: string; fastestTime: number } | null
|
responseSpeed?: { avgResponseTime: number; fastestFriend: string; fastestTime: number } | null
|
||||||
topPhrases?: { phrase: string; count: number }[]
|
topPhrases?: { phrase: string; count: number }[]
|
||||||
|
snsStats?: {
|
||||||
|
totalPosts: number
|
||||||
|
typeCounts?: Record<string, number>
|
||||||
|
topLikers: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
topLiked: { username: string; displayName: string; avatarUrl?: string; count: number }[]
|
||||||
|
}
|
||||||
|
lostFriend: {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
avatarUrl?: string
|
||||||
|
earlyCount: number
|
||||||
|
lateCount: number
|
||||||
|
periodDesc: string
|
||||||
|
} | null
|
||||||
}
|
}
|
||||||
|
|
||||||
interface SectionInfo {
|
interface SectionInfo {
|
||||||
@@ -249,6 +263,7 @@ function AnnualReportWindow() {
|
|||||||
const [fabOpen, setFabOpen] = useState(false)
|
const [fabOpen, setFabOpen] = useState(false)
|
||||||
const [loadingProgress, setLoadingProgress] = useState(0)
|
const [loadingProgress, setLoadingProgress] = useState(0)
|
||||||
const [loadingStage, setLoadingStage] = useState('正在初始化...')
|
const [loadingStage, setLoadingStage] = useState('正在初始化...')
|
||||||
|
const [exportMode, setExportMode] = useState<'separate' | 'long'>('separate')
|
||||||
|
|
||||||
const { currentTheme, themeMode } = useThemeStore()
|
const { currentTheme, themeMode } = useThemeStore()
|
||||||
|
|
||||||
@@ -273,6 +288,8 @@ function AnnualReportWindow() {
|
|||||||
responseSpeed: useRef<HTMLElement>(null),
|
responseSpeed: useRef<HTMLElement>(null),
|
||||||
topPhrases: useRef<HTMLElement>(null),
|
topPhrases: useRef<HTMLElement>(null),
|
||||||
ranking: useRef<HTMLElement>(null),
|
ranking: useRef<HTMLElement>(null),
|
||||||
|
sns: useRef<HTMLElement>(null),
|
||||||
|
lostFriend: useRef<HTMLElement>(null),
|
||||||
ending: useRef<HTMLElement>(null),
|
ending: useRef<HTMLElement>(null),
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -281,7 +298,8 @@ function AnnualReportWindow() {
|
|||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const params = new URLSearchParams(window.location.hash.split('?')[1] || '')
|
const params = new URLSearchParams(window.location.hash.split('?')[1] || '')
|
||||||
const yearParam = params.get('year')
|
const yearParam = params.get('year')
|
||||||
const year = yearParam ? parseInt(yearParam) : new Date().getFullYear()
|
const parsedYear = yearParam ? parseInt(yearParam, 10) : new Date().getFullYear()
|
||||||
|
const year = Number.isNaN(parsedYear) ? new Date().getFullYear() : parsedYear
|
||||||
generateReport(year)
|
generateReport(year)
|
||||||
}, [])
|
}, [])
|
||||||
|
|
||||||
@@ -336,6 +354,11 @@ function AnnualReportWindow() {
|
|||||||
return `${Math.round(seconds / 3600)}小时`
|
return `${Math.round(seconds / 3600)}小时`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const formatYearLabel = (value: number, withSuffix: boolean = true) => {
|
||||||
|
if (value === 0) return '历史以来'
|
||||||
|
return withSuffix ? `${value}年` : `${value}`
|
||||||
|
}
|
||||||
|
|
||||||
// 获取可用的板块列表
|
// 获取可用的板块列表
|
||||||
const getAvailableSections = (): SectionInfo[] => {
|
const getAvailableSections = (): SectionInfo[] => {
|
||||||
if (!reportData) return []
|
if (!reportData) return []
|
||||||
@@ -366,10 +389,16 @@ function AnnualReportWindow() {
|
|||||||
if (reportData.responseSpeed) {
|
if (reportData.responseSpeed) {
|
||||||
sections.push({ id: 'responseSpeed', name: '回应速度', ref: sectionRefs.responseSpeed })
|
sections.push({ id: 'responseSpeed', name: '回应速度', ref: sectionRefs.responseSpeed })
|
||||||
}
|
}
|
||||||
|
if (reportData.lostFriend) {
|
||||||
|
sections.push({ id: 'lostFriend', name: '曾经的好朋友', ref: sectionRefs.lostFriend })
|
||||||
|
}
|
||||||
if (reportData.topPhrases && reportData.topPhrases.length > 0) {
|
if (reportData.topPhrases && reportData.topPhrases.length > 0) {
|
||||||
sections.push({ id: 'topPhrases', name: '年度常用语', ref: sectionRefs.topPhrases })
|
sections.push({ id: 'topPhrases', name: '年度常用语', ref: sectionRefs.topPhrases })
|
||||||
}
|
}
|
||||||
sections.push({ id: 'ranking', name: '好友排行', ref: sectionRefs.ranking })
|
sections.push({ id: 'ranking', name: '好友排行', ref: sectionRefs.ranking })
|
||||||
|
if (reportData.snsStats && reportData.snsStats.totalPosts > 0) {
|
||||||
|
sections.push({ id: 'sns', name: '朋友圈', ref: sectionRefs.sns })
|
||||||
|
}
|
||||||
sections.push({ id: 'ending', name: '尾声', ref: sectionRefs.ending })
|
sections.push({ id: 'ending', name: '尾声', ref: sectionRefs.ending })
|
||||||
return sections
|
return sections
|
||||||
}
|
}
|
||||||
@@ -490,7 +519,7 @@ function AnnualReportWindow() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 导出整个报告为长图
|
// 导出整个报告为长图
|
||||||
const exportFullReport = async () => {
|
const exportFullReport = async (filterIds?: Set<string>) => {
|
||||||
if (!containerRef.current) {
|
if (!containerRef.current) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -516,6 +545,16 @@ function AnnualReportWindow() {
|
|||||||
el.style.padding = '40px 0'
|
el.style.padding = '40px 0'
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// 如果有筛选,隐藏未选中的板块
|
||||||
|
if (filterIds) {
|
||||||
|
const available = getAvailableSections()
|
||||||
|
available.forEach(s => {
|
||||||
|
if (!filterIds.has(s.id) && s.ref.current) {
|
||||||
|
s.ref.current.style.display = 'none'
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
// 修复词云导出问题
|
// 修复词云导出问题
|
||||||
const wordCloudInner = container.querySelector('.word-cloud-inner') as HTMLElement
|
const wordCloudInner = container.querySelector('.word-cloud-inner') as HTMLElement
|
||||||
const wordTags = container.querySelectorAll('.word-tag') as NodeListOf<HTMLElement>
|
const wordTags = container.querySelectorAll('.word-tag') as NodeListOf<HTMLElement>
|
||||||
@@ -584,7 +623,8 @@ function AnnualReportWindow() {
|
|||||||
|
|
||||||
const dataUrl = outputCanvas.toDataURL('image/png')
|
const dataUrl = outputCanvas.toDataURL('image/png')
|
||||||
const link = document.createElement('a')
|
const link = document.createElement('a')
|
||||||
link.download = `${reportData?.year}年度报告.png`
|
const yearFilePrefix = reportData ? formatYearLabel(reportData.year, false) : ''
|
||||||
|
link.download = `${yearFilePrefix}年度报告${filterIds ? '_自定义' : ''}.png`
|
||||||
link.href = dataUrl
|
link.href = dataUrl
|
||||||
document.body.appendChild(link)
|
document.body.appendChild(link)
|
||||||
link.click()
|
link.click()
|
||||||
@@ -607,6 +647,13 @@ function AnnualReportWindow() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (exportMode === 'long') {
|
||||||
|
setShowExportModal(false)
|
||||||
|
await exportFullReport(selectedSections)
|
||||||
|
setSelectedSections(new Set())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
setIsExporting(true)
|
setIsExporting(true)
|
||||||
setShowExportModal(false)
|
setShowExportModal(false)
|
||||||
|
|
||||||
@@ -640,11 +687,12 @@ function AnnualReportWindow() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
setExportProgress('正在写入文件...')
|
setExportProgress('正在写入文件...')
|
||||||
|
const yearFilePrefix = reportData ? formatYearLabel(reportData.year, false) : ''
|
||||||
const exportResult = await window.electronAPI.annualReport.exportImages({
|
const exportResult = await window.electronAPI.annualReport.exportImages({
|
||||||
baseDir: dirResult.filePaths[0],
|
baseDir: dirResult.filePaths[0],
|
||||||
folderName: `${reportData?.year}年度报告_分模块`,
|
folderName: `${yearFilePrefix}年度报告_分模块`,
|
||||||
images: exportedImages.map((img) => ({
|
images: exportedImages.map((img) => ({
|
||||||
name: `${reportData?.year}年度报告_${img.name}.png`,
|
name: `${yearFilePrefix}年度报告_${img.name}.png`,
|
||||||
dataUrl: img.data
|
dataUrl: img.data
|
||||||
}))
|
}))
|
||||||
})
|
})
|
||||||
@@ -715,10 +763,14 @@ function AnnualReportWindow() {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
const { year, totalMessages, totalFriends, coreFriends, monthlyTopFriends, peakDay, longestStreak, activityHeatmap, midnightKing, selfAvatarUrl, mutualFriend, socialInitiative, responseSpeed, topPhrases } = reportData
|
const { year, totalMessages, totalFriends, coreFriends, monthlyTopFriends, peakDay, longestStreak, activityHeatmap, midnightKing, selfAvatarUrl, mutualFriend, socialInitiative, responseSpeed, topPhrases, lostFriend } = reportData
|
||||||
const topFriend = coreFriends[0]
|
const topFriend = coreFriends[0]
|
||||||
const mostActive = getMostActiveTime(activityHeatmap.data)
|
const mostActive = getMostActiveTime(activityHeatmap.data)
|
||||||
const socialStoryName = topFriend?.displayName || '好友'
|
const socialStoryName = topFriend?.displayName || '好友'
|
||||||
|
const yearTitle = formatYearLabel(year, true)
|
||||||
|
const yearTitleShort = formatYearLabel(year, false)
|
||||||
|
const monthlyTitle = year === 0 ? '历史以来月度好友' : `${year}年月度好友`
|
||||||
|
const phrasesTitle = year === 0 ? '你在历史以来的常用语' : `你在${year}年的年度常用语`
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="annual-report-window">
|
<div className="annual-report-window">
|
||||||
@@ -735,9 +787,12 @@ function AnnualReportWindow() {
|
|||||||
|
|
||||||
{/* 浮动操作按钮 */}
|
{/* 浮动操作按钮 */}
|
||||||
<div className={`fab-container ${fabOpen ? 'open' : ''}`}>
|
<div className={`fab-container ${fabOpen ? 'open' : ''}`}>
|
||||||
<button className="fab-item" onClick={() => { setFabOpen(false); setShowExportModal(true) }} title="分模块导出">
|
<button className="fab-item" onClick={() => { setFabOpen(false); setExportMode('separate'); setShowExportModal(true) }} title="分模块导出">
|
||||||
<Image size={18} />
|
<Image size={18} />
|
||||||
</button>
|
</button>
|
||||||
|
<button className="fab-item" onClick={() => { setFabOpen(false); setExportMode('long'); setShowExportModal(true) }} title="自定义导出长图">
|
||||||
|
<SlidersHorizontal size={18} />
|
||||||
|
</button>
|
||||||
<button className="fab-item" onClick={() => { setFabOpen(false); exportFullReport() }} title="导出长图">
|
<button className="fab-item" onClick={() => { setFabOpen(false); exportFullReport() }} title="导出长图">
|
||||||
<Download size={18} />
|
<Download size={18} />
|
||||||
</button>
|
</button>
|
||||||
@@ -765,7 +820,7 @@ function AnnualReportWindow() {
|
|||||||
<div className="export-overlay" onClick={() => setShowExportModal(false)}>
|
<div className="export-overlay" onClick={() => setShowExportModal(false)}>
|
||||||
<div className="export-modal section-selector" onClick={e => e.stopPropagation()}>
|
<div className="export-modal section-selector" onClick={e => e.stopPropagation()}>
|
||||||
<div className="modal-header">
|
<div className="modal-header">
|
||||||
<h3>选择要导出的板块</h3>
|
<h3>{exportMode === 'long' ? '自定义导出长图' : '选择要导出的板块'}</h3>
|
||||||
<button className="close-btn" onClick={() => setShowExportModal(false)}>
|
<button className="close-btn" onClick={() => setShowExportModal(false)}>
|
||||||
<X size={20} />
|
<X size={20} />
|
||||||
</button>
|
</button>
|
||||||
@@ -793,7 +848,7 @@ function AnnualReportWindow() {
|
|||||||
onClick={exportSelectedSections}
|
onClick={exportSelectedSections}
|
||||||
disabled={selectedSections.size === 0}
|
disabled={selectedSections.size === 0}
|
||||||
>
|
>
|
||||||
导出 {selectedSections.size > 0 ? `(${selectedSections.size})` : ''}
|
{exportMode === 'long' ? '生成长图' : '导出'} {selectedSections.size > 0 ? `(${selectedSections.size})` : ''}
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -806,7 +861,7 @@ function AnnualReportWindow() {
|
|||||||
{/* 封面 */}
|
{/* 封面 */}
|
||||||
<section className="section" ref={sectionRefs.cover}>
|
<section className="section" ref={sectionRefs.cover}>
|
||||||
<div className="label-text">WEFLOW · ANNUAL REPORT</div>
|
<div className="label-text">WEFLOW · ANNUAL REPORT</div>
|
||||||
<h1 className="hero-title">{year}年<br />微信聊天报告</h1>
|
<h1 className="hero-title">{yearTitle}<br />微信聊天报告</h1>
|
||||||
<hr className="divider" />
|
<hr className="divider" />
|
||||||
<p className="hero-desc">每一条消息背后<br />都藏着一段独特的故事</p>
|
<p className="hero-desc">每一条消息背后<br />都藏着一段独特的故事</p>
|
||||||
</section>
|
</section>
|
||||||
@@ -848,7 +903,7 @@ function AnnualReportWindow() {
|
|||||||
{/* 月度好友 */}
|
{/* 月度好友 */}
|
||||||
<section className="section" ref={sectionRefs.monthlyFriends}>
|
<section className="section" ref={sectionRefs.monthlyFriends}>
|
||||||
<div className="label-text">月度好友</div>
|
<div className="label-text">月度好友</div>
|
||||||
<h2 className="hero-title">{year}年月度好友</h2>
|
<h2 className="hero-title">{monthlyTitle}</h2>
|
||||||
<p className="hero-desc">根据12个月的聊天习惯</p>
|
<p className="hero-desc">根据12个月的聊天习惯</p>
|
||||||
<div className="monthly-orbit">
|
<div className="monthly-orbit">
|
||||||
{monthlyTopFriends.map((m, i) => (
|
{monthlyTopFriends.map((m, i) => (
|
||||||
@@ -862,7 +917,7 @@ function AnnualReportWindow() {
|
|||||||
<Avatar url={selfAvatarUrl} name="我" size="lg" />
|
<Avatar url={selfAvatarUrl} name="我" size="lg" />
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<p className="hero-desc">只要你想<br />我一直在</p>
|
<p className="hero-desc">你只管说<br />我一直在</p>
|
||||||
</section>
|
</section>
|
||||||
|
|
||||||
{/* 双向奔赴 */}
|
{/* 双向奔赴 */}
|
||||||
@@ -962,15 +1017,15 @@ function AnnualReportWindow() {
|
|||||||
{midnightKing && (
|
{midnightKing && (
|
||||||
<section className="section" ref={sectionRefs.midnightKing}>
|
<section className="section" ref={sectionRefs.midnightKing}>
|
||||||
<div className="label-text">深夜好友</div>
|
<div className="label-text">深夜好友</div>
|
||||||
<h2 className="hero-title">当城市睡去</h2>
|
<h2 className="hero-title">月光下的你</h2>
|
||||||
<p className="hero-desc">这一年你留下了</p>
|
<p className="hero-desc">在这一年你留下了</p>
|
||||||
<div className="big-stat">
|
<div className="big-stat">
|
||||||
<span className="stat-num">{midnightKing.count}</span>
|
<span className="stat-num">{midnightKing.count}</span>
|
||||||
<span className="stat-unit">条深夜的消息</span>
|
<span className="stat-unit">条深夜的消息</span>
|
||||||
</div>
|
</div>
|
||||||
<p className="hero-desc">
|
<p className="hero-desc">
|
||||||
其中 <span className="hl">{midnightKing.displayName}</span> 常常在深夜中陪着你。
|
其中 <span className="hl">{midnightKing.displayName}</span> 常常在深夜中陪着你胡思乱想。
|
||||||
<br />你和Ta的对话占深夜期间聊天的 <span className="gold">{midnightKing.percentage}%</span>。
|
<br />你和Ta的对话占你深夜期间聊天的 <span className="gold">{midnightKing.percentage}%</span>。
|
||||||
</p>
|
</p>
|
||||||
</section>
|
</section>
|
||||||
)}
|
)}
|
||||||
@@ -991,11 +1046,46 @@ function AnnualReportWindow() {
|
|||||||
</section>
|
</section>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{/* 曾经的好朋友 */}
|
||||||
|
{lostFriend && (
|
||||||
|
<section className="section" ref={sectionRefs.lostFriend}>
|
||||||
|
<div className="label-text">曾经的好朋友</div>
|
||||||
|
<h2 className="hero-title">{lostFriend.displayName}</h2>
|
||||||
|
<div className="big-stat">
|
||||||
|
<span className="stat-num">{formatNumber(lostFriend.earlyCount)}</span>
|
||||||
|
<span className="stat-unit">条消息</span>
|
||||||
|
</div>
|
||||||
|
<p className="hero-desc">
|
||||||
|
在 <span className="hl">{lostFriend.periodDesc}</span>
|
||||||
|
<br />你们曾有聊不完的话题
|
||||||
|
</p>
|
||||||
|
<div className="lost-friend-visual">
|
||||||
|
<div className="avatar-group sender">
|
||||||
|
<Avatar url={lostFriend.avatarUrl} name={lostFriend.displayName} size="lg" />
|
||||||
|
<span className="avatar-label">TA</span>
|
||||||
|
</div>
|
||||||
|
<div className="fading-line">
|
||||||
|
<div className="line-path" />
|
||||||
|
<div className="line-glow" />
|
||||||
|
<div className="flow-particle" />
|
||||||
|
</div>
|
||||||
|
<div className="avatar-group receiver">
|
||||||
|
<Avatar url={selfAvatarUrl} name="我" size="lg" />
|
||||||
|
<span className="avatar-label">我</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<p className="hero-desc fading">
|
||||||
|
人类发明后悔
|
||||||
|
<br />来证明拥有的珍贵
|
||||||
|
</p>
|
||||||
|
</section>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* 年度常用语 - 词云 */}
|
{/* 年度常用语 - 词云 */}
|
||||||
{topPhrases && topPhrases.length > 0 && (
|
{topPhrases && topPhrases.length > 0 && (
|
||||||
<section className="section" ref={sectionRefs.topPhrases}>
|
<section className="section" ref={sectionRefs.topPhrases}>
|
||||||
<div className="label-text">年度常用语</div>
|
<div className="label-text">年度常用语</div>
|
||||||
<h2 className="hero-title">你在{year}年的年度常用语</h2>
|
<h2 className="hero-title">{phrasesTitle}</h2>
|
||||||
<p className="hero-desc">
|
<p className="hero-desc">
|
||||||
这一年,你说得最多的是:
|
这一年,你说得最多的是:
|
||||||
<br />
|
<br />
|
||||||
@@ -1008,6 +1098,57 @@ function AnnualReportWindow() {
|
|||||||
</section>
|
</section>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{/* 朋友圈 */}
|
||||||
|
{reportData.snsStats && reportData.snsStats.totalPosts > 0 && (
|
||||||
|
<section className="section" ref={sectionRefs.sns}>
|
||||||
|
<div className="label-text">朋友圈</div>
|
||||||
|
<h2 className="hero-title">记录生活时刻</h2>
|
||||||
|
<p className="hero-desc">
|
||||||
|
这一年,你发布了
|
||||||
|
</p>
|
||||||
|
<div className="big-stat">
|
||||||
|
<span className="stat-num">{reportData.snsStats.totalPosts}</span>
|
||||||
|
<span className="stat-unit">条朋友圈</span>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="sns-stats-container" style={{ display: 'flex', gap: '60px', marginTop: '40px', justifyContent: 'center' }}>
|
||||||
|
{reportData.snsStats.topLikers.length > 0 && (
|
||||||
|
<div className="sns-sub-stat" style={{ textAlign: 'left' }}>
|
||||||
|
<h3 className="sub-title" style={{ fontSize: '18px', marginBottom: '16px', opacity: 0.8, borderBottom: '1px solid currentColor', paddingBottom: '8px' }}>更关心你的Ta</h3>
|
||||||
|
<div className="mini-ranking">
|
||||||
|
{reportData.snsStats.topLikers.slice(0, 3).map((u, i) => (
|
||||||
|
<div key={i} className="mini-rank-item" style={{ display: 'flex', alignItems: 'center', gap: '12px', marginBottom: '14px' }}>
|
||||||
|
<Avatar url={u.avatarUrl} name={u.displayName} size="sm" />
|
||||||
|
<div style={{ display: 'flex', flexDirection: 'column' }}>
|
||||||
|
<span className="name" style={{ fontSize: '15px', fontWeight: 500, maxWidth: '120px', overflow: 'hidden', textOverflow: 'ellipsis', whiteSpace: 'nowrap' }}>{u.displayName}</span>
|
||||||
|
</div>
|
||||||
|
<span className="count hl" style={{ fontSize: '14px', marginLeft: 'auto' }}>{u.count}赞</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{reportData.snsStats.topLiked.length > 0 && (
|
||||||
|
<div className="sns-sub-stat" style={{ textAlign: 'left' }}>
|
||||||
|
<h3 className="sub-title" style={{ fontSize: '18px', marginBottom: '16px', opacity: 0.8, borderBottom: '1px solid currentColor', paddingBottom: '8px' }}>你最关心的Ta</h3>
|
||||||
|
<div className="mini-ranking">
|
||||||
|
{reportData.snsStats.topLiked.slice(0, 3).map((u, i) => (
|
||||||
|
<div key={i} className="mini-rank-item" style={{ display: 'flex', alignItems: 'center', gap: '12px', marginBottom: '14px' }}>
|
||||||
|
<Avatar url={u.avatarUrl} name={u.displayName} size="sm" />
|
||||||
|
<div style={{ display: 'flex', flexDirection: 'column' }}>
|
||||||
|
<span className="name" style={{ fontSize: '15px', fontWeight: 500, maxWidth: '120px', overflow: 'hidden', textOverflow: 'ellipsis', whiteSpace: 'nowrap' }}>{u.displayName}</span>
|
||||||
|
</div>
|
||||||
|
<span className="count hl" style={{ fontSize: '14px', marginLeft: 'auto' }}>{u.count}赞</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* 好友排行 */}
|
{/* 好友排行 */}
|
||||||
<section className="section" ref={sectionRefs.ranking}>
|
<section className="section" ref={sectionRefs.ranking}>
|
||||||
<div className="label-text">好友排行</div>
|
<div className="label-text">好友排行</div>
|
||||||
@@ -1064,7 +1205,7 @@ function AnnualReportWindow() {
|
|||||||
<br />愿新的一年,
|
<br />愿新的一年,
|
||||||
<br />所有期待,皆有回声。
|
<br />所有期待,皆有回声。
|
||||||
</p>
|
</p>
|
||||||
<div className="ending-year">{year}</div>
|
<div className="ending-year">{yearTitleShort}</div>
|
||||||
<div className="ending-brand">WEFLOW</div>
|
<div className="ending-brand">WEFLOW</div>
|
||||||
</section>
|
</section>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
132
src/pages/ChatHistoryPage.scss
Normal file
132
src/pages/ChatHistoryPage.scss
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
.chat-history-page {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
height: 100vh;
|
||||||
|
background: var(--bg-primary);
|
||||||
|
|
||||||
|
.history-list {
|
||||||
|
flex: 1;
|
||||||
|
overflow-y: auto;
|
||||||
|
padding: 16px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
|
||||||
|
.status-msg {
|
||||||
|
text-align: center;
|
||||||
|
padding: 40px 20px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 14px;
|
||||||
|
|
||||||
|
&.error {
|
||||||
|
color: var(--danger);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.empty {
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.history-item {
|
||||||
|
display: flex;
|
||||||
|
gap: 12px;
|
||||||
|
align-items: flex-start;
|
||||||
|
|
||||||
|
.avatar {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 4px;
|
||||||
|
overflow: hidden;
|
||||||
|
flex-shrink: 0;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
|
||||||
|
img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar-placeholder {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 500;
|
||||||
|
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.content-wrapper {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
|
||||||
|
.header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
margin-bottom: 6px;
|
||||||
|
|
||||||
|
.sender {
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.time {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
flex-shrink: 0;
|
||||||
|
margin-left: 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.bubble {
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
padding: 10px 14px;
|
||||||
|
border-radius: 18px 18px 18px 4px;
|
||||||
|
word-wrap: break-word;
|
||||||
|
max-width: 100%;
|
||||||
|
display: inline-block;
|
||||||
|
|
||||||
|
&.image-bubble {
|
||||||
|
padding: 0;
|
||||||
|
background: transparent;
|
||||||
|
}
|
||||||
|
|
||||||
|
.text-content {
|
||||||
|
font-size: 14px;
|
||||||
|
line-height: 1.6;
|
||||||
|
color: var(--text-primary);
|
||||||
|
white-space: pre-wrap;
|
||||||
|
word-break: break-word;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-content {
|
||||||
|
img {
|
||||||
|
max-width: 100%;
|
||||||
|
max-height: 300px;
|
||||||
|
border-radius: 8px;
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-tip {
|
||||||
|
padding: 8px 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-placeholder {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
padding: 4px 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
250
src/pages/ChatHistoryPage.tsx
Normal file
250
src/pages/ChatHistoryPage.tsx
Normal file
@@ -0,0 +1,250 @@
|
|||||||
|
import { useEffect, useState } from 'react'
|
||||||
|
import { useParams, useLocation } from 'react-router-dom'
|
||||||
|
import { ChatRecordItem } from '../types/models'
|
||||||
|
import TitleBar from '../components/TitleBar'
|
||||||
|
import './ChatHistoryPage.scss'
|
||||||
|
|
||||||
|
export default function ChatHistoryPage() {
|
||||||
|
const params = useParams<{ sessionId: string; messageId: string }>()
|
||||||
|
const location = useLocation()
|
||||||
|
const [recordList, setRecordList] = useState<ChatRecordItem[]>([])
|
||||||
|
const [loading, setLoading] = useState(true)
|
||||||
|
const [title, setTitle] = useState('聊天记录')
|
||||||
|
const [error, setError] = useState('')
|
||||||
|
|
||||||
|
// 简单的 XML 标签内容提取
|
||||||
|
const extractXmlValue = (xml: string, tag: string): string => {
|
||||||
|
const match = new RegExp(`<${tag}>([\\s\\S]*?)<\\/${tag}>`).exec(xml)
|
||||||
|
return match ? match[1] : ''
|
||||||
|
}
|
||||||
|
|
||||||
|
// 简单的 HTML 实体解码
|
||||||
|
const decodeHtmlEntities = (text?: string): string | undefined => {
|
||||||
|
if (!text) return text
|
||||||
|
return text
|
||||||
|
.replace(/</g, '<')
|
||||||
|
.replace(/>/g, '>')
|
||||||
|
.replace(/&/g, '&')
|
||||||
|
.replace(/"/g, '"')
|
||||||
|
.replace(/'/g, "'")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 前端兜底解析合并转发聊天记录
|
||||||
|
const parseChatHistory = (content: string): ChatRecordItem[] | undefined => {
|
||||||
|
try {
|
||||||
|
const type = extractXmlValue(content, 'type')
|
||||||
|
if (type !== '19') return undefined
|
||||||
|
|
||||||
|
const match = /<recorditem>[\s\S]*?<!\[CDATA\[([\s\S]*?)\]\]>[\s\S]*?<\/recorditem>/.exec(content)
|
||||||
|
if (!match) return undefined
|
||||||
|
|
||||||
|
const innerXml = match[1]
|
||||||
|
const items: ChatRecordItem[] = []
|
||||||
|
const itemRegex = /<dataitem\s+(.*?)>([\s\S]*?)<\/dataitem>/g
|
||||||
|
let itemMatch: RegExpExecArray | null
|
||||||
|
|
||||||
|
while ((itemMatch = itemRegex.exec(innerXml)) !== null) {
|
||||||
|
const attrs = itemMatch[1]
|
||||||
|
const body = itemMatch[2]
|
||||||
|
|
||||||
|
const datatypeMatch = /datatype="(\d+)"/.exec(attrs)
|
||||||
|
const datatype = datatypeMatch ? parseInt(datatypeMatch[1]) : 0
|
||||||
|
|
||||||
|
const sourcename = extractXmlValue(body, 'sourcename')
|
||||||
|
const sourcetime = extractXmlValue(body, 'sourcetime')
|
||||||
|
const sourceheadurl = extractXmlValue(body, 'sourceheadurl')
|
||||||
|
const datadesc = extractXmlValue(body, 'datadesc')
|
||||||
|
const datatitle = extractXmlValue(body, 'datatitle')
|
||||||
|
const fileext = extractXmlValue(body, 'fileext')
|
||||||
|
const datasize = parseInt(extractXmlValue(body, 'datasize') || '0')
|
||||||
|
const messageuuid = extractXmlValue(body, 'messageuuid')
|
||||||
|
|
||||||
|
const dataurl = extractXmlValue(body, 'dataurl')
|
||||||
|
const datathumburl = extractXmlValue(body, 'datathumburl') || extractXmlValue(body, 'thumburl')
|
||||||
|
const datacdnurl = extractXmlValue(body, 'datacdnurl') || extractXmlValue(body, 'cdnurl')
|
||||||
|
const aeskey = extractXmlValue(body, 'aeskey') || extractXmlValue(body, 'qaeskey')
|
||||||
|
const md5 = extractXmlValue(body, 'md5') || extractXmlValue(body, 'datamd5')
|
||||||
|
const imgheight = parseInt(extractXmlValue(body, 'imgheight') || '0')
|
||||||
|
const imgwidth = parseInt(extractXmlValue(body, 'imgwidth') || '0')
|
||||||
|
const duration = parseInt(extractXmlValue(body, 'duration') || '0')
|
||||||
|
|
||||||
|
items.push({
|
||||||
|
datatype,
|
||||||
|
sourcename,
|
||||||
|
sourcetime,
|
||||||
|
sourceheadurl,
|
||||||
|
datadesc: decodeHtmlEntities(datadesc),
|
||||||
|
datatitle: decodeHtmlEntities(datatitle),
|
||||||
|
fileext,
|
||||||
|
datasize,
|
||||||
|
messageuuid,
|
||||||
|
dataurl: decodeHtmlEntities(dataurl),
|
||||||
|
datathumburl: decodeHtmlEntities(datathumburl),
|
||||||
|
datacdnurl: decodeHtmlEntities(datacdnurl),
|
||||||
|
aeskey: decodeHtmlEntities(aeskey),
|
||||||
|
md5,
|
||||||
|
imgheight,
|
||||||
|
imgwidth,
|
||||||
|
duration
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return items.length > 0 ? items : undefined
|
||||||
|
} catch (e) {
|
||||||
|
console.error('前端解析聊天记录失败:', e)
|
||||||
|
return undefined
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 统一从路由参数或 pathname 中解析 sessionId / messageId
|
||||||
|
const getIds = () => {
|
||||||
|
const sessionId = params.sessionId || ''
|
||||||
|
const messageId = params.messageId || ''
|
||||||
|
|
||||||
|
if (sessionId && messageId) {
|
||||||
|
return { sid: sessionId, mid: messageId }
|
||||||
|
}
|
||||||
|
|
||||||
|
// 独立窗口场景下没有 Route 包裹,用 pathname 手动解析
|
||||||
|
const match = /^\/chat-history\/([^/]+)\/([^/]+)/.exec(location.pathname)
|
||||||
|
if (match) {
|
||||||
|
return { sid: match[1], mid: match[2] }
|
||||||
|
}
|
||||||
|
|
||||||
|
return { sid: '', mid: '' }
|
||||||
|
}
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const loadData = async () => {
|
||||||
|
const { sid, mid } = getIds()
|
||||||
|
if (!sid || !mid) {
|
||||||
|
setError('无效的聊天记录链接')
|
||||||
|
setLoading(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.chat.getMessage(sid, parseInt(mid, 10))
|
||||||
|
if (result.success && result.message) {
|
||||||
|
const msg = result.message
|
||||||
|
// 优先使用后端解析好的列表
|
||||||
|
let records: ChatRecordItem[] | undefined = msg.chatRecordList
|
||||||
|
|
||||||
|
// 如果后端没有解析到,则在前端兜底解析一次
|
||||||
|
if ((!records || records.length === 0) && msg.content) {
|
||||||
|
records = parseChatHistory(msg.content) || []
|
||||||
|
}
|
||||||
|
|
||||||
|
if (records && records.length > 0) {
|
||||||
|
setRecordList(records)
|
||||||
|
const match = /<title>(.*?)<\/title>/.exec(msg.content || '')
|
||||||
|
if (match) setTitle(match[1])
|
||||||
|
} else {
|
||||||
|
setError('暂时无法解析这条聊天记录')
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
setError(result.error || '获取消息失败')
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error(e)
|
||||||
|
setError('加载详情失败')
|
||||||
|
} finally {
|
||||||
|
setLoading(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
loadData()
|
||||||
|
}, [params.sessionId, params.messageId, location.pathname])
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="chat-history-page">
|
||||||
|
<TitleBar title={title} />
|
||||||
|
<div className="history-list">
|
||||||
|
{loading ? (
|
||||||
|
<div className="status-msg">加载中...</div>
|
||||||
|
) : error ? (
|
||||||
|
<div className="status-msg error">{error}</div>
|
||||||
|
) : recordList.length === 0 ? (
|
||||||
|
<div className="status-msg empty">暂无可显示的聊天记录</div>
|
||||||
|
) : (
|
||||||
|
recordList.map((item, i) => (
|
||||||
|
<HistoryItem key={i} item={item} />
|
||||||
|
))
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
function HistoryItem({ item }: { item: ChatRecordItem }) {
|
||||||
|
// sourcetime 在合并转发里有两种格式:
|
||||||
|
// 1) 时间戳(秒) 2) 已格式化的字符串 "2026-01-21 09:56:46"
|
||||||
|
let time = ''
|
||||||
|
if (item.sourcetime) {
|
||||||
|
if (/^\d+$/.test(item.sourcetime)) {
|
||||||
|
time = new Date(parseInt(item.sourcetime, 10) * 1000).toLocaleString()
|
||||||
|
} else {
|
||||||
|
time = item.sourcetime
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const renderContent = () => {
|
||||||
|
if (item.datatype === 1) {
|
||||||
|
// 文本消息
|
||||||
|
return <div className="text-content">{item.datadesc || ''}</div>
|
||||||
|
}
|
||||||
|
if (item.datatype === 3) {
|
||||||
|
// 图片
|
||||||
|
const src = item.datathumburl || item.datacdnurl
|
||||||
|
if (src) {
|
||||||
|
return (
|
||||||
|
<div className="media-content">
|
||||||
|
<img
|
||||||
|
src={src}
|
||||||
|
alt="图片"
|
||||||
|
referrerPolicy="no-referrer"
|
||||||
|
onError={(e) => {
|
||||||
|
const target = e.target as HTMLImageElement
|
||||||
|
target.style.display = 'none'
|
||||||
|
const placeholder = document.createElement('div')
|
||||||
|
placeholder.className = 'media-tip'
|
||||||
|
placeholder.textContent = '图片无法加载'
|
||||||
|
target.parentElement?.appendChild(placeholder)
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
return <div className="media-placeholder">[图片]</div>
|
||||||
|
}
|
||||||
|
if (item.datatype === 43) {
|
||||||
|
return <div className="media-placeholder">[视频] {item.datatitle}</div>
|
||||||
|
}
|
||||||
|
if (item.datatype === 34) {
|
||||||
|
return <div className="media-placeholder">[语音] {item.duration ? (item.duration / 1000).toFixed(0) + '"' : ''}</div>
|
||||||
|
}
|
||||||
|
// Fallback
|
||||||
|
return <div className="text-content">{item.datadesc || item.datatitle || '[不支持的消息类型]'}</div>
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="history-item">
|
||||||
|
<div className="avatar">
|
||||||
|
{item.sourceheadurl ? (
|
||||||
|
<img src={item.sourceheadurl} alt="" referrerPolicy="no-referrer" />
|
||||||
|
) : (
|
||||||
|
<div className="avatar-placeholder">
|
||||||
|
{item.sourcename?.slice(0, 1)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="content-wrapper">
|
||||||
|
<div className="header">
|
||||||
|
<span className="sender">{item.sourcename || '未知发送者'}</span>
|
||||||
|
<span className="time">{time}</span>
|
||||||
|
</div>
|
||||||
|
<div className={`bubble ${item.datatype === 3 ? 'image-bubble' : ''}`}>
|
||||||
|
{renderContent()}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
551
src/pages/ContactsPage.scss
Normal file
551
src/pages/ContactsPage.scss
Normal file
@@ -0,0 +1,551 @@
|
|||||||
|
.contacts-page {
|
||||||
|
display: flex;
|
||||||
|
height: calc(100% + 48px);
|
||||||
|
margin: -24px;
|
||||||
|
background: var(--bg-primary);
|
||||||
|
overflow: hidden;
|
||||||
|
|
||||||
|
// 左侧联系人面板
|
||||||
|
.contacts-panel {
|
||||||
|
width: 380px;
|
||||||
|
min-width: 380px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
border-right: 1px solid var(--border-color);
|
||||||
|
background: var(--card-bg);
|
||||||
|
}
|
||||||
|
|
||||||
|
.panel-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 20px 24px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
|
||||||
|
h2 {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.icon-btn {
|
||||||
|
width: 32px;
|
||||||
|
height: 32px;
|
||||||
|
border: none;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
|
||||||
|
.spin {
|
||||||
|
animation: contactsSpin 1s linear infinite;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.search-bar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
margin: 16px 20px;
|
||||||
|
padding: 10px 14px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border-radius: 10px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
transition: border-color 0.2s;
|
||||||
|
|
||||||
|
&:focus-within {
|
||||||
|
border-color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
input {
|
||||||
|
flex: 1;
|
||||||
|
border: none;
|
||||||
|
background: none;
|
||||||
|
outline: none;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
&::placeholder {
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.clear-btn {
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
padding: 4px;
|
||||||
|
cursor: pointer;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
border-radius: 4px;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.type-filters {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 0 20px 16px;
|
||||||
|
flex-wrap: nowrap;
|
||||||
|
overflow-x: auto;
|
||||||
|
|
||||||
|
&::-webkit-scrollbar {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.filter-chip {
|
||||||
|
display: 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;
|
||||||
|
|
||||||
|
input[type="checkbox"] {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
svg {
|
||||||
|
opacity: 0.7;
|
||||||
|
transition: transform 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
border-color: var(--text-tertiary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: var(--primary-light);
|
||||||
|
border-color: var(--primary);
|
||||||
|
color: var(--primary);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
opacity: 1;
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.contacts-count {
|
||||||
|
padding: 0 20px 12px;
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading-state,
|
||||||
|
.empty-state {
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
font-size: 14px;
|
||||||
|
|
||||||
|
.spin {
|
||||||
|
animation: contactsSpin 1s linear infinite;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.contacts-list {
|
||||||
|
flex: 1;
|
||||||
|
overflow-y: auto;
|
||||||
|
padding: 0 12px 12px;
|
||||||
|
|
||||||
|
&::-webkit-scrollbar {
|
||||||
|
width: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
&::-webkit-scrollbar-thumb {
|
||||||
|
background: var(--text-tertiary);
|
||||||
|
border-radius: 3px;
|
||||||
|
opacity: 0.3;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 12px;
|
||||||
|
border-radius: 10px;
|
||||||
|
transition: all 0.2s;
|
||||||
|
margin-bottom: 4px;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-avatar {
|
||||||
|
width: 44px;
|
||||||
|
height: 44px;
|
||||||
|
border-radius: 10px;
|
||||||
|
background: linear-gradient(135deg, var(--primary), var(--primary-hover));
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
overflow: hidden;
|
||||||
|
flex-shrink: 0;
|
||||||
|
|
||||||
|
img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
|
||||||
|
span {
|
||||||
|
color: #fff;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-info {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-name {
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-remark {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
margin-top: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.contact-type {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 4px;
|
||||||
|
padding: 4px 10px;
|
||||||
|
border-radius: 6px;
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: 500;
|
||||||
|
flex-shrink: 0;
|
||||||
|
|
||||||
|
&.friend {
|
||||||
|
background: rgba(var(--primary-rgb), 0.1);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.group {
|
||||||
|
background: rgba(52, 211, 153, 0.1);
|
||||||
|
color: rgb(52, 211, 153);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.official {
|
||||||
|
background: rgba(251, 191, 36, 0.1);
|
||||||
|
color: rgb(251, 191, 36);
|
||||||
|
}
|
||||||
|
|
||||||
|
svg {
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 右侧设置面板
|
||||||
|
.settings-panel {
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.settings-content {
|
||||||
|
flex: 1;
|
||||||
|
overflow-y: auto;
|
||||||
|
padding: 20px 24px;
|
||||||
|
|
||||||
|
&::-webkit-scrollbar {
|
||||||
|
width: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
&::-webkit-scrollbar-thumb {
|
||||||
|
background: var(--text-tertiary);
|
||||||
|
border-radius: 3px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.setting-section {
|
||||||
|
margin-bottom: 28px;
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
text-transform: uppercase;
|
||||||
|
letter-spacing: 0.5px;
|
||||||
|
margin: 0 0 14px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.format-select {
|
||||||
|
position: relative;
|
||||||
|
/* margin-bottom 移到 .setting-section */
|
||||||
|
|
||||||
|
.select-trigger {
|
||||||
|
width: 100%;
|
||||||
|
padding: 10px 16px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 9999px;
|
||||||
|
/* Rounded pill shape */
|
||||||
|
font-size: 14px;
|
||||||
|
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: 20;
|
||||||
|
max-height: 260px;
|
||||||
|
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: 14px;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: color-mix(in srgb, var(--primary) 12%, transparent);
|
||||||
|
color: var(--primary);
|
||||||
|
|
||||||
|
.option-desc {
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.option-label {
|
||||||
|
font-weight: 500;
|
||||||
|
}
|
||||||
|
|
||||||
|
.option-desc {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.checkbox-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
|
input[type="checkbox"] {
|
||||||
|
width: 18px;
|
||||||
|
height: 18px;
|
||||||
|
accent-color: var(--primary);
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.export-path-display {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 12px 16px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border-radius: 10px;
|
||||||
|
font-size: 13px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin-bottom: 12px;
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--primary);
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
span {
|
||||||
|
flex: 1;
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-folder-btn {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 10px 16px;
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 13px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary);
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
border-color: var(--primary);
|
||||||
|
color: var(--primary);
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
transform: scale(0.98);
|
||||||
|
}
|
||||||
|
|
||||||
|
svg {
|
||||||
|
color: var(--text-secondary);
|
||||||
|
transition: color 0.2s;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.export-action {
|
||||||
|
padding: 20px 24px;
|
||||||
|
border-top: 1px solid var(--border-color);
|
||||||
|
}
|
||||||
|
|
||||||
|
.export-btn {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 10px;
|
||||||
|
padding: 14px 24px;
|
||||||
|
background: var(--primary);
|
||||||
|
color: #fff;
|
||||||
|
border: none;
|
||||||
|
border-radius: 12px;
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover:not(:disabled) {
|
||||||
|
background: var(--primary-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
|
||||||
|
.spin {
|
||||||
|
animation: contactsSpin 1s linear infinite;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes contactsSpin {
|
||||||
|
from {
|
||||||
|
transform: rotate(0deg);
|
||||||
|
}
|
||||||
|
|
||||||
|
to {
|
||||||
|
transform: rotate(360deg);
|
||||||
|
}
|
||||||
|
}
|
||||||
379
src/pages/ContactsPage.tsx
Normal file
379
src/pages/ContactsPage.tsx
Normal file
@@ -0,0 +1,379 @@
|
|||||||
|
import { useState, useEffect, useCallback, useRef } from 'react'
|
||||||
|
import { Search, RefreshCw, X, User, Users, MessageSquare, Loader2, FolderOpen, Download, ChevronDown } from 'lucide-react'
|
||||||
|
import './ContactsPage.scss'
|
||||||
|
|
||||||
|
interface ContactInfo {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
remark?: string
|
||||||
|
nickname?: string
|
||||||
|
avatarUrl?: string
|
||||||
|
type: 'friend' | 'group' | 'official' | 'other'
|
||||||
|
}
|
||||||
|
|
||||||
|
function ContactsPage() {
|
||||||
|
const [contacts, setContacts] = useState<ContactInfo[]>([])
|
||||||
|
const [filteredContacts, setFilteredContacts] = useState<ContactInfo[]>([])
|
||||||
|
const [isLoading, setIsLoading] = useState(true)
|
||||||
|
const [searchKeyword, setSearchKeyword] = useState('')
|
||||||
|
const [contactTypes, setContactTypes] = useState({
|
||||||
|
friends: true,
|
||||||
|
groups: true,
|
||||||
|
officials: true
|
||||||
|
})
|
||||||
|
|
||||||
|
// 导出相关状态
|
||||||
|
const [exportFormat, setExportFormat] = useState<'json' | 'csv' | 'vcf'>('json')
|
||||||
|
const [exportAvatars, setExportAvatars] = useState(true)
|
||||||
|
const [exportFolder, setExportFolder] = useState('')
|
||||||
|
const [isExporting, setIsExporting] = useState(false)
|
||||||
|
const [showFormatSelect, setShowFormatSelect] = useState(false)
|
||||||
|
const formatDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
|
// 加载通讯录
|
||||||
|
const loadContacts = useCallback(async () => {
|
||||||
|
setIsLoading(true)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.chat.connect()
|
||||||
|
if (!result.success) {
|
||||||
|
console.error('连接失败:', result.error)
|
||||||
|
setIsLoading(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
const contactsResult = await window.electronAPI.chat.getContacts()
|
||||||
|
|
||||||
|
if (contactsResult.success && contactsResult.contacts) {
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// 获取头像URL
|
||||||
|
const usernames = contactsResult.contacts.map((c: ContactInfo) => c.username)
|
||||||
|
if (usernames.length > 0) {
|
||||||
|
const avatarResult = await window.electronAPI.chat.enrichSessionsContactInfo(usernames)
|
||||||
|
if (avatarResult.success && avatarResult.contacts) {
|
||||||
|
contactsResult.contacts.forEach((contact: ContactInfo) => {
|
||||||
|
const enriched = avatarResult.contacts?.[contact.username]
|
||||||
|
if (enriched?.avatarUrl) {
|
||||||
|
contact.avatarUrl = enriched.avatarUrl
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
setContacts(contactsResult.contacts)
|
||||||
|
setFilteredContacts(contactsResult.contacts)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('加载通讯录失败:', e)
|
||||||
|
} finally {
|
||||||
|
setIsLoading(false)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
loadContacts()
|
||||||
|
}, [loadContacts])
|
||||||
|
|
||||||
|
// 搜索和类型过滤
|
||||||
|
useEffect(() => {
|
||||||
|
let filtered = contacts
|
||||||
|
|
||||||
|
// 类型过滤
|
||||||
|
filtered = filtered.filter(c => {
|
||||||
|
if (c.type === 'friend' && !contactTypes.friends) return false
|
||||||
|
if (c.type === 'group' && !contactTypes.groups) return false
|
||||||
|
if (c.type === 'official' && !contactTypes.officials) return false
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
// 关键词过滤
|
||||||
|
if (searchKeyword.trim()) {
|
||||||
|
const lower = searchKeyword.toLowerCase()
|
||||||
|
filtered = filtered.filter(c =>
|
||||||
|
c.displayName?.toLowerCase().includes(lower) ||
|
||||||
|
c.remark?.toLowerCase().includes(lower) ||
|
||||||
|
c.username.toLowerCase().includes(lower)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
setFilteredContacts(filtered)
|
||||||
|
}, [searchKeyword, contacts, contactTypes])
|
||||||
|
|
||||||
|
// 点击外部关闭下拉菜单
|
||||||
|
useEffect(() => {
|
||||||
|
const handleClickOutside = (event: MouseEvent) => {
|
||||||
|
const target = event.target as Node
|
||||||
|
if (showFormatSelect && formatDropdownRef.current && !formatDropdownRef.current.contains(target)) {
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
document.addEventListener('mousedown', handleClickOutside)
|
||||||
|
return () => document.removeEventListener('mousedown', handleClickOutside)
|
||||||
|
}, [showFormatSelect])
|
||||||
|
|
||||||
|
const getAvatarLetter = (name: string) => {
|
||||||
|
if (!name) return '?'
|
||||||
|
return [...name][0] || '?'
|
||||||
|
}
|
||||||
|
|
||||||
|
const getContactTypeIcon = (type: string) => {
|
||||||
|
switch (type) {
|
||||||
|
case 'friend': return <User size={14} />
|
||||||
|
case 'group': return <Users size={14} />
|
||||||
|
case 'official': return <MessageSquare size={14} />
|
||||||
|
default: return <User size={14} />
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const getContactTypeName = (type: string) => {
|
||||||
|
switch (type) {
|
||||||
|
case 'friend': return '好友'
|
||||||
|
case 'group': return '群聊'
|
||||||
|
case 'official': return '公众号'
|
||||||
|
default: return '其他'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 选择导出文件夹
|
||||||
|
const selectExportFolder = async () => {
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.dialog.openDirectory({
|
||||||
|
title: '选择导出位置'
|
||||||
|
})
|
||||||
|
if (result && !result.canceled && result.filePaths && result.filePaths.length > 0) {
|
||||||
|
setExportFolder(result.filePaths[0])
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('选择文件夹失败:', e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 开始导出
|
||||||
|
const startExport = async () => {
|
||||||
|
if (!exportFolder) {
|
||||||
|
alert('请先选择导出位置')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
setIsExporting(true)
|
||||||
|
try {
|
||||||
|
const exportOptions = {
|
||||||
|
format: exportFormat,
|
||||||
|
exportAvatars,
|
||||||
|
contactTypes: {
|
||||||
|
friends: contactTypes.friends,
|
||||||
|
groups: contactTypes.groups,
|
||||||
|
officials: contactTypes.officials
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await window.electronAPI.export.exportContacts(exportFolder, exportOptions)
|
||||||
|
|
||||||
|
if (result.success) {
|
||||||
|
alert(`导出成功!共导出 ${result.successCount} 个联系人`)
|
||||||
|
} else {
|
||||||
|
alert(`导出失败:${result.error}`)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('导出失败:', e)
|
||||||
|
alert(`导出失败:${String(e)}`)
|
||||||
|
} finally {
|
||||||
|
setIsExporting(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const exportFormatOptions = [
|
||||||
|
{ value: 'json', label: 'JSON', desc: '详细格式,包含完整联系人信息' },
|
||||||
|
{ value: 'csv', label: 'CSV (Excel)', desc: '电子表格格式,适合Excel查看' },
|
||||||
|
{ value: 'vcf', label: 'VCF (vCard)', desc: '标准名片格式,支持导入手机' }
|
||||||
|
]
|
||||||
|
|
||||||
|
const getOptionLabel = (value: string) => {
|
||||||
|
return exportFormatOptions.find(opt => opt.value === value)?.label || value
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="contacts-page">
|
||||||
|
{/* 左侧:联系人列表 */}
|
||||||
|
<div className="contacts-panel">
|
||||||
|
<div className="panel-header">
|
||||||
|
<h2>通讯录</h2>
|
||||||
|
<button className="icon-btn" onClick={loadContacts} disabled={isLoading}>
|
||||||
|
<RefreshCw size={18} className={isLoading ? 'spin' : ''} />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="search-bar">
|
||||||
|
<Search size={16} />
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="搜索联系人..."
|
||||||
|
value={searchKeyword}
|
||||||
|
onChange={e => setSearchKeyword(e.target.value)}
|
||||||
|
/>
|
||||||
|
{searchKeyword && (
|
||||||
|
<button className="clear-btn" onClick={() => setSearchKeyword('')}>
|
||||||
|
<X size={14} />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="type-filters">
|
||||||
|
<label className={`filter-chip ${contactTypes.friends ? 'active' : ''}`}>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={contactTypes.friends}
|
||||||
|
onChange={e => setContactTypes({ ...contactTypes, friends: e.target.checked })}
|
||||||
|
/>
|
||||||
|
<User size={16} />
|
||||||
|
<span>好友</span>
|
||||||
|
</label>
|
||||||
|
<label className={`filter-chip ${contactTypes.groups ? 'active' : ''}`}>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={contactTypes.groups}
|
||||||
|
onChange={e => setContactTypes({ ...contactTypes, groups: e.target.checked })}
|
||||||
|
/>
|
||||||
|
<Users size={16} />
|
||||||
|
<span>群聊</span>
|
||||||
|
</label>
|
||||||
|
<label className={`filter-chip ${contactTypes.officials ? 'active' : ''}`}>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={contactTypes.officials}
|
||||||
|
onChange={e => setContactTypes({ ...contactTypes, officials: e.target.checked })}
|
||||||
|
/>
|
||||||
|
<MessageSquare size={16} />
|
||||||
|
<span>公众号</span>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="contacts-count">
|
||||||
|
共 {filteredContacts.length} 个联系人
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{isLoading ? (
|
||||||
|
<div className="loading-state">
|
||||||
|
<Loader2 size={32} className="spin" />
|
||||||
|
<span>加载中...</span>
|
||||||
|
</div>
|
||||||
|
) : filteredContacts.length === 0 ? (
|
||||||
|
<div className="empty-state">
|
||||||
|
<span>暂无联系人</span>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="contacts-list">
|
||||||
|
{filteredContacts.map(contact => (
|
||||||
|
<div key={contact.username} className="contact-item">
|
||||||
|
<div className="contact-avatar">
|
||||||
|
{contact.avatarUrl ? (
|
||||||
|
<img src={contact.avatarUrl} alt="" />
|
||||||
|
) : (
|
||||||
|
<span>{getAvatarLetter(contact.displayName)}</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="contact-info">
|
||||||
|
<div className="contact-name">{contact.displayName}</div>
|
||||||
|
{contact.remark && contact.remark !== contact.displayName && (
|
||||||
|
<div className="contact-remark">备注: {contact.remark}</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className={`contact-type ${contact.type}`}>
|
||||||
|
{getContactTypeIcon(contact.type)}
|
||||||
|
<span>{getContactTypeName(contact.type)}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 右侧:导出设置 */}
|
||||||
|
<div className="settings-panel">
|
||||||
|
<div className="panel-header">
|
||||||
|
<h2>导出设置</h2>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="settings-content">
|
||||||
|
<div className="setting-section">
|
||||||
|
<h3>导出格式</h3>
|
||||||
|
<div className="format-select" ref={formatDropdownRef}>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`select-trigger ${showFormatSelect ? 'open' : ''}`}
|
||||||
|
onClick={() => setShowFormatSelect(!showFormatSelect)}
|
||||||
|
>
|
||||||
|
<span className="select-value">{getOptionLabel(exportFormat)}</span>
|
||||||
|
<ChevronDown size={16} />
|
||||||
|
</button>
|
||||||
|
{showFormatSelect && (
|
||||||
|
<div className="select-dropdown">
|
||||||
|
{exportFormatOptions.map(option => (
|
||||||
|
<button
|
||||||
|
key={option.value}
|
||||||
|
type="button"
|
||||||
|
className={`select-option ${exportFormat === option.value ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setExportFormat(option.value as 'json' | 'csv' | 'vcf')
|
||||||
|
setShowFormatSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="option-label">{option.label}</span>
|
||||||
|
<span className="option-desc">{option.desc}</span>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="setting-section">
|
||||||
|
<h3>导出选项</h3>
|
||||||
|
<label className="checkbox-item">
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={exportAvatars}
|
||||||
|
onChange={e => setExportAvatars(e.target.checked)}
|
||||||
|
/>
|
||||||
|
<span>导出头像</span>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="setting-section">
|
||||||
|
<h3>导出位置</h3>
|
||||||
|
<div className="export-path-display">
|
||||||
|
<FolderOpen size={16} />
|
||||||
|
<span>{exportFolder || '未设置'}</span>
|
||||||
|
</div>
|
||||||
|
<button className="select-folder-btn" onClick={selectExportFolder}>
|
||||||
|
<FolderOpen size={16} />
|
||||||
|
<span>选择导出目录</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="export-action">
|
||||||
|
<button
|
||||||
|
className="export-btn"
|
||||||
|
onClick={startExport}
|
||||||
|
disabled={!exportFolder || isExporting}
|
||||||
|
>
|
||||||
|
{isExporting ? (
|
||||||
|
<>
|
||||||
|
<Loader2 size={18} className="spin" />
|
||||||
|
<span>导出中...</span>
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<Download size={18} />
|
||||||
|
<span>开始导出</span>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default ContactsPage
|
||||||
@@ -1,569 +0,0 @@
|
|||||||
.page-header {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: space-between;
|
|
||||||
margin-bottom: 24px;
|
|
||||||
|
|
||||||
h1 {
|
|
||||||
font-size: 24px;
|
|
||||||
font-weight: 600;
|
|
||||||
color: var(--text-primary);
|
|
||||||
margin: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
.header-tabs {
|
|
||||||
display: flex;
|
|
||||||
gap: 8px;
|
|
||||||
|
|
||||||
.tab-btn {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 6px;
|
|
||||||
padding: 8px 16px;
|
|
||||||
border: none;
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-secondary);
|
|
||||||
font-size: 14px;
|
|
||||||
cursor: pointer;
|
|
||||||
border-radius: 9999px;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
background: var(--border-color);
|
|
||||||
color: var(--text-primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.active {
|
|
||||||
background: var(--primary);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.page-scroll {
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
gap: 24px;
|
|
||||||
}
|
|
||||||
|
|
||||||
.page-section {
|
|
||||||
background: var(--bg-secondary);
|
|
||||||
border-radius: 16px;
|
|
||||||
padding: 20px 24px;
|
|
||||||
|
|
||||||
h2 {
|
|
||||||
font-size: 16px;
|
|
||||||
font-weight: 600;
|
|
||||||
color: var(--text-primary);
|
|
||||||
margin: 0 0 4px;
|
|
||||||
}
|
|
||||||
|
|
||||||
.section-desc {
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
margin: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
.section-header {
|
|
||||||
display: flex;
|
|
||||||
justify-content: space-between;
|
|
||||||
align-items: flex-start;
|
|
||||||
margin-bottom: 20px;
|
|
||||||
|
|
||||||
.section-actions {
|
|
||||||
display: flex;
|
|
||||||
gap: 10px;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.btn {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 6px;
|
|
||||||
padding: 8px 16px;
|
|
||||||
border: none;
|
|
||||||
border-radius: 9999px;
|
|
||||||
font-size: 14px;
|
|
||||||
font-weight: 500;
|
|
||||||
cursor: pointer;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:disabled {
|
|
||||||
opacity: 0.6;
|
|
||||||
cursor: not-allowed;
|
|
||||||
}
|
|
||||||
|
|
||||||
.spin {
|
|
||||||
animation: spin 1s linear infinite;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.btn-primary {
|
|
||||||
background: var(--primary);
|
|
||||||
color: white;
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
background: var(--primary-hover);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.btn-secondary {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-primary);
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
background: var(--border-color);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.btn-warning {
|
|
||||||
background: #f59e0b;
|
|
||||||
color: white;
|
|
||||||
|
|
||||||
&:hover:not(:disabled) {
|
|
||||||
background: #d97706;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.database-list {
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
gap: 8px;
|
|
||||||
}
|
|
||||||
|
|
||||||
.database-item {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 12px;
|
|
||||||
padding: 12px 16px;
|
|
||||||
background: var(--bg-primary);
|
|
||||||
border-radius: 12px;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
}
|
|
||||||
|
|
||||||
.status-icon {
|
|
||||||
width: 28px;
|
|
||||||
height: 28px;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
border-radius: 50%;
|
|
||||||
|
|
||||||
&.decrypted {
|
|
||||||
background: var(--primary);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.needs-update {
|
|
||||||
background: #f59e0b;
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.pending {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.db-info {
|
|
||||||
flex: 1;
|
|
||||||
min-width: 0;
|
|
||||||
|
|
||||||
.db-name {
|
|
||||||
font-size: 14px;
|
|
||||||
font-weight: 500;
|
|
||||||
color: var(--text-primary);
|
|
||||||
margin-bottom: 2px;
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
}
|
|
||||||
|
|
||||||
.db-meta {
|
|
||||||
display: flex;
|
|
||||||
gap: 6px;
|
|
||||||
font-size: 12px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.db-status {
|
|
||||||
padding: 4px 10px;
|
|
||||||
border-radius: 9999px;
|
|
||||||
font-size: 12px;
|
|
||||||
font-weight: 500;
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
&.decrypted {
|
|
||||||
background: rgba(34, 197, 94, 0.15);
|
|
||||||
color: #16a34a;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.needs-update {
|
|
||||||
background: rgba(245, 158, 11, 0.15);
|
|
||||||
color: #b45309;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.pending {
|
|
||||||
background: rgba(234, 179, 8, 0.15);
|
|
||||||
color: #b45309;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.empty-state {
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
padding: 48px 20px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
|
|
||||||
svg {
|
|
||||||
margin-bottom: 16px;
|
|
||||||
opacity: 0.5;
|
|
||||||
}
|
|
||||||
|
|
||||||
p {
|
|
||||||
margin: 0;
|
|
||||||
font-size: 14px;
|
|
||||||
|
|
||||||
&.hint {
|
|
||||||
margin-top: 6px;
|
|
||||||
font-size: 13px;
|
|
||||||
opacity: 0.7;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.unavailable-state {
|
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
padding: 64px 20px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
|
|
||||||
svg {
|
|
||||||
margin-bottom: 20px;
|
|
||||||
opacity: 0.4;
|
|
||||||
}
|
|
||||||
|
|
||||||
p {
|
|
||||||
margin: 0;
|
|
||||||
font-size: 15px;
|
|
||||||
color: var(--text-secondary);
|
|
||||||
|
|
||||||
&.hint {
|
|
||||||
margin-top: 8px;
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.message-toast {
|
|
||||||
position: fixed;
|
|
||||||
top: 60px;
|
|
||||||
left: 50%;
|
|
||||||
transform: translateX(-50%);
|
|
||||||
padding: 10px 24px;
|
|
||||||
border-radius: 9999px;
|
|
||||||
font-size: 14px;
|
|
||||||
z-index: 100;
|
|
||||||
animation: slideDown 0.3s ease;
|
|
||||||
|
|
||||||
&.success {
|
|
||||||
background: var(--primary);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.error {
|
|
||||||
background: var(--danger);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@keyframes slideDown {
|
|
||||||
from {
|
|
||||||
opacity: 0;
|
|
||||||
transform: translateX(-50%) translateY(-10px);
|
|
||||||
}
|
|
||||||
to {
|
|
||||||
opacity: 1;
|
|
||||||
transform: translateX(-50%) translateY(0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@keyframes spin {
|
|
||||||
from {
|
|
||||||
transform: rotate(0deg);
|
|
||||||
}
|
|
||||||
to {
|
|
||||||
transform: rotate(360deg);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
.decrypt-progress-overlay {
|
|
||||||
position: fixed;
|
|
||||||
top: 0;
|
|
||||||
left: 0;
|
|
||||||
right: 0;
|
|
||||||
bottom: 0;
|
|
||||||
background: rgba(0, 0, 0, 0.5);
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
z-index: 1000;
|
|
||||||
|
|
||||||
.progress-card {
|
|
||||||
background: var(--bg-primary);
|
|
||||||
border-radius: 16px;
|
|
||||||
padding: 32px 40px;
|
|
||||||
min-width: 400px;
|
|
||||||
text-align: center;
|
|
||||||
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
|
|
||||||
|
|
||||||
h3 {
|
|
||||||
margin: 0 0 8px;
|
|
||||||
font-size: 18px;
|
|
||||||
font-weight: 600;
|
|
||||||
color: var(--text-primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
.progress-file {
|
|
||||||
margin: 0 0 20px;
|
|
||||||
font-size: 14px;
|
|
||||||
color: var(--text-secondary);
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
}
|
|
||||||
|
|
||||||
.progress-bar {
|
|
||||||
height: 8px;
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
border-radius: 9999px;
|
|
||||||
overflow: hidden;
|
|
||||||
margin-bottom: 12px;
|
|
||||||
|
|
||||||
.progress-fill {
|
|
||||||
height: 100%;
|
|
||||||
background: var(--primary);
|
|
||||||
border-radius: 9999px;
|
|
||||||
transition: width 0.2s ease;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.progress-text {
|
|
||||||
margin: 0;
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// 图片列表样式
|
|
||||||
.current-dir {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 8px;
|
|
||||||
padding: 10px 14px;
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
border-radius: 8px;
|
|
||||||
margin-bottom: 16px;
|
|
||||||
font-size: 13px;
|
|
||||||
|
|
||||||
.dir-label {
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
.dir-path {
|
|
||||||
color: var(--text-primary);
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.image-list {
|
|
||||||
display: grid;
|
|
||||||
grid-template-columns: repeat(auto-fill, minmax(280px, 1fr));
|
|
||||||
gap: 8px;
|
|
||||||
max-height: 500px;
|
|
||||||
overflow-y: auto;
|
|
||||||
padding-right: 4px;
|
|
||||||
|
|
||||||
&::-webkit-scrollbar {
|
|
||||||
width: 6px;
|
|
||||||
}
|
|
||||||
|
|
||||||
&::-webkit-scrollbar-track {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
border-radius: 3px;
|
|
||||||
}
|
|
||||||
|
|
||||||
&::-webkit-scrollbar-thumb {
|
|
||||||
background: var(--border-color);
|
|
||||||
border-radius: 3px;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
background: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.image-item {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 10px;
|
|
||||||
padding: 10px 14px;
|
|
||||||
background: var(--bg-primary);
|
|
||||||
border-radius: 10px;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.clickable {
|
|
||||||
cursor: pointer;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
|
|
||||||
.decrypt-hint {
|
|
||||||
opacity: 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.status-icon {
|
|
||||||
width: 24px;
|
|
||||||
height: 24px;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
border-radius: 50%;
|
|
||||||
flex-shrink: 0;
|
|
||||||
|
|
||||||
&.decrypted {
|
|
||||||
background: var(--primary);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.pending {
|
|
||||||
background: var(--bg-tertiary);
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
|
|
||||||
.spin {
|
|
||||||
animation: spin 1s linear infinite;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.img-info {
|
|
||||||
flex: 1;
|
|
||||||
min-width: 0;
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: space-between;
|
|
||||||
gap: 8px;
|
|
||||||
|
|
||||||
.img-name {
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-primary);
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
}
|
|
||||||
|
|
||||||
.img-meta {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 8px;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
.version-tag {
|
|
||||||
font-size: 10px;
|
|
||||||
padding: 2px 6px;
|
|
||||||
border-radius: 4px;
|
|
||||||
font-weight: 500;
|
|
||||||
|
|
||||||
&.v3 {
|
|
||||||
background: rgba(59, 130, 246, 0.15);
|
|
||||||
color: #3b82f6;
|
|
||||||
}
|
|
||||||
|
|
||||||
&.v4 {
|
|
||||||
background: rgba(168, 85, 247, 0.15);
|
|
||||||
color: #a855f7;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.img-size {
|
|
||||||
font-size: 12px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.decrypt-hint {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
justify-content: center;
|
|
||||||
width: 20px;
|
|
||||||
height: 20px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
opacity: 0;
|
|
||||||
transition: opacity 0.2s;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.more-hint {
|
|
||||||
grid-column: 1 / -1;
|
|
||||||
text-align: center;
|
|
||||||
padding: 16px;
|
|
||||||
font-size: 13px;
|
|
||||||
color: var(--text-tertiary);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// 账号选择器
|
|
||||||
.account-selector {
|
|
||||||
display: flex;
|
|
||||||
gap: 8px;
|
|
||||||
margin-bottom: 12px;
|
|
||||||
flex-wrap: wrap;
|
|
||||||
|
|
||||||
.account-btn {
|
|
||||||
padding: 6px 14px;
|
|
||||||
border: 1px solid var(--border-color);
|
|
||||||
background: var(--bg-primary);
|
|
||||||
color: var(--text-secondary);
|
|
||||||
font-size: 13px;
|
|
||||||
border-radius: 9999px;
|
|
||||||
cursor: pointer;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:hover {
|
|
||||||
border-color: var(--primary);
|
|
||||||
color: var(--primary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.active {
|
|
||||||
background: var(--primary);
|
|
||||||
border-color: var(--primary);
|
|
||||||
color: white;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,62 +0,0 @@
|
|||||||
import { useEffect, useState } from 'react'
|
|
||||||
import * as configService from '../services/config'
|
|
||||||
import './DataManagementPage.scss'
|
|
||||||
|
|
||||||
function DataManagementPage() {
|
|
||||||
const [dbPath, setDbPath] = useState<string | null>(null)
|
|
||||||
const [wxid, setWxid] = useState<string | null>(null)
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
const loadConfig = async () => {
|
|
||||||
const [path, id] = await Promise.all([
|
|
||||||
configService.getDbPath(),
|
|
||||||
configService.getMyWxid()
|
|
||||||
])
|
|
||||||
setDbPath(path)
|
|
||||||
setWxid(id)
|
|
||||||
}
|
|
||||||
loadConfig()
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
return (
|
|
||||||
<>
|
|
||||||
<div className="page-header">
|
|
||||||
<h1>数据管理</h1>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="page-scroll">
|
|
||||||
<section className="page-section">
|
|
||||||
<div className="section-header">
|
|
||||||
<div>
|
|
||||||
<h2>WCDB 直连模式</h2>
|
|
||||||
<p className="section-desc">
|
|
||||||
当前版本通过 WCDB DLL 直接读取加密数据库,不再需要解密流程。
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="database-list">
|
|
||||||
<div className="database-item decrypted">
|
|
||||||
<div className="db-info">
|
|
||||||
<div className="db-name">
|
|
||||||
数据库目录
|
|
||||||
</div>
|
|
||||||
<div className="db-path">{dbPath || '未配置'}</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div className="database-item decrypted">
|
|
||||||
<div className="db-info">
|
|
||||||
<div className="db-name">
|
|
||||||
微信ID
|
|
||||||
</div>
|
|
||||||
<div className="db-path">{wxid || '未配置'}</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</section>
|
|
||||||
</div>
|
|
||||||
</>
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
export default DataManagementPage
|
|
||||||
171
src/pages/DualReportPage.scss
Normal file
171
src/pages/DualReportPage.scss
Normal file
@@ -0,0 +1,171 @@
|
|||||||
|
.dual-report-page {
|
||||||
|
padding: 32px 28px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-report-page.loading {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
min-height: 60vh;
|
||||||
|
gap: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.page-header {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
justify-content: space-between;
|
||||||
|
gap: 16px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 700;
|
||||||
|
}
|
||||||
|
|
||||||
|
p {
|
||||||
|
margin: 8px 0 0;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.year-badge {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 6px 10px;
|
||||||
|
border-radius: 999px;
|
||||||
|
background: color-mix(in srgb, var(--primary) 12%, transparent);
|
||||||
|
color: var(--primary);
|
||||||
|
border: 1px solid color-mix(in srgb, var(--primary) 30%, transparent);
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: 600;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.search-bar {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 10px 12px;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 12px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
|
||||||
|
input {
|
||||||
|
flex: 1;
|
||||||
|
border: none;
|
||||||
|
outline: none;
|
||||||
|
background: transparent;
|
||||||
|
color: var(--text-primary);
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.ranking-list {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.ranking-item {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: auto auto 1fr auto;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 12px 16px;
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 14px;
|
||||||
|
text-align: left;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
border-color: var(--primary);
|
||||||
|
box-shadow: 0 8px 20px rgba(0, 0, 0, 0.08);
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.rank-badge {
|
||||||
|
width: 28px;
|
||||||
|
height: 28px;
|
||||||
|
border-radius: 8px;
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: var(--border-color);
|
||||||
|
color: var(--text-secondary);
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: 700;
|
||||||
|
|
||||||
|
&.top {
|
||||||
|
background: color-mix(in srgb, var(--primary) 18%, transparent);
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.avatar {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 50%;
|
||||||
|
overflow: hidden;
|
||||||
|
background: var(--primary-light);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: var(--primary);
|
||||||
|
font-weight: 700;
|
||||||
|
|
||||||
|
img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.info {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 4px;
|
||||||
|
|
||||||
|
.name {
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
.sub {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.meta {
|
||||||
|
text-align: right;
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
|
||||||
|
.count {
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.empty {
|
||||||
|
text-align: center;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
padding: 40px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.spin {
|
||||||
|
animation: spin 1s linear infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes spin {
|
||||||
|
from { transform: rotate(0deg); }
|
||||||
|
to { transform: rotate(360deg); }
|
||||||
|
}
|
||||||
138
src/pages/DualReportPage.tsx
Normal file
138
src/pages/DualReportPage.tsx
Normal file
@@ -0,0 +1,138 @@
|
|||||||
|
import { useEffect, useMemo, useState } from 'react'
|
||||||
|
import { useNavigate } from 'react-router-dom'
|
||||||
|
import { Loader2, Search, Users } from 'lucide-react'
|
||||||
|
import './DualReportPage.scss'
|
||||||
|
|
||||||
|
interface ContactRanking {
|
||||||
|
username: string
|
||||||
|
displayName: string
|
||||||
|
avatarUrl?: string
|
||||||
|
messageCount: number
|
||||||
|
sentCount: number
|
||||||
|
receivedCount: number
|
||||||
|
lastMessageTime?: number | null
|
||||||
|
}
|
||||||
|
|
||||||
|
function DualReportPage() {
|
||||||
|
const navigate = useNavigate()
|
||||||
|
const [year, setYear] = useState<number>(0)
|
||||||
|
const [rankings, setRankings] = useState<ContactRanking[]>([])
|
||||||
|
const [isLoading, setIsLoading] = useState(true)
|
||||||
|
const [loadError, setLoadError] = useState<string | null>(null)
|
||||||
|
const [keyword, setKeyword] = useState('')
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const params = new URLSearchParams(window.location.hash.split('?')[1] || '')
|
||||||
|
const yearParam = params.get('year')
|
||||||
|
const parsedYear = yearParam ? parseInt(yearParam, 10) : 0
|
||||||
|
setYear(Number.isNaN(parsedYear) ? 0 : parsedYear)
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
loadRankings()
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const loadRankings = async () => {
|
||||||
|
setIsLoading(true)
|
||||||
|
setLoadError(null)
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.analytics.getContactRankings(200)
|
||||||
|
if (result.success && result.data) {
|
||||||
|
setRankings(result.data)
|
||||||
|
} else {
|
||||||
|
setLoadError(result.error || '加载好友列表失败')
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
setLoadError(String(e))
|
||||||
|
} finally {
|
||||||
|
setIsLoading(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const yearLabel = year === 0 ? '全部时间' : `${year}年`
|
||||||
|
|
||||||
|
const filteredRankings = useMemo(() => {
|
||||||
|
if (!keyword.trim()) return rankings
|
||||||
|
const q = keyword.trim().toLowerCase()
|
||||||
|
return rankings.filter((item) => {
|
||||||
|
return item.displayName.toLowerCase().includes(q) || item.username.toLowerCase().includes(q)
|
||||||
|
})
|
||||||
|
}, [rankings, keyword])
|
||||||
|
|
||||||
|
const handleSelect = (username: string) => {
|
||||||
|
const yearParam = year === 0 ? 0 : year
|
||||||
|
navigate(`/dual-report/view?username=${encodeURIComponent(username)}&year=${yearParam}`)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isLoading) {
|
||||||
|
return (
|
||||||
|
<div className="dual-report-page loading">
|
||||||
|
<Loader2 size={32} className="spin" />
|
||||||
|
<p>正在加载聊天排行...</p>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (loadError) {
|
||||||
|
return (
|
||||||
|
<div className="dual-report-page loading">
|
||||||
|
<p>加载失败:{loadError}</p>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="dual-report-page">
|
||||||
|
<div className="page-header">
|
||||||
|
<div>
|
||||||
|
<h1>双人年度报告</h1>
|
||||||
|
<p>选择一位好友,生成你们的专属聊天报告</p>
|
||||||
|
</div>
|
||||||
|
<div className="year-badge">
|
||||||
|
<Users size={14} />
|
||||||
|
<span>{yearLabel}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="search-bar">
|
||||||
|
<Search size={16} />
|
||||||
|
<input
|
||||||
|
value={keyword}
|
||||||
|
onChange={(e) => setKeyword(e.target.value)}
|
||||||
|
placeholder="搜索好友(昵称/备注/wxid)"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="ranking-list">
|
||||||
|
{filteredRankings.map((item, index) => (
|
||||||
|
<button
|
||||||
|
key={item.username}
|
||||||
|
className="ranking-item"
|
||||||
|
onClick={() => handleSelect(item.username)}
|
||||||
|
>
|
||||||
|
<span className={`rank-badge ${index < 3 ? 'top' : ''}`}>{index + 1}</span>
|
||||||
|
<div className="avatar">
|
||||||
|
{item.avatarUrl
|
||||||
|
? <img src={item.avatarUrl} alt={item.displayName} />
|
||||||
|
: <span>{item.displayName.slice(0, 1) || '?'}</span>
|
||||||
|
}
|
||||||
|
</div>
|
||||||
|
<div className="info">
|
||||||
|
<div className="name">{item.displayName}</div>
|
||||||
|
<div className="sub">{item.username}</div>
|
||||||
|
</div>
|
||||||
|
<div className="meta">
|
||||||
|
<div className="count">{item.messageCount.toLocaleString()} 条</div>
|
||||||
|
<div className="hint">总消息</div>
|
||||||
|
</div>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
{filteredRankings.length === 0 ? (
|
||||||
|
<div className="empty">没有匹配的好友</div>
|
||||||
|
) : null}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default DualReportPage
|
||||||
253
src/pages/DualReportWindow.scss
Normal file
253
src/pages/DualReportWindow.scss
Normal file
@@ -0,0 +1,253 @@
|
|||||||
|
.annual-report-window.dual-report-window {
|
||||||
|
.hero-title {
|
||||||
|
font-size: clamp(22px, 4vw, 34px);
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-cover-title {
|
||||||
|
font-size: clamp(26px, 5vw, 44px);
|
||||||
|
white-space: normal;
|
||||||
|
}
|
||||||
|
.dual-names {
|
||||||
|
font-size: clamp(24px, 4vw, 40px);
|
||||||
|
font-weight: 700;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
margin: 8px 0 16px;
|
||||||
|
color: var(--ar-text-main);
|
||||||
|
|
||||||
|
.amp {
|
||||||
|
color: var(--ar-primary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-info-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(2, minmax(0, 1fr));
|
||||||
|
gap: 16px;
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-info-card {
|
||||||
|
background: var(--ar-card-bg);
|
||||||
|
border: 1px solid var(--bg-tertiary, rgba(0, 0, 0, 0.05));
|
||||||
|
border-radius: 14px;
|
||||||
|
padding: 16px;
|
||||||
|
|
||||||
|
&.full {
|
||||||
|
grid-column: 1 / -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-label {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
margin-bottom: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.info-value {
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--ar-text-main);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-message-list {
|
||||||
|
margin-top: 16px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-message {
|
||||||
|
background: var(--ar-card-bg);
|
||||||
|
border-radius: 14px;
|
||||||
|
padding: 14px;
|
||||||
|
|
||||||
|
&.received {
|
||||||
|
background: var(--ar-card-bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-meta {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
margin-bottom: 6px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-content {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--ar-text-main);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.first-chat-scene {
|
||||||
|
background: linear-gradient(180deg, #8f5b85 0%, #e38aa0 50%, #f6d0c8 100%);
|
||||||
|
border-radius: 20px;
|
||||||
|
padding: 28px 24px 24px;
|
||||||
|
color: #fff;
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.first-chat-scene::before {
|
||||||
|
content: "";
|
||||||
|
position: absolute;
|
||||||
|
inset: 0;
|
||||||
|
background-image:
|
||||||
|
radial-gradient(circle at 20% 20%, rgba(255, 255, 255, 0.2), transparent 40%),
|
||||||
|
radial-gradient(circle at 80% 10%, rgba(255, 255, 255, 0.15), transparent 35%),
|
||||||
|
radial-gradient(circle at 50% 80%, rgba(255, 255, 255, 0.12), transparent 45%);
|
||||||
|
opacity: 0.6;
|
||||||
|
pointer-events: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-title {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 700;
|
||||||
|
text-align: center;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-subtitle {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 500;
|
||||||
|
text-align: center;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
opacity: 0.95;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-messages {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 14px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-message {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-end;
|
||||||
|
gap: 12px;
|
||||||
|
|
||||||
|
&.sent {
|
||||||
|
flex-direction: row-reverse;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-avatar {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 12px;
|
||||||
|
background: rgba(255, 255, 255, 0.25);
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
font-weight: 700;
|
||||||
|
color: #fff;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-bubble {
|
||||||
|
background: rgba(255, 255, 255, 0.85);
|
||||||
|
color: #5a4d5e;
|
||||||
|
padding: 10px 14px;
|
||||||
|
border-radius: 14px;
|
||||||
|
max-width: 60%;
|
||||||
|
box-shadow: 0 10px 24px rgba(0, 0, 0, 0.12);
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-message.sent .scene-bubble {
|
||||||
|
background: rgba(255, 224, 168, 0.9);
|
||||||
|
color: #4a3a2f;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-meta {
|
||||||
|
font-size: 11px;
|
||||||
|
opacity: 0.7;
|
||||||
|
margin-bottom: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-content {
|
||||||
|
font-size: 14px;
|
||||||
|
line-height: 1.4;
|
||||||
|
word-break: break-word;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scene-message.sent .scene-avatar {
|
||||||
|
background: rgba(255, 224, 168, 0.9);
|
||||||
|
color: #4a3a2f;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-stat-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(5, minmax(140px, 1fr));
|
||||||
|
gap: 14px;
|
||||||
|
margin: 20px -28px 24px;
|
||||||
|
padding: 0 28px;
|
||||||
|
overflow: visible;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-stat-card {
|
||||||
|
background: var(--ar-card-bg);
|
||||||
|
border-radius: 14px;
|
||||||
|
padding: 14px 12px;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-num {
|
||||||
|
font-size: clamp(20px, 2.8vw, 30px);
|
||||||
|
font-variant-numeric: tabular-nums;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-unit {
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dual-stat-card.long .stat-num {
|
||||||
|
font-size: clamp(18px, 2.4vw, 26px);
|
||||||
|
letter-spacing: -0.02em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.emoji-row {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(2, minmax(260px, 1fr));
|
||||||
|
gap: 20px;
|
||||||
|
margin: 0 -12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.emoji-card {
|
||||||
|
border: 1px solid var(--bg-tertiary, rgba(0, 0, 0, 0.08));
|
||||||
|
border-radius: 16px;
|
||||||
|
padding: 18px 16px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 10px;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: var(--ar-card-bg);
|
||||||
|
|
||||||
|
img {
|
||||||
|
width: 64px;
|
||||||
|
height: 64px;
|
||||||
|
object-fit: contain;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.emoji-title {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
}
|
||||||
|
|
||||||
|
.emoji-placeholder {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
word-break: break-all;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.word-cloud-empty {
|
||||||
|
color: var(--ar-text-sub);
|
||||||
|
font-size: 14px;
|
||||||
|
text-align: center;
|
||||||
|
padding: 24px 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
472
src/pages/DualReportWindow.tsx
Normal file
472
src/pages/DualReportWindow.tsx
Normal file
@@ -0,0 +1,472 @@
|
|||||||
|
import { useEffect, useState, type CSSProperties } from 'react'
|
||||||
|
import './AnnualReportWindow.scss'
|
||||||
|
import './DualReportWindow.scss'
|
||||||
|
|
||||||
|
interface DualReportMessage {
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
}
|
||||||
|
|
||||||
|
interface DualReportData {
|
||||||
|
year: number
|
||||||
|
selfName: string
|
||||||
|
friendUsername: string
|
||||||
|
friendName: string
|
||||||
|
firstChat: {
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
senderUsername?: string
|
||||||
|
} | null
|
||||||
|
firstChatMessages?: DualReportMessage[]
|
||||||
|
yearFirstChat?: {
|
||||||
|
createTime: number
|
||||||
|
createTimeStr: string
|
||||||
|
content: string
|
||||||
|
isSentByMe: boolean
|
||||||
|
friendName: string
|
||||||
|
firstThreeMessages: DualReportMessage[]
|
||||||
|
} | null
|
||||||
|
stats: {
|
||||||
|
totalMessages: number
|
||||||
|
totalWords: number
|
||||||
|
imageCount: number
|
||||||
|
voiceCount: number
|
||||||
|
emojiCount: number
|
||||||
|
myTopEmojiMd5?: string
|
||||||
|
friendTopEmojiMd5?: string
|
||||||
|
myTopEmojiUrl?: string
|
||||||
|
friendTopEmojiUrl?: string
|
||||||
|
}
|
||||||
|
topPhrases: Array<{ phrase: string; count: number }>
|
||||||
|
}
|
||||||
|
|
||||||
|
const WordCloud = ({ words }: { words: { phrase: string; count: number }[] }) => {
|
||||||
|
if (!words || words.length === 0) {
|
||||||
|
return <div className="word-cloud-empty">暂无高频语句</div>
|
||||||
|
}
|
||||||
|
const sortedWords = [...words].sort((a, b) => b.count - a.count)
|
||||||
|
const maxCount = sortedWords.length > 0 ? sortedWords[0].count : 1
|
||||||
|
const topWords = sortedWords.slice(0, 32)
|
||||||
|
const baseSize = 520
|
||||||
|
|
||||||
|
const seededRandom = (seed: number) => {
|
||||||
|
const x = Math.sin(seed) * 10000
|
||||||
|
return x - Math.floor(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
const placedItems: { x: number; y: number; w: number; h: number }[] = []
|
||||||
|
|
||||||
|
const canPlace = (x: number, y: number, w: number, h: number): boolean => {
|
||||||
|
const halfW = w / 2
|
||||||
|
const halfH = h / 2
|
||||||
|
const dx = x - 50
|
||||||
|
const dy = y - 50
|
||||||
|
const dist = Math.sqrt(dx * dx + dy * dy)
|
||||||
|
const maxR = 49 - Math.max(halfW, halfH)
|
||||||
|
if (dist > maxR) return false
|
||||||
|
|
||||||
|
const pad = 1.8
|
||||||
|
for (const p of placedItems) {
|
||||||
|
if ((x - halfW - pad) < (p.x + p.w / 2) &&
|
||||||
|
(x + halfW + pad) > (p.x - p.w / 2) &&
|
||||||
|
(y - halfH - pad) < (p.y + p.h / 2) &&
|
||||||
|
(y + halfH + pad) > (p.y - p.h / 2)) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
const wordItems = topWords.map((item, i) => {
|
||||||
|
const ratio = item.count / maxCount
|
||||||
|
const fontSize = Math.round(12 + Math.pow(ratio, 0.65) * 20)
|
||||||
|
const opacity = Math.min(1, Math.max(0.35, 0.35 + ratio * 0.65))
|
||||||
|
const delay = (i * 0.04).toFixed(2)
|
||||||
|
|
||||||
|
const charCount = Math.max(1, item.phrase.length)
|
||||||
|
const hasCjk = /[\u4e00-\u9fff]/.test(item.phrase)
|
||||||
|
const hasLatin = /[A-Za-z0-9]/.test(item.phrase)
|
||||||
|
const widthFactor = hasCjk && hasLatin ? 0.85 : hasCjk ? 0.98 : 0.6
|
||||||
|
const widthPx = fontSize * (charCount * widthFactor)
|
||||||
|
const heightPx = fontSize * 1.1
|
||||||
|
const widthPct = (widthPx / baseSize) * 100
|
||||||
|
const heightPct = (heightPx / baseSize) * 100
|
||||||
|
|
||||||
|
let x = 50, y = 50
|
||||||
|
let placedOk = false
|
||||||
|
const tries = i === 0 ? 1 : 420
|
||||||
|
|
||||||
|
for (let t = 0; t < tries; t++) {
|
||||||
|
if (i === 0) {
|
||||||
|
x = 50
|
||||||
|
y = 50
|
||||||
|
} else {
|
||||||
|
const idx = i + t * 0.28
|
||||||
|
const radius = Math.sqrt(idx) * 7.6 + (seededRandom(i * 1000 + t) * 1.2 - 0.6)
|
||||||
|
const angle = idx * 2.399963 + seededRandom(i * 2000 + t) * 0.35
|
||||||
|
x = 50 + radius * Math.cos(angle)
|
||||||
|
y = 50 + radius * Math.sin(angle)
|
||||||
|
}
|
||||||
|
if (canPlace(x, y, widthPct, heightPct)) {
|
||||||
|
placedOk = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!placedOk) return null
|
||||||
|
placedItems.push({ x, y, w: widthPct, h: heightPct })
|
||||||
|
|
||||||
|
return (
|
||||||
|
<span
|
||||||
|
key={i}
|
||||||
|
className="word-tag"
|
||||||
|
style={{
|
||||||
|
'--final-opacity': opacity,
|
||||||
|
left: `${x.toFixed(2)}%`,
|
||||||
|
top: `${y.toFixed(2)}%`,
|
||||||
|
fontSize: `${fontSize}px`,
|
||||||
|
animationDelay: `${delay}s`,
|
||||||
|
} as CSSProperties}
|
||||||
|
title={`${item.phrase} (出现 ${item.count} 次)`}
|
||||||
|
>
|
||||||
|
{item.phrase}
|
||||||
|
</span>
|
||||||
|
)
|
||||||
|
}).filter(Boolean)
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="word-cloud-wrapper">
|
||||||
|
<div className="word-cloud-inner">
|
||||||
|
{wordItems}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
function DualReportWindow() {
|
||||||
|
const [reportData, setReportData] = useState<DualReportData | null>(null)
|
||||||
|
const [isLoading, setIsLoading] = useState(true)
|
||||||
|
const [error, setError] = useState<string | null>(null)
|
||||||
|
const [loadingStage, setLoadingStage] = useState('准备中')
|
||||||
|
const [loadingProgress, setLoadingProgress] = useState(0)
|
||||||
|
const [myEmojiUrl, setMyEmojiUrl] = useState<string | null>(null)
|
||||||
|
const [friendEmojiUrl, setFriendEmojiUrl] = useState<string | null>(null)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const params = new URLSearchParams(window.location.hash.split('?')[1] || '')
|
||||||
|
const username = params.get('username')
|
||||||
|
const yearParam = params.get('year')
|
||||||
|
const parsedYear = yearParam ? parseInt(yearParam, 10) : 0
|
||||||
|
const year = Number.isNaN(parsedYear) ? 0 : parsedYear
|
||||||
|
if (!username) {
|
||||||
|
setError('缺少好友信息')
|
||||||
|
setIsLoading(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
generateReport(username, year)
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
const generateReport = async (friendUsername: string, year: number) => {
|
||||||
|
setIsLoading(true)
|
||||||
|
setError(null)
|
||||||
|
setLoadingProgress(0)
|
||||||
|
|
||||||
|
const removeProgressListener = window.electronAPI.dualReport.onProgress?.((payload: { status: string; progress: number }) => {
|
||||||
|
setLoadingProgress(payload.progress)
|
||||||
|
setLoadingStage(payload.status)
|
||||||
|
})
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await window.electronAPI.dualReport.generateReport({ friendUsername, year })
|
||||||
|
removeProgressListener?.()
|
||||||
|
setLoadingProgress(100)
|
||||||
|
setLoadingStage('完成')
|
||||||
|
|
||||||
|
if (result.success && result.data) {
|
||||||
|
setReportData(result.data)
|
||||||
|
setIsLoading(false)
|
||||||
|
} else {
|
||||||
|
setError(result.error || '生成报告失败')
|
||||||
|
setIsLoading(false)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
removeProgressListener?.()
|
||||||
|
setError(String(e))
|
||||||
|
setIsLoading(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const loadEmojis = async () => {
|
||||||
|
if (!reportData) return
|
||||||
|
const stats = reportData.stats
|
||||||
|
if (stats.myTopEmojiUrl) {
|
||||||
|
const res = await window.electronAPI.chat.downloadEmoji(stats.myTopEmojiUrl, stats.myTopEmojiMd5)
|
||||||
|
if (res.success && res.localPath) {
|
||||||
|
setMyEmojiUrl(res.localPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (stats.friendTopEmojiUrl) {
|
||||||
|
const res = await window.electronAPI.chat.downloadEmoji(stats.friendTopEmojiUrl, stats.friendTopEmojiMd5)
|
||||||
|
if (res.success && res.localPath) {
|
||||||
|
setFriendEmojiUrl(res.localPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void loadEmojis()
|
||||||
|
}, [reportData])
|
||||||
|
|
||||||
|
if (isLoading) {
|
||||||
|
return (
|
||||||
|
<div className="annual-report-window loading">
|
||||||
|
<div className="loading-ring">
|
||||||
|
<svg viewBox="0 0 100 100">
|
||||||
|
<circle className="ring-bg" cx="50" cy="50" r="42" />
|
||||||
|
<circle
|
||||||
|
className="ring-progress"
|
||||||
|
cx="50" cy="50" r="42"
|
||||||
|
style={{ strokeDashoffset: 264 - (264 * loadingProgress / 100) }}
|
||||||
|
/>
|
||||||
|
</svg>
|
||||||
|
<span className="ring-text">{loadingProgress}%</span>
|
||||||
|
</div>
|
||||||
|
<p className="loading-stage">{loadingStage}</p>
|
||||||
|
<p className="loading-hint">进行中</p>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (error) {
|
||||||
|
return (
|
||||||
|
<div className="annual-report-window error">
|
||||||
|
<p>生成报告失败: {error}</p>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!reportData) {
|
||||||
|
return (
|
||||||
|
<div className="annual-report-window error">
|
||||||
|
<p>暂无数据</p>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
const yearTitle = reportData.year === 0 ? '全部时间' : `${reportData.year}年`
|
||||||
|
const firstChat = reportData.firstChat
|
||||||
|
const firstChatMessages = (reportData.firstChatMessages && reportData.firstChatMessages.length > 0)
|
||||||
|
? reportData.firstChatMessages.slice(0, 3)
|
||||||
|
: firstChat
|
||||||
|
? [{
|
||||||
|
content: firstChat.content,
|
||||||
|
isSentByMe: firstChat.isSentByMe,
|
||||||
|
createTime: firstChat.createTime,
|
||||||
|
createTimeStr: firstChat.createTimeStr
|
||||||
|
}]
|
||||||
|
: []
|
||||||
|
const daysSince = firstChat
|
||||||
|
? Math.max(0, Math.floor((Date.now() - firstChat.createTime) / 86400000))
|
||||||
|
: null
|
||||||
|
const yearFirstChat = reportData.yearFirstChat
|
||||||
|
const stats = reportData.stats
|
||||||
|
const statItems = [
|
||||||
|
{ label: '总消息数', value: stats.totalMessages },
|
||||||
|
{ label: '总字数', value: stats.totalWords },
|
||||||
|
{ label: '图片', value: stats.imageCount },
|
||||||
|
{ label: '语音', value: stats.voiceCount },
|
||||||
|
{ label: '表情', value: stats.emojiCount },
|
||||||
|
]
|
||||||
|
|
||||||
|
const decodeEntities = (text: string) => (
|
||||||
|
text
|
||||||
|
.replace(/&/g, '&')
|
||||||
|
.replace(/</g, '<')
|
||||||
|
.replace(/>/g, '>')
|
||||||
|
.replace(/"/g, '"')
|
||||||
|
.replace(/'/g, "'")
|
||||||
|
)
|
||||||
|
|
||||||
|
const stripCdata = (text: string) => text.replace(/<!\[CDATA\[([\s\S]*?)\]\]>/g, '$1')
|
||||||
|
|
||||||
|
const extractXmlText = (content: string) => {
|
||||||
|
const titleMatch = content.match(/<title>([\s\S]*?)<\/title>/i)
|
||||||
|
if (titleMatch?.[1]) return titleMatch[1]
|
||||||
|
const descMatch = content.match(/<des>([\s\S]*?)<\/des>/i)
|
||||||
|
if (descMatch?.[1]) return descMatch[1]
|
||||||
|
const summaryMatch = content.match(/<summary>([\s\S]*?)<\/summary>/i)
|
||||||
|
if (summaryMatch?.[1]) return summaryMatch[1]
|
||||||
|
const contentMatch = content.match(/<content>([\s\S]*?)<\/content>/i)
|
||||||
|
if (contentMatch?.[1]) return contentMatch[1]
|
||||||
|
return ''
|
||||||
|
}
|
||||||
|
|
||||||
|
const formatMessageContent = (content?: string) => {
|
||||||
|
const raw = String(content || '').trim()
|
||||||
|
if (!raw) return '(空)'
|
||||||
|
const hasXmlTag = /<\s*[a-zA-Z]+[^>]*>/.test(raw)
|
||||||
|
const looksLikeXml = /<\?xml|<msg\b|<appmsg\b|<sysmsg\b|<appattach\b|<emoji\b|<img\b|<voip\b/i.test(raw)
|
||||||
|
|| hasXmlTag
|
||||||
|
if (!looksLikeXml) return raw
|
||||||
|
const extracted = extractXmlText(raw)
|
||||||
|
if (!extracted) return '(XML消息)'
|
||||||
|
return decodeEntities(stripCdata(extracted).trim()) || '(XML消息)'
|
||||||
|
}
|
||||||
|
const formatFullDate = (timestamp: number) => {
|
||||||
|
const d = new Date(timestamp)
|
||||||
|
const year = d.getFullYear()
|
||||||
|
const month = String(d.getMonth() + 1).padStart(2, '0')
|
||||||
|
const day = String(d.getDate()).padStart(2, '0')
|
||||||
|
const hour = String(d.getHours()).padStart(2, '0')
|
||||||
|
const minute = String(d.getMinutes()).padStart(2, '0')
|
||||||
|
return `${year}/${month}/${day} ${hour}:${minute}`
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="annual-report-window dual-report-window">
|
||||||
|
<div className="drag-region" />
|
||||||
|
|
||||||
|
<div className="bg-decoration">
|
||||||
|
<div className="deco-circle c1" />
|
||||||
|
<div className="deco-circle c2" />
|
||||||
|
<div className="deco-circle c3" />
|
||||||
|
<div className="deco-circle c4" />
|
||||||
|
<div className="deco-circle c5" />
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="report-scroll-view">
|
||||||
|
<div className="report-container">
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">WEFLOW · DUAL REPORT</div>
|
||||||
|
<h1 className="hero-title dual-cover-title">{yearTitle}<br />双人聊天报告</h1>
|
||||||
|
<hr className="divider" />
|
||||||
|
<div className="dual-names">
|
||||||
|
<span>{reportData.selfName}</span>
|
||||||
|
<span className="amp">&</span>
|
||||||
|
<span>{reportData.friendName}</span>
|
||||||
|
</div>
|
||||||
|
<p className="hero-desc">每一次对话都值得被珍藏</p>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">首次聊天</div>
|
||||||
|
<h2 className="hero-title">故事的开始</h2>
|
||||||
|
{firstChat ? (
|
||||||
|
<>
|
||||||
|
<div className="dual-info-grid">
|
||||||
|
<div className="dual-info-card">
|
||||||
|
<div className="info-label">第一次聊天时间</div>
|
||||||
|
<div className="info-value">{formatFullDate(firstChat.createTime)}</div>
|
||||||
|
</div>
|
||||||
|
<div className="dual-info-card">
|
||||||
|
<div className="info-label">距今天数</div>
|
||||||
|
<div className="info-value">{daysSince} 天</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{firstChatMessages.length > 0 ? (
|
||||||
|
<div className="dual-message-list">
|
||||||
|
{firstChatMessages.map((msg, idx) => (
|
||||||
|
<div
|
||||||
|
key={idx}
|
||||||
|
className={`dual-message ${msg.isSentByMe ? 'sent' : 'received'}`}
|
||||||
|
>
|
||||||
|
<div className="message-meta">
|
||||||
|
{msg.isSentByMe ? reportData.selfName : reportData.friendName} · {formatFullDate(msg.createTime)}
|
||||||
|
</div>
|
||||||
|
<div className="message-content">{formatMessageContent(msg.content)}</div>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<p className="hero-desc">暂无首条消息</p>
|
||||||
|
)}
|
||||||
|
</section>
|
||||||
|
|
||||||
|
{yearFirstChat ? (
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">第一段对话</div>
|
||||||
|
<h2 className="hero-title">
|
||||||
|
{reportData.year === 0 ? '你们的第一段对话' : `${reportData.year}年的第一段对话`}
|
||||||
|
</h2>
|
||||||
|
<div className="dual-info-grid">
|
||||||
|
<div className="dual-info-card">
|
||||||
|
<div className="info-label">第一段对话时间</div>
|
||||||
|
<div className="info-value">{formatFullDate(yearFirstChat.createTime)}</div>
|
||||||
|
</div>
|
||||||
|
<div className="dual-info-card">
|
||||||
|
<div className="info-label">发起者</div>
|
||||||
|
<div className="info-value">{yearFirstChat.isSentByMe ? reportData.selfName : reportData.friendName}</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="dual-message-list">
|
||||||
|
{yearFirstChat.firstThreeMessages.map((msg, idx) => (
|
||||||
|
<div key={idx} className={`dual-message ${msg.isSentByMe ? 'sent' : 'received'}`}>
|
||||||
|
<div className="message-meta">
|
||||||
|
{msg.isSentByMe ? reportData.selfName : reportData.friendName} · {formatFullDate(msg.createTime)}
|
||||||
|
</div>
|
||||||
|
<div className="message-content">{formatMessageContent(msg.content)}</div>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
) : null}
|
||||||
|
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">常用语</div>
|
||||||
|
<h2 className="hero-title">{yearTitle}常用语</h2>
|
||||||
|
<WordCloud words={reportData.topPhrases} />
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">年度统计</div>
|
||||||
|
<h2 className="hero-title">{yearTitle}数据概览</h2>
|
||||||
|
<div className="dual-stat-grid">
|
||||||
|
{statItems.map((item) => {
|
||||||
|
const valueText = item.value.toLocaleString()
|
||||||
|
const isLong = valueText.length > 7
|
||||||
|
return (
|
||||||
|
<div key={item.label} className={`dual-stat-card ${isLong ? 'long' : ''}`}>
|
||||||
|
<div className="stat-num">{valueText}</div>
|
||||||
|
<div className="stat-unit">{item.label}</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="emoji-row">
|
||||||
|
<div className="emoji-card">
|
||||||
|
<div className="emoji-title">我常用的表情</div>
|
||||||
|
{myEmojiUrl ? (
|
||||||
|
<img src={myEmojiUrl} alt="my-emoji" />
|
||||||
|
) : (
|
||||||
|
<div className="emoji-placeholder">{stats.myTopEmojiMd5 || '暂无'}</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="emoji-card">
|
||||||
|
<div className="emoji-title">{reportData.friendName}常用的表情</div>
|
||||||
|
{friendEmojiUrl ? (
|
||||||
|
<img src={friendEmojiUrl} alt="friend-emoji" />
|
||||||
|
) : (
|
||||||
|
<div className="emoji-placeholder">{stats.friendTopEmojiMd5 || '暂无'}</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section className="section">
|
||||||
|
<div className="label-text">尾声</div>
|
||||||
|
<h2 className="hero-title">谢谢你一直在</h2>
|
||||||
|
<p className="hero-desc">愿我们继续把故事写下去</p>
|
||||||
|
</section>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default DualReportWindow
|
||||||
@@ -338,64 +338,129 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.time-options {
|
.time-range-picker-item {
|
||||||
display: flex;
|
|
||||||
flex-direction: column;
|
|
||||||
gap: 12px;
|
|
||||||
}
|
|
||||||
|
|
||||||
.checkbox-item {
|
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
gap: 10px;
|
justify-content: space-between;
|
||||||
|
padding: 14px 16px;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
font-size: 14px;
|
transition: background 0.2s;
|
||||||
color: var(--text-primary);
|
background: transparent;
|
||||||
|
|
||||||
input[type="checkbox"] {
|
|
||||||
width: 18px;
|
|
||||||
height: 18px;
|
|
||||||
accent-color: var(--primary);
|
|
||||||
cursor: pointer;
|
|
||||||
}
|
|
||||||
|
|
||||||
svg {
|
|
||||||
color: var(--text-secondary);
|
|
||||||
}
|
|
||||||
|
|
||||||
&.main-toggle {
|
|
||||||
padding: 12px 16px;
|
|
||||||
background: var(--bg-secondary);
|
|
||||||
border-radius: 10px;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
.date-range {
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
gap: 10px;
|
|
||||||
padding: 12px 16px;
|
|
||||||
background: var(--bg-secondary);
|
|
||||||
border-radius: 10px;
|
|
||||||
font-size: 14px;
|
|
||||||
color: var(--text-primary);
|
|
||||||
cursor: pointer;
|
|
||||||
transition: all 0.2s;
|
|
||||||
|
|
||||||
&:hover {
|
&:hover {
|
||||||
background: var(--bg-hover);
|
background: var(--bg-hover);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.time-picker-info {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
|
||||||
svg {
|
svg {
|
||||||
color: var(--text-tertiary);
|
color: var(--primary);
|
||||||
flex-shrink: 0;
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
span {
|
svg {
|
||||||
flex: 1;
|
color: var(--text-tertiary);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.select-field {
|
||||||
|
position: relative;
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-trigger {
|
||||||
|
width: 100%;
|
||||||
|
padding: 10px 16px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 9999px;
|
||||||
|
font-size: 14px;
|
||||||
|
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: 20;
|
||||||
|
max-height: 260px;
|
||||||
|
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: 14px;
|
||||||
|
|
||||||
|
&: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-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.select-option.active .option-desc {
|
||||||
|
color: var(--primary);
|
||||||
|
}
|
||||||
|
|
||||||
.media-options {
|
.media-options {
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-wrap: wrap;
|
flex-wrap: wrap;
|
||||||
@@ -602,6 +667,87 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.export-layout-modal {
|
||||||
|
background: var(--card-bg);
|
||||||
|
padding: 28px 32px;
|
||||||
|
border-radius: 16px;
|
||||||
|
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.25);
|
||||||
|
text-align: center;
|
||||||
|
width: min(520px, 90vw);
|
||||||
|
|
||||||
|
h3 {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
margin: 0 0 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-subtitle {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
margin: 0 0 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-options {
|
||||||
|
display: grid;
|
||||||
|
gap: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-option-btn {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 14px 18px;
|
||||||
|
border-radius: 12px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
text-align: left;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
border-color: var(--primary);
|
||||||
|
background: rgba(var(--primary-rgb), 0.08);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.primary {
|
||||||
|
border-color: var(--primary);
|
||||||
|
background: rgba(var(--primary-rgb), 0.12);
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-title {
|
||||||
|
font-size: 15px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-desc {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-actions {
|
||||||
|
margin-top: 18px;
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.layout-cancel-btn {
|
||||||
|
padding: 8px 20px;
|
||||||
|
border-radius: 8px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
background: var(--bg-secondary);
|
||||||
|
color: var(--text-primary);
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
.export-result-modal {
|
.export-result-modal {
|
||||||
background: var(--card-bg);
|
background: var(--card-bg);
|
||||||
padding: 32px 40px;
|
padding: 32px 40px;
|
||||||
@@ -920,3 +1066,94 @@
|
|||||||
transform: rotate(360deg);
|
transform: rotate(360deg);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 媒体导出选项卡片样式
|
||||||
|
.setting-subtitle {
|
||||||
|
font-size: 12px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
margin: 4px 0 12px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-options-card {
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 12px;
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-switch-row {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 14px 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-switch-info {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-switch-title {
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-switch-desc {
|
||||||
|
font-size: 11px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-option-divider {
|
||||||
|
height: 1px;
|
||||||
|
background: var(--border-color);
|
||||||
|
margin-left: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-checkbox-row {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: space-between;
|
||||||
|
padding: 12px 16px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: background 0.2s;
|
||||||
|
|
||||||
|
&:hover:not(.disabled) {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
|
||||||
|
input[type="checkbox"] {
|
||||||
|
width: 18px;
|
||||||
|
height: 18px;
|
||||||
|
accent-color: var(--primary);
|
||||||
|
cursor: pointer;
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-checkbox-info {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-checkbox-title {
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
.media-checkbox-desc {
|
||||||
|
font-size: 11px;
|
||||||
|
color: var(--text-tertiary);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 全局样式已在 main.scss 中定义
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { useState, useEffect, useCallback } from 'react'
|
import { useState, useEffect, useCallback, useRef } from 'react'
|
||||||
import { Search, Download, FolderOpen, RefreshCw, Check, Calendar, FileJson, FileText, Table, Loader2, X, ChevronDown, ChevronLeft, ChevronRight, FileSpreadsheet, Database, FileCode, CheckCircle, XCircle, ExternalLink } from 'lucide-react'
|
import { Search, Download, FolderOpen, RefreshCw, Check, Calendar, FileJson, FileText, Table, Loader2, X, ChevronDown, ChevronLeft, ChevronRight, FileSpreadsheet, Database, FileCode, CheckCircle, XCircle, ExternalLink } from 'lucide-react'
|
||||||
import * as configService from '../services/config'
|
import * as configService from '../services/config'
|
||||||
import './ExportPage.scss'
|
import './ExportPage.scss'
|
||||||
@@ -16,6 +16,16 @@ interface ExportOptions {
|
|||||||
dateRange: { start: Date; end: Date } | null
|
dateRange: { start: Date; end: Date } | null
|
||||||
useAllTime: boolean
|
useAllTime: boolean
|
||||||
exportAvatars: boolean
|
exportAvatars: boolean
|
||||||
|
exportMedia: boolean
|
||||||
|
exportImages: boolean
|
||||||
|
exportVoices: boolean
|
||||||
|
exportVideos: boolean
|
||||||
|
exportEmojis: boolean
|
||||||
|
exportVoiceAsText: boolean
|
||||||
|
excelCompactColumns: boolean
|
||||||
|
txtColumns: string[]
|
||||||
|
displayNamePreference: 'group-nickname' | 'remark' | 'nickname'
|
||||||
|
exportConcurrency: number
|
||||||
}
|
}
|
||||||
|
|
||||||
interface ExportResult {
|
interface ExportResult {
|
||||||
@@ -25,7 +35,10 @@ interface ExportResult {
|
|||||||
error?: string
|
error?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type SessionLayout = 'shared' | 'per-session'
|
||||||
|
|
||||||
function ExportPage() {
|
function ExportPage() {
|
||||||
|
const defaultTxtColumns = ['index', 'time', 'senderRole', 'messageType', 'content']
|
||||||
const [sessions, setSessions] = useState<ChatSession[]>([])
|
const [sessions, setSessions] = useState<ChatSession[]>([])
|
||||||
const [filteredSessions, setFilteredSessions] = useState<ChatSession[]>([])
|
const [filteredSessions, setFilteredSessions] = useState<ChatSession[]>([])
|
||||||
const [selectedSessions, setSelectedSessions] = useState<Set<string>>(new Set())
|
const [selectedSessions, setSelectedSessions] = useState<Set<string>>(new Set())
|
||||||
@@ -38,17 +51,49 @@ 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 [showMediaLayoutPrompt, setShowMediaLayoutPrompt] = useState(false)
|
||||||
|
const [showDisplayNameSelect, setShowDisplayNameSelect] = useState(false)
|
||||||
|
const displayNameDropdownRef = useRef<HTMLDivElement>(null)
|
||||||
|
|
||||||
const [options, setOptions] = useState<ExportOptions>({
|
const [options, setOptions] = useState<ExportOptions>({
|
||||||
format: 'chatlab',
|
format: 'excel',
|
||||||
dateRange: {
|
dateRange: {
|
||||||
start: new Date(Date.now() - 7 * 24 * 60 * 60 * 1000),
|
start: new Date(new Date().setHours(0, 0, 0, 0)),
|
||||||
end: new Date()
|
end: new Date()
|
||||||
},
|
},
|
||||||
useAllTime: true,
|
useAllTime: false,
|
||||||
exportAvatars: true
|
exportAvatars: true,
|
||||||
|
exportMedia: false,
|
||||||
|
exportImages: true,
|
||||||
|
exportVoices: true,
|
||||||
|
exportVideos: true,
|
||||||
|
exportEmojis: true,
|
||||||
|
exportVoiceAsText: true,
|
||||||
|
excelCompactColumns: true,
|
||||||
|
txtColumns: defaultTxtColumns,
|
||||||
|
displayNamePreference: 'remark',
|
||||||
|
exportConcurrency: 2
|
||||||
})
|
})
|
||||||
|
|
||||||
|
const buildDateRangeFromPreset = (preset: string) => {
|
||||||
|
const now = new Date()
|
||||||
|
if (preset === 'all') {
|
||||||
|
return { useAllTime: true, dateRange: { start: now, end: now } }
|
||||||
|
}
|
||||||
|
let rangeMs = 0
|
||||||
|
if (preset === '7d') rangeMs = 7 * 24 * 60 * 60 * 1000
|
||||||
|
if (preset === '30d') rangeMs = 30 * 24 * 60 * 60 * 1000
|
||||||
|
if (preset === '90d') rangeMs = 90 * 24 * 60 * 60 * 1000
|
||||||
|
if (preset === 'today' || rangeMs === 0) {
|
||||||
|
const start = new Date(now)
|
||||||
|
start.setHours(0, 0, 0, 0)
|
||||||
|
return { useAllTime: false, dateRange: { start, end: now } }
|
||||||
|
}
|
||||||
|
const start = new Date(now.getTime() - rangeMs)
|
||||||
|
start.setHours(0, 0, 0, 0)
|
||||||
|
return { useAllTime: false, dateRange: { start, end: now } }
|
||||||
|
}
|
||||||
|
|
||||||
const loadSessions = useCallback(async () => {
|
const loadSessions = useCallback(async () => {
|
||||||
setIsLoading(true)
|
setIsLoading(true)
|
||||||
try {
|
try {
|
||||||
@@ -84,10 +129,87 @@ function ExportPage() {
|
|||||||
}
|
}
|
||||||
}, [])
|
}, [])
|
||||||
|
|
||||||
|
const loadExportDefaults = useCallback(async () => {
|
||||||
|
try {
|
||||||
|
const [
|
||||||
|
savedFormat,
|
||||||
|
savedRange,
|
||||||
|
savedMedia,
|
||||||
|
savedVoiceAsText,
|
||||||
|
savedExcelCompactColumns,
|
||||||
|
savedTxtColumns,
|
||||||
|
savedConcurrency
|
||||||
|
] = await Promise.all([
|
||||||
|
configService.getExportDefaultFormat(),
|
||||||
|
configService.getExportDefaultDateRange(),
|
||||||
|
configService.getExportDefaultMedia(),
|
||||||
|
configService.getExportDefaultVoiceAsText(),
|
||||||
|
configService.getExportDefaultExcelCompactColumns(),
|
||||||
|
configService.getExportDefaultTxtColumns(),
|
||||||
|
configService.getExportDefaultConcurrency()
|
||||||
|
])
|
||||||
|
|
||||||
|
const preset = savedRange || 'today'
|
||||||
|
const rangeDefaults = buildDateRangeFromPreset(preset)
|
||||||
|
const txtColumns = savedTxtColumns && savedTxtColumns.length > 0 ? savedTxtColumns : defaultTxtColumns
|
||||||
|
|
||||||
|
setOptions((prev) => ({
|
||||||
|
...prev,
|
||||||
|
format: (savedFormat as ExportOptions['format']) || 'excel',
|
||||||
|
useAllTime: rangeDefaults.useAllTime,
|
||||||
|
dateRange: rangeDefaults.dateRange,
|
||||||
|
exportMedia: savedMedia ?? false,
|
||||||
|
exportVoiceAsText: savedVoiceAsText ?? true,
|
||||||
|
excelCompactColumns: savedExcelCompactColumns ?? true,
|
||||||
|
txtColumns,
|
||||||
|
exportConcurrency: savedConcurrency ?? 2
|
||||||
|
}))
|
||||||
|
} catch (e) {
|
||||||
|
console.error('加载导出默认设置失败:', e)
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
loadSessions()
|
loadSessions()
|
||||||
loadExportPath()
|
loadExportPath()
|
||||||
}, [loadSessions, loadExportPath])
|
loadExportDefaults()
|
||||||
|
}, [loadSessions, loadExportPath, loadExportDefaults])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const handleChange = () => {
|
||||||
|
setSelectedSessions(new Set())
|
||||||
|
setSearchKeyword('')
|
||||||
|
setExportResult(null)
|
||||||
|
setSessions([])
|
||||||
|
setFilteredSessions([])
|
||||||
|
loadSessions()
|
||||||
|
}
|
||||||
|
window.addEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
return () => window.removeEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
}, [loadSessions])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const removeListener = window.electronAPI.export.onProgress?.((payload: { current: number; total: number; currentSession: string; phase: string }) => {
|
||||||
|
setExportProgress({
|
||||||
|
current: payload.current,
|
||||||
|
total: payload.total,
|
||||||
|
currentName: payload.currentSession
|
||||||
|
})
|
||||||
|
})
|
||||||
|
return () => {
|
||||||
|
removeListener?.()
|
||||||
|
}
|
||||||
|
}, [])
|
||||||
|
useEffect(() => {
|
||||||
|
const handleClickOutside = (event: MouseEvent) => {
|
||||||
|
const target = event.target as Node
|
||||||
|
if (showDisplayNameSelect && displayNameDropdownRef.current && !displayNameDropdownRef.current.contains(target)) {
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
document.addEventListener('mousedown', handleClickOutside)
|
||||||
|
return () => document.removeEventListener('mousedown', handleClickOutside)
|
||||||
|
}, [showDisplayNameSelect])
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!searchKeyword.trim()) {
|
if (!searchKeyword.trim()) {
|
||||||
@@ -128,13 +250,31 @@ function ExportPage() {
|
|||||||
return date.toLocaleDateString('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
|
return date.toLocaleDateString('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const handleFormatChange = (format: ExportOptions['format']) => {
|
||||||
|
setOptions((prev) => {
|
||||||
|
const next = { ...prev, format }
|
||||||
|
if (format === 'html') {
|
||||||
|
return {
|
||||||
|
...next,
|
||||||
|
exportMedia: true,
|
||||||
|
exportImages: true,
|
||||||
|
exportVoices: true,
|
||||||
|
exportVideos: true,
|
||||||
|
exportEmojis: true,
|
||||||
|
exportVoiceAsText: true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return next
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
const openExportFolder = async () => {
|
const openExportFolder = async () => {
|
||||||
if (exportFolder) {
|
if (exportFolder) {
|
||||||
await window.electronAPI.shell.openPath(exportFolder)
|
await window.electronAPI.shell.openPath(exportFolder)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const startExport = async () => {
|
const runExport = async (sessionLayout: SessionLayout) => {
|
||||||
if (selectedSessions.size === 0 || !exportFolder) return
|
if (selectedSessions.size === 0 || !exportFolder) return
|
||||||
|
|
||||||
setIsExporting(true)
|
setIsExporting(true)
|
||||||
@@ -146,14 +286,25 @@ function ExportPage() {
|
|||||||
const exportOptions = {
|
const exportOptions = {
|
||||||
format: options.format,
|
format: options.format,
|
||||||
exportAvatars: options.exportAvatars,
|
exportAvatars: options.exportAvatars,
|
||||||
|
exportMedia: options.exportMedia,
|
||||||
|
exportImages: options.exportMedia && options.exportImages,
|
||||||
|
exportVoices: options.exportMedia && options.exportVoices,
|
||||||
|
exportVideos: options.exportMedia && options.exportVideos,
|
||||||
|
exportEmojis: options.exportMedia && options.exportEmojis,
|
||||||
|
exportVoiceAsText: options.exportVoiceAsText, // 即使不导出媒体,也可以导出语音转文字内容
|
||||||
|
excelCompactColumns: options.excelCompactColumns,
|
||||||
|
txtColumns: options.txtColumns,
|
||||||
|
displayNamePreference: options.displayNamePreference,
|
||||||
|
exportConcurrency: options.exportConcurrency,
|
||||||
|
sessionLayout,
|
||||||
dateRange: options.useAllTime ? null : options.dateRange ? {
|
dateRange: options.useAllTime ? null : options.dateRange ? {
|
||||||
start: Math.floor(options.dateRange.start.getTime() / 1000),
|
start: Math.floor(options.dateRange.start.getTime() / 1000),
|
||||||
// 将结束日期设置为当天的 23:59:59,以包含当天的所有消息
|
// 将结束日期设置为当天的 23:59:59,确保包含当天的所有记录
|
||||||
end: Math.floor(new Date(options.dateRange.end.getFullYear(), options.dateRange.end.getMonth(), options.dateRange.end.getDate(), 23, 59, 59).getTime() / 1000)
|
end: Math.floor(new Date(options.dateRange.end.getFullYear(), options.dateRange.end.getMonth(), options.dateRange.end.getDate(), 23, 59, 59).getTime() / 1000)
|
||||||
} : null
|
} : null
|
||||||
}
|
}
|
||||||
|
|
||||||
if (options.format === 'chatlab' || options.format === 'chatlab-jsonl' || options.format === 'json') {
|
if (options.format === 'chatlab' || options.format === 'chatlab-jsonl' || options.format === 'json' || options.format === 'excel' || options.format === 'txt' || options.format === 'html') {
|
||||||
const result = await window.electronAPI.export.exportSessions(
|
const result = await window.electronAPI.export.exportSessions(
|
||||||
sessionList,
|
sessionList,
|
||||||
exportFolder,
|
exportFolder,
|
||||||
@@ -161,16 +312,28 @@ function ExportPage() {
|
|||||||
)
|
)
|
||||||
setExportResult(result)
|
setExportResult(result)
|
||||||
} else {
|
} else {
|
||||||
setExportResult({ success: false, error: `${options.format.toUpperCase()} 格式导出功能开发中...` })
|
setExportResult({ success: false, error: `${options.format.toUpperCase()} 格式目前暂未实现,请选择其他格式。` })
|
||||||
}
|
}
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('导出失败:', e)
|
console.error('导出过程中发生异常:', e)
|
||||||
setExportResult({ success: false, error: String(e) })
|
setExportResult({ success: false, error: String(e) })
|
||||||
} finally {
|
} finally {
|
||||||
setIsExporting(false)
|
setIsExporting(false)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const startExport = () => {
|
||||||
|
if (selectedSessions.size === 0 || !exportFolder) return
|
||||||
|
|
||||||
|
if (options.exportMedia && selectedSessions.size > 1) {
|
||||||
|
setShowMediaLayoutPrompt(true)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
const layout: SessionLayout = options.exportMedia ? 'per-session' : 'shared'
|
||||||
|
runExport(layout)
|
||||||
|
}
|
||||||
|
|
||||||
const getDaysInMonth = (date: Date) => {
|
const getDaysInMonth = (date: Date) => {
|
||||||
const year = date.getFullYear()
|
const year = date.getFullYear()
|
||||||
const month = date.getMonth()
|
const month = date.getMonth()
|
||||||
@@ -203,18 +366,54 @@ function ExportPage() {
|
|||||||
const year = calendarDate.getFullYear()
|
const year = calendarDate.getFullYear()
|
||||||
const month = calendarDate.getMonth()
|
const month = calendarDate.getMonth()
|
||||||
const selectedDate = new Date(year, month, day)
|
const selectedDate = new Date(year, month, day)
|
||||||
|
// 设置时间为当天的开始或结束
|
||||||
|
selectedDate.setHours(selectingStart ? 0 : 23, selectingStart ? 0 : 59, selectingStart ? 0 : 59, selectingStart ? 0 : 999)
|
||||||
|
|
||||||
|
const now = new Date()
|
||||||
|
// 如果选择的日期晚于当前时间,限制为当前时间
|
||||||
|
if (selectedDate > now) {
|
||||||
|
selectedDate.setTime(now.getTime())
|
||||||
|
}
|
||||||
|
|
||||||
if (selectingStart) {
|
if (selectingStart) {
|
||||||
|
// 选择开始日期
|
||||||
|
const currentEnd = options.dateRange?.end || new Date()
|
||||||
|
// 如果选择的开始日期晚于结束日期,则同时更新结束日期
|
||||||
|
if (selectedDate > currentEnd) {
|
||||||
|
const newEnd = new Date(selectedDate)
|
||||||
|
newEnd.setHours(23, 59, 59, 999)
|
||||||
|
// 确保结束日期也不晚于当前时间
|
||||||
|
if (newEnd > now) {
|
||||||
|
newEnd.setTime(now.getTime())
|
||||||
|
}
|
||||||
|
setOptions({
|
||||||
|
...options,
|
||||||
|
dateRange: { start: selectedDate, end: newEnd }
|
||||||
|
})
|
||||||
|
} else {
|
||||||
setOptions({
|
setOptions({
|
||||||
...options,
|
...options,
|
||||||
dateRange: options.dateRange ? { ...options.dateRange, start: selectedDate } : { start: selectedDate, end: new Date() }
|
dateRange: options.dateRange ? { ...options.dateRange, start: selectedDate } : { start: selectedDate, end: new Date() }
|
||||||
})
|
})
|
||||||
|
}
|
||||||
setSelectingStart(false)
|
setSelectingStart(false)
|
||||||
|
} else {
|
||||||
|
// 选择结束日期
|
||||||
|
const currentStart = options.dateRange?.start || new Date(Date.now() - 7 * 24 * 60 * 60 * 1000)
|
||||||
|
// 如果选择的结束日期早于开始日期,则同时更新开始日期
|
||||||
|
if (selectedDate < currentStart) {
|
||||||
|
const newStart = new Date(selectedDate)
|
||||||
|
newStart.setHours(0, 0, 0, 0)
|
||||||
|
setOptions({
|
||||||
|
...options,
|
||||||
|
dateRange: { start: newStart, end: selectedDate }
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
setOptions({
|
setOptions({
|
||||||
...options,
|
...options,
|
||||||
dateRange: options.dateRange ? { ...options.dateRange, end: selectedDate } : { start: new Date(), end: selectedDate }
|
dateRange: options.dateRange ? { ...options.dateRange, end: selectedDate } : { start: new Date(), end: selectedDate }
|
||||||
})
|
})
|
||||||
|
}
|
||||||
setSelectingStart(true)
|
setSelectingStart(true)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -228,6 +427,25 @@ function ExportPage() {
|
|||||||
{ value: 'excel', label: 'Excel', icon: FileSpreadsheet, desc: '电子表格,适合统计分析' },
|
{ value: 'excel', label: 'Excel', icon: FileSpreadsheet, desc: '电子表格,适合统计分析' },
|
||||||
{ value: 'sql', label: 'PostgreSQL', icon: Database, desc: '数据库脚本,便于导入到数据库' }
|
{ value: 'sql', label: 'PostgreSQL', icon: Database, desc: '数据库脚本,便于导入到数据库' }
|
||||||
]
|
]
|
||||||
|
const displayNameOptions = [
|
||||||
|
{
|
||||||
|
value: 'group-nickname',
|
||||||
|
label: '群昵称优先',
|
||||||
|
desc: '仅群聊有效,私聊显示备注/昵称'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'remark',
|
||||||
|
label: '备注优先',
|
||||||
|
desc: '有备注显示备注,否则显示昵称'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'nickname',
|
||||||
|
label: '微信昵称',
|
||||||
|
desc: '始终显示微信昵称'
|
||||||
|
}
|
||||||
|
]
|
||||||
|
const displayNameOption = displayNameOptions.find(option => option.value === options.displayNamePreference)
|
||||||
|
const displayNameLabel = displayNameOption?.label || '备注优先'
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="export-page">
|
<div className="export-page">
|
||||||
@@ -311,7 +529,7 @@ function ExportPage() {
|
|||||||
<div
|
<div
|
||||||
key={fmt.value}
|
key={fmt.value}
|
||||||
className={`format-card ${options.format === fmt.value ? 'active' : ''}`}
|
className={`format-card ${options.format === fmt.value ? 'active' : ''}`}
|
||||||
onClick={() => setOptions({ ...options, format: fmt.value as any })}
|
onClick={() => handleFormatChange(fmt.value as ExportOptions['format'])}
|
||||||
>
|
>
|
||||||
<fmt.icon size={24} />
|
<fmt.icon size={24} />
|
||||||
<span className="format-label">{fmt.label}</span>
|
<span className="format-label">{fmt.label}</span>
|
||||||
@@ -323,38 +541,191 @@ function ExportPage() {
|
|||||||
|
|
||||||
<div className="setting-section">
|
<div className="setting-section">
|
||||||
<h3>时间范围</h3>
|
<h3>时间范围</h3>
|
||||||
<div className="time-options">
|
<p className="setting-subtitle">选择要导出的消息时间区间</p>
|
||||||
<label className="checkbox-item">
|
<div className="media-options-card">
|
||||||
|
<div className="media-switch-row">
|
||||||
|
<div className="media-switch-info">
|
||||||
|
<span className="media-switch-title">导出全部时间</span>
|
||||||
|
<span className="media-switch-desc">关闭此项以选择特定的起止日期</span>
|
||||||
|
</div>
|
||||||
|
<label className="switch">
|
||||||
<input
|
<input
|
||||||
type="checkbox"
|
type="checkbox"
|
||||||
checked={options.useAllTime}
|
checked={options.useAllTime}
|
||||||
onChange={e => setOptions({ ...options, useAllTime: e.target.checked })}
|
onChange={e => setOptions({ ...options, useAllTime: e.target.checked })}
|
||||||
/>
|
/>
|
||||||
<span>导出全部时间</span>
|
<span className="switch-slider"></span>
|
||||||
</label>
|
</label>
|
||||||
|
</div>
|
||||||
|
|
||||||
{!options.useAllTime && options.dateRange && (
|
{!options.useAllTime && options.dateRange && (
|
||||||
<div className="date-range" onClick={() => setShowDatePicker(true)}>
|
<>
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
<div className="time-range-picker-item" onClick={() => setShowDatePicker(true)}>
|
||||||
|
<div className="time-picker-info">
|
||||||
<Calendar size={16} />
|
<Calendar size={16} />
|
||||||
<span>{formatDate(options.dateRange.start)} - {formatDate(options.dateRange.end)}</span>
|
<span>{formatDate(options.dateRange.start)} - {formatDate(options.dateRange.end)}</span>
|
||||||
|
</div>
|
||||||
<ChevronDown size={14} />
|
<ChevronDown size={14} />
|
||||||
</div>
|
</div>
|
||||||
|
</>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{/* 发送者名称显示偏好 */}
|
||||||
|
{(options.format === 'html' || options.format === 'json' || options.format === 'txt') && (
|
||||||
<div className="setting-section">
|
<div className="setting-section">
|
||||||
<h3>导出头像</h3>
|
<h3>发送者名称显示</h3>
|
||||||
<div className="time-options">
|
<p className="setting-subtitle">选择导出时优先显示的名称</p>
|
||||||
<label className="checkbox-item">
|
<div className="select-field" ref={displayNameDropdownRef}>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className={`select-trigger ${showDisplayNameSelect ? 'open' : ''}`}
|
||||||
|
onClick={() => setShowDisplayNameSelect(!showDisplayNameSelect)}
|
||||||
|
>
|
||||||
|
<span className="select-value">{displayNameLabel}</span>
|
||||||
|
<ChevronDown size={16} />
|
||||||
|
</button>
|
||||||
|
{showDisplayNameSelect && (
|
||||||
|
<div className="select-dropdown">
|
||||||
|
{displayNameOptions.map(option => (
|
||||||
|
<button
|
||||||
|
key={option.value}
|
||||||
|
type="button"
|
||||||
|
className={`select-option ${options.displayNamePreference === option.value ? 'active' : ''}`}
|
||||||
|
onClick={() => {
|
||||||
|
setOptions({
|
||||||
|
...options,
|
||||||
|
displayNamePreference: option.value as ExportOptions['displayNamePreference']
|
||||||
|
})
|
||||||
|
setShowDisplayNameSelect(false)
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="option-label">{option.label}</span>
|
||||||
|
<span className="option-desc">{option.desc}</span>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
<div className="setting-section">
|
||||||
|
<h3>媒体文件</h3>
|
||||||
|
<p className="setting-subtitle">导出图片/语音/视频/表情并在记录内写入相对路径</p>
|
||||||
|
<div className="media-options-card">
|
||||||
|
<div className="media-switch-row">
|
||||||
|
<div className="media-switch-info">
|
||||||
|
<span className="media-switch-title">导出媒体文件</span>
|
||||||
|
<span className="media-switch-desc">会创建子文件夹并保存媒体资源</span>
|
||||||
|
</div>
|
||||||
|
<label className="switch">
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportMedia}
|
||||||
|
onChange={e => setOptions({ ...options, exportMedia: e.target.checked })}
|
||||||
|
/>
|
||||||
|
<span className="switch-slider"></span>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
|
||||||
|
<label className={`media-checkbox-row ${!options.exportMedia ? 'disabled' : ''}`}>
|
||||||
|
<div className="media-checkbox-info">
|
||||||
|
<span className="media-checkbox-title">图片</span>
|
||||||
|
<span className="media-checkbox-desc">已有文件直接复制,缺失时尝试解密</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportImages}
|
||||||
|
disabled={!options.exportMedia}
|
||||||
|
onChange={e => setOptions({ ...options, exportImages: e.target.checked })}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
|
||||||
|
<label className={`media-checkbox-row ${!options.exportMedia ? 'disabled' : ''}`}>
|
||||||
|
<div className="media-checkbox-info">
|
||||||
|
<span className="media-checkbox-title">语音</span>
|
||||||
|
<span className="media-checkbox-desc">缺失时会解码生成 MP3</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportVoices}
|
||||||
|
disabled={!options.exportMedia}
|
||||||
|
onChange={e => setOptions({ ...options, exportVoices: e.target.checked })}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
|
||||||
|
<label className="media-checkbox-row">
|
||||||
|
<div className="media-checkbox-info">
|
||||||
|
<span className="media-checkbox-title">语音转文字</span>
|
||||||
|
<span className="media-checkbox-desc">将语音消息转换为文字导出</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportVoiceAsText}
|
||||||
|
onChange={e => setOptions({ ...options, exportVoiceAsText: e.target.checked })}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
|
||||||
|
<label className={`media-checkbox-row ${!options.exportMedia ? 'disabled' : ''}`}>
|
||||||
|
<div className="media-checkbox-info">
|
||||||
|
<span className="media-checkbox-title">视频</span>
|
||||||
|
<span className="media-checkbox-desc">直接复制视频文件到导出目录</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportVideos}
|
||||||
|
disabled={!options.exportMedia}
|
||||||
|
onChange={e => setOptions({ ...options, exportVideos: e.target.checked })}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<div className="media-option-divider"></div>
|
||||||
|
|
||||||
|
<label className={`media-checkbox-row ${!options.exportMedia ? 'disabled' : ''}`}>
|
||||||
|
<div className="media-checkbox-info">
|
||||||
|
<span className="media-checkbox-title">表情</span>
|
||||||
|
<span className="media-checkbox-desc">本地无缓存时尝试下载</span>
|
||||||
|
</div>
|
||||||
|
<input
|
||||||
|
type="checkbox"
|
||||||
|
checked={options.exportEmojis}
|
||||||
|
disabled={!options.exportMedia}
|
||||||
|
onChange={e => setOptions({ ...options, exportEmojis: e.target.checked })}
|
||||||
|
/>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="setting-section">
|
||||||
|
<h3>头像</h3>
|
||||||
|
<p className="setting-subtitle">可选导出头像索引,关闭则不下载头像</p>
|
||||||
|
<div className="media-options-card">
|
||||||
|
<div className="media-switch-row">
|
||||||
|
<div className="media-switch-info">
|
||||||
|
<span className="media-switch-title">导出头像</span>
|
||||||
|
<span className="media-switch-desc">用于展示发送者头像,可能会读取或下载头像文件</span>
|
||||||
|
</div>
|
||||||
|
<label className="switch">
|
||||||
<input
|
<input
|
||||||
type="checkbox"
|
type="checkbox"
|
||||||
checked={options.exportAvatars}
|
checked={options.exportAvatars}
|
||||||
onChange={e => setOptions({ ...options, exportAvatars: e.target.checked })}
|
onChange={e => setOptions({ ...options, exportAvatars: e.target.checked })}
|
||||||
/>
|
/>
|
||||||
<span>导出头像图片</span>
|
<span className="switch-slider"></span>
|
||||||
</label>
|
</label>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
<div className="setting-section">
|
<div className="setting-section">
|
||||||
<h3>导出位置</h3>
|
<h3>导出位置</h3>
|
||||||
@@ -406,6 +777,43 @@ function ExportPage() {
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{/* 媒体导出布局选择弹窗 */}
|
||||||
|
{showMediaLayoutPrompt && (
|
||||||
|
<div className="export-overlay" onClick={() => setShowMediaLayoutPrompt(false)}>
|
||||||
|
<div className="export-layout-modal" onClick={e => e.stopPropagation()}>
|
||||||
|
<h3>导出文件夹布局</h3>
|
||||||
|
<p className="layout-subtitle">检测到同时导出多个会话并包含媒体文件,请选择存放方式:</p>
|
||||||
|
<div className="layout-options">
|
||||||
|
<button
|
||||||
|
className="layout-option-btn primary"
|
||||||
|
onClick={() => {
|
||||||
|
setShowMediaLayoutPrompt(false)
|
||||||
|
runExport('shared')
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="layout-title">所有会话在同一文件夹</span>
|
||||||
|
<span className="layout-desc">媒体会按会话名归档到 media 子目录</span>
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
className="layout-option-btn"
|
||||||
|
onClick={() => {
|
||||||
|
setShowMediaLayoutPrompt(false)
|
||||||
|
runExport('per-session')
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<span className="layout-title">每个会话一个文件夹</span>
|
||||||
|
<span className="layout-desc">每个会话单独包含导出文件和媒体</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div className="layout-actions">
|
||||||
|
<button className="layout-cancel-btn" onClick={() => setShowMediaLayoutPrompt(false)}>
|
||||||
|
取消
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* 导出进度弹窗 */}
|
{/* 导出进度弹窗 */}
|
||||||
{isExporting && (
|
{isExporting && (
|
||||||
<div className="export-overlay">
|
<div className="export-overlay">
|
||||||
@@ -462,6 +870,9 @@ function ExportPage() {
|
|||||||
<div className="export-overlay" onClick={() => setShowDatePicker(false)}>
|
<div className="export-overlay" onClick={() => setShowDatePicker(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>
|
||||||
<div className="quick-select">
|
<div className="quick-select">
|
||||||
<button
|
<button
|
||||||
className="quick-btn"
|
className="quick-btn"
|
||||||
@@ -556,12 +967,16 @@ function ExportPage() {
|
|||||||
const isStart = options.dateRange?.start.toDateString() === currentDate.toDateString()
|
const isStart = options.dateRange?.start.toDateString() === currentDate.toDateString()
|
||||||
const isEnd = options.dateRange?.end.toDateString() === currentDate.toDateString()
|
const isEnd = options.dateRange?.end.toDateString() === currentDate.toDateString()
|
||||||
const isInRange = options.dateRange && currentDate >= options.dateRange.start && currentDate <= options.dateRange.end
|
const isInRange = options.dateRange && currentDate >= options.dateRange.start && currentDate <= options.dateRange.end
|
||||||
|
const today = new Date()
|
||||||
|
today.setHours(0, 0, 0, 0)
|
||||||
|
const isFuture = currentDate > today
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div
|
<div
|
||||||
key={day}
|
key={day}
|
||||||
className={`calendar-day ${isStart ? 'start' : ''} ${isEnd ? 'end' : ''} ${isInRange ? 'in-range' : ''}`}
|
className={`calendar-day ${isStart ? 'start' : ''} ${isEnd ? 'end' : ''} ${isInRange ? 'in-range' : ''} ${isFuture ? 'disabled' : ''}`}
|
||||||
onClick={() => handleDateSelect(day)}
|
onClick={() => !isFuture && handleDateSelect(day)}
|
||||||
|
style={{ cursor: isFuture ? 'not-allowed' : 'pointer', opacity: isFuture ? 0.3 : 1 }}
|
||||||
>
|
>
|
||||||
{day}
|
{day}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -333,7 +333,7 @@
|
|||||||
.group-avatar {
|
.group-avatar {
|
||||||
width: 44px;
|
width: 44px;
|
||||||
height: 44px;
|
height: 44px;
|
||||||
border-radius: 50%;
|
border-radius: 8px;
|
||||||
overflow: hidden;
|
overflow: hidden;
|
||||||
flex-shrink: 0;
|
flex-shrink: 0;
|
||||||
|
|
||||||
@@ -346,11 +346,11 @@
|
|||||||
.avatar-placeholder {
|
.avatar-placeholder {
|
||||||
width: 100%;
|
width: 100%;
|
||||||
height: 100%;
|
height: 100%;
|
||||||
background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
|
background: var(--bg-tertiary);
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
justify-content: center;
|
justify-content: center;
|
||||||
color: #fff;
|
color: var(--text-secondary);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -390,7 +390,7 @@
|
|||||||
.skeleton-avatar {
|
.skeleton-avatar {
|
||||||
width: 44px;
|
width: 44px;
|
||||||
height: 44px;
|
height: 44px;
|
||||||
border-radius: 50%;
|
border-radius: 8px;
|
||||||
background: var(--bg-tertiary);
|
background: var(--bg-tertiary);
|
||||||
animation: pulse 1.5s infinite;
|
animation: pulse 1.5s infinite;
|
||||||
}
|
}
|
||||||
@@ -500,7 +500,7 @@
|
|||||||
.group-avatar.large {
|
.group-avatar.large {
|
||||||
width: 80px;
|
width: 80px;
|
||||||
height: 80px;
|
height: 80px;
|
||||||
border-radius: 50%;
|
border-radius: 10px;
|
||||||
overflow: hidden;
|
overflow: hidden;
|
||||||
margin: 0 auto 16px;
|
margin: 0 auto 16px;
|
||||||
|
|
||||||
@@ -513,11 +513,11 @@
|
|||||||
.avatar-placeholder {
|
.avatar-placeholder {
|
||||||
width: 100%;
|
width: 100%;
|
||||||
height: 100%;
|
height: 100%;
|
||||||
background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
|
background: var(--bg-tertiary);
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
justify-content: center;
|
justify-content: center;
|
||||||
color: #fff;
|
color: var(--text-secondary);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -656,6 +656,32 @@
|
|||||||
cursor: not-allowed;
|
cursor: not-allowed;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.export-btn {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 6px 10px;
|
||||||
|
border: none;
|
||||||
|
background: var(--bg-tertiary);
|
||||||
|
border-radius: 8px;
|
||||||
|
color: var(--text-secondary);
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.2s;
|
||||||
|
-webkit-app-region: no-drag;
|
||||||
|
font-size: 12px;
|
||||||
|
flex-shrink: 0;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: var(--bg-hover);
|
||||||
|
color: var(--text-primary);
|
||||||
|
}
|
||||||
|
|
||||||
|
&:disabled {
|
||||||
|
opacity: 0.4;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
.content-body {
|
.content-body {
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import { useState, useEffect, useRef } from 'react'
|
import { useState, useEffect, useRef, useCallback } from 'react'
|
||||||
import { Users, BarChart3, Clock, Image, Loader2, RefreshCw, User, Medal, Search, X, ChevronLeft, Copy, Check } from 'lucide-react'
|
import { Users, BarChart3, Clock, Image, Loader2, RefreshCw, User, Medal, Search, X, ChevronLeft, Copy, Check, Download } from 'lucide-react'
|
||||||
|
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 './GroupAnalyticsPage.scss'
|
import './GroupAnalyticsPage.scss'
|
||||||
@@ -15,6 +16,10 @@ interface GroupMember {
|
|||||||
username: string
|
username: string
|
||||||
displayName: string
|
displayName: string
|
||||||
avatarUrl?: string
|
avatarUrl?: string
|
||||||
|
nickname?: string
|
||||||
|
alias?: string
|
||||||
|
remark?: string
|
||||||
|
groupNickname?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
interface GroupMessageRank {
|
interface GroupMessageRank {
|
||||||
@@ -38,6 +43,7 @@ function GroupAnalyticsPage() {
|
|||||||
const [activeHours, setActiveHours] = useState<Record<number, number>>({})
|
const [activeHours, setActiveHours] = useState<Record<number, number>>({})
|
||||||
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 [selectedMember, setSelectedMember] = useState<GroupMember | null>(null)
|
const [selectedMember, setSelectedMember] = useState<GroupMember | null>(null)
|
||||||
@@ -92,7 +98,7 @@ function GroupAnalyticsPage() {
|
|||||||
}
|
}
|
||||||
}, [dateRangeReady])
|
}, [dateRangeReady])
|
||||||
|
|
||||||
const loadGroups = async () => {
|
const loadGroups = useCallback(async () => {
|
||||||
setIsLoading(true)
|
setIsLoading(true)
|
||||||
try {
|
try {
|
||||||
const result = await window.electronAPI.groupAnalytics.getGroupChats()
|
const result = await window.electronAPI.groupAnalytics.getGroupChats()
|
||||||
@@ -105,7 +111,23 @@ function GroupAnalyticsPage() {
|
|||||||
} finally {
|
} finally {
|
||||||
setIsLoading(false)
|
setIsLoading(false)
|
||||||
}
|
}
|
||||||
|
}, [])
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const handleChange = () => {
|
||||||
|
setGroups([])
|
||||||
|
setFilteredGroups([])
|
||||||
|
setSelectedGroup(null)
|
||||||
|
setSelectedFunction(null)
|
||||||
|
setMembers([])
|
||||||
|
setRankings([])
|
||||||
|
setActiveHours({})
|
||||||
|
setMediaStats(null)
|
||||||
|
void loadGroups()
|
||||||
}
|
}
|
||||||
|
window.addEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
return () => window.removeEventListener('wxid-changed', handleChange as EventListener)
|
||||||
|
}, [loadGroups])
|
||||||
|
|
||||||
const handleGroupSelect = (group: GroupChatInfo) => {
|
const handleGroupSelect = (group: GroupChatInfo) => {
|
||||||
if (selectedGroup?.username !== group.username) {
|
if (selectedGroup?.username !== group.username) {
|
||||||
@@ -164,6 +186,10 @@ function GroupAnalyticsPage() {
|
|||||||
return num.toLocaleString()
|
return num.toLocaleString()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const sanitizeFileName = (name: string) => {
|
||||||
|
return name.replace(/[<>:"/\\|?*]+/g, '_').trim()
|
||||||
|
}
|
||||||
|
|
||||||
const getHourlyOption = () => {
|
const getHourlyOption = () => {
|
||||||
const hours = Array.from({ length: 24 }, (_, i) => i)
|
const hours = Array.from({ length: 24 }, (_, i) => i)
|
||||||
const data = hours.map(h => activeHours[h] || 0)
|
const data = hours.map(h => activeHours[h] || 0)
|
||||||
@@ -235,6 +261,35 @@ function GroupAnalyticsPage() {
|
|||||||
setCopiedField(null)
|
setCopiedField(null)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const handleExportMembers = async () => {
|
||||||
|
if (!selectedGroup || isExportingMembers) return
|
||||||
|
setIsExportingMembers(true)
|
||||||
|
try {
|
||||||
|
const downloadsPath = await window.electronAPI.app.getDownloadsPath()
|
||||||
|
const baseName = sanitizeFileName(`${selectedGroup.displayName || selectedGroup.username}_群成员列表`)
|
||||||
|
const separator = downloadsPath && downloadsPath.includes('\\') ? '\\' : '/'
|
||||||
|
const defaultPath = downloadsPath ? `${downloadsPath}${separator}${baseName}.xlsx` : `${baseName}.xlsx`
|
||||||
|
const saveResult = await window.electronAPI.dialog.saveFile({
|
||||||
|
title: '导出群成员列表',
|
||||||
|
defaultPath,
|
||||||
|
filters: [{ name: 'Excel', extensions: ['xlsx'] }]
|
||||||
|
})
|
||||||
|
if (!saveResult || saveResult.canceled || !saveResult.filePath) return
|
||||||
|
|
||||||
|
const result = await window.electronAPI.groupAnalytics.exportGroupMembers(selectedGroup.username, saveResult.filePath)
|
||||||
|
if (result.success) {
|
||||||
|
alert(`导出成功,共 ${result.count ?? members.length} 人`)
|
||||||
|
} else {
|
||||||
|
alert(`导出失败:${result.error || '未知错误'}`)
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('导出群成员失败:', e)
|
||||||
|
alert(`导出失败:${String(e)}`)
|
||||||
|
} finally {
|
||||||
|
setIsExportingMembers(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)
|
||||||
@@ -247,6 +302,10 @@ function GroupAnalyticsPage() {
|
|||||||
|
|
||||||
const renderMemberModal = () => {
|
const renderMemberModal = () => {
|
||||||
if (!selectedMember) return null
|
if (!selectedMember) return null
|
||||||
|
const nickname = (selectedMember.nickname || '').trim()
|
||||||
|
const alias = (selectedMember.alias || '').trim()
|
||||||
|
const remark = (selectedMember.remark || '').trim()
|
||||||
|
const groupNickname = (selectedMember.groupNickname || '').trim()
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="member-modal-overlay" onClick={() => setSelectedMember(null)}>
|
<div className="member-modal-overlay" onClick={() => setSelectedMember(null)}>
|
||||||
@@ -256,11 +315,7 @@ function GroupAnalyticsPage() {
|
|||||||
</button>
|
</button>
|
||||||
<div className="modal-content">
|
<div className="modal-content">
|
||||||
<div className="member-avatar large">
|
<div className="member-avatar large">
|
||||||
{selectedMember.avatarUrl ? (
|
<Avatar src={selectedMember.avatarUrl} name={selectedMember.displayName} size={96} />
|
||||||
<img src={selectedMember.avatarUrl} alt="" />
|
|
||||||
) : (
|
|
||||||
<div className="avatar-placeholder"><User size={48} /></div>
|
|
||||||
)}
|
|
||||||
</div>
|
</div>
|
||||||
<h3 className="member-display-name">{selectedMember.displayName}</h3>
|
<h3 className="member-display-name">{selectedMember.displayName}</h3>
|
||||||
<div className="member-details">
|
<div className="member-details">
|
||||||
@@ -273,11 +328,40 @@ function GroupAnalyticsPage() {
|
|||||||
</div>
|
</div>
|
||||||
<div className="detail-row">
|
<div className="detail-row">
|
||||||
<span className="detail-label">昵称</span>
|
<span className="detail-label">昵称</span>
|
||||||
<span className="detail-value">{selectedMember.displayName}</span>
|
<span className="detail-value">{nickname || '未设置'}</span>
|
||||||
<button className="copy-btn" onClick={() => handleCopy(selectedMember.displayName, 'displayName')}>
|
{nickname && (
|
||||||
{copiedField === 'displayName' ? <Check size={14} /> : <Copy size={14} />}
|
<button className="copy-btn" onClick={() => handleCopy(nickname, 'nickname')}>
|
||||||
|
{copiedField === 'nickname' ? <Check size={14} /> : <Copy size={14} />}
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
{alias && (
|
||||||
|
<div className="detail-row">
|
||||||
|
<span className="detail-label">微信号</span>
|
||||||
|
<span className="detail-value">{alias}</span>
|
||||||
|
<button className="copy-btn" onClick={() => handleCopy(alias, 'alias')}>
|
||||||
|
{copiedField === 'alias' ? <Check size={14} /> : <Copy size={14} />}
|
||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
)}
|
||||||
|
{groupNickname && (
|
||||||
|
<div className="detail-row">
|
||||||
|
<span className="detail-label">群昵称</span>
|
||||||
|
<span className="detail-value">{groupNickname}</span>
|
||||||
|
<button className="copy-btn" onClick={() => handleCopy(groupNickname, 'groupNickname')}>
|
||||||
|
{copiedField === 'groupNickname' ? <Check size={14} /> : <Copy size={14} />}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
{remark && (
|
||||||
|
<div className="detail-row">
|
||||||
|
<span className="detail-label">备注</span>
|
||||||
|
<span className="detail-value">{remark}</span>
|
||||||
|
<button className="copy-btn" onClick={() => handleCopy(remark, 'remark')}>
|
||||||
|
{copiedField === 'remark' ? <Check size={14} /> : <Copy size={14} />}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -334,7 +418,7 @@ function GroupAnalyticsPage() {
|
|||||||
onClick={() => handleGroupSelect(group)}
|
onClick={() => handleGroupSelect(group)}
|
||||||
>
|
>
|
||||||
<div className="group-avatar">
|
<div className="group-avatar">
|
||||||
{group.avatarUrl ? <img src={group.avatarUrl} alt="" /> : <div className="avatar-placeholder"><Users size={20} /></div>}
|
<Avatar src={group.avatarUrl} name={group.displayName} size={44} />
|
||||||
</div>
|
</div>
|
||||||
<div className="group-info">
|
<div className="group-info">
|
||||||
<span className="group-name">{group.displayName}</span>
|
<span className="group-name">{group.displayName}</span>
|
||||||
@@ -352,7 +436,7 @@ function GroupAnalyticsPage() {
|
|||||||
<div className="function-menu">
|
<div className="function-menu">
|
||||||
<div className="selected-group-info">
|
<div className="selected-group-info">
|
||||||
<div className="group-avatar large">
|
<div className="group-avatar large">
|
||||||
{selectedGroup?.avatarUrl ? <img src={selectedGroup.avatarUrl} alt="" /> : <div className="avatar-placeholder"><Users size={40} /></div>}
|
<Avatar src={selectedGroup?.avatarUrl} name={selectedGroup?.displayName} size={80} />
|
||||||
</div>
|
</div>
|
||||||
<h2>{selectedGroup?.displayName}</h2>
|
<h2>{selectedGroup?.displayName}</h2>
|
||||||
<p>{selectedGroup?.memberCount} 位成员</p>
|
<p>{selectedGroup?.memberCount} 位成员</p>
|
||||||
@@ -410,6 +494,12 @@ function GroupAnalyticsPage() {
|
|||||||
onRangeComplete={handleDateRangeComplete}
|
onRangeComplete={handleDateRangeComplete}
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
|
{selectedFunction === 'members' && (
|
||||||
|
<button className="export-btn" onClick={handleExportMembers} disabled={functionLoading || isExportingMembers}>
|
||||||
|
{isExportingMembers ? <Loader2 size={16} className="spin" /> : <Download size={16} />}
|
||||||
|
<span>导出成员</span>
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
<button className="refresh-btn" onClick={handleRefresh} disabled={functionLoading}>
|
<button className="refresh-btn" onClick={handleRefresh} disabled={functionLoading}>
|
||||||
<RefreshCw size={16} className={functionLoading ? 'spin' : ''} />
|
<RefreshCw size={16} className={functionLoading ? 'spin' : ''} />
|
||||||
</button>
|
</button>
|
||||||
@@ -424,7 +514,7 @@ function GroupAnalyticsPage() {
|
|||||||
{members.map(member => (
|
{members.map(member => (
|
||||||
<div key={member.username} className="member-card" onClick={() => handleMemberClick(member)}>
|
<div key={member.username} className="member-card" onClick={() => handleMemberClick(member)}>
|
||||||
<div className="member-avatar">
|
<div className="member-avatar">
|
||||||
{member.avatarUrl ? <img src={member.avatarUrl} alt="" /> : <div className="avatar-placeholder"><User size={20} /></div>}
|
<Avatar src={member.avatarUrl} name={member.displayName} size={48} />
|
||||||
</div>
|
</div>
|
||||||
<span className="member-name">{member.displayName}</span>
|
<span className="member-name">{member.displayName}</span>
|
||||||
</div>
|
</div>
|
||||||
@@ -437,7 +527,7 @@ function GroupAnalyticsPage() {
|
|||||||
<div key={item.member.username} className="ranking-item">
|
<div key={item.member.username} className="ranking-item">
|
||||||
<span className={`rank ${index < 3 ? 'top' : ''}`}>{index + 1}</span>
|
<span className={`rank ${index < 3 ? 'top' : ''}`}>{index + 1}</span>
|
||||||
<div className="contact-avatar">
|
<div className="contact-avatar">
|
||||||
{item.member.avatarUrl ? <img src={item.member.avatarUrl} alt="" /> : <div className="avatar-placeholder"><User size={20} /></div>}
|
<Avatar src={item.member.avatarUrl} name={item.member.displayName} size={40} />
|
||||||
{index < 3 && <div className={`medal medal-${index + 1}`}><Medal size={10} /></div>}
|
{index < 3 && <div className={`medal medal-${index + 1}`}><Medal size={10} /></div>}
|
||||||
</div>
|
</div>
|
||||||
<div className="contact-info">
|
<div className="contact-info">
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user