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

package com.hypixel.hytale.builtin.parkour;

import com.hypixel.hytale.component.Holder;
import com.hypixel.hytale.server.core.modules.entity.tracker.NetworkId;
import com.hypixel.hytale.component.system.HolderSystem;
import java.util.concurrent.TimeUnit;
import com.hypixel.hytale.server.core.Message;
import it.unimi.dsi.fastutil.objects.Object2LongMap;
import java.util.UUID;
import it.unimi.dsi.fastutil.objects.Object2IntMap;
import com.hypixel.hytale.math.vector.Vector3d;
import it.unimi.dsi.fastutil.objects.ObjectList;
import java.util.List;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.dependency.SystemDependency;
import com.hypixel.hytale.component.dependency.OrderPriority;
import com.hypixel.hytale.server.core.modules.entity.system.PlayerSpatialSystem;
import com.hypixel.hytale.component.dependency.Order;
import com.hypixel.hytale.component.dependency.Dependency;
import java.util.Set;
import com.hypixel.hytale.component.spatial.SpatialResource;
import com.hypixel.hytale.component.ResourceType;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.component.system.tick.EntityTickingSystem;
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.server.core.modules.entity.component.TransformComponent;
import com.hypixel.hytale.component.query.Query;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.entity.UUIDComponent;
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 ParkourCheckpointSystems
{
    public static class Init extends RefSystem<EntityStore>
    {
        private final ComponentType<EntityStore, ParkourCheckpoint> parkourCheckpointComponentType;
        @Nonnull
        private final ComponentType<EntityStore, UUIDComponent> uuidComponentComponentType;
        @Nonnull
        private final Query<EntityStore> query;
        
        public Init(final ComponentType<EntityStore, ParkourCheckpoint> parkourCheckpointComponentType) {
            this.parkourCheckpointComponentType = parkourCheckpointComponentType;
            this.uuidComponentComponentType = UUIDComponent.getComponentType();
            final ComponentType<EntityStore, TransformComponent> transformComponentType = TransformComponent.getComponentType();
            this.query = (Query<EntityStore>)Query.and(parkourCheckpointComponentType, transformComponentType);
        }
        
        @Nonnull
        @Override
        public Query<EntityStore> getQuery() {
            return this.query;
        }
        
        @Override
        public void onEntityAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
            final ParkourCheckpoint entity = store.getComponent(ref, this.parkourCheckpointComponentType);
            ParkourPlugin.get().updateLastIndex(entity.getIndex());
            ParkourPlugin.get().getCheckpointUUIDMap().put(entity.getIndex(), store.getComponent(ref, this.uuidComponentComponentType).getUuid());
        }
        
        @Override
        public void onEntityRemove(@Nonnull final Ref<EntityStore> ref, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        }
    }
    
    public static class Ticking extends EntityTickingSystem<EntityStore>
    {
        private final ComponentType<EntityStore, ParkourCheckpoint> parkourCheckpointComponentType;
        private final ComponentType<EntityStore, Player> playerComponentType;
        private final ResourceType<EntityStore, SpatialResource<Ref<EntityStore>, EntityStore>> playerSpatialComponent;
        private final ComponentType<EntityStore, TransformComponent> transformComponentType;
        @Nonnull
        private final ComponentType<EntityStore, UUIDComponent> uuidComponentType;
        @Nonnull
        private final Query<EntityStore> query;
        @Nonnull
        private final Set<Dependency<EntityStore>> dependencies;
        
        public Ticking(final ComponentType<EntityStore, ParkourCheckpoint> parkourCheckpointComponentType, final ComponentType<EntityStore, Player> playerComponentType, final ResourceType<EntityStore, SpatialResource<Ref<EntityStore>, EntityStore>> playerSpatialComponent) {
            this.parkourCheckpointComponentType = parkourCheckpointComponentType;
            this.playerComponentType = playerComponentType;
            this.playerSpatialComponent = playerSpatialComponent;
            this.transformComponentType = TransformComponent.getComponentType();
            this.uuidComponentType = UUIDComponent.getComponentType();
            this.query = (Query<EntityStore>)Query.and(parkourCheckpointComponentType, this.transformComponentType);
            this.dependencies = (Set<Dependency<EntityStore>>)Set.of(new SystemDependency(Order.AFTER, PlayerSpatialSystem.class, OrderPriority.CLOSEST));
        }
        
        @Nonnull
        @Override
        public Query<EntityStore> getQuery() {
            return this.query;
        }
        
        @Nonnull
        @Override
        public Set<Dependency<EntityStore>> getDependencies() {
            return this.dependencies;
        }
        
        @Override
        public void tick(final float dt, final int index, @Nonnull final ArchetypeChunk<EntityStore> archetypeChunk, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
            final int lastIndex = ParkourPlugin.get().getLastIndex();
            if (lastIndex == 0) {
                return;
            }
            final int parkourCheckpointIndex = archetypeChunk.getComponent(index, this.parkourCheckpointComponentType).getIndex();
            final SpatialResource<Ref<EntityStore>, EntityStore> spatialResource = store.getResource(this.playerSpatialComponent);
            final ObjectList<Ref<EntityStore>> results = SpatialResource.getThreadLocalReferenceList();
            final Vector3d position = archetypeChunk.getComponent(index, this.transformComponentType).getPosition();
            spatialResource.getSpatialStructure().ordered(position, 1.0, results);
            final ParkourPlugin parkourPlugin = ParkourPlugin.get();
            final Object2IntMap<UUID> currentCheckpointByPlayerMap = parkourPlugin.getCurrentCheckpointByPlayerMap();
            final Object2LongMap<UUID> startTimeByPlayerMap = parkourPlugin.getStartTimeByPlayerMap();
            for (int i = 0; i < results.size(); ++i) {
                final Ref<EntityStore> otherReference = results.get(i);
                final UUIDComponent uuidComponent = commandBuffer.getComponent(otherReference, this.uuidComponentType);
                final UUID playerUuid = uuidComponent.getUuid();
                final Player player = commandBuffer.getComponent(otherReference, this.playerComponentType);
                handleCheckpointUpdate(currentCheckpointByPlayerMap, startTimeByPlayerMap, player, playerUuid, parkourCheckpointIndex, lastIndex);
            }
        }
        
        private static void handleCheckpointUpdate(@Nonnull final Object2IntMap<UUID> currentCheckpointByPlayerMap, @Nonnull final Object2LongMap<UUID> startTimeByPlayerMap, @Nonnull final Player player, final UUID playerUuid, final int checkpointIndex, final int lastIndex) {
            final int currentCheckpoint = currentCheckpointByPlayerMap.getOrDefault(playerUuid, -1);
            if (currentCheckpoint == -1) {
                if (checkpointIndex != 0) {
                    return;
                }
                currentCheckpointByPlayerMap.put(playerUuid, 0);
                startTimeByPlayerMap.put(playerUuid, java.lang.System.nanoTime());
                player.sendMessage(Message.translation("server.general.parkourRun.started"));
            }
            else {
                if (currentCheckpoint + 1 != checkpointIndex) {
                    return;
                }
                if (lastIndex == checkpointIndex) {
                    final long completionTimeNano = java.lang.System.nanoTime() - startTimeByPlayerMap.getLong(playerUuid);
                    final long completionTimeMillis = TimeUnit.NANOSECONDS.toMillis(completionTimeNano);
                    player.sendMessage(Message.translation("server.general.parkourRun.completed").param("seconds", completionTimeMillis / 1000.0));
                    currentCheckpointByPlayerMap.remove(playerUuid, currentCheckpoint);
                    return;
                }
                currentCheckpointByPlayerMap.put(playerUuid, checkpointIndex);
                player.sendMessage(Message.translation("server.general.parkourRun.checkpointReached").param("checkpoint", checkpointIndex).param("checkpoints", lastIndex));
            }
        }
    }
    
    public static class EnsureNetworkSendable extends HolderSystem<EntityStore>
    {
        private final Query<EntityStore> query;
        
        public EnsureNetworkSendable() {
            this.query = (Query<EntityStore>)Query.and(ParkourCheckpoint.getComponentType(), Query.not((Query<Object>)NetworkId.getComponentType()));
        }
        
        @Override
        public void onEntityAdd(@Nonnull final Holder<EntityStore> holder, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store) {
            holder.addComponent(NetworkId.getComponentType(), new NetworkId(store.getExternalData().takeNextNetworkId()));
        }
        
        @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 this.query;
        }
    }
}