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

package com.hypixel.hytale.server.core.universe.system;

import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.dependency.RootDependency;
import com.hypixel.hytale.component.dependency.Dependency;
import java.util.Set;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.system.RefSystem;

public class PlayerRefAddedSystem extends RefSystem<EntityStore>
{
    @Nonnull
    private final ComponentType<EntityStore, PlayerRef> playerRefComponentType;
    
    public PlayerRefAddedSystem(@Nonnull final ComponentType<EntityStore, PlayerRef> playerRefComponentType) {
        this.playerRefComponentType = playerRefComponentType;
    }
    
    @Nonnull
    @Override
    public Set<Dependency<EntityStore>> getDependencies() {
        return RootDependency.firstSet();
    }
    
    @Override
    public Query<EntityStore> getQuery() {
        return this.playerRefComponentType;
    }
    
    @Override
    public void onEntityAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        final PlayerRef playerRefComponent = store.getComponent(ref, this.playerRefComponentType);
        assert playerRefComponent != null;
        playerRefComponent.addedToStore(ref);
        store.getExternalData().getWorld().trackPlayerRef(playerRefComponent);
    }
    
    @Override
    public void onEntityRemove(@Nonnull final Ref<EntityStore> ref, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        final PlayerRef playerRefComponent = store.getComponent(ref, this.playerRefComponentType);
        assert playerRefComponent != null;
        store.getExternalData().getWorld().untrackPlayerRef(playerRefComponent);
    }
}
