import pymysql
from datetime import datetime, timedelta
import redis
import xml.etree.ElementTree as ET
from message_summary.message_summary_4o import message_summary

# 配置数据库连接
db_config = {
    'host': '192.168.2.32',  # 替换为你的MariaDB服务器地址
    'user': 'root',  # 替换为你的MariaDB用户名
    'password': 'lw123456',  # 替换为你的MariaDB密码
    'database': 'message_archive'
}

# 连接到Redis
r = redis.Redis(host='192.168.2.32', port=6379, db=0)


def archive_message(group_id, timestamp_str, sender, content, message_type, attachment_url=None):
    # 连接到数据库
    connection = pymysql.connect(**db_config)

    try:
        with connection.cursor() as cursor:
            # 插入消息信息
            sql = """
            INSERT INTO messages (group_id,timestamp, sender, content, message_type, attachment_url)
            VALUES (%s, %s, %s, %s, %s, %s)
            """
            cursor.execute(sql, (group_id, timestamp_str, sender, content, message_type, attachment_url))

        # 提交事务
        connection.commit()
        print(f"Archived：{timestamp_str}:{group_id}:{sender}: {content}")

    except Exception as e:
        print(f"Error archiving message: {e}")
        connection.rollback()

    finally:
        # 关闭连接
        connection.close()


def get_messages(group_id, all_contacts: dict):
    # 连接到数据库
    with pymysql.connect(**db_config) as connection:
        # 获取 redis 中的上次总结时间，本次从上次开始算，若没有，则从 8 小时之前开始计算
        key = f"{group_id}:summary_time"
        last_summary_time = r.get(key)

        # 如果 Redis 返回值为字节类型，转换为字符串
        if last_summary_time:
            last_summary_time = last_summary_time.decode('utf-8')

        current_time = datetime.now()
        current_date = current_time.strftime('%Y-%m-%d %H:%M:%S')

        if not last_summary_time:
            # 获取当前时间并计算 8 小时前的时间
            eight_hours_ago = current_time - timedelta(hours=8)
            last_summary_time = eight_hours_ago.strftime('%Y-%m-%d %H:%M:%S')
        else:
            # 检查 redis 中的时间与当前时间差是否小于 3 小时
            last_summary_time_obj = datetime.strptime(last_summary_time, '%Y-%m-%d %H:%M:%S')
            time_diff = current_time - last_summary_time_obj

            if time_diff < timedelta(hours=3):
                # 如果小于 3 小时，取当天的内容
                last_summary_time = current_time.replace(hour=0, minute=0, second=0, microsecond=0).strftime(
                    '%Y-%m-%d %H:%M:%S')
            elif time_diff > timedelta(days=1):
                # 如果超过 24 小时，将时间设置为当天 0 点
                last_summary_time = current_time.replace(hour=0, minute=0, second=0, microsecond=0).strftime(
                    '%Y-%m-%d %H:%M:%S')

        # 更新 Redis 存储的当前时间
        # r.set(key, current_date)

        with connection.cursor() as cursor:
            # 执行查询，获取最近 8 小时的消息
            query = """
                SELECT timestamp, sender, content,message_type
                FROM messages
                WHERE timestamp >= %s AND message_type in(1,49) AND group_id = %s
            """
            cursor.execute(query, (last_summary_time, group_id))

            # 构建最终的结果字符串
            #  message_type 需要加入49类型，因为49是引用之后的发言。但是49是xml ，需要将content进行xml解析

            result = []
            for row in cursor.fetchall():
                timestamp, sender, content, message_type = row
                try:
                    if message_type == "49":
                        # 解析 XML 字符串
                        root = ET.fromstring(content)
                        # 提取 title 内容
                        content = root.find('.//title').text

                except Exception as e:
                    print(f"message_type 49 error: {e}")
                sender_name = all_contacts.get(sender, sender)  # 获取发送者的名字，若找不到则使用原 ID
                result.append(f"{timestamp},{sender_name},{content}")

            result_str = "\n".join(result)  # 将结果拼接为最终字符串
            print(result_str)
            return result_str


# 示例用法
if __name__ == "__main__":
    # group_id = 'XXX@123123'
    # timestamp_str = "2025-02-06 11:15:28"
    # sender = "XXX"
    # content = "This is a test message with a string timestamp."
    # message_type = "text"
    # attachment_url = "http://example.com/attachment.pdf"  # 可以为None如果没有附件
    #
    # archive_message(group_id, timestamp_str, sender, content, message_type, attachment_url)
    # get_messages("45317011307@chatroom", {})
    message_summary(get_messages("45317011307@chatroom", {}))