DeathComponent.java

package com.hypixel.hytale.server.core.modules.entity.damage;

import com.hypixel.hytale.assetstore.AssetRegistry;
import com.hypixel.hytale.assetstore.map.IndexedLookupTableAssetMap;
import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.codecs.array.ArrayCodec;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Component;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.server.core.asset.type.gameplay.DeathConfig;
import com.hypixel.hytale.server.core.entity.InteractionChain;
import com.hypixel.hytale.server.core.inventory.ItemStack;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import java.util.List;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class DeathComponent implements Component<EntityStore> {
   public static final BuilderCodec<DeathComponent> CODEC;
   private String deathCause;
   @Nullable
   private Message deathMessage;
   private boolean showDeathMenu = true;
   private ItemStack[] itemsLostOnDeath;
   private double itemsAmountLossPercentage;
   private double itemsDurabilityLossPercentage;
   private boolean displayDataOnDeathScreen;
   @Nullable
   private Damage deathInfo;
   private DeathConfig.ItemsLossMode itemsLossMode;
   @Nullable
   private InteractionChain interactionChain;

   public static ComponentType<EntityStore, DeathComponent> getComponentType() {
      return DamageModule.get().getDeathComponentType();
   }

   protected DeathComponent(@Nonnull Damage deathInfo) {
      this.itemsLossMode = DeathConfig.ItemsLossMode.ALL;
      this.deathInfo = deathInfo;
      this.deathCause = deathInfo.getCause().getId();
   }

   protected DeathComponent() {
      this.itemsLossMode = DeathConfig.ItemsLossMode.ALL;
   }

   @Nullable
   public DamageCause getDeathCause() {
      return (DamageCause)((IndexedLookupTableAssetMap)AssetRegistry.getAssetStore(DamageCause.class).getAssetMap()).getAsset(this.deathCause);
   }

   @Nullable
   public Message getDeathMessage() {
      return this.deathMessage;
   }

   public void setDeathMessage(@Nullable Message deathMessage) {
      this.deathMessage = deathMessage;
   }

   public boolean isShowDeathMenu() {
      return this.showDeathMenu;
   }

   public void setShowDeathMenu(boolean showDeathMenu) {
      this.showDeathMenu = showDeathMenu;
   }

   public ItemStack[] getItemsLostOnDeath() {
      return this.itemsLostOnDeath;
   }

   public void setItemsLostOnDeath(List<ItemStack> itemsLostOnDeath) {
      this.itemsLostOnDeath = (ItemStack[])itemsLostOnDeath.toArray((x$0) -> new ItemStack[x$0]);
   }

   public double getItemsAmountLossPercentage() {
      return this.itemsAmountLossPercentage;
   }

   public void setItemsAmountLossPercentage(double itemsAmountLossPercentage) {
      this.itemsAmountLossPercentage = itemsAmountLossPercentage;
   }

   public double getItemsDurabilityLossPercentage() {
      return this.itemsDurabilityLossPercentage;
   }

   public void setItemsDurabilityLossPercentage(double itemsDurabilityLossPercentage) {
      this.itemsDurabilityLossPercentage = itemsDurabilityLossPercentage;
   }

   public boolean displayDataOnDeathScreen() {
      return this.displayDataOnDeathScreen;
   }

   public void setDisplayDataOnDeathScreen(boolean displayDataOnDeathScreen) {
      this.displayDataOnDeathScreen = displayDataOnDeathScreen;
   }

   @Nullable
   public Damage getDeathInfo() {
      return this.deathInfo;
   }

   public DeathConfig.ItemsLossMode getItemsLossMode() {
      return this.itemsLossMode;
   }

   public void setItemsLossMode(DeathConfig.ItemsLossMode itemsLossMode) {
      this.itemsLossMode = itemsLossMode;
   }

   public DeathItemLoss getDeathItemLoss() {
      return new DeathItemLoss(this.itemsLossMode, this.itemsLostOnDeath, this.itemsAmountLossPercentage, this.itemsDurabilityLossPercentage);
   }

   @Nullable
   public InteractionChain getInteractionChain() {
      return this.interactionChain;
   }

   public void setInteractionChain(@Nullable InteractionChain interactionChain) {
      this.interactionChain = interactionChain;
   }

   @Nonnull
   public Component<EntityStore> clone() {
      DeathComponent death = new DeathComponent();
      death.deathCause = this.deathCause;
      death.deathMessage = this.deathMessage;
      death.showDeathMenu = this.showDeathMenu;
      death.itemsLostOnDeath = this.itemsLostOnDeath;
      death.itemsAmountLossPercentage = this.itemsAmountLossPercentage;
      death.itemsDurabilityLossPercentage = this.itemsDurabilityLossPercentage;
      death.displayDataOnDeathScreen = this.displayDataOnDeathScreen;
      death.deathInfo = this.deathInfo;
      death.itemsLossMode = this.itemsLossMode;
      return death;
   }

   public static void tryAddComponent(@Nonnull CommandBuffer<EntityStore> commandBuffer, @Nonnull Ref<EntityStore> ref, @Nonnull Damage damage) {
      if (!commandBuffer.getArchetype(ref).contains(getComponentType())) {
         commandBuffer.run((store) -> tryAddComponent(store, ref, damage));
      }
   }

   public static void tryAddComponent(@Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> ref, @Nonnull Damage damage) {
      if (!store.getArchetype(ref).contains(getComponentType())) {
         store.addComponent(ref, getComponentType(), new DeathComponent(damage));
      }
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(DeathComponent.class, DeathComponent::new).append(new KeyedCodec("DeathCause", Codec.STRING), (o, i) -> o.deathCause = i, (o) -> o.deathCause).add()).append(new KeyedCodec("DeathMessage", Message.CODEC), (deathComponent, message) -> deathComponent.deathMessage = message, (deathComponent) -> deathComponent.deathMessage).add()).append(new KeyedCodec("ShowDeathMenu", BuilderCodec.BOOLEAN), (deathComponent, showDeathMenu) -> deathComponent.showDeathMenu = showDeathMenu, (deathComponent) -> deathComponent.showDeathMenu).add()).append(new KeyedCodec("ItemsLostOnDeath", new ArrayCodec(ItemStack.CODEC, (x$0) -> new ItemStack[x$0])), (deathComponent, itemStacks) -> deathComponent.itemsLostOnDeath = itemStacks, (deathComponent) -> deathComponent.itemsLostOnDeath).add()).append(new KeyedCodec("ItemsAmountLossPercentage", Codec.DOUBLE), (deathComponent, aDouble) -> deathComponent.itemsAmountLossPercentage = aDouble, (deathComponent) -> deathComponent.itemsAmountLossPercentage).add()).append(new KeyedCodec("ItemsDurabilityLossPercentage", Codec.DOUBLE), (deathComponent, aDouble) -> deathComponent.itemsDurabilityLossPercentage = aDouble, (deathComponent) -> deathComponent.itemsDurabilityLossPercentage).add()).append(new KeyedCodec("DisplayDataOnDeathScreen", Codec.BOOLEAN), (deathComponent, aBoolean) -> deathComponent.displayDataOnDeathScreen = aBoolean, (deathComponent) -> deathComponent.displayDataOnDeathScreen).add()).build();
   }
}