hide secondary beacon effect if level < 4

badasintended · Nov 23, 2023 · def0ee3 · def0ee3
1 parent 7db6c3a
commit def0ee3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/mixin/java/mcp/mobius/waila/mixin/BeaconBlockEntityAccess.java b/src/mixin/java/mcp/mobius/waila/mixin/BeaconBlockEntityAccess.java
@@ -17,4 +17,7 @@ public interface BeaconBlockEntityAccess {
     @Accessor("secondaryPower")
     MobEffect wthit_secondaryPower();
 
+    @Accessor("levels")
+    int wthit_levels();
+
 }
diff --git a/src/pluginVanilla/java/mcp/mobius/waila/plugin/vanilla/provider/BeaconProvider.java b/src/pluginVanilla/java/mcp/mobius/waila/plugin/vanilla/provider/BeaconProvider.java
@@ -50,7 +50,7 @@ public void appendBody(ITooltip tooltip, IBlockAccessor accessor, IPluginConfig
     public void appendData(IDataWriter data, IServerAccessor<BeaconBlockEntity> accessor, IPluginConfig config) {
         if (config.getBoolean(Options.ATTRIBUTE_BEACON_EFFECTS)) data.add(Data.class, res -> {
             var beacon = (BeaconBlockEntity & BeaconBlockEntityAccess) accessor.getTarget();
-            res.add(new Data(beacon.wthit_primaryPower(), beacon.wthit_secondaryPower()));
+            res.add(new Data(beacon.wthit_primaryPower(), beacon.wthit_levels() >= 4 ? beacon.wthit_secondaryPower() : null));
         });
     }