Day 2 vibes

ET/Web/Services/DocsService.cs

@@ -1,4 +1,7 @@
+using System.Net;
 using System.Net.Http.Json;
+using System.Text;
+using System.Text.RegularExpressions;
 using Markdig;
 using Web.Models;
 
@@ -6,12 +9,19 @@ namespace Web.Services;
 
 public class DocsService
 {
-    private readonly HttpClient _http;
-    private NotesIndex? _index;
     private static readonly MarkdownPipeline Pipeline = new MarkdownPipelineBuilder()
         .UseYamlFrontMatter()
         .Build();
 
+    private static readonly HashSet<string> ImageExtensions = new(StringComparer.OrdinalIgnoreCase)
+    {
+        ".png", ".jpg", ".jpeg", ".gif", ".svg", ".webp", ".bmp"
+    };
+
+    private readonly HttpClient _http;
+    private NotesIndex? _index;
+    private readonly Dictionary<string, string> _markdownCache = new();
+
     public DocsService(HttpClient http)
     {
         _http = http;
@@ -38,7 +48,7 @@ public class DocsService
         try
         {
             var markdown = await _http.GetStringAsync($"docs/notes/{slug}.md");
-            return ParseDocument(slug, noteInfo.Title, markdown);
+            return await ParseDocument(slug, noteInfo.Title, markdown);
         }
         catch
         {
@@ -46,7 +56,7 @@ public class DocsService
         }
     }
 
-    private static NoteDocument ParseDocument(string slug, string title, string markdown)
+    private async Task<NoteDocument> ParseDocument(string slug, string title, string markdown, int depth = 0)
     {
         var doc = new NoteDocument
         {
@@ -69,6 +79,7 @@ public class DocsService
                     inFrontmatter = true;
                     continue;
                 }
+
                 inFrontmatter = false;
                 frontmatterDone = true;
                 continue;
@@ -89,13 +100,11 @@ public class DocsService
                 {
                     var key = line[..colonIdx].Trim();
                     if (string.Equals(key, "category", StringComparison.OrdinalIgnoreCase))
-                    {
                         doc.Category = line[(colonIdx + 1)..].Trim().Trim('"');
-                    }
                 }
             }
 
-            var fmHtml = new System.Text.StringBuilder();
+            var fmHtml = new StringBuilder();
             fmHtml.Append("<table class=\"frontmatter\">");
             foreach (var line in frontmatterLines)
             {
@@ -105,33 +114,91 @@ public class DocsService
                     var key = line[..colonIdx].Trim();
                     var value = line[(colonIdx + 1)..].Trim().Trim('"');
                     fmHtml.Append("<tr><td class=\"fm-key\">");
-                    fmHtml.Append(System.Net.WebUtility.HtmlEncode(key));
+                    fmHtml.Append(WebUtility.HtmlEncode(key));
                     fmHtml.Append("</td><td class=\"fm-value\">");
-                    var encoded = System.Net.WebUtility.HtmlEncode(value);
+                    var encoded = WebUtility.HtmlEncode(value);
                     fmHtml.Append(ConvertWikiLinks(encoded));
                     fmHtml.Append("</td></tr>");
                 }
                 else
                 {
                     fmHtml.Append("<tr><td colspan=\"2\">");
-                    fmHtml.Append(ConvertWikiLinks(System.Net.WebUtility.HtmlEncode(line.Trim())));
+                    fmHtml.Append(ConvertWikiLinks(WebUtility.HtmlEncode(line.Trim())));
                     fmHtml.Append("</td></tr>");
                 }
             }
+
             fmHtml.Append("</table>");
             doc.FrontmatterHtml = fmHtml.ToString();
         }
 
         var body = string.Join("\n", bodyLines);
         body = ConvertWikiLinks(body);
-        doc.HtmlContent = Markdown.ToHtml(body, Pipeline);
+        var html = Markdown.ToHtml(body, Pipeline);
+        html = await ResolveEmbedsInHtml(html, depth);
+        doc.HtmlContent = html;
 
         return doc;
     }
 
+    private async Task<string> ResolveEmbedsInHtml(string html, int depth)
+    {
+        if (depth > 10) return html;
+
+        var regex = new Regex(@"(?:<p>)?!\[\[([^\]]+)\]\](?:</p>)?");
+        var sb = new StringBuilder();
+        var lastIndex = 0;
+
+        foreach (Match match in regex.Matches(html))
+        {
+            sb.Append(html, lastIndex, match.Index - lastIndex);
+
+            var filename = match.Groups[1].Value.Trim();
+            var replacement = await ResolveEmbed(filename, depth);
+            sb.Append(replacement);
+
+            lastIndex = match.Index + match.Length;
+        }
+
+        sb.Append(html, lastIndex, html.Length - lastIndex);
+        return sb.ToString();
+    }
+
+    private async Task<string> ResolveEmbed(string filename, int depth)
+    {
+        var ext = Path.GetExtension(filename);
+
+        if (ImageExtensions.Contains(ext))
+        {
+            return $"<img src=\"/docs/images/{filename}\" alt=\"{WebUtility.HtmlEncode(filename)}\" />";
+        }
+
+        var slug = Slugify(filename);
+        try
+        {
+            var markdown = await GetMarkdownAsync(slug);
+            var embedded = await ParseDocument(slug, filename, markdown, depth + 1);
+            return $"<div class=\"embed\">{embedded.HtmlContent}</div>";
+        }
+        catch
+        {
+            return $"<div class=\"embed-error\">[Embed not found: {WebUtility.HtmlEncode(filename)}]</div>";
+        }
+    }
+
+    private async Task<string> GetMarkdownAsync(string slug)
+    {
+        if (_markdownCache.TryGetValue(slug, out var cached))
+            return cached;
+
+        var markdown = await _http.GetStringAsync($"docs/notes/{slug}.md");
+        _markdownCache[slug] = markdown;
+        return markdown;
+    }
+
     private static string ConvertWikiLinks(string text)
     {
-        return System.Text.RegularExpressions.Regex.Replace(
+        return Regex.Replace(
             text,
             @"\[\[([^\]]+)\]\]",
             match =>
@@ -141,7 +208,7 @@ public class DocsService
                 var linkText = parts.Length > 1 ? parts[1].Trim() : parts[0].Trim();
                 var target = parts[0].Trim();
                 var slug = Slugify(target);
-                return $"<a href=\"/docs/{slug}\">{System.Net.WebUtility.HtmlEncode(linkText)}</a>";
+                return $"<a href=\"/docs/{slug}\">{WebUtility.HtmlEncode(linkText)}</a>";
             });
     }
 
@@ -153,8 +220,8 @@ public class DocsService
             .Replace(".", "")
             .Replace("(", "")
             .Replace(")", "");
-        slug = System.Text.RegularExpressions.Regex.Replace(slug, @"[^a-z0-9\-]", "");
-        slug = System.Text.RegularExpressions.Regex.Replace(slug, @"-+", "-");
+        slug = Regex.Replace(slug, @"[^a-z0-9\-]", "");
+        slug = Regex.Replace(slug, @"-+", "-");
         return slug.Trim('-');
     }
 }