using QSB.Events;

namespace QSB.Tools
{
    public static class ProbePatches
    {
        private static void ProbeAnchor()
        {
            GlobalMessenger.FireEvent(EventNames.QSBOnProbeAnchor);
        }

        private static bool ProbeWarp(ref bool ____isRetrieving)
        {
            if (!____isRetrieving)
            {
                GlobalMessenger.FireEvent(EventNames.QSBOnProbeWarp);
            }
            return true;
        }

        public static void DoPatches()
        {
            QSB.Helper.HarmonyHelper.AddPostfix<SurveyorProbe>("OnAnchor", typeof(ProbePatches), nameof(ProbeAnchor));
            QSB.Helper.HarmonyHelper.AddPrefix<SurveyorProbe>("Retrieve", typeof(ProbePatches), nameof(ProbeWarp));
        }
    }
}