From 9e974ae73775c7388ab43f73043a82677755ea59 Mon Sep 17 00:00:00 2001
From: james <75051061+notsceptor@users.noreply.github.com>
Date: Tue, 4 Jun 2024 15:23:51 +0100
Subject: [PATCH] Add a /tempmute command (#96)

* Update WhoHasCMD.java

Added functionality that clears all players with a specified item

* Added functionality that clears all players with a specified item

* Altered tab completion to include "clear" argument

* Update WhoHasCMD.java

* Implemented a mob limiter that can be configured via a command

* Fixed some imports

* Refined tab complete

* Implemented reviewed changes

* Added a /tempmute command

* Removed unnecessary imports & simplified the reason assignment

* Fixed the usage for the moblimit command

---------

Co-authored-by: Telesphoreo <me@telesphoreo.me>
---
 .../dev/plex/command/impl/MobLimitCMD.java    |  2 +-
 .../dev/plex/command/impl/TempmuteCMD.java    | 92 +++++++++++++++++++
 .../main/java/dev/plex/util/TimeUtils.java    | 38 ++++++++
 server/src/main/resources/messages.yml        |  9 +-
 4 files changed, 139 insertions(+), 2 deletions(-)
 create mode 100644 server/src/main/java/dev/plex/command/impl/TempmuteCMD.java

diff --git a/server/src/main/java/dev/plex/command/impl/MobLimitCMD.java b/server/src/main/java/dev/plex/command/impl/MobLimitCMD.java
index 66bbfe6..70dfcaa 100644
--- a/server/src/main/java/dev/plex/command/impl/MobLimitCMD.java
+++ b/server/src/main/java/dev/plex/command/impl/MobLimitCMD.java
@@ -18,7 +18,7 @@ import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 
-@CommandParameters(name = "moblimit", usage = "/<command> [on/off/setmax <limit>]", aliases = "entitylimit", description = "Manages the mob limit per chunk.")
+@CommandParameters(name = "moblimit", usage = "/<command> [on | off | setmax <limit>]", aliases = "entitylimit", description = "Manages the mob limit per chunk.")
 @CommandPermissions(permission = "plex.moblimit", source = RequiredCommandSource.ANY)
 public class MobLimitCMD extends PlexCommand
 {
diff --git a/server/src/main/java/dev/plex/command/impl/TempmuteCMD.java b/server/src/main/java/dev/plex/command/impl/TempmuteCMD.java
new file mode 100644
index 0000000..12004c8
--- /dev/null
+++ b/server/src/main/java/dev/plex/command/impl/TempmuteCMD.java
@@ -0,0 +1,92 @@
+package dev.plex.command.impl;
+
+import com.google.common.collect.ImmutableList;
+import dev.plex.command.PlexCommand;
+import dev.plex.command.annotation.CommandParameters;
+import dev.plex.command.annotation.CommandPermissions;
+import dev.plex.player.PlexPlayer;
+import dev.plex.punishment.Punishment;
+import dev.plex.punishment.PunishmentType;
+import dev.plex.util.PlexUtils;
+import dev.plex.util.TimeUtils; // Import your TimeUtils
+import net.kyori.adventure.text.Component;
+import org.bukkit.command.CommandSender;
+import org.bukkit.entity.Player;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.time.ZonedDateTime;
+import java.util.Arrays;
+import java.util.List;
+
+@CommandParameters(name = "tempmute", description = "Temporarily mute a player on the server",
+        usage = "/<command> <player> <time> [reason]", aliases = "tmute")
+@CommandPermissions(permission = "plex.tempmute")
+public class TempmuteCMD extends PlexCommand
+{
+    @Override
+    protected Component execute(@NotNull CommandSender sender, @Nullable Player playerSender, String[] args)
+    {
+        if (args.length < 2)
+        {
+            return usage();
+        }
+
+        Player player = getNonNullPlayer(args[0]);
+        PlexPlayer punishedPlayer = getOfflinePlexPlayer(player.getUniqueId());
+
+        if (punishedPlayer.isMuted())
+        {
+            return messageComponent("playerMuted");
+        }
+
+        if (silentCheckPermission(player, "plex.tempmute"))
+        {
+            send(sender, messageComponent("higherRankThanYou"));
+            return null;
+        }
+
+        ZonedDateTime endDate;
+        try
+        {
+            endDate = TimeUtils.createDate(args[1]);
+        }
+        catch (NumberFormatException e)
+        {
+            return messageComponent("invalidTimeFormat");
+        }
+
+        if (endDate.isBefore(ZonedDateTime.now()))
+        {
+            return messageComponent("timeMustBeFuture");
+        }
+
+        ZonedDateTime oneWeekFromNow = ZonedDateTime.now().plusWeeks(1);
+        if (endDate.isAfter(oneWeekFromNow))
+        {
+            return messageComponent("maxTimeExceeded");
+        }
+
+        final String reason = args.length >= 3 ? String.join(" ", Arrays.copyOfRange(args, 2, args.length))
+                : messageString("noReasonProvided");
+
+        Punishment punishment = new Punishment(punishedPlayer.getUuid(), getUUID(sender));
+        punishment.setCustomTime(true);
+        punishment.setEndDate(endDate);
+        punishment.setType(PunishmentType.MUTE);
+        punishment.setPunishedUsername(player.getName());
+        punishment.setIp(player.getAddress().getAddress().getHostAddress().trim());
+        punishment.setReason(reason);
+        punishment.setActive(true);
+
+        plugin.getPunishmentManager().punish(punishedPlayer, punishment);
+        PlexUtils.broadcast(messageComponent("tempMutedPlayer", sender.getName(), player.getName(), TimeUtils.formatRelativeTime(endDate)));
+        return null;
+    }
+
+    @Override
+    public @NotNull List<String> smartTabComplete(@NotNull CommandSender sender, @NotNull String alias, @NotNull String[] args) throws IllegalArgumentException
+    {
+        return args.length == 1 && silentCheckPermission(sender, this.getPermission()) ? PlexUtils.getPlayerNameList() : ImmutableList.of();
+    }
+}
diff --git a/server/src/main/java/dev/plex/util/TimeUtils.java b/server/src/main/java/dev/plex/util/TimeUtils.java
index 220efb2..10ff962 100644
--- a/server/src/main/java/dev/plex/util/TimeUtils.java
+++ b/server/src/main/java/dev/plex/util/TimeUtils.java
@@ -7,6 +7,7 @@ import java.time.LocalDateTime;
 import java.time.ZoneId;
 import java.time.ZonedDateTime;
 import java.time.format.DateTimeFormatter;
+import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Set;
@@ -79,4 +80,41 @@ public class TimeUtils
         }
         return DATE_FORMAT.withZone(ZoneId.of(TIMEZONE)).format(date);
     }
