package com.alttd.altitudequests.objects.quests; import com.alttd.altitudequests.config.Config; import com.alttd.altitudequests.config.QuestsConfig; import com.alttd.altitudequests.database.Database; import com.alttd.altitudequests.objects.Quest; import com.alttd.altitudequests.objects.variants.CollectDropsQuestObject; import com.alttd.altitudequests.util.Logger; import com.alttd.altitudequests.util.Utilities; import net.kyori.adventure.text.Component; import net.kyori.adventure.text.minimessage.MiniMessage; import net.kyori.adventure.text.minimessage.tag.resolver.Placeholder; import net.kyori.adventure.text.minimessage.tag.resolver.TagResolver; import org.bukkit.entity.Player; import org.bukkit.inventory.ItemStack; import org.bukkit.inventory.PlayerInventory; import java.sql.PreparedStatement; import java.sql.SQLException; import java.util.Arrays; import java.util.List; import java.util.Objects; import java.util.UUID; public class CollectDropsQuest extends Quest { private final CollectDropsQuestObject collectDropsQuestObject; public CollectDropsQuest(UUID uuid) { super(uuid, 0, 0, QuestsConfig.COLLECT_DROPS_QUEST.get(Utilities.randomOr0(QuestsConfig.COLLECT_DROPS_QUEST.size() - 1))); if (getVariant() instanceof CollectDropsQuestObject collectDropsQuestObject) this.collectDropsQuestObject = collectDropsQuestObject; else this.collectDropsQuestObject = null; if (collectDropsQuestObject == null) { Logger.warning("Tried to create collectDropsQuest but unable to find variant: %.", "unknown"); return; } } public CollectDropsQuest(UUID uuid, int step1, int step2, String variant) { super(uuid, step1, step2, QuestsConfig.COLLECT_DROPS_QUEST.stream() .filter(object -> variant.equals(object.getInternalName())) .findAny().orElse(null)); if (getVariant() instanceof CollectDropsQuestObject collectDropsQuestObject) this.collectDropsQuestObject = collectDropsQuestObject; else this.collectDropsQuestObject = null; if (collectDropsQuestObject == null) { Logger.warning("Tried to create collectDropsQuest but unable to find variant: %.", variant); return; } checkDone(); } @Override public void save() { String sql = "INSERT INTO generic_quest_progress " + "(year_day, uuid, quest, quest_variant, step_1_progress, step_2_progress) " + "VALUES (?, ?, ?, ?, ?, ?) " + "ON DUPLICATE KEY UPDATE " + "quest = ?, quest_variant = ?, step_1_progress = ?, step_2_progress = ?, year_day = ?"; try { PreparedStatement statement = Database.getDatabase().getConnection().prepareStatement(sql); int yearDay = Utilities.getYearDay(); if (Config.DEBUG) Logger.info("Saving user for year day %.", String.valueOf(yearDay)); statement.setInt(1, yearDay); statement.setString(2, getUuid().toString()); statement.setString(3, this.getClass().getSimpleName()); statement.setString(4, collectDropsQuestObject.getInternalName()); statement.setInt(5, getStep1()); statement.setInt(6, getStep2()); statement.setString(7, this.getClass().getSimpleName()); statement.setString(8, collectDropsQuestObject.getInternalName()); statement.setInt(9, getStep1()); statement.setInt(10, getStep2()); statement.setInt(11, yearDay); statement.execute(); } catch (SQLException exception) { exception.printStackTrace(); } } @Override public TagResolver getTagResolvers() { return TagResolver.resolver( Placeholder.unparsed("item", collectDropsQuestObject.getMaterial().name()), Placeholder.parsed("step_1_progress", getStep1() == collectDropsQuestObject.getAmount() ? "" + getStep1() + "" : "" + getStep1() + ""), Placeholder.parsed("step_1_total", String.valueOf(collectDropsQuestObject.getAmount())), Placeholder.parsed("step_2_progress", getStep2() == collectDropsQuestObject.getAmount() ? "" + getStep2() + "" : "" + getStep2() + ""), Placeholder.parsed("step_2_total", String.valueOf(collectDropsQuestObject.getAmount())) ); } @Override public int turnIn(Player player) { PlayerInventory inventory = player.getInventory(); int maxToTurnIn = getMaxToTurnIn(); if (maxToTurnIn == 0) return 0; var ref = new Object() { int tmpAmount = maxToTurnIn; }; Arrays.stream(inventory.getContents()) .filter(Objects::nonNull) .filter(itemStack -> itemStack.getType().equals(collectDropsQuestObject.getMaterial())) .forEach(itemStack -> { if (ref.tmpAmount == 0) return; if (itemStack.getAmount() > ref.tmpAmount) { itemStack.setAmount(itemStack.getAmount() - ref.tmpAmount); ref.tmpAmount = 0; } else { ref.tmpAmount -= itemStack.getAmount(); itemStack.setAmount(0); } }); int totalTurnedIn = maxToTurnIn - ref.tmpAmount; addStep2(totalTurnedIn); checkDone(player); return totalTurnedIn; } @Override public Component getDisplayName() { return MiniMessage.miniMessage().deserialize(QuestsConfig.COLLECT_DROPS_QUEST_NAME); } public void collectDrops(List drops) { if (isDone()) return; int total = drops.stream() .filter(itemStack -> itemStack.getType().equals(collectDropsQuestObject.getMaterial())) .mapToInt(ItemStack::getAmount) .sum(); if (total == 0) return; addStep1(total); checkDone(); } }