CreatureTrackerComponent.cs

using UnityEngine;

namespace CreatureTrackerMinimap
{
    internal class CreatureTrackerComponent : MonoBehaviour
    {
        private string displayName;
        private string displayLevel;
        private Sprite displayIcon;
        private Minimap.PinData pinData;
        private bool isFish;
        private bool showName;

        public void Initialize(string displayName, Sprite displayIcon, int level, bool isFish)
        {
            this.displayName = displayName;
            this.displayIcon = displayIcon;
            this.isFish = isFish;
            SetLevel(level);
        }

        public void SetName(string displayName)
        {
            this.displayName = displayName;
            if (pinData != null) pinData.m_name = GetDisplayName();
        }

        public void SetLevel(int level)
        {
            if (level == 2)
                this.displayLevel = " ⭐";
            else if (level == 3)
                this.displayLevel += " ⭐⭐";
            else
                this.displayLevel = string.Empty;

            if (pinData != null) pinData.m_name = GetDisplayName();
        }

        private string GetDisplayName()
        {
            return showName ? displayName + displayLevel : "";
        }

        private void Update()
        {
            if (isFish && CreatureTrackerMinimap.cfgTrackFish.Value == false)
            {
                OnDestroy();
                return;
            }

            if (pinData == null)
            {
                showName = CreatureTrackerMinimap.cfgShowNames.Value;
                pinData = Minimap.instance.AddPin(base.transform.position, Minimap.PinType.Hildir1, GetDisplayName(), false, false);
                if (displayIcon != null)
                {
                    pinData.m_icon = displayIcon;
                    pinData.m_worldSize = 0f;
                }
            }
            else if (showName != CreatureTrackerMinimap.cfgShowNames.Value)
            {
                showName = CreatureTrackerMinimap.cfgShowNames.Value;
                pinData.m_name = GetDisplayName();
            }

            pinData.m_pos = base.transform.position;
        }

        private void OnDestroy()
        {
            if (pinData == null) return;
            Minimap.instance.RemovePin(pinData);
            pinData = null;
        }
    }
}