mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-08 14:23:27 +08:00
1364 lines
44 KiB
Python
1364 lines
44 KiB
Python
import math
|
||
import re
|
||
import uuid
|
||
from dataclasses import dataclass, field
|
||
from datetime import datetime, timedelta
|
||
from threading import Lock
|
||
from typing import Any, Dict, List, Optional, Tuple, Union
|
||
|
||
from app.chain import ChainBase
|
||
from app.chain.download import DownloadChain
|
||
from app.chain.media import MediaChain
|
||
from app.chain.search import SearchChain
|
||
from app.chain.subscribe import SubscribeChain
|
||
from app.core.config import settings
|
||
from app.core.context import Context, MediaInfo
|
||
from app.core.meta import MetaBase
|
||
from app.db.user_oper import UserOper
|
||
from app.helper.torrent import TorrentHelper
|
||
from app.log import logger
|
||
from app.schemas import Notification, NotExistMediaInfo
|
||
from app.schemas.message import ChannelCapabilityManager
|
||
from app.schemas.types import MediaType, MessageChannel
|
||
from app.utils.string import StringUtils
|
||
|
||
|
||
@dataclass(frozen=True)
|
||
class AgentInteractionOption:
|
||
"""
|
||
Agent 交互选项。
|
||
"""
|
||
|
||
label: str
|
||
value: str
|
||
|
||
|
||
@dataclass
|
||
class PendingAgentInteraction:
|
||
"""
|
||
待处理的 Agent 客户端交互请求。
|
||
"""
|
||
|
||
request_id: str
|
||
session_id: str
|
||
user_id: str
|
||
channel: Optional[str]
|
||
source: Optional[str]
|
||
username: Optional[str]
|
||
title: Optional[str]
|
||
prompt: str
|
||
options: List[AgentInteractionOption]
|
||
created_at: datetime = field(default_factory=datetime.now)
|
||
|
||
|
||
class AgentInteractionManager:
|
||
"""
|
||
管理 Agent 发起的客户端交互请求。
|
||
"""
|
||
|
||
_ttl = timedelta(hours=24)
|
||
|
||
def __init__(self):
|
||
self._pending_interactions: Dict[str, PendingAgentInteraction] = {}
|
||
self._lock = Lock()
|
||
|
||
def _cleanup_locked(self) -> None:
|
||
expire_before = datetime.now() - self._ttl
|
||
expired_ids = [
|
||
request_id
|
||
for request_id, request in self._pending_interactions.items()
|
||
if request.created_at < expire_before
|
||
]
|
||
for request_id in expired_ids:
|
||
self._pending_interactions.pop(request_id, None)
|
||
|
||
def create_request(
|
||
self,
|
||
session_id: str,
|
||
user_id: str,
|
||
channel: Optional[str],
|
||
source: Optional[str],
|
||
username: Optional[str],
|
||
title: Optional[str],
|
||
prompt: str,
|
||
options: List[AgentInteractionOption],
|
||
) -> PendingAgentInteraction:
|
||
"""
|
||
创建一条待用户确认的 Agent 交互请求。
|
||
"""
|
||
with self._lock:
|
||
self._cleanup_locked()
|
||
request_id = uuid.uuid4().hex[:12]
|
||
while request_id in self._pending_interactions:
|
||
request_id = uuid.uuid4().hex[:12]
|
||
request = PendingAgentInteraction(
|
||
request_id=request_id,
|
||
session_id=session_id,
|
||
user_id=str(user_id),
|
||
channel=channel,
|
||
source=source,
|
||
username=username,
|
||
title=title,
|
||
prompt=prompt,
|
||
options=options,
|
||
)
|
||
self._pending_interactions[request_id] = request
|
||
return request
|
||
|
||
def resolve(
|
||
self,
|
||
request_id: str,
|
||
option_index: int,
|
||
user_id: Optional[str] = None,
|
||
) -> Optional[tuple[PendingAgentInteraction, AgentInteractionOption]]:
|
||
"""
|
||
消费一条 Agent 交互请求，并返回选中的选项。
|
||
"""
|
||
with self._lock:
|
||
self._cleanup_locked()
|
||
request = self._pending_interactions.get(request_id)
|
||
if not request:
|
||
return None
|
||
if user_id is not None and str(request.user_id) != str(user_id):
|
||
return None
|
||
if option_index < 1 or option_index > len(request.options):
|
||
return None
|
||
option = request.options[option_index - 1]
|
||
self._pending_interactions.pop(request_id, None)
|
||
return request, option
|
||
|
||
def clear(self) -> None:
|
||
"""
|
||
清空所有 Agent 交互请求。
|
||
"""
|
||
with self._lock:
|
||
self._pending_interactions.clear()
|
||
|
||
|
||
agent_interaction_manager = AgentInteractionManager()
|
||
|
||
|
||
@dataclass
|
||
class PendingMediaInteraction:
|
||
"""
|
||
记录一次搜索/下载/订阅交互的当前上下文。
|
||
"""
|
||
|
||
request_id: str
|
||
user_id: str
|
||
channel: Optional[MessageChannel]
|
||
source: Optional[str]
|
||
username: Optional[str]
|
||
action: str
|
||
keyword: str
|
||
phase: str = "media"
|
||
page: int = 0
|
||
title: str = ""
|
||
meta: Optional[MetaBase] = None
|
||
current_media: Optional[MediaInfo] = None
|
||
items: List[Any] = field(default_factory=list)
|
||
created_at: datetime = field(default_factory=datetime.now)
|
||
|
||
|
||
class MediaInteractionManager:
|
||
"""
|
||
管理用户当前激活的媒体交互状态。
|
||
|
||
每个用户只保留一个有效会话，避免旧按钮与新一轮搜索混用。
|
||
"""
|
||
|
||
_ttl = timedelta(hours=24)
|
||
|
||
def __init__(self):
|
||
self._by_id: Dict[str, PendingMediaInteraction] = {}
|
||
self._by_user: Dict[str, str] = {}
|
||
self._lock = Lock()
|
||
|
||
def _cleanup_locked(self) -> None:
|
||
"""
|
||
清理超时会话，避免内存中残留旧交互状态。
|
||
"""
|
||
expire_before = datetime.now() - self._ttl
|
||
expired = [
|
||
request_id
|
||
for request_id, request in self._by_id.items()
|
||
if request.created_at < expire_before
|
||
]
|
||
for request_id in expired:
|
||
request = self._by_id.pop(request_id, None)
|
||
if request:
|
||
self._by_user.pop(str(request.user_id), None)
|
||
|
||
def create_or_replace(
|
||
self,
|
||
user_id: Union[str, int],
|
||
channel: Optional[MessageChannel],
|
||
source: Optional[str],
|
||
username: Optional[str],
|
||
action: str,
|
||
keyword: str,
|
||
title: str = "",
|
||
meta: Optional[MetaBase] = None,
|
||
items: Optional[List[Any]] = None,
|
||
) -> PendingMediaInteraction:
|
||
"""
|
||
为用户创建新的交互状态，并替换旧会话。
|
||
"""
|
||
with self._lock:
|
||
self._cleanup_locked()
|
||
user_key = str(user_id)
|
||
old_request_id = self._by_user.get(user_key)
|
||
if old_request_id:
|
||
self._by_id.pop(old_request_id, None)
|
||
|
||
request = PendingMediaInteraction(
|
||
request_id=uuid.uuid4().hex[:12],
|
||
user_id=user_key,
|
||
channel=channel,
|
||
source=source,
|
||
username=username,
|
||
action=action,
|
||
keyword=keyword,
|
||
title=title,
|
||
meta=meta,
|
||
items=list(items or []),
|
||
)
|
||
self._by_id[request.request_id] = request
|
||
self._by_user[user_key] = request.request_id
|
||
return request
|
||
|
||
def get_by_user(
|
||
self, user_id: Union[str, int]
|
||
) -> Optional[PendingMediaInteraction]:
|
||
"""
|
||
按用户读取当前会话，供文本回复和旧按钮兼容使用。
|
||
"""
|
||
with self._lock:
|
||
self._cleanup_locked()
|
||
request_id = self._by_user.get(str(user_id))
|
||
if not request_id:
|
||
return None
|
||
return self._by_id.get(request_id)
|
||
|
||
def get_by_id(
|
||
self, request_id: str, user_id: Union[str, int]
|
||
) -> Optional[PendingMediaInteraction]:
|
||
"""
|
||
按请求 ID 读取会话，并校验用户归属。
|
||
"""
|
||
with self._lock:
|
||
self._cleanup_locked()
|
||
request = self._by_id.get(request_id)
|
||
if not request or str(request.user_id) != str(user_id):
|
||
return None
|
||
return request
|
||
|
||
def remove(self, request_id: str) -> None:
|
||
"""
|
||
主动结束一条会话。
|
||
"""
|
||
with self._lock:
|
||
request = self._by_id.pop(request_id, None)
|
||
if request:
|
||
self._by_user.pop(str(request.user_id), None)
|
||
|
||
def clear(self) -> None:
|
||
"""
|
||
清空所有交互状态，主要用于测试。
|
||
"""
|
||
with self._lock:
|
||
self._by_id.clear()
|
||
self._by_user.clear()
|
||
|
||
|
||
media_interaction_manager = MediaInteractionManager()
|
||
|
||
|
||
class MediaInteractionChain(ChainBase):
|
||
"""
|
||
处理媒体搜索、订阅、资源选择和翻页等交互流程。
|
||
"""
|
||
|
||
_button_page_size = 8
|
||
_text_page_size = 8
|
||
|
||
@staticmethod
|
||
def has_pending_interaction(user_id: Union[str, int]) -> bool:
|
||
"""
|
||
判断用户当前是否存在未结束的媒体交互。
|
||
"""
|
||
return media_interaction_manager.get_by_user(user_id) is not None
|
||
|
||
@staticmethod
|
||
def _get_noexits_info(
|
||
meta: MetaBase, mediainfo: MediaInfo
|
||
) -> Dict[Union[int, str], Dict[int, NotExistMediaInfo]]:
|
||
"""
|
||
构造媒体缺失集信息，用于全量重搜或自动下载补全集数。
|
||
"""
|
||
if mediainfo.type == MediaType.TV:
|
||
if not mediainfo.seasons:
|
||
mediainfo = MediaChain().recognize_media(
|
||
mtype=mediainfo.type,
|
||
tmdbid=mediainfo.tmdb_id,
|
||
doubanid=mediainfo.douban_id,
|
||
cache=False,
|
||
)
|
||
if not mediainfo:
|
||
logger.warn("媒体信息识别失败，无法补充季集信息")
|
||
return {}
|
||
if not mediainfo.seasons:
|
||
logger.warn(
|
||
"媒体信息中没有季集信息，标题：%s，tmdbid：%s，doubanid：%s",
|
||
mediainfo.title,
|
||
mediainfo.tmdb_id,
|
||
mediainfo.douban_id,
|
||
)
|
||
return {}
|
||
|
||
mediakey = mediainfo.tmdb_id or mediainfo.douban_id
|
||
no_exists = {mediakey: {}}
|
||
if meta.begin_season:
|
||
episodes = mediainfo.seasons.get(meta.begin_season)
|
||
if not episodes:
|
||
return {}
|
||
no_exists[mediakey][meta.begin_season] = NotExistMediaInfo(
|
||
season=meta.begin_season,
|
||
episodes=[],
|
||
total_episode=len(episodes),
|
||
start_episode=episodes[0],
|
||
)
|
||
else:
|
||
for sea, eps in mediainfo.seasons.items():
|
||
if not eps:
|
||
continue
|
||
no_exists[mediakey][sea] = NotExistMediaInfo(
|
||
season=sea,
|
||
episodes=[],
|
||
total_episode=len(eps),
|
||
start_episode=eps[0],
|
||
)
|
||
return no_exists
|
||
return {}
|
||
|
||
@staticmethod
|
||
def parse_callback(
|
||
callback_data: str,
|
||
) -> Optional[Tuple[Optional[str], str, Optional[int]]]:
|
||
"""
|
||
解析新旧两种媒体交互按钮格式。
|
||
"""
|
||
if callback_data.startswith("media:"):
|
||
parts = callback_data.split(":")
|
||
if len(parts) < 3:
|
||
return None
|
||
request_id = parts[1]
|
||
action = parts[2]
|
||
index = None
|
||
if len(parts) >= 4 and parts[3].isdigit():
|
||
index = int(parts[3])
|
||
return request_id, action, index
|
||
|
||
match = re.match(r"^(select|download)_(\d+)$", callback_data)
|
||
if match:
|
||
return None, match.group(1), int(match.group(2))
|
||
if callback_data == "page_p":
|
||
return None, "page-prev", None
|
||
if callback_data == "page_n":
|
||
return None, "page-next", None
|
||
return None
|
||
|
||
def handle_callback_interaction(
|
||
self,
|
||
callback_data: str,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> bool:
|
||
"""
|
||
处理按钮回调，并将当前视图刷新到原消息上。
|
||
"""
|
||
parsed = self.parse_callback(callback_data)
|
||
if not parsed:
|
||
return False
|
||
|
||
request_id, action, index = parsed
|
||
if request_id:
|
||
request = media_interaction_manager.get_by_id(request_id, userid)
|
||
else:
|
||
request = media_interaction_manager.get_by_user(userid)
|
||
|
||
if not request:
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="交互已失效，请重新搜索或订阅",
|
||
)
|
||
)
|
||
return True
|
||
|
||
request.channel = channel
|
||
request.source = source
|
||
request.username = username
|
||
|
||
if action == "page-prev":
|
||
if request.page <= 0:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="已经是第一页了！",
|
||
)
|
||
return True
|
||
request.page -= 1
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
return True
|
||
|
||
if action == "page-next":
|
||
if not self._has_next_page(request):
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="已经是最后一页了！",
|
||
)
|
||
return True
|
||
request.page += 1
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
return True
|
||
|
||
if action == "select":
|
||
self._handle_media_selection(
|
||
request=request,
|
||
page_index=index,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
return True
|
||
|
||
if action == "download":
|
||
self._handle_torrent_selection(
|
||
request=request,
|
||
page_index=index,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
|
||
return False
|
||
|
||
def handle_text_interaction(
|
||
self,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
text: str,
|
||
) -> bool:
|
||
"""
|
||
处理文本式交互。
|
||
|
||
有会话时优先处理数字选择和翻页；无会话时负责识别搜索/订阅类入口。
|
||
"""
|
||
request = media_interaction_manager.get_by_user(userid)
|
||
normalized = (text or "").strip()
|
||
lowered = normalized.lower()
|
||
|
||
if request and lowered in {"退出", "关闭", "q", "quit", "exit"}:
|
||
media_interaction_manager.remove(request.request_id)
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="媒体交互已结束",
|
||
)
|
||
)
|
||
return True
|
||
|
||
if normalized.isdigit():
|
||
if not request:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
request.channel = channel
|
||
request.source = source
|
||
request.username = username
|
||
index = int(normalized)
|
||
if request.phase == "torrent":
|
||
self._handle_torrent_selection(
|
||
request=request,
|
||
page_index=index,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
else:
|
||
self._handle_media_selection(
|
||
request=request,
|
||
page_index=index,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
|
||
if lowered in {"p", "prev", "上一页"}:
|
||
if not request:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
if request.page <= 0:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="已经是第一页了！",
|
||
)
|
||
return True
|
||
request.page -= 1
|
||
request.channel = channel
|
||
request.source = source
|
||
request.username = username
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
)
|
||
return True
|
||
|
||
if lowered in {"n", "next", "下一页"}:
|
||
if not request:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
if not self._has_next_page(request):
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="已经是最后一页了！",
|
||
)
|
||
return True
|
||
request.page += 1
|
||
request.channel = channel
|
||
request.source = source
|
||
request.username = username
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
)
|
||
return True
|
||
|
||
action, content = self._resolve_action(normalized)
|
||
if not action:
|
||
return False
|
||
|
||
self._start_media_interaction(
|
||
action=action,
|
||
content=content,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return True
|
||
|
||
@staticmethod
|
||
def _resolve_action(text: str) -> Tuple[Optional[str], str]:
|
||
"""
|
||
将用户输入归类为搜索、订阅或普通聊天。
|
||
"""
|
||
if text.startswith("订阅"):
|
||
return "Subscribe", re.sub(r"订阅[:：\s]*", "", text)
|
||
if text.startswith("洗版"):
|
||
return "ReSubscribe", re.sub(r"洗版[:：\s]*", "", text)
|
||
if text.startswith("搜索") or text.startswith("下载"):
|
||
return "ReSearch", re.sub(r"(搜索|下载)[:：\s]*", "", text)
|
||
if StringUtils.is_link(text):
|
||
return None, text
|
||
if not StringUtils.is_media_title_like(text):
|
||
return None, text
|
||
return "Search", text
|
||
|
||
def _start_media_interaction(
|
||
self,
|
||
action: str,
|
||
content: str,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
) -> None:
|
||
"""
|
||
根据用户输入搜索媒体，并进入媒体选择阶段。
|
||
"""
|
||
meta, medias = MediaChain().search(content)
|
||
if not meta.name:
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title="无法识别输入内容！",
|
||
)
|
||
return
|
||
if not medias:
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"{meta.name} 没有找到对应的媒体信息！",
|
||
)
|
||
)
|
||
return
|
||
|
||
logger.info("搜索到 %s 条相关媒体信息", len(medias))
|
||
request = media_interaction_manager.create_or_replace(
|
||
user_id=userid,
|
||
channel=channel,
|
||
source=source,
|
||
username=username,
|
||
action=action,
|
||
keyword=content,
|
||
title=meta.name,
|
||
meta=meta,
|
||
items=medias,
|
||
)
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
)
|
||
|
||
def _handle_media_selection(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
page_index: Optional[int],
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> None:
|
||
"""
|
||
处理媒体选择阶段的序号输入。
|
||
"""
|
||
page_items, page, _ = self._page_items(
|
||
items=request.items,
|
||
page=request.page,
|
||
page_size=self._page_size(request.channel),
|
||
)
|
||
request.page = page
|
||
if not page_index or page_index < 1 or page_index > len(page_items):
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return
|
||
|
||
mediainfo: MediaInfo = page_items[page_index - 1]
|
||
request.current_media = mediainfo
|
||
|
||
if request.action in {"Search", "ReSearch"}:
|
||
self._search_media_resources(
|
||
request=request,
|
||
mediainfo=mediainfo,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
return
|
||
|
||
if request.action in {"Subscribe", "ReSubscribe"}:
|
||
self._subscribe_media(
|
||
request=request,
|
||
mediainfo=mediainfo,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
|
||
def _search_media_resources(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
mediainfo: MediaInfo,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> None:
|
||
"""
|
||
根据已选媒体搜索资源，并切换到资源选择阶段。
|
||
"""
|
||
exist_flag, no_exists = DownloadChain().get_no_exists_info(
|
||
meta=request.meta,
|
||
mediainfo=mediainfo,
|
||
)
|
||
if exist_flag and request.action == "Search":
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"【{mediainfo.title_year}{request.meta.sea} 媒体库中已存在，如需重新下载请发送：搜索 名称 或 下载 名称】",
|
||
)
|
||
)
|
||
return
|
||
if exist_flag:
|
||
no_exists = self._get_noexits_info(request.meta, mediainfo)
|
||
|
||
messages = self._build_no_exists_messages(
|
||
mediainfo=mediainfo,
|
||
no_exists=no_exists,
|
||
show_missing_only=request.action == "Search",
|
||
)
|
||
if messages:
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"{mediainfo.title_year}：\n" + "\n".join(messages),
|
||
)
|
||
)
|
||
|
||
logger.info("开始搜索 %s ...", mediainfo.title_year)
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"开始搜索 {mediainfo.type.value} {mediainfo.title_year} ...",
|
||
)
|
||
)
|
||
|
||
contexts = SearchChain().process(mediainfo=mediainfo, no_exists=no_exists)
|
||
if not contexts:
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"{mediainfo.title}{request.meta.sea} 未搜索到需要的资源！",
|
||
)
|
||
)
|
||
return
|
||
|
||
contexts = TorrentHelper().sort_torrents(contexts)
|
||
if self._should_auto_download(userid):
|
||
logger.info("用户 %s 在自动下载用户中，开始自动择优下载 ...", userid)
|
||
self._auto_download(
|
||
request=request,
|
||
cache_list=contexts,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
no_exists=no_exists,
|
||
)
|
||
return
|
||
|
||
request.phase = "torrent"
|
||
request.page = 0
|
||
request.title = mediainfo.title
|
||
request.items = list(contexts)
|
||
self._render_interaction(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
|
||
def _subscribe_media(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
mediainfo: MediaInfo,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
) -> None:
|
||
"""
|
||
根据已选媒体创建订阅或洗版订阅。
|
||
"""
|
||
best_version = request.action == "ReSubscribe"
|
||
if not best_version:
|
||
exist_flag, _ = DownloadChain().get_no_exists_info(
|
||
meta=request.meta,
|
||
mediainfo=mediainfo,
|
||
)
|
||
if exist_flag:
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=f"【{mediainfo.title_year}{request.meta.sea} 媒体库中已存在，如需洗版请发送：洗版 XXX】",
|
||
)
|
||
)
|
||
return
|
||
|
||
mp_name = (
|
||
UserOper().get_name(**{f"{channel.name.lower()}_userid": userid})
|
||
if channel
|
||
else None
|
||
)
|
||
SubscribeChain().add(
|
||
title=mediainfo.title,
|
||
year=mediainfo.year,
|
||
mtype=mediainfo.type,
|
||
tmdbid=mediainfo.tmdb_id,
|
||
season=request.meta.begin_season,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=mp_name or username,
|
||
best_version=best_version,
|
||
)
|
||
|
||
def _handle_torrent_selection(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
page_index: Optional[int],
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
) -> None:
|
||
"""
|
||
处理资源选择阶段的下载操作。
|
||
"""
|
||
if request.phase != "torrent":
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return
|
||
|
||
if page_index == 0:
|
||
self._auto_download(
|
||
request=request,
|
||
cache_list=request.items,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return
|
||
|
||
page_items, page, _ = self._page_items(
|
||
items=request.items,
|
||
page=request.page,
|
||
page_size=self._page_size(request.channel),
|
||
)
|
||
request.page = page
|
||
if not page_index or page_index < 1 or page_index > len(page_items):
|
||
self._post_invalid_input(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
return
|
||
|
||
context: Context = page_items[page_index - 1]
|
||
DownloadChain().download_single(
|
||
context,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
|
||
def _auto_download(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
cache_list: List[Context],
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: str,
|
||
no_exists: Optional[Dict[Union[int, str], Dict[int, NotExistMediaInfo]]] = None,
|
||
) -> None:
|
||
"""
|
||
自动择优下载当前资源列表，并在未完成时补建订阅。
|
||
"""
|
||
downloadchain = DownloadChain()
|
||
if no_exists is None:
|
||
exist_flag, no_exists = downloadchain.get_no_exists_info(
|
||
meta=request.meta,
|
||
mediainfo=request.current_media,
|
||
)
|
||
if exist_flag:
|
||
no_exists = self._get_noexits_info(request.meta, request.current_media)
|
||
|
||
downloads, lefts = downloadchain.batch_download(
|
||
contexts=cache_list,
|
||
no_exists=no_exists,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
)
|
||
if downloads and not lefts:
|
||
logger.info("%s 下载完成", request.current_media.title_year)
|
||
return
|
||
|
||
logger.info("%s 未下载未完整，添加订阅 ...", request.current_media.title_year)
|
||
if downloads and request.current_media.type == MediaType.TV:
|
||
note = [
|
||
download.meta_info.begin_episode
|
||
for download in downloads
|
||
if download.meta_info.begin_episode
|
||
]
|
||
else:
|
||
note = None
|
||
|
||
mp_name = (
|
||
UserOper().get_name(**{f"{channel.name.lower()}_userid": userid})
|
||
if channel
|
||
else None
|
||
)
|
||
SubscribeChain().add(
|
||
title=request.current_media.title,
|
||
year=request.current_media.year,
|
||
mtype=request.current_media.type,
|
||
tmdbid=request.current_media.tmdb_id,
|
||
season=request.meta.begin_season,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=mp_name or username,
|
||
state="R",
|
||
note=note,
|
||
)
|
||
|
||
def _render_interaction(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> None:
|
||
"""
|
||
按当前阶段渲染媒体列表或资源列表。
|
||
"""
|
||
if request.phase == "torrent":
|
||
self._post_torrents_message(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
else:
|
||
self._post_medias_message(
|
||
request=request,
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
)
|
||
|
||
def _post_medias_message(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> None:
|
||
"""
|
||
发送或更新媒体选择列表。
|
||
"""
|
||
page_items, page, total_pages = self._page_items(
|
||
items=request.items,
|
||
page=request.page,
|
||
page_size=self._page_size(channel),
|
||
)
|
||
request.page = page
|
||
total = len(request.items)
|
||
if self._supports_interactive_buttons(channel):
|
||
title = f"【{request.title}】共找到{total}条相关信息，请选择操作"
|
||
buttons = self._create_media_buttons(
|
||
channel=channel,
|
||
request=request,
|
||
items=page_items,
|
||
total=total,
|
||
total_pages=total_pages,
|
||
)
|
||
else:
|
||
if total > self._page_size(channel):
|
||
title = f"【{request.title}】共找到{total}条相关信息，请回复对应数字选择（p: 上一页 n: 下一页）"
|
||
else:
|
||
title = f"【{request.title}】共找到{total}条相关信息，请回复对应数字选择"
|
||
buttons = None
|
||
|
||
self.post_medias_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
title=title,
|
||
userid=userid,
|
||
buttons=buttons,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
),
|
||
medias=page_items,
|
||
)
|
||
|
||
def _post_torrents_message(
|
||
self,
|
||
request: PendingMediaInteraction,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
original_message_id: Optional[Union[str, int]] = None,
|
||
original_chat_id: Optional[str] = None,
|
||
) -> None:
|
||
"""
|
||
发送或更新资源选择列表。
|
||
"""
|
||
page_items, page, total_pages = self._page_items(
|
||
items=request.items,
|
||
page=request.page,
|
||
page_size=self._page_size(channel),
|
||
)
|
||
request.page = page
|
||
total = len(request.items)
|
||
if self._supports_interactive_buttons(channel):
|
||
title = f"【{request.title}】共找到{total}条相关资源，请选择下载"
|
||
buttons = self._create_torrent_buttons(
|
||
channel=channel,
|
||
request=request,
|
||
items=page_items,
|
||
total=total,
|
||
total_pages=total_pages,
|
||
)
|
||
else:
|
||
if total > self._page_size(channel):
|
||
title = f"【{request.title}】共找到{total}条相关资源，请回复对应数字下载（0: 自动选择 p: 上一页 n: 下一页）"
|
||
else:
|
||
title = f"【{request.title}】共找到{total}条相关资源，请回复对应数字下载（0: 自动选择）"
|
||
buttons = None
|
||
|
||
self.post_torrents_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
title=title,
|
||
userid=userid,
|
||
link=settings.MP_DOMAIN("#/resource"),
|
||
buttons=buttons,
|
||
original_message_id=original_message_id,
|
||
original_chat_id=original_chat_id,
|
||
),
|
||
torrents=page_items,
|
||
)
|
||
|
||
def _create_media_buttons(
|
||
self,
|
||
channel: MessageChannel,
|
||
request: PendingMediaInteraction,
|
||
items: List[MediaInfo],
|
||
total: int,
|
||
total_pages: int,
|
||
) -> List[List[Dict[str, str]]]:
|
||
"""
|
||
为媒体列表生成选择和翻页按钮。
|
||
"""
|
||
buttons: List[List[Dict[str, str]]] = []
|
||
max_text_length = ChannelCapabilityManager.get_max_button_text_length(channel)
|
||
max_per_row = ChannelCapabilityManager.get_max_buttons_per_row(channel)
|
||
|
||
current_row: List[Dict[str, str]] = []
|
||
for index, media in enumerate(items, start=1):
|
||
if max_per_row == 1:
|
||
button_text = f"{index}. {media.title_year}"
|
||
if len(button_text) > max_text_length:
|
||
button_text = button_text[: max_text_length - 3] + "..."
|
||
buttons.append(
|
||
[
|
||
{
|
||
"text": button_text,
|
||
"callback_data": f"media:{request.request_id}:select:{index}",
|
||
}
|
||
]
|
||
)
|
||
continue
|
||
|
||
current_row.append(
|
||
{
|
||
"text": f"{index}",
|
||
"callback_data": f"media:{request.request_id}:select:{index}",
|
||
}
|
||
)
|
||
if len(current_row) == max_per_row or index == len(items):
|
||
buttons.append(current_row)
|
||
current_row = []
|
||
|
||
if total > self._page_size(channel):
|
||
buttons.extend(self._navigation_buttons(request, total_pages))
|
||
return buttons
|
||
|
||
def _create_torrent_buttons(
|
||
self,
|
||
channel: MessageChannel,
|
||
request: PendingMediaInteraction,
|
||
items: List[Context],
|
||
total: int,
|
||
total_pages: int,
|
||
) -> List[List[Dict[str, str]]]:
|
||
"""
|
||
为资源列表生成下载和翻页按钮。
|
||
"""
|
||
buttons: List[List[Dict[str, str]]] = [
|
||
[
|
||
{
|
||
"text": "🤖 自动选择下载",
|
||
"callback_data": f"media:{request.request_id}:download:0",
|
||
}
|
||
]
|
||
]
|
||
max_text_length = ChannelCapabilityManager.get_max_button_text_length(channel)
|
||
max_per_row = ChannelCapabilityManager.get_max_buttons_per_row(channel)
|
||
|
||
current_row: List[Dict[str, str]] = []
|
||
for index, context in enumerate(items, start=1):
|
||
torrent = context.torrent_info
|
||
if max_per_row == 1:
|
||
button_text = f"{index}. {torrent.site_name} - {torrent.seeders}↑"
|
||
if len(button_text) > max_text_length:
|
||
button_text = button_text[: max_text_length - 3] + "..."
|
||
buttons.append(
|
||
[
|
||
{
|
||
"text": button_text,
|
||
"callback_data": f"media:{request.request_id}:download:{index}",
|
||
}
|
||
]
|
||
)
|
||
continue
|
||
|
||
current_row.append(
|
||
{
|
||
"text": f"{index}",
|
||
"callback_data": f"media:{request.request_id}:download:{index}",
|
||
}
|
||
)
|
||
if len(current_row) == max_per_row or index == len(items):
|
||
buttons.append(current_row)
|
||
current_row = []
|
||
|
||
if total > self._page_size(channel):
|
||
buttons.extend(self._navigation_buttons(request, total_pages))
|
||
return buttons
|
||
|
||
def _has_next_page(self, request: PendingMediaInteraction) -> bool:
|
||
"""
|
||
判断当前视图是否还有下一页。
|
||
"""
|
||
_, page, total_pages = self._page_items(
|
||
items=request.items,
|
||
page=request.page,
|
||
page_size=self._page_size(request.channel),
|
||
)
|
||
return page < total_pages - 1
|
||
|
||
@staticmethod
|
||
def _navigation_buttons(
|
||
request: PendingMediaInteraction,
|
||
total_pages: int,
|
||
) -> List[List[Dict[str, str]]]:
|
||
"""
|
||
按当前页状态生成上一页和下一页按钮。
|
||
"""
|
||
buttons: List[List[Dict[str, str]]] = []
|
||
nav_row: List[Dict[str, str]] = []
|
||
if request.page > 0:
|
||
nav_row.append(
|
||
{
|
||
"text": "⬅️ 上一页",
|
||
"callback_data": f"media:{request.request_id}:page-prev",
|
||
}
|
||
)
|
||
if request.page < total_pages - 1:
|
||
nav_row.append(
|
||
{
|
||
"text": "下一页 ➡️",
|
||
"callback_data": f"media:{request.request_id}:page-next",
|
||
}
|
||
)
|
||
if nav_row:
|
||
buttons.append(nav_row)
|
||
return buttons
|
||
|
||
@staticmethod
|
||
def _page_items(
|
||
items: List[Any],
|
||
page: int,
|
||
page_size: int,
|
||
) -> Tuple[List[Any], int, int]:
|
||
"""
|
||
返回当前页数据，并把页码限制在有效范围内。
|
||
"""
|
||
total_pages = max(1, math.ceil(len(items) / page_size)) if page_size else 1
|
||
page = min(max(0, page), total_pages - 1)
|
||
start = page * page_size
|
||
end = start + page_size
|
||
return items[start:end], page, total_pages
|
||
|
||
def _page_size(self, channel: Optional[MessageChannel]) -> int:
|
||
"""
|
||
按渠道交互能力选择分页大小。
|
||
"""
|
||
return (
|
||
self._button_page_size
|
||
if self._supports_interactive_buttons(channel)
|
||
else self._text_page_size
|
||
)
|
||
|
||
@staticmethod
|
||
def _supports_interactive_buttons(channel: Optional[MessageChannel]) -> bool:
|
||
"""
|
||
判断渠道是否同时支持按钮展示与按钮回调。
|
||
"""
|
||
return bool(
|
||
channel
|
||
and ChannelCapabilityManager.supports_buttons(channel)
|
||
and ChannelCapabilityManager.supports_callbacks(channel)
|
||
)
|
||
|
||
@staticmethod
|
||
def _build_no_exists_messages(
|
||
mediainfo: MediaInfo,
|
||
no_exists: Optional[Dict[Union[int, str], Dict[int, NotExistMediaInfo]]],
|
||
show_missing_only: bool,
|
||
) -> List[str]:
|
||
"""
|
||
将缺失集信息转换为可发送的文案。
|
||
"""
|
||
if not no_exists:
|
||
return []
|
||
mediakey = mediainfo.tmdb_id or mediainfo.douban_id
|
||
season_map = no_exists.get(mediakey) or {}
|
||
if show_missing_only:
|
||
return [
|
||
f"第 {sea} 季缺失 {StringUtils.str_series(no_exist.episodes) if no_exist.episodes else no_exist.total_episode} 集"
|
||
for sea, no_exist in season_map.items()
|
||
]
|
||
return [
|
||
f"第 {sea} 季总 {no_exist.total_episode} 集"
|
||
for sea, no_exist in season_map.items()
|
||
]
|
||
|
||
@staticmethod
|
||
def _should_auto_download(userid: Union[str, int]) -> bool:
|
||
"""
|
||
判断当前用户是否命中自动下载名单。
|
||
"""
|
||
auto_download_user = settings.AUTO_DOWNLOAD_USER
|
||
return bool(
|
||
auto_download_user
|
||
and (
|
||
auto_download_user == "all"
|
||
or any(userid == user for user in auto_download_user.split(","))
|
||
)
|
||
)
|
||
|
||
def _post_invalid_input(
|
||
self,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
userid: Union[str, int],
|
||
username: Optional[str],
|
||
title: str = "输入有误！",
|
||
) -> None:
|
||
"""
|
||
发送统一的非法输入提示。
|
||
"""
|
||
self.post_message(
|
||
Notification(
|
||
channel=channel,
|
||
source=source,
|
||
userid=userid,
|
||
username=username,
|
||
title=title,
|
||
)
|
||
)
|