AetherBags/AetherBags/Addons/AddonCategoryConfigurationWindow.cs

using System.Collections.Generic;
using System.Linq;
using System.Numerics;
using AetherBags.Configuration;
using AetherBags.Inventory;
using AetherBags.Nodes.Configuration.Category;
using FFXIVClientStructs.FFXIV.Component.GUI;
using KamiToolKit;
using KamiToolKit.Classes;
using KamiToolKit.Nodes;
using KamiToolKit.Premade.Nodes;

namespace AetherBags.Addons;

public class AddonCategoryConfigurationWindow : NativeAddon
{
    private ModifyListNode<CategoryWrapper, CategoryListItemNode>? _selectionListNode;
    private VerticalLineNode? _separatorLine;
    private CategoryConfigurationNode? _configNode;
    private TextNode? _nothingSelectedTextNode;

    private List<CategoryWrapper> _categoryWrappers = new();

    private bool _suppressSelectionListRefresh;
    private bool _pendingSelectionListRefresh;

    protected override unsafe void OnSetup(AtkUnitBase* addon)
    {
        _categoryWrappers = CreateCategoryWrappers();

        _selectionListNode = new ModifyListNode<CategoryWrapper, CategoryListItemNode>
        {
            Position = ContentStartPosition,
            Size = ContentSize with { X = 250.0f },
            Options = _categoryWrappers,
            SelectionChanged = OnOptionChanged,
            AddNewEntry = OnAddNewCategory,
            RemoveEntry = OnRemoveCategory,
            SortOptions = [ "Order" ],
            ItemComparer = (left, right, mode) => left.Compare(right, mode),
            IsSearchMatch = (data, search) => data.GetLabel().Contains(search, global::System.StringComparison.OrdinalIgnoreCase)
        };
        _selectionListNode.AttachNode(this);

        _separatorLine = new VerticalLineNode
        {
            Position = ContentStartPosition + new Vector2(250.0f + 8.0f, 0.0f),
            Size = ContentSize with { X = 4.0f },
        };
        _separatorLine.AttachNode(this);

        _nothingSelectedTextNode = new TextNode
        {
            Position = ContentStartPosition + new Vector2(250.0f + 16.0f, 0.0f),
            Size = ContentSize - new Vector2(250.0f + 16.0f, 0.0f),
            AlignmentType = AlignmentType.Center,
            TextFlags = TextFlags.WordWrap | TextFlags.MultiLine,
            FontSize = 14,
            LineSpacing = 22,
            FontType = FontType.Axis,
            String = "Please select a category on the left or add one.",
            TextColor = ColorHelper.GetColor(1),
        };
        _nothingSelectedTextNode.AttachNode(this);

        _configNode = new CategoryConfigurationNode
        {
            Position = ContentStartPosition + new Vector2(250.0f + 16.0f, 0.0f),
            Size = ContentSize - new Vector2(250.0f + 16.0f, 0.0f),
            IsVisible = false,
            OnCategoryChanged = RefreshSelectionList,
        };

        _configNode.AttachNode(this);
    }

    private List<CategoryWrapper> CreateCategoryWrappers()
    {
        return System.Config.Categories.UserCategories
            .Select(categoryDefinition => new CategoryWrapper(categoryDefinition))
            .ToList();
    }

    private void OnAddNewCategory()
    {
        var newCategory = new UserCategoryDefinition
        {
            Name = $"New Category {System.Config.Categories.UserCategories.Count + 1}",
            Order = System.Config.Categories.UserCategories.Count,
        };

        System.Config.Categories.UserCategories.Add(newCategory);

        var newWrapper = new CategoryWrapper(newCategory);
        _categoryWrappers.Add(newWrapper);

        RefreshSelectionList();
        _selectionListNode?.RefreshList();
        InventoryOrchestrator.RefreshAll(updateMaps: true);
    }

    private void OnOptionChanged(CategoryWrapper? newOption)
    {
        if (_configNode is null) return;

        _suppressSelectionListRefresh = true;
        try
        {
            _configNode.IsVisible = newOption is not null;

            if (_nothingSelectedTextNode is not null)
                _nothingSelectedTextNode.IsVisible = newOption is null;

            _configNode.ConfigurationOption = newOption;
        }
        finally
        {
            _suppressSelectionListRefresh = false;

            if (_pendingSelectionListRefresh)
            {
                _pendingSelectionListRefresh = false;
                _selectionListNode?.RefreshList();
            }
        }
    }

    private void OnRemoveCategory(CategoryWrapper categoryWrapper)
    {
        if (categoryWrapper.CategoryDefinition is null) return;

        System.Config.Categories.UserCategories.Remove(categoryWrapper.CategoryDefinition);
        _categoryWrappers.Remove(categoryWrapper);

        RefreshSelectionList();

        if (_configNode is not null && ReferenceEquals(_configNode.ConfigurationOption, categoryWrapper))
        {
            OnOptionChanged(null);
        }
        InventoryOrchestrator.RefreshAll(updateMaps: true);
    }

    private void RefreshSelectionList()
    {
        if (_suppressSelectionListRefresh)
        {
            _pendingSelectionListRefresh = true;
            return;
        }

        _selectionListNode?.RefreshList();
    }

    protected override unsafe void OnFinalize(AtkUnitBase* addon)
    {
        _selectionListNode = null;
        _configNode = null;
        _separatorLine = null;
        _nothingSelectedTextNode = null;
        base.OnFinalize(addon);
    }
}