Vendor.cs

using UnityEngine;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using Lidgren.Network;
using Janus;

[System.Serializable]
public class VendorItemInfo
{
    public string ItemID;
    public int Price;
    public string Currency;
    public Item Item { get { return ItemRegistry.GetItem<Item>(ItemID); } }
}

[AddComponentMenu("Liberi/Vendor")]
[RequireComponent(typeof(Inspectable))]
public class Vendor : MonoBehaviour, IActionInfoProvider
{
    public string VendorName;
    public AudioClip PurchaseSound;
    public AudioClip EquipSound;
    public AudioClip ErrorSound;
    public AudioClip MenuScrollSound;

    Timeline<GameMessage> _purchaseAttempts;
    Timeline<GameMessage> _equipAttempts;

    VendorItemInfo[] _vendorItems;
    Inspectable _inspectable;
    int _selectionIndex;
    bool _selectionOwned;
    bool _selectionEquipped;

    bool _actionTimedout;
    float _timeoutLength = 0.1f;

    void Awake()
    {
        _purchaseAttempts = Sync.CreateEvent<GameMessage>();
        _equipAttempts = Sync.CreateEvent<GameMessage>();
        _vendorItems = new VendorItemInfo[0];
        _inspectable = GetComponent<Inspectable>();
    }

    void OnSpawn()
    {
        if (Game.IsServer)
        {
            _purchaseAttempts.EntryPassed += OnPurchaseAttempted;
            _equipAttempts.EntryPassed += OnEquipAttempt;
            GameServer.Instance.PullWorldData(name, OnVendorDataReceived);
            return;
        }

        GameClient.Instance.PullWorldData(name, OnVendorDataReceived);
    }

    void OnPurchaseAttempted(Timeline<GameMessage> timeline, TimelineEntry<GameMessage> entry)
    {
        GameMessage purchaseMessage = entry.Value;

        string playerID = purchaseMessage.ReadString();
        string itemID = purchaseMessage.ReadString();

        VendorItemInfo itemInfo = GetVendorItemInfoByID(itemID);
        PlayerProfile profile = GameServer.Instance.GetPlayerProfile(playerID);

        if (CanAfford(profile, itemInfo))
        {
            if (!profile.Inventory["Items"].HasChild(itemID))
                profile.Inventory["Items"].AddChild(itemID, 0);

            profile.Inventory["Items"][itemID].IntValue++;

            GameServer.Instance.PushPlayerData(playerID, "Inventory", "Items/" + itemID, profile.Inventory["Items"][itemID].Value);

            profile.Inventory["Currency"][itemInfo.Currency].IntValue -= itemInfo.Price;
            GameServer.Instance.PushPlayerData(playerID, "Inventory", "Currency/" + itemInfo.Currency, profile.Inventory["Currency"][itemInfo.Currency].Value);

            GameMessage itemMsg = new GameMessage();
            itemMsg.Write((byte)PlayerProfileOperationType.InventoryUpdate);
            itemMsg.Write("Items/" + itemID);
            itemMsg.Write(profile.Inventory["Items"][itemID].Value);

            GameServer.Instance.SendPlayerProfileOperation(profile, itemMsg);

            GameMessage currencyMsg = new GameMessage();
            currencyMsg.Write((byte)PlayerProfileOperationType.InventoryUpdate);
            currencyMsg.Write("Currency/" + itemInfo.Currency);
            currencyMsg.Write(profile.Inventory["Currency"][itemInfo.Currency].Value);

            GameServer.Instance.SendPlayerProfileOperation(profile, currencyMsg);
        }
    }

    void OnEquipAttempt(Timeline<GameMessage> timeline, TimelineEntry<GameMessage> entry)
    {
        GameMessage equipMessage = entry.Value;

        string playerID = equipMessage.ReadString();
        string itemID = equipMessage.ReadString();
        string itemType = ItemRegistry.GetItem<Item>(itemID).Type;

        PlayerProfile profile = GameServer.Instance.GetPlayerProfile(playerID);

        if (itemType == "hero")
        {
            HeroItem heroItem = ItemRegistry.GetItem<HeroItem>(itemID);
            string slot = heroItem.Slot.ToString().ToLower();

            if (profile.Inventory["Items"].HasChild(itemID))
            {
                GameServer.Instance.PushPlayerData(playerID, "Styles", "hero/" + slot, itemID);

                GameMessage stylesMsg = new GameMessage();
                stylesMsg.Write((byte)PlayerProfileOperationType.StylesUpdate);
                stylesMsg.Write("hero");
                stylesMsg.Write(slot);
                stylesMsg.Write(itemID);

                GameServer.Instance.SendPlayerProfileOperation(profile, stylesMsg, true);

                profile.Styles["hero"][slot].Value = itemID;
            }
        }

        else if (itemType == "guardian")
        {
            GameServer.Instance.PushPlayerData(playerID, "Styles", "guardian/weapon", itemID);

            GameMessage stylesMsg = new GameMessage();
            stylesMsg.Write((byte)PlayerProfileOperationType.StylesUpdate);
            stylesMsg.Write("guardian");
            stylesMsg.Write("weapon");
            stylesMsg.Write(itemID);

            GameServer.Instance.SendPlayerProfileOperation(profile, stylesMsg, true);

            profile.Styles["guardian"]["weapon"].Value = itemID;
        }
    }

    public string GetActionCaption(ControlsAction actionType)
    {
        if (actionType == ControlsAction.Primary)
        {
            if (_selectionOwned && !_selectionEquipped && (_vendorItems[_selectionIndex].Item.Equippable))
                return Localizer.GetString("caption.equip");
            else if (!_selectionOwned)
                return Localizer.GetString("caption.buy");
        }
        else if (actionType == ControlsAction.Back)
        {
            return Localizer.GetString("caption.close"); ;
        }

        return "";
    }

