hytale-server/com/hypixel/hytale/builtin/crafting/window/BenchWindow.java

package com.hypixel.hytale.builtin.crafting.window;

import com.google.gson.JsonObject;
import com.hypixel.hytale.builtin.adventure.memories.MemoriesPlugin;
import com.hypixel.hytale.builtin.crafting.component.CraftingManager;
import com.hypixel.hytale.builtin.crafting.state.BenchState;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.protocol.SoundCategory;
import com.hypixel.hytale.protocol.packets.window.WindowType;
import com.hypixel.hytale.server.core.asset.type.blocktype.config.bench.Bench;
import com.hypixel.hytale.server.core.asset.type.item.config.Item;
import com.hypixel.hytale.server.core.entity.entities.player.windows.BlockWindow;
import com.hypixel.hytale.server.core.entity.entities.player.windows.MaterialContainerWindow;
import com.hypixel.hytale.server.core.entity.entities.player.windows.MaterialExtraResourcesSection;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.SoundUtil;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import javax.annotation.Nonnull;

public abstract class BenchWindow extends BlockWindow implements MaterialContainerWindow {
   private static final float CRAFTING_UPDATE_MIN_PERCENT = 0.05F;
   private static final long CRAFTING_UPDATE_INTERVAL_MS = 500L;
   protected static final String BENCH_UPGRADING = "BenchUpgrading";
   private float lastUpdatePercent;
   private long lastUpdateTimeMs;
   protected final Bench bench;
   protected final BenchState benchState;
   protected final JsonObject windowData = new JsonObject();
   @Nonnull
   private MaterialExtraResourcesSection extraResourcesSection = new MaterialExtraResourcesSection();

   public BenchWindow(@Nonnull WindowType windowType, @Nonnull BenchState benchState) {
      super(windowType, benchState.getBlockX(), benchState.getBlockY(), benchState.getBlockZ(), benchState.getRotationIndex(), benchState.getBlockType());
      this.bench = this.blockType.getBench();
      this.benchState = benchState;
      Item item = this.blockType.getItem();
      this.windowData.addProperty("type", this.bench.getType().ordinal());
      this.windowData.addProperty("id", this.bench.getId());
      this.windowData.addProperty("name", item.getTranslationKey());
      this.windowData.addProperty("blockItemId", item.getId());
      this.windowData.addProperty("tierLevel", this.getBenchTierLevel());
   }

   @Nonnull
   @Override
   public JsonObject getData() {
      return this.windowData;
   }

   @Override
   protected boolean onOpen0() {
      PlayerRef playerRef = this.getPlayerRef();
      Ref<EntityStore> ref = playerRef.getReference();
      Store<EntityStore> store = ref.getStore();
      CraftingManager craftingManager = store.getComponent(ref, CraftingManager.getComponentType());
      craftingManager.setBench(this.x, this.y, this.z, this.blockType);
      World world = store.getExternalData().getWorld();
      this.windowData.addProperty("worldMemoriesLevel", MemoriesPlugin.get().getMemoriesLevel(world.getGameplayConfig()));
      return true;
   }

   protected int getBenchTierLevel() {
      return this.benchState != null ? this.benchState.getTierLevel() : 1;
   }

   @Override
   public void onClose0() {
      PlayerRef playerRef = this.getPlayerRef();
      Ref<EntityStore> ref = playerRef.getReference();
      Store<EntityStore> store = ref.getStore();
      CraftingManager craftingManagerComponent = store.getComponent(ref, CraftingManager.getComponentType());

      assert craftingManagerComponent != null;

      if (craftingManagerComponent.clearBench(ref, store) && this.bench.getFailedSoundEventIndex() != 0) {
         SoundUtil.playSoundEvent2d(ref, this.bench.getFailedSoundEventIndex(), SoundCategory.UI, store);
      }
   }

   public void updateCraftingJob(float percent) {
      this.windowData.addProperty("progress", percent);
      this.checkProgressInvalidate(percent);
   }

   public void updateBenchUpgradeJob(float percent) {
      this.windowData.addProperty("tierUpgradeProgress", percent);
      this.checkProgressInvalidate(percent);
   }

   private void checkProgressInvalidate(float percent) {
      if (this.lastUpdatePercent != percent) {
         long time = System.currentTimeMillis();
         if (percent >= 1.0F
            || percent < this.lastUpdatePercent
            || percent - this.lastUpdatePercent > 0.05F
            || time - this.lastUpdateTimeMs > 500L
            || this.lastUpdateTimeMs == 0L) {
            this.lastUpdatePercent = percent;
            this.lastUpdateTimeMs = time;
            this.invalidate();
         }
      }
   }

   public void updateBenchTierLevel(int newValue) {
      this.windowData.addProperty("tierLevel", newValue);
      this.updateBenchUpgradeJob(0.0F);
      this.setNeedRebuild();
      this.invalidate();
   }

   @Nonnull
   @Override
   public MaterialExtraResourcesSection getExtraResourcesSection() {
      if (!this.extraResourcesSection.isValid()) {
         CraftingManager.feedExtraResourcesSection(this.benchState, this.extraResourcesSection);
      }

      return this.extraResourcesSection;
   }

   @Override
   public void invalidateExtraResources() {
      this.extraResourcesSection.setValid(false);
      this.invalidate();
   }

   @Override
   public boolean isValid() {
      return this.extraResourcesSection.isValid();
   }
}