89 lines
3.5 KiB
Java
89 lines
3.5 KiB
Java
package me.xginko.villageroptimizer.models;
|
|
|
|
import me.xginko.villageroptimizer.VillagerOptimizer;
|
|
import me.xginko.villageroptimizer.enums.Keys;
|
|
import me.xginko.villageroptimizer.enums.OptimizationType;
|
|
import org.bukkit.entity.Villager;
|
|
import org.bukkit.persistence.PersistentDataContainer;
|
|
import org.bukkit.persistence.PersistentDataType;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
public final class WrappedVillager {
|
|
|
|
private final @NotNull Villager villager;
|
|
private final @NotNull PersistentDataContainer villagerData;
|
|
|
|
public WrappedVillager(@NotNull Villager villager) {
|
|
this.villager = villager;
|
|
this.villagerData = villager.getPersistentDataContainer();
|
|
}
|
|
|
|
public @NotNull Villager villager() {
|
|
return villager;
|
|
}
|
|
|
|
public static @NotNull WrappedVillager fromVillager(Villager villager) {
|
|
return VillagerOptimizer.getVillagerCache().get(villager);
|
|
}
|
|
|
|
public boolean isOptimized() {
|
|
return villagerData.has(Keys.OPTIMIZED.key());
|
|
}
|
|
|
|
public @NotNull OptimizationType computePossibleOptimization() {
|
|
return VillagerOptimizer.computeOptimization(this);
|
|
}
|
|
|
|
public boolean setOptimization(OptimizationType type) {
|
|
if (type.equals(OptimizationType.OFF) && isOptimized()) {
|
|
villagerData.remove(Keys.OPTIMIZED.key());
|
|
villager.setAware(true);
|
|
villager.setAI(true);
|
|
} else {
|
|
if (isOnOptimizeCooldown()) return false;
|
|
setOptimizeCooldown(VillagerOptimizer.getConfiguration().state_change_cooldown);
|
|
villagerData.set(Keys.OPTIMIZED.key(), PersistentDataType.STRING, type.name());
|
|
villager.setAware(false);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public @NotNull OptimizationType getOptimizationType() {
|
|
return isOptimized() ? OptimizationType.valueOf(villagerData.get(Keys.OPTIMIZED.key(), PersistentDataType.STRING)) : OptimizationType.OFF;
|
|
}
|
|
|
|
public void setOptimizeCooldown(long milliseconds) {
|
|
villagerData.set(Keys.COOLDOWN_OPTIMIZE.key(), PersistentDataType.LONG, System.currentTimeMillis() + milliseconds);
|
|
}
|
|
|
|
public long getOptimizeCooldown() {
|
|
return villagerData.has(Keys.COOLDOWN_OPTIMIZE.key(), PersistentDataType.LONG) ? System.currentTimeMillis() - villagerData.get(Keys.COOLDOWN_OPTIMIZE.key(), PersistentDataType.LONG) : 0L;
|
|
}
|
|
|
|
public boolean isOnOptimizeCooldown() {
|
|
return villagerData.has(Keys.COOLDOWN_OPTIMIZE.key(), PersistentDataType.LONG) && villagerData.get(Keys.COOLDOWN_OPTIMIZE.key(), PersistentDataType.LONG) <= System.currentTimeMillis();
|
|
}
|
|
|
|
public void restock() {
|
|
villager.getRecipes().forEach(recipe -> recipe.setUses(0));
|
|
}
|
|
|
|
public void setExpCooldown(long milliseconds) {
|
|
villagerData.set(Keys.COOLDOWN_EXPERIENCE.key(), PersistentDataType.LONG, System.currentTimeMillis() + milliseconds);
|
|
}
|
|
|
|
public boolean isOnExpCooldown() {
|
|
return villagerData.has(Keys.COOLDOWN_EXPERIENCE.key(), PersistentDataType.LONG) && villagerData.get(Keys.COOLDOWN_EXPERIENCE.key(), PersistentDataType.LONG) <= System.currentTimeMillis();
|
|
}
|
|
|
|
public long saveWorldTime() {
|
|
final long worldTime = villager.getWorld().getFullTime();
|
|
villagerData.set(Keys.WORLDTIME.key(), PersistentDataType.LONG, worldTime);
|
|
return worldTime;
|
|
}
|
|
|
|
public long getSavedWorldTime() {
|
|
return villagerData.has(Keys.WORLDTIME.key(), PersistentDataType.LONG) ? villagerData.get(Keys.WORLDTIME.key(), PersistentDataType.LONG) : saveWorldTime();
|
|
}
|
|
}
|