feat: add message download path selection

tests/test_media_interaction.py

@@ -3,10 +3,11 @@ from unittest.mock import patch
 import pytest
 
 from app.chain.message import MediaInteractionChain, MessageChain
-from app.core.context import MediaInfo
+from app.core.context import Context, MediaInfo, TorrentInfo
 from app.core.meta import MetaBase
 from app.helper.interaction import media_interaction_manager
-from app.schemas.types import MessageChannel
+from app.schemas import TransferDirectoryConf
+from app.schemas.types import MediaType, MessageChannel
 
 
 @pytest.fixture(autouse=True)
@@ -24,6 +25,43 @@ def _build_meta(name: str) -> MetaBase:
     return meta
 
 
+def _build_context(title: str = "星际穿越") -> Context:
+    """构造可用于媒体交互下载测试的资源上下文。"""
+    return Context(
+        meta_info=_build_meta(title),
+        media_info=MediaInfo(
+            type=MediaType.MOVIE,
+            title=title,
+            year="2014",
+            tmdb_id=1,
+        ),
+        torrent_info=TorrentInfo(
+            title=f"{title}.2014.1080p",
+            site_name="TestSite",
+            enclosure="https://example.com/demo.torrent",
+            seeders=10,
+        ),
+    )
+
+
+def _build_download_dirs() -> list[TransferDirectoryConf]:
+    """构造消息交互可选择的下载目录配置。"""
+    return [
+        TransferDirectoryConf(
+            name="电影下载",
+            storage="local",
+            download_path="/downloads/movies",
+            priority=1,
+        ),
+        TransferDirectoryConf(
+            name="动画下载",
+            storage="rclone",
+            download_path="/media/anime",
+            priority=2,
+        ),
+    ]
+
+
 def test_message_routes_text_reply_to_media_interaction_before_ai():
     """已有传统媒体交互时，用户回复应优先交给传统交互处理。"""
     chain = MessageChain()
@@ -230,3 +268,164 @@ def test_media_interaction_legacy_page_callback_updates_existing_request():
     notification = post_medias_message.call_args.args[0]
     assert notification.original_message_id == 123
     assert notification.original_chat_id == "456"
