Misc.

xoofx · Mar 18, 2024 · 03390e4 · 03390e4
1 parent 42bd65c
commit 03390e4
Showing 1 changed file with 16 additions and 14 deletions.
diff --git a/src/Markdig/Extensions/Bootstrap/BootstrapExtension.cs b/src/Markdig/Extensions/Bootstrap/BootstrapExtension.cs
@@ -41,6 +41,7 @@ public void Setup(MarkdownPipeline pipeline, IMarkdownRenderer renderer)
 
     private static void PipelineOnDocumentProcessed(MarkdownDocument document)
     {
+        Span<char> upperKind = new char[16];
         foreach (var node in document.Descendants())
         {
             if (node.IsInline)
@@ -61,22 +62,23 @@ private static void PipelineOnDocumentProcessed(MarkdownDocument document)
                     var attributes = node.GetAttributes();
                     attributes.AddClass("alert");
                     attributes.AddProperty("role", "alert");
-
-                    Span<char> upperKind = new char[alertBlock.Kind.Length];
-                    alertBlock.Kind.AsSpan().ToUpperInvariant(upperKind);
-                    string? @class = upperKind switch
+                    if (alertBlock.Kind.Length <= upperKind.Length)
                     {
-                        "NOTE" => "alert-primary",
-                        "TIP" => "alert-success",
-                        "IMPORTANT" => "alert-info",
-                        "WARNING" => "alert-warning",
-                        "CAUTION" => "alert-danger",
-                        _ => null,
-                    };
+                        alertBlock.Kind.AsSpan().ToUpperInvariant(upperKind);
+                        string? @class = upperKind.Slice(0, alertBlock.Kind.Length) switch
+                        {
+                            "NOTE" => "alert-primary",
+                            "TIP" => "alert-success",
+                            "IMPORTANT" => "alert-info",
+                            "WARNING" => "alert-warning",
+                            "CAUTION" => "alert-danger",
+                            _ => null,
+                        };
 
-                    if (@class is not null)
-                    {
-                        attributes.AddClass(@class);
+                        if (@class is not null)
+                        {
+                            attributes.AddClass(@class);
+                        }
                     }
 
                     var lastParagraph = alertBlock.Descendants().OfType<ParagraphBlock>().LastOrDefault();