#if CARBON
#else
// Reference: 0Harmony
using Harmony;
#endif
using System.Collections.Generic;
using System.Reflection.Emit;

namespace Oxide.Plugins
{
    [Info("Server Projectile Hit Fix", "nivex", "1.0.1")]
    [Description("Fix to set initiator for fire entities.")]

    public class ServerProjectileHitFix : RustPlugin
    {
#if CARBON
        public override bool AutoPatch => true;
#else
        private HarmonyInstance _harmony;
#endif

#if CARBON
#else
        private void Loaded()
        {
            _harmony = HarmonyInstance.Create(Name + "Patch");
            System.Type[] patchType =
            {
                AccessTools.Inner(typeof(ServerProjectileHitFix), "ItemModProjectileSpawn_ServerProjectileHit"),
            };
            foreach (var t in patchType) { new PatchProcessor(_harmony, t, HarmonyMethod.Merge(t.GetHarmonyMethods())).Patch(); }
        }
#endif

#if CARBON
#else
        private void Unload()
        {
            _harmony?.UnpatchAll(Name + "Patch");
        }
#endif

        [HarmonyPatch(typeof(ItemModProjectileSpawn), "ServerProjectileHit", typeof(HitInfo))]
        internal class ItemModProjectileSpawn_ServerProjectileHit
        {
            [HarmonyTranspiler]
            private static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstruction> instructions)
            {
                List<CodeInstruction> codes = new(instructions);
                List<CodeInstruction> patch = new()
                {
                    new(OpCodes.Ldarg_1),
                    new(OpCodes.Ldfld, AccessTools.Field(typeof(HitInfo), "Initiator")),
                    new(OpCodes.Stfld, AccessTools.Field(typeof(BaseEntity), "creatorEntity")),
                };
                for (int i = 0; i < codes.Count; i++)
                {
                    if (i - 1 >= 0 && codes[i - 1].ToString() == "callvirt Void set_rotation(UnityEngine.Quaternion)")
                    {
                        CodeInstruction copy = new(OpCodes.Ldloc_S, codes[i].operand);
                        codes.InsertRange(i, patch);
                        codes.Insert(i, copy);
                    }
                }
                return codes;
            }
        }
    }
}