using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.Versioning;
using BepInEx;
using BepInEx.Configuration;
using BepInEx.Logging;
using HarmonyLib;
using TMPro;
using Unity.Netcode;
using UnityEngine;
using UnityEngine.InputSystem;
using UnityEngine.InputSystem.Controls;
using UnityEngine.SceneManagement;
using UnityEngine.UI;

[assembly: CompilationRelaxations(8)]
[assembly: RuntimeCompatibility(WrapNonExceptionThrows = true)]
[assembly: Debuggable(DebuggableAttribute.DebuggingModes.Default | DebuggableAttribute.DebuggingModes.DisableOptimizations | DebuggableAttribute.DebuggingModes.IgnoreSymbolStoreSequencePoints | DebuggableAttribute.DebuggingModes.EnableEditAndContinue)]
[assembly: TargetFramework(".NETFramework,Version=v4.8", FrameworkDisplayName = ".NET Framework 4.8")]
[assembly: AssemblyVersion("0.0.0.0")]
namespace HostRestartFix;

[BepInPlugin("net.vanilla.hostrestartfix", "HostRestartFix", "1.0.0")]
public class Plugin : BaseUnityPlugin
{
	private class DebugLogPatch
	{
		private static bool betaSaveDataDetected;

		private static bool hostStartDetected;

		private static void Prefix(object message)
		{
			if (message != null)
			{
				string text = message.ToString();
				if (text.Contains("host"))
				{
					logger.LogInfo((object)("Host-related message detected: '" + text + "'"));
				}
				if (text.Contains("Has beta save data"))
				{
					logger.LogInfo((object)("Beta save data message detected: '" + text + "'"));
					betaSaveDataDetected = true;
					CheckBothConditions();
				}
				if (text.Contains("[Info   : Unity Log] started host!") || text.Contains("started host") || (text.Contains("host") && text.Contains("start")))
				{
					logger.LogInfo((object)("Host start message detected: '" + text + "'"));
					logger.LogInfo((object)$"Current flags - hasBeenForceQuit: {hasBeenForceQuit}, firstHostDetected: {firstHostDetected}, betaSaveDataDetected: {betaSaveDataDetected}");
					hostStartDetected = true;
					CheckBothConditions();
				}
			}
		}

		private static void CheckBothConditions()
		{
			logger.LogInfo((object)$"Checking both conditions - hostStartDetected: {hostStartDetected}, betaSaveDataDetected: {betaSaveDataDetected}, hasBeenForceQuit: {hasBeenForceQuit}");
			if (!hasBeenForceQuit && hostStartDetected && betaSaveDataDetected)
			{
				logger.LogInfo((object)"Both conditions met! Will force quit to menu in 10 seconds.");
				firstHostDetected = true;
				if (!((Object)(object)Instance == (Object)null))
				{
					try
					{
						logger.LogInfo((object)"Starting ForceQuitToMenu coroutine");
						((MonoBehaviour)Instance).StartCoroutine(ForceQuitToMenu());
						logger.LogInfo((object)"Coroutine started successfully");
						return;
					}
					catch (Exception ex)
					{
						logger.LogError((object)("Failed to start coroutine: " + ex.Message));
						logger.LogError((object)ex.StackTrace);
						hasBeenForceQuit = true;
						ForceReturnToMainMenu();
						return;
					}
				}
				logger.LogError((object)"Plugin Instance is null! Cannot start coroutine!");
			}
			else
			{
				logger.LogInfo((object)$"Not all conditions met: hostStartDetected={hostStartDetected}, betaSaveDataDetected={betaSaveDataDetected}, hasBeenForceQuit={hasBeenForceQuit}");
			}
		}
	}

	[CompilerGenerated]
	private sealed class <ForceQuitToMenu>d__13 : IEnumerator<object>, IDisposable, IEnumerator
	{
		private int <>1__state;

		private object <>2__current;

		private float <timer>5__1;

		object IEnumerator<object>.Current
		{
			[DebuggerHidden]
			get
			{
				return <>2__current;
			}
		}

		object IEnumerator.Current
		{
			[DebuggerHidden]
			get
			{
				return <>2__current;
			}
		}

		[DebuggerHidden]
		public <ForceQuitToMenu>d__13(int <>1__state)
		{
			this.<>1__state = <>1__state;
		}

