using CounterStrikeSharp.API.Core;
using CounterStrikeSharp.API.Modules.Commands;
using CounterStrikeSharp.API.Modules.Menu;

namespace WeaponPaints
{
	public partial class WeaponPaints
	{
		private void OnCommandRefresh(CCSPlayerController? player, CommandInfo command)
		{
			if (!Config.AdditionalSetting.CommandWpEnabled || !Config.AdditionalSetting.SkinEnabled || !g_bCommandsAllowed) return;
			if (!Utility.IsPlayerValid(player)) return;
            if (player == null || !player.IsValid || player.UserId == null || player.Index <= 0 || player.IsBot) return;

			PlayerInfo playerInfo = new PlayerInfo
			{
				UserId = player.UserId,
				Index = (int)player.Index,
				SteamId = player?.AuthorizedSteamID?.SteamId64,
				Name = player?.PlayerName,
				IpAddress = player?.IpAddress?.Split(":")[0]
			};

			if (!commandsCooldown.TryGetValue((int)player!.UserId, out DateTime cooldownEndTime) ||
				DateTime.UtcNow >= (commandsCooldown.TryGetValue((int)player.UserId, out cooldownEndTime) ? cooldownEndTime : DateTime.UtcNow))
			{
				commandsCooldown[(int)player.UserId] = DateTime.UtcNow.AddSeconds(Config.CmdRefreshCooldownSeconds);
				if (weaponSync != null)
					Task.Run(async () => await weaponSync.GetWeaponPaintsFromDatabase(playerInfo));
				if (Config.AdditionalSetting.KnifeEnabled)
				{
					if (weaponSync != null)
						Task.Run(async () => await weaponSync.GetKnifeFromDatabase(playerInfo));

					RefreshWeapons(player);
				}
				if (!string.IsNullOrEmpty(Localizer["wp_command_refresh_done"]))
				{
					player!.Print(Localizer["wp_command_refresh_done"]);
				}
				return;
			}
			if (!string.IsNullOrEmpty(Localizer["wp_command_cooldown"]))
			{
				player!.Print(Localizer["wp_command_cooldown"]);
			}
		}

		private void OnCommandWS(CCSPlayerController? player, CommandInfo command)
		{
            if (!Config.AdditionalSetting.SkinEnabled) return;
			if (!Utility.IsPlayerValid(player)) return;

			if (!string.IsNullOrEmpty(Localizer["wp_info_website"]))
			{
				player!.Print(Localizer["wp_info_website", Config.Website]);
			}
			if (!string.IsNullOrEmpty(Localizer["wp_info_refresh"]))
			{
				player!.Print(Localizer["wp_info_refresh"]);
			}
			if (!Config.AdditionalSetting.KnifeEnabled) return;
			if (!string.IsNullOrEmpty(Localizer["wp_info_knife"]))
			{
				player!.Print(Localizer["wp_info_knife"]);
			}
		}

		private void RegisterCommands()
		{
			AddCommand($"css_{Config.AdditionalSetting.CommandSkin}", "Skins info", (player, info) =>
			{
				if (!Utility.IsPlayerValid(player)) return;
				OnCommandWS(player, info);
			});
			AddCommand($"css_{Config.AdditionalSetting.CommandRefresh}", "Skins refresh", (player, info) =>
			{
				if (!Utility.IsPlayerValid(player) || !g_bCommandsAllowed) return;
				OnCommandRefresh(player, info);
			});
			if (Config.AdditionalSetting.CommandKillEnabled)
			{
				AddCommand($"css_{Config.AdditionalSetting.CommandKill}", "kill yourself", (player, info) =>
				{
					if (player == null || !Utility.IsPlayerValid(player) || player.PlayerPawn.Value == null || !player!.PlayerPawn.IsValid) return;

					player.PlayerPawn.Value.CommitSuicide(true, false);
				});
			}
		}

