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)), false); 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, boolean rewardReceived) { super(uuid, step1, step2, QuestsConfig.COLLECT_DROPS_QUEST.stream() .filter(object -> variant.equals(object.getInternalName())) .findAny().orElse(null), rewardReceived); 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, reward_received) " + "VALUES (?, ?, ?, ?, ?, ?, ?) " + "ON DUPLICATE KEY UPDATE " + "quest = ?, quest_variant = ?, step_1_progress = ?, step_2_progress = ?, year_day = ?, reward_received = ?"; 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.setInt(7, isRewardReceived() ? 1 : 0); statement.setString(8, this.getClass().getSimpleName()); statement.setString(9, collectDropsQuestObject.getInternalName()); statement.setInt(10, getStep1()); statement.setInt(11, getStep2()); statement.setInt(12, yearDay); statement.setInt(13, isRewardReceived() ? 1 : 0); statement.execute(); } catch (SQLException exception) { exception.printStackTrace(); } } @Override public TagResolver getTagResolvers() { TagResolver resolver = TagResolver.resolver( Placeholder.unparsed("item", Utilities.formatName(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())), Placeholder.unparsed("step_1", QuestsConfig.COLLECT_DROPS_STEP_1), Placeholder.unparsed("step_2", QuestsConfig.COLLECT_DROPS_STEP_2), Placeholder.unparsed("turn_in_text", QuestsConfig.COLLECT_DROPS_TURN_IN) ); Component turnInText = MiniMessage.miniMessage().deserialize(QuestsConfig.COLLECT_DROPS_TURN_IN, resolver); return TagResolver.resolver( resolver, Placeholder.component("turn_in_text", turnInText) ); } @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); } @Override public List getRewardCommand() { return QuestsConfig.COLLECT_DROPS_COMMANDS; } 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(); } }