Fix dancing bug.

Signed-off-by: 秋雨落 <[email protected]>
CuteNekoOwO · Feb 23, 2024 · c79eb0b · c79eb0b
1 parent 2e84091
commit c79eb0b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/main/java/cuteneko/catsplus/mixins/mixin/dancing/LivingEntityMixin.java b/src/main/java/cuteneko/catsplus/mixins/mixin/dancing/LivingEntityMixin.java
@@ -11,11 +11,16 @@
 
 @Mixin(LivingEntity.class)
 public abstract class LivingEntityMixin {
-    @Inject(method = "setNearbySongPlaying", at = @At("HEAD"))
-    private void beforeSetNearbySongPlaying(BlockPos songPosition, boolean playing, CallbackInfo ci) {
+    @Inject(method = "setNearbySongPlaying", at = @At("TAIL"))
+    private void afterSetNearbySongPlaying(BlockPos songPosition, boolean playing, CallbackInfo ci) {
         if ((Object) this instanceof CatEntity cat) {
             var geniusCat = CatsPlusData.getGeniusCat(cat);
-            geniusCat.songStartPlay(songPosition);
+
+            if (playing) {
+                geniusCat.songStartPlay(songPosition);
+            } else {
+                geniusCat.songStopPlay();
+            }
         }
     }
 }