From bc892a75f8a67c2d1321c127168b261d994f8658 Mon Sep 17 00:00:00 2001
From: lvtomatoj <lvtomatoj@gmail.com>
Date: Thu, 1 Feb 2024 13:01:54 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=BF=94=E5=9B=9E=E4=BC=9A?=
 =?UTF-8?q?=E8=AF=9D=E6=89=80=E6=9C=89=E7=94=A8=E6=88=B7=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pywxdump/analyzer/export_chat.py | 62 ++++++++++++++++++++++++++++++++
 pywxdump/api/api.py              | 25 +++++++++++++
 2 files changed, 87 insertions(+)

diff --git a/pywxdump/analyzer/export_chat.py b/pywxdump/analyzer/export_chat.py
index a997dee..8158338 100644
--- a/pywxdump/analyzer/export_chat.py
+++ b/pywxdump/analyzer/export_chat.py
@@ -24,6 +24,30 @@ from .utils import get_md5, attach_databases, execute_sql, get_type_name, match_
 from .db_parsing import parse_xml_string, decompress_CompressContent, read_BytesExtra
 
 
+def get_contact(MicroMsg_db_path,wx_id):
+    """
+    获取联系人信息
+    :param MicroMsg_db_path: MicroMsg.db 文件路径
+    :param wx_id: 微信id
+    :return: 联系人信息
+    """
+    db = sqlite3.connect(MicroMsg_db_path)
+    cursor = db.cursor()
+    # 获取username是wx_id的用户
+    sql = ("SELECT A.UserName, A.NickName, A.Remark,A.Alias,A.Reserved6,B.bigHeadImgUrl "
+           "FROM Contact A,ContactHeadImgUrl B "
+           f"WHERE A.UserName = '{wx_id}' AND A.UserName = B.usrName "
+           "ORDER BY NickName ASC;")
+    cursor.execute(sql)
+    result = cursor.fetchone()
+    print('联系人信息：', result)
+    if not result:
+        print('居然没找到！')
+        print(wx_id)
+        return None
+    return {"username": result[0], "nickname": result[1], "remark": result[2], "account": result[3], "describe": result[4], "headImgUrl": result[5]}
+    
+
 def get_contact_list(MicroMsg_db_path):
     """
     获取联系人列表
@@ -79,6 +103,44 @@ def get_chatroom_list(MicroMsg_db_path):
              "Announcement": Announcement, "AnnouncementEditor": AnnouncementEditor})
     return rooms
 
+def get_room_user_list(MSG_db_path, selected_talker):
+    """
+    获取群聊中包含的所有用户列表
+    :param MSG_db_path: MSG.db 文件路径
+    :param selected_talker: 选中的聊天对象 wxid
+    :return: 聊天用户列表
+    """
+    
+    # 连接 MSG_ALL.db 数据库，并执行查询
+    db1 = sqlite3.connect(MSG_db_path)
+    cursor1 = db1.cursor()
+
+    sql = (
+        "SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id "
+        "FROM MSG WHERE StrTalker=? "
+        "ORDER BY CreateTime ASC")
+
+    cursor1.execute(sql, (selected_talker,))
+    result1 = cursor1.fetchall()
+    cursor1.close()
+    db1.close()
+    user_list = []
+    
+    for row  in result1:
+        localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType, CreateTime, MsgSvrID, DisplayContent, CompressContent, BytesExtra, id = row
+        bytes_extra = read_BytesExtra(BytesExtra)
+        if bytes_extra:
+            try:
+                talker = bytes_extra['3'][0]['2'].decode('utf-8', errors='ignore')
+            except:
+                continue
+        user = get_contact(MSG_db_path, talker)
+        if not user:
+            continue
+        user_list.append(user)
+    return user_list
+            
+
 
 def get_msg_list(MSG_db_path, selected_talker="", start_index=0, page_size=500):
     """
diff --git a/pywxdump/api/api.py b/pywxdump/api/api.py
index c50928b..ec91aed 100644
--- a/pywxdump/api/api.py
+++ b/pywxdump/api/api.py
@@ -13,6 +13,7 @@ import shutil
 
 from flask import Flask, request, render_template, g, Blueprint, send_file, make_response, session
 from pywxdump import analyzer, read_img_dat, read_audio, get_wechat_db, get_core_db
+from pywxdump.analyzer.export_chat import get_contact, get_room_user_list
 from pywxdump.api.rjson import ReJson, RqJson
 from pywxdump.api.utils import read_session, save_session, error9999
 from pywxdump import read_info, VERSION_LIST, batch_decrypt, BiasAddr, merge_db, decrypt_merge
@@ -190,6 +191,30 @@ def contact_count_list():
     except Exception as e:
         return ReJson(9999, msg=str(e))
 
+@api.route('/api/msgs_user_list',methods=['GET','POST'])
+@error9999
+def get_msg_user_list():
+    msg_path = request.headers.get("msg_path")
+    micro_path = request.headers.get("micro_path")
+    if not msg_path:
+        msg_path = read_session(g.sf, "msg_path")
+    if not micro_path:
+        micro_path = read_session(g.sf, "micro_path")
+    wxid = request.json.get("wxid")
+    # msg_list = analyzer.get_msg_list(msg_path, wxid, start_index=start, page_size=limit)
+    my_wxid = read_session(g.sf, "my_wxid")
+    userlist = []
+    if wxid.endswith("@chatroom"):
+        # 群聊
+        userlist = get_room_user_list(msg_path, wxid)
+    else:
+        # 单聊
+        user = get_contact(micro_path, wxid)
+        my_user = get_contact(micro_path,my_wxid)
+        userlist.append(user)
+        userlist.append(my_user)
+    return ReJson(0, {"user_list":userlist})
+    
 
 @api.route('/api/msgs', methods=["GET", 'POST'])
 @error9999