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.server.core.modules.entity.component.EntityScaleComponent;
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">public class EntityScaleSnapshot class="kw">implements EntitySnapshot<EntityScaleSnapshot> {
   @Nonnull
   class="kw">private Ref<EntityStore> ref;
   class="kw">private class="kw">final float scale;

   class="kw">public EntityScaleSnapshot(@Nonnull Ref<EntityStore> var1, float var2) {
      this.ref = var1;
      this.scale = var2;
   }

   class="kw">public EntityScaleSnapshot(@Nonnull Ref<EntityStore> var1, @Nonnull ComponentAccessor<EntityStore> var2) {
      this.ref = var1;
      EntityScaleComponent var3 = var2.getComponent(var1, EntityScaleComponent.getComponentType());
      this.scale = var3 != null ? var3.getScale() : 1.0F;
   }

   @Override
   class="kw">public void updateEntityRef(@Nonnull Ref<EntityStore> var1, @Nonnull Ref<EntityStore> var2) {
      if (this.ref == var1) {
         this.ref = var2;
      }
   }

   @Nonnull
   class="kw">public Ref<EntityStore> getRef() {
      class="kw">return this.ref;
   }

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

      EntityScaleComponent var4 = var3.getComponent(this.ref, EntityScaleComponent.getComponentType());
      float var5 = var4 != null ? var4.getScale() : 1.0F;
      if (var4 == null) {
         var4 = new EntityScaleComponent(this.scale);
         var3.addComponent(this.ref, EntityScaleComponent.getComponentType(), var4);
      } else {
         var4.setScale(this.scale);
      }

      class="kw">return new EntityScaleSnapshot(this.ref, var5);
   }
}