From a9aba743514d13de403b6685d1475afa914beb63 Mon Sep 17 00:00:00 2001
From: Salman Ahmed <frenzibyte@gmail.com>
Date: Sat, 8 Oct 2022 21:26:20 +0300
Subject: [PATCH] Add markdown country flag support

---
 .../Markdown/OsuMarkdownTextFlowContainer.cs  | 27 +++++++++++++++++++
 .../Wiki/Markdown/WikiMarkdownContainer.cs    |  7 +++++
 2 files changed, 34 insertions(+)

diff --git a/osu.Game/Graphics/Containers/Markdown/OsuMarkdownTextFlowContainer.cs b/osu.Game/Graphics/Containers/Markdown/OsuMarkdownTextFlowContainer.cs
index db8abfb269..fb8f13ab84 100644
--- a/osu.Game/Graphics/Containers/Markdown/OsuMarkdownTextFlowContainer.cs
+++ b/osu.Game/Graphics/Containers/Markdown/OsuMarkdownTextFlowContainer.cs
@@ -3,6 +3,9 @@
 
 #nullable disable
 
+using System;
+using System.Linq;
+using Markdig.Extensions.CustomContainers;
 using Markdig.Syntax.Inlines;
 using osu.Framework.Allocation;
 using osu.Framework.Graphics;
@@ -11,6 +14,9 @@
 using osu.Framework.Graphics.Shapes;
 using osu.Framework.Graphics.Sprites;
 using osu.Game.Overlays;
+using osu.Game.Users;
+using osu.Game.Users.Drawables;
+using osuTK;
 
 namespace osu.Game.Graphics.Containers.Markdown
 {
@@ -33,6 +39,27 @@ protected override void AddCodeInLine(CodeInline codeInline) => AddDrawable(new
         protected override SpriteText CreateEmphasisedSpriteText(bool bold, bool italic)
             => CreateSpriteText().With(t => t.Font = t.Font.With(weight: bold ? FontWeight.Bold : FontWeight.Regular, italics: italic));
 
+        protected override void AddCustomComponent(CustomContainerInline inline)
+        {
+            if (!(inline.FirstChild is LiteralInline literal))
+            {
+                base.AddCustomComponent(inline);
+                return;
+            }
+
+            string[] attributes = literal.Content.ToString().Trim(' ', '{', '}').Split();
+            string flagAttribute = attributes.SingleOrDefault(a => a.StartsWith(@"flag", StringComparison.Ordinal));
+
+            if (flagAttribute == null)
+            {
+                base.AddCustomComponent(inline);
+                return;
+            }
+
+            string flag = flagAttribute.Split('=').Last().Trim('"');
+            AddDrawable(new DrawableFlag(Enum.Parse<CountryCode>(flag)) { Size = new Vector2(20, 15) });
+        }
+
         private class OsuMarkdownInlineCode : Container
         {
             [Resolved]
diff --git a/osu.Game/Overlays/Wiki/Markdown/WikiMarkdownContainer.cs b/osu.Game/Overlays/Wiki/Markdown/WikiMarkdownContainer.cs
index 4f1083a75c..15c455416c 100644
--- a/osu.Game/Overlays/Wiki/Markdown/WikiMarkdownContainer.cs
+++ b/osu.Game/Overlays/Wiki/Markdown/WikiMarkdownContainer.cs
@@ -4,6 +4,7 @@
 #nullable disable
 
 using System.Linq;
+using Markdig.Extensions.CustomContainers;
 using Markdig.Extensions.Yaml;
 using Markdig.Syntax;
 using Markdig.Syntax.Inlines;
@@ -16,6 +17,7 @@ namespace osu.Game.Overlays.Wiki.Markdown
     public class WikiMarkdownContainer : OsuMarkdownContainer
     {
         protected override bool Footnotes => true;
+        protected override bool CustomContainers => true;
 
         public string CurrentPath
         {
@@ -26,6 +28,11 @@ protected override void AddMarkdownComponent(IMarkdownObject markdownObject, Fil
         {
             switch (markdownObject)
             {
+                case CustomContainer:
+                    // infoboxes are parsed into CustomContainer objects, but we don't have support for infoboxes yet.
+                    // todo: add support for infobox.
+                    break;
+
                 case YamlFrontMatterBlock yamlFrontMatterBlock:
                     container.Add(new WikiNoticeContainer(yamlFrontMatterBlock));
                     break;