BountyObjectiveTask.java

package com.hypixel.hytale.builtin.adventure.npcobjectives.task;

import com.hypixel.hytale.builtin.adventure.npcobjectives.assets.BountyObjectiveTaskAsset;
import com.hypixel.hytale.builtin.adventure.npcobjectives.resources.KillTrackerResource;
import com.hypixel.hytale.builtin.adventure.npcobjectives.transaction.KillTaskTransaction;
import com.hypixel.hytale.builtin.adventure.objectives.Objective;
import com.hypixel.hytale.builtin.adventure.objectives.ObjectivePlugin;
import com.hypixel.hytale.builtin.adventure.objectives.config.task.ObjectiveTaskAsset;
import com.hypixel.hytale.builtin.adventure.objectives.task.ObjectiveTask;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.RegistrationTransactionRecord;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.SpawnEntityTransactionRecord;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.TransactionRecord;
import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.logger.HytaleLogger;
import com.hypixel.hytale.math.vector.Transform;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.math.vector.Vector3f;
import com.hypixel.hytale.math.vector.Vector3i;
import com.hypixel.hytale.protocol.packets.worldmap.ContextMenuItem;
import com.hypixel.hytale.protocol.packets.worldmap.MapMarker;
import com.hypixel.hytale.server.core.entity.UUIDComponent;
import com.hypixel.hytale.server.core.modules.entity.damage.Damage;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.npc.INonPlayerCharacter;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.core.util.PositionUtil;
import com.hypixel.hytale.server.npc.NPCPlugin;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import it.unimi.dsi.fastutil.Pair;
import java.util.UUID;
import java.util.logging.Level;
import javax.annotation.Nonnull;

public class BountyObjectiveTask extends ObjectiveTask implements KillTask {
   public static final BuilderCodec<BountyObjectiveTask> CODEC;
   boolean completed;
   UUID entityUuid;

   public BountyObjectiveTask(@Nonnull ObjectiveTaskAsset asset, int taskSetIndex, int taskIndex) {
      super(asset, taskSetIndex, taskIndex);
   }

   protected BountyObjectiveTask() {
   }

   @Nonnull
   public BountyObjectiveTaskAsset getAsset() {
      return (BountyObjectiveTaskAsset)super.getAsset();
   }

   @Nonnull
   protected TransactionRecord[] setup0(@Nonnull Objective objective, @Nonnull World world, @Nonnull Store<EntityStore> store) {
      if (this.serializedTransactionRecords != null) {
         return RegistrationTransactionRecord.append(this.serializedTransactionRecords, this.eventRegistry);
      } else {
         Vector3d objectivePosition = objective.getPosition(store);

         assert objectivePosition != null;

         Vector3i spawnPosition = this.getAsset().getWorldLocationProvider().runCondition(world, objectivePosition.clone().floor().toVector3i());
         TransactionRecord[] transactionRecords = new TransactionRecord[2];
         Pair<Ref<EntityStore>, INonPlayerCharacter> npcPair = NPCPlugin.get().spawnNPC(store, this.getAsset().getNpcId(), (String)null, spawnPosition.toVector3d(), Vector3f.ZERO);
         Ref<EntityStore> npcReference = npcPair.first();
         UUIDComponent npcUuidComponent = (UUIDComponent)store.getComponent(npcReference, UUIDComponent.getComponentType());

         assert npcUuidComponent != null;

         UUID npcUuid = npcUuidComponent.getUuid();
         HytaleLogger.Api var10000 = ObjectivePlugin.get().getLogger().at(Level.INFO);
         String var10001 = this.getAsset().getNpcId();
         var10000.log("Spawned Entity '" + var10001 + "' at position: " + String.valueOf(spawnPosition));
         transactionRecords[0] = new SpawnEntityTransactionRecord(world.getWorldConfig().getUuid(), npcUuid);
         this.entityUuid = npcUuid;
         this.addMarker(new MapMarker(getBountyMarkerIDFromUUID(npcUuid), "Bounty Target", "Home.png", PositionUtil.toTransformPacket(new Transform(spawnPosition)), (ContextMenuItem[])null));
         KillTaskTransaction transaction = new KillTaskTransaction(this, objective, store);
         ((KillTrackerResource)store.getResource(KillTrackerResource.getResourceType())).watch(transaction);
         transactionRecords[1] = transaction;
         return transactionRecords;
      }
   }

   public boolean checkCompletion() {
      return this.completed;
   }

   @Nonnull
   public static String getBountyMarkerIDFromUUID(@Nonnull UUID uuid) {
      return "Bounty_" + String.valueOf(uuid);
   }

   public void checkKilledEntity(@Nonnull Store<EntityStore> store, @Nonnull Ref<EntityStore> npcRef, @Nonnull Objective objective, NPCEntity npc, Damage damageInfo) {
      UUIDComponent uuidComponent = (UUIDComponent)store.getComponent(npcRef, UUIDComponent.getComponentType());

      assert uuidComponent != null;

      UUID uuid = uuidComponent.getUuid();
      if (this.entityUuid.equals(uuid)) {
         this.completed = true;
         this.consumeTaskConditions(store, npcRef, objective.getPlayerUUIDs());
         this.complete(objective, store);
         objective.checkTaskSetCompletion(store);
         this.removeMarker(getBountyMarkerIDFromUUID(uuid));
      }
   }

   @Nonnull
   public com.hypixel.hytale.protocol.ObjectiveTask toPacket(@Nonnull Objective objective) {
      com.hypixel.hytale.protocol.ObjectiveTask packet = new com.hypixel.hytale.protocol.ObjectiveTask();
      packet.taskDescriptionKey = this.asset.getDescriptionKey(objective.getObjectiveId(), this.taskSetIndex, this.taskIndex);
      packet.currentCompletion = this.completed ? 1 : 0;
      packet.completionNeeded = 1;
      return packet;
   }

   @Nonnull
   public String toString() {
      boolean var10000 = this.completed;
      return "BountyObjectiveTask{completed=" + var10000 + ", entityUuid=" + String.valueOf(this.entityUuid) + "} " + super.toString();
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(BountyObjectiveTask.class, BountyObjectiveTask::new, ObjectiveTask.BASE_CODEC).append(new KeyedCodec("Completed", Codec.BOOLEAN), (bountyObjectiveTask, aBoolean) -> bountyObjectiveTask.completed = aBoolean, (bountyObjectiveTask) -> bountyObjectiveTask.completed).add()).append(new KeyedCodec("EntityUUID", Codec.UUID_BINARY), (bountyObjectiveTask, uuid) -> bountyObjectiveTask.entityUuid = uuid, (bountyObjectiveTask) -> bountyObjectiveTask.entityUuid).add()).build();
   }
}