    public bool IsActionEnabled(ControlsAction actionType)
    {
        if (actionType == ControlsAction.Primary)
        {
            if (_selectionOwned && !_selectionEquipped)
                return true;
            else if (!_selectionOwned && CanAfford(Game.LocalPlayerProfile, _vendorItems[_selectionIndex]))
                return true;
        }
        else if (actionType == ControlsAction.Back)
        {
            return true;
        }

        return false;
    }

    bool CanAfford(PlayerProfile profile, VendorItemInfo itemInfo)
    {
        if (!profile.Inventory["Currency"].HasChild("libi"))
            return false;

        return profile.Inventory["Currency"][itemInfo.Currency].IntValue >= itemInfo.Price;
    }

    void OnVendorDataReceived(UJeli vendorData)
    {
        var inventoryData = vendorData["Inventory"];

        if (inventoryData != null)
        {
            var itemDatas = inventoryData.GetChildren("Item");
            _vendorItems = new VendorItemInfo[itemDatas.Length];

            for (int i = 0; i < itemDatas.Length; i++)
            {
                var itemData = itemDatas[i];

                var item = new VendorItemInfo()
                {
                    ItemID = itemData.Value,
                    Price = itemData["Price"].IntValue,
                    Currency = itemData["Currency"].Value,
                };

                _vendorItems[i] = item;
            }
        }
    }

    VendorItemInfo GetVendorItemInfoByID(string itemID)
    {
        return _vendorItems.Where(i => i.ItemID == itemID).FirstOrDefault();
    }

    void OnStartInspecting()
    {
        Controls.RegisterActionInfoProvider(ControlsAction.Primary, this);
        Controls.RegisterActionInfoProvider(ControlsAction.Back, this);
        StartCoroutine(ActionTimeout());
        _selectionIndex = 0;
        StoreUIManager.Show(_vendorItems, VendorName);
        UpdateSelectionOwned(_vendorItems[_selectionIndex]);
        StoreUIManager.UpdateSelectionIndex(_selectionIndex, _selectionOwned);
    }

    IEnumerator ActionTimeout()
    {
        _actionTimedout = true;
        yield return new WaitForSeconds(_timeoutLength);
        _actionTimedout = false;
    }

    void Update()
    {
        if (Game.IsServer)
            return;

        if (_inspectable.IsInspecting && !_actionTimedout)
        {
            if (Controls.GetActionDown(ControlsAction.Primary))
            {
                if (!_selectionOwned && CanAfford(Game.LocalPlayerProfile, _vendorItems[_selectionIndex]))
                {
                    MakePurchase(_vendorItems[_selectionIndex]);
                }
                else if (_selectionOwned && (_vendorItems[_selectionIndex].Item.Equippable))
                {
                    EquipItem(_vendorItems[_selectionIndex]);
                }
            }
            else if (Controls.GetActionDown(ControlsAction.Back))
            {
                _inspectable.StopInspecting();
            }
        }
    }

    void OnStopInspecting()
    {
        Controls.UnregisterActionInfoProvider(ControlsAction.Primary, this);
        Controls.UnregisterActionInfoProvider(ControlsAction.Back, this);
        StoreUIManager.Hide();
    }

    void OnScroll(Vector3 scrollDir)
    {
        int newSelectionIndex = _selectionIndex + (int)scrollDir.x + (int)scrollDir.y * -4;

        if (newSelectionIndex >= _vendorItems.Length || newSelectionIndex < 0)
            return;

        _selectionIndex = newSelectionIndex;
        UpdateSelectionOwned(_vendorItems[_selectionIndex]);
        StoreUIManager.UpdateSelectionIndex(_selectionIndex, _selectionOwned);
        this.PlayPooledSound(MenuScrollSound);
    }

    void UpdateSelectionOwned(VendorItemInfo itemInfo)
    {
        _selectionOwned = Game.LocalPlayerProfile.Inventory["Items"].HasChild(itemInfo.ItemID);

        if (_selectionOwned)
            _selectionEquipped = Game.LocalPlayerProfile.Styles[itemInfo.Item.Type].GetChildValues<string>(s => s).Contains(itemInfo.ItemID);
        else
            _selectionEquipped = false;
    }

    private void EquipItem(VendorItemInfo itemInfo)
    {
        GameMessage equipAttempt = new GameMessage();

        equipAttempt.Write(Game.LocalPlayerID);
        equipAttempt.Write(itemInfo.ItemID);

        _equipAttempts[0] = equipAttempt;

        StoreUIManager.BounceSelectedIcon();

        this.PlayPooledSound(EquipSound);
    }

    private void MakePurchase(VendorItemInfo itemInfo)
    {
        GameMessage purchaseMessage = new GameMessage();

        purchaseMessage.Write(Game.LocalPlayerID);
        purchaseMessage.Write(itemInfo.ItemID);

        _purchaseAttempts[0] = purchaseMessage;

        // Update the local state instantly so the item appears bought (assumes no cheating, server can correct otherwise)
        PlayerProfile profile = Game.LocalPlayerProfile;

        if (!profile.Inventory["Items"].HasChild(itemInfo.ItemID))
            profile.Inventory["Items"].AddChild(itemInfo.ItemID, 0);

        profile.Inventory["Items"][itemInfo.ItemID].IntValue++;

        UpdateSelectionOwned(itemInfo);
        StoreUIManager.UpdateSelectionIndex(_selectionIndex, _selectionOwned);
        StoreUIManager.BounceSelectedIcon();

        if (itemInfo.Item.Equippable)
            EquipItem(itemInfo);

        this.PlayPooledSound(PurchaseSound);
    }
}