/*
* Project: UHC
* Class: gg.uhc.uhc.modules.team.requests.RequestManager
*
* The MIT License (MIT)
*
* Copyright (c) 2015 Graham Howden <graham_howden1 at yahoo.co.uk>.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package gg.uhc.uhc.modules.team.requests;
import gg.uhc.uhc.messages.MessageTemplates;
import gg.uhc.uhc.modules.team.FunctionalUtil;
import gg.uhc.uhc.modules.team.TeamModule;
import com.google.common.base.Joiner;
import com.google.common.base.Optional;
import com.google.common.base.Preconditions;
import com.google.common.base.Predicates;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import com.google.common.collect.Maps;
import net.md_5.bungee.api.ChatColor;
import net.md_5.bungee.api.chat.BaseComponent;
import net.md_5.bungee.api.chat.ClickEvent;
import net.md_5.bungee.api.chat.TextComponent;
import org.bukkit.Bukkit;
import org.bukkit.OfflinePlayer;
import org.bukkit.entity.Player;
import org.bukkit.plugin.Plugin;
import org.bukkit.scheduler.BukkitRunnable;
import org.bukkit.scoreboard.Team;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.UUID;
public class RequestManager {
public enum AcceptState {
ACCEPT,
DENY,
CANCEL
}
public static final String ADMIN_PERMISSION = "uhc.command.teamrequestadmin";
public static final String AUTO_WHITELIST_KEY = "auto whitelist accepted teams";
protected final Map<UUID, TeamRequest> byUUID = Maps.newHashMap();
protected final Map<Integer, TeamRequest> byId = Maps.newHashMap();
protected final Collection<TeamRequest> requests = byUUID.values();
protected final Map<Integer, BukkitRunnable> timers = Maps.newHashMap();
protected final Plugin plugin;
protected final MessageTemplates messages;
protected final TeamModule module;
protected final long autoCancelTime;
protected final boolean autoWhitelistAcceptedTeams;
public RequestManager(Plugin plugin, MessageTemplates messages, TeamModule module,
long autoCancelTime, boolean autoWhitelistAcceptedTeams) {
this.plugin = plugin;
this.messages = messages;
this.module = module;
this.autoWhitelistAcceptedTeams = autoWhitelistAcceptedTeams;
Preconditions.checkArgument(autoCancelTime > 0);
this.autoCancelTime = autoCancelTime;
}
public Optional<TeamRequest> getByUUID(UUID uuid) {
return Optional.fromNullable(byUUID.get(uuid));
}
public Optional<TeamRequest> getById(int id) {
return Optional.fromNullable(byId.get(id));
}
public List<TeamRequest> getRequests() {
return ImmutableList.copyOf(requests);
}
/**
* Removes the request and processes the response.
*
* @param id the id of the request
* @param accepted the state to set
* @return true if id existed, false otherwise
*/
public boolean finalizeRequest(int id, AcceptState accepted) {
final Optional<TeamRequest> optional = removeRequest(id);
if (!optional.isPresent()) return false;
final TeamRequest request = optional.get();
final Map<String, Object> context = ImmutableMap.<String, Object>builder()
.put("name", request.getOwnerName())
.put("id", request.getId())
.put("members", Joiner.on(", ").join(request.getOthers()))
.build();
broadcast(messages.evalTemplate("on." + accepted.name().toLowerCase() + ".broadcast", context));
final Player player = Bukkit.getPlayer(request.getOwner());
if (player != null) {
player.sendMessage(messages.evalTemplate("on." + accepted.name().toLowerCase() + ".notify", context));
}
if (accepted == AcceptState.ACCEPT) {
final Iterable<OfflinePlayer> players = Iterables.filter(
Iterables.transform(
request.getOthers(),
FunctionalUtil.OFFLINE_PLAYER_FROM_NAME
),
Predicates.notNull()
);
final Optional<Team> potentialTeam = module.findFirstEmptyTeam();
if (!potentialTeam.isPresent()) {
broadcast(messages.getRaw("no empty teams"));
return true;
}
final Team team = potentialTeam.get();
for (final OfflinePlayer p : players) {
team.addPlayer(p);
if (autoWhitelistAcceptedTeams) p.setWhitelisted(true);
}
final OfflinePlayer ownerOfflinePlayer = Bukkit.getOfflinePlayer(request.getOwner());
team.addPlayer(ownerOfflinePlayer);
if (autoWhitelistAcceptedTeams) {
ownerOfflinePlayer.setWhitelisted(true);
}
}
// otherwise do nothing
return true;
}
public boolean acceptRequest(int id) {
return finalizeRequest(id, AcceptState.ACCEPT);
}
public boolean denyRequest(int id) {
return finalizeRequest(id, AcceptState.DENY);
}
public boolean cancelRequest(int id) {
return finalizeRequest(id, AcceptState.CANCEL);
}
protected Optional<TeamRequest> removeRequest(int id) {
final TeamRequest request = byId.remove(id);
if (request == null) return Optional.absent();
// cancel the timer and UUID mapping too
timers.remove(id).cancel();
byUUID.remove(request.getOwner());
return Optional.of(request);
}
/**
* Adds a new team request. If a request already existed for the owner, it is cancelled before this one is added.
*
* @param request the request to add
*/
public void addRequest(TeamRequest request) {
final UUID owner = request.getOwner();
final int id = request.getId();
final TeamRequest existing = byUUID.get(owner);
if (existing != null) {
cancelRequest(existing.getId());
}
byUUID.put(owner, request);
byId.put(id, request);
// start timer to auto-remove
final AutoCancelTask task = new AutoCancelTask(id);
task.runTaskLater(plugin, autoCancelTime);
timers.put(id, task);
final Player player = Bukkit.getPlayer(request.getOwner());
if (player != null) {
player.sendMessage(messages.getRaw("added request"));
}
final TextComponent others = new TextComponent(Joiner.on(", ").join(request.getOthers()));
others.setColor(ChatColor.DARK_PURPLE);
final TextComponent accept = new TextComponent("Accept");
accept.setColor(ChatColor.GREEN);
accept.setClickEvent(new ClickEvent(ClickEvent.Action.RUN_COMMAND, "/teamrequest accept " + request.getId()));
final TextComponent deny = new TextComponent("Deny");
deny.setColor(ChatColor.RED);
deny.setClickEvent(new ClickEvent(ClickEvent.Action.RUN_COMMAND, "/teamrequest deny " + request.getId()));
final TextComponent component = new TextComponent(
"New request (ID: " + request.getId() + ") from " + request.getOwnerName() + " to team with: "
);
component.setColor(ChatColor.GRAY);
component.addExtra(others);
component.addExtra(" ");
component.addExtra(accept);
component.addExtra(" | ");
component.addExtra(deny);
broadcast(component);
}
protected void broadcast(BaseComponent... components) {
for (final Player player : Bukkit.getOnlinePlayers()) {
if (player.hasPermission(ADMIN_PERMISSION)) {
player.spigot().sendMessage(components);
}
}
}
protected void broadcast(String... broadcastMessages) {
for (final Player player : Bukkit.getOnlinePlayers()) {
if (player.hasPermission(ADMIN_PERMISSION)) {
player.sendMessage(broadcastMessages);
}
}
}
protected class AutoCancelTask extends BukkitRunnable {
protected final int id;
public AutoCancelTask(int id) {
this.id = id;
}
@Override
public void run() {
cancelRequest(id);
}
}
}