		[DebuggerHidden]
		void IDisposable.Dispose()
		{
			<>1__state = -2;
		}

		private bool MoveNext()
		{
			//IL_0114: Unknown result type (might be due to invalid IL or missing references)
			//IL_011e: Expected O, but got Unknown
			//IL_01ac: Unknown result type (might be due to invalid IL or missing references)
			//IL_01b6: Expected O, but got Unknown
			//IL_021a: Unknown result type (might be due to invalid IL or missing references)
			//IL_0224: Expected O, but got Unknown
			switch (<>1__state)
			{
			default:
				return false;
			case 0:
				<>1__state = -1;
				logger.LogInfo((object)"ForceQuitToMenu coroutine started");
				if ((Object)(object)messageObject != (Object)null)
				{
					messageObject.SetActive(true);
					if ((Object)(object)messageText != (Object)null)
					{
						((TMP_Text)messageText).text = "The Mods have Pre-loaded. Returning to menu in 10 seconds.\nPlease re-host the game for proper weather loading.";
					}
					logger.LogInfo((object)"Message displayed");
				}
				else
				{
					logger.LogError((object)"messageObject is null!");
				}
				<timer>5__1 = 10f;
				goto IL_0145;
			case 1:
				<>1__state = -1;
				<timer>5__1--;
				goto IL_0145;
			case 2:
				<>1__state = -1;
				logger.LogInfo((object)"Showing completion message");
				if ((Object)(object)messageObject != (Object)null && (Object)(object)messageText != (Object)null)
				{
					messageObject.SetActive(true);
					((TMP_Text)messageText).text = "Please start hosting again for proper weather loading.";
					<>2__current = (object)new WaitForSeconds(5f);
					<>1__state = 3;
					return true;
				}
				break;
			case 3:
				{
					<>1__state = -1;
					messageObject.SetActive(false);
					break;
				}
				IL_0145:
				if (<timer>5__1 > 0f)
				{
					if ((Object)(object)messageText != (Object)null)
					{
						((TMP_Text)messageText).text = $"The Mods have Pre-loaded. Returning to menu in {<timer>5__1:0} seconds.\nPlease re-host the game for proper weather loading.";
					}
					logger.LogInfo((object)$"Timer: {<timer>5__1}");
					<>2__current = (object)new WaitForSeconds(1f);
					<>1__state = 1;
					return true;
				}
				logger.LogInfo((object)"Timer completed, setting hasBeenForceQuit=true");
				hasBeenForceQuit = true;
				if ((Object)(object)messageObject != (Object)null)
				{
					messageObject.SetActive(false);
				}
				logger.LogInfo((object)"Calling ForceReturnToMainMenu");
				ForceReturnToMainMenu();
				<>2__current = (object)new WaitForSeconds(2f);
				<>1__state = 2;
				return true;
			}
			logger.LogInfo((object)"ForceQuitToMenu coroutine completed successfully");
			return false;
		}

		bool IEnumerator.MoveNext()
		{
			//ILSpy generated this explicit interface implementation from .override directive in MoveNext
			return this.MoveNext();
		}

		[DebuggerHidden]
		void IEnumerator.Reset()
		{
			throw new NotSupportedException();
		}
	}

	internal static ManualLogSource logger;

	private static bool firstHostDetected;

	private static bool hasBeenForceQuit;

	private static GameObject messageObject;

	private static TextMeshProUGUI messageText;

	internal static Plugin Instance;

	internal static ConfigEntry<Key> RestartKey;

	private static bool keyPressedNow;

