HuntBuddy/Interface.cs

using System;
using System.Linq;
using System.Numerics;
using System.Threading.Tasks;
using Dalamud.Game.Text.SeStringHandling.Payloads;
using Dalamud.Interface;
using Dalamud.Logging;
using HuntBuddy.Structs;
using ImGuiNET;
using Lumina.Excel.GeneratedSheets;

namespace HuntBuddy
{
	public class Interface
	{
		private readonly Plugin _plugin;

		public bool DrawInterface;
		private bool _drawConfigurationInterface;

		public Interface(Plugin plugin)
		{
			_plugin = plugin;
		}

		public unsafe bool Draw()
		{
			var draw = true;

			var fontGlobalScale = ImGui.GetIO().FontGlobalScale;

			ImGui.SetNextWindowSize(new Vector2(400 * ImGui.GetIO().FontGlobalScale, 500), ImGuiCond.Once);

			if (!ImGui.Begin($"{_plugin.Name}", ref draw, ImGuiWindowFlags.NoDocking))
			{
				return draw;
			}

			if (!_plugin.MobHuntEntriesReady)
			{
				ImGui.Text("Reloading data ...");
				ImGui.End();
				return draw;
			}

			if (IconButton(FontAwesomeIcon.Redo, "Reload"))
			{
				ImGui.End();
				_plugin.MobHuntEntriesReady = false;
				Task.Run(() => _plugin.ReloadData());
				return draw;
			}

			if (ImGui.IsItemHovered())
			{
				ImGui.BeginTooltip();
				ImGui.Text("Click this button to reload daily hunt data");
				ImGui.EndTooltip();
			}
			
			ImGui.SameLine();

			if (IconButton(FontAwesomeIcon.Cog, "Config"))
			{
				this._drawConfigurationInterface = !this._drawConfigurationInterface;
			}

			foreach (var expansionEntry in _plugin.MobHuntEntries.Where(expansionEntry =>
				         ImGui.TreeNode(expansionEntry.Key)))
			{
				foreach (var entry in expansionEntry.Value.Where(entry =>
					         ImGui.TreeNode(
						         $"{entry.Key.Value} ({entry.Value.Count(x => _plugin.MobHuntStruct->CurrentKills[x.CurrentKillsOffset] == x.NeededKills)}/{entry.Value.Count})")))
				{
					ImGui.Indent();
					foreach (var mobHuntEntry in entry.Value)
					{
						if (Location.Database.ContainsKey(mobHuntEntry.MobHuntId))
						{
							if (IconButton(FontAwesomeIcon.Search, $"##{mobHuntEntry.MobHuntId}"))
							{
								Location.OpenMapLink(mobHuntEntry.TerritoryType, mobHuntEntry.MapId,
									mobHuntEntry.MobHuntId);
							}

							ImGui.SameLine();
						}

						var currentKills = _plugin.MobHuntStruct->CurrentKills[mobHuntEntry.CurrentKillsOffset];
						ImGui.Text(mobHuntEntry.Name);
						if (ImGui.IsItemHovered())
						{
							ImGui.PushStyleColor(ImGuiCol.PopupBg, Vector4.Zero);
							ImGui.BeginTooltip();
							var imageSize = 256f * fontGlobalScale;
							var cursorPos = ImGui.GetCursorScreenPos();
							ImGui.InvisibleButton("canvas", new Vector2(imageSize));

							var drawList = ImGui.GetWindowDrawList();
							if (mobHuntEntry.ExpansionId == 4 &&
							    mobHuntEntry.MobHuntType == 1) // Endwalker uses circle for non elite mobs
							{
								drawList.AddCircleFilled(cursorPos + new Vector2(imageSize / 2f), imageSize / 2f,
									_plugin.Configuration.IconBackgroundColourU32);
							}
							else
							{
								drawList.AddRectFilled(cursorPos, cursorPos + new Vector2(imageSize), _plugin.Configuration.IconBackgroundColourU32);
							}

							drawList.AddImage(mobHuntEntry.Icon.ImGuiHandle, cursorPos,
								new Vector2(cursorPos.X + imageSize, cursorPos.Y + imageSize));
							ImGui.PopStyleColor();
							ImGui.EndTooltip();
						}

						ImGui.SameLine();
						if (currentKills != mobHuntEntry.NeededKills)
						{
							ImGui.Text($"({currentKills}/{mobHuntEntry.NeededKills})");
						}
						else
						{
							ImGui.TextColored(new Vector4(0f, 1f, 0f, 1f),
								$"({currentKills}/{mobHuntEntry.NeededKills})");
						}
					}

					ImGui.Unindent();
					ImGui.TreePop();
				}

				ImGui.TreePop();
			}

			ImGui.End();

			if (_drawConfigurationInterface)
			{
				this.DrawConfiguration();
			}

			return draw;
		}

