TrackedPlacement.java

package com.hypixel.hytale.server.core.modules.interaction.blocktrack;

import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.component.AddReason;
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.RemoveReason;
import com.hypixel.hytale.component.ResourceType;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.system.RefSystem;
import com.hypixel.hytale.math.util.ChunkUtil;
import com.hypixel.hytale.server.core.asset.type.blocktype.config.BlockType;
import com.hypixel.hytale.server.core.modules.block.BlockModule;
import com.hypixel.hytale.server.core.modules.interaction.InteractionModule;
import com.hypixel.hytale.server.core.universe.world.chunk.BlockChunk;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class TrackedPlacement implements Component<ChunkStore> {
   public static final BuilderCodec<TrackedPlacement> CODEC;
   private String blockName;

   public static ComponentType<ChunkStore, TrackedPlacement> getComponentType() {
      return InteractionModule.get().getTrackedPlacementComponentType();
   }

   public TrackedPlacement() {
   }

   public TrackedPlacement(String blockName) {
      this.blockName = blockName;
   }

   @Nullable
   public Component<ChunkStore> clone() {
      return new TrackedPlacement(this.blockName);
   }

   static {
      CODEC = ((BuilderCodec.Builder)BuilderCodec.builder(TrackedPlacement.class, TrackedPlacement::new).append(new KeyedCodec("BlockName", Codec.STRING), (o, v) -> o.blockName = v, (o) -> o.blockName).add()).build();
   }

   public static class OnAddRemove extends RefSystem<ChunkStore> {
      private static final ComponentType<ChunkStore, TrackedPlacement> COMPONENT_TYPE = TrackedPlacement.getComponentType();
      private static final ResourceType<ChunkStore, BlockCounter> BLOCK_COUNTER_RESOURCE_TYPE = BlockCounter.getResourceType();

      public OnAddRemove() {
      }

      public void onEntityAdded(@Nonnull Ref<ChunkStore> ref, @Nonnull AddReason reason, @Nonnull Store<ChunkStore> store, @Nonnull CommandBuffer<ChunkStore> commandBuffer) {
         if (reason == AddReason.SPAWN) {
            BlockModule.BlockStateInfo blockInfo = (BlockModule.BlockStateInfo)commandBuffer.getComponent(ref, BlockModule.BlockStateInfo.getComponentType());

            assert blockInfo != null;

            Ref<ChunkStore> chunkRef = blockInfo.getChunkRef();
            if (chunkRef != null && chunkRef.isValid()) {
               BlockChunk blockChunk = (BlockChunk)commandBuffer.getComponent(chunkRef, BlockChunk.getComponentType());

               assert blockChunk != null;

               int blockId = blockChunk.getBlock(ChunkUtil.xFromBlockInColumn(blockInfo.getIndex()), ChunkUtil.yFromBlockInColumn(blockInfo.getIndex()), ChunkUtil.zFromBlockInColumn(blockInfo.getIndex()));
               BlockType blockType = (BlockType)BlockType.getAssetMap().getAsset(blockId);
               if (blockType != null) {
                  BlockCounter counter = (BlockCounter)commandBuffer.getResource(BLOCK_COUNTER_RESOURCE_TYPE);
                  String blockName = blockType.getId();
                  counter.trackBlock(blockName);
                  TrackedPlacement tracked = (TrackedPlacement)commandBuffer.getComponent(ref, COMPONENT_TYPE);

                  assert tracked != null;

                  tracked.blockName = blockName;
               }
            }
         }
      }

      public void onEntityRemove(@Nonnull Ref<ChunkStore> ref, @Nonnull RemoveReason reason, @Nonnull Store<ChunkStore> store, @Nonnull CommandBuffer<ChunkStore> commandBuffer) {
         if (reason == RemoveReason.REMOVE) {
            TrackedPlacement tracked = (TrackedPlacement)commandBuffer.getComponent(ref, COMPONENT_TYPE);

            assert tracked != null;

            BlockCounter counter = (BlockCounter)commandBuffer.getResource(BLOCK_COUNTER_RESOURCE_TYPE);
            counter.untrackBlock(tracked.blockName);
         }
      }

      @Nullable
      public Query<ChunkStore> getQuery() {
         return COMPONENT_TYPE;
      }
   }
}