#coding=utf-8
import itchat
from itchat.content import TEXT
from itchat.content import *
import sys
import time
import re
import os

msg_information = {}
face_bug=None  #針對表情包的內容
@itchat.msg_register([TEXT,PICTURE,FRIENDS,CARD,MAP,SHARING,RECORDING,ATTACHMENT,VIDEO],isFriendChat=True,isGroupChat=True)
def receive_msg(msg):
    global face_bug
    msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) #接收訊息的時間
    if 'ActualNickName' in msg:
        from_user = msg['ActualUserName'] #群訊息的傳送者,使用者的唯一標識
        msg_from = msg['ActualNickName']#傳送者群內的暱稱
        friends = itchat.get_friends(update=True)#獲取所有好友
        for f in friends:
            if from_user == f['UserName']: #如果群訊息是好友發的
                if f['RemarkName']: # 優先使用好友的備註名稱，沒有則使用暱稱
                    msg_from = f['RemarkName']
                else:
                    msg_from = f['NickName']
                break
        groups = itchat.get_chatrooms(update=True)#獲取所有的群
        for g in groups:
            if msg['FromUserName'] == g['UserName']:#根據群訊息的FromUserName匹配是哪個群
                group_name = g['NickName']
                group_menbers = g['MemberCount']
                break
        group_name = group_name + "(" + str(group_menbers) +")"
    else:
        if itchat.search_friends(userName=msg['FromUserName'])['RemarkName']:#優先使用備註名稱
            msg_from = itchat.search_friends(userName=msg['FromUserName'])['RemarkName']
        else:
            msg_from = itchat.search_friends(userName=msg['FromUserName'])['NickName'] #在好友列表中查詢傳送資訊的好友暱稱
        group_name = ""
    msg_time = msg['CreateTime'] #資訊傳送的時間
    msg_id = msg['MsgId']    #每條資訊的id
    msg_content = None      #儲存資訊的內容
    msg_share_url = None    #儲存分享的連結，比如分享的文章和音樂
    if msg['Type'] == 'Text' or msg['Type'] == 'Friends':     #如果傳送的訊息是文字或者好友推薦
        msg_content = msg['Text']

    #如果傳送的訊息是附件、視訊、圖片、語音
    elif msg['Type'] == "Attachment" or msg['Type'] == "Video" \
            or msg['Type'] == 'Picture' \
            or msg['Type'] == 'Recording':
        msg_content = msg['FileName']    #內容就是他們的檔名
        msg['Text'](str(msg_content))    #下載檔案

    elif msg['Type'] == 'Map':    #如果訊息為分享的位置資訊
        x, y, location = re.search(
            "<location x=\"(.*?)\" y=\"(.*?)\".*label=\"(.*?)\".*", msg['OriContent']).group(1, 2, 3)
        if location is None:
            msg_content = r"緯度->" + x.__str__() + " 經度->" + y.__str__()     #內容為詳細的地址
        else:
            msg_content = r"" + location

    elif msg['Type'] == 'Sharing':     #如果訊息為分享的音樂或者文章，詳細的內容為文章的標題或者是分享的名字
        msg_content = msg['Text']
        msg_share_url = msg['Url']       #記錄分享的url
    face_bug = msg_content
    #將資訊儲存在字典中，每一個msg_id對應一條資訊
    time.sleep(2)
    msg_information.update(
        {
            msg_id: {
                "msg_from": msg_from,
                "msg_time": msg_time,
                "msg_time_rec": msg_time_rec,
                "msg_type": msg["Type"],
                "msg_content": msg_content,
                "msg_share_url": msg_share_url,
                "group_name":group_name
            }
        }
    )

    del_info = []
    for k in msg_information:
        m_time = msg_information[k]['msg_time'] #取得訊息時間
        if int(time.time()) - m_time > 130: #如果訊息時間是130秒甚至更久之前的，則刪除。
            del_info.append(k)
    if del_info:
        for i in del_info:
            msg_information.pop(i)

#監聽是否有訊息撤回
@itchat.msg_register(NOTE,isFriendChat=True,isGroupChat=True,isMpChat=True)
def information(msg):
    #如果這裡的msg['Content']中包含訊息撤回和id，就執行下面的語句
    if '撤回了一條訊息' in msg['Content']:
        old_msg_id = re.search("\<msgid\>(.*?)\<\/msgid\>", msg['Content']).group(1) #在返回的content查詢撤回的訊息的id
        old_msg = msg_information.get(old_msg_id)    #獲取到訊息原文,型別：字典
        print(old_msg)
        if len(old_msg_id)<11:  #如果傳送的是表情包
            itchat.send_file(face_bug,toUserName='filehelper')

        else:  #傳送撤回的提示給檔案助手
            msg_body = old_msg['group_name'] + old_msg['msg_from'] +"\n" + old_msg['msg_time_rec'] \
                       + "撤回了:" + "\n" + r"" + old_msg['msg_content']

            #如果是分享的檔案被撤回了，那麼就將分享的url加在msg_body中傳送給檔案助手
            if old_msg['msg_type'] == "Sharing":
                msg_body += "\n連結是:" + old_msg.get('msg_share_url')
            #print(msg_body)
            itchat.send_msg(msg_body, toUserName='filehelper')#將撤回訊息發給檔案助手

            #有檔案的話也要將檔案傳送回去
            if old_msg["msg_type"] == "Picture" \
                    or old_msg["msg_type"] == "Recording" \
                    or old_msg["msg_type"] == "Video" \
                    or old_msg["msg_type"] == "Attachment":
                file = '@
 
[email protected]%s' % (old_msg['msg_content'])
                itchat.send(msg=file, toUserName='filehelper')
                os.remove(old_msg['msg_content'])
            msg_information.pop(old_msg_id)# 刪除字典舊訊息

itchat.auto_login()
itchat.run()