TreasureChestState.java

package com.hypixel.hytale.builtin.adventure.objectives.blockstates;

import com.hypixel.hytale.builtin.adventure.objectives.Objective;
import com.hypixel.hytale.builtin.adventure.objectives.ObjectivePlugin;
import com.hypixel.hytale.builtin.adventure.objectives.events.TreasureChestOpeningEvent;
import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.event.IEventDispatcher;
import com.hypixel.hytale.server.core.HytaleServer;
import com.hypixel.hytale.server.core.entity.UUIDComponent;
import com.hypixel.hytale.server.core.inventory.ItemStack;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.meta.BlockState;
import com.hypixel.hytale.server.core.universe.world.meta.state.BreakValidatedBlockState;
import com.hypixel.hytale.server.core.universe.world.meta.state.ItemContainerState;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import java.util.List;
import java.util.UUID;
import javax.annotation.Nonnull;

public class TreasureChestState extends ItemContainerState implements BreakValidatedBlockState {
   public static final BuilderCodec<TreasureChestState> CODEC;
   protected UUID objectiveUUID;
   protected UUID chestUUID;
   protected boolean opened;

   public TreasureChestState() {
   }

   public boolean canOpen(@Nonnull Ref<EntityStore> ref, @Nonnull ComponentAccessor<EntityStore> componentAccessor) {
      if (!this.opened) {
         UUIDComponent uuidComponent = (UUIDComponent)componentAccessor.getComponent(ref, UUIDComponent.getComponentType());

         assert uuidComponent != null;

         Objective objective = ObjectivePlugin.get().getObjectiveDataStore().getObjective(this.objectiveUUID);
         return objective != null && objective.getActivePlayerUUIDs().contains(uuidComponent.getUuid());
      } else {
         return true;
      }
   }

   public boolean canDestroy(@Nonnull Ref<EntityStore> playerRef, @Nonnull ComponentAccessor<EntityStore> componentAccessor) {
      return this.opened;
   }

   public void onOpen(@Nonnull Ref<EntityStore> ref, @Nonnull World world, @Nonnull Store<EntityStore> store) {
      IEventDispatcher<TreasureChestOpeningEvent, TreasureChestOpeningEvent> dispatcher = HytaleServer.get().getEventBus().dispatchFor(TreasureChestOpeningEvent.class, world.getName());
      if (dispatcher.hasListener()) {
         dispatcher.dispatch(new TreasureChestOpeningEvent(this.objectiveUUID, this.chestUUID, ref, store));
      }

      this.setOpened(true);
   }

   public void setOpened(boolean opened) {
      this.opened = opened;
   }

   public void setObjectiveData(UUID objectiveUUID, UUID chestUUID, List<ItemStack> itemStacks) {
      this.objectiveUUID = objectiveUUID;
      this.chestUUID = chestUUID;
      this.itemContainer.addItemStacks(itemStacks);
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(TreasureChestState.class, TreasureChestState::new, BlockState.BASE_CODEC).append(new KeyedCodec("ObjectiveUUID", Codec.UUID_BINARY), (treasureChestState, uuid) -> treasureChestState.objectiveUUID = uuid, (treasureChestState) -> treasureChestState.objectiveUUID).add()).append(new KeyedCodec("ChestUUID", Codec.UUID_BINARY), (treasureChestState, uuid) -> treasureChestState.chestUUID = uuid, (treasureChestState) -> treasureChestState.chestUUID).add()).append(new KeyedCodec("Opened", Codec.BOOLEAN), (treasureChestState, aBoolean) -> treasureChestState.opened = aBoolean, (treasureChestState) -> treasureChestState.opened).add()).build();
   }
}