CodeMatcherExtensions.cs

using System;
using System.Linq;
using System.Runtime.CompilerServices;
using HarmonyLib;

namespace AdofaiQolMod;

internal static class CodeMatcherExtensions
{
    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    private static string MatchesToString(params CodeMatch[] matches)
    {
        return string.Join(';', matches.Select(i => i.ToString()));
    }

#if !DEBUG
    [MethodImpl(MethodImplOptions.AggressiveInlining)]
#endif
    internal static CodeMatcher MatchForward(
        this CodeMatcher codeMatcher,
        params CodeMatch[] matches
    )
    {
        var pos = Math.Clamp(codeMatcher.Pos, 0, codeMatcher.Length - 1);
        if (codeMatcher.MatchForward(true, matches).IsInvalid)
        {
            var _pos = -1;
            AdofaiQolMod.Logger.LogError(
                $"Start position: {pos}\n{string.Join('\n', codeMatcher.Instructions().Select(i => $"{(++_pos > pos ? "=" : _pos == pos ? "v" : " ")} #{_pos, 2} {i.opcode}{(i.operand == null ? string.Empty : " " + i.operand)}{(i.labels.Count <= 0 ? string.Empty : " " + string.Join(" ", i.labels.Select(l => $"[{l}]")))}"))}"
            );
            throw new InvalidOperationException(
                $"{nameof(CodeMatcher.MatchForward)} failed: {MatchesToString(matches)}"
            );
        }

        return codeMatcher;
    }

#if !DEBUG
    [MethodImpl(MethodImplOptions.AggressiveInlining)]
#endif
    internal static CodeMatcher MatchBack(this CodeMatcher codeMatcher, params CodeMatch[] matches)
    {
        var pos = Math.Clamp(codeMatcher.Pos, 0, codeMatcher.Length - 1);
        if (codeMatcher.MatchBack(true, matches).IsInvalid)
        {
            var _pos = -1;
            AdofaiQolMod.Logger.LogError(
                $"Start position: {pos}\n{string.Join('\n', codeMatcher.Instructions().Select(i => $"{(++_pos < pos ? "=" : _pos == pos ? "^" : " ")} #{_pos, 2} {i.opcode}{(i.operand == null ? string.Empty : " " + i.operand)}{(i.labels.Count <= 0 ? string.Empty : " " + string.Join(" ", i.labels.Select(l => $"[{l}]")))}"))}"
            );
            throw new InvalidOperationException(
                $"{nameof(CodeMatcher.MatchBack)} failed: {MatchesToString(matches)}"
            );
        }

        return codeMatcher;
    }

#if DEBUG
    [Obsolete("Logs should be removed before release")]
    internal static CodeMatcher LogDebug(this CodeMatcher codeMatcher)
    {
        var pos = 0;
        AdofaiQolMod.Logger.LogDebug(
            $"Current position: {codeMatcher.Pos}\n{string.Join('\n', codeMatcher.Instructions().Select(i => $"{(++pos == codeMatcher.Pos ? ">" : " ")} #{pos, 2} {i.opcode}{(i.operand == null ? string.Empty : " " + i.operand)}{(i.labels.Count <= 0 ? string.Empty : " " + string.Join(" ", i.labels.Select(l => $"[{l}]")))}"))}"
        );
        return codeMatcher;
    }
#endif
}