AI-Studio/app/MindWork AI Studio/Components/JsonTreeView.razor.cs

using System.Text.Json.Nodes;

using AIStudio.Tools.ToolCallingSystem;

using Microsoft.AspNetCore.Components;

namespace AIStudio.Components;

public partial class JsonTreeView : ComponentBase
{
    [Parameter]
    public JsonNode? Value { get; set; }

    private IReadOnlyCollection<TreeItemData<JsonTreeNode>> items = [];

    protected override void OnParametersSet()
    {
        this.items = [CreateTreeItem("$", this.Value)];
    }

    private static TreeItemData<JsonTreeNode> CreateTreeItem(string label, JsonNode? value)
    {
        var children = CreateChildren(value);
        return new TreeItemData<JsonTreeNode>
        {
            Expanded = false,
            Expandable = children.Count > 0,
            Value = new JsonTreeNode
            {
                Text = $"{label}: {FormatValue(value)}",
                Icon = GetIcon(value),
                Expandable = children.Count > 0,
            },
            Children = children,
        };
    }

    private static List<TreeItemData<JsonTreeNode>> CreateChildren(JsonNode? value) => value switch
    {
        JsonObject jsonObject => jsonObject
            .Select(property => CreateTreeItem(FormatKey(property.Key), property.Value))
            .ToList(),
        JsonArray jsonArray => jsonArray
            .Select((item, index) => CreateTreeItem($"[{index}]", item))
            .ToList(),
        _ => [],
    };

    internal static string FormatKey(string key) => JsonValue.Create(key).ToJsonString(ToolExecutionResult.MODEL_CONTENT_OPTIONS);

    /// <summary>
    /// Writes a value as JSON, objects and arrays only as a placeholder.
    /// </summary>
    /// <remarks>
    /// The tree shows the result of a tool, so it writes like the model reads it, see
    /// ToolExecutionResult.MODEL_CONTENT_OPTIONS. The default options would escape every umlaut and
    /// every quotation mark for a web page, although Blazor encodes the text for the page anyway.
    /// </remarks>
    internal static string FormatValue(JsonNode? value) => value switch
    {
        JsonObject jsonObject when jsonObject.Count == 0 => "{}",
        JsonObject => "{...}",
        JsonArray jsonArray when jsonArray.Count == 0 => "[]",
        JsonArray => "[...]",
        null => "null",
        _ => value.ToJsonString(ToolExecutionResult.MODEL_CONTENT_OPTIONS),
    };

    private static string GetIcon(JsonNode? value) => value switch
    {
        JsonObject => Icons.Material.Filled.DataObject,
        JsonArray => Icons.Material.Filled.DataArray,
        _ => Icons.Material.Filled.Code,
    };

    private sealed class JsonTreeNode
    {
        public string Text { get; init; } = string.Empty;

        public string Icon { get; init; } = string.Empty;

        public bool Expandable { get; init; }
    }
}