PlotSquared/src/main/java/com/intellectualcrafters/plot/commands/Rate.java

////////////////////////////////////////////////////////////////////////////////////////////////////
// PlotSquared - A plot manager and world generator for the Bukkit API                             /
// Copyright (c) 2014 IntellectualSites/IntellectualCrafters                                       /
//                                                                                                 /
// 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, write to the Free Software Foundation,                         /
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA                               /
//                                                                                                 /
// You can contact us via: support@intellectualsites.com                                           /
////////////////////////////////////////////////////////////////////////////////////////////////////
package com.intellectualcrafters.plot.commands;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map.Entry;
import java.util.UUID;

import org.apache.commons.lang.mutable.MutableInt;

import com.intellectualcrafters.plot.PS;
import com.intellectualcrafters.plot.config.C;
import com.intellectualcrafters.plot.config.Settings;
import com.intellectualcrafters.plot.database.DBFunc;
import com.intellectualcrafters.plot.object.Location;
import com.intellectualcrafters.plot.object.Plot;
import com.intellectualcrafters.plot.object.PlotInventory;
import com.intellectualcrafters.plot.object.PlotItemStack;
import com.intellectualcrafters.plot.object.PlotPlayer;
import com.intellectualcrafters.plot.object.Rating;
import com.intellectualcrafters.plot.util.EventUtil;
import com.intellectualcrafters.plot.util.MainUtil;
import com.intellectualcrafters.plot.util.MathMan;
import com.intellectualcrafters.plot.util.Permissions;
import com.intellectualcrafters.plot.util.TaskManager;
import com.plotsquared.general.commands.Command;
import com.plotsquared.general.commands.CommandDeclaration;

@CommandDeclaration(
command = "rate",
permission = "plots.rate",
description = "Rate the plot",
usage = "/plot rate [#|next]",
aliases = { "rt" },
category = CommandCategory.ACTIONS,
requiredType = RequiredType.NONE)
public class Rate extends SubCommand {
    
