feat: 美化文章界面 (#3)

Reviewed-on: #3

YaeBlog.Core/Processors/CodeBlockPostRenderProcessor.cs

@@ -0,0 +1,29 @@
+using AngleSharp;
+using AngleSharp.Dom;
+using YaeBlog.Core.Abstractions;
+using YaeBlog.Core.Models;
+
+namespace YaeBlog.Core.Processors;
+
+public class CodeBlockPostRenderProcessor : IPostRenderProcessor
+{
+    public async Task<BlogEssay> ProcessAsync(BlogEssay essay)
+    {
+        BrowsingContext context = new(Configuration.Default);
+        IDocument document = await context.OpenAsync(
+            req => req.Content(essay.HtmlContent));
+
+        IEnumerable<IElement> preElements = from e in document.All
+            where e.LocalName == "pre"
+            select e;
+
+        foreach (IElement element in preElements)
+        {
+            element.ClassList.Add("p-3 text-bg-secondary rounded-1");
+        }
+
+        return essay.WithNewHtmlContent(document.DocumentElement.OuterHtml);
+    }
+
+    public string Name => nameof(CodeBlockPostRenderProcessor);
+}

YaeBlog.Core/Processors/HeadlinePostRenderProcessor.cs

@@ -0,0 +1,105 @@
+using AngleSharp;
+using AngleSharp.Dom;
+using YaeBlog.Core.Abstractions;
+using YaeBlog.Core.Models;
+using YaeBlog.Core.Services;
+
+namespace YaeBlog.Core.Processors;
+
+public class HeadlinePostRenderProcessor(
+    IConfiguration angleConfiguration,
+    TableOfContentService tableOfContentService) : IPostRenderProcessor
+{
+    public async Task<BlogEssay> ProcessAsync(BlogEssay essay)
+    {
+        BrowsingContext browsingContext = new(angleConfiguration);
+        IDocument document = await browsingContext.OpenAsync(req => req.Content(essay.HtmlContent));
+
+        IEnumerable<IElement> elements = from item in document.All
+            where item.LocalName is "h2" or "h3" or "h4"
+            select item;
+
+        BlogHeadline topHeadline = new(essay.Title, "#title");
+        List<BlogHeadline> level2List = [];
+        List<BlogHeadline> level3List = [];
+        List<BlogHeadline> level4List = [];
+
+        foreach (IElement element in elements)
+        {
+            switch (element.LocalName)
+            {
+                case "h2":
+                    {
+                        FindParentHeadline(topHeadline, level2List, level3List).Children.AddRange(level4List);
+                        level4List.Clear();
+                        FindParentHeadline(topHeadline, level2List).Children.AddRange(level3List);
+                        level3List.Clear();
+
+                        BlogHeadline headline = ParserHeadlineElement(element);
+                        level2List.Add(headline);
+                        break;
+                    }
+                case "h3":
+                    {
+                        FindParentHeadline(topHeadline, level2List, level3List).Children.AddRange(level4List);
+                        level4List.Clear();
+
+                        BlogHeadline headline = ParserHeadlineElement(element);
+                        level3List.Add(headline);
+                        break;
+                    }
+                case "h4":
+                    {
+                        BlogHeadline headline = ParserHeadlineElement(element);
+                        level4List.Add(headline);
+                        break;
+                    }
+            }
+        }
+
+        // 太抽象了（（（
+        FindParentHeadline(topHeadline, level2List, level3List).Children.AddRange(level4List);
+        FindParentHeadline(topHeadline, level2List).Children.AddRange(level3List);
+        topHeadline.Children.AddRange(level2List);
+
+        tableOfContentService.AddHeadline(essay.FileName, topHeadline);
+
+        return essay.WithNewHtmlContent(document.DocumentElement.OuterHtml);
+    }
+
+    private static BlogHeadline ParserHeadlineElement(IElement element)
+    {
+        element.Id ??= element.TextContent;
+        return new BlogHeadline(element.TextContent, element.Id);
+    }
+
+    /// <summary>
+    /// 找到h4标题的父级标题
+    /// </summary>
+    /// <param name="topHeadline"></param>
+    /// <param name="level2"></param>
+    /// <param name="level3"></param>
+    /// <returns></returns>
+    private static BlogHeadline FindParentHeadline(BlogHeadline topHeadline, List<BlogHeadline> level2,
+        List<BlogHeadline> level3)
+    {
+        BlogHeadline? result = level3.LastOrDefault();
+        if (result is not null)
+        {
+            return result;
+        }
+
+        return level2.LastOrDefault() ?? topHeadline;
+    }
+
+    /// <summary>
+    /// 找到h3标题的父级标题
+    /// </summary>
+    /// <param name="topHeadline"></param>
+    /// <param name="level2"></param>
+    /// <returns></returns>
+    private static BlogHeadline FindParentHeadline(BlogHeadline topHeadline, List<BlogHeadline> level2) =>
+        FindParentHeadline(topHeadline, level2, []);
+
+    public string Name => nameof(HeadlinePostRenderProcessor);
+}

YaeBlog.Core/Processors/ImagePostRenderProcessor.cs

@@ -38,7 +38,7 @@ public class ImagePostRenderProcessor(ILogger<ImagePostRenderProcessor> logger,
         return essay.WithNewHtmlContent(html.DocumentElement.OuterHtml);
     }
 
-    public string Name => "ImagePostRenderProcessor";
+    public string Name => nameof(ImagePostRenderProcessor);
 
     private string GenerateImageLink(string filename, string essayFilename)
     {

YaeBlog.Core/Processors/TablePostRenderProcessor.cs

@@ -0,0 +1,34 @@
+using AngleSharp;
+using AngleSharp.Dom;
+using AngleSharp.Html.Dom;
+using YaeBlog.Core.Abstractions;
+using YaeBlog.Core.Models;
+
+namespace YaeBlog.Core.Processors;
+
+public class TablePostRenderProcessor: IPostRenderProcessor
+{
+    public async Task<BlogEssay> ProcessAsync(BlogEssay essay)
+    {
+        BrowsingContext browsingContext = new(Configuration.Default);
+        IDocument document = await browsingContext.OpenAsync(
+            req => req.Content(essay.HtmlContent));
+
+        IEnumerable<IHtmlTableElement> tableElements = from item in document.All
+            where item.LocalName == "table"
+            select item as IHtmlTableElement;
+
+        foreach (IHtmlTableElement element in tableElements)
+        {
+            IHtmlDivElement divElement = document.CreateElement<IHtmlDivElement>();
+            divElement.InnerHtml = element.OuterHtml;
+            divElement.ClassList.Add("py-2", "table-wrapper");
+
+            element.Replace(divElement);
+        }
+
+        return essay.WithNewHtmlContent(document.DocumentElement.OuterHtml);
+    }
+
+    public string Name => nameof(TablePostRenderProcessor);
+}