Implement support for iron's spellbooks

2025-11-08 22:27:38 +01:00
parent 73a4f2549f
commit f34f644d33
6 changed files with 512 additions and 114 deletions
@@ -0,0 +1,70 @@
+package net.fretux.ascend.event;
+
+import net.fretux.ascend.AscendMod;
+import net.fretux.ascend.player.PlayerStatsProvider;
+import net.minecraft.world.damagesource.DamageSource;
+import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.entity.ai.attributes.Attributes;
+import net.minecraftforge.event.entity.living.LivingHurtEvent;
+import net.minecraftforge.eventbus.api.SubscribeEvent;
+import net.minecraftforge.fml.common.Mod;
+
+@Mod.EventBusSubscriber(modid = AscendMod.MODID)
+public class WeaponMagicScalingEvents {
+
+    @SubscribeEvent
+    public static void onLivingHurt(LivingHurtEvent event) {
+        if (!(event.getSource().getEntity() instanceof Player player)) return;
+        if (event.getEntity() == player) return;
+        if (player.level().isClientSide) return;
+        player.getCapability(PlayerStatsProvider.PLAYER_STATS).ifPresent(stats -> {
+            float base = event.getAmount();
+            float modified = base;
+            if (isMagicDamage(event.getSource())) {
+                int magicScaling = stats.getAttributeLevel("magic_scaling");
+                if (magicScaling > 0) {
+                    double perPoint = 0.005d;
+                    double maxBonus = 0.40d;
+                    double bonusMult = 1.0d + Math.min(magicScaling * perPoint, maxBonus);
+                    modified = (float) (modified * bonusMult);
+                }
+            } else {
+                double attackSpeed = 4.0d;
+                if (player.getAttribute(Attributes.ATTACK_SPEED) != null) {
+                    attackSpeed = player.getAttribute(Attributes.ATTACK_SPEED).getValue();
+                }
+
+                String key =
+                        (attackSpeed >= 1.8d) ? "light_scaling" :
+                                (attackSpeed >= 1.2d) ? "medium_scaling" :
+                                        "heavy_scaling";
+
+                int level = stats.getAttributeLevel(key);
+                if (level > 0) {
+                    double perPoint =
+                            key.equals("light_scaling") ? 0.003d :
+                                    key.equals("medium_scaling") ? 0.004d :
+                                            0.005d;
+                    double maxBonus = 0.40d;
+                    double bonusMult = 1.0d + Math.min(level * perPoint, maxBonus);
+                    modified = (float) (modified * bonusMult);
+                }
+            }
+
+            if (modified != base) {
+                event.setAmount(modified);
+            }
+        });
+    }
+
+    private static boolean isMagicDamage(DamageSource source) {
+        String id = source.getMsgId();
+        if (id == null) return false;
+        return id.equals("magic")
+                || id.equals("indirectMagic")
+                || id.equals("wither")
+                || id.equals("dragonBreath")
+                || id.equals("thrown")     
+                || id.equals("potion");
+    }
+}