+
+
+def test_torrent_selection_prompts_download_dir_buttons_before_download():
+    """支持按钮的渠道选择资源后，应先发送下载目录按钮而不是立即下载。"""
+    chain = MediaInteractionChain()
+    context = _build_context()
+    request = media_interaction_manager.create_or_replace(
+        user_id="10001",
+        channel=MessageChannel.Telegram,
+        source="telegram-test",
+        username="tester",
+        action="Search",
+        keyword="星际穿越",
+        title="星际穿越",
+        meta=_build_meta("星际穿越"),
+        items=[context],
+    )
+    request.phase = "torrent"
+
+    with patch(
+        "app.chain.message.DirectoryHelper.get_download_dirs",
+        return_value=_build_download_dirs(),
+    ), patch.object(chain, "post_message") as post_message, patch(
+        "app.chain.message.DownloadChain.download_single"
+    ) as download_single:
+        handled = chain.handle_text_interaction(
+            channel=MessageChannel.Telegram,
+            source="telegram-test",
+            userid="10001",
+            username="tester",
+            text="1",
+        )
+
+    assert handled
+    download_single.assert_not_called()
+    assert request.phase == "download-dir"
+    post_message.assert_called_once()
+    notification = post_message.call_args.args[0]
+    assert "请选择下载目录" in notification.title
+    assert "电影下载 (/downloads/movies)" in notification.text
+    assert notification.buttons[0][0]["callback_data"] == f"media:{request.request_id}:download-dir:1"
+
+
+def test_torrent_selection_prompts_text_download_dir_for_plain_channel():
+    """不支持按钮的渠道选择资源后，应提示用户回复数字选择下载目录。"""
+    chain = MediaInteractionChain()
+    context = _build_context()
+    request = media_interaction_manager.create_or_replace(
+        user_id="wechat-user",
+        channel=MessageChannel.Wechat,
+        source="wechat-test",
+        username="tester",
+        action="Search",
+        keyword="星际穿越",
+        title="星际穿越",
+        meta=_build_meta("星际穿越"),
+        items=[context],
+    )
+    request.phase = "torrent"
+
+    with patch(
+        "app.chain.message.DirectoryHelper.get_download_dirs",
+        return_value=_build_download_dirs(),
+    ), patch.object(chain, "post_message") as post_message:
+        handled = chain.handle_text_interaction(
+            channel=MessageChannel.Wechat,
+            source="wechat-test",
+            userid="wechat-user",
+            username="tester",
+            text="1",
+        )
+
+    assert handled
+    notification = post_message.call_args.args[0]
+    assert "请回复对应数字" in notification.title
+    assert notification.buttons is None
+    assert "2. 动画下载 (rclone:/media/anime)" in notification.text
+
+
+def test_download_dir_callback_runs_pending_single_download_with_save_path():
+    """下载目录按钮回调应使用所选 save_path 继续执行挂起的单资源下载。"""
+    chain = MediaInteractionChain()
+    context = _build_context()
+    request = media_interaction_manager.create_or_replace(
+        user_id="10001",
+        channel=MessageChannel.Telegram,
+        source="telegram-test",
+        username="tester",
+        action="Search",
+        keyword="星际穿越",
+        title="星际穿越",
+        meta=_build_meta("星际穿越"),
+        items=[context],
+    )
+    request.phase = "download-dir"
+    request.pending_download_mode = "single"
+    request.pending_download_context = context
+
+    with patch(
+        "app.chain.message.DirectoryHelper.get_download_dirs",
+        return_value=_build_download_dirs(),
+    ), patch(
+        "app.chain.message.DownloadChain.download_single",
+        return_value="hash",
+    ) as download_single:
+        request.download_dirs = chain._get_download_dirs()
+        handled = chain.handle_callback_interaction(
+            callback_data=f"media:{request.request_id}:download-dir:2",
+            channel=MessageChannel.Telegram,
+            source="telegram-test",
+            userid="10001",
+            username="tester",
+        )
+
+    assert handled
+    assert request.phase == "torrent"
+    download_single.assert_called_once()
+    assert download_single.call_args.args[0] is context
+    assert download_single.call_args.kwargs["save_path"] == "rclone:/media/anime"
+
+
+def test_download_dir_text_reply_runs_pending_single_download_with_save_path():
+    """下载目录文本回复应使用所选 save_path 继续执行挂起的单资源下载。"""
+    chain = MediaInteractionChain()
+    context = _build_context()
+    request = media_interaction_manager.create_or_replace(
+        user_id="wechat-user",
+        channel=MessageChannel.Wechat,
+        source="wechat-test",
+        username="tester",
+        action="Search",
+        keyword="星际穿越",
+        title="星际穿越",
+        meta=_build_meta("星际穿越"),
+        items=[context],
+    )
+    request.phase = "download-dir"
+    request.pending_download_mode = "single"
+    request.pending_download_context = context
+
+    with patch(
+        "app.chain.message.DirectoryHelper.get_download_dirs",
+        return_value=_build_download_dirs(),
+    ), patch(
+        "app.chain.message.DownloadChain.download_single",
+        return_value="hash",
+    ) as download_single:
+        request.download_dirs = chain._get_download_dirs()
+        handled = chain.handle_text_interaction(
+            channel=MessageChannel.Wechat,
+            source="wechat-test",
+            userid="wechat-user",
+            username="tester",
+            text="1",
+        )
+
+    assert handled
+    assert request.phase == "torrent"
+    download_single.assert_called_once()
+    assert download_single.call_args.args[0] is context
+    assert download_single.call_args.kwargs["save_path"] == "/downloads/movies"