using HarmonyLib;
using QSB.CampfireSync.Messages;
using QSB.CampfireSync.WorldObjects;
using QSB.Messaging;
using QSB.Patches;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.CampfireSync.Patches;

[HarmonyPatch]
internal class CampfirePatches : QSBPatch
{
	public override QSBPatchTypes Type => QSBPatchTypes.OnClientConnect;

	[HarmonyPrefix]
	[HarmonyPatch(typeof(Campfire), nameof(Campfire.OnPressInteract))]
	public static bool LightCampfireEvent(Campfire __instance)
	{
		var qsbCampfire = __instance.GetWorldObject<QSBCampfire>();
		if (__instance._state == Campfire.State.LIT)
		{
			qsbCampfire.StartRoasting();
		}
		else
		{
			qsbCampfire.SetState(Campfire.State.LIT);
			qsbCampfire.SendMessage(new CampfireStateMessage(Campfire.State.LIT));
			Locator.GetFlashlight().TurnOff(false);
		}

		return false;
	}

	[HarmonyPrefix]
	[HarmonyPatch(typeof(Campfire), nameof(Campfire.Update))]
	public static bool UpdateReplacement(Campfire __instance)
	{
		var targetLitFraction = 0f;
		switch (__instance._state)
		{
			case Campfire.State.UNLIT:
				targetLitFraction = 0f;
				break;
			case Campfire.State.LIT:
				targetLitFraction = 1f;
				break;
			case Campfire.State.SMOLDERING:
				targetLitFraction = 0.4f;
				break;
		}

		if (__instance._litFraction != targetLitFraction)
		{
			__instance.SetLitFraction(Mathf.MoveTowards(__instance._litFraction, targetLitFraction, Time.deltaTime));
		}

		if (__instance._canSleepHere)
		{
			__instance._sleepPrompt.SetVisibility(false);
			if (__instance._interactVolumeFocus && !__instance._isPlayerSleeping && !__instance._isPlayerRoasting && OWInput.IsInputMode(InputMode.Character))
			{
				__instance._sleepPrompt.SetVisibility(true);
				__instance._sleepPrompt.SetDisplayState(__instance.CanSleepHereNow() ? ScreenPrompt.DisplayState.Normal : ScreenPrompt.DisplayState.GrayedOut);
				if (OWInput.IsNewlyPressed(InputLibrary.interactSecondary, InputMode.All) && __instance.CanSleepHereNow())
				{
					__instance.StartSleeping();
				}
			}
		}

		if (__instance._isPlayerSleeping)
		{
			__instance._wakePrompt.SetVisibility(OWInput.IsInputMode(InputMode.None) && Time.timeSinceLevelLoad - __instance._fastForwardStartTime > __instance.GetWakePromptDelay());
			if (__instance.ShouldWakeUp())
			{
				__instance.StopSleeping(false);
				return false;
			}
		}

		return false;
	}
}