fix: SpriteButtonComponent to initialize sprites in onLoad (#3302)

flame-engine · Oct 15, 2024 · b5eefc4 · b5eefc4
1 parent ce97eb4
commit b5eefc4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/packages/flame/lib/src/components/input/sprite_button_component.dart b/packages/flame/lib/src/components/input/sprite_button_component.dart
@@ -56,16 +56,21 @@ class SpriteButtonComponent extends SpriteGroupComponent<ButtonState>
   }
 
   @override
-  void onMount() {
-    assert(
-      _button != null,
-      'The button sprite has to be set either in onLoad or in the constructor',
-    );
+  void onLoad() {
+    super.onLoad();
     sprites = {
       ButtonState.up: button,
       ButtonState.down: buttonDown,
     };
+  }
+
+  @override
+  void onMount() {
     super.onMount();
+    assert(
+      _button != null,
+      'The button sprite has to be set either in onLoad or in the constructor',
+    );
   }
 
   @override

diff --git a/packages/flame/lib/src/components/sprite_group_component.dart b/packages/flame/lib/src/components/sprite_group_component.dart
@@ -100,7 +100,8 @@ class SpriteGroupComponent<T> extends PositionComponent
 
   /// Updates the sprite for the given key.
   void updateSprite(T key, Sprite sprite) {
-    _sprites![key] = sprite;
+    assert(_sprites != null, 'Sprites map is not yet initialized');
+    _sprites?[key] = sprite;
     _resizeToSprite();
   }