import traceback import time from typing import Optional, List, Dict, Any from src.common.logger_manager import get_logger from src.chat.heart_flow.observation.chatting_observation import ChattingObservation from src.chat.focus_chat.hfc_utils import create_empty_anchor_message # 以下为类型注解需要 from src.chat.message_receive.chat_stream import ChatStream, chat_manager from src.chat.focus_chat.expressors.default_expressor import DefaultExpressor from src.chat.focus_chat.replyer.default_replyer import DefaultReplyer from src.chat.focus_chat.info.obs_info import ObsInfo # 新增导入 from src.chat.focus_chat.heartFC_sender import HeartFCSender from src.chat.message_receive.message import MessageSending from maim_message import Seg, UserInfo from src.config.config import global_config logger = get_logger("message_api") class MessageAPI: """消息API模块 提供了发送消息、获取消息历史等功能 """ async def send_message_to_target( self, message_type: str, content: str, platform: str, target_id: str, is_group: bool = True, display_message: str = "", ) -> bool: """直接向指定目标发送消息 Args: message_type: 消息类型，如"text"、"image"、"emoji"等 content: 消息内容 platform: 目标平台，如"qq" target_id: 目标ID（群ID或用户ID） is_group: 是否为群聊，True为群聊，False为私聊 display_message: 显示消息（可选） Returns: bool: 是否发送成功 """ try: # 构建目标聊天流ID if is_group: # 群聊：从数据库查找对应的聊天流 target_stream = None for stream_id, stream in chat_manager.streams.items(): if ( stream.group_info and str(stream.group_info.group_id) == str(target_id) and stream.platform == platform ): target_stream = stream break if not target_stream: logger.error(f"{getattr(self, 'log_prefix', '')} 未找到群ID为 {target_id} 的聊天流") return False else: # 私聊：从数据库查找对应的聊天流 target_stream = None for stream_id, stream in chat_manager.streams.items(): if ( not stream.group_info and str(stream.user_info.user_id) == str(target_id) and stream.platform == platform ): target_stream = stream break if not target_stream: logger.error(f"{getattr(self, 'log_prefix', '')} 未找到用户ID为 {target_id} 的私聊流") return False # 创建HeartFCSender实例 heart_fc_sender = HeartFCSender() # 生成消息ID和thinking_id current_time = time.time() message_id = f"plugin_msg_{int(current_time * 1000)}" thinking_id = f"plugin_thinking_{int(current_time * 1000)}" # 构建机器人用户信息 bot_user_info = UserInfo( user_id=global_config.bot.qq_account, user_nickname=global_config.bot.nickname, platform=platform, ) # 创建消息段 message_segment = Seg(type=message_type, data=content) # 创建空锚点消息（用于回复） anchor_message = await create_empty_anchor_message(platform, target_stream.group_info, target_stream) # 构建发送消息对象 bot_message = MessageSending( message_id=message_id, chat_stream=target_stream, bot_user_info=bot_user_info, sender_info=target_stream.user_info, # 目标用户信息 message_segment=message_segment, display_message=display_message, reply=anchor_message, is_head=True, is_emoji=(message_type == "emoji"), thinking_start_time=current_time, ) # 发送消息 sent_msg = await heart_fc_sender.send_message(bot_message, has_thinking=True, typing=False, set_reply=False) if sent_msg: logger.info(f"{getattr(self, 'log_prefix', '')} 成功发送消息到 {platform}:{target_id}") return True else: logger.error(f"{getattr(self, 'log_prefix', '')} 发送消息失败") return False except Exception as e: logger.error(f"{getattr(self, 'log_prefix', '')} 向目标发送消息时出错: {e}") traceback.print_exc() return False async def send_text_to_group(self, text: str, group_id: str, platform: str = "qq") -> bool: """便捷方法：向指定群聊发送文本消息 Args: text: 要发送的文本内容 group_id: 群聊ID platform: 平台，默认为"qq" Returns: bool: 是否发送成功 """ return await self.send_message_to_target( message_type="text", content=text, platform=platform, target_id=group_id, is_group=True ) async def send_text_to_user(self, text: str, user_id: str, platform: str = "qq") -> bool: """便捷方法：向指定用户发送私聊文本消息 Args: text: 要发送的文本内容 user_id: 用户ID platform: 平台，默认为"qq" Returns: bool: 是否发送成功 """ return await self.send_message_to_target( message_type="text", content=text, platform=platform, target_id=user_id, is_group=False ) async def send_message(self, type: str, data: str, target: Optional[str] = "", display_message: str = "") -> bool: """发送消息的简化方法 Args: type: 消息类型，如"text"、"image"等 data: 消息内容 target: 目标消息（可选） display_message: 显示的消息内容（可选） Returns: bool: 是否发送成功 """ try: expressor: DefaultExpressor = self._services.get("expressor") chat_stream: ChatStream = self._services.get("chat_stream") if not expressor or not chat_stream: logger.error(f"{self.log_prefix} 无法发送消息：缺少必要的内部服务") return False # 获取锚定消息（如果有） observations = self._services.get("observations", []) if len(observations) > 0: chatting_observation: ChattingObservation = next( (obs for obs in observations if isinstance(obs, ChattingObservation)), None ) if chatting_observation: anchor_message = chatting_observation.search_message_by_text(target) else: anchor_message = None else: anchor_message = None # 如果没有找到锚点消息，创建一个占位符 if not anchor_message: logger.info(f"{self.log_prefix} 未找到锚点消息，创建占位符") anchor_message = await create_empty_anchor_message( chat_stream.platform, chat_stream.group_info, chat_stream ) else: anchor_message.update_chat_stream(chat_stream) response_set = [ (type, data), ] # 调用内部方法发送消息 success = await expressor.send_response_messages( anchor_message=anchor_message, response_set=response_set, display_message=display_message, ) return success except Exception as e: logger.error(f"{self.log_prefix} 发送消息时出错: {e}") traceback.print_exc() return False async def send_message_by_expressor(self, text: str, target: Optional[str] = None) -> bool: """通过expressor发送文本消息的简化方法 Args: text: 要发送的消息文本 target: 目标消息（可选） Returns: bool: 是否发送成功 """ expressor: DefaultExpressor = self._services.get("expressor") chat_stream: ChatStream = self._services.get("chat_stream") if not expressor or not chat_stream: logger.error(f"{self.log_prefix} 无法发送消息：缺少必要的内部服务") return False # 构造简化的动作数据 reply_data = {"text": text, "target": target or "", "emojis": []} # 获取锚定消息（如果有） observations = self._services.get("observations", []) # 查找 ChattingObservation 实例 chatting_observation = None for obs in observations: if isinstance(obs, ChattingObservation): chatting_observation = obs break if not chatting_observation: logger.warning(f"{self.log_prefix} 未找到 ChattingObservation 实例，创建占位符") anchor_message = await create_empty_anchor_message( chat_stream.platform, chat_stream.group_info, chat_stream ) else: anchor_message = chatting_observation.search_message_by_text(reply_data["target"]) if not anchor_message: logger.info(f"{self.log_prefix} 未找到锚点消息，创建占位符") anchor_message = await create_empty_anchor_message( chat_stream.platform, chat_stream.group_info, chat_stream ) else: anchor_message.update_chat_stream(chat_stream) # 调用内部方法发送消息 success, _ = await expressor.deal_reply( cycle_timers=self.cycle_timers, action_data=reply_data, anchor_message=anchor_message, reasoning=self.reasoning, thinking_id=self.thinking_id, ) return success async def send_message_by_replyer( self, target: Optional[str] = None, extra_info_block: Optional[str] = None ) -> bool: """通过replyer发送消息的简化方法 Args: target: 目标消息（可选） extra_info_block: 额外信息块（可选） Returns: bool: 是否发送成功 """ replyer: DefaultReplyer = self._services.get("replyer") chat_stream: ChatStream = self._services.get("chat_stream") if not replyer or not chat_stream: logger.error(f"{self.log_prefix} 无法发送消息：缺少必要的内部服务") return False # 构造简化的动作数据 reply_data = {"target": target or "", "extra_info_block": extra_info_block} # 获取锚定消息（如果有） observations = self._services.get("observations", []) # 查找 ChattingObservation 实例 chatting_observation = None for obs in observations: if isinstance(obs, ChattingObservation): chatting_observation = obs break if not chatting_observation: logger.warning(f"{self.log_prefix} 未找到 ChattingObservation 实例，创建占位符") anchor_message = await create_empty_anchor_message( chat_stream.platform, chat_stream.group_info, chat_stream ) else: anchor_message = chatting_observation.search_message_by_text(reply_data["target"]) if not anchor_message: logger.info(f"{self.log_prefix} 未找到锚点消息，创建占位符") anchor_message = await create_empty_anchor_message( chat_stream.platform, chat_stream.group_info, chat_stream ) else: anchor_message.update_chat_stream(chat_stream) # 调用内部方法发送消息 success, _ = await replyer.deal_reply( cycle_timers=self.cycle_timers, action_data=reply_data, anchor_message=anchor_message, reasoning=self.reasoning, thinking_id=self.thinking_id, ) return success def get_chat_type(self) -> str: """获取当前聊天类型 Returns: str: 聊天类型 ("group" 或 "private") """ chat_stream: ChatStream = self._services.get("chat_stream") if chat_stream and hasattr(chat_stream, "group_info"): return "group" if chat_stream.group_info else "private" return "unknown" def get_recent_messages(self, count: int = 5) -> List[Dict[str, Any]]: """获取最近的消息 Args: count: 要获取的消息数量 Returns: List[Dict]: 消息列表，每个消息包含发送者、内容等信息 """ messages = [] observations = self._services.get("observations", []) if observations and len(observations) > 0: obs = observations[0] if hasattr(obs, "get_talking_message"): obs: ObsInfo raw_messages = obs.get_talking_message() # 转换为简化格式 for msg in raw_messages[-count:]: simple_msg = { "sender": msg.get("sender", "未知"), "content": msg.get("content", ""), "timestamp": msg.get("timestamp", 0), } messages.append(simple_msg) return messages