WechatBakTool/WXUserReader.cs

using K4os.Compression.LZ4.Encoders;
using K4os.Compression.LZ4;
using SQLite;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.IO;
using System.Linq;
using System.Net.Mime;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Interop;
using System.Windows.Media.Imaging;
using System.Xml;
using System.Xml.Linq;
using WechatBakTool.Helpers;
using WechatBakTool.Model;
using System.Windows;
using System.Net.Http;
using System.Reflection.Metadata;
using System.Threading;

namespace WechatBakTool
{
    public class WXUserReader
    {
        private Dictionary<string, SQLiteConnection> DBInfo = new Dictionary<string, SQLiteConnection>();
        private UserBakConfig? UserBakConfig = null;
        private Hashtable HeadImgCache = new Hashtable();
        private Hashtable UserNameCache = new Hashtable();
        private Hashtable EmojiCache = new Hashtable();
        private HttpClient httpClient = new HttpClient();
        public WXUserReader(UserBakConfig userBakConfig) {
            string path = Path.Combine(userBakConfig.UserWorkspacePath, "DecDB");
            UserBakConfig = userBakConfig;
            LoadDB(path);
            InitCache();
        }

        public void LoadDB(string path)
        {
            string[] dbFileList = Directory.GetFiles(path);
            foreach (var item in dbFileList)
            {
                FileInfo fileInfo = new FileInfo(item);
                if (fileInfo.Extension != ".db")
                    continue;
                SQLiteConnection con = new SQLiteConnection(item);
                string dbName = fileInfo.Name.Split('.')[0];
                DBInfo.Add(dbName, con);
            }
        }

        private SQLiteConnection? getCon(string name)
        {
            if (DBInfo.ContainsKey(name))
            {
                return DBInfo[name];
            }
            else
            {
                return null;
            }
        }

        public void InitCache()
        {
            SQLiteConnection? con = getCon("Misc");
            if (con == null)
                return;

            string query = @"SELECT * FROM ContactHeadImg1";
            List<ContactHeadImg> imgs = con.Query<ContactHeadImg>(query);
            foreach(ContactHeadImg item in imgs)
            {
                if (!HeadImgCache.ContainsKey(item.usrName))
                {
                    HeadImgCache.Add(item.usrName, item);
                }
            }

            List<WXContact> contacts = GetWXContacts(null, true).ToList();
            foreach(WXContact contact in contacts)
            {
                if (!UserNameCache.ContainsKey(contact.UserName))
                    UserNameCache.Add(contact.UserName, contact);
            }
        }

        public void EmojiCacheInit()
        {
            string emoji_path = Path.Combine(UserBakConfig!.UserWorkspacePath, "Emoji");
            if (Directory.Exists(emoji_path))
            {
                string[] files = Directory.GetFiles(emoji_path);
                foreach (string file in files)
                {
                    FileInfo fileInfo = new FileInfo(file);
                    string[] names = fileInfo.Name.Split(".");
                    if (!EmojiCache.ContainsKey(names[0]))
                    {
                        EmojiCache.Add(names[0], 1);
                    }
                }
            }
        }

        public void PreDownloadEmoji()
        {
            if (UserBakConfig == null)
                return;

            HttpClientHandler handler = new HttpClientHandler() { AutomaticDecompression = System.Net.DecompressionMethods.GZip | System.Net.DecompressionMethods.Deflate };
            HttpClient httpClient = new HttpClient(handler);

            List<WXMsg> msgs = GetTypeMsg("47");
            int i = 0;

            EmojiCacheInit();

            foreach (var msg in msgs)
            {
                i++;
                if (i % 5 == 0)
                {
                    // 每5次让下载线程休息1秒
                    Thread.Sleep(1000);
                }
                try
                {
                    XmlDocument xmlDocument = new XmlDocument();
                    xmlDocument.LoadXml(msg.StrContent);
                    XmlNode? node = xmlDocument.SelectSingleNode("/msg/emoji");
                    if (node != null)
                    {
                        if (node.Attributes != null)
                        {
                            string type = "";
                            string md5 = "";
                            string url = "";
                            XmlNode? item = node.Attributes.GetNamedItem("type");
                            type = item != null ? item.InnerText : "";

                            item = node.Attributes.GetNamedItem("md5");
                            md5 = item != null ? item.InnerText : "";

                            item = node.Attributes.GetNamedItem("cdnurl");
                            url = item != null ? item.InnerText : "";

                            if (EmojiCache.ContainsKey(md5))
                                continue;

                            if (url == "")
                                continue;

                            else
                            {
                                string path = Path.Combine(UserBakConfig.UserWorkspacePath, "Emoji", md5 + ".jpg");
                                try
                                {
                                    HttpResponseMessage res = httpClient.GetAsync(url).Result;
                                    if (res.IsSuccessStatusCode)
                                    {
                                        using (FileStream fs = File.Create(path))
                                        {
                                            res.Content.ReadAsStream().CopyTo(fs);
                                        }
                                    }
                                }
                                catch (Exception ex)
                                {

                                }
                            }
                        }
                    }
                }
                catch (Exception ex)
                {

                }
            }
        }

