Make UseLlmMarkdownRenderer use of static action

Author: reakaleek

src/Elastic.Markdown/Exporters/LlmMarkdownExporter.cs

@@ -61,9 +61,9 @@ await fileContext.SourceFile.SourceFile.FileSystem.File.WriteAllTextAsync(
 	}
 
 	public static string ConvertToLlmMarkdown(MarkdownDocument document, BuildContext context) =>
-		DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(context, renderer =>
+		DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(context, document, static (obj, renderer) =>
 		{
-			_ = renderer.Render(document);
+			_ = renderer.Render(obj);
 		});
 
 	private static IFileInfo GetLlmOutputFile(MarkdownExportFileContext fileContext)

src/Elastic.Markdown/Helpers/DocumentationObjectPoolProvider.cs

@@ -8,6 +8,7 @@
 using Elastic.Markdown.Myst;
 using Elastic.Markdown.Myst.Renderers.LlmMarkdown;
 using Markdig.Renderers;
+using Markdig.Syntax;
 using Microsoft.Extensions.ObjectPool;
 
 namespace Elastic.Markdown.Helpers;
@@ -21,13 +22,13 @@ internal static class DocumentationObjectPoolProvider
 	public static readonly ObjectPool<HtmlRenderSubscription> HtmlRendererPool = PoolProvider.Create(new HtmlRendererPooledObjectPolicy());
 	private static readonly ObjectPool<LlmMarkdownRenderSubscription> LlmMarkdownRendererPool = PoolProvider.Create(new LlmMarkdownRendererPooledObjectPolicy());
 
-	public static string UseLlmMarkdownRenderer(BuildContext buildContext, Action<LlmMarkdownRenderer> action)
+	public static string UseLlmMarkdownRenderer<TContext>(BuildContext buildContext, TContext context, Action<TContext, LlmMarkdownRenderer> action)
 	{
 		var subscription = LlmMarkdownRendererPool.Get();
 		subscription.SetBuildContext(buildContext);
 		try
 		{
-			action(subscription.LlmMarkdownRenderer);
+			action(context, subscription.LlmMarkdownRenderer);
 			return subscription.RentedStringBuilder!.ToString();
 		}
 		finally

src/Elastic.Markdown/Myst/Renderers/LlmMarkdown/LlmBlockRenderers.cs

@@ -23,9 +23,9 @@ public static class LlmRenderingHelpers
 {
 	public static void RenderBlockWithIndentation(LlmMarkdownRenderer renderer, MarkdownObject block, string indentation = "  ")
 	{
-		var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+		var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, block, static (obj, tmpRenderer) =>
 		{
-			_ = tempRenderer.Render(block);
+			_ = tmpRenderer.Render(obj);
 		});
 
 		if (string.IsNullOrEmpty(content))
@@ -190,11 +190,10 @@ private static string GetContinuationIndent(string baseIndent, bool isOrdered) =
 
 	private static void RenderBlockWithIndentation(LlmMarkdownRenderer renderer, Block block, string baseIndent, bool isOrdered)
 	{
-		var blockOutput = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+		var blockOutput = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, block, static (obj, tmpRenderer) =>
 		{
-			_ = tempRenderer.Render(block);
+			_ = tmpRenderer.Render(obj);
 		});
-
 		var continuationIndent = GetContinuationIndent(baseIndent, isOrdered);
 		var lines = blockOutput.Split('\n');
 		for (var i = 0; i < lines.Length; i++)
@@ -287,15 +286,11 @@ protected override void Write(LlmMarkdownRenderer renderer, Table table)
 				renderer.Writer.Write(" ");
 
 				// Capture cell content
-				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, cell.Descendants().OfType<Inline>(), static (obj, tmpRenderer) =>
 				{
-					// Render cell content to temporary writer
-					foreach (var inline in cell.Descendants().OfType<Inline>())
-					{
-						tempRenderer.Write(inline);
-					}
+					foreach (var inline in obj)
+						tmpRenderer.Write(inline);
 				});
-
 				// Write padded content
 				renderer.Writer.Write(content.PadRight(columnWidths[cellIndex]));
 				renderer.Writer.Write(" |");
@@ -325,13 +320,10 @@ protected override void Write(LlmMarkdownRenderer renderer, Table table)
 				renderer.Writer.Write(" ");
 
 				// Capture cell content
-				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, cell.Descendants().OfType<Inline>(), static (obj, tmpRenderer) =>
 				{
-					// Render cell content to temporary writer
-					foreach (var inline in cell.Descendants().OfType<Inline>())
-					{
-						tempRenderer.Write(inline);
-					}
+					foreach (var inline in obj)
+						tmpRenderer.Write(inline);
 				});
 
 				// Write padded content
@@ -364,13 +356,10 @@ private static int[] CalculateColumnWidths(LlmMarkdownRenderer renderer, Table t
 			foreach (var cell in row.Cast<TableCell>())
 			{
 				// Capture cell content
-				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+				var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, cell.Descendants().OfType<Inline>(), static (obj, tmpRenderer) =>
 				{
-					// Render cell content to temporary writer
-					foreach (var inline in cell.Descendants().OfType<Inline>())
-					{
-						tempRenderer.Write(inline);
-					}
+					foreach (var inline in obj)
+						tmpRenderer.Write(inline);
 				});
 
 				// Update width if this cell is wider
@@ -494,15 +483,15 @@ private void WriteIncludeBlock(LlmMarkdownRenderer renderer, IncludeBlock block)
 	private static void WriteChildrenWithIndentation(LlmMarkdownRenderer renderer, Block container, string indent)
 	{
 		// Capture output and manually add indentation
-		var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+		var content = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, container, static (obj, tmpRenderer) =>
 		{
-			switch (container)
+			switch (obj)
 			{
 				case ContainerBlock containerBlock:
-					tempRenderer.WriteChildren(containerBlock);
+					tmpRenderer.WriteChildren(containerBlock);
 					break;
 				case LeafBlock leafBlock:
-					tempRenderer.WriteLeafInline(leafBlock);
+					tmpRenderer.WriteLeafInline(leafBlock);
 					break;
 			}
 		});
@@ -538,11 +527,10 @@ protected override void Write(LlmMarkdownRenderer renderer, DefinitionItem obj)
 
 	private static string GetPlainTextFromLeafBlock(LlmMarkdownRenderer renderer, LeafBlock leafBlock)
 	{
-		var markdownText = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, tempRenderer =>
+		var markdownText = DocumentationObjectPoolProvider.UseLlmMarkdownRenderer(renderer.BuildContext, leafBlock, static (obj, tmpRenderer) =>
 		{
-			tempRenderer.WriteLeafInline(leafBlock);
+			tmpRenderer.WriteLeafInline(obj);
 		});
-
 		return markdownText.StripMarkdown();
 	}
 }