mcMMO/src/main/java/com/gmail/nossr50/util/AttributeMapper.java

197 lines
9.9 KiB
Java

package com.gmail.nossr50.util;
import com.gmail.nossr50.mcMMO;
import org.bukkit.attribute.Attribute;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Optional;
import java.util.stream.Stream;
public class AttributeMapper {
public static final String ATTRIBUTE = "ATTRIBUTE";
public static final String ORG_BUKKIT_REGISTRY = "org.bukkit.Registry";
// Prevent instantiation
private AttributeMapper() {
}
// Define constants for attribute keys and their legacy counterparts
private static final String MAX_HEALTH_1_21_3_STR = "max_health";
private static final String MAX_HEALTH_1_18_2_STR = "generic.max_health";
public static final Attribute MAPPED_MAX_HEALTH;
private static final String JUMP_STRENGTH_1_23_1 = "jump_strength";
private static final String JUMP_STRENGTH_1_21_1 = "generic.jump_strength";
private static final String JUMP_STR_1_18_2 = "horse.jump_strength";
public static final Attribute MAPPED_JUMP_STRENGTH;
public static final Attribute MAPPED_MOVEMENT_SPEED;
private static final String MOVEMENT_SPEED_1_18_2 = "generic.movement_speed";
private static final String MOVEMENT_SPEED_1_21_1 = "generic.movement_speed";
private static final String MOVEMENT_SPEED_1_21_3 = "movement_speed";
// Add other attributes similarly...
// For brevity, only key attributes are shown
static {
MAPPED_MAX_HEALTH = findAttribute(MAX_HEALTH_1_21_3_STR, MAX_HEALTH_1_18_2_STR);
MAPPED_JUMP_STRENGTH = findAttribute(JUMP_STRENGTH_1_23_1, JUMP_STRENGTH_1_21_1, JUMP_STR_1_18_2);
MAPPED_MOVEMENT_SPEED = findAttribute(MOVEMENT_SPEED_1_18_2, MOVEMENT_SPEED_1_21_1, MOVEMENT_SPEED_1_21_3);
}
private static Attribute findAttribute(String... keys) {
Stream<?> attributeStream;
try {
// Try to get Registry.ATTRIBUTE using reflection
Class<?> registryClass = Class.forName(ORG_BUKKIT_REGISTRY);
Field attributeField = registryClass.getField(ATTRIBUTE);
Object attributeRegistry = attributeField.get(null);
// Get the stream() method of the attribute registry
Method streamMethod = attributeRegistry.getClass().getMethod("stream");
attributeStream = (Stream<?>) streamMethod.invoke(attributeRegistry);
} catch (ClassNotFoundException | NoSuchFieldException | IllegalAccessException | NoSuchMethodException |
InvocationTargetException e) {
// Fallback to older versions where Attribute is an enum
Object[] enumConstants = Attribute.class.getEnumConstants();
attributeStream = Arrays.stream(enumConstants);
}
Optional<?> optionalAttribute = attributeStream
.filter(attr -> {
try {
String attrKey = null;
String attrName = null;
// Try to get attr.getKey().getKey()
Method getKeyMethod = attr.getClass().getMethod("getKey");
Object namespacedKey = getKeyMethod.invoke(attr);
if (namespacedKey != null) {
Method getKeyStringMethod = namespacedKey.getClass().getMethod("getKey");
attrKey = (String) getKeyStringMethod.invoke(namespacedKey);
}
// Try to get attr.name()
Method nameMethod;
try {
nameMethod = attr.getClass().getMethod("name");
attrName = (String) nameMethod.invoke(attr);
} catch (NoSuchMethodException e) {
// name() method doesn't exist in newer versions
attrName = null;
}
// Compare with provided keys
for (String key : keys) {
if ((attrKey != null && attrKey.equalsIgnoreCase(key)) ||
(attrName != null && attrName.equalsIgnoreCase(key))) {
return true;
}
}
} catch (Exception e) {
mcMMO.p.getLogger().severe("Unable to find the attribute with possible keys: "
+ Arrays.toString(keys) + ", mcMMO will not function properly.");
throw new RuntimeException(e);
}
return false;
})
.findFirst();
if (optionalAttribute.isPresent()) {
return (Attribute) optionalAttribute.get();
} else {
mcMMO.p.getLogger().severe("Unable to find the attribute with possible keys: "
+ Arrays.toString(keys) + ", mcMMO will not function properly.");
throw new IllegalStateException("Unable to find the attribute with possible keys: "
+ Arrays.toString(keys));
}
}
/*
For easy reference...
List of 1.18 Attributes by name...
GENERIC_MAX_HEALTH("generic.max_health"),
GENERIC_FOLLOW_RANGE("generic.follow_range"),
GENERIC_KNOCKBACK_RESISTANCE("generic.knockback_resistance"),
GENERIC_MOVEMENT_SPEED("generic.movement_speed"),
GENERIC_FLYING_SPEED("generic.flying_speed"),
GENERIC_ATTACK_DAMAGE("generic.attack_damage"),
GENERIC_ATTACK_KNOCKBACK("generic.attack_knockback"),
GENERIC_ATTACK_SPEED("generic.attack_speed"),
GENERIC_ARMOR("generic.armor"),
GENERIC_ARMOR_TOUGHNESS("generic.armor_toughness"),
GENERIC_LUCK("generic.luck"),
HORSE_JUMP_STRENGTH("horse.jump_strength"),
ZOMBIE_SPAWN_REINFORCEMENTS("zombie.spawn_reinforcements");
List of 1.21.1 Attributes by name...
GENERIC_MAX_HEALTH("generic.max_health"),
GENERIC_FOLLOW_RANGE("generic.follow_range"),
GENERIC_KNOCKBACK_RESISTANCE("generic.knockback_resistance"),
GENERIC_MOVEMENT_SPEED("generic.movement_speed"),
GENERIC_FLYING_SPEED("generic.flying_speed"),
GENERIC_ATTACK_DAMAGE("generic.attack_damage"),
GENERIC_ATTACK_KNOCKBACK("generic.attack_knockback"),
GENERIC_ATTACK_SPEED("generic.attack_speed"),
GENERIC_ARMOR("generic.armor"),
GENERIC_ARMOR_TOUGHNESS("generic.armor_toughness"),
GENERIC_FALL_DAMAGE_MULTIPLIER("generic.fall_damage_multiplier"),
GENERIC_LUCK("generic.luck"),
GENERIC_MAX_ABSORPTION("generic.max_absorption"),
GENERIC_SAFE_FALL_DISTANCE("generic.safe_fall_distance"),
GENERIC_SCALE("generic.scale"),
GENERIC_STEP_HEIGHT("generic.step_height"),
GENERIC_GRAVITY("generic.gravity"),
GENERIC_JUMP_STRENGTH("generic.jump_strength"),
GENERIC_EXPLOSION_KNOCKBACK_RESISTANCE("generic.explosion_knockback_resistance"),
GENERIC_MOVEMENT_EFFICIENCY("generic.movement_efficiency"),
GENERIC_OXYGEN_BONUS("generic.oxygen_bonus"),
GENERIC_WATER_MOVEMENT_EFFICIENCY("generic.water_movement_efficiency"),
PLAYER_BLOCK_INTERACTION_RANGE("player.block_interaction_range"),
PLAYER_ENTITY_INTERACTION_RANGE("player.entity_interaction_range"),
PLAYER_BLOCK_BREAK_SPEED("player.block_break_speed"),
PLAYER_MINING_EFFICIENCY("player.mining_efficiency"),
PLAYER_SNEAKING_SPEED("player.sneaking_speed"),
PLAYER_SUBMERGED_MINING_SPEED("player.submerged_mining_speed"),
PLAYER_SWEEPING_DAMAGE_RATIO("player.sweeping_damage_ratio"),
ZOMBIE_SPAWN_REINFORCEMENTS("zombie.spawn_reinforcements");
List of 1.21.3 Attributes...
Attribute MAX_HEALTH = getAttribute("max_health");
Attribute FOLLOW_RANGE = getAttribute("follow_range");
Attribute KNOCKBACK_RESISTANCE = getAttribute("knockback_resistance");
Attribute MOVEMENT_SPEED = getAttribute("movement_speed");
Attribute FLYING_SPEED = getAttribute("flying_speed");
Attribute ATTACK_DAMAGE = getAttribute("attack_damage");
Attribute ATTACK_KNOCKBACK = getAttribute("attack_knockback");
Attribute ATTACK_SPEED = getAttribute("attack_speed");
Attribute ARMOR = getAttribute("armor");
Attribute ARMOR_TOUGHNESS = getAttribute("armor_toughness");
Attribute FALL_DAMAGE_MULTIPLIER = getAttribute("fall_damage_multiplier");
Attribute LUCK = getAttribute("luck");
Attribute MAX_ABSORPTION = getAttribute("max_absorption");
Attribute SAFE_FALL_DISTANCE = getAttribute("safe_fall_distance");
Attribute SCALE = getAttribute("scale");
Attribute STEP_HEIGHT = getAttribute("step_height");
Attribute GRAVITY = getAttribute("gravity");
Attribute JUMP_STRENGTH = getAttribute("jump_strength");
Attribute BURNING_TIME = getAttribute("burning_time");
Attribute EXPLOSION_KNOCKBACK_RESISTANCE = getAttribute("explosion_knockback_resistance");
Attribute MOVEMENT_EFFICIENCY = getAttribute("movement_efficiency");
Attribute OXYGEN_BONUS = getAttribute("oxygen_bonus");
Attribute WATER_MOVEMENT_EFFICIENCY = getAttribute("water_movement_efficiency");
Attribute TEMPT_RANGE = getAttribute("tempt_range");
Attribute BLOCK_INTERACTION_RANGE = getAttribute("block_interaction_range");
Attribute ENTITY_INTERACTION_RANGE = getAttribute("entity_interaction_range");
Attribute BLOCK_BREAK_SPEED = getAttribute("block_break_speed");
Attribute MINING_EFFICIENCY = getAttribute("mining_efficiency");
Attribute SNEAKING_SPEED = getAttribute("sneaking_speed");
Attribute SUBMERGED_MINING_SPEED = getAttribute("submerged_mining_speed");
Attribute SWEEPING_DAMAGE_RATIO = getAttribute("sweeping_damage_ratio");
Attribute SPAWN_REINFORCEMENTS = getAttribute("spawn_reinforcements");
*/
}