		public unsafe void DrawLocalHunts()
		{
			if (!_plugin.Configuration.ShowLocalHunts ||
			    _plugin.CurrentAreaMobHuntEntries.IsEmpty ||
			    _plugin.CurrentAreaMobHuntEntries.Count(x =>
				    _plugin.MobHuntStruct->CurrentKills[x.CurrentKillsOffset] == x.NeededKills) ==
			    _plugin.CurrentAreaMobHuntEntries.Count)
			{
				return;
			}

			var fontGlobalScale = ImGui.GetIO().FontGlobalScale;

			ImGui.SetNextWindowSize(Vector2.Zero, ImGuiCond.Always);

			var windowFlags = ImGuiWindowFlags.NoNavInputs | ImGuiWindowFlags.NoDocking;

			if (_plugin.Configuration.HideLocalHuntBackground)
			{
				windowFlags |= ImGuiWindowFlags.NoBackground;
			}

			if (!ImGui.Begin("Hunts in current area", windowFlags))
			{
				return;
			}

			foreach (var mobHuntEntry in _plugin.CurrentAreaMobHuntEntries)
			{
				var currentKills = _plugin.MobHuntStruct->CurrentKills[mobHuntEntry.CurrentKillsOffset];

				if (_plugin.Configuration.HideCompletedHunts && currentKills == mobHuntEntry.NeededKills)
				{
					continue;
				}

				if (Location.Database.ContainsKey(mobHuntEntry.MobHuntId))
				{
					if (IconButton(FontAwesomeIcon.Search, $"##{mobHuntEntry.MobHuntId}"))
					{
						Location.OpenMapLink(mobHuntEntry.TerritoryType, mobHuntEntry.MapId,
							mobHuntEntry.MobHuntId);
					}

					ImGui.SameLine();
				}

				ImGui.Text($"{mobHuntEntry.Name} ({currentKills}/{mobHuntEntry.NeededKills})");

				if (!_plugin.Configuration.ShowLocalHuntIcons)
				{
					continue;
				}

				var imageSize = 128f * fontGlobalScale;
				ImGui.SetCursorPosX(ImGui.GetWindowWidth() / 2f * fontGlobalScale - imageSize / 2f);
				var cursorPos = ImGui.GetCursorScreenPos();
				ImGui.InvisibleButton("canvas", new Vector2(imageSize));

				var drawList = ImGui.GetWindowDrawList();
				if (mobHuntEntry.ExpansionId == 4 &&
				    mobHuntEntry.MobHuntType == 1) // Endwalker uses circle for non elite mobs
				{
					drawList.AddCircleFilled(cursorPos + new Vector2(imageSize / 2f), imageSize / 2f, _plugin.Configuration.IconBackgroundColourU32);
				}
				else
				{
					drawList.AddRectFilled(cursorPos, cursorPos + new Vector2(imageSize), _plugin.Configuration.IconBackgroundColourU32);
				}

				drawList.AddImage(mobHuntEntry.Icon.ImGuiHandle, cursorPos,
					new Vector2(cursorPos.X + imageSize, cursorPos.Y + imageSize));
			}

			ImGui.End();
		}

		private void DrawConfiguration()
		{
			ImGui.SetNextWindowSize(Vector2.Zero, ImGuiCond.Always);

			if (!ImGui.Begin($"{_plugin.Name} settings", ImGuiWindowFlags.NoDocking))
			{
				return;
			}

			var save = false;

			save |= ImGui.Checkbox("Show hunts in local area", ref _plugin.Configuration.ShowLocalHunts);
			save |= ImGui.Checkbox("Show icons of hunts in local area", ref _plugin.Configuration.ShowLocalHuntIcons);
			save |= ImGui.Checkbox("Hide background of local hunts window",
				ref _plugin.Configuration.HideLocalHuntBackground);
			save |= ImGui.Checkbox("Hide completed targets in local hunts window",
				ref _plugin.Configuration.HideCompletedHunts);
			if (ImGui.ColorEdit4("Hunt icon background colour", ref _plugin.Configuration.IconBackgroundColour))
			{
				_plugin.Configuration.IconBackgroundColourU32 =
					ImGui.ColorConvertFloat4ToU32(_plugin.Configuration.IconBackgroundColour);
				save = true;
			}

			if (save)
			{
				_plugin.Configuration.Save();
			}


			ImGui.End();
		}

		private static bool IconButton(FontAwesomeIcon icon, string? id = null)
		{
			ImGui.PushFont(UiBuilder.IconFont);

			var text = icon.ToIconString();
			if (id != null)
			{
				text += $"##{id}";
			}

			var result = ImGui.Button(text);

			ImGui.PopFont();

			return result;
		}
	}
}