Essentials/Essentials/src/main/java/com/earth2me/essentials/commands/Commandpweather.java

122 lines
4.7 KiB
Java

package com.earth2me.essentials.commands;
import com.earth2me.essentials.CommandSource;
import com.earth2me.essentials.IUser;
import com.earth2me.essentials.User;
import com.google.common.collect.Lists;
import org.bukkit.Server;
import org.bukkit.WeatherType;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.StringJoiner;
import static com.earth2me.essentials.I18n.tl;
public class Commandpweather extends EssentialsLoopCommand {
private static final List<String> getAliases = Arrays.asList("get", "list", "show", "display");
private static final Map<String, WeatherType> weatherAliases = new HashMap<>();
static {
weatherAliases.put("sun", WeatherType.CLEAR);
weatherAliases.put("clear", WeatherType.CLEAR);
weatherAliases.put("storm", WeatherType.DOWNFALL);
weatherAliases.put("thunder", WeatherType.DOWNFALL);
}
public Commandpweather() {
super("pweather");
}
@Override
public void run(final Server server, final CommandSource sender, final String commandLabel, final String[] args) throws Exception {
if (args.length == 0 || getAliases.contains(args[0].toLowerCase())) {
if (args.length > 1) { // /pweather get md_5 || /pweather get *
if (args[1].equals("*") || args[1].equals("**")) {
sender.sendMessage(tl("pWeatherPlayers"));
}
loopOnlinePlayersConsumer(server, sender, false, true, args[1], player -> getUserWeather(sender, player));
return;
}
if (args.length == 1 || sender.isPlayer()) { // /pweather get
if (sender.isPlayer()) {
getUserWeather(sender, sender.getUser(ess));
return;
}
throw new NotEnoughArgumentsException(); // We cannot imply the target for console
}
// Default to showing the weather of all online users for console when no arguments are provided
if (ess.getOnlinePlayers().size() > 1) {
sender.sendMessage(tl("pWeatherPlayers"));
}
for (final User player : ess.getOnlineUsers()) {
getUserWeather(sender, player);
}
}
if (args.length > 1 && !sender.isAuthorized("essentials.pweather.others", ess) && !args[1].equalsIgnoreCase(sender.getSelfSelector())) {
sender.sendMessage(tl("pWeatherOthersPermission"));
return;
}
final String weather = args[0].toLowerCase();
if (!weatherAliases.containsKey(weather) && !weather.equalsIgnoreCase("reset")) {
throw new NotEnoughArgumentsException(tl("pWeatherInvalidAlias"));
}
final StringJoiner joiner = new StringJoiner(", ");
loopOnlinePlayersConsumer(server, sender, false, true, args.length > 1 ? args[1] : sender.getSelfSelector(), player -> {
setUserWeather(player, weather);
joiner.add(player.getName());
});
if (weather.equalsIgnoreCase("reset")) {
sender.sendMessage(tl("pWeatherReset", joiner.toString()));
return;
}
sender.sendMessage(tl("pWeatherSet", weather, joiner.toString()));
}
private void getUserWeather(final CommandSource sender, final IUser user) {
if (user == null) {
return;
}
if (user.getBase().getPlayerWeather() == null) {
sender.sendMessage(tl("pWeatherNormal", user.getName()));
return;
}
sender.sendMessage(tl("pWeatherCurrent", user.getName(), user.getBase().getPlayerWeather().toString().toLowerCase(Locale.ENGLISH)));
}
private void setUserWeather(final User user, final String weatherType) {
if (weatherType.equalsIgnoreCase("reset")) {
user.getBase().resetPlayerWeather();
return;
}
user.getBase().setPlayerWeather(weatherAliases.get(weatherType));
}
@Override
protected List<String> getTabCompleteOptions(final Server server, final User user, final String commandLabel, final String[] args) {
if (args.length == 1) {
return Lists.newArrayList("get", "reset", "storm", "sun");
} else if (args.length == 2 && (getAliases.contains(args[0]) || user == null || user.isAuthorized("essentials.pweather.others"))) {
return getPlayers(server, user);
} else {
return Collections.emptyList();
}
}
@Override
protected void updatePlayer(final Server server, final CommandSource sender, final User user, final String[] args) {
}
}