lsh4/Assets/SoftKitty/InventoryEngine/Scripts/UiWindows/MerchantUi.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;

namespace SoftKitty.InventoryEngine
{
    public class MerchantUi : UiWindow
{
        #region Variables
        public RectTransform HoverInfoAnchorPlayer;
        public RectTransform HoverInfoAnchorMerchant;
        public RawImage MerchantAvatar;
        public InventoryItem PlayerItemPrefab;
        public InventoryItem MerchantItemPrefab;
        public CurrenyInfo PlayerCurrencyPrefab;
        public CurrenyInfo MerchantCurrencyPrefab;
        public InputField SearchInput;
        public GameObject ClearFilterButton;
        public GameObject FavActiveIcon;
        public Text MerchantNameText;
        public Text SubtitleText;
        private InventoryItem[] playerItems;
        private InventoryItem[] merchantItems;
        private InventoryHolder playerHolder;
        private InventoryHolder merchantHolder;
        public Button TradeButton;
        private bool filterFav = false;
        private bool inited = false;
        private InventorySnapShot playerSnapshot;
        private InventorySnapShot merchantSnapshot;
        private List<CurrenyInfo> playerCurrencyItems = new List<CurrenyInfo>();
        private List<CurrenyInfo> merchantCurrencyItems = new List<CurrenyInfo>();
        public Image BlockImage;
        #endregion

        #region MonoBehaviour
        void Start()
        {
            FavActiveIcon.GetComponent<Image>().color = InventorySkin.instance.FavoriteColor;
        }

        public override void Update()
        {
            if (!inited) return;
            base.Update();
            FilterItems();
            CheckBlock();
        }
        #endregion



        public void SetMerchantAvatar(Texture _avatar) //Set the avatar image of the merchant to display on the top right of the window
        {
            MerchantAvatar.texture = _avatar;
        }

        public override void Initialize(InventoryHolder _playerHolder, InventoryHolder _merchantHolder, string _merchantName= "Merchant")//Initialize this interface
        {
            playerHolder = _playerHolder;
            merchantHolder = _merchantHolder;
            MerchantNameText.text = _merchantName;
            SubtitleText.text= " "+_merchantName.ToUpper();
            PlayerItemPrefab.Outline.color = InventorySkin.instance.ItemSelectedColor;
            PlayerItemPrefab.Hover.color = InventorySkin.instance.ItemHoverColor;
            PlayerItemPrefab.Fav.GetComponent<Image>().color = InventorySkin.instance.FavoriteColor;
            PlayerItemPrefab.LimitedByOwner = _merchantHolder;
            MerchantItemPrefab.Outline.color = InventorySkin.instance.ItemSelectedColor;
            MerchantItemPrefab.Hover.color = InventorySkin.instance.ItemHoverColor;
            MerchantItemPrefab.Fav.GetComponent<Image>().color = InventorySkin.instance.FavoriteColor;
            MerchantItemPrefab.LimitedByOwner = _playerHolder;
            playerHolder.RegisterItemChangeCallback(OnPlayerItemChange);
            merchantHolder.RegisterItemChangeCallback(OnMerchantItemChange);
            ShowList();
            for (int i = 0; i < playerHolder.Currency.Count; i++)
            {
                GameObject _newItem = Instantiate(PlayerCurrencyPrefab.gameObject, PlayerCurrencyPrefab.transform.parent);
                _newItem.transform.localScale = Vector3.one;
                _newItem.SetActive(true);
                _newItem.GetComponent<CurrenyInfo>().AutoCalculateChanges = false;
                _newItem.GetComponent<CurrenyInfo>().Initialize(i, playerHolder);
                playerCurrencyItems.Add(_newItem.GetComponent<CurrenyInfo>());
            }
            for (int i = 0; i < merchantHolder.Currency.Count; i++)
            {
                GameObject _newItem = Instantiate(MerchantCurrencyPrefab.gameObject, MerchantCurrencyPrefab.transform.parent);
                _newItem.transform.localScale = Vector3.one;
                _newItem.SetActive(true);
                _newItem.GetComponent<CurrenyInfo>().AutoCalculateChanges = false;
                _newItem.GetComponent<CurrenyInfo>().Initialize(i, merchantHolder);
                merchantCurrencyItems.Add(_newItem.GetComponent<CurrenyInfo>());
            }
            playerSnapshot = playerHolder.GetSnapShot();
            merchantSnapshot = merchantHolder.GetSnapShot();
            playerHolder.AutoSave = false;
            merchantHolder.BlockAutoSave = true;
            playerHolder.BlockAutoSave = true;
            inited = true;
        }

        

