// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.builtin.beds.sleep.systems.player;

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

public class WakeUpOnDismountSystem extends RefChangeSystem<EntityStore, MountedComponent>
{
    @Override
    public ComponentType<EntityStore, MountedComponent> componentType() {
        return MountedComponent.getComponentType();
    }
    
    @Override
    public Query<EntityStore> getQuery() {
        return MountedComponent.getComponentType();
    }
    
    @Override
    public void onComponentAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final MountedComponent component, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
    }
    
    @Override
    public void onComponentSet(@Nonnull final Ref<EntityStore> ref, @Nullable final MountedComponent oldComponent, @Nonnull final MountedComponent newComponent, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
    }
    
    @Override
    public void onComponentRemoved(@Nonnull final Ref<EntityStore> ref, @Nonnull final MountedComponent component, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        if (component.getBlockMountType() == BlockMountType.Bed) {
            commandBuffer.putComponent(ref, PlayerSomnolence.getComponentType(), PlayerSomnolence.AWAKE);
        }
    }
}