class="kw">package com.hypixel.hytale.builtin.buildertools.snapshot;

class="kw">import com.hypixel.hytale.component.ComponentAccessor;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.RemoveReason;
class="kw">import com.hypixel.hytale.server.core.universe.PlayerRef;
class="kw">import com.hypixel.hytale.server.core.universe.world.World;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class EntityAddSnapshot class="kw">implements EntitySnapshot<EntityRemoveSnapshot> {
   class="kw">private class="kw">final Ref<EntityStore> entityRef;

   class="kw">public EntityAddSnapshot(Ref<EntityStore> var1) {
      this.entityRef = var1;
   }

   class="kw">public Ref<EntityStore> getEntityRef() {
      class="kw">return this.entityRef;
   }

   @Nullable
   class="kw">public EntityRemoveSnapshot restoreEntity(@Nonnull PlayerRef var1, @Nonnull World var2, @Nonnull ComponentAccessor<EntityStore> var3) {
      if (!this.entityRef.isValid()) {
         class="kw">return null;
      }

      EntityRemoveSnapshot var4 = new EntityRemoveSnapshot(this.entityRef);
      var2.getEntityStore().getStore().removeEntity(this.entityRef, RemoveReason.BUILDER_TOOLS_UNDO);
      class="kw">return var4;
   }
}