        private void ShowList()//Show item list for both side
        {
            playerItems = new InventoryItem[playerHolder.Stacks.Count];
            for (int i = 0; i < playerHolder.Stacks.Count; i++)
            {
                GameObject _newItem = Instantiate(PlayerItemPrefab.gameObject, PlayerItemPrefab.transform.parent);
                _newItem.transform.localScale = Vector3.one;
                _newItem.GetComponent<InventoryItem>().Initialize(playerHolder.Stacks[i]);
                _newItem.GetComponent<InventoryItem>().SetHolder(playerHolder);
                _newItem.GetComponent<InventoryItem>().HoverInfoAnchorPoint = HoverInfoAnchorPlayer;
                _newItem.GetComponent<InventoryItem>().RegisterClickCallback(i, OnPlayerItemClick);
                _newItem.GetComponent<InventoryItem>().PriceMultiplier = merchantHolder.BuyPriceMultiplier;
                _newItem.gameObject.SetActive(true);
                playerItems[i] = _newItem.GetComponent<InventoryItem>();
            }

            merchantItems = new InventoryItem[merchantHolder.Stacks.Count];
            for (int i = 0; i < merchantHolder.Stacks.Count; i++)
            {
                GameObject _newItem = Instantiate(MerchantItemPrefab.gameObject, MerchantItemPrefab.transform.parent);
                _newItem.transform.localScale = Vector3.one;
                _newItem.GetComponent<InventoryItem>().Initialize(merchantHolder.Stacks[i]);
                _newItem.GetComponent<InventoryItem>().SetHolder(merchantHolder);
                _newItem.GetComponent<InventoryItem>().HoverInfoAnchorPoint = HoverInfoAnchorMerchant;
                _newItem.GetComponent<InventoryItem>().RegisterClickCallback(i, OnMerchantItemClick);
                _newItem.GetComponent<InventoryItem>().PriceMultiplier = merchantHolder.SellPriceMultiplier;
                _newItem.GetComponent<InventoryItem>().RecieveUntradeable = false;
                _newItem.gameObject.SetActive(true);
                merchantItems[i] = _newItem.GetComponent<InventoryItem>();
            }
        }
        private bool isKeyMatch(AlterKeys _key)
        {
            switch (_key)
            {
                case AlterKeys.None:
                    return true;
                case AlterKeys.LeftCtrl:
                    return InputProxy.GetKey(KeyCode.LeftControl);
                case AlterKeys.LeftAlt:
                    return InputProxy.GetKey(KeyCode.LeftAlt);
                case AlterKeys.LeftShift:
                    return InputProxy.GetKey(KeyCode.LeftShift);
            }
            return false;
        }
        public void OnPlayerItemClick(int _index, int _button)//Callback for when player click an item in player's inventory.
        {
            if ( !playerItems[_index].isEmpty() && playerItems[_index].isTradeable()
                 && (merchantHolder.TradeAllItems || merchantHolder.TradeList.Contains(playerItems[_index].GetItemId()) || merchantHolder.TradeCategoryList.Contains(playerItems[_index].GetCategory())))
            {
                foreach (var setting in ItemManager.instance.clickSettings)
                {
                    if (_button == (int)setting.mouseButton && isKeyMatch(setting.key))
                    {
                        if (setting.function == ClickFunctions.Use )
                        {
                            playerHolder.RemoveItem(playerItems[_index].GetItemId(), 1 - merchantHolder.AddItem(playerItems[_index].GetItem(), 1).Number, _index);
                            SoundManager.Play2D("ItemDrop");
                        }
                        else if (setting.function == ClickFunctions.MarkFavorite)
                        {
                            playerItems[_index].MarkFav();
                        }
                    }
                }
            }
        }

       

