fix(mattermost): address review comments

- Attachments now only attached to first chunk when message is split
- Filename sanitized with Path(name).name to prevent path traversal
- Updated streaming tests to match buffer-and-post-at-end pattern
  (iOS compatibility)
This commit is contained in:
Kenneth Zhao
2026-07-06 12:14:57 +08:00
committed by Xubin Ren
parent fff38f11a7
commit a710a7d6f7
2 changed files with 13 additions and 12 deletions
+6 -6
View File
@@ -459,11 +459,11 @@ class MattermostChannel(BaseChannel):
if msg.content or file_ids:
text = msg.content or " "
chunks = split_message(text, MATTERMOST_MAX_MESSAGE_LEN)
for chunk in chunks:
for i, chunk in enumerate(chunks):
await self._create_post(
chat_id, chunk,
root_id=root_id if self.config.reply_in_thread else None,
file_ids=file_ids or None,
file_ids=(file_ids if i == 0 else None) or None,
)
if not meta.get("_progress") and meta.get("message_id"):
@@ -616,10 +616,10 @@ class MattermostChannel(BaseChannel):
async def _download_file(self, file_id: str) -> str | None:
try:
resp = await self._http_client.get(f"/api/v4/files/{file_id}")
resp.raise_for_status()
info = resp.json()
name = info.get("name", file_id)
info_resp = await self._http_client.get(f"/api/v4/files/{file_id}/info")
info_resp.raise_for_status()
info = info_resp.json()
name = Path(info.get("name", file_id)).name
out = Path(get_media_dir("mattermost")) / f"{file_id}_{name}"
out.parent.mkdir(parents=True, exist_ok=True)
+7 -6
View File
@@ -492,9 +492,9 @@ async def test_stream_first_delta_creates_post():
await channel.send_delta("chan_1", "Hello", {"_stream_id": "s1"})
posts = [c for c in fake.post_calls if c["path"] == "/api/v4/posts"]
assert len(posts) == 1
assert "Hello" in posts[0]["json"]["message"]
assert channel._stream_posts["s1"] == "stream_post_1"
assert len(posts) == 0
assert channel._stream_buffers["s1"] == "Hello"
assert channel._stream_committed["s1"] == "Hello"
@pytest.mark.asyncio
@@ -508,8 +508,8 @@ async def test_stream_subsequent_delta_edits_post():
await channel.send_delta("chan_1", " world", {"_stream_id": "s1"})
edits = [c for c in fake.put_calls if c["path"] == "/api/v4/posts/stream_post_1"]
assert len(edits) == 1
assert edits[0]["json"]["message"] == "Hello world"
assert len(edits) == 0
assert channel._stream_buffers["s1"] == "Hello world"
@pytest.mark.asyncio
@@ -535,7 +535,8 @@ async def test_stream_chunk_boundary_finalizes_and_creates_new():
await channel.send_delta("chan_1", "world", {"_stream_id": "s1"})
posts = [c for c in fake.post_calls if c["path"] == "/api/v4/posts"]
assert len(posts) == 2
assert len(posts) == 0
assert channel._stream_buffers["s1"] == "Hello world"
# ---------------------------------------------------------------------------