Refresh registrations on game load

pragasette · Feb 3, 2020 · 7c6a22a · 7c6a22a
1 parent d5208e6
commit 7c6a22a
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,13 @@
 The format is based on [Keep a Changelog], and this project adheres to
 [Semantic Versioning].
 
+## Unreleased
+
+### Fixed
+
+- Health was not reverted after level up in loaded games, this apparently
+affected only Skyrim Special Edition.
+
 ## [1.0.0] - 2019-12-18
 
 ### Added

diff --git a/No Health Reset.esp b/No Health Reset.esp
diff --git a/Scripts/Source/_p7NHR_PlayerAlias.psc b/Scripts/Source/_p7NHR_PlayerAlias.psc
@@ -0,0 +1,7 @@
+Scriptname _p7NHR_PlayerAlias extends ReferenceAlias
+
+_p7NHR_Quest Property PlayerQuest Auto
+
+Event OnPlayerLoadGame()
+	PlayerQuest.RefreshRegistrations()
+EndEvent
diff --git a/Scripts/Source/_p7NHR_Quest.psc b/Scripts/Source/_p7NHR_Quest.psc
@@ -7,10 +7,7 @@ Float controlBaseHealth
 Float sleepStart
 
 Event OnInit()
-	RegisterForSleep()
-
-	; requires SKSE >= 1.06.00
-	RegisterForMenu("LevelUp Menu")
+	RefreshRegistrations()
 EndEvent
 
 Event OnSleepStart(Float afSleepStartTime, Float afDesiredSleepEndTime)
@@ -79,6 +76,15 @@ Event OnMenuClose(String MenuName)
 	EndIf
 EndEvent
 
+Function RefreshRegistrations()
+	UnregisterForSleep()
+	UnregisterForAllMenus()
+
+	RegisterForSleep()
+	; requires SKSE >= 1.6.0
+	RegisterForMenu("LevelUp Menu")
+EndFunction
+
 Function Log(String text)
 	Debug.Trace("[No Health Reset] " + text)
 EndFunction