        public void OnMerchantItemClick(int _index, int _button)//Callback for when player click an item in merchant's inventory.
        {
            if (!merchantItems[_index].isEmpty() && merchantItems[_index].isTradeable())
            {
                foreach (var setting in ItemManager.instance.clickSettings)
                {
                    if (_button == (int)setting.mouseButton && isKeyMatch(setting.key))
                    {
                        if (setting.function == ClickFunctions.Use)
                        {
                            merchantHolder.RemoveItem(merchantItems[_index].GetItemId(), 1 - playerHolder.AddItem(merchantItems[_index].GetItem(), 1).Number, _index);
                            SoundManager.Play2D("ItemDrop");
                        }
                    }
                }
            }
        }

        private void OnDestroy()//UnRegisterI all callbacks
        {
            playerHolder.UnRegisterItemChangeCallback(OnPlayerItemChange);
            merchantHolder.UnRegisterItemChangeCallback(OnMerchantItemChange);
            merchantHolder.BlockAutoSave = false;
            playerHolder.BlockAutoSave = false;
        }

        public void OnPlayerItemChange(Dictionary<Item, int> _changedItems)//Callback for when items in player's inventory has changed.
        {
            GetCurrencyChange();
        }

        public void OnMerchantItemChange(Dictionary<Item, int> _changedItems)//Callback for when items in merchant's inventory has changed.
        {
            GetCurrencyChange();
        }

        private void GetCurrencyChange()//Compare the items with the snapeshot, calculate the difference by their cost.
        {
            Dictionary<int, int> _playerDic = new Dictionary<int, int>();
            Dictionary<int ,int> _merchantDic = new Dictionary<int, int>();
            foreach (var obj in playerHolder.Stacks)
            {
                if (!obj.isEmpty())
                {
                    if (!_playerDic.ContainsKey(obj.GetItemId()))
                    {
                        _playerDic.Add(obj.GetItemId(), obj.Number);
                    }
                    else
                    {
                        _playerDic[obj.GetItemId()] += obj.Number;
                    }
                }
            }
            foreach (var obj in playerSnapshot.Stacks)
            {
                if (!obj.isEmpty())
                {
                    if (!_playerDic.ContainsKey(obj.GetItemId())) _playerDic.Add(obj.GetItemId(),0);
                }
            }

            foreach (var obj in merchantHolder.Stacks)
            {
                if (!obj.isEmpty())
                {
                    if (!_merchantDic.ContainsKey(obj.GetItemId()))
                    {
                        _merchantDic.Add(obj.GetItemId(), obj.Number);
                    }
                    else
                    {
                        _merchantDic[obj.GetItemId()] += obj.Number;
                    }
                }
            }
            foreach (var obj in merchantSnapshot.Stacks)
            {
                if (!obj.isEmpty())
                {
                    if (!_merchantDic.ContainsKey(obj.GetItemId())) _merchantDic.Add(obj.GetItemId(), 0);
                }
            }

            int[] _currencyChange = new int[ItemManager.instance.currencies.Count];
            foreach (var key in _playerDic.Keys)
            {
                int _change = _playerDic[key] - playerSnapshot.GetItemNumber(key);
                if (_merchantDic.ContainsKey(key)) { //Compare with the change of merchant's item, make sure player is not cheating.
                    int _merchantChange = _merchantDic[key] - merchantSnapshot.GetItemNumber(key);
                    int _finalChange = Mathf.Max(_change, -_merchantChange);//Clamp on what merchant get or lost, make sure player is not cheating.
                    if (_finalChange > 0)
                    {//Buy
                        _currencyChange[ItemManager.itemDic[key].currency] -= Mathf.CeilToInt(ItemManager.itemDic[key].price * merchantHolder.SellPriceMultiplier * _finalChange);
                    }
                    else if (_finalChange < 0)
                    {//Sell
                        _currencyChange[ItemManager.itemDic[key].currency] += Mathf.CeilToInt(ItemManager.itemDic[key].price * merchantHolder.BuyPriceMultiplier * -_finalChange);
                    }
                }
            }
            bool _lackOfCurrency = false;
            for (int i = 0; i < _currencyChange.Length; i++)
            {
                playerHolder.SetCurrency(i,playerSnapshot.GetCurrency(i,false)+ _currencyChange[i]);
                merchantHolder.SetCurrency(i,merchantSnapshot.GetCurrency(i, false) - _currencyChange[i]);
                if (playerHolder.GetCurrency(i, true) < 0 || merchantHolder.GetCurrency(i,true) < 0) _lackOfCurrency = true;
            }
            TradeButton.interactable = !_lackOfCurrency;
            TradeButton.GetComponent<CanvasGroup>().alpha = _lackOfCurrency ? 0.2F : 1F;
            List<int> _playerCurrencyDifference = playerHolder.GetCurrencyDifference(playerSnapshot.CurrencyValue);
            for (int i = 0; i < playerCurrencyItems.Count; i++) playerCurrencyItems[i].SetChangeText(_playerCurrencyDifference[i]);
            List<int> _merchantCurrencyDifference = merchantHolder.GetCurrencyDifference(merchantSnapshot.CurrencyValue);
            for (int i = 0; i < merchantCurrencyItems.Count; i++) merchantCurrencyItems[i].SetChangeText(_merchantCurrencyDifference[i]);
        }

