Improve performance (#162)

2026-03-02 05:03:42 +00:00 · 2019-04-10 23:45:21 +03:00
parent 359278afec
commit 4bfb2ec7fd
86 changed files with 1242 additions and 900 deletions
--- a/DiscordChatExporter.Core.Rendering/PlainTextChatLogRenderer.cs
+++ b/DiscordChatExporter.Core.Rendering/PlainTextChatLogRenderer.cs
@@ -0,0 +1,128 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.IO;
+using System.Linq;
+using System.Threading.Tasks;
+using DiscordChatExporter.Core.Markdown;
+using DiscordChatExporter.Core.Markdown.Nodes;
+using DiscordChatExporter.Core.Models;
+using Tyrrrz.Extensions;
+
+namespace DiscordChatExporter.Core.Rendering
+{
+    public class PlainTextChatLogRenderer : IChatLogRenderer
+    {
+        private readonly ChatLog _chatLog;
+        private readonly string _dateFormat;
+
+        public PlainTextChatLogRenderer(ChatLog chatLog, string dateFormat)
+        {
+            _chatLog = chatLog;
+            _dateFormat = dateFormat;
+        }
+
+        private string FormatDate(DateTime date) => date.ToString(_dateFormat, CultureInfo.InvariantCulture);
+
+        private string FormatDateRange(DateTime? from, DateTime? to)
+        {
+            // Both 'from' and 'to'
+            if (from.HasValue && to.HasValue)
+                return $"{FormatDate(from.Value)} to {FormatDate(to.Value)}";
+
+            // Just 'from'
+            if (from.HasValue)
+                return $"after {FormatDate(from.Value)}";
+
+            // Just 'to'
+            if (to.HasValue)
+                return $"before {FormatDate(to.Value)}";
+
+            // Neither
+            return null;
+        }
+
+        private string FormatMarkdown(Node node)
+        {
+            // Formatted node
+            if (node is FormattedNode formattedNode)
+            {
+                // Recursively get inner text
+                var innerText = FormatMarkdown(formattedNode.Children);
+
+                return $"{formattedNode.Token}{innerText}{formattedNode.Token}";
+            }
+
+            // Non-meta mention node
+            if (node is MentionNode mentionNode && mentionNode.Type != MentionType.Meta)
+            {
+                // User mention node
+                if (mentionNode.Type == MentionType.User)
+                {
+                    var user = _chatLog.Mentionables.GetUser(mentionNode.Id);
+                    return $"@{user.Name}";
+                }
+
+                // Channel mention node
+                if (mentionNode.Type == MentionType.Channel)
+                {
+                    var channel = _chatLog.Mentionables.GetChannel(mentionNode.Id);
+                    return $"#{channel.Name}";
+                }
+
+                // Role mention node
+                if (mentionNode.Type == MentionType.Role)
+                {
+                    var role = _chatLog.Mentionables.GetRole(mentionNode.Id);
+                    return $"@{role.Name}";
+                }
+            }
+
+            // Custom emoji node
+            if (node is EmojiNode emojiNode && emojiNode.IsCustomEmoji)
+            {
+                return $":{emojiNode.Name}:";
+            }
+
+            // All other nodes - simply return source
+            return node.Source;
+        }
+
+        private string FormatMarkdown(IEnumerable<Node> nodes) => nodes.Select(FormatMarkdown).JoinToString("");
+
+        private string FormatMarkdown(string markdown) => FormatMarkdown(MarkdownParser.Parse(markdown));
+
+        private async Task RenderMessageAsync(TextWriter writer, Message message)
+        {
+            // Timestamp and author
+            await writer.WriteLineAsync($"[{FormatDate(message.Timestamp)}] {message.Author.FullName}");
+
+            // Content
+            await writer.WriteLineAsync(FormatMarkdown(message.Content));
+
+            // Attachments
+            foreach (var attachment in message.Attachments)
+                await writer.WriteLineAsync(attachment.Url);
+        }
+
+        public async Task RenderAsync(TextWriter writer)
+        {
+            // Metadata
+            await writer.WriteLineAsync('='.Repeat(62));
+            await writer.WriteLineAsync($"Guild: {_chatLog.Guild.Name}");
+            await writer.WriteLineAsync($"Channel: {_chatLog.Channel.Name}");
+            await writer.WriteLineAsync($"Topic: {_chatLog.Channel.Topic}");
+            await writer.WriteLineAsync($"Messages: {_chatLog.Messages.Count:N0}");
+            await writer.WriteLineAsync($"Range: {FormatDateRange(_chatLog.From, _chatLog.To)}");
+            await writer.WriteLineAsync('='.Repeat(62));
+            await writer.WriteLineAsync();
+
+            // Log
+            foreach (var message in _chatLog.Messages)
+            {
+                await RenderMessageAsync(writer, message);
+                await writer.WriteLineAsync();
+            }
+        }
+    }
+}