#!/usr/bin/env python
# -*- coding: utf-8 -*-

import re

class MessageProcessor:
    """
    消息处理器类，负责消息的解析、分割和格式化
    """
    
    def __init__(self):
        self.cmd_pattern = r'nlpexec\([^)]*\)'
    
    def split_message(self, msg_str):
        """
        将包含系统命令的消息分割为多个部分
        例如："老板，您好！nlpexec(异常端口信息)好的，已获取异常端口信息。"
        会被分割为：
        ["老板，您好！", "nlpexec(异常端口信息)", "好的，已获取异常端口信息。"]
        """
        parts = []
        
        # 查找所有系统命令
        cmd_matches = list(re.finditer(self.cmd_pattern, msg_str))
        
        if cmd_matches:
            # 如果有系统命令，进行分割
            last_end = 0
            for match in cmd_matches:
                start, end = match.span()
                # 添加命令前的文本
                if start > last_end:
                    text_part = msg_str[last_end:start].strip()
                    if text_part:
                        parts.append(text_part)
                # 添加系统命令
                parts.append(msg_str[start:end])
                last_end = end
            # 添加命令后的文本
            if last_end < len(msg_str):
                text_part = msg_str[last_end:].strip()
                if text_part:
                    parts.append(text_part)
        else:
            # 没有系统命令，检查是否包含多个句子
            # 使用正则表达式分割句子
            sentences = re.split(r'[。！？]', msg_str)
            for sentence in sentences:
                if sentence.strip():
                    parts.append(sentence.strip() + '。')
        
        return parts
    
    def format_message(self, role_name, content, timestamp=None):
        """
        格式化消息，添加角色名和时间戳
        """
        if timestamp:
            return f"{timestamp}\n{role_name}: {content}\n"
        else:
            return f"{role_name}: {content}\n"
    
    def is_end_of_stream(self, msg):
        """
        检查消息是否是流式输出结束标记
        """
        return msg == '<END>'