Penumbra/Penumbra/Api/IpcTester/TemporaryIpcTester.cs

using Dalamud.Interface;
using Dalamud.Plugin;
using ImGuiNET;
using OtterGui;
using OtterGui.Raii;
using OtterGui.Services;
using OtterGui.Text;
using Penumbra.Api.Api;
using Penumbra.Api.Enums;
using Penumbra.Api.IpcSubscribers;
using Penumbra.Collections.Manager;
using Penumbra.Meta.Manipulations;
using Penumbra.Mods;
using Penumbra.Mods.Manager;
using Penumbra.Services;

namespace Penumbra.Api.IpcTester;

public class TemporaryIpcTester(
    DalamudPluginInterface pi,
    ModManager modManager,
    CollectionManager collections,
    TempModManager tempMods,
    TempCollectionManager tempCollections,
    SaveService saveService,
    Configuration config)
    : IUiService
{
    public Guid LastCreatedCollectionId = Guid.Empty;

    private Guid?         _tempGuid;
    private string        _tempCollectionName     = string.Empty;
    private string        _tempCollectionGuidName = string.Empty;
    private string        _tempModName            = string.Empty;
    private string        _tempGamePath           = "test/game/path.mtrl";
    private string        _tempFilePath           = "test/success.mtrl";
    private string        _tempManipulation       = string.Empty;
    private PenumbraApiEc _lastTempError;
    private int           _tempActorIndex;
    private bool          _forceOverwrite;

    public void Draw()
    {
        using var _ = ImRaii.TreeNode("Temporary");
        if (!_)
            return;

        ImGui.InputTextWithHint("##tempCollection", "Collection Name...", ref _tempCollectionName, 128);
        ImGuiUtil.GuidInput("##guid", "Collection GUID...", string.Empty, ref _tempGuid, ref _tempCollectionGuidName);
        ImGui.InputInt("##tempActorIndex", ref _tempActorIndex, 0, 0);
        ImGui.InputTextWithHint("##tempMod",   "Temporary Mod Name...",         ref _tempModName,      32);
        ImGui.InputTextWithHint("##tempGame",  "Game Path...",                  ref _tempGamePath,     256);
        ImGui.InputTextWithHint("##tempFile",  "File Path...",                  ref _tempFilePath,     256);
        ImUtf8.InputText("##tempManip"u8, ref _tempManipulation, "Manipulation Base64 String..."u8);
        ImGui.Checkbox("Force Character Collection Overwrite", ref _forceOverwrite);

        using var table = ImRaii.Table(string.Empty, 3, ImGuiTableFlags.SizingFixedFit);
        if (!table)
            return;

        IpcTester.DrawIntro("Last Error", _lastTempError.ToString());
        ImGuiUtil.DrawTableColumn("Last Created Collection");
        ImGui.TableNextColumn();
        using (ImRaii.PushFont(UiBuilder.MonoFont))
        {
            ImGuiUtil.CopyOnClickSelectable(LastCreatedCollectionId.ToString());
        }

        IpcTester.DrawIntro(CreateTemporaryCollection.Label, "Create Temporary Collection");
        if (ImGui.Button("Create##Collection"))
        {
            LastCreatedCollectionId = new CreateTemporaryCollection(pi).Invoke(_tempCollectionName);
            if (_tempGuid == null)
            {
                _tempGuid               = LastCreatedCollectionId;
                _tempCollectionGuidName = LastCreatedCollectionId.ToString();
            }
        }

        var guid = _tempGuid.GetValueOrDefault(Guid.Empty);

        IpcTester.DrawIntro(DeleteTemporaryCollection.Label, "Delete Temporary Collection");
        if (ImGui.Button("Delete##Collection"))
            _lastTempError = new DeleteTemporaryCollection(pi).Invoke(guid);
        ImGui.SameLine();
        if (ImGui.Button("Delete Last##Collection"))
            _lastTempError = new DeleteTemporaryCollection(pi).Invoke(LastCreatedCollectionId);

        IpcTester.DrawIntro(AssignTemporaryCollection.Label, "Assign Temporary Collection");
        if (ImGui.Button("Assign##NamedCollection"))
            _lastTempError = new AssignTemporaryCollection(pi).Invoke(guid, _tempActorIndex, _forceOverwrite);

        IpcTester.DrawIntro(AddTemporaryMod.Label, "Add Temporary Mod to specific Collection");
        if (ImGui.Button("Add##Mod"))
            _lastTempError = new AddTemporaryMod(pi).Invoke(_tempModName, guid,
                new Dictionary<string, string> { { _tempGamePath, _tempFilePath } },
                _tempManipulation.Length > 0 ? _tempManipulation : string.Empty, int.MaxValue);

        IpcTester.DrawIntro(CreateTemporaryCollection.Label, "Copy Existing Collection");
        if (ImGuiUtil.DrawDisabledButton("Copy##Collection", Vector2.Zero,
                "Copies the effective list from the collection named in Temporary Mod Name...",
                !collections.Storage.ByName(_tempModName, out var copyCollection))
         && copyCollection is { HasCache: true })
        {
            var files = copyCollection.ResolvedFiles.ToDictionary(kvp => kvp.Key.ToString(), kvp => kvp.Value.Path.ToString());
            var manips = MetaApi.CompressMetaManipulations(copyCollection);
            _lastTempError = new AddTemporaryMod(pi).Invoke(_tempModName, guid, files, manips, 999);
        }

        IpcTester.DrawIntro(AddTemporaryModAll.Label, "Add Temporary Mod to all Collections");
        if (ImGui.Button("Add##All"))
            _lastTempError = new AddTemporaryModAll(pi).Invoke(_tempModName,
                new Dictionary<string, string> { { _tempGamePath, _tempFilePath } },
                _tempManipulation.Length > 0 ? _tempManipulation : string.Empty, int.MaxValue);

        IpcTester.DrawIntro(RemoveTemporaryMod.Label, "Remove Temporary Mod from specific Collection");
        if (ImGui.Button("Remove##Mod"))
            _lastTempError = new RemoveTemporaryMod(pi).Invoke(_tempModName, guid, int.MaxValue);

        IpcTester.DrawIntro(RemoveTemporaryModAll.Label, "Remove Temporary Mod from all Collections");
        if (ImGui.Button("Remove##ModAll"))
            _lastTempError = new RemoveTemporaryModAll(pi).Invoke(_tempModName, int.MaxValue);
    }

    public void DrawCollections()
    {
        using var collTree = ImRaii.TreeNode("Temporary Collections##TempCollections");
        if (!collTree)
            return;

        using var table = ImRaii.Table("##collTree", 6, ImGuiTableFlags.SizingFixedFit);
        if (!table)
            return;

        foreach (var (collection, idx) in tempCollections.Values.WithIndex())
        {
            using var id = ImRaii.PushId(idx);
            ImGui.TableNextColumn();
            var character = tempCollections.Collections.Where(p => p.Collection == collection).Select(p => p.DisplayName)
                    .FirstOrDefault()
             ?? "Unknown";
            if (ImGui.Button("Save##Collection"))
                TemporaryMod.SaveTempCollection(config, saveService, modManager, collection, character);

            using (ImRaii.PushFont(UiBuilder.MonoFont))
            {
                ImGui.TableNextColumn();
                ImGuiUtil.CopyOnClickSelectable(collection.Identifier);
            }

            ImGuiUtil.DrawTableColumn(collection.Name);
            ImGuiUtil.DrawTableColumn(collection.ResolvedFiles.Count.ToString());
            ImGuiUtil.DrawTableColumn(collection.MetaCache?.Count.ToString() ?? "0");
            ImGuiUtil.DrawTableColumn(string.Join(", ",
                tempCollections.Collections.Where(p => p.Collection == collection).Select(c => c.DisplayName)));
        }
    }

    public void DrawMods()
    {
        using var modTree = ImRaii.TreeNode("Temporary Mods##TempMods");
        if (!modTree)
            return;

        using var table = ImRaii.Table("##modTree", 5, ImGuiTableFlags.SizingFixedFit);

        void PrintList(string collectionName, IReadOnlyList<TemporaryMod> list)
        {
            foreach (var mod in list)
            {
                ImGui.TableNextColumn();
                ImGui.TextUnformatted(mod.Name);
                ImGui.TableNextColumn();
                ImGui.TextUnformatted(mod.Priority.ToString());
                ImGui.TableNextColumn();
                ImGui.TextUnformatted(collectionName);
                ImGui.TableNextColumn();
                ImGui.TextUnformatted(mod.Default.Files.Count.ToString());
                if (ImGui.IsItemHovered())
                {
                    using var tt = ImRaii.Tooltip();
                    foreach (var (path, file) in mod.Default.Files)
                        ImGui.TextUnformatted($"{path} -> {file}");
                }

                ImGui.TableNextColumn();
                ImGui.TextUnformatted(mod.TotalManipulations.ToString());
                if (ImGui.IsItemHovered())
                {
                    using var tt = ImRaii.Tooltip();
                    foreach (var identifier in mod.Default.Manipulations.Identifiers)
                        ImGui.TextUnformatted(identifier.ToString());
                }
            }
        }

        if (table)
        {
            PrintList("All", tempMods.ModsForAllCollections);
            foreach (var (collection, list) in tempMods.Mods)
                PrintList(collection.Name, list);
        }
    }
}