WM
/
WechatBakTool
зеркало из https://github.com/SuxueCode/WechatBakTool


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186
							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;

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 = "<!DOCTYPE html><html><head><meta charset=\"utf-8\"><title>WechatBakTool</title><style>p{margin:0px;}.msg{padding-bottom:10px;}.nickname{font-size:10px;}.content{font-size:14px;}</style></head><body>";

            HtmlBody += string.Format("<div class=\"msg\"><p class=\"nickname\"><b>与 {0}({1}) 的聊天记录</b></p>", Session.NickName, Session.UserName);
            HtmlBody += string.Format("<div class=\"msg\"><p class=\"nickname\"><b>导出时间：{0}</b></p><hr/>", 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 += "</body></html>";
            File.AppendAllText(Path, HtmlBody);
        }

        public void SetMsg(WXUserReader reader, WXContact contact,WorkspaceViewModel viewModel)
        {
            if (Session == null)
                throw new Exception("请初始化模版：Not Use InitTemplate");

            List<WXMsg>? 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("<div class=\"msg\"><p class=\"nickname\">{0} <span style=\"padding-left:10px;\">{1}</span></p>", msg.IsSender ? "我" : msg.NickName, TimeStampToDateTime(msg.CreateTime).ToString("yyyy-MM-dd HH:mm:ss"));

                if (msg.Type == 1)
                    HtmlBody += string.Format("<p class=\"content\">{0}</p></div>", 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}", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"), "Img Error Path=>", path));
                        File.AppendAllText("debug.log", string.Format("[D]{0} {1}:{2}", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"),"Img Error Msg=>",JsonConvert.SerializeObject(msg)));
#endif
                        HtmlBody += string.Format("<p class=\"content\">{0}</p></div>", "图片转换出现错误或文件不存在");
                        continue;
                    }
                    HtmlBody += string.Format("<p class=\"content\"><img src=\"{0}\" style=\"max-height:1000px;max-width:1000px;\"/></p></div>", path);
                }
                else if (msg.Type == 43)
                {
                    string? path = reader.GetAttachment(WXMsgType.Video, msg);
                    if (path == null)
                    {
                        HtmlBody += string.Format("<p class=\"content\">{0}</p></div>", "视频不存在");
                        continue;
                    }
                    HtmlBody += string.Format("<p class=\"content\"><video controls style=\"max-height:300px;max-width:300px;\"><source src=\"{0}\" type=\"video/mp4\" /></video></p></div>", path);
                }
                else if (msg.Type == 49)
                {
                    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("<p class=\"content\">{0}|{1}</p><p><a href=\"{2}\">点击访问</a></p></div>", appName, title, url);

                            }
                        }
                    }
                }
                else if (msg.Type == 34)
                {
                    string? path = reader.GetAttachment(WXMsgType.Audio, msg);
                    if (path == null)
                    {
                        HtmlBody += string.Format("<p class=\"content\">{0}</p></div>", "语音不存在");
                        continue;
                    }
                    HtmlBody += string.Format("<p class=\"content\"><audio controls src=\"{0}\"></audio></p></div>", path);
                }
                else
                {
                    HtmlBody += string.Format("<p class=\"content\">{0}</p></div>", "暂未支持的消息");
                }
                
                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;
        }
    }
}