AI-Studio/MudBlazor.Markdown/Components/MudMathJax.razor.cs

namespace MudBlazor;

internal sealed class MudMathJax : ComponentBase
{
	private const string ScriptId = "mudblazor-markdown-mathjax";

	[Parameter]
	public string Delimiter { get; set; } = string.Empty;

	[Parameter]
	public StringSlice Value { get; set; }

	[Inject]
	private IJSRuntime Js { get; init; } = default!;

	protected override void BuildRenderTree(RenderTreeBuilder builder)
	{
		var elementIndex = 0;

		var delimiter = GetDelimiter(Delimiter);

		builder.AddContent(elementIndex++, delimiter.Start);
		builder.AddContent(elementIndex++, Value);
		builder.AddContent(elementIndex, delimiter.End);
	}

	protected override async Task OnAfterRenderAsync(bool firstRender)
	{
		if (firstRender)
		{
			await Js.InvokeVoidAsync("appendMathJaxScript", ScriptId)
				.ConfigureAwait(false);
		}

		await Js.InvokeVoidAsync("refreshMathJaxScript")
			.ConfigureAwait(false);
	}

	private static MathDelimiter GetDelimiter(in string delimiter) =>
		delimiter switch
		{
			"$" => new MathDelimiter("\\(", "\\)"),
			"$$" => new MathDelimiter(delimiter),
			_ => new MathDelimiter(delimiter)
		};

	private readonly ref struct MathDelimiter
	{
		public MathDelimiter(string delimiter)
		{
			Start = End = delimiter;
		}

		public MathDelimiter(string start, string end)
		{
			Start = start;
			End = end;
		}

		public string Start { get; }
		
		public string End { get; }
	}
}