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

package com.hypixel.hytale.server.core.command.commands.world.entity;

import it.unimi.dsi.fastutil.objects.ObjectListIterator;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.server.core.asset.type.entityeffect.config.OverlapBehavior;
import com.hypixel.hytale.server.core.entity.effect.EffectControllerComponent;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.Ref;
import it.unimi.dsi.fastutil.objects.ObjectList;
import com.hypixel.hytale.server.core.command.system.CommandContext;
import com.hypixel.hytale.codec.validation.Validator;
import com.hypixel.hytale.codec.validation.Validators;
import com.hypixel.hytale.server.core.command.system.arguments.types.ArgumentType;
import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
import com.hypixel.hytale.server.core.command.system.arguments.system.DefaultArg;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.asset.type.entityeffect.config.EntityEffect;
import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
import com.hypixel.hytale.server.core.command.system.basecommands.AbstractTargetEntityCommand;

public class EntityEffectCommand extends AbstractTargetEntityCommand
{
    @Nonnull
    private final RequiredArg<EntityEffect> effectArg;
    @Nonnull
    private final DefaultArg<Float> durationArg;
    
    public EntityEffectCommand() {
        super("effect", "server.commands.entity.effect.desc");
        this.effectArg = this.withRequiredArg("effect", "server.commands.entity.effect.effect.desc", ArgTypes.EFFECT_ASSET);
        this.durationArg = this.withDefaultArg("duration", "server.commands.entity.effect.duration.desc", ArgTypes.FLOAT, 100.0f, "server.commands.entity.effect.duration.default").addValidator((Validator<Float>)Validators.greaterThan((DataType)0.0f));
    }
    
    @Override
    protected void execute(@Nonnull final CommandContext context, @Nonnull final ObjectList<Ref<EntityStore>> entities, @Nonnull final World world, @Nonnull final Store<EntityStore> store) {
        final EntityEffect entityEffect = this.effectArg.get(context);
        final float duration = this.durationArg.get(context);
        for (final Ref<EntityStore> entityRef : entities) {
            final EffectControllerComponent effectControllerComponent = store.getComponent(entityRef, EffectControllerComponent.getComponentType());
            if (effectControllerComponent != null) {
                effectControllerComponent.addEffect(entityRef, entityEffect, duration, OverlapBehavior.OVERWRITE, store);
            }
        }
    }
}