EssentialsExpansion.java

/*
 *
 * Essentials-Expansion
 * Copyright (C) 2019 Ryan McCarthy
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 *
 */
package com.extendedclip.papi.expansion.essentials;

import com.earth2me.essentials.Essentials;
import com.earth2me.essentials.Kit;
import com.earth2me.essentials.User;
import com.earth2me.essentials.utils.DateUtil;
import com.earth2me.essentials.utils.DescParseTickFormat;
import com.google.common.primitives.Ints;
import me.clip.placeholderapi.PlaceholderAPIPlugin;
import me.clip.placeholderapi.expansion.PlaceholderExpansion;
import org.bukkit.*;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemStack;
import org.jetbrains.annotations.NotNull;

import java.math.BigDecimal;
import java.text.DateFormat;
import java.text.NumberFormat;
import java.time.Instant;
import java.time.temporal.ChronoUnit;
import java.util.Date;
import java.util.stream.StreamSupport;

public class EssentialsExpansion extends PlaceholderExpansion {

    private Essentials essentials;

    private final String VERSION = getClass().getPackage().getImplementationVersion();

    @Override
    public boolean canRegister() {
        return Bukkit.getPluginManager().getPlugin("Essentials") != null;
    }

    @Override
    public boolean register() {
        essentials = (Essentials) Bukkit.getPluginManager().getPlugin("Essentials");
        if (essentials != null) {
            return super.register();
        }
        return false;
    }

    @Override
    public @NotNull String getAuthor() {
        return "clip";
    }

    @Override
    public @NotNull String getIdentifier() {
        return "essentials";
    }

    @Override
    public @NotNull String getVersion() {
        return VERSION;
    }

    @Override
    public String onRequest(OfflinePlayer player, @NotNull String identifier) {
        final String papiTrue = PlaceholderAPIPlugin.booleanTrue();
        final String papiFalse = PlaceholderAPIPlugin.booleanFalse();

        if (player == null) return "";

        if (identifier.startsWith("kit_last_use_")) {
            String kitName = identifier.split("kit_last_use_")[1].toLowerCase();
            Kit kit;

            try {
                kit = new Kit(kitName, essentials);
            } catch (Exception e) {
                return "Invalid kit name";
            }

            long time = essentials.getUser(player.getUniqueId()).getKitTimestamp(kit.getName());

            if (time == 1 || time <= 0) {
                return "1";
            }
            return PlaceholderAPIPlugin.getDateFormat().format(new Date(time));
        }

        if (identifier.startsWith("kit_is_available_")) {
            String kitName = identifier.split("kit_is_available_")[1].toLowerCase();
            Kit kit;
            User user = essentials.getUser(player.getUniqueId());
            long time;

            try {
                kit = new Kit(kitName, essentials);
            } catch (Exception e) {
                return "Invalid kit name";
            }

            try {
                time = kit.getNextUse(user);
            } catch (Exception e) {
                return papiFalse;
            }

            return time == 0 ? papiTrue : papiFalse;
        }

        if (identifier.startsWith("kit_time_until_available_")) {
            String kitName = identifier.split("kit_time_until_available_")[1].toLowerCase();
            boolean raw = false;
            User user = essentials.getUser(player.getUniqueId());
            Kit kit;
            long time;

            if (kitName.startsWith("raw_")) {
                raw = true;
                kitName = kitName.substring(4);

                if (kitName.isEmpty()) {
                    return "Invalid kit name";
                }
            }

            try {
                kit = new Kit(kitName, essentials);
            } catch (Exception e) {
                return "Invalid kit name";
            }

            try {
                time = kit.getNextUse(user);
            } catch (Exception e) {
                return "-1";
            }

            if (time <= System.currentTimeMillis()) {
                return raw ? "0" : DateUtil.formatDateDiff(System.currentTimeMillis());
            }

            if (raw) {
                return String.valueOf(Instant.now().until(Instant.ofEpochMilli(time), ChronoUnit.MILLIS));
            } else {
                return DateUtil.formatDateDiff(time);
            }
        }

        if (identifier.startsWith("has_kit_")) {
            Player oPlayer = player.getPlayer();
            if (oPlayer == null) return papiFalse;

            String kit = identifier.split("has_kit_")[1];
            return oPlayer.hasPermission("essentials.kits." + kit) ? papiTrue : papiFalse;
        }

        if (identifier.startsWith("home_")) {
            Integer homeNumber;
            final User user = essentials.getUser(player.getUniqueId());

            // Removes all the letters from the identifier to get the home slot.
            // Checks if the number slot is an integer or not.
            if ((homeNumber = Ints.tryParse(identifier.replaceAll("\\D+", ""))) == null) return null;

            // Since it is easier for users to type from 1-x I subtract one from the original number to work from 0-x.
            homeNumber -= 1;

            // checks if the home is out of bounds and returns and empty string if it is.
            if (homeNumber >= user.getHomes().size() || homeNumber < 0) return "";

            // checks if the identifier matches the pattern home_%d
            if (identifier.matches("(\\w+_)(\\d)")) return user.getHomes().get(homeNumber);

            //checks if the identifier matches the pattern home_%d_(x/y/z)
            if (identifier.matches("(\\w+_)(\\d)(_\\w)")) {

                try {
                    final Location home = user.getHome(user.getHomes().get(homeNumber));
                    final StringBuilder stringBuilder = new StringBuilder();

                    switch (identifier.charAt(identifier.length() - 1)) {
                        case 'x':
                            stringBuilder.append(home.getX()).append(".5");
                            break;
                        case 'y':
                            stringBuilder.append((int) home.getY());
                            break;
                        case 'z':
                            stringBuilder.append(home.getZ()).append(".5");
                            break;
                    }

                    return stringBuilder.toString();
                } catch (Exception e) {
                    return null;
                }
            }
        }

        if (identifier.startsWith("worth")) {
            ItemStack item;

            if (identifier.contains(":")){
                Material material = Material.getMaterial(identifier.replace("worth:", "").toUpperCase());

                if (material == null) return "";
                item = new ItemStack(material,1);
            } else {
                Player oPlayer = player.getPlayer();
                if (oPlayer == null) return "";

                if (oPlayer.getInventory().getItemInMainHand().getType() == Material.AIR) return "";
                item = oPlayer.getInventory().getItemInMainHand();
            }

            BigDecimal worth = essentials.getWorth().getPrice(essentials, item);
            if (worth == null) return "";
            return String.valueOf(worth.doubleValue());
        }

        final User user = essentials.getUser(player.getUniqueId());

        switch (identifier) {
            case "is_pay_confirm":
                return user.isPromptingPayConfirm() ? papiTrue : papiFalse;
            case "is_pay_enabled":
                return user.isAcceptingPay() ? papiTrue : papiFalse;
            case "is_teleport_enabled":
                return user.isTeleportEnabled() ? papiTrue : papiFalse;
            case "is_muted":
                return user.isMuted() ? papiTrue : papiFalse;
            case "vanished":
                return user.isVanished() ? papiTrue : papiFalse;
            case "afk":
                return user.isAfk() ? papiTrue : papiFalse;
            case "afk_reason":
                if (user.getAfkMessage() == null) return "";
                return ChatColor.translateAlternateColorCodes('&', user.getAfkMessage());
            case "afk_player_count":
                return String.valueOf(essentials.getUserMap().getAllUniqueUsers().stream()
                        .map(UUID -> essentials.getUser(UUID)).filter(User::isAfk)
                        .count());
            case "msg_ignore":
                return user.isIgnoreMsg() ? papiTrue : papiFalse;
            case "fly":
                return user.getBase().getAllowFlight() ? papiTrue : papiFalse;
            case "nickname":
                return user.getNickname() != null ? essentials.getUser(player.getUniqueId()).getNickname() : player.getName();
            case "nickname_stripped":
                return ChatColor.stripColor(user.getNickname() != null ? essentials.getUser(player.getUniqueId()).getNickname() : player.getName());
            case "nickname_hex_stripped":
                return user.getNickname() != null ? user.getNickname().replaceAll("§x§[0-9a-f]§[0-9a-f]§[0-9a-f]§[0-9a-f]§[0-9a-f]§[0-9a-f]", "") : player.getName();
            case "godmode":
                return user.isGodModeEnabled() ? papiTrue : papiFalse;
            case "unique":
                return NumberFormat.getInstance().format(essentials.getUserMap().getUniqueUsers());
            case "homes_set":
                return user.getHomes().isEmpty() ? "0" : String.valueOf(user.getHomes().size());
            case "homes_max":
                return String.valueOf(essentials.getSettings().getHomeLimit(user));
            case "jailed":
                return user.isJailed() ? papiTrue : papiFalse;
            case "jailed_time_remaining":
                return user.getFormattedJailTime();
            case "pm_recipient":
                return user.getReplyRecipient() != null ? user.getReplyRecipient().getName() : "";
            case "safe_online":
                return String.valueOf(StreamSupport.stream(essentials.getOnlineUsers().spliterator(), false)
                        .filter(user1 -> !user1.isHidden())
                        .count());
            case "world_date":
                return DateFormat.getDateInstance(DateFormat.MEDIUM, essentials.getI18n().getCurrentLocale())
                        .format(DescParseTickFormat.ticksToDate(user.getWorld() == null ? 0 : user.getWorld().getFullTime()));
            case "world_time":
                return DescParseTickFormat.format12(user.getWorld() == null ? 0 : user.getWorld().getTime());
            case "world_time_24":
                return DescParseTickFormat.format24(user.getWorld() == null ? 0 : user.getWorld().getTime());
        }
        return null;
    }
}