        public override void Close()//Revert to the snapshot for both side if the window is closed.
        {
            playerHolder.RevertSnapShot(playerSnapshot);
            merchantHolder.RevertSnapShot(merchantSnapshot);
            base.Close();
        }

        public void Cancel()//Revert to the snapshot for both side
        {
            SoundManager.Play2D("Sort");
            playerHolder.RevertSnapShot(playerSnapshot);
            merchantHolder.RevertSnapShot(merchantSnapshot);
            foreach (var obj in playerCurrencyItems)
            {
                obj.SetChangeText(0);
            }
            foreach (var obj in merchantCurrencyItems)
            {
                obj.SetChangeText(0);
            }
        }

        public void Purchase()//Confirm the purchase and close the window
        {
            SoundManager.Play2D("Trade");
            SoundManager.Play2D("DoneDeal");
            merchantHolder.BlockAutoSave = false;
            playerHolder.BlockAutoSave = false;
            playerHolder.Save();
            merchantHolder.Save();
            playerSnapshot = playerHolder.GetSnapShot();
            merchantSnapshot = merchantHolder.GetSnapShot();
            foreach (var obj in playerCurrencyItems) {
                obj.SetChangeText(0);
            }
            foreach (var obj in merchantCurrencyItems)
            {
                obj.SetChangeText(0);
            }
        }

        public void FilterItems()//Filter the items by their name and "isTradeable" mark
        {
            for (int i = 0; i < playerItems.Length; i++)
            {
                playerItems[i].SetVisible(playerItems[i].isNameMatch(SearchInput.text) && (!filterFav || playerItems[i].Fav.activeSelf) && playerItems[i].isTradeable(true)
                     && (merchantHolder.TradeAllItems || merchantHolder.TradeList.Contains(playerItems[i].GetItemId()) || merchantHolder.TradeCategoryList.Contains(playerItems[i].GetCategory())));
            }
            for (int i = 0; i < merchantItems.Length; i++)
            {
                merchantItems[i].SetVisible(merchantItems[i].isNameMatch(SearchInput.text));
            }
            ClearFilterButton.SetActive(SearchInput.text != "");
        }

        public void ClearFilter()//Clear all filters.
        {
            SearchInput.text = "";
            ClearFilterButton.SetActive(false);
            filterFav = false;
            for (int i = 0; i < playerItems.Length; i++)
            {
                playerItems[i].SetVisible(true);
            }
            for (int i = 0; i < merchantItems.Length; i++)
            {
                merchantItems[i].SetVisible(true);
            }
        }

        public void ShowFavItems()//Only show favorite items
        {
            SoundManager.Play2D("Tab");
            filterFav = !filterFav;
            FavActiveIcon.SetActive(filterFav);
            FilterItems();
        }

        private void CheckBlock()//Check if the interface should be blocked.
        {
            if (NumberInput.instance != null)
            {
                BlockImage.gameObject.SetActive(true);
                BlockImage.color = new Color(0F, 0F, 0F, Mathf.Lerp(BlockImage.color.a, 0.94F, Time.deltaTime * 3F));
            }
            else
            {
                if (BlockImage.color.a > 0F)
                {
                    BlockImage.color = new Color(0F, 0F, 0F, Mathf.MoveTowards(BlockImage.color.a, 0F, Time.deltaTime * 2F));
                }
                else
                {
                    if (BlockImage.gameObject.activeSelf) BlockImage.gameObject.SetActive(false);
                }
            }
        }
    }
}