From ef3e0e135e30e4f59b4cba17b31f9e944aa46e38 Mon Sep 17 00:00:00 2001
From: Lucky56 <55949451+Lucky-56@users.noreply.github.com>
Date: Fri, 18 Jun 2021 21:44:05 +0200
Subject: [PATCH] renamed 'lock weeks' to 'reset story mode progress' + added
 reset score

---
 source/Options.hx     | 53 ++++++++++++++++++++++++++++++++++++++++---
 source/OptionsMenu.hx |  4 ++--
 2 files changed, 52 insertions(+), 5 deletions(-)

diff --git a/source/Options.hx b/source/Options.hx
index a3210c6..cc82e65 100644
--- a/source/Options.hx
+++ b/source/Options.hx
@@ -637,6 +637,8 @@ class CamZoomOption extends Option
 
 class LockWeeksOption extends Option
 {
+	var confirm:Bool = false;
+
 	public function new(desc:String)
 	{
 		super();
@@ -644,8 +646,14 @@ class LockWeeksOption extends Option
 	}
 	public override function press():Bool
 	{
-		FlxG.save.data.weekUnlocked = 0;
-		StoryMenuState.weekUnlocked = [true];
+		if(!confirm)
+		{
+			confirm = true;
+			display = updateDisplay();
+			return true;
+		}
+		FlxG.save.data.weekUnlocked = 1;
+		StoryMenuState.weekUnlocked = [true, true];
 		trace('Weeks Locked');
 		display = updateDisplay();
 		return true;
@@ -653,6 +661,45 @@ class LockWeeksOption extends Option
 
 	private override function updateDisplay():String
 	{
-		return "Lock Weeks";
+		return confirm ? "Confirm" : "Reset Story Progression";
+	}
+}
+
+class ResetScoreOption extends Option
+{
+	var confirm:Bool = false;
+
+	public function new(desc:String)
+	{
+		super();
+		description = desc;
+	}
+	public override function press():Bool
+	{
+		if(!confirm)
+		{
+			confirm = true;
+			display = updateDisplay();
+			return true;
+		}
+		FlxG.save.data.songScores = null;
+		for(key in Highscore.songScores.keys())
+		{
+			Highscore.songScores[key] = 0;
+		}
+		FlxG.save.data.songCombos = null;
+		for(key in Highscore.songCombos.keys())
+		{
+			Highscore.songCombos[key] = '';
+		}	
+		confirm = false;
+		trace('Save Data Wiped');
+		display = updateDisplay();
+		return true;
+	}
+
+	private override function updateDisplay():String
+	{
+		return confirm ? "Confirm" : "Reset Score";
 	}
 }
diff --git a/source/OptionsMenu.hx b/source/OptionsMenu.hx
index 71a82ee..c2a4446 100644
--- a/source/OptionsMenu.hx
+++ b/source/OptionsMenu.hx
@@ -61,8 +61,8 @@ class OptionsMenu extends MusicBeatState
 		]),
 		
 		new OptionCategory("Saved Data", [
-			new LockWeeksOption("Lock all weeks. (exept Tutorial of course)")
-			// TODO reset score option
+			new ResetScoreOption("Reset your score on all songs and weeks."),
+			new LockWeeksOption("Reset your storymode progress. (only Tutorial + Week 1 is unlocked)")
 		])
 		
 	];