mcMMO/src/main/java/com/gmail/nossr50/skills/acrobatics/AcrobaticsManager.java

package com.gmail.nossr50.skills.acrobatics;

import com.gmail.nossr50.config.experience.ExperienceConfig;
import com.gmail.nossr50.datatypes.BlockLocationHistory;
import com.gmail.nossr50.datatypes.experience.XPGainReason;
import com.gmail.nossr50.datatypes.experience.XPGainSource;
import com.gmail.nossr50.datatypes.interactions.NotificationType;
import com.gmail.nossr50.datatypes.player.McMMOPlayer;
import com.gmail.nossr50.datatypes.skills.PrimarySkillType;
import com.gmail.nossr50.datatypes.skills.SubSkillType;
import com.gmail.nossr50.mcMMO;
import com.gmail.nossr50.runnables.MobDodgeMetaCleanup;
import com.gmail.nossr50.skills.SkillManager;
import com.gmail.nossr50.util.MetadataConstants;
import com.gmail.nossr50.util.Misc;
import com.gmail.nossr50.util.Permissions;
import com.gmail.nossr50.util.player.NotificationManager;
import com.gmail.nossr50.util.player.UserManager;
import com.gmail.nossr50.util.random.ProbabilityUtil;
import com.gmail.nossr50.util.skills.ParticleEffectUtils;
import com.gmail.nossr50.util.skills.RankUtils;
import com.gmail.nossr50.util.skills.SkillUtils;
import com.tcoded.folialib.wrapper.task.WrappedTask;
import java.util.function.Consumer;
import org.bukkit.Location;
import org.bukkit.entity.Entity;
import org.bukkit.entity.LightningStrike;
import org.bukkit.entity.Mob;
import org.bukkit.entity.Player;
import org.bukkit.metadata.FixedMetadataValue;
import org.bukkit.metadata.MetadataValue;

public class AcrobaticsManager extends SkillManager {

    public AcrobaticsManager(McMMOPlayer mmoPlayer) {
        super(mmoPlayer, PrimarySkillType.ACROBATICS);
        fallLocationMap = new BlockLocationHistory(50);
    }

    private long rollXPCooldown = 0;
    private final long rollXPInterval = (1000 * 3); //1 Minute
    private long rollXPIntervalLengthen = (1000 * 10); //10 Seconds
    private final BlockLocationHistory fallLocationMap;

    public boolean hasFallenInLocationBefore(Location location) {
        return fallLocationMap.contains(location);
    }

    public void addLocationToFallMap(Location location) {
        fallLocationMap.add(location);
    }

    public boolean canGainRollXP() {
        if (!ExperienceConfig.getInstance().isAcrobaticsExploitingPrevented()) {
            return true;
        }

        if (System.currentTimeMillis() >= rollXPCooldown) {
            rollXPCooldown = System.currentTimeMillis() + rollXPInterval;
            rollXPIntervalLengthen = (1000 * 10); //5 Seconds
            return true;
        } else {
            rollXPCooldown += rollXPIntervalLengthen;
            rollXPIntervalLengthen += 1000; //Add another second to the next penalty
            return false;
        }
    }

    public boolean canDodge(Entity damager) {
        if (getPlayer().isBlocking()) {
            return false;
        }

        if (!RankUtils.hasUnlockedSubskill(getPlayer(), SubSkillType.ACROBATICS_DODGE)) {
            return false;
        }

        if (Permissions.isSubSkillEnabled(getPlayer(), SubSkillType.ACROBATICS_DODGE)) {
            if (damager instanceof LightningStrike && Acrobatics.dodgeLightningDisabled) {
                return false;
            }

            return mcMMO.p.getSkillTools().canCombatSkillsTrigger(skill, damager);
        }

        return false;
    }

    /**
     * Handle the damage reduction and XP gain from the Dodge ability
     *
     * @param damage The amount of damage initially dealt by the event
     * @return the modified event damage if the ability was successful, the original event damage
     * otherwise
     */
    public double dodgeCheck(Entity attacker, double damage) {
        double modifiedDamage = Acrobatics.calculateModifiedDodgeDamage(damage,
                Acrobatics.dodgeDamageModifier);
        Player player = getPlayer();

        if (!isFatal(modifiedDamage)
                && ProbabilityUtil.isSkillRNGSuccessful(SubSkillType.ACROBATICS_DODGE,
                UserManager.getPlayer(player))) {
            ParticleEffectUtils.playDodgeEffect(player);

            if (mmoPlayer.useChatNotifications()) {
                NotificationManager.sendPlayerInformation(player, NotificationType.SUBSKILL_MESSAGE,
                        "Acrobatics.Combat.Proc");
            }

            if (SkillUtils.cooldownExpired(mmoPlayer.getRespawnATS(),
                    Misc.PLAYER_RESPAWN_COOLDOWN_SECONDS)) {
                if (attacker instanceof Mob mob) {
                    //Check to see how many dodge XP rewards this mob has handed out
                    if (mob.hasMetadata(MetadataConstants.METADATA_KEY_DODGE_TRACKER)
                            && ExperienceConfig.getInstance().isAcrobaticsExploitingPrevented()) {
                        //If Dodge XP has been handed out 5 times then consider it being exploited
                        MetadataValue metadataValue = mob.getMetadata(
                                MetadataConstants.METADATA_KEY_DODGE_TRACKER).get(0);
                        int count = metadataValue.asInt();

                        if (count <= 5) {
                            applyXpGain((float) (damage * Acrobatics.dodgeXpModifier),
                                    XPGainReason.PVE, XPGainSource.SELF);
                            mob.setMetadata(MetadataConstants.METADATA_KEY_DODGE_TRACKER,
                                    new FixedMetadataValue(mcMMO.p, count + 1));
                            final Consumer<WrappedTask> metaCleanupTask = new MobDodgeMetaCleanup(mob,
                                    mcMMO.p);
                            mcMMO.p.getFoliaLib().getScheduler()
                                    .runAtEntityTimer(mob, metaCleanupTask, 20,
                                            20 * 60); //one minute
                        }
                    } else {
                        applyXpGain((float) (damage * Acrobatics.dodgeXpModifier),
                                XPGainReason.PVE, XPGainSource.SELF);
                    }
                }
            }

            //Check respawn to prevent abuse
            return modifiedDamage;
        }

        return damage;
    }

    private boolean isFatal(double damage) {
        return getPlayer().getHealth() - damage <= 0;
    }
}