From 6103c5387921c6076d1487c62e2d39b096a4942c Mon Sep 17 00:00:00 2001
From: CuckyDev <regan_green27@hotmail.com>
Date: Mon, 2 Aug 2021 22:16:53 -0400
Subject: [PATCH] Fix chart load/save difficulty

---
 source/ChartingState.hx | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/source/ChartingState.hx b/source/ChartingState.hx
index 6357545..2e8fbab 100644
--- a/source/ChartingState.hx
+++ b/source/ChartingState.hx
@@ -2584,12 +2584,13 @@ class ChartingState extends MusicBeatState
 
 	function loadJson(song:String):Void
 	{
+		var difficultyArray:Array<String> = ['-easy', "", "-hard"];
 		var format = StringTools.replace(PlayState.SONG.song.toLowerCase(), " ", "-");
 		switch (format) {
 			case 'Dad-Battle': format = 'Dadbattle';
 			case 'Philly-Nice': format = 'Philly';
 		}
-		PlayState.SONG = Song.loadFromJson(format, format);
+		PlayState.SONG = Song.loadFromJson(format + difficultyArray[PlayState.storyDifficulty], format);
 		LoadingState.loadAndSwitchState(new ChartingState());
 	}
 
@@ -2609,6 +2610,7 @@ class ChartingState extends MusicBeatState
 
 	private function saveLevel()
 	{
+		var difficultyArray:Array<String> = ['-easy', "", "-hard"];
 		var json = {
 			"song": _song
 		};
@@ -2621,7 +2623,7 @@ class ChartingState extends MusicBeatState
 			_file.addEventListener(Event.COMPLETE, onSaveComplete);
 			_file.addEventListener(Event.CANCEL, onSaveCancel);
 			_file.addEventListener(IOErrorEvent.IO_ERROR, onSaveError);
-			_file.save(data.trim(), _song.song.toLowerCase() + ".json");
+			_file.save(data.trim(), _song.song.toLowerCase() + difficultyArray[PlayState.storyDifficulty] + ".json");
 		}
 	}