TeleporterSettingsPageSupplier.java

package com.hypixel.hytale.builtin.adventure.teleporter.page;

import com.hypixel.hytale.builtin.adventure.teleporter.component.Teleporter;
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.ComponentAccessor;
import com.hypixel.hytale.component.Holder;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.math.util.ChunkUtil;
import com.hypixel.hytale.protocol.BlockPosition;
import com.hypixel.hytale.server.core.entity.InteractionContext;
import com.hypixel.hytale.server.core.entity.entities.player.pages.CustomUIPage;
import com.hypixel.hytale.server.core.modules.block.BlockModule;
import com.hypixel.hytale.server.core.modules.interaction.interaction.config.server.OpenCustomUIInteraction;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.chunk.BlockComponentChunk;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class TeleporterSettingsPageSupplier implements OpenCustomUIInteraction.CustomPageSupplier {
   public static final BuilderCodec<TeleporterSettingsPageSupplier> CODEC;
   private boolean create = true;
   private TeleporterSettingsPage.Mode mode;
   @Nullable
   private String activeState;

   public TeleporterSettingsPageSupplier() {
      this.mode = TeleporterSettingsPage.Mode.FULL;
   }

   @Nullable
   public CustomUIPage tryCreate(@Nonnull Ref<EntityStore> ref, ComponentAccessor<EntityStore> componentAccessor, @Nonnull PlayerRef playerRef, @Nonnull InteractionContext context) {
      BlockPosition targetBlock = context.getTargetBlock();
      if (targetBlock == null) {
         return null;
      } else {
         Store<EntityStore> store = ref.getStore();
         World world = ((EntityStore)store.getExternalData()).getWorld();
         ChunkStore chunkStore = world.getChunkStore();
         Ref<ChunkStore> chunkRef = chunkStore.getChunkReference(ChunkUtil.indexChunkFromBlock(targetBlock.x, targetBlock.z));
         BlockComponentChunk blockComponentChunk = chunkRef == null ? null : (BlockComponentChunk)chunkStore.getStore().getComponent(chunkRef, BlockComponentChunk.getComponentType());
         if (blockComponentChunk == null) {
            return null;
         } else {
            int blockIndex = ChunkUtil.indexBlockInColumn(targetBlock.x, targetBlock.y, targetBlock.z);
            Ref<ChunkStore> blockRef = blockComponentChunk.getEntityReference(blockIndex);
            if (blockRef == null || !blockRef.isValid()) {
               if (!this.create) {
                  return null;
               }

               Holder<ChunkStore> holder = ChunkStore.REGISTRY.newHolder();
               holder.putComponent(BlockModule.BlockStateInfo.getComponentType(), new BlockModule.BlockStateInfo(blockIndex, chunkRef));
               holder.ensureComponent(Teleporter.getComponentType());
               blockRef = world.getChunkStore().getStore().addEntity(holder, AddReason.SPAWN);
            }

            return new TeleporterSettingsPage(playerRef, blockRef, this.mode, this.activeState);
         }
      }
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(TeleporterSettingsPageSupplier.class, TeleporterSettingsPageSupplier::new).appendInherited(new KeyedCodec("Create", Codec.BOOLEAN), (supplier, b) -> supplier.create = b, (supplier) -> supplier.create, (supplier, parent) -> supplier.create = parent.create).add()).appendInherited(new KeyedCodec("Mode", TeleporterSettingsPage.Mode.CODEC), (supplier, o) -> supplier.mode = o, (supplier) -> supplier.mode, (supplier, parent) -> supplier.mode = parent.mode).add()).appendInherited(new KeyedCodec("ActiveState", Codec.STRING), (supplier, o) -> supplier.activeState = o, (supplier) -> supplier.activeState, (supplier, parent) -> supplier.activeState = parent.activeState).add()).build();
   }
}