112 lines
3.4 KiB
C#
112 lines
3.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UnityEngine;
|
|
using UnityEngine.UI;
|
|
|
|
public class AbilityManager : MonoBehaviour
|
|
{
|
|
#region Statication
|
|
|
|
public static AbilityManager instance;
|
|
|
|
private void Awake()
|
|
{
|
|
if (instance != null && instance != this)
|
|
{
|
|
Destroy(gameObject);
|
|
return;
|
|
}
|
|
instance = this;
|
|
}
|
|
|
|
#endregion
|
|
|
|
public class StoredUpgrade
|
|
{
|
|
public AbilityUpgrade upgrade;
|
|
public int count;
|
|
}
|
|
public MarisaAbilityHandler player;
|
|
public AbilityUpgrade upgradeToAdd;
|
|
public Button upgradeButton;
|
|
|
|
[Header("Upgrades")]
|
|
public HashSet<StoredUpgrade> upgradesInventory = new();
|
|
private void Start()
|
|
{
|
|
// upgradeButton.onClick.AddListener((() => AddUpgrade(upgradeToAdd, player.mainAttackInstance)));
|
|
StoreUpgrade(upgradeToAdd);
|
|
}
|
|
|
|
public void StoreUpgrade(AbilityUpgrade upgradeToStore)
|
|
{
|
|
foreach (StoredUpgrade storedUpgrade in upgradesInventory)
|
|
{
|
|
if (storedUpgrade.upgrade == upgradeToStore)
|
|
{
|
|
storedUpgrade.count++;
|
|
Debug.Log($"Added upgrade {storedUpgrade.upgrade.upgradeName}. Current count: {storedUpgrade.count}");
|
|
AbilityUIHandler.instance.UpdateInventory();
|
|
return;
|
|
}
|
|
}
|
|
|
|
StoredUpgrade newUpgrade = new()
|
|
{
|
|
upgrade = upgradeToStore,
|
|
count = 1
|
|
};
|
|
upgradesInventory.Add(newUpgrade);
|
|
Debug.Log($"Added upgrade {newUpgrade.upgrade.upgradeName}. Current count: {newUpgrade.count}");
|
|
AbilityUIHandler.instance.UpdateInventory();
|
|
}
|
|
public void AddUpgrade(AbilityUpgrade upgrade, PlayerAbility ability)
|
|
{
|
|
foreach (StoredUpgrade storedUpgrade in upgradesInventory.ToArray())
|
|
{
|
|
if (storedUpgrade.upgrade == upgrade)
|
|
{
|
|
storedUpgrade.count--;
|
|
if (storedUpgrade.count <= 0)
|
|
{
|
|
upgradesInventory.Remove(storedUpgrade);
|
|
AbilityUIHandler.instance.UpdateInventory();
|
|
}
|
|
if (!ability.attachedUpgrades.Contains(upgrade))
|
|
{
|
|
AbilityUpgrade newUpgrade = Instantiate(upgrade, ability.transform);
|
|
ability.attachedUpgrades.Add(newUpgrade);
|
|
newUpgrade.thisPlayerAbility = ability;
|
|
newUpgrade.ApplyUpgrade();
|
|
}
|
|
else
|
|
{
|
|
ability.attachedUpgrades.TryGetValue(upgrade, out AbilityUpgrade foundUpgrade);
|
|
if (foundUpgrade)
|
|
{
|
|
foundUpgrade.count++;
|
|
foundUpgrade.ApplyUpgrade();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void RemoveUpgrade(AbilityUpgrade upgrade, PlayerAbility ability)
|
|
{
|
|
if (ability.attachedUpgrades.TryGetValue(upgrade, out AbilityUpgrade foundUpgrade))
|
|
{
|
|
if (foundUpgrade.count > 1)
|
|
{
|
|
foundUpgrade.ApplyRemoval();
|
|
foundUpgrade.count--;
|
|
}
|
|
else
|
|
{
|
|
foundUpgrade.ApplyRemoval();
|
|
ability.attachedUpgrades.Remove(foundUpgrade);
|
|
}
|
|
}
|
|
}
|
|
}
|