package com.atsuishio.superbwarfare.perk; import com.atsuishio.superbwarfare.init.ModItems; import com.atsuishio.superbwarfare.init.ModPerks; import com.atsuishio.superbwarfare.item.PerkItem; import net.minecraft.nbt.CompoundTag; import net.minecraft.resources.ResourceLocation; import net.minecraft.util.Mth; import net.minecraft.world.item.Item; import net.minecraft.world.item.ItemStack; import net.minecraftforge.registries.RegistryObject; import javax.annotation.Nullable; import java.util.Optional; import java.util.function.Predicate; public class PerkHelper { private static final String TAG_PERK_ID = "id"; private static final String TAG_PERK_LEVEL = "level"; private static final String TAG_PERK = "Perks"; /** * 把一个Perk封装成nbt进行存储 */ public static CompoundTag makePerk(@Nullable ResourceLocation pId, int pLevel) { CompoundTag compoundtag = new CompoundTag(); compoundtag.putString(TAG_PERK_ID, String.valueOf(pId)); compoundtag.putShort(TAG_PERK_LEVEL, (short) pLevel); return compoundtag; } public static CompoundTag setPerkLevel(CompoundTag pCompound, int pLevel) { pCompound.putShort(TAG_PERK_LEVEL, (short) pLevel); return pCompound; } public static int getPerkLevel(CompoundTag pCompound) { return Mth.clamp(pCompound.getInt(TAG_PERK_LEVEL), 0, 255); } @Nullable public static ResourceLocation getPerkId(CompoundTag pCompoundTag) { return ResourceLocation.tryParse(pCompoundTag.getString(TAG_PERK_ID)); } @Nullable public static ResourceLocation getPerkId(Perk perk) { return switch (perk.type) { case AMMO -> ModPerks.AMMO_PERKS.getEntries().stream() .filter(p -> p.get().descriptionId.equals(perk.descriptionId)) .findFirst() .map(RegistryObject::getId) .orElse(null); case FUNCTIONAL -> ModPerks.FUNC_PERKS.getEntries().stream() .filter(p -> p.get().descriptionId.equals(perk.descriptionId)) .findFirst() .map(RegistryObject::getId) .orElse(null); case DAMAGE -> ModPerks.DAMAGE_PERKS.getEntries().stream() .filter(p -> p.get().descriptionId.equals(perk.descriptionId)) .findFirst() .map(RegistryObject::getId) .orElse(null); }; } public static int getItemPerkLevel(Perk perk, ItemStack stack) { if (stack.isEmpty()) { return 0; } var tag = stack.getTag(); if (tag == null) { return 0; } var tagPerk = tag.getCompound(TAG_PERK); if (!tagPerk.contains(perk.type.getName())) { return 0; } var pt = tagPerk.getCompound(perk.type.getName()); ResourceLocation id = getPerkId(perk); if (id == null) { return 0; } if (!pt.getString(TAG_PERK_ID).equals(makeId(id))) { return 0; } return getPerkLevel(getPerkTag(stack, perk.type)); } public static CompoundTag getPerkTag(ItemStack stack, Perk.Type type) { var tag = stack.getTag(); if (tag == null) return new CompoundTag(); var tagPerk = tag.getCompound(TAG_PERK); if (!tagPerk.contains(type.getName())) return new CompoundTag(); return tagPerk.getCompound(type.getName()); } public static void setPerk(ItemStack stack, Perk perk, int level) { var tag = stack.getTag(); if (tag == null) { tag = new CompoundTag(); } var perkTag = tag.getCompound(TAG_PERK); if (perkTag.isEmpty()) { perkTag = new CompoundTag(); } perkTag.put(perk.type.getName(), makePerk(getPerkId(perk), level)); stack.addTagElement(TAG_PERK, perkTag); } public static void setPerk(ItemStack stack, Perk perk) { setPerk(stack, perk, 1); } @Nullable public static Perk getPerkByType(ItemStack stack, Perk.Type type) { var tag = stack.getTag(); if (tag == null) { return null; } var tagPerk = tag.getCompound(TAG_PERK); if (!tagPerk.contains(type.getName())) { return null; } return switch (type) { case AMMO -> ModPerks.AMMO_PERKS.getEntries().stream() .filter(p -> makeId(p.getId()).equals(tagPerk.getCompound(type.getName()).getString(TAG_PERK_ID))) .findFirst() .map(RegistryObject::get) .orElse(null); case FUNCTIONAL -> ModPerks.FUNC_PERKS.getEntries().stream() .filter(p -> makeId(p.getId()).equals(tagPerk.getCompound(type.getName()).getString(TAG_PERK_ID))) .findFirst() .map(RegistryObject::get) .orElse(null); case DAMAGE -> ModPerks.DAMAGE_PERKS.getEntries().stream() .filter(p -> makeId(p.getId()).equals(tagPerk.getCompound(type.getName()).getString(TAG_PERK_ID))) .findFirst() .map(RegistryObject::get) .orElse(null); }; } public static void removePerkByType(ItemStack stack, Perk.Type type) { var tag = stack.getTag(); if (tag == null) { return; } var tagPerk = tag.getCompound(TAG_PERK); if (!tagPerk.contains(type.getName())) { return; } tagPerk.remove(type.getName()); stack.addTagElement(TAG_PERK, tagPerk); } public static Optional> getPerkItem(Perk perk) { return ModItems.PERKS.getEntries().stream().filter(p -> { if (p.get() instanceof PerkItem perkItem) { return perkItem.getPerk() == perk; } return false; }).findFirst(); } public static String makeId(ResourceLocation resourceLocation) { return resourceLocation.getNamespace() + ":" + resourceLocation.getPath(); } public static final Predicate SHOTGUN_PERKS = perk -> { switch (perk.type) { case AMMO -> { return !perk.descriptionId.equals("butterfly_bullet"); } case FUNCTIONAL -> { return perk == ModPerks.SUBSISTENCE.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.HEAL_CLIP.get() || perk == ModPerks.FIELD_DOCTOR.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.GUTSHOT_STRAIGHT.get() || perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.KILL_CLIP.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate RIFLE_PERKS = perk -> { switch (perk.type) { case AMMO -> { return true; } case FUNCTIONAL -> { return perk == ModPerks.HEAL_CLIP.get() || perk == ModPerks.FIELD_DOCTOR.get() || perk == ModPerks.FOURTH_TIMES_CHARM.get() || perk == ModPerks.SUBSISTENCE.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.KILL_CLIP.get() || perk == ModPerks.GUTSHOT_STRAIGHT.get() || perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.HEAD_SEEKER.get() || perk == ModPerks.DESPERADO.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate HANDGUN_PERKS = perk -> { switch (perk.type) { case AMMO -> { return !perk.descriptionId.equals("butterfly_bullet"); } case FUNCTIONAL -> { return perk == ModPerks.HEAL_CLIP.get() || perk == ModPerks.FIELD_DOCTOR.get() || perk == ModPerks.SUBSISTENCE.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.KILL_CLIP.get() || perk == ModPerks.GUTSHOT_STRAIGHT.get() || perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate SNIPER_RIFLE_PERKS = perk -> { switch (perk.type) { case AMMO -> { return !perk.descriptionId.equals("butterfly_bullet"); } case FUNCTIONAL -> { return perk == ModPerks.HEAL_CLIP.get() || perk == ModPerks.FOURTH_TIMES_CHARM.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.KILL_CLIP.get() || perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate SMG_PERKS = perk -> { switch (perk.type) { case AMMO -> { return true; } case FUNCTIONAL -> { return perk == ModPerks.HEAL_CLIP.get() || perk == ModPerks.FOURTH_TIMES_CHARM.get() || perk == ModPerks.SUBSISTENCE.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.KILL_CLIP.get() || perk == ModPerks.GUTSHOT_STRAIGHT.get() || perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.HEAD_SEEKER.get() || perk == ModPerks.DESPERADO.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate MACHINE_GUN_PERKS = perk -> { switch (perk.type) { case AMMO -> { return true; } case FUNCTIONAL -> { return perk == ModPerks.FOURTH_TIMES_CHARM.get() || perk == ModPerks.SUBSISTENCE.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get(); } case DAMAGE -> { return perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.KILLING_TALLY.get() || perk == ModPerks.VORPAL_WEAPON.get(); } default -> { return false; } } }; public static final Predicate MAGAZINE_PERKS = perk -> false; public static final Predicate LAUNCHER_PERKS = perk -> perk == ModPerks.MONSTER_HUNTER.get() || perk == ModPerks.POWERFUL_ATTRACTION.get() || perk == ModPerks.INTELLIGENT_CHIP.get() || perk == ModPerks.VORPAL_WEAPON.get(); }