1 files changed, 46 insertions, 0 deletions
diff --git a/Thronefall_v1.0/Decompile/UpgradeCastleUp.cs b/Thronefall_v1.0/Decompile/UpgradeCastleUp.cs
new file mode 100644
index 0000000..802d0f6
--- /dev/null
+++ b/Thronefall_v1.0/Decompile/UpgradeCastleUp.cs
@@ -0,0 +1,46 @@
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+
+public class UpgradeCastleUp : MonoBehaviour, DayNightCycle.IDaytimeSensitive
+{
+	private List<TagManager.ETag> mustHaveTags = new List<TagManager.ETag>();
+
+	private List<TagManager.ETag> mayNotHaveTags = new List<TagManager.ETag>();
+
+	private List<TaggedObject> allWallsAndTowers = new List<TaggedObject>();
+
+	private void OnEnable()
+	{
+		mustHaveTags.Add(TagManager.ETag.WallOrTower);
+		DayNightCycle.Instance.RegisterDaytimeSensitiveObject(this);
+	}
+
+	public void OnDusk()
+	{
+	}
+
+	public void OnDawn_BeforeSunrise()
+	{
+	}
+
+	public void OnDawn_AfterSunrise()
+	{
+		StartCoroutine(AfterSunriseDelayed());
+	}
+
+	public IEnumerator AfterSunriseDelayed()
+	{
+		yield return null;
+		yield return null;
+		TagManager.instance.FindAllTaggedObjectsWithTags(allWallsAndTowers, mustHaveTags, mayNotHaveTags);
+		for (int i = 0; i < allWallsAndTowers.Count; i++)
+		{
+			BuildSlot componentInParent = allWallsAndTowers[i].GetComponentInParent<BuildSlot>();
+			if (componentInParent.NextUpgradeOrBuildCost > 1)
+			{
+				componentInParent.NextUpgradeOrBuildCost--;
+			}
+		}
+	}
+}