NPCMountSystems.java

package com.hypixel.hytale.builtin.mounts;

import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.system.RefSystem;
import com.hypixel.hytale.protocol.Packet;
import com.hypixel.hytale.protocol.packets.interaction.MountNPC;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.server.core.modules.entity.damage.DeathComponent;
import com.hypixel.hytale.server.core.modules.entity.damage.DeathSystems;
import com.hypixel.hytale.server.core.modules.entity.tracker.NetworkId;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.server.npc.systems.RoleChangeSystem;
import javax.annotation.Nonnull;

public class NPCMountSystems {
   public NPCMountSystems() {
   }

   public static class OnAdd extends RefSystem<EntityStore> {
      @Nonnull
      private final ComponentType<EntityStore, NPCMountComponent> mountComponentType;

      public OnAdd(@Nonnull ComponentType<EntityStore, NPCMountComponent> mountRoleChangeComponentType) {
         this.mountComponentType = mountRoleChangeComponentType;
      }

      public Query<EntityStore> getQuery() {
         return this.mountComponentType;
      }

      public void onEntityAdded(@Nonnull Ref<EntityStore> ref, @Nonnull AddReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer) {
         NPCMountComponent mountComponent = (NPCMountComponent)store.getComponent(ref, this.mountComponentType);

         assert mountComponent != null;

         PlayerRef playerRef = mountComponent.getOwnerPlayerRef();
         if (playerRef == null) {
            resetOriginalRoleMount(ref, store, commandBuffer, mountComponent);
         } else {
            NPCEntity npcComponent = (NPCEntity)store.getComponent(ref, NPCEntity.getComponentType());

            assert npcComponent != null;

            NetworkId networkIdComponent = (NetworkId)store.getComponent(ref, NetworkId.getComponentType());

            assert networkIdComponent != null;

            int networkId = networkIdComponent.getId();
            MountNPC packet = new MountNPC(mountComponent.getAnchorX(), mountComponent.getAnchorY(), mountComponent.getAnchorZ(), networkId);
            Player playerComponent = (Player)playerRef.getComponent(Player.getComponentType());

            assert playerComponent != null;

            playerComponent.setMountEntityId(networkId);
            playerRef.getPacketHandler().write((Packet)packet);
         }
      }

      private static void resetOriginalRoleMount(@Nonnull Ref<EntityStore> ref, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer, @Nonnull NPCMountComponent mountComponent) {
         NPCEntity npcComponent = (NPCEntity)store.getComponent(ref, NPCEntity.getComponentType());

         assert npcComponent != null;

         RoleChangeSystem.requestRoleChange(ref, npcComponent.getRole(), mountComponent.getOriginalRoleIndex(), false, "Idle", (String)null, store);
         commandBuffer.removeComponent(ref, NPCMountComponent.getComponentType());
      }

      public void onEntityRemove(@Nonnull Ref<EntityStore> ref, @Nonnull RemoveReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer) {
      }
   }

   public static class DismountOnPlayerDeath extends DeathSystems.OnDeathSystem {
      public DismountOnPlayerDeath() {
      }

      @Nonnull
      public Query<EntityStore> getQuery() {
         return Player.getComponentType();
      }

      public void onComponentAdded(@Nonnull Ref<EntityStore> ref, @Nonnull DeathComponent component, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer) {
         Player playerComponent = (Player)store.getComponent(ref, Player.getComponentType());

         assert playerComponent != null;

         MountPlugin.checkDismountNpc(commandBuffer, playerComponent);
      }
   }

   public static class DismountOnMountDeath extends DeathSystems.OnDeathSystem {
      public DismountOnMountDeath() {
      }

      public Query<EntityStore> getQuery() {
         return NPCMountComponent.getComponentType();
      }

      public void onComponentAdded(@Nonnull Ref<EntityStore> ref, @Nonnull DeathComponent component, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer) {
         NPCMountComponent mountComponent = (NPCMountComponent)store.getComponent(ref, NPCMountComponent.getComponentType());

         assert mountComponent != null;

         PlayerRef playerRef = mountComponent.getOwnerPlayerRef();
         if (playerRef != null) {
            MountPlugin.resetOriginalPlayerMovementSettings(playerRef, store);
         }

      }
   }
}