From e7346e53d95b4791240e33d7623e6c19a613e6c1 Mon Sep 17 00:00:00 2001
From: micle <micle@micle.dev>
Date: Thu, 30 Sep 2021 22:26:14 +0100
Subject: [PATCH] Updated the grace period messages sent to players. Added a
 new message to notify player when he is seen as active. Removed old console
 message when a player is no longer protected.

---
 .../com/micle/loginprotection/data/ProtectedPlayers.java   | 7 ++++---
 .../com/micle/loginprotection/network/C2SKeyPress.java     | 3 ++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/micle/loginprotection/data/ProtectedPlayers.java b/src/main/java/com/micle/loginprotection/data/ProtectedPlayers.java
index 27ac120..4b32f1d 100755
--- a/src/main/java/com/micle/loginprotection/data/ProtectedPlayers.java
+++ b/src/main/java/com/micle/loginprotection/data/ProtectedPlayers.java
@@ -39,8 +39,10 @@ public class ProtectedPlayers {
         protected_players.remove(protected_player);
 
         ServerPlayerEntity player = ServerLifecycleHooks.getCurrentServer().getPlayerList().getPlayer(player_uuid);
-        if (player == null || !Config.POST_GRACE_ENABLED.get()) { return; }
-        player.sendMessage(new StringTextComponent("Grace Period over!"), player_uuid);
+        if (player == null) { return; }
+
+        if (!Config.POST_GRACE_ENABLED.get()) { return; }
+        player.sendMessage(new StringTextComponent("[LoginProtection] Grace period ended!"), player_uuid);
     }
 
     public void updateGracePeriod(UUID player_uuid) {
@@ -51,7 +53,6 @@ public class ProtectedPlayers {
         protected_player.setGracePeriod(grace_period);
         if (grace_period <= 0) {
             removePlayer(player_uuid);
-            System.out.println(protected_player.getPlayerUUID() + " is no longer being protected!");
         }
     }
 }
diff --git a/src/main/java/com/micle/loginprotection/network/C2SKeyPress.java b/src/main/java/com/micle/loginprotection/network/C2SKeyPress.java
index 393e329..6cff8ce 100755
--- a/src/main/java/com/micle/loginprotection/network/C2SKeyPress.java
+++ b/src/main/java/com/micle/loginprotection/network/C2SKeyPress.java
@@ -30,9 +30,10 @@ public class C2SKeyPress {
 
             if (!Config.POST_GRACE_ENABLED.get()) {
                 LoginProtection.protected_players.removePlayer(sender.getUUID());
+                sender.sendMessage(new StringTextComponent("[LoginProtection] You are now seen as active."), sender.getUUID());
             } else {
                 LoginProtection.protected_players.getPlayer(sender.getUUID()).setLoading(false);
-                sender.sendMessage(new StringTextComponent("Grace Period started!"), sender.getUUID());
+                sender.sendMessage(new StringTextComponent("[LoginProtection] Grace period started!"), sender.getUUID());
             }
             if (sender.isInWater()) {
                 if (Config.POST_DROWN_ENABLED.get()) {