Fix Shop Removal and 0 Hp Battle Start

Author: LifeHckr

Scenes/BattleDirector/Scripts/BattleDirector.cs

@@ -87,6 +87,8 @@ private void BeginPlayback()
             CM.BeginTweens();
             Audio.Play();
         }
+        if (Player.GetCurrentHealth() <= 0 && !_battleEnding)
+            CheckBattleStatus(Player);
         HasPlayed = true;
         _initializedPlaying = true;
     }
@@ -332,10 +334,13 @@ private Timing CheckTiming(double beatDif)
     #endregion
 
     #region Battle End
+    private bool _battleEnding = false;
+
     private void CheckBattleStatus(PuppetTemplate puppet) //Called when a puppet dies
     {
         var tween = CreateTween();
         tween.TweenProperty(puppet, "modulate:a", 0, 2f);
+        _battleEnding = true;
         if (puppet == Player)
         {
             CM.ProcessMode = ProcessModeEnum.Disabled;

Scenes/Puppets/Scripts/PlayerStats.cs

@@ -80,6 +80,7 @@ public void RemoveNote(int index)
         CurNotes = CurNotes.Where((_, i) => i != index).ToArray();
     }
 
+    //Can remove more than 1
     public void RemoveNote(Note nSelection)
     {
         CurNotes = CurNotes.Where(n => n != nSelection).ToArray();

Scenes/ShopScene/Scripts/ShopScene.cs

@@ -332,7 +332,9 @@ private void RemoveNote()
         StageProducer.PlayerStats.Money -= RemovalCost;
         _removalButton.Disabled = true;
         _hasRemoved = true;
-        StageProducer.PlayerStats.RemoveNote(_toRemove);
+        StageProducer.PlayerStats.RemoveNote(
+            Array.IndexOf(StageProducer.PlayerStats.CurNotes, _toRemove)
+        );
         _selectedRemoveButton.QueueFree();
         CloseRemovalPane();
         UpdateMoneyLabel();