	private void Awake()
	{
		//IL_0059: Unknown result type (might be due to invalid IL or missing references)
		//IL_005f: Expected O, but got Unknown
		//IL_02b5: Unknown result type (might be due to invalid IL or missing references)
		//IL_02bc: Expected O, but got Unknown
		//IL_00e6: Unknown result type (might be due to invalid IL or missing references)
		//IL_00f4: Expected O, but got Unknown
		//IL_0161: Unknown result type (might be due to invalid IL or missing references)
		//IL_0168: Expected O, but got Unknown
		//IL_0194: Unknown result type (might be due to invalid IL or missing references)
		//IL_019e: Expected O, but got Unknown
		//IL_01ce: Unknown result type (might be due to invalid IL or missing references)
		//IL_01e5: Unknown result type (might be due to invalid IL or missing references)
		//IL_01fc: Unknown result type (might be due to invalid IL or missing references)
		//IL_0213: Unknown result type (might be due to invalid IL or missing references)
		//IL_0252: Unknown result type (might be due to invalid IL or missing references)
		logger = ((BaseUnityPlugin)this).Logger;
		logger.LogInfo((object)"Plugin net.vanilla.hostrestartfix is loaded!");
		Instance = this;
		logger.LogInfo((object)"Plugin instance set");
		RestartKey = ((BaseUnityPlugin)this).Config.Bind<Key>("General", "RestartKey", (Key)32, "This key, when pressed in combination with Ctrl+Shift+Alt, will force return to main menu (useful if mods break)");
		try
		{
			Harmony val = new Harmony("net.vanilla.hostrestartfix");
			MethodInfo methodInfo = AccessTools.Method(typeof(Debug), "Log", new Type[1] { typeof(object) }, (Type[])null);
			if (methodInfo == null)
			{
				logger.LogError((object)"Failed to find Debug.Log method to patch!");
				return;
			}
			MethodInfo methodInfo2 = AccessTools.Method(typeof(DebugLogPatch), "Prefix", (Type[])null, (Type[])null);
			if (methodInfo2 == null)
			{
				logger.LogError((object)"Failed to find prefix method for patching!");
				return;
			}
			val.Patch((MethodBase)methodInfo, new HarmonyMethod(methodInfo2), (HarmonyMethod)null, (HarmonyMethod)null, (HarmonyMethod)null, (HarmonyMethod)null);
			logger.LogInfo((object)"Debug.Log patch applied successfully");
			val.PatchAll(Assembly.GetExecutingAssembly());
			logger.LogInfo((object)"All patches applied successfully");
		}
		catch (Exception ex)
		{
			logger.LogError((object)("Failed to apply Harmony patch: " + ex.Message));
			logger.LogError((object)ex.StackTrace);
			return;
		}
		try
		{
			GameObject val2 = new GameObject("RestartFixMessageCanvas");
			Object.DontDestroyOnLoad((Object)(object)val2);
			Canvas val3 = val2.AddComponent<Canvas>();
			val3.renderMode = (RenderMode)0;
			val3.sortingOrder = 10000;
			messageObject = new GameObject("RestartFixMessage");
			messageObject.transform.SetParent(((Component)val3).transform, false);
			RectTransform val4 = messageObject.AddComponent<RectTransform>();
			val4.anchorMin = new Vector2(0.5f, 0.5f);
			val4.anchorMax = new Vector2(0.5f, 0.5f);
			val4.pivot = new Vector2(0.5f, 0.5f);
			val4.sizeDelta = new Vector2(800f, 100f);
			messageText = messageObject.AddComponent<TextMeshProUGUI>();
			((TMP_Text)messageText).fontSize = 24f;
			((TMP_Text)messageText).alignment = (TextAlignmentOptions)514;
			((Graphic)messageText).color = Color.red;
			((TMP_Text)messageText).text = "";
			messageObject.SetActive(false);
			logger.LogInfo((object)"UI components created successfully");
		}
		catch (Exception ex2)
		{
			logger.LogError((object)("Error creating UI: " + ex2.Message));
		}
		try
		{
			GameObject val5 = new GameObject("ReHosterKeyHandler");
			val5.AddComponent<KeybindHandlerBehaviour>();
			Object.DontDestroyOnLoad((Object)(object)val5);
			logger.LogInfo((object)"Keybind handler created successfully");
		}
		catch (Exception ex3)
		{
			logger.LogError((object)("Error creating keybind handler: " + ex3.Message));
		}
		hasBeenForceQuit = false;
		firstHostDetected = false;
		logger.LogInfo((object)("Flags reset: hasBeenForceQuit=" + hasBeenForceQuit));
	}

	private void Update()
	{
		if (Keyboard.current != null)
		{
			CheckForceRestartKeybind();
		}
	}

