Glamourer/Glamourer/Interop/Penumbra/PenumbraAutoRedraw.cs

using Dalamud.Plugin.Services;
using Glamourer.Api.Enums;
using Glamourer.Designs.History;
using Glamourer.Events;
using Glamourer.Interop.Structs;
using Glamourer.State;
using OtterGui.Classes;
using OtterGui.Services;
using Penumbra.Api.Enums;

namespace Glamourer.Interop.Penumbra;

public class PenumbraAutoRedraw : IDisposable, IRequiredService
{
    private const    int             WaitFrames = 5;
    private readonly Configuration   _config;
    private readonly PenumbraService _penumbra;
    private readonly StateManager    _state;
    private readonly ObjectManager   _objects;
    private readonly IFramework      _framework;
    private readonly StateChanged    _stateChanged;

    public PenumbraAutoRedraw(PenumbraService penumbra, Configuration config, StateManager state, ObjectManager objects, IFramework framework,
        StateChanged stateChanged)
    {
        _penumbra                   =  penumbra;
        _config                     =  config;
        _state                      =  state;
        _objects                    =  objects;
        _framework                  =  framework;
        _stateChanged               =  stateChanged;
        _penumbra.ModSettingChanged += OnModSettingChange;
        _framework.Update           += OnFramework;
        _stateChanged.Subscribe(OnStateChanged, StateChanged.Priority.PenumbraAutoRedraw);
    }

    public void Dispose()
    {
        _penumbra.ModSettingChanged -= OnModSettingChange;
        _framework.Update           -= OnFramework;
        _stateChanged.Unsubscribe(OnStateChanged);
    }

    private readonly ConcurrentQueue<(ActorState, Action, int)> _actions = [];
    private readonly ConcurrentSet<ActorState>                  _skips   = [];
    private          DateTime                                   _frame;

    private void OnStateChanged(StateChangeType type, StateSource source, ActorState state, ActorData _1, ITransaction? _2)
    {
        if (type is StateChangeType.Design && source.IsIpc())
            _skips.TryAdd(state);
    }

    private void OnFramework(IFramework _)
    {
        var count = _actions.Count;
        while (_actions.TryDequeue(out var tuple) && count-- > 0)
        {
            if (_skips.ContainsKey(tuple.Item1))
            {
                _skips.TryRemove(tuple.Item1);
                continue;
            }

            if (tuple.Item3 > 0)
                _actions.Enqueue((tuple.Item1, tuple.Item2, tuple.Item3 - 1));
            else
                tuple.Item2();
        }
    }

    private void OnModSettingChange(ModSettingChange type, Guid collectionId, string mod, bool inherited)
    {
        if (type is ModSettingChange.TemporaryMod)
        {
            _framework.RunOnFrameworkThread(() =>
            {
                _objects.Update();
                foreach (var (id, state) in _state)
                {
                    if (!_objects.TryGetValue(id, out var actors) || !actors.Valid)
                        continue;

                    var collection = _penumbra.GetActorCollection(actors.Objects[0], out _);
                    if (collection != collectionId)
                        continue;

                    _actions.Enqueue((state, () =>
                    {
                        foreach (var actor in actors.Objects)
                            _state.ReapplyState(actor, state, false, StateSource.IpcManual);
                        Glamourer.Log.Debug($"Automatically applied mod settings of type {type} to {id.Incognito(null)}.");
                    }, WaitFrames));
                }
            });
        }
        else if (_config.AutoRedrawEquipOnChanges)
        {
            // Only update once per frame.
            var playerName = _penumbra.GetCurrentPlayerCollection();
            if (playerName != collectionId)
                return;

            var currentFrame = _framework.LastUpdateUTC;
            if (currentFrame == _frame)
                return;

            _frame = currentFrame;
            _framework.RunOnFrameworkThread(() =>
            {
                _state.ReapplyState(_objects.Player, false, StateSource.IpcManual);
                Glamourer.Log.Debug(
                    $"Automatically applied mod settings of type {type} to {_objects.PlayerData.Identifier.Incognito(null)} (Local Player).");
            });
        }
    }
}