Pf2eDataScrapper/Pf2eViewer/MainViewForm.cs

using Microsoft.EntityFrameworkCore;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using Pf2eModel;
using Pf2eModel.Model;
using Pf2eModel.Model.Base;
using Pf2eModel.Model.Enum;

namespace Pf2eViewer;

public partial class MainViewForm : Form
{
    public Pf2eDbContext Context = new();

    public MainViewForm()
    {
        InitializeComponent();
        Reset();

        EntityGridView.SelectionMode = DataGridViewSelectionMode.FullRowSelect;
    }

    public void Reset()
    {
        Context.Dispose();
        Context = new();

        TypesListBox.DataSource = Context.Entities.Select(e => e.Type).Distinct().ToList();
        TypesListBox.SelectedIndex = 0;
    }

    private void TypesListBox_SelectedIndexChanged(object sender, EventArgs _)
    {
        if (TypesListBox.SelectedItem is null)
        {
            return;
        }

        EntityGridView.DataSource = Context.Entities
            .Where(e => e.Type == (EntityType) TypesListBox.SelectedItem)
            .OrderBy(e => e.Id)
            .Include(e => e.Sources)
            .Include(e => e.Aliases)
            .Include(e => e.Keywords)
            .Include(e => e.Externals)
            .Include(e => e.Traits)
            .Select(e => new EntityGridRow(e))
            .ToList();
        EntityGridView.Rows[0].Selected = true;
    }

    private void EntityGridView_RowStateChanged(object sender, DataGridViewRowStateChangedEventArgs e)
    {
        int rowsCount = EntityGridView.SelectedRows.Count;
        if (rowsCount is 0 or > 1)
        {
            return;
        }

        DataGridViewRow row = EntityGridView.SelectedRows[0];

        EntityGridRow data = (EntityGridRow) row.DataBoundItem;

        string json = SerializeEntity(data.GetEntity());

        JsonToTreeview(EntityTreeView, json, "Entity");

        DescriptionTextBox.Text = data.GetEntity().Description.Replace("\\r", "").Replace("\\n", "\\r\\n");
    }

    public string SerializeEntity(Entity entity)
    {
        JsonSerializerSettings settings = new()
        {
            Converters = [new EntityConverter()]
        };

        string result = JsonConvert.SerializeObject(entity, settings);

        Type type = EntityOfTypeAttribute.Resolve(entity);
        string subResult = JsonConvert.SerializeObject(Context.Find(type, entity.Id), settings);

        return result[..^1] + $",\"SubEntity\":{subResult}}}";
    }



    public static void JsonToTreeview(TreeView treeView, string jsonString, string rootName)
    {
        object json = jsonString.StartsWith('[')
                        ? JArray.Parse(jsonString)
                        : JObject.Parse(jsonString);

        treeView.Nodes.Clear();

        foreach (TreeNode node in Element2TreeNode(json, rootName).Nodes)
        {
            treeView.Nodes.Add(node);
        }
    }

    private static TreeNode Element2TreeNode(object element, string text = "")
    {
        List<TreeNode> subNodes = [];

        switch (element.GetType().Name)
        {
            case "JObject":
                JObject jObject = (JObject) element;
                foreach (KeyValuePair<string, JToken> token in jObject)
                {
                    subNodes.Add(Element2TreeNode(token.Value, token.Key));
                }
                break;

            case "JArray":
                JArray jArray = (JArray) element;
                int i = 0;
                subNodes.AddRange(jArray.Select(token => Element2TreeNode(token, $"[{i++}]")));

                if (i == 0)
                {
                    text += ": []";
                }
                break;

            case "JValue":
                JValue jValue = (JValue) element;
                text += $": ({jValue.Type}) {jValue}";
                break;

            default:
                throw new Exception("clsJSON2Treeview can't interpret object:" + element.GetType().Name);
        }

        TreeNode treeNode = new(text);

        if (subNodes.Count > 0)
        {
            treeNode.Nodes.AddRange(subNodes
                .OrderBy(node => node.Nodes.Count > 0 ? 1 : 0)
                .ToArray()
                );
        }

        return treeNode;
    }

    private class EntityGridRow(Entity e)
    {
        public int Id { get; set; } = e.Id;
        public string Name { get; set; } = e.Name;
        public EntityRarity Rarity { get; set; } = e.Rarity;

        private readonly Entity _e = e;
        public Entity GetEntity() => _e;

    }
}