Firstborn/Assets/RPG Creation Kit/Scripts/Looting System/Items In Looting UI/ArmorItemInLootingUI_DrawDeposit.cs

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

namespace RPGCreationKit
{
	public class ArmorItemInLootingUI_DrawDeposit : ArmorItemInInventoryUI
	{
		public override void OnClick(bool takeAll = false)
		{
			if (LootingInventoryUI.instance.curIsPickPocket && LootingInventoryUI.instance.curLootingPoint != null)
			{
				// Player is attempting to pickpocket

				Vector3 forward = Inventory.PlayerInventory.transform.TransformDirection(Vector3.forward);
				Vector3 toOther = LootingInventoryUI.instance.curLootingPoint.transform.position - transform.position;
				
				float A = (1f - ((float)armorItemInInventory.item.Value / 100f));
				float B = (float)EntityAttributes.PlayerAttributes.attributes.Dexterity / 100f;
				float C = (1f - ((float)armorItemInInventory.item.Weight / 100f));
				float D = (((Vector3.Dot(forward, toOther)) + 360) / 720);
				
				float ChanceForSuccess = ((A+B+C+D)/4);
				if (base.armorItemInInventory.isEquipped)
					ChanceForSuccess = (ChanceForSuccess/2);

				float RandomChance = (Random.value);
				if (ChanceForSuccess < RandomChance) {
					RckAI ai = LootingInventoryUI.instance.curLootingPoint.inventory.gameObject.GetComponent<RckAI>();
					ai.TryEngageWithPlayer(); // Engage with player - Lets get faction instead of every fucker in the area.
					LootingInventoryUI.instance.CloseUI();
					return;
				} else { // Reward Player
					EntityAttributes.PlayerAttributes.attributes.DexterityPoints += 1;
					A = (float)EntityAttributes.PlayerAttributes.attributes.DexterityPoints;
					B = (float)EntityAttributes.PlayerAttributes.attributes.Dexterity;
					while (A >= (B*3)) {
						EntityAttributes.PlayerAttributes.attributes.Dexterity += 1;
						float E = (A-(B*3f));
						EntityAttributes.PlayerAttributes.attributes.DexterityPoints = (int)D;
						A = (float)EntityAttributes.PlayerAttributes.attributes.DexterityPoints;
						B = (float)EntityAttributes.PlayerAttributes.attributes.Dexterity;
					}
				}
			}
			// Some palce in the codebase seems to be calling this method after the CloseUI randomly, not sure why or where yet
			// this is a backup in that case
			if (LootingInventoryUI.instance.curIsPickPocket && LootingInventoryUI.instance.curLootingPoint == null) {
				return;
			}

			if (LootingInventoryUI.instance.isDrawing)
			{
				// if the amount is 1, add it one time
				if (base.armorItemInInventory.Amount <= 1)
				{
					Inventory.PlayerInventory.AddItem(base.armorItemInInventory.item, base.armorItemInInventory.metadata, 1, !takeAll);

					if(base.armorItemInInventory.isEquipped)
						LootingInventoryUI.instance.curLootingPoint.equipment.Unequip(base.armorItemInInventory);

					// Remove the item from the current loot inventory
					LootingInventoryUI.instance.curLootingPoint.inventory.RemoveItem(base.armorItemInInventory, 1);

					LootingInventoryUI.instance.SelectNextButton();

					// Disable this object
					pool.usedObjects.Remove(this);
					pool.ArmorsPool.usedObjects.Remove(this);

					gameObject.GetComponent<Button>().onClick.RemoveAllListeners();
					gameObject.SetActive(false);
				}
				else if (base.armorItemInInventory.item.isCumulable && base.armorItemInInventory.Amount > 1)
				{
					if (!takeAll)
					{
						LootingInventoryUI.instance.takeDepositItemsPanel.gameObject.SetActive(true);
						LootingInventoryUI.instance.takeDepositItemsPanel.Init(armorItemInInventory, this);
					}
					else
					{
						ConfirmButtonCumulableItem(base.armorItemInInventory.Amount);
					}
				}
			}
			else // we're depositing
			{
				if (base.armorItemInInventory.item.QuestItem)
				{
					AlertMessage.instance.InitAlertMessage("You can't leave Quest Items", AlertMessage.DEFAULT_MESSAGE_DURATION_MEDIUM);
					return;
				}

				// if the amount is 1, add it one time
				if (base.armorItemInInventory.Amount <= 1)
				{
					// Remove the item from the current loot inventory
					LootingInventoryUI.instance.curLootingPoint.inventory.AddItem(base.armorItemInInventory.item, base.armorItemInInventory.metadata, 1);

					if (base.armorItemInInventory.isEquipped)
						Equipment.PlayerEquipment.Unequip(base.armorItemInInventory);

					Inventory.PlayerInventory.RemoveItem(base.armorItemInInventory, 1);

					LootingInventoryUI.instance.SelectNextButton();

					// Disable this object
					pool.usedObjects.Remove(this);
					pool.ArmorsPool.usedObjects.Remove(this);

					gameObject.GetComponent<Button>().onClick.RemoveAllListeners();
					gameObject.SetActive(false);
				}
				else if (base.armorItemInInventory.item.isCumulable && base.armorItemInInventory.Amount > 1)
				{
					LootingInventoryUI.instance.takeDepositItemsPanel.gameObject.SetActive(true);
					LootingInventoryUI.instance.takeDepositItemsPanel.Init(armorItemInInventory, this);
				}
			}
		}

