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

package com.hypixel.hytale.builtin.adventure.npcobjectives.task;

import com.hypixel.hytale.builtin.adventure.objectives.config.task.ObjectiveTaskAsset;
import com.hypixel.hytale.builtin.adventure.objectives.config.task.CountObjectiveTaskAsset;
import it.unimi.dsi.fastutil.Pair;
import com.hypixel.hytale.server.spawning.wrappers.BeaconSpawnWrapper;
import com.hypixel.hytale.builtin.adventure.objectives.config.worldlocationproviders.WorldLocationProvider;
import com.hypixel.hytale.logger.HytaleLogger;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.SpawnEntityTransactionRecord;
import java.util.logging.Level;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.entity.UUIDComponent;
import com.hypixel.hytale.server.spawning.beacons.LegacySpawnBeaconEntity;
import com.hypixel.hytale.math.vector.Vector3f;
import com.hypixel.hytale.server.spawning.SpawningPlugin;
import java.util.Arrays;
import com.hypixel.hytale.server.spawning.assets.spawns.config.BeaconNPCSpawn;
import com.hypixel.hytale.builtin.adventure.objectives.ObjectivePlugin;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.WorldTransactionRecord;
import com.hypixel.hytale.common.util.ArrayUtil;
import com.hypixel.hytale.builtin.adventure.npcobjectives.resources.KillTrackerResource;
import com.hypixel.hytale.builtin.adventure.npcobjectives.transaction.KillTaskTransaction;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.TransactionUtil;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.builtin.adventure.objectives.transaction.TransactionRecord;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.builtin.adventure.objectives.Objective;
import com.hypixel.hytale.builtin.adventure.npcobjectives.assets.KillObjectiveTaskAsset;
import javax.annotation.Nonnull;
import com.hypixel.hytale.builtin.adventure.npcobjectives.assets.KillSpawnBeaconObjectiveTaskAsset;
import com.hypixel.hytale.codec.builder.BuilderCodec;

public class KillSpawnBeaconObjectiveTask extends KillObjectiveTask
{
    public static final BuilderCodec<KillSpawnBeaconObjectiveTask> CODEC;
    
    public KillSpawnBeaconObjectiveTask(@Nonnull final KillSpawnBeaconObjectiveTaskAsset asset, final int taskSetIndex, final int taskIndex) {
        super(asset, taskSetIndex, taskIndex);
    }
    
    protected KillSpawnBeaconObjectiveTask() {
    }
    
    @Nonnull
    @Override
    public KillSpawnBeaconObjectiveTaskAsset getAsset() {
        return (KillSpawnBeaconObjectiveTaskAsset)super.getAsset();
    }
    
    @Nonnull
    @Override
    protected TransactionRecord[] setup0(@Nonnull final Objective objective, @Nonnull final World world, @Nonnull final Store<EntityStore> store) {
        TransactionRecord[] transactionRecords = this.serializedTransactionRecords;
        if (transactionRecords == null) {
            transactionRecords = this.setupSpawnBeacons(objective, world, store);
            if (TransactionUtil.anyFailed(transactionRecords)) {
                return transactionRecords;
            }
        }
        final KillTaskTransaction transaction = new KillTaskTransaction(this, objective, store);
        store.getResource(KillTrackerResource.getResourceType()).watch(transaction);
        return ArrayUtil.append(transactionRecords, transaction);
    }
    
    @Nonnull
    private TransactionRecord[] setupSpawnBeacons(@Nonnull final Objective objective, @Nonnull final World world, @Nonnull final ComponentAccessor<EntityStore> componentAccessor) {
        final Vector3d position = objective.getPosition(componentAccessor);
        if (position == null) {
            return TransactionRecord.appendFailedTransaction(null, new WorldTransactionRecord(), "No valid position found for the objective.");
        }
        final KillSpawnBeaconObjectiveTaskAsset.ObjectiveSpawnBeacon[] spawnBeaconConfigs = this.getAsset().getSpawnBeacons();
        final TransactionRecord[] transactionRecords = new TransactionRecord[spawnBeaconConfigs.length];
        final HytaleLogger logger = ObjectivePlugin.get().getLogger();
        for (int i = 0; i < spawnBeaconConfigs.length; ++i) {
            Vector3d spawnPosition = position.clone();
            final KillSpawnBeaconObjectiveTaskAsset.ObjectiveSpawnBeacon spawnBeaconConfig = spawnBeaconConfigs[i];
            final String spawnBeaconId = spawnBeaconConfig.getSpawnBeaconId();
            final int index = BeaconNPCSpawn.getAssetMap().getIndex(spawnBeaconId);
            if (index == Integer.MIN_VALUE) {
                transactionRecords[i] = new WorldTransactionRecord().fail("Failed to find spawn beacon " + spawnBeaconId);
                return Arrays.copyOf(transactionRecords, i + 1);
            }
            final Vector3d offset = spawnBeaconConfig.getOffset();
            if (offset != null) {
                spawnPosition.add(offset);
            }
            final WorldLocationProvider worldLocationCondition = spawnBeaconConfig.getWorldLocationProvider();
            if (worldLocationCondition != null) {
                spawnPosition = worldLocationCondition.runCondition(world, spawnPosition.toVector3i()).toVector3d();
            }
            if (spawnPosition == null) {
                transactionRecords[i] = new WorldTransactionRecord().fail("Failed to find a valid position to spawn beacon " + spawnBeaconId);
            }
            else {
                final BeaconSpawnWrapper wrapper = SpawningPlugin.get().getBeaconSpawnWrapper(index);
                final Pair<Ref<EntityStore>, LegacySpawnBeaconEntity> spawnBeaconPair = LegacySpawnBeaconEntity.create(wrapper, spawnPosition, Vector3f.FORWARD, componentAccessor);
                spawnBeaconPair.second().setObjectiveUUID(objective.getObjectiveUUID());
                final UUIDComponent spawnBeaconUuidComponent = componentAccessor.getComponent(spawnBeaconPair.first(), UUIDComponent.getComponentType());
                assert spawnBeaconUuidComponent != null;
                logger.at(Level.INFO).log("Spawned SpawnBeacon '" + spawnBeaconId + "' at position: " + String.valueOf(position));
                transactionRecords[i] = new SpawnEntityTransactionRecord(world.getWorldConfig().getUuid(), spawnBeaconUuidComponent.getUuid());
            }
        }
        return transactionRecords;
    }
    
    @Nonnull
    @Override
    public String toString() {
        return "KillSpawnBeaconObjectiveTask{} " + super.toString();
    }
    
    static {
        CODEC = BuilderCodec.builder(KillSpawnBeaconObjectiveTask.class, KillSpawnBeaconObjectiveTask::new, KillObjectiveTask.CODEC).build();
    }
}