		private void SetupKnifeMenu()
		{
			if (!Config.AdditionalSetting.KnifeEnabled || !g_bCommandsAllowed) return;

			var knivesOnly = weaponList
				.Where(pair => pair.Key.StartsWith("weapon_knife") || pair.Key.StartsWith("weapon_bayonet"))
				.ToDictionary(pair => pair.Key, pair => pair.Value);

			var giveItemMenu = new ChatMenu(Localizer["wp_knife_menu_title"]);
			var handleGive = (CCSPlayerController? player, ChatMenuOption option) =>
			{
				if (Utility.IsPlayerValid(player))
				{
					if (player == null) return;
					var knifeName = option.Text;
					var knifeKey = knivesOnly.FirstOrDefault(x => x.Value == knifeName).Key;
					if (!string.IsNullOrEmpty(knifeKey))
					{
						if (!string.IsNullOrEmpty(Localizer["wp_knife_menu_select"]))
						{
							player!.Print(Localizer["wp_knife_menu_select", knifeName]);
						}

						if (!string.IsNullOrEmpty(Localizer["wp_knife_menu_kill"]) && Config.AdditionalSetting.CommandKillEnabled)
						{
							player!.Print(Localizer["wp_knife_menu_kill"]);
						}

						PlayerInfo playerInfo = new PlayerInfo
						{
							UserId = player.UserId,
							Index = (int)player.Index,
							SteamId = player?.AuthorizedSteamID?.SteamId64,
							Name = player?.PlayerName,
							IpAddress = player?.IpAddress?.Split(":")[0]
						};

						g_playersKnife[(int)player!.Index] = knifeKey;

						if (player!.PawnIsAlive && g_bCommandsAllowed)
						{
							RefreshWeapons(player);
						}

						if (weaponSync != null)
							Task.Run(async () => await weaponSync.SyncKnifeToDatabase(playerInfo, knifeKey));
					}
				}
			};
			foreach (var knifePair in knivesOnly)
			{
				giveItemMenu.AddMenuOption(knifePair.Value, handleGive);
			}
			AddCommand($"css_{Config.AdditionalSetting.CommandKnife}", "Knife Menu", (player, info) =>
			{
				if (!Utility.IsPlayerValid(player) || !g_bCommandsAllowed) return;

				if (player == null || player.UserId == null) return;

				if (!commandsCooldown.TryGetValue((int)player.UserId, out DateTime cooldownEndTime) ||
					DateTime.UtcNow >= (commandsCooldown.TryGetValue((int)player.UserId, out cooldownEndTime) ? cooldownEndTime : DateTime.UtcNow))
				{
					commandsCooldown[(int)player.UserId] = DateTime.UtcNow.AddSeconds(Config.CmdRefreshCooldownSeconds);
					ChatMenus.OpenMenu(player, giveItemMenu);
					return;
				}
				if (!string.IsNullOrEmpty(Localizer["wp_command_cooldown"]))
				{
					player!.Print(Localizer["wp_command_cooldown"]);
				}
			});
		}

