mirror of https://github.com/jackwener/wx-cli.git
joshua-deng
GitHub
commit
3e79c8e093
|
|
@ -132,13 +132,15 @@ claude mcp add wechat -- python C:\Users\你的用户名\wechat-decrypt\mcp_serv
|
|||
| Tool | 功能 |
|
||||
|------|------|
|
||||
| `get_recent_sessions(limit)` | 最近会话列表（含消息摘要、未读数） |
|
||||
| `get_chat_history(chat_name, limit)` | 指定聊天的消息记录（支持模糊匹配名字） |
|
||||
| `search_messages(keyword, limit)` | 全库搜索消息内容 |
|
||||
| `get_chat_history(chat_name, limit, offset, start_time, end_time)` | 指定聊天的消息记录，支持时间范围和分页 |
|
||||
| `search_messages(keyword, chat_name, start_time, end_time, limit, offset)` | 统一搜索消息；支持全库、单个聊天对象、多个聊天对象、时间范围和分页 |
|
||||
| `get_contacts(query, limit)` | 搜索/列出联系人 |
|
||||
| `get_new_messages()` | 获取自上次调用以来的新消息 |
|
||||
|
||||
前置条件：需要先运行 `python main.py` 或 `python find_all_keys.py` 完成密钥提取。
|
||||
|
||||
说明：`search_messages` 的 `limit` 最大为 `500`；`get_chat_history` 支持更大的 `limit`，但消息很多时仍建议配合 `offset` 分页读取。
|
||||
|
||||
**[查看使用案例 →](USAGE.md)**
|
||||
|
||||
### 图片解密 (V2 格式)
|
||||
|
|
|
|||
80
USAGE.md
80
USAGE.md
|
|
@ -66,7 +66,7 @@ Claude 调用 `get_chat_history`，然后自动分析总结：
|
|||
> 搜一下谁提过"claude"
|
||||
```
|
||||
|
||||
Claude 调用 `search_messages(keyword="claude")`：
|
||||
Claude 调用 `search_messages(keyword="claude")`：
|
||||
|
||||
```
|
||||
搜索 "claude" 找到 20 条结果:
|
||||
|
|
@ -78,7 +78,77 @@ Claude 调用 `search_messages(keyword="claude")`：
|
|||
...
|
||||
```
|
||||
|
||||
## 4. 搜索联系人
|
||||
## 4. 时间范围 + 分页查看聊天记录
|
||||
|
||||
```
|
||||
> 帮我看一下██群 3 月 1 日到 3 月 7 日的聊天，先给我前 20 条
|
||||
```
|
||||
|
||||
Claude 可以调用：
|
||||
|
||||
```python
|
||||
get_chat_history(
|
||||
chat_name="██群",
|
||||
start_time="2026-03-01",
|
||||
end_time="2026-03-07",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
```
|
||||
|
||||
下一页：
|
||||
|
||||
```python
|
||||
get_chat_history(
|
||||
chat_name="██群",
|
||||
start_time="2026-03-01",
|
||||
end_time="2026-03-07",
|
||||
limit=20,
|
||||
offset=20,
|
||||
)
|
||||
```
|
||||
|
||||
## 5. 搜索指定联系人/群聊在某个时间段内的消息
|
||||
|
||||
```
|
||||
> 帮我搜一下██群这周谁提到过 Claude
|
||||
```
|
||||
|
||||
Claude 可以调用统一接口：
|
||||
|
||||
```python
|
||||
search_messages(
|
||||
keyword="Claude",
|
||||
chat_name="██群",
|
||||
start_time="2026-03-01",
|
||||
end_time="2026-03-07",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
```
|
||||
|
||||
## 6. 多个联系人/群聊联合搜索
|
||||
|
||||
```
|
||||
> 帮我看看联系人A、联系人B 和 ██项目群 这周谁提到过“项目”
|
||||
```
|
||||
|
||||
Claude 可以调用统一接口：
|
||||
|
||||
```python
|
||||
search_messages(
|
||||
keyword="项目",
|
||||
chat_name=["联系人A", "联系人B", "██项目群"],
|
||||
start_time="2026-03-01",
|
||||
end_time="2026-03-07",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
```
|
||||
|
||||
如果某些名字没匹配到联系人，或没有对应消息表，结果里会单独说明。
|
||||
|
||||
## 7. 搜索联系人
|
||||
|
||||
```
|
||||
> 帮我找一下姓张的联系人
|
||||
|
|
@ -95,7 +165,7 @@ wxid_████ 备注: 张██ 昵称: 小██
|
|||
...
|
||||
```
|
||||
|
||||
## 5. 获取新消息
|
||||
## 8. 获取新消息
|
||||
|
||||
```
|
||||
> 有没有新消息
|
||||
|
|
@ -113,7 +183,7 @@ Claude 调用 `get_new_messages()`：
|
|||
[16:22] ██群 [群] (19条未读): (图片)
|
||||
```
|
||||
|
||||
## 6. 高级用法：群聊分析
|
||||
## 9. 高级用法：群聊分析
|
||||
|
||||
Claude 可以获取大量消息后自动分析活跃度、话题分布、关键人物：
|
||||
|
||||
|
|
@ -121,7 +191,7 @@ Claude 可以获取大量消息后自动分析活跃度、话题分布、关键
|
|||
> 帮我分析一下██群最近一周的情况
|
||||
```
|
||||
|
||||
Claude 会调用 `get_chat_history(chat_name="██群", limit=500)` 获取消息，然后输出：
|
||||
Claude 会调用 `get_chat_history(chat_name="██群", limit=500)` 获取消息，然后输出。消息很多时，也可以把 `limit` 设得更大，或配合 `offset` 分页读取：
|
||||
|
||||
```
|
||||
## ██群最近一周分析
|
||||
|
|
|
|||
1205
mcp_server.py
1205
mcp_server.py
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,686 @@
|
|||
import hashlib
|
||||
import os
|
||||
import sqlite3
|
||||
import tempfile
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
import mcp_server
|
||||
|
||||
|
||||
class _FakeCache:
|
||||
# 用最小缓存桩替代真实解密缓存，避免单元测试依赖本地微信环境。
|
||||
def __init__(self, mapping):
|
||||
self._mapping = mapping
|
||||
|
||||
def get(self, rel_key):
|
||||
return self._mapping.get(rel_key)
|
||||
|
||||
|
||||
def _msg_table_name(username):
|
||||
# 生产代码使用 username 的 md5 作为消息表名，测试里保持一致。
|
||||
return f"Msg_{hashlib.md5(username.encode()).hexdigest()}"
|
||||
|
||||
|
||||
def _create_message_db(path, chats):
|
||||
# 构造最小可用消息库，只包含搜索/历史查询依赖的字段。
|
||||
conn = sqlite3.connect(path)
|
||||
try:
|
||||
conn.execute("CREATE TABLE Name2Id (user_name TEXT)")
|
||||
for username, messages in chats.items():
|
||||
conn.execute("INSERT INTO Name2Id(user_name) VALUES (?)", (username,))
|
||||
table_name = _msg_table_name(username)
|
||||
conn.execute(
|
||||
f"""
|
||||
CREATE TABLE [{table_name}] (
|
||||
local_id INTEGER,
|
||||
local_type INTEGER,
|
||||
create_time INTEGER,
|
||||
real_sender_id INTEGER,
|
||||
message_content TEXT,
|
||||
WCDB_CT_message_content INTEGER
|
||||
)
|
||||
"""
|
||||
)
|
||||
for local_id, create_time, content in messages:
|
||||
conn.execute(
|
||||
f"""
|
||||
INSERT INTO [{table_name}] (
|
||||
local_id, local_type, create_time, real_sender_id,
|
||||
message_content, WCDB_CT_message_content
|
||||
) VALUES (?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(local_id, 1, create_time, 0, content, 0),
|
||||
)
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
class SearchMessagesTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.temp_dir = tempfile.TemporaryDirectory()
|
||||
self.addCleanup(self.temp_dir.cleanup)
|
||||
|
||||
def create_db(self, filename, chats):
|
||||
path = os.path.join(self.temp_dir.name, filename)
|
||||
_create_message_db(path, chats)
|
||||
return path
|
||||
|
||||
def test_validate_pagination_rejects_large_limit(self):
|
||||
# 防止单次查询过大，保证 limit 上限校验存在。
|
||||
with self.assertRaisesRegex(ValueError, "limit 不能大于 500"):
|
||||
mcp_server._validate_pagination(501, 0)
|
||||
|
||||
def test_validate_pagination_allows_large_limit_when_limit_is_unbounded(self):
|
||||
# get_chat_history 允许更大的 limit，只校验正数和 offset。
|
||||
mcp_server._validate_pagination(999999, 0, limit_max=None)
|
||||
|
||||
def test_page_search_entries_returns_chronological_results_with_offset(self):
|
||||
# 结果应先按最新时间分页，再把当前页恢复成时间正序输出。
|
||||
entries = [(1, "a"), (5, "e"), (3, "c"), (4, "d"), (2, "b")]
|
||||
|
||||
paged = mcp_server._page_search_entries(entries, limit=2, offset=1)
|
||||
|
||||
self.assertEqual(paged, [(3, "c"), (4, "d")])
|
||||
|
||||
def test_search_messages_single_chat_uses_offset_and_returns_page(self):
|
||||
# 单聊分页应只返回当前页，并按聊天阅读顺序展示。
|
||||
db_path = self.create_db(
|
||||
"single.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 100, "foo newest"),
|
||||
(2, 90, "foo middle"),
|
||||
(3, 80, "foo oldest"),
|
||||
]
|
||||
},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
):
|
||||
result = mcp_server.search_messages("foo", chat_name="Alice", limit=2, offset=1)
|
||||
|
||||
self.assertIn('在 Alice 中搜索 "foo" 找到 2 条结果（offset=1, limit=2）', result)
|
||||
self.assertLess(result.index("foo oldest"), result.index("foo middle"))
|
||||
self.assertNotIn("foo newest", result)
|
||||
|
||||
def test_search_messages_multiple_chats_applies_global_pagination(self):
|
||||
# 多个聊天联合搜索时，分页必须基于合并后的全局结果。
|
||||
db_path = self.create_db(
|
||||
"multi.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 110, "foo a1"),
|
||||
(2, 90, "foo a2"),
|
||||
],
|
||||
"bob": [
|
||||
(1, 100, "foo b1"),
|
||||
(2, 80, "foo b2"),
|
||||
],
|
||||
},
|
||||
)
|
||||
contexts = [
|
||||
{
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
},
|
||||
{
|
||||
"query": "Bob",
|
||||
"username": "bob",
|
||||
"display_name": "Bob",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("bob"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("bob")}],
|
||||
"is_group": False,
|
||||
},
|
||||
]
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice", "bob": "Bob"}), patch.object(
|
||||
mcp_server, "_resolve_chat_contexts", return_value=(contexts, [], [])
|
||||
):
|
||||
result = mcp_server.search_messages("foo", chat_name=["Alice", "Bob"], limit=2, offset=1)
|
||||
|
||||
self.assertIn('在 2 个聊天对象中搜索 "foo" 找到 2 条结果（offset=1, limit=2）', result)
|
||||
self.assertLess(result.index("foo a2"), result.index("foo b1"))
|
||||
self.assertNotIn("foo a1", result)
|
||||
self.assertNotIn("foo b2", result)
|
||||
|
||||
def test_search_messages_all_messages_merges_global_results_before_paging(self):
|
||||
# 全库搜索要基于跨库合并后的全局时间线分页，不能被单个分库提前截断。
|
||||
older_db = self.create_db(
|
||||
"older.db",
|
||||
{"older_user": [(1, 10, "foo older 1"), (2, 9, "foo older 2"), (3, 8, "foo older 3")]},
|
||||
)
|
||||
newer_db = self.create_db(
|
||||
"newer.db",
|
||||
{"newer_user": [(1, 30, "foo newer 1"), (2, 20, "foo newer 2")]},
|
||||
)
|
||||
fake_cache = _FakeCache({"older": older_db, "newer": newer_db})
|
||||
|
||||
with patch.object(mcp_server, "MSG_DB_KEYS", ["older", "newer"]), patch.object(
|
||||
mcp_server, "_cache", fake_cache
|
||||
), patch.object(
|
||||
mcp_server,
|
||||
"get_contact_names",
|
||||
return_value={"older_user": "Older", "newer_user": "Newer"},
|
||||
):
|
||||
result = mcp_server.search_messages("foo", limit=2, offset=0)
|
||||
|
||||
self.assertIn('搜索 "foo" 找到 2 条结果（offset=0, limit=2）', result)
|
||||
self.assertLess(result.index("foo newer 2"), result.index("foo newer 1"))
|
||||
self.assertNotIn("foo older 1", result)
|
||||
|
||||
def test_search_messages_all_messages_uses_bounded_sql_pagination(self):
|
||||
# 每个消息表都只应查询当前页所需的候选窗口，不能回退到 limit=None 的全量扫描。
|
||||
older_db = self.create_db(
|
||||
"older_paged.db",
|
||||
{"older_user": [(1, 10, "foo older 1"), (2, 9, "foo older 2"), (3, 8, "foo older 3")]},
|
||||
)
|
||||
newer_db = self.create_db(
|
||||
"newer_paged.db",
|
||||
{"newer_user": [(1, 30, "foo newer 1"), (2, 20, "foo newer 2"), (3, 19, "foo newer 3")]},
|
||||
)
|
||||
fake_cache = _FakeCache({"older": older_db, "newer": newer_db})
|
||||
original_query_messages = mcp_server._query_messages
|
||||
calls = []
|
||||
|
||||
def recording_query_messages(*args, **kwargs):
|
||||
calls.append((args[1], kwargs.get("limit"), kwargs.get("offset", 0)))
|
||||
return original_query_messages(*args, **kwargs)
|
||||
|
||||
with patch.object(mcp_server, "MSG_DB_KEYS", ["older", "newer"]), patch.object(
|
||||
mcp_server, "_cache", fake_cache
|
||||
), patch.object(
|
||||
mcp_server,
|
||||
"get_contact_names",
|
||||
return_value={"older_user": "Older", "newer_user": "Newer"},
|
||||
), patch.object(
|
||||
mcp_server, "_query_messages", side_effect=recording_query_messages
|
||||
):
|
||||
result = mcp_server.search_messages("foo", limit=2, offset=1)
|
||||
|
||||
self.assertIn('搜索 "foo" 找到 2 条结果（offset=1, limit=2）', result)
|
||||
self.assertEqual(
|
||||
calls,
|
||||
[
|
||||
(_msg_table_name("older_user"), 3, 0),
|
||||
(_msg_table_name("newer_user"), 3, 0),
|
||||
],
|
||||
)
|
||||
|
||||
def test_search_messages_single_chat_respects_time_range(self):
|
||||
# 单聊搜索的开始/结束时间都必须严格生效。
|
||||
db_path = self.create_db(
|
||||
"single_time.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 300, "foo in range"),
|
||||
(2, 200, "foo too early"),
|
||||
(3, 400, "foo too late"),
|
||||
]
|
||||
},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
), patch.object(
|
||||
mcp_server, "_parse_time_range", return_value=(250, 350)
|
||||
):
|
||||
result = mcp_server.search_messages(
|
||||
"foo",
|
||||
chat_name="Alice",
|
||||
start_time="custom-start",
|
||||
end_time="custom-end",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
|
||||
self.assertIn("时间范围: custom-start ~ custom-end", result)
|
||||
self.assertIn("foo in range", result)
|
||||
self.assertNotIn("foo too early", result)
|
||||
self.assertNotIn("foo too late", result)
|
||||
|
||||
def test_search_messages_multiple_chats_respects_time_range(self):
|
||||
# 多聊联合搜索时，每个聊天对象都要套用同一时间范围。
|
||||
db_path = self.create_db(
|
||||
"multi_time.db",
|
||||
{
|
||||
"alice": [(1, 300, "foo alice in range"), (2, 150, "foo alice too early")],
|
||||
"bob": [(1, 320, "foo bob in range"), (2, 500, "foo bob too late")],
|
||||
},
|
||||
)
|
||||
contexts = [
|
||||
{
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
},
|
||||
{
|
||||
"query": "Bob",
|
||||
"username": "bob",
|
||||
"display_name": "Bob",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("bob"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("bob")}],
|
||||
"is_group": False,
|
||||
},
|
||||
]
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice", "bob": "Bob"}), patch.object(
|
||||
mcp_server, "_resolve_chat_contexts", return_value=(contexts, [], [])
|
||||
), patch.object(
|
||||
mcp_server, "_parse_time_range", return_value=(250, 400)
|
||||
):
|
||||
result = mcp_server.search_messages(
|
||||
"foo",
|
||||
chat_name=["Alice", "Bob"],
|
||||
start_time="range-start",
|
||||
end_time="range-end",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
|
||||
self.assertIn("时间范围: range-start ~ range-end", result)
|
||||
self.assertIn("foo alice in range", result)
|
||||
self.assertIn("foo bob in range", result)
|
||||
self.assertNotIn("foo alice too early", result)
|
||||
self.assertNotIn("foo bob too late", result)
|
||||
|
||||
def test_search_messages_all_messages_respects_time_range(self):
|
||||
# 全库搜索也不能返回时间范围外的消息。
|
||||
db_path = self.create_db(
|
||||
"all_time.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 100, "foo too early"),
|
||||
(2, 300, "foo in range"),
|
||||
(3, 500, "foo too late"),
|
||||
]
|
||||
},
|
||||
)
|
||||
fake_cache = _FakeCache({"all": db_path})
|
||||
|
||||
with patch.object(mcp_server, "MSG_DB_KEYS", ["all"]), patch.object(
|
||||
mcp_server, "_cache", fake_cache
|
||||
), patch.object(
|
||||
mcp_server,
|
||||
"get_contact_names",
|
||||
return_value={"alice": "Alice"},
|
||||
), patch.object(
|
||||
mcp_server, "_parse_time_range", return_value=(250, 350)
|
||||
):
|
||||
result = mcp_server.search_messages(
|
||||
"foo",
|
||||
start_time="range-start",
|
||||
end_time="range-end",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
|
||||
self.assertIn("时间范围: range-start ~ range-end", result)
|
||||
self.assertIn("foo in range", result)
|
||||
self.assertNotIn("foo too early", result)
|
||||
self.assertNotIn("foo too late", result)
|
||||
|
||||
def test_get_chat_history_merges_sharded_message_tables(self):
|
||||
# 同一联系人跨多个 message_N.db 分片时，历史查询要先合并再分页。
|
||||
older_db = self.create_db("history_older.db", {"alice": [(1, 100, "old message")]})
|
||||
newer_db = self.create_db(
|
||||
"history_newer.db",
|
||||
{"alice": [(1, 300, "new message"), (2, 250, "middle message")]},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": newer_db,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [
|
||||
{"db_path": older_db, "table_name": _msg_table_name("alice")},
|
||||
{"db_path": newer_db, "table_name": _msg_table_name("alice")},
|
||||
],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=2, offset=0)
|
||||
|
||||
self.assertIn("Alice 的消息记录（返回 2 条，offset=0, limit=2）", result)
|
||||
self.assertIn("middle message", result)
|
||||
self.assertIn("new message", result)
|
||||
self.assertNotIn("old message", result)
|
||||
|
||||
def test_get_chat_history_large_limit_reads_all_rows_across_shards(self):
|
||||
# 大 limit 下，跨分片历史查询不能只返回较旧分片里的少量消息。
|
||||
older_messages = [
|
||||
(index, 1000 + index, f"old shard message {index}")
|
||||
for index in range(1, 18)
|
||||
]
|
||||
newer_messages = [
|
||||
(index, 2000 + index, f"new shard message {index}")
|
||||
for index in range(1, 296)
|
||||
]
|
||||
older_db = self.create_db("history_cross_shard_older.db", {"alice": older_messages})
|
||||
newer_db = self.create_db("history_cross_shard_newer.db", {"alice": newer_messages})
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": newer_db,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [
|
||||
{"db_path": newer_db, "table_name": _msg_table_name("alice")},
|
||||
{"db_path": older_db, "table_name": _msg_table_name("alice")},
|
||||
],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=500, offset=0)
|
||||
|
||||
self.assertIn("Alice 的消息记录（返回 312 条，offset=0, limit=500）", result)
|
||||
self.assertIn("new shard message 295", result)
|
||||
self.assertIn("old shard message 17", result)
|
||||
|
||||
body = result.split(":\n\n", 1)[1]
|
||||
self.assertEqual(len(body.splitlines()), 312)
|
||||
|
||||
def test_get_chat_history_uses_bounded_sql_pagination(self):
|
||||
# 历史查询应把 offset+limit 下推到 SQL，避免把整张消息表读出来后再切片。
|
||||
db_path = self.create_db(
|
||||
"history_paged.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 400, "newest"),
|
||||
(2, 300, "middle"),
|
||||
(3, 200, "older"),
|
||||
(4, 100, "oldest"),
|
||||
]
|
||||
},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
original_query_messages = mcp_server._query_messages
|
||||
calls = []
|
||||
|
||||
def recording_query_messages(*args, **kwargs):
|
||||
calls.append((args[1], kwargs.get("limit"), kwargs.get("offset", 0)))
|
||||
return original_query_messages(*args, **kwargs)
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
), patch.object(
|
||||
mcp_server, "_query_messages", side_effect=recording_query_messages
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=2, offset=1)
|
||||
|
||||
self.assertIn("middle", result)
|
||||
self.assertIn("older", result)
|
||||
self.assertNotIn("newest", result)
|
||||
self.assertNotIn("oldest", result)
|
||||
self.assertEqual(calls, [(_msg_table_name("alice"), 3, 0)])
|
||||
|
||||
def test_get_chat_history_allows_large_limit_values(self):
|
||||
# 历史查询不应再把大 limit 直接拒绝掉。
|
||||
db_path = self.create_db(
|
||||
"history_large_limit.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 200, "message 1"),
|
||||
(2, 100, "message 2"),
|
||||
]
|
||||
},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=999999, offset=0)
|
||||
|
||||
self.assertNotIn("错误:", result)
|
||||
self.assertIn("message 1", result)
|
||||
self.assertIn("message 2", result)
|
||||
|
||||
def test_get_chat_history_keeps_partial_results_when_formatting_fails(self):
|
||||
# 单条坏消息不应让整个历史查询失败，已有结果仍应返回并附带失败说明。
|
||||
db_path = self.create_db(
|
||||
"history_partial_failure.db",
|
||||
{"alice": [(1, 200, "good message"), (2, 100, "bad message")]},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
original_build_history_line = mcp_server._build_history_line
|
||||
|
||||
def flaky_build_history_line(row, *args, **kwargs):
|
||||
if row[2] == 100:
|
||||
raise ValueError("bad row")
|
||||
return original_build_history_line(row, *args, **kwargs)
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
), patch.object(
|
||||
mcp_server, "_build_history_line", side_effect=flaky_build_history_line
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=2, offset=0)
|
||||
|
||||
self.assertIn("good message", result)
|
||||
self.assertIn("查询失败:", result)
|
||||
self.assertIn("bad row", result)
|
||||
|
||||
def test_get_chat_history_does_not_truncate_long_messages(self):
|
||||
# 历史记录应返回完整消息内容，而不是固定截断到 500 字符。
|
||||
long_message = "x" * 600
|
||||
db_path = self.create_db(
|
||||
"history_long_message.db",
|
||||
{"alice": [(1, 200, long_message)]},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
):
|
||||
result = mcp_server.get_chat_history("Alice", limit=1, offset=0)
|
||||
|
||||
self.assertIn(long_message, result)
|
||||
self.assertNotIn(("x" * 500) + "...", result)
|
||||
|
||||
def test_search_messages_single_chat_merges_sharded_message_tables(self):
|
||||
# 单聊搜索也要跨分片合并，否则最近消息可能查不到。
|
||||
older_db = self.create_db("search_older.db", {"alice": [(1, 100, "foo old")]})
|
||||
newer_db = self.create_db(
|
||||
"search_newer.db",
|
||||
{"alice": [(1, 300, "foo new"), (2, 200, "foo middle")]},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": newer_db,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [
|
||||
{"db_path": older_db, "table_name": _msg_table_name("alice")},
|
||||
{"db_path": newer_db, "table_name": _msg_table_name("alice")},
|
||||
],
|
||||
"is_group": False,
|
||||
}
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
), patch.object(
|
||||
mcp_server, "_parse_time_range", return_value=(150, 350)
|
||||
):
|
||||
result = mcp_server.search_messages(
|
||||
"foo",
|
||||
chat_name="Alice",
|
||||
start_time="range-start",
|
||||
end_time="range-end",
|
||||
limit=20,
|
||||
offset=0,
|
||||
)
|
||||
|
||||
self.assertIn("foo middle", result)
|
||||
self.assertIn("foo new", result)
|
||||
self.assertNotIn("foo old", result)
|
||||
|
||||
def test_search_messages_keeps_partial_results_when_later_batch_fails(self):
|
||||
# 后续批次失败时，前面已经拿到的有效结果不应被丢弃。
|
||||
db_path = self.create_db(
|
||||
"search_partial_failure.db",
|
||||
{
|
||||
"alice": [
|
||||
(1, 400, "foo newest"),
|
||||
(2, 300, "foo skipped"),
|
||||
(3, 200, "foo older"),
|
||||
(4, 100, "foo bad"),
|
||||
]
|
||||
},
|
||||
)
|
||||
ctx = {
|
||||
"query": "Alice",
|
||||
"username": "alice",
|
||||
"display_name": "Alice",
|
||||
"db_path": db_path,
|
||||
"table_name": _msg_table_name("alice"),
|
||||
"message_tables": [{"db_path": db_path, "table_name": _msg_table_name("alice")}],
|
||||
"is_group": False,
|
||||
}
|
||||
original_build_search_entry = mcp_server._build_search_entry
|
||||
|
||||
def flaky_build_search_entry(row, *args, **kwargs):
|
||||
if row[2] == 300:
|
||||
return None
|
||||
if row[2] == 100:
|
||||
raise ValueError("bad row")
|
||||
return original_build_search_entry(row, *args, **kwargs)
|
||||
|
||||
with patch.object(mcp_server, "get_contact_names", return_value={"alice": "Alice"}), patch.object(
|
||||
mcp_server, "_resolve_chat_context", return_value=ctx
|
||||
), patch.object(
|
||||
mcp_server, "_build_search_entry", side_effect=flaky_build_search_entry
|
||||
):
|
||||
result = mcp_server.search_messages("foo", chat_name="Alice", limit=3, offset=0)
|
||||
|
||||
self.assertIn("foo newest", result)
|
||||
self.assertIn("foo older", result)
|
||||
self.assertIn("查询失败:", result)
|
||||
self.assertIn("bad row", result)
|
||||
|
||||
def test_get_recent_sessions_closes_connection_when_query_fails(self):
|
||||
# 会话查询抛异常时也必须关闭 sqlite3 连接，避免资源泄漏。
|
||||
fake_cache = _FakeCache({os.path.join("session", "session.db"): "session.db"})
|
||||
|
||||
class _FakeConn:
|
||||
def __init__(self):
|
||||
self.closed = False
|
||||
|
||||
def execute(self, *args, **kwargs):
|
||||
raise sqlite3.OperationalError("boom")
|
||||
|
||||
def close(self):
|
||||
self.closed = True
|
||||
|
||||
fake_conn = _FakeConn()
|
||||
|
||||
with patch.object(mcp_server, "_cache", fake_cache), patch.object(
|
||||
mcp_server, "get_contact_names", return_value={}
|
||||
), patch.object(
|
||||
mcp_server.sqlite3, "connect", return_value=fake_conn
|
||||
):
|
||||
with self.assertRaisesRegex(sqlite3.OperationalError, "boom"):
|
||||
mcp_server.get_recent_sessions()
|
||||
|
||||
self.assertTrue(fake_conn.closed)
|
||||
|
||||
def test_get_new_messages_closes_connection_when_query_fails(self):
|
||||
# 新消息轮询失败时也要释放 sqlite3 连接。
|
||||
fake_cache = _FakeCache({os.path.join("session", "session.db"): "session.db"})
|
||||
|
||||
class _FakeConn:
|
||||
def __init__(self):
|
||||
self.closed = False
|
||||
|
||||
def execute(self, *args, **kwargs):
|
||||
raise sqlite3.OperationalError("boom")
|
||||
|
||||
def close(self):
|
||||
self.closed = True
|
||||
|
||||
fake_conn = _FakeConn()
|
||||
|
||||
with patch.object(mcp_server, "_cache", fake_cache), patch.object(
|
||||
mcp_server, "get_contact_names", return_value={}
|
||||
), patch.object(
|
||||
mcp_server.sqlite3, "connect", return_value=fake_conn
|
||||
):
|
||||
with self.assertRaisesRegex(sqlite3.OperationalError, "boom"):
|
||||
mcp_server.get_new_messages()
|
||||
|
||||
self.assertTrue(fake_conn.closed)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Loading…
Reference in New Issue