		public override void ConfirmButtonCumulableItem(int amount)
		{
			if (LootingInventoryUI.instance.isDrawing)
			{
				// Add in PlayerInventory
				Inventory.PlayerInventory.AddItem(base.armorItemInInventory.item, base.armorItemInInventory.metadata, amount);

				if(base.armorItemInInventory.Amount <= amount)
					if (base.armorItemInInventory.isEquipped)
						LootingInventoryUI.instance.curLootingPoint.equipment.Unequip(base.armorItemInInventory);

				// Remove from Loot Inventory
				LootingInventoryUI.instance.curLootingPoint.inventory.RemoveItem(base.armorItemInInventory, amount);

				if (armorItemInInventory.Amount <= 0)
				{
					//If we've took all the items
					// Remove item from inventory (whole)
					armorItemInInventory = null;
					armorItemInInventory.item = null;

					LootingInventoryUI.instance.takeDepositItemsPanel.SetPreviousSelected();
					LootingInventoryUI.instance.SelectNextButton(true);

					// Disable this object
					pool.usedObjects.Remove(this);
					pool.ArmorsPool.usedObjects.Remove(this);

					gameObject.GetComponent<Button>().onClick.RemoveAllListeners();
					gameObject.SetActive(false);
				}
				else
				{
					// Update the UI
					Init();
				}
			}
			else // we're depositing
			{
				if(base.armorItemInInventory.Amount <= amount)
					if (base.armorItemInInventory.isEquipped)
					Equipment.PlayerEquipment.Unequip(base.armorItemInInventory);

				// Remove from Loot Inventory
				LootingInventoryUI.instance.curLootingPoint.inventory.AddItem(base.armorItemInInventory.item, base.armorItemInInventory.metadata, amount);



				// Add in PlayerInventory
				Inventory.PlayerInventory.RemoveItem(base.armorItemInInventory, amount);

				if (armorItemInInventory.Amount <= 0)
				{
					LootingInventoryUI.instance.takeDepositItemsPanel.SetPreviousSelected();
					LootingInventoryUI.instance.SelectNextButton(true);

					//If we've took all the items
					pool.usedObjects.Remove(this);
					pool.ArmorsPool.usedObjects.Remove(this);

					// Disable this object
					gameObject.GetComponent<Button>().onClick.RemoveAllListeners();
					gameObject.SetActive(false);
				}
				else
				{
					// Update the UI
					Init();
				}
			}
		}
	}
}