package com.hypixel.hytale.server.flock;

import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.codecs.EnumCodec;
import com.hypixel.hytale.component.Component;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import java.util.UUID;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class FlockMembership implements Component<EntityStore> {
   public static final int VERSION = 5;
   public static final BuilderCodec<FlockMembership> CODEC = BuilderCodec.builder(FlockMembership.class, FlockMembership::new)
      .legacyVersioned()
      .codecVersion(5)
      .<UUID>append(new KeyedCodec<>("FlockId", Codec.UUID_BINARY), (membership, uuid) -> membership.flockId = uuid, membership -> membership.flockId)
      .setVersionRange(5, 5)
      .add()
      .append(
         new KeyedCodec<>("Type", new EnumCodec<>(FlockMembership.Type.class, EnumCodec.EnumStyle.LEGACY)),
         (membership, type) -> membership.membershipType = type,
         membership -> membership.membershipType
      )
      .add()
      .build();
   private UUID flockId;
   private FlockMembership.Type membershipType;
   @Nullable
   private Ref<EntityStore> flockRef;

   public static ComponentType<EntityStore, FlockMembership> getComponentType() {
      return FlockPlugin.get().getFlockMembershipComponentType();
   }

   public UUID getFlockId() {
      return this.flockId;
   }

   public void setFlockId(UUID flockId) {
      this.flockId = flockId;
   }

   @Nullable
   public Ref<EntityStore> getFlockRef() {
      return this.flockRef;
   }

   public void setFlockRef(Ref<EntityStore> flockRef) {
      this.flockRef = flockRef;
   }

   public void setMembershipType(FlockMembership.Type membershipType) {
      this.membershipType = membershipType;
   }

   public FlockMembership.Type getMembershipType() {
      return this.membershipType;
   }

   public void unload() {
      this.flockRef = null;
   }

   @Nonnull
   @Override
   public Component<EntityStore> clone() {
      FlockMembership membership = new FlockMembership();
      membership.setFlockId(this.flockId);
      membership.setFlockRef(this.flockRef);
      membership.setMembershipType(this.membershipType);
      return membership;
   }

   public static enum Type {
      JOINING(false),
      MEMBER(false),
      LEADER(true),
      INTERIM_LEADER(true);

      private final boolean actsAsLeader;

      private Type(boolean actsAsLeader) {
         this.actsAsLeader = actsAsLeader;
      }

      public boolean isActingAsLeader() {
         return this.actsAsLeader;
      }
   }
}