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

package com.hypixel.hytale.builtin.portals.systems.curse;

import javax.annotation.Nullable;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.server.core.inventory.ItemStack;
import java.util.Set;
import com.hypixel.hytale.server.core.asset.type.item.config.metadata.AdventureMetadata;
import com.hypixel.hytale.server.core.modules.entity.item.ItemComponent;
import com.hypixel.hytale.builtin.portals.resources.PortalWorld;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.AddReason;
import javax.annotation.Nonnull;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.system.RefSystem;

public class CurseItemDropsSystem extends RefSystem<EntityStore>
{
    @Override
    public void onEntityAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        final PortalWorld portalWorld = store.getResource(PortalWorld.getResourceType());
        if (!portalWorld.exists()) {
            return;
        }
        final Set<String> cursedItemsInWorld = portalWorld.getPortalType().getCursedItems();
        if (cursedItemsInWorld.isEmpty()) {
            return;
        }
        final ItemComponent itemComponent = store.getComponent(ref, ItemComponent.getComponentType());
        final ItemStack itemStack = itemComponent.getItemStack();
        if (itemStack == null) {
            return;
        }
        final String itemId = itemStack.getItemId().toString();
        if (!cursedItemsInWorld.contains(itemId)) {
            return;
        }
        final AdventureMetadata adventureMeta = itemStack.getFromMetadataOrDefault("Adventure", AdventureMetadata.CODEC);
        adventureMeta.setCursed(true);
        final ItemStack cursed = itemStack.withMetadata(AdventureMetadata.KEYED_CODEC, adventureMeta);
        itemComponent.setItemStack(cursed);
    }
    
    @Override
    public void onEntityRemove(@Nonnull final Ref<EntityStore> ref, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
    }
    
    @Nullable
    @Override
    public Query<EntityStore> getQuery() {
        return ItemComponent.getComponentType();
    }
}