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

import com.hypixel.hytale.builtin.instances.removal.InstanceDataResource;
import com.hypixel.hytale.builtin.portals.resources.PortalWorld;
import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.system.RefSystem;
import com.hypixel.hytale.component.system.tick.DelayedEntitySystem;
import com.hypixel.hytale.protocol.packets.interface_.UpdatePortal;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.server.core.universe.PlayerRef;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import java.time.Instant;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public final class PortalTrackerSystems {
   private PortalTrackerSystems() {
   }

   public static class TrackerSystem extends RefSystem<EntityStore> {
      @Override
      public void onEntityAdded(
         @Nonnull Ref<EntityStore> ref, @Nonnull AddReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
         PortalWorld portalWorld = store.getResource(PortalWorld.getResourceType());
         if (portalWorld.exists()) {
            World world = store.getExternalData().getWorld();
            PlayerRef playerRef = commandBuffer.getComponent(ref, PlayerRef.getComponentType());
            if (portalWorld.getSeesUi().add(playerRef.getUuid())) {
               UpdatePortal packet = portalWorld.createFullPacket(world);
               playerRef.getPacketHandler().write(packet);
            }
         }
      }

      @Override
      public void onEntityRemove(
         @Nonnull Ref<EntityStore> ref, @Nonnull RemoveReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
         PortalWorld portalWorld = store.getResource(PortalWorld.getResourceType());
         if (portalWorld.exists()) {
            PlayerRef playerRef = commandBuffer.getComponent(ref, PlayerRef.getComponentType());
            playerRef.getPacketHandler().write(new UpdatePortal(null, null));
            portalWorld.getSeesUi().remove(playerRef.getUuid());
         }
      }

      @Override
      public Query<EntityStore> getQuery() {
         return Query.and(Player.getComponentType(), PlayerRef.getComponentType());
      }
   }

   public static class UiTickingSystem extends DelayedEntitySystem<EntityStore> {
      public UiTickingSystem() {
         super(1.0F);
      }

      @Override
      public void tick(
         float dt,
         int index,
         @Nonnull ArchetypeChunk<EntityStore> archetypeChunk,
         @Nonnull Store<EntityStore> store,
         @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
         PortalWorld portalWorld = store.getResource(PortalWorld.getResourceType());
         if (portalWorld.exists()) {
            World world = store.getExternalData().getWorld();
            Store<ChunkStore> chunkStore = world.getChunkStore().getStore();
            InstanceDataResource instanceData = chunkStore.getResource(InstanceDataResource.getResourceType());
            Instant timeout = instanceData.getTimeoutTimer();
            if (timeout != null) {
               PlayerRef playerRef = archetypeChunk.getComponent(index, PlayerRef.getComponentType());
               UpdatePortal packet = portalWorld.getSeesUi().add(playerRef.getUuid())
                  ? portalWorld.createFullPacket(world)
                  : portalWorld.createUpdatePacket(world);
               playerRef.getPacketHandler().write(packet);
            }
         }
      }

      @Nullable
      @Override
      public Query<EntityStore> getQuery() {
         return Query.and(Player.getComponentType(), PlayerRef.getComponentType());
      }
   }
}