// Reference: Fleck

using System.Collections.Generic;
using Fleck;
using HarmonyLib;
using JetBrains.Annotations;
using Oxide.Core.Plugins;

namespace Oxide.Plugins;

[UsedImplicitly]
[Info("Fleck Limit Lifter", "misticos", "1.0.0")]
public sealed class FleckLimitLifter : CovalencePlugin
{
    [UsedImplicitly]
    [AutoPatch]
    [HarmonyPatch(typeof(WebSocketConnection), "SendBytes")]
    private static class WebSocketConnection_SendBytes_Patch
    {
        [UsedImplicitly]
        [HarmonyTranspiler]
        private static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstruction> instructions)
        {
            var codeMatcher = new CodeMatcher(instructions);

            var pendingSendCount = AccessTools.Field(typeof(WebSocketConnection), "_pendingSendCount");

            codeMatcher.Start()
                .MatchEndForward(
                    CodeMatch.LoadsField(pendingSendCount, true)
                )
                .ThrowIfInvalid($"Failed to match {pendingSendCount}")
                .MatchStartForward(
                    CodeMatch.LoadsConstant(256)
                )
                .ThrowIfInvalid($"Failed to match {256}")
                .Operand = int.MaxValue;

            return codeMatcher.Instructions();
        }
    }
}