Workspace.xaml.cs 9.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257
  1. using JiebaNet.Segmenter;
  2. using JiebaNet.Segmenter.Common;
  3. using System;
  4. using System.Collections;
  5. using System.Collections.Generic;
  6. using System.ComponentModel;
  7. using System.Diagnostics;
  8. using System.IO;
  9. using System.Linq;
  10. using System.Text;
  11. using System.Threading.Tasks;
  12. using System.Windows;
  13. using System.Windows.Controls.Primitives;
  14. using System.Windows.Data;
  15. using System.Windows.Documents;
  16. using System.Windows.Input;
  17. using System.Windows.Media;
  18. using System.Windows.Media.Imaging;
  19. using System.Windows.Navigation;
  20. using WechatBakTool.Export;
  21. using WechatBakTool.Helpers;
  22. using WechatBakTool.Model;
  23. using WechatBakTool.ViewModel;
  24. using WordCloudSharp;
  25. using System.Drawing;
  26. using System.Windows.Controls;
  27. using System.Text.RegularExpressions;
  28. namespace WechatBakTool.Pages
  29. {
  30. /// <summary>
  31. /// Workspace.xaml 的交互逻辑
  32. /// </summary>
  33. public partial class Workspace : Page
  34. {
  35. public WXUserReader? UserReader;
  36. private WorkspaceViewModel ViewModel = new WorkspaceViewModel();
  37. public Workspace()
  38. {
  39. ViewModel.ExportItems = new System.Collections.ObjectModel.ObservableCollection<ExportItem> {
  40. new ExportItem(){ Name="导出为HTML",Value=1 },
  41. new ExportItem(){ Name="导出为TXT",Value=2 },
  42. new ExportItem(){ Name="与他的词云",Value=3 },
  43. };
  44. ViewModel.SelectExportItem = ViewModel.ExportItems[0];
  45. InitializeComponent();
  46. list_users.Items.Clear();
  47. DataContext = ViewModel;
  48. UserBakConfig? config = Main2.CurrentUserBakConfig;
  49. if (config != null)
  50. {
  51. UserReader = new WXUserReader(config);
  52. if (config.Decrypt)
  53. {
  54. ViewModel.Contacts = UserReader.GetWXContacts();
  55. }
  56. }
  57. }
  58. private void btn_read_Click(object sender, RoutedEventArgs e)
  59. {
  60. if (Main2.CurrentUserBakConfig == null)
  61. {
  62. MessageBox.Show("工作区配置加载失败,请检查配置文件是否正常","错误");
  63. return;
  64. }
  65. }
  66. private void list_users_SelectionChanged(object sender, SelectionChangedEventArgs e)
  67. {
  68. ViewModel.ExportCount = "";
  69. ViewModel.WXContact = list_users.SelectedItem as WXContact;
  70. if(ViewModel.WXContact == null || UserReader == null)
  71. {
  72. return;
  73. }
  74. List<WXMsg>? msgs = UserReader.GetWXMsgs(ViewModel.WXContact.UserName);
  75. list_msg.ItemsSource = msgs;
  76. }
  77. private void txt_find_user_TextChanged(object sender, TextChangedEventArgs e)
  78. {
  79. if (UserReader == null)
  80. return;
  81. string findName = txt_find_user.Text;
  82. if (txt_find_user.Text == "搜索...")
  83. findName = "";
  84. ViewModel.Contacts = UserReader.GetWXContacts(findName);
  85. }
  86. private void txt_find_user_GotFocus(object sender, RoutedEventArgs e)
  87. {
  88. if (txt_find_user.Text == "搜索...")
  89. txt_find_user.Text = "";
  90. Debug.WriteLine(ViewModel.SearchString);
  91. }
  92. private void btn_export_Click(object sender, RoutedEventArgs e)
  93. {
  94. if(ViewModel.WXContact == null || UserReader == null)
  95. {
  96. MessageBox.Show("请选择联系人", "错误");
  97. return;
  98. }
  99. try
  100. {
  101. string path = Path.Combine(Main2.CurrentUserBakConfig!.UserWorkspacePath, ViewModel.WXContact.UserName + ".txt");
  102. IExport export = new TXTExport();
  103. export.InitTemplate(ViewModel.WXContact, path);
  104. export.SetMsg(UserReader, ViewModel.WXContact, ViewModel);
  105. export.SetEnd();
  106. export.Save(path);
  107. }catch(Exception ex)
  108. {
  109. File.AppendAllText("1.log", ex.Message);
  110. MessageBox.Show(ex.Message);
  111. }
  112. MessageBox.Show("导出完成");
  113. }
  114. private void btn_open_workspace_Click(object sender, RoutedEventArgs e)
  115. {
  116. Process.Start("explorer.exe ", Main2.CurrentUserBakConfig!.UserWorkspacePath);
  117. }
  118. private void Export_Click(object sender, RoutedEventArgs e)
  119. {
  120. Task.Run(() =>
  121. {
  122. if (ViewModel.WXContact == null || UserReader == null)
  123. {
  124. MessageBox.Show("请选择联系人", "错误");
  125. return;
  126. }
  127. if (ViewModel.SelectExportItem == null)
  128. {
  129. MessageBox.Show("请选择导出方式", "错误");
  130. return;
  131. }
  132. if(ViewModel.SelectExportItem.Value == 3)
  133. {
  134. if(UserReader != null && ViewModel.WXContact != null)
  135. {
  136. WordCloudSettingViewModel setting = new WordCloudSettingViewModel() {
  137. ImgWidth = "1000",
  138. ImgHeight = "1000",
  139. EnableRemoveOneKey = true,
  140. };
  141. Dispatcher.Invoke(() => {
  142. WordCloudSetting wordCloudSetting = new WordCloudSetting(setting);
  143. wordCloudSetting.ShowDialog();
  144. });
  145. var jieba = new JiebaSegmenter();
  146. Counter<string> counter = new Counter<string>();
  147. ViewModel.ExportCount = "词频统计ing...";
  148. List<WXMsg>? msgs = UserReader.GetWXMsgs(ViewModel.WXContact.UserName);
  149. if(msgs!= null)
  150. {
  151. foreach(WXMsg msg in msgs)
  152. {
  153. if(msg.Type == 1)
  154. {
  155. List<string> list = jieba.Cut(msg.StrContent).ToList();
  156. counter.Add(list);
  157. }
  158. }
  159. }
  160. var orderBy = counter.MostCommon();
  161. ViewModel.ExportCount = "移除部分词语...";
  162. string[] remove_string_list = setting.RemoveKey.Split(",");
  163. foreach(string remove_string in remove_string_list)
  164. {
  165. counter.Remove(remove_string);
  166. }
  167. foreach(var key in orderBy)
  168. {
  169. if (key.Key.Length == 1 && setting.EnableRemoveOneKey)
  170. counter.Remove(key.Key);
  171. }
  172. ViewModel.ExportCount = "渲染词云结果";
  173. string resultPath = "result.jpg";
  174. var wordCloud = new WordCloud(int.Parse(setting.ImgWidth), int.Parse(setting.ImgHeight), allowVerical: true, fontname: setting.Font);
  175. if(orderBy.Count() >= setting.MaxKeyCount)
  176. orderBy = orderBy.Take(setting.MaxKeyCount);
  177. //var wordCloud = new WordCloud(1000, 1000,false, null,-1,1,null, false);
  178. var result = wordCloud.Draw(orderBy.Select(it => it.Key).ToList(), orderBy.Select(it => it.Value).ToList());
  179. result.Save(resultPath);
  180. ViewModel.ExportCount = "完成";
  181. MessageBox.Show("生成完毕", "提示");
  182. }
  183. return;
  184. }
  185. string name = ViewModel.WXContact.NickName;
  186. name = name.Replace(@"\", "");
  187. name = Regex.Replace(name, "[ \\[ \\] \\^ \\-_*×――(^)$%~!/@#$…&%¥—+=<>《》|!!???::•`·、。,;,.;\"‘’“”-]", "");
  188. string path = Path.Combine(
  189. Main2.CurrentUserBakConfig!.UserWorkspacePath,
  190. string.Format(
  191. "{0}-{1}",
  192. ViewModel.WXContact.UserName,
  193. ViewModel.WXContact.Remark == "" ? name : ViewModel.WXContact.Remark
  194. )
  195. );
  196. IExport export;
  197. #if DEBUG
  198. Stopwatch stopwatch = new Stopwatch();
  199. stopwatch.Start();
  200. #endif
  201. if (ViewModel.SelectExportItem.Value == 2)
  202. {
  203. path += ".txt";
  204. export = new TXTExport();
  205. }
  206. else
  207. {
  208. path += ".html";
  209. export = new HtmlExport();
  210. }
  211. export.InitTemplate(ViewModel.WXContact, path);
  212. export.SetMsg(UserReader, ViewModel.WXContact, ViewModel);
  213. export.SetEnd();
  214. export.Save(path);
  215. #if DEBUG
  216. stopwatch.Stop();
  217. MessageBox.Show(stopwatch.Elapsed.ToString());
  218. #endif
  219. MessageBox.Show("导出完成", "提示");
  220. });
  221. }
  222. private void btn_pre_emoji_Click(object sender, RoutedEventArgs e)
  223. {
  224. if(UserReader != null && ViewModel.WXContact != null)
  225. {
  226. Task.Run(() => {
  227. UserReader.PreDownloadEmoji(ViewModel.WXContact.UserName);
  228. MessageBox.Show("用户所有表情预下载完毕");
  229. });
  230. }
  231. }
  232. }
  233. }