WechatBakTool

using K4os.Compression.LZ4.Encoders; using K4os.Compression.LZ4; using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using System.Threading.Tasks; using WechatBakTool.Model; using System.Xml; using Newtonsoft.Json; using WechatBakTool.ViewModel; using System.Security.Policy; namespace WechatBakTool.Export { public class HtmlExport : IExport { private string HtmlBody = ""; private WXSession? Session = null; private string Path = ""; public void InitTemplate(WXSession session) { Session = session; HtmlBody = "WechatBakTool"; HtmlBody += string.Format("

与 {0}({1}) 的聊天记录

", Session.NickName, Session.UserName); HtmlBody += string.Format("

导出时间：{0}

", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")); File.WriteAllText(Path, HtmlBody); } public void InitTemplate(WXContact contact, string p) { Path = p; WXSession session = new WXSession(); session.NickName = contact.NickName; session.UserName = contact.UserName; InitTemplate(session); } public void Save(string path = "") { } public void SetEnd() { HtmlBody += ""; File.AppendAllText(Path, HtmlBody); } public void SetMsg(WXUserReader reader, WXContact contact,WorkspaceViewModel viewModel) { if (Session == null) throw new Exception("请初始化模版：Not Use InitTemplate"); List? msgList = reader.GetWXMsgs(contact.UserName); if (msgList == null) throw new Exception("获取消息失败，请确认数据库读取正常"); msgList.Sort((x, y) => x.CreateTime.CompareTo(y.CreateTime)); int msgCount = 0; HtmlBody = ""; StreamWriter streamWriter = new StreamWriter(Path, true); foreach (var msg in msgList) { HtmlBody += string.Format("

{0} {1}

", msg.IsSender ? "我" : msg.NickName, TimeStampToDateTime(msg.CreateTime).ToString("yyyy-MM-dd HH:mm:ss")); if (msg.Type == 1) HtmlBody += string.Format("

{0}

", msg.StrContent); else if (msg.Type == 3) { string? path = reader.GetAttachment(WXMsgType.Image, msg); if (path == null) { #if DEBUG File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}\n", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"), "Img Error Path=>", path)); File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}\n", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"),"Img Error Msg=>",JsonConvert.SerializeObject(msg))); #endif HtmlBody += string.Format("

{0}

", "图片转换出现错误或文件不存在"); continue; } HtmlBody += string.Format("

", path); } else if (msg.Type == 43) { string? path = reader.GetAttachment(WXMsgType.Video, msg); if (path == null) { HtmlBody += string.Format("

{0}

", "视频不存在"); continue; } HtmlBody += string.Format("

", path); } else if(msg.Type == 47) { string? path = reader.GetAttachment(WXMsgType.Emoji, msg); if (path == null) { #if DEBUG File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}\n", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"), "Emoji Error Path=>", path)); File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}\n", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"), "Emoji Error Msg=>", JsonConvert.SerializeObject(msg))); #endif HtmlBody += string.Format("

{0}

", "表情未预下载或加密表情"); continue; } HtmlBody += string.Format("

", path); } else if (msg.Type == 49) { if(msg.SubType == 6||msg.SubType == 19||msg.SubType == 40) { string? path = reader.GetAttachment(WXMsgType.File, msg); if(path == null) { HtmlBody += string.Format("

{0}

", "文件不存在"); continue; } else { HtmlBody += string.Format("

{0}

", "文件：" + path, path); } } else { using (var decoder = LZ4Decoder.Create(true, 64)) { byte[] target = new byte[10240]; int res = 0; if (msg.CompressContent != null) res = LZ4Codec.Decode(msg.CompressContent, 0, msg.CompressContent.Length, target, 0, target.Length); byte[] data = target.Skip(0).Take(res).ToArray(); string xml = Encoding.UTF8.GetString(data); if (!string.IsNullOrEmpty(xml)) { xml = xml.Replace("\n", ""); XmlDocument xmlObj = new XmlDocument(); xmlObj.LoadXml(xml); if (xmlObj.DocumentElement != null) { string title = ""; string appName = ""; string url = ""; XmlNodeList? findNode = xmlObj.DocumentElement.SelectNodes("/msg/appmsg/title"); if (findNode != null) { if (findNode.Count > 0) { title = findNode[0]!.InnerText; } } findNode = xmlObj.DocumentElement.SelectNodes("/msg/appmsg/sourcedisplayname"); if (findNode != null) { if (findNode.Count > 0) { appName = findNode[0]!.InnerText; } } findNode = xmlObj.DocumentElement.SelectNodes("/msg/appmsg/url"); if (findNode != null) { if (findNode.Count > 0) { url = findNode[0]!.InnerText; } } HtmlBody += string.Format("

{0}|{1}

", appName, title, url); } } } } } else if (msg.Type == 34) { string? path = reader.GetAttachment(WXMsgType.Audio, msg); if (path == null) { HtmlBody += string.Format("

{0}

", "语音不存在"); continue; } HtmlBody += string.Format("

", path); } else { HtmlBody += string.Format("

{0}

", "暂未支持的消息"); } msgCount++; if(msgCount % 50 == 0) { streamWriter.WriteLine(HtmlBody); HtmlBody = ""; viewModel.ExportCount = msgCount.ToString(); } } if(msgCount % 50 != 0) { streamWriter.WriteLine(HtmlBody); HtmlBody = ""; viewModel.ExportCount = msgCount.ToString(); } streamWriter.Close(); streamWriter.Dispose(); } private static DateTime TimeStampToDateTime(long timeStamp, bool inMilli = false) { DateTimeOffset dateTimeOffset = inMilli ? DateTimeOffset.FromUnixTimeMilliseconds(timeStamp) : DateTimeOffset.FromUnixTimeSeconds(timeStamp); return dateTimeOffset.LocalDateTime; } } }