        public byte[]? GetHeadImgCahce(string username)
        {
            if (HeadImgCache.ContainsKey(username))
            {
                ContactHeadImg? img = HeadImgCache[username] as ContactHeadImg;
                if (img == null)
                    return null;
                else
                    return img.smallHeadBuf;
            }
            return null;
        }

        public int[] GetWXCount()
        {
            SQLiteConnection? con = getCon("MicroMsg");
            if (con == null)
                return new int[] { 0, 0 };

            string query = @"select count(*) as count from contact where type != 4";
            int userCount = con.Query<WXCount>(query)[0].Count;

            int msgCount = 0;
            for (int i = 0; i <= 99; i++)
            {
                con = getCon("MSG" + i.ToString());
                if (con == null)
                    return new int[] { userCount, msgCount };

                query = "select count(*) as count from MSG";
                msgCount += con.Query<WXCount>(query)[0].Count;
            }
            return new int[] { userCount, msgCount };
        }

        public ObservableCollection<WXContact> GetWXContacts(string? name = null,bool all = false)
        {
            SQLiteConnection? con = getCon("MicroMsg");
            if (con == null)
                return new ObservableCollection<WXContact>();
            string query = @"select contact.*,session.strContent,contactHeadImgUrl.smallHeadImgUrl,contactHeadImgUrl.bigHeadImgUrl from contact 
            left join session on session.strUsrName = contact.username
            left join contactHeadImgUrl on contactHeadImgUrl.usrName = contact.username
            where type != 4 {searchName}
            order by nOrder desc";

            if (all)
            {
                query = query.Replace("where type != 4 ", "");
            }

            List<WXContact>? contacts = null;
            if (name != null)
            {
                query = query.Replace("{searchName}", " and (username like ? or alias like ? or nickname like ? or remark like ?)");
                contacts = con.Query<WXContact>(query, $"%{name}%", $"%{name}%", $"%{name}%", $"%{name}%");
            }
            else
            {
                query = query.Replace("{searchName}", "");
                contacts = con.Query<WXContact>(query);
            }

            foreach (WXContact contact in contacts)
            {
                if(contact.Remark != "")
                    contact.NickName = contact.Remark;

                byte[]? imgBytes = GetHeadImgCahce(contact.UserName);
                if (imgBytes != null)
                {
                    try
                    {
                        MemoryStream stream = new MemoryStream(imgBytes);
                        BitmapImage bitmapImage = new BitmapImage();
                        bitmapImage.BeginInit();
                        bitmapImage.CreateOptions = BitmapCreateOptions.IgnoreColorProfile;
                        bitmapImage.StreamSource = stream;
                        bitmapImage.EndInit();
                        bitmapImage.Freeze();
                        contact.Avatar = bitmapImage;
                    }
                    catch
                    {
#if DEBUG
                        File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}\n", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"), "BitmapConvert Err=>Length", imgBytes.Length));
#endif
                    }
                }
                else
                    continue;
            }

            return new ObservableCollection<WXContact>(contacts);
        }

        public List<WXUserImg>? GetUserImgs()
        {
            SQLiteConnection? con = getCon("MicroMsg");
            if (con == null)
                return null;
            string query = "select * from contactHeadImgUrl";
            return con.Query<WXUserImg>(query);
        }

        public List<WXChatRoom>? GetWXChatRooms()
        {
            SQLiteConnection? con = getCon("MicroMsg");
            if (con == null)
                return null;
            string query = "select * from ChatRoom";
            return con.Query<WXChatRoom>(query);
        }

        public List<WXMsg> GetTypeMsg(string type)
        {
            List<WXMsg> tmp = new List<WXMsg>();
            for (int i = 0; i <= 99; i++)
            {
                SQLiteConnection? con = getCon("MSG" + i.ToString());
                if (con == null)
                    return tmp;

                string query = "select * from MSG where Type=?";
                List<WXMsg> wXMsgs = con.Query<WXMsg>(query, type);
                tmp.AddRange(wXMsgs);
            }
            return tmp;
        }
        public List<WXMsg>? GetWXMsgs(string uid,string msg = "")
        {
            List<WXMsg> tmp = new List<WXMsg>();
            for (int i = 0; i <= 99; i++)
            {
                SQLiteConnection? con = getCon("MSG" + i.ToString());
                if (con == null)
                    return tmp;

                List<WXMsg>? wXMsgs = null;
                if (msg == "")
                {
                    string query = "select * from MSG where StrTalker=?";
                    wXMsgs = con.Query<WXMsg>(query, uid);
                }
                else if (uid == "")
                {
                    string query = "select * from MSG where StrContent like ?";
                    wXMsgs = con.Query<WXMsg>(query, string.Format("%{0}%", msg));
                }
                else
                {
                    string query = "select * from MSG where StrTalker=? and StrContent like ?";
                    wXMsgs = con.Query<WXMsg>(query, uid, string.Format("%{0}%", msg));
                }

                foreach (WXMsg w in wXMsgs)
                {
                    if (UserNameCache.ContainsKey(w.StrTalker))
                    {
                        WXContact? contact = UserNameCache[w.StrTalker] as WXContact;
                        if (contact != null)
                        {
                            if (contact.Remark != "")
                                w.NickName = contact.Remark;
                            else
                                w.NickName = contact.NickName;
                        }

                    }

                    if (uid.Contains("@chatroom"))
                    {
                        string userId = "";

                        if (w.BytesExtra == null)
                            continue;

                        string sl = BitConverter.ToString(w.BytesExtra).Replace("-", "");

                        ProtoMsg protoMsg;
                        using (MemoryStream stream = new MemoryStream(w.BytesExtra))
                        {
                            protoMsg = ProtoBuf.Serializer.Deserialize<ProtoMsg>(stream);
                        }

                        if (protoMsg.TVMsg != null)
                        {
                            foreach (TVType _tmp in protoMsg.TVMsg)
                            {
                                if (_tmp.Type == 1)
                                    userId = _tmp.TypeValue;
                            }
                        }


                        if (!w.IsSender)
                        {
                            if (UserNameCache.ContainsKey(userId))
                            {
                                WXContact? contact = UserNameCache[userId] as WXContact;
                                if (contact != null)
                                    w.NickName = contact.Remark == "" ? contact.NickName : contact.Remark;
                            }
                            else
                            {
                                w.NickName = userId;
                            }
                        }
                    }
                    if (w.IsSender)
                        w.NickName = "我";

                    if (w.Type != 1)
                    {
                        if (w.Type == 10000)
                        {
                            w.Type = 1;
                            w.NickName = "系统消息";
                            w.StrContent = w.StrContent.Replace("<revokemsg>", "").Replace("</revokemsg>", "");
                        }
                        else
                        {
                            w.StrContent = "[界面未支持格式]Type=" + w.Type;
                        }
                    }
                    tmp.Add(w);

                }
            }
            return tmp;
        }
        public List<WXSessionAttachInfo>? GetWXMsgAtc()
        {
            SQLiteConnection? con = getCon("MultiSearchChatMsg");
            if (con == null)
                return null;

            string query = "select * from SessionAttachInfo";
            List<WXSessionAttachInfo> list = con.Query<WXSessionAttachInfo>(query);
            if (list.Count != 0)
            {
                return list;
            }
            else
                return null;
        }
        public WXSessionAttachInfo? GetWXMsgAtc(WXMsg msg)
        {
            SQLiteConnection? con = getCon("MultiSearchChatMsg");
            if (con == null)
                return null;

            string query = "select * from SessionAttachInfo where msgId = ? order by attachsize desc";
            List<WXSessionAttachInfo> list = con.Query<WXSessionAttachInfo>(query, msg.MsgSvrID);
            if (list.Count != 0)
            {
                //部分附件可能有md5校验，这里移除校验，给的是正确路径
                WXSessionAttachInfo acc = list[0];
                int index = acc.attachPath.IndexOf(".dat");
                int index2 = acc.attachPath.IndexOf(".dat");
                if (acc.attachPath.Length - index > 10 && index != -1)
                {
                    acc.attachPath = acc.attachPath.Substring(0, acc.attachPath.Length - 32);
                }
                if (acc.attachPath.Length - index2 > 10 && index2 != -1)
                {
                    acc.attachPath = acc.attachPath.Substring(0, acc.attachPath.Length - 32);
                }
                return acc;
            }
            else
                return null;
        }
        public WXMediaMsg? GetVoiceMsg(WXMsg msg)
        {
            for (int i = 0; i <= 99; i++)
            {
                SQLiteConnection? con = getCon("MediaMSG" + i.ToString());
                if (con == null)
                    continue;

                string query = "select * from Media where Reserved0=?";
                List<WXMediaMsg> wXMsgs = con.Query<WXMediaMsg>(query, msg.MsgSvrID);
                if (wXMsgs.Count != 0)
                    return wXMsgs[0];
            }
            return null;
        }
        public string? GetAttachment(WXMsgType type, WXMsg msg)
        {
            if (UserBakConfig == null)
                return null;

            string? tmpPath = Path.Combine(UserBakConfig.UserWorkspacePath, "Temp");
            if (!Directory.Exists(tmpPath))
                Directory.CreateDirectory(tmpPath);

            // 如果是图片和视频，从附件库中搜索
            string? path = null;
            if (type == WXMsgType.Image || type == WXMsgType.Video)
            {
                WXSessionAttachInfo? atcInfo = GetWXMsgAtc(msg);
                if (atcInfo == null)
                    return null;
                path = atcInfo.attachPath;
            }
            // 如果是从语音，从媒体库查找
            else if (type == WXMsgType.Audio)
            {
                WXMediaMsg? voiceMsg = GetVoiceMsg(msg);
                if (voiceMsg == null)
                    return null;
                if (voiceMsg.Buf == null)
                    return null;

                // 从DB取音频文件到临时目录
                string tmp_file_path = Path.Combine(tmpPath, voiceMsg.Key + ".arm");
                using (FileStream stream = new FileStream(tmp_file_path, FileMode.OpenOrCreate))
                {
                    stream.Write(voiceMsg.Buf, 0, voiceMsg.Buf.Length);
                }
                path = tmp_file_path;
            }

            if (path == null)
                return null;

            // 获取到原路径后，开始进行解密转移,只有图片和语音需要解密，解密后是直接归档目录
            if(type == WXMsgType.Image || type== WXMsgType.Audio)
            {
                path = DecryptAttachment(type, path);
            }
            else if (type == WXMsgType.Video)
            {
                string video_dir = Path.Combine(UserBakConfig.UserWorkspacePath, "Video");
                if(!Directory.Exists(video_dir))
                    Directory.CreateDirectory(video_dir);
                FileInfo fileInfo = new FileInfo(path);
                // 目标视频路径
                string video_file_path = Path.Combine(video_dir, fileInfo.Name);
                // 视频的路径是相对路径，需要加上资源目录
                path = Path.Combine(UserBakConfig.UserResPath, path);
                // 原文件存在，目标不存在
                if (!File.Exists(video_file_path) && File.Exists(path))
                {
                    // 复制
                    File.Copy(path, video_file_path);
                    path = video_file_path;
                }
                else if (File.Exists(video_file_path))
                {
                    path = video_file_path;
                }
                else
                    return null;
                    
            }

            if (path == null)
                return null;

            // 改相对路径
            path = path.Replace(UserBakConfig.UserWorkspacePath + "\\", "");
            return path;

        }
        public string? DecryptAttachment(WXMsgType type, string path)
        {
            if (UserBakConfig == null)
                return null;

            string? file_path = null;
            switch (type)
            {
                case WXMsgType.Image:
                    string img_dir = Path.Combine(UserBakConfig.UserWorkspacePath, "Image");
                    if (!Directory.Exists(img_dir))
                        Directory.CreateDirectory(img_dir);
                    // 图片的路径是相对路径，需要加上资源目录
                    path = Path.Combine(UserBakConfig.UserResPath, path);
                    byte[] decFileByte = DecryptionHelper.DecImage(path);
                    string decFiletype = DecryptionHelper.CheckFileType(decFileByte);
                    file_path = DecryptionHelper.SaveDecImage(decFileByte, path, img_dir, decFiletype);
                    break;
                case WXMsgType.Audio:
                    string audio_dir = Path.Combine(UserBakConfig.UserWorkspacePath, "Audio");
                    if (!Directory.Exists(audio_dir))
                        Directory.CreateDirectory(audio_dir);
                    FileInfo fileInfo = new FileInfo(path);
                    string audio_file_dir = Path.Combine(audio_dir, fileInfo.Name + ".mp3");
                    ToolsHelper.DecodeVoice(path, path + ".pcm", audio_file_dir);
                    file_path = audio_file_dir;
                    break;
            }
            return file_path;
        }
        public List<WXMsgGroup> GetWXMsgGroup()
        {
            List<WXMsgGroup> g = new List<WXMsgGroup>();
            for (int i = 0; i <= 99; i++)
            {
                SQLiteConnection? con = getCon("MSG" + i.ToString());
                if (con == null)
                    return g;

                string query = "select StrTalker,Count(localId) as MsgCount from MSG GROUP BY StrTalker";
                List<WXMsgGroup> wXMsgs = con.Query<WXMsgGroup>(query);
                foreach (WXMsgGroup w in wXMsgs)
                {
                    WXMsgGroup? tmp = g.Find(x => x.UserName == w.UserName);
                    if (tmp == null)
                        g.Add(w);
                    else
                        tmp.MsgCount += g.Count;
                }

            }
            return g;
        }
    }

    public enum WXMsgType
    {
        Image = 0,
        Video = 1,
        Audio = 2,
        File = 3,
    }
}