		private void SetupSkinsMenu()
		{
			var classNamesByWeapon = weaponList.ToDictionary(kvp => kvp.Value, kvp => kvp.Key);
			var weaponSelectionMenu = new ChatMenu(Localizer["wp_skin_menu_weapon_title"]);

			// Function to handle skin selection for a specific weapon
			var handleWeaponSelection = (CCSPlayerController? player, ChatMenuOption option) =>
			{
				if (!Utility.IsPlayerValid(player)) return;

				int playerIndex = (int)player!.Index;
				string selectedWeapon = option.Text;
				if (classNamesByWeapon.TryGetValue(selectedWeapon, out string? selectedWeaponClassname))
				{
					if (selectedWeaponClassname == null) return;
					var skinsForSelectedWeapon = skinsList?.Where(skin =>
					skin != null &&
					skin.TryGetValue("weapon_name", out var weaponName) &&
					weaponName?.ToString() == selectedWeaponClassname
				)?.ToList();

					var skinSubMenu = new ChatMenu(Localizer["wp_skin_menu_skin_title", selectedWeapon]);

					// Function to handle skin selection for the chosen weapon
					var handleSkinSelection = (CCSPlayerController? p, ChatMenuOption opt) =>
					{
						if (p == null || !p.IsValid || p.Index <= 0) return;

						playerIndex = (int)p.Index;

						if (p.AuthorizedSteamID == null) return;

						string steamId = p.AuthorizedSteamID.SteamId64.ToString();
						var firstSkin = skinsList?.FirstOrDefault(skin =>
						{
							if (skin != null && skin.TryGetValue("weapon_name", out var weaponName))
							{
								return weaponName?.ToString() == selectedWeaponClassname;
							}
							return false;
						});
						string selectedSkin = opt.Text;
						string selectedPaintID = selectedSkin.Split('(')[1].Trim(')').Trim();

						if (firstSkin != null &&
							firstSkin.TryGetValue("weapon_defindex", out var weaponDefIndexObj) &&
							weaponDefIndexObj != null &&
							ushort.TryParse(weaponDefIndexObj.ToString(), out var weaponDefIndex) &&
                            ushort.TryParse(selectedPaintID, out var paintID))
						{
							p!.Print(Localizer["wp_skin_menu_select", selectedSkin]);

                            if (!gPlayerWeaponsInfo[playerIndex].TryGetValue(weaponDefIndex, out _))
                            {
                                gPlayerWeaponsInfo[playerIndex][weaponDefIndex] = new WeaponInfo();
                            }

                            gPlayerWeaponsInfo[playerIndex][weaponDefIndex].Paint = paintID;
							gPlayerWeaponsInfo[playerIndex][weaponDefIndex].Wear = 0.00001f;
							gPlayerWeaponsInfo[playerIndex][weaponDefIndex].Seed = 0;

							PlayerInfo playerInfo = new PlayerInfo
							{
								UserId = player.UserId,
								Index = (int)player.Index,
								SteamId = player?.AuthorizedSteamID?.SteamId64,
								Name = player?.PlayerName,
								IpAddress = player?.IpAddress?.Split(":")[0]
							};

							if (!Config.GlobalShare)
							{
								if (weaponSync != null)
									Task.Run(async () => await weaponSync.SyncWeaponPaintToDatabase(playerInfo, weaponDefIndex));
							}
						}
					};

					// Add skin options to the submenu for the selected weapon
					if (skinsForSelectedWeapon != null)
					{
						foreach (var skin in skinsForSelectedWeapon.Where(s => s != null))
						{
							if (skin.TryGetValue("paint_name", out var paintNameObj) && skin.TryGetValue("paint", out var paintObj))
							{
								var paintName = paintNameObj?.ToString();
								var paint = paintObj?.ToString();

								if (!string.IsNullOrEmpty(paintName) && !string.IsNullOrEmpty(paint))
								{
									skinSubMenu.AddMenuOption($"{paintName} ({paint})", handleSkinSelection);
								}
							}
						}
					}

					// Open the submenu for skin selection of the chosen weapon
					ChatMenus.OpenMenu(player, skinSubMenu);
				}
			};

			// Add weapon options to the weapon selection menu
			foreach (var weaponClass in weaponList.Keys)
			{
				string weaponName = weaponList[weaponClass];
				weaponSelectionMenu.AddMenuOption(weaponName, handleWeaponSelection);
			}
			// Command to open the weapon selection menu for players
			AddCommand($"css_{Config.AdditionalSetting.CommandSkinSelection}", "Skins selection menu", (player, info) =>
			{
				if (!Utility.IsPlayerValid(player)) return;

				if (player == null || player.UserId == null) return;

				if (!commandsCooldown.TryGetValue((int)player.UserId, out DateTime cooldownEndTime) ||
					DateTime.UtcNow >= (commandsCooldown.TryGetValue((int)player.UserId, out cooldownEndTime) ? cooldownEndTime : DateTime.UtcNow))
				{
					commandsCooldown[(int)player.UserId] = DateTime.UtcNow.AddSeconds(Config.CmdRefreshCooldownSeconds);
					ChatMenus.OpenMenu(player, weaponSelectionMenu);
					return;
				}
				if (!string.IsNullOrEmpty(Localizer["wp_command_cooldown"]))
				{
					player!.Print(Localizer["wp_command_cooldown"]);
				}
			});
		}
	}
}