+
+    public static String formatRelativeTime(ZonedDateTime date)
+    {
+        long seconds = ChronoUnit.SECONDS.between(ZonedDateTime.now(), date);
+
+        if (seconds <= 0)
+        {
+            return "now";
+        }
+
+        long minute = seconds / 60;
+        long hour = minute / 60;
+        long day = hour / 24;
+        long week = day / 7;
+
+        if (week > 0)
+        {
+            return week + " week" + (week > 1 ? "s" : "");
+        }
+        else if (day > 0)
+        {
+            return day + " day" + (day > 1 ? "s" : "");
+        }
+        else if (hour > 0)
+        {
+            return hour + " hour" + (hour > 1 ? "s" : "");
+        }
+        else if (minute > 0)
+        {
+            return minute + " minute" + (minute > 1 ? "s" : "");
+        }
+        else
+        {
+            return seconds + " second" + (seconds > 1 ? "s" : "");
+        }
+    }
+
 }
diff --git a/server/src/main/resources/messages.yml b/server/src/main/resources/messages.yml
index fa5114f..38cd95c 100644
--- a/server/src/main/resources/messages.yml
+++ b/server/src/main/resources/messages.yml
@@ -43,6 +43,13 @@ mutedPlayer: "<red>{0} - Muted {1}"
 # 0 - The command sender
 # 1 - The person who has been unmuted
 unmutedPlayer: "<aqua>{0} - Unmuted {1}"
+invalidTimeFormat: "<red>Invalid time format. Use s, m, h, d, w, mo, or y (e.g., 1h30m)."
+timeMustBeFuture: "<red>The specified time must be in the future."
+# 0 - The command sender
+# 1 - The person who has been muted
+# 2 - The time that the person is muted for
+tempMutedPlayer: "<red>{0} - Muted {1} for {2}"
+maxTimeExceeded: "<red>The specified time must be under a week."
 # 0 - The person who is locking up
 # 1 - The person who has been locked up
 lockedUpPlayer: "<aqua>{0} - Locking up {1}"
@@ -271,4 +278,4 @@ reappliedGamerules: "<aqua>All game rules have been re-applied!"
 commandNotFound: "<red>That command could not be found!"
 # 0 - The command
 # 1 - A list of aliases found
-commandAliases: "<aqua>Aliases for {0} are: {1}"
+commandAliases: "<aqua>Aliases for {0} are: {1}"
\ No newline at end of file