Ported 1.17.1-2.1.1 to 1.18.1.

src/main/java/dev/micle/loginprotection/data/ProtectedPlayer.java

@@ -0,0 +1,37 @@
+package dev.micle.loginprotection.data;
+
+import dev.micle.loginprotection.setup.Config;
+
+import java.util.UUID;
+
+public class ProtectedPlayer {
+    private final UUID player_uuid;
+    private int grace_period;
+    private boolean is_loading;
+
+    public ProtectedPlayer(UUID player_uuid) {
+        this.player_uuid = player_uuid;
+        this.grace_period = (Config.Server.POST_GRACE_DURATION.get() * 40);
+        this.is_loading = true;
+    }
+
+    public UUID getPlayerUUID() {
+        return this.player_uuid;
+    }
+
+    public int getGracePeriod() {
+        return this.grace_period;
+    }
+
+    public void setGracePeriod(int new_grace_period) {
+        this.grace_period = new_grace_period;
+    }
+
+    public boolean isLoading() {
+        return this.is_loading;
+    }
+
+    public void setLoading(boolean new_loading) {
+        this.is_loading = new_loading;
+    }
+}

src/main/java/dev/micle/loginprotection/data/ProtectedPlayers.java

@@ -0,0 +1,58 @@
+package dev.micle.loginprotection.data;
+
+import dev.micle.loginprotection.setup.Config;
+import net.minecraft.network.chat.TextComponent;
+import net.minecraft.server.level.ServerPlayer;
+import net.minecraftforge.server.ServerLifecycleHooks;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+
+public class ProtectedPlayers {
+    private final List<ProtectedPlayer> protected_players = new ArrayList<>();
+
+    public ProtectedPlayers() { }
+
+    public int size() {
+        return protected_players.size();
+    }
+
+    public void addPlayer(UUID player_uuid) {
+        protected_players.add(new ProtectedPlayer(player_uuid));
+    }
+
+    public ProtectedPlayer getPlayer(UUID player_uuid) {
+        ProtectedPlayer player;
+        for (ProtectedPlayer protected_player : protected_players) {
+            player = protected_player;
+            if (player.getPlayerUUID() == player_uuid) {
+                return player;
+            }
+        }
+        return null;
+    }
+
+    public void removePlayer(UUID player_uuid) {
+        ProtectedPlayer protected_player = getPlayer(player_uuid);
+        if (protected_player == null) { return; }
+        protected_players.remove(protected_player);
+
+        ServerPlayer player = ServerLifecycleHooks.getCurrentServer().getPlayerList().getPlayer(player_uuid);
+        if (player == null) { return; }
+
+        if (!Config.Server.POST_GRACE_ENABLED.get()) { return; }
+        player.sendMessage(new TextComponent("[LoginProtection] Grace period ended!"), player_uuid);
+    }
+
+    public void updateGracePeriod(UUID player_uuid) {
+        ProtectedPlayer protected_player = getPlayer(player_uuid);
+        if (protected_player.isLoading()) { return; }
+
+        int grace_period = protected_player.getGracePeriod()-1;
+        protected_player.setGracePeriod(grace_period);
+        if (grace_period <= 0) {
+            removePlayer(player_uuid);
+        }
+    }
+}