	public static void CheckForceRestartKeybind()
	{
		//IL_0053: Unknown result type (might be due to invalid IL or missing references)
		if (Keyboard.current == null)
		{
			return;
		}
		if (Keyboard.current.ctrlKey.isPressed && Keyboard.current.altKey.isPressed && Keyboard.current.shiftKey.isPressed)
		{
			KeyControl val = ((IEnumerable<KeyControl>)(object)Keyboard.current.allKeys).FirstOrDefault((Func<KeyControl, bool>)((KeyControl key) => key.keyCode == RestartKey.Value));
			if (val != null && ((ButtonControl)val).isPressed)
			{
				if (!keyPressedNow)
				{
					logger.LogInfo((object)"Force restart keybind pressed!");
					ForceReturnToMainMenu();
				}
				keyPressedNow = true;
			}
			else
			{
				keyPressedNow = false;
			}
		}
		else
		{
			keyPressedNow = false;
		}
	}

	public static void ForceReturnToMainMenu()
	{
		logger.LogInfo((object)"ForceReturnToMainMenu called");
		if ((Object)(object)NetworkManager.Singleton != (Object)null && NetworkManager.Singleton.IsConnectedClient)
		{
			try
			{
				GameNetworkManager instance = GameNetworkManager.Instance;
				if (instance != null)
				{
					instance.Disconnect();
				}
				logger.LogInfo((object)"Attempted clean disconnect through GameNetworkManager");
			}
			catch (Exception ex)
			{
				logger.LogError((object)("Error during disconnect: " + ex.Message));
			}
			try
			{
				NetworkManager.Singleton.Shutdown(false);
				logger.LogInfo((object)"Force shutdown NetworkManager");
			}
			catch (Exception ex2)
			{
				logger.LogError((object)("Error during NetworkManager shutdown: " + ex2.Message));
			}
		}
		else
		{
			logger.LogInfo((object)"Not connected to network or NetworkManager.Singleton is null");
		}
		if ((Object)(object)GameNetworkManager.Instance != (Object)null)
		{
			try
			{
				MethodInfo method = typeof(GameNetworkManager).GetMethod("ResetGameValuesToDefault", BindingFlags.Instance | BindingFlags.NonPublic);
				if (method != null)
				{
					method.Invoke(GameNetworkManager.Instance, new object[0]);
					logger.LogInfo((object)"Reset game values to default");
				}
				else
				{
					logger.LogError((object)"ResetGameValuesToDefault method not found");
				}
			}
			catch (Exception ex3)
			{
				logger.LogError((object)("Error resetting game values: " + ex3.Message));
			}
		}
		else
		{
			logger.LogInfo((object)"GameNetworkManager.Instance is null");
		}
		try
		{
			logger.LogInfo((object)"Attempting to load main menu scene by index");
			SceneManager.LoadScene(0);
			logger.LogInfo((object)"Loaded main menu scene");
		}
		catch (Exception ex4)
		{
			logger.LogError((object)("Error loading main menu scene: " + ex4.Message));
			try
			{
				logger.LogInfo((object)"Attempting to load main menu scene by name");
				SceneManager.LoadScene("MainMenu");
				logger.LogInfo((object)"Loaded main menu scene by name");
			}
			catch (Exception ex5)
			{
				logger.LogError((object)("Error loading main menu scene by name: " + ex5.Message));
			}
		}
	}

	[IteratorStateMachine(typeof(<ForceQuitToMenu>d__13))]
	private static IEnumerator ForceQuitToMenu()
	{
		//yield-return decompiler failed: Unexpected instruction in Iterator.Dispose()
		return new <ForceQuitToMenu>d__13(0);
	}
}
public class KeybindHandlerBehaviour : MonoBehaviour
{
	private void Update()
	{
		Plugin.CheckForceRestartKeybind();
	}
}
[HarmonyPatch]
internal class ReHosterPatches
{
	[HarmonyPostfix]
	[HarmonyPatch(typeof(StartOfRound), "Start")]
	public static void InGameSetup()
	{
		//IL_001d: Unknown result type (might be due to invalid IL or missing references)
		//IL_0023: Expected O, but got Unknown
		GameObject val = GameObject.Find("ReHosterKeyHandler");
		if ((Object)(object)val == (Object)null)
		{
			val = new GameObject("ReHosterKeyHandler");
			val.AddComponent<KeybindHandlerBehaviour>();
			Object.DontDestroyOnLoad((Object)(object)val);
			Plugin.logger.LogInfo((object)"Created KeybindHandlerBehaviour in game scene");
		}
	}
}
public static class PluginInfo
{
	public const string PLUGIN_GUID = "net.vanilla.hostrestartfix";

	public const string PLUGIN_NAME = "HostRestartFix";

	public const string PLUGIN_VERSION = "1.0.0";
}