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

package com.hypixel.hytale.server.core.entity.damage;

import com.hypixel.hytale.server.core.modules.entity.AllLegacyLivingEntityTypesQuery;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.Holder;
import javax.annotation.Nonnull;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.system.HolderSystem;

public class DamageDataSetupSystem extends HolderSystem<EntityStore>
{
    @Nonnull
    private final ComponentType<EntityStore, DamageDataComponent> damageDataComponentType;
    
    public DamageDataSetupSystem(@Nonnull final ComponentType<EntityStore, DamageDataComponent> damageDataComponentType) {
        this.damageDataComponentType = damageDataComponentType;
    }
    
    @Override
    public void onEntityAdd(@Nonnull final Holder<EntityStore> holder, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store) {
        holder.ensureComponent(this.damageDataComponentType);
    }
    
    @Override
    public void onEntityRemoved(@Nonnull final Holder<EntityStore> holder, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store) {
    }
    
    @Nonnull
    @Override
    public Query<EntityStore> getQuery() {
        return AllLegacyLivingEntityTypesQuery.INSTANCE;
    }
}