import threading

from common import kafka_helper,redis_helper,utils

import json,time,re,random
from common.log import logger, log_exception 

from wechat import gewe_chat


def wx_messages_process_callback(message):
    wxchat=gewe_chat.wxchat
    msg_content= message
    cleaned_content = clean_json_string(msg_content)
    content=json.loads(cleaned_content)
    data = content.get("data", {})

    msg_type_data=data.get("msg_type",None)  
    content_data = data.get("content",{})  
    agent_tel=content_data.get("agent_tel",None)
    # print(message)
    print(msg_type_data)


    # # 更新好友缓存
    # save_friends_to_redis(agent_tel,agent_nickname_data,friends)
    if msg_type_data=='group-sending':
        hash_key = f"__AI_OPS_WX__:LOGININFO:{agent_tel}"
        # print('群发')
        
        logininfo=redis_helper.redis_helper.get_hash(hash_key)
        if logininfo:
            token_id=logininfo.get('tokenId')
            app_id=logininfo.get('appId')
            agent_wxid=logininfo.get('wxid')
            # print(token_id)
            # print(app_id)
            contacts_list=wxchat.fetch_contacts_list(token_id, app_id)
            friend_wxids=contacts_list['friends']
            wxid_contact_list_content_data=[c['wxid'] for c in content_data.get("contact_list",None)]
            intersection = list(set(friend_wxids) & set(wxid_contact_list_content_data))
            # print(intersection)
            wx_content_list=content_data.get("wx_content",[])
            for wx_content in wx_content_list:
                if wx_content["type"]=="text":
                      for t in intersection:
                        res=wxchat.post_text(token_id,app_id,t,wx_content["text"])
                        logger.info(f'{agent_wxid} 向 {t} 发送 文字【{wx_content["text"]}】')
                        
                        input_wx_content_dialogue_message=[{"type": "text", "text": wx_content["text"]}]
                        input_message=utils.dialogue_message(agent_wxid,t,input_wx_content_dialogue_message)
                        kafka_helper.kafka_client.produce_message(input_message)
                        logger.info("发送对话 %s",input_message)
                         # 等待随机时间
                        time.sleep(random.uniform(5, 15))
                if wx_content["type"]=="image_url":
                    aeskey=""
                    cdnthumburl=""
                    cdnthumblength=0
                    cdnthumbheight=0
                    cdnthumbwidth=0
                    length=0
                    md5=""
                    image_url= wx_content.get("image_url",{})
                    url=image_url.get("url",None)
                    for t in intersection:
                        if t == intersection[0]:
                            res=wxchat.post_image(token_id,app_id,t,url)
                            aeskey=res["aesKey"]
                            cdnthumburl=res["fileId"]
                            cdnthumblength=res["cdnThumbLength"]
                            cdnthumbheight=res["height"]
                            cdnthumbwidth=res["width"]
                            length=res["length"]
                            md5=res["md5"]
                            logger.info(f'{agent_wxid} 向 {t} 发送 图片【{url}】')
                        else:
                            res=wxchat.forward_image(token_id,app_id,t,aeskey,cdnthumburl,cdnthumblength,cdnthumbheight,cdnthumbwidth,length,md5)
                            logger.info(f'{agent_wxid} 向 {t} 转发送 图片【{url}】')
                        
                        wx_content_dialogue_message=[{"type": "image_url", "image_url": {"url": url}}]
                        input_message=utils.dialogue_message(agent_wxid,t,wx_content_dialogue_message)
                        kafka_helper.kafka_client.produce_message(input_message)
                        logger.info("发送对话 %s",input_message)

                        time.sleep(random.uniform(5, 15))
                        

            



def clean_json_string(json_str):

    # 删除所有控制字符（非打印字符），包括换行符、回车符等
    return re.sub(r'[\x00-\x1f\x7f]', '', json_str)

# 启动 Kafka 消费者线程
def start_kafka_consumer_thread():
    consumer_thread = threading.Thread(target=kafka_helper.kafka_client.consume_messages, args=(wx_messages_process_callback,))
    consumer_thread.daemon = True  # 设置为守护线程，应用退出时会自动结束
    consumer_thread.start()