Preserve constructor arguments for scripted states after hot reloading

Author: nykwono

hmm.json

@@ -205,7 +205,7 @@
     {
       "name": "polymod",
       "type": "git",
-      "ref": "d426bdfdafd5e225b81956b217db5fe3a86a90b2",
+      "ref": "af4d8ab44a76551a9fcc20f26ee191699ef8c7ff",
       "url": "https://github.com/FunkinCrew/polymod"
     },
     {

source/funkin/ui/MusicBeatState.hx

@@ -276,7 +276,9 @@ class MusicBeatState extends FlxTransitionableState implements IEventHandler
     var scriptedNextState:NextState = () ->
     {
       var path:String = _asc?.fullyQualifiedName ?? '';
-      var newState:Null<MusicBeatState> = MusicBeatState.scriptInit(path);
+      var constructorArgs:Array<Dynamic> = _asc?.getConstructorArgs() ?? [];
+
+      var newState:Null<MusicBeatState> = MusicBeatState.scriptInit(path, ...constructorArgs);
       if (newState == null) return new funkin.ui.mainmenu.MainMenuState();
       return newState;
     }

source/funkin/ui/MusicBeatSubState.hx

@@ -203,7 +203,9 @@ class MusicBeatSubState extends FlxSubState implements IEventHandler
     var scriptedNextState:NextState = () ->
     {
       var path:String = _asc?.fullyQualifiedName ?? '';
-      var newState:Null<MusicBeatSubState> = MusicBeatSubState.scriptInit(path);
+      var constructorArgs:Array<Dynamic> = _asc?.getConstructorArgs() ?? [];
+
+      var newState:Null<MusicBeatSubState> = MusicBeatSubState.scriptInit(path, ...constructorArgs);
       if (newState == null) return new funkin.ui.mainmenu.MainMenuState();
       return newState;
     }

source/funkin/util/plugins/ReloadAssetsDebugPlugin.hx

@@ -56,6 +56,8 @@ class ReloadAssetsDebugPlugin extends FlxBasic
     {
       @:privateAccess
       var path:String = state._asc?.fullyQualifiedName ?? '';
+      var constructorArgs:Array<Dynamic> = state._asc?.getConstructorArgs() ?? [];
+
       trace('Hot-reloading scripted state: ' + path);
 
       if (Std.isOfType(state, MusicBeatState) || Std.isOfType(state, MusicBeatSubState))
@@ -73,11 +75,11 @@ class ReloadAssetsDebugPlugin extends FlxBasic
         {
           if (Std.isOfType(state, FlxState))
           {
-            newState = FlxState.scriptInit(path);
+            newState = FlxState.scriptInit(path, ...constructorArgs);
           }
           else if (Std.isOfType(state, FlxSubState))
           {
-            newState = FlxSubState.scriptInit(path);
+            newState = FlxSubState.scriptInit(path, ...constructorArgs);
           }
           if (newState == null) return new funkin.ui.mainmenu.MainMenuState();
           return newState;