    @Override
    public boolean onCommand(final PlotPlayer player, final String[] args) {
        if (args.length == 1) {
            if (args[0].equalsIgnoreCase("next")) {
                final ArrayList<Plot> plots = new ArrayList<>(PS.get().getBasePlots());
                Collections.sort(plots, new Comparator<Plot>() {
                    @Override
                    public int compare(final Plot p1, final Plot p2) {
                        double v1 = 0;
                        double v2 = 0;
                        if (p1.getSettings().ratings != null) {
                            for (final Entry<UUID, Rating> entry : p1.getRatings().entrySet()) {
                                v1 -= 11 - entry.getValue().getAverageRating();
                            }
                        }
                        if (p2.getSettings().ratings != null) {
                            for (final Entry<UUID, Rating> entry : p2.getRatings().entrySet()) {
                                v2 -= 11 - entry.getValue().getAverageRating();
                            }
                        }
                        return v2 > v1 ? 1 : -1;
                    }
                });
                final UUID uuid = player.getUUID();
                for (final Plot p : plots) {
                    if ((!Settings.REQUIRE_DONE || p.getFlags().containsKey("done"))
                    && p.isBasePlot()
                    && ((p.getSettings().ratings == null) || !p.getSettings().ratings.containsKey(uuid))
                    && !p.isAdded(uuid)) {
                        MainUtil.teleportPlayer(player, player.getLocation(), p);
                        MainUtil.sendMessage(player, C.RATE_THIS);
                        return true;
                    }
                }
                MainUtil.sendMessage(player, C.FOUND_NO_PLOTS);
                return false;
            }
        }
        final Location loc = player.getLocation();
        final Plot plot = MainUtil.getPlot(loc);
        if (plot == null) {
            return !sendMessage(player, C.NOT_IN_PLOT);
        }
        if (!plot.hasOwner()) {
            sendMessage(player, C.RATING_NOT_OWNED);
            return false;
        }
        if (plot.isOwner(player.getUUID())) {
            sendMessage(player, C.RATING_NOT_YOUR_OWN);
            return false;
        }
        if (Settings.REQUIRE_DONE && !plot.getFlags().containsKey("done")) {
            sendMessage(player, C.RATING_NOT_DONE);
            return false;
        }
        if ((Settings.RATING_CATEGORIES != null) && (Settings.RATING_CATEGORIES.size() != 0)) {
            final Runnable run = new Runnable() {
                @Override
                public void run() {
                    if (plot.getSettings().ratings.containsKey(player.getUUID())) {
                        sendMessage(player, C.RATING_ALREADY_EXISTS, plot.getId().toString());
                        return;
                    }
                    final MutableInt index = new MutableInt(0);
                    final MutableInt rating = new MutableInt(0);
                    final String title = Settings.RATING_CATEGORIES.get(0);
                    final PlotInventory inventory = new PlotInventory(player, 1, title) {
                        @Override
                        public boolean onClick(final int i) {
                            rating.add((i + 1) * Math.pow(10, index.intValue()));
                            index.increment();
                            if (index.intValue() >= Settings.RATING_CATEGORIES.size()) {
                                close();
                                final int rV = rating.intValue();
                                final Rating result = EventUtil.manager.callRating(player, plot, new Rating(rV));
                                plot.getSettings().ratings.put(player.getUUID(), result.getAggregate());
                                DBFunc.setRating(plot, player.getUUID(), rV);
                                sendMessage(player, C.RATING_APPLIED, plot.getId().toString());
                                sendMessage(player, C.RATING_APPLIED, plot.getId().toString());
                                return false;
                            }
                            setTitle(Settings.RATING_CATEGORIES.get(index.intValue()));
                            if (Permissions.hasPermission(player, "plots.comment")) {
                                final Command<PlotPlayer> command = MainCommand.getInstance().getCommand("comment");
                                if (command != null) {
                                    MainUtil.sendMessage(player, C.COMMENT_THIS, command.getUsage().replaceAll("{label}", "plot"));
                                }
                            }
                            return false;
                        }
                    };
                    inventory.setItem(0, new PlotItemStack(35, (short) 12, 0, "0/8"));
                    inventory.setItem(1, new PlotItemStack(35, (short) 14, 1, "1/8"));
                    inventory.setItem(2, new PlotItemStack(35, (short) 1, 2, "2/8"));
                    inventory.setItem(3, new PlotItemStack(35, (short) 4, 3, "3/8"));
                    inventory.setItem(4, new PlotItemStack(35, (short) 5, 4, "4/8"));
                    inventory.setItem(5, new PlotItemStack(35, (short) 9, 5, "5/8"));
                    inventory.setItem(6, new PlotItemStack(35, (short) 11, 6, "6/8"));
                    inventory.setItem(7, new PlotItemStack(35, (short) 10, 7, "7/8"));
                    inventory.setItem(8, new PlotItemStack(35, (short) 2, 8, "8/8"));
                    inventory.openInventory();
                }
            };
            if (plot.getSettings().ratings == null) {
                if (!Settings.CACHE_RATINGS) {
                    TaskManager.runTaskAsync(new Runnable() {
                        @Override
                        public void run() {
                            plot.getSettings().ratings = DBFunc.getRatings(plot);
                            run.run();
                        }
                    });
                    return true;
                }
                plot.getSettings().ratings = new HashMap<UUID, Integer>();
            }
            run.run();
            return true;
        }
        if (args.length < 1) {
            sendMessage(player, C.RATING_NOT_VALID);
            return true;
        }
        final String arg = args[0];
        final int rating;
        if (MathMan.isInteger(arg) && (arg.length() < 3) && (arg.length() > 0)) {
            rating = Integer.parseInt(arg);
            if (rating > 10) {
                sendMessage(player, C.RATING_NOT_VALID);
                return false;
            }
        } else {
            sendMessage(player, C.RATING_NOT_VALID);
            return false;
        }
        final UUID uuid = player.getUUID();
        final Runnable run = new Runnable() {
            @Override
            public void run() {
                if (plot.getSettings().ratings.containsKey(uuid)) {
                    sendMessage(player, C.RATING_ALREADY_EXISTS, plot.getId().toString());
                    return;
                }
                final Rating result = EventUtil.manager.callRating(player, plot, new Rating(rating));
                final int resultVal = result.getAggregate();
                plot.getSettings().ratings.put(uuid, resultVal);
                DBFunc.setRating(plot, uuid, resultVal);
                sendMessage(player, C.RATING_APPLIED, plot.getId().toString());
            }
        };
        if (plot.getSettings().ratings == null) {
            if (!Settings.CACHE_RATINGS) {
                TaskManager.runTaskAsync(new Runnable() {
                    @Override
                    public void run() {
                        plot.getSettings().ratings = DBFunc.getRatings(plot);
                        run.run();
                    }
                });
                return true;
            }
            plot.getSettings().ratings = new HashMap<UUID, Integer>();
        }
        run.run();
        return true;
    }
}