CustomizeTool/CustomizePlus/Core/Services/HookingService.cs

using Dalamud.Game;
using Dalamud.Hooking;
using Dalamud.Plugin.Services;
using System;
using System.Runtime.InteropServices;
using OtterGui.Log;
using CustomizePlus.Core.Data;
using CustomizePlus.Game.Services;
using CustomizePlus.Configuration.Data;
using CustomizePlus.Profiles;
using CustomizePlus.Armatures.Services;
using CustomizePlus.GameData.Data;
using Penumbra.GameData.Interop;

namespace CustomizePlus.Core.Services;

public class HookingService : IDisposable
{
    private readonly PluginConfiguration _configuration;
    private readonly ISigScanner _sigScanner;
    private readonly IGameInteropProvider _hooker;
    private readonly ProfileManager _profileManager;
    private readonly ArmatureManager _armatureManager;
    private readonly GameStateService _gameStateService;
    private readonly FantasiaPlusDetectService _fantasiaPlusDetectService;
    private readonly Logger _logger;

    private Hook<RenderDelegate>? _renderManagerHook;
    private Hook<GameObjectMovementDelegate>? _gameObjectMovementHook;

    private delegate nint RenderDelegate(nint a1, nint a2, int a3, int a4);

    [UnmanagedFunctionPointer(CallingConvention.StdCall)]
    private delegate void GameObjectMovementDelegate(nint gameObject);

    public bool RenderHookFailed { get; private set; }
    public bool MovementHookFailed { get; private set; }

    public HookingService(
        PluginConfiguration configuration,
        ISigScanner sigScanner,
        IGameInteropProvider hooker,
        ProfileManager profileManager,
        ArmatureManager armatureManager,
        GameStateService gameStateService,
        FantasiaPlusDetectService fantasiaPlusDetectService,
        Logger logger)
    {
        _configuration = configuration;
        _sigScanner = sigScanner;
        _hooker = hooker;
        _profileManager = profileManager;
        _armatureManager = armatureManager;
        _gameStateService = gameStateService;
        _fantasiaPlusDetectService = fantasiaPlusDetectService;
        _logger = logger;

        ReloadHooks();
    }

    public void ReloadHooks()
    {
        RenderHookFailed = false;
        MovementHookFailed = false;

        try
        {
            if (_configuration.PluginEnabled)
            {
                if (_renderManagerHook == null)
                {
                    var renderAddress = _sigScanner.ScanText(Constants.RenderHookAddress);
                    _renderManagerHook = _hooker.HookFromAddress<RenderDelegate>(renderAddress, OnRender);
                    _logger.Debug("Render hook established");
                }

                if (_gameObjectMovementHook == null)
                {
                    var movementAddress = _sigScanner.ScanText(Constants.MovementHookAddress);
                    _gameObjectMovementHook = _hooker.HookFromAddress<GameObjectMovementDelegate>(movementAddress, OnGameObjectMove);
                    _logger.Debug("Movement hook established");
                }


                _logger.Debug("Hooking render manager");
                _renderManagerHook.Enable();

                _logger.Debug("Hooking movement functions");
                _gameObjectMovementHook.Enable();
            }
            else
            {
                _logger.Debug("Unhooking...");
                _renderManagerHook?.Disable();
                _gameObjectMovementHook?.Disable();
            }
        }
        catch (Exception e)
        {
            _logger.Error($"Failed to hook into the game: {e}");
            RenderHookFailed = true;
            MovementHookFailed = true;
            throw;
        }
    }

    private nint OnRender(nint a1, nint a2, int a3, int a4)
    {
        if (_renderManagerHook == null)
        {
            throw new Exception();
        }

        if (_fantasiaPlusDetectService.IsFantasiaPlusInstalled)
        {
            _logger.Error($"Fantasia+ detected, disabling all hooks");
            _renderManagerHook.Disable();
            _gameObjectMovementHook?.Disable();

            return _renderManagerHook.Original(a1, a2, a3, a4);
        }

        try
        {
            _armatureManager.OnRender();
            _profileManager.OnRender();
        }
        catch (Exception e)
        {
            RenderHookFailed = true;
            _logger.Error($"Error in Customize+ render hook {e}");
            _renderManagerHook?.Disable();
        }

        return _renderManagerHook!.Original(a1, a2, a3, a4);
    }

    private unsafe void OnGameObjectMove(nint gameObjectPtr)
    {
        if (_gameObjectMovementHook == null)
        {
            throw new Exception();
        }

        // Call the original function.
        _gameObjectMovementHook.Original(gameObjectPtr);

        // If GPose and a 3rd-party posing service are active simultneously, abort
        if (_gameStateService.GameInPosingMode())
            return;

        try
        {
            var actor = (Actor)gameObjectPtr;
            if (actor.Valid)
                _armatureManager.OnGameObjectMove((Actor)gameObjectPtr);
        }
        catch (Exception ex)
        {
            MovementHookFailed = true;
            _logger.Error($"Exception in Customize+ movement hook: {ex}");
            _gameObjectMovementHook?.Disable();
        }
    }

    public void Dispose()
    {
        _gameObjectMovementHook?.Disable();
        _gameObjectMovementHook?.Dispose();

        _renderManagerHook?.Disable();
        _renderManagerHook?.Dispose();
    }
}