mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-13 07:26:45 +00:00
290 lines
11 KiB
Python
290 lines
11 KiB
Python
from typing import Any, List, Optional, Tuple, Union
|
||
|
||
from app.core.context import Context, MediaInfo
|
||
from app.log import logger
|
||
from app.modules import _ModuleBase, _MessageBase
|
||
from app.modules.feishu.feishu import Feishu
|
||
from app.schemas import CommingMessage, MessageChannel, MessageResponse, Notification
|
||
from app.schemas.types import ModuleType
|
||
|
||
|
||
class FeishuModule(_ModuleBase, _MessageBase[Feishu]):
|
||
def init_module(self) -> None:
|
||
self.stop()
|
||
super().init_service(service_name=Feishu.__name__.lower(), service_type=Feishu)
|
||
self._channel = MessageChannel.Feishu
|
||
|
||
@staticmethod
|
||
def get_name() -> str:
|
||
return "飞书"
|
||
|
||
@staticmethod
|
||
def get_type() -> ModuleType:
|
||
return ModuleType.Notification
|
||
|
||
@staticmethod
|
||
def get_subtype() -> MessageChannel:
|
||
return MessageChannel.Feishu
|
||
|
||
@staticmethod
|
||
def get_priority() -> int:
|
||
return 2
|
||
|
||
def stop(self):
|
||
for client in self.get_instances().values():
|
||
if hasattr(client, "stop"):
|
||
try:
|
||
client.stop()
|
||
except Exception as err:
|
||
logger.error(f"停止飞书模块实例失败：{err}")
|
||
|
||
def test(self) -> Optional[Tuple[bool, str]]:
|
||
if not self.get_instances():
|
||
return None
|
||
for name, client in self.get_instances().items():
|
||
state = client.get_state()
|
||
if not state:
|
||
return False, f"飞书 {name} 未就绪"
|
||
return True, ""
|
||
|
||
def init_setting(self) -> Tuple[str, Union[str, bool]]:
|
||
"""通知模块通过系统通知配置控制实例化，这里不额外设置环境开关。"""
|
||
return None
|
||
|
||
@staticmethod
|
||
def _resolve_message_target(
|
||
message: Notification,
|
||
) -> Tuple[Optional[str], Optional[str], Optional[str]]:
|
||
"""优先使用 open_id，其次回退 user_id 或 chat_id。"""
|
||
userid = str(message.userid).strip() if message.userid else None
|
||
chat_id = None
|
||
receive_id_type = "open_id" if userid else None
|
||
|
||
targets = message.targets or {}
|
||
if not userid and targets:
|
||
open_id = str(targets.get("feishu_openid") or "").strip() or None
|
||
user_id = str(targets.get("feishu_userid") or "").strip() or None
|
||
chat_id = str(targets.get("feishu_chat_id") or "").strip() or None
|
||
if open_id:
|
||
userid = open_id
|
||
receive_id_type = "open_id"
|
||
elif user_id:
|
||
userid = user_id
|
||
receive_id_type = "user_id"
|
||
|
||
return userid, chat_id, receive_id_type
|
||
|
||
def message_parser(
|
||
self, source: str, body: Any, form: Any, args: Any
|
||
) -> Optional[CommingMessage]:
|
||
client_config = self.get_config(source)
|
||
if not client_config:
|
||
return None
|
||
client: Feishu = self.get_instance(client_config.name)
|
||
if not client:
|
||
return None
|
||
return client.parse_message(body)
|
||
|
||
def post_message(self, message: Notification, **kwargs) -> None:
|
||
for conf in self.get_configs().values():
|
||
if not self.check_message(message, conf.name):
|
||
continue
|
||
userid, chat_id, receive_id_type = self._resolve_message_target(message)
|
||
client: Feishu = self.get_instance(conf.name)
|
||
if client:
|
||
if message.file_path:
|
||
client.send_file(
|
||
file_path=message.file_path,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
title=message.title,
|
||
text=message.text,
|
||
file_name=message.file_name,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
elif message.voice_path:
|
||
client.send_voice(
|
||
voice_path=message.voice_path,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
caption=message.voice_caption,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
else:
|
||
client.send_notification(
|
||
message=message,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
|
||
def post_medias_message(self, message: Notification, medias: List[MediaInfo]) -> None:
|
||
for conf in self.get_configs().values():
|
||
if not self.check_message(message, conf.name):
|
||
continue
|
||
userid, chat_id, receive_id_type = self._resolve_message_target(message)
|
||
client: Feishu = self.get_instance(conf.name)
|
||
if client:
|
||
client.send_medias_message(
|
||
message=message,
|
||
medias=medias,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
receive_id_type=receive_id_type,
|
||
)
|
||
|
||
def post_torrents_message(self, message: Notification, torrents: List[Context]) -> None:
|
||
for conf in self.get_configs().values():
|
||
if not self.check_message(message, conf.name):
|
||
continue
|
||
userid, chat_id, receive_id_type = self._resolve_message_target(message)
|
||
client: Feishu = self.get_instance(conf.name)
|
||
if client:
|
||
client.send_torrents_message(
|
||
message=message,
|
||
torrents=torrents,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
receive_id_type=receive_id_type,
|
||
)
|
||
|
||
def edit_message(
|
||
self,
|
||
channel: MessageChannel,
|
||
source: str,
|
||
message_id: Union[str, int],
|
||
chat_id: Union[str, int],
|
||
text: str,
|
||
title: Optional[str] = None,
|
||
buttons: Optional[List[List[dict]]] = None,
|
||
) -> bool:
|
||
if channel != self._channel:
|
||
return False
|
||
for conf in self.get_configs().values():
|
||
if source != conf.name:
|
||
continue
|
||
client: Feishu = self.get_instance(conf.name)
|
||
if client and client.edit_message(
|
||
message_id=str(message_id),
|
||
title=title,
|
||
text=text,
|
||
buttons=buttons,
|
||
):
|
||
return True
|
||
return False
|
||
|
||
def send_direct_message(self, message: Notification) -> Optional[MessageResponse]:
|
||
for conf in self.get_configs().values():
|
||
if not self.check_message(message, conf.name):
|
||
continue
|
||
userid, chat_id, receive_id_type = self._resolve_message_target(message)
|
||
client: Feishu = self.get_instance(conf.name)
|
||
if not client:
|
||
continue
|
||
if message.file_path:
|
||
result = client.send_file(
|
||
file_path=message.file_path,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
title=message.title,
|
||
text=message.text,
|
||
file_name=message.file_name,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
elif message.voice_path:
|
||
result = client.send_voice(
|
||
voice_path=message.voice_path,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
caption=message.voice_caption,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
else:
|
||
result = client.send_notification(
|
||
message=message,
|
||
userid=userid,
|
||
chat_id=chat_id,
|
||
receive_id_type=receive_id_type,
|
||
original_message_id=str(message.original_message_id) if message.original_message_id else None,
|
||
)
|
||
if result and result.get("success"):
|
||
return MessageResponse(
|
||
message_id=result.get("message_id"),
|
||
chat_id=result.get("chat_id"),
|
||
channel=MessageChannel.Feishu,
|
||
source=conf.name,
|
||
success=True,
|
||
)
|
||
return None
|
||
|
||
def download_feishu_image_to_data_url(self, image_ref: str, source: str) -> Optional[str]:
|
||
if not image_ref or not image_ref.startswith("feishu://image/"):
|
||
return None
|
||
client_config = self.get_config(source)
|
||
if not client_config:
|
||
return None
|
||
client = self.get_instance(client_config.name)
|
||
if not client:
|
||
return None
|
||
image_key = image_ref.replace("feishu://image/", "", 1)
|
||
downloaded = client._download_image_bytes(image_key)
|
||
if not downloaded:
|
||
return None
|
||
content, _, content_type = downloaded
|
||
mime_type = content_type or "image/jpeg"
|
||
import base64
|
||
|
||
return f"data:{mime_type};base64,{base64.b64encode(content).decode()}"
|
||
|
||
def download_feishu_file_bytes(self, file_ref: str, source: str) -> Optional[bytes]:
|
||
if not file_ref or not file_ref.startswith("feishu://file/"):
|
||
return None
|
||
client_config = self.get_config(source)
|
||
if not client_config:
|
||
return None
|
||
client = self.get_instance(client_config.name)
|
||
if not client:
|
||
return None
|
||
parts = file_ref.replace("feishu://file/", "", 1).split("/", 1)
|
||
file_key = parts[0].strip() if parts else ""
|
||
if not file_key:
|
||
return None
|
||
downloaded = client._download_file_bytes(file_key)
|
||
if not downloaded:
|
||
return None
|
||
content, _, _ = downloaded
|
||
return content
|
||
|
||
def add_feishu_message_reaction(
|
||
self,
|
||
message_id: str,
|
||
emoji_type: str,
|
||
source: str,
|
||
) -> Optional[str]:
|
||
client_config = self.get_config(source)
|
||
if not client_config:
|
||
return None
|
||
client = self.get_instance(client_config.name)
|
||
if not client:
|
||
return None
|
||
return client.add_message_reaction(message_id=message_id, emoji_type=emoji_type)
|
||
|
||
def delete_feishu_message_reaction(
|
||
self,
|
||
message_id: str,
|
||
reaction_id: str,
|
||
source: str,
|
||
) -> bool:
|
||
client_config = self.get_config(source)
|
||
if not client_config:
|
||
return False
|
||
client = self.get_instance(client_config.name)
|
||
if not client:
|
||
return False
|
||
return client.delete_message_reaction(message_id=message_id, reaction_id=reaction_id)
|