class="kw">package com.hypixel.hytale.builtin.beds.sleep.systems.player;

class="kw">import com.hypixel.hytale.builtin.beds.sleep.components.PlayerSomnolence;
class="kw">import com.hypixel.hytale.builtin.mounts.MountedComponent;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.ComponentType;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.component.query.Query;
class="kw">import com.hypixel.hytale.component.system.RefChangeSystem;
class="kw">import com.hypixel.hytale.protocol.BlockMountType;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class WakeUpOnDismountSystem class="kw">extends RefChangeSystem<EntityStore, MountedComponent> {
   @Nonnull
   class="kw">private class="kw">final ComponentType<EntityStore, MountedComponent> mountedComponentType;
   @Nonnull
   class="kw">private class="kw">final ComponentType<EntityStore, PlayerSomnolence> playerSomnolenceComponentType;

   class="kw">public WakeUpOnDismountSystem(@Nonnull ComponentType<EntityStore, MountedComponent> var1, @Nonnull ComponentType<EntityStore, PlayerSomnolence> var2) {
      this.mountedComponentType = var1;
      this.playerSomnolenceComponentType = var2;
   }

   @Nonnull
   @Override
   class="kw">public ComponentType<EntityStore, MountedComponent> componentType() {
      class="kw">return this.mountedComponentType;
   }

   @Nonnull
   @Override
   class="kw">public Query<EntityStore> getQuery() {
      class="kw">return this.mountedComponentType;
   }

   class="kw">public void onComponentAdded(
      @Nonnull Ref<EntityStore> var1, @Nonnull MountedComponent var2, @Nonnull Store<EntityStore> var3, @Nonnull CommandBuffer<EntityStore> var4
   ) {
   }

   class="kw">public void onComponentSet(
      @Nonnull Ref<EntityStore> var1,
      @Nullable MountedComponent var2,
      @Nonnull MountedComponent var3,
      @Nonnull Store<EntityStore> var4,
      @Nonnull CommandBuffer<EntityStore> var5
   ) {
   }

   class="kw">public void onComponentRemoved(
      @Nonnull Ref<EntityStore> var1, @Nonnull MountedComponent var2, @Nonnull Store<EntityStore> var3, @Nonnull CommandBuffer<EntityStore> var4
   ) {
      if (var2.getBlockMountType() == BlockMountType.Bed) {
         var4.putComponent(var1, this.playerSomnolenceComponentType, PlayerSomnolence.AWAKE);
      }
   }
}