package com.ogzz.login.manager; import com.ogzz.login.OGzzLogin; import com.ogzz.login.model.PlayerData; import com.ogzz.login.model.Session; import net.kyori.adventure.text.minimessage.tag.resolver.Placeholder; import org.bukkit.Bukkit; import org.bukkit.Location; import org.bukkit.entity.Player; import org.bukkit.scheduler.BukkitTask; import java.util.HashMap; import java.util.Map; import java.util.UUID; import java.util.concurrent.ConcurrentHashMap; /** * Manages player authentication state and sessions. */ public class AuthManager { private final OGzzLogin plugin; private final Map authenticatedPlayers; private final Map activeSessions; private final Map loginAttempts; private final Map timeoutTasks; private final Map reminderTasks; private final Map savedLocations; public AuthManager(OGzzLogin plugin) { this.plugin = plugin; this.authenticatedPlayers = new ConcurrentHashMap<>(); this.activeSessions = new ConcurrentHashMap<>(); this.loginAttempts = new ConcurrentHashMap<>(); this.timeoutTasks = new HashMap<>(); this.reminderTasks = new HashMap<>(); this.savedLocations = new HashMap<>(); } /** * Checks if a player is authenticated. * * @param uuid the player's UUID * @return true if authenticated */ public boolean isAuthenticated(UUID uuid) { return authenticatedPlayers.getOrDefault(uuid, false); } /** * Sets a player's authentication status. * * @param uuid the player's UUID * @param authenticated the authentication status */ public void setAuthenticated(UUID uuid, boolean authenticated) { authenticatedPlayers.put(uuid, authenticated); if (authenticated) { loginAttempts.remove(uuid); cancelTimeoutTask(uuid); cancelReminderTask(uuid); } } /** * Removes a player from the authentication system (on quit). * * @param uuid the player's UUID */ public void removePlayer(UUID uuid) { authenticatedPlayers.remove(uuid); loginAttempts.remove(uuid); cancelTimeoutTask(uuid); cancelReminderTask(uuid); savedLocations.remove(uuid); } /** * Creates a new session for a player. * * @param uuid the player's UUID * @param ipAddress the player's IP address */ public void createSession(UUID uuid, String ipAddress) { long autoLoginDuration = plugin.getConfig().getLong("session.auto-login-duration", 300) * 1000; if (autoLoginDuration > 0) { long expiryTime = System.currentTimeMillis() + autoLoginDuration; Session session = new Session(uuid, ipAddress, System.currentTimeMillis(), expiryTime); activeSessions.put(uuid, session); } } /** * Gets a player's session. * * @param uuid the player's UUID * @return the session, or null if not found */ public Session getSession(UUID uuid) { Session session = activeSessions.get(uuid); if (session != null && !session.isValid()) { activeSessions.remove(uuid); return null; } return session; } /** * Removes a player's session. * * @param uuid the player's UUID */ public void removeSession(UUID uuid) { activeSessions.remove(uuid); } /** * Increments login attempts for a player. * * @param uuid the player's UUID * @return the new attempt count */ public int incrementLoginAttempts(UUID uuid) { int attempts = loginAttempts.getOrDefault(uuid, 0) + 1; loginAttempts.put(uuid, attempts); return attempts; } /** * Gets the number of login attempts for a player. * * @param uuid the player's UUID * @return the attempt count */ public int getLoginAttempts(UUID uuid) { return loginAttempts.getOrDefault(uuid, 0); } /** * Starts the login timeout task for a player. * * @param player the player */ public void startTimeoutTask(Player player) { int timeout = plugin.getConfig().getInt("security.login-timeout", 60); if (timeout > 0) { BukkitTask task = Bukkit.getScheduler().runTaskLater(plugin, () -> { if (!isAuthenticated(player.getUniqueId())) { plugin.getMessageUtil().sendMessage(player, "login-timeout"); player.kick(plugin.getMessageUtil().deserialize( plugin.getMessageUtil().getMessage("login-timeout") )); } }, timeout * 20L); timeoutTasks.put(player.getUniqueId(), task); } } /** * Cancels the timeout task for a player. * * @param uuid the player's UUID */ private void cancelTimeoutTask(UUID uuid) { BukkitTask task = timeoutTasks.remove(uuid); if (task != null) { task.cancel(); } } /** * Starts the action bar reminder task for a player. * * @param player the player * @param isRegistered whether the player is registered */ public void startReminderTask(Player player, boolean isRegistered) { int interval = plugin.getConfig().getInt("messages.actionbar-interval", 5); if (interval > 0) { String messageKey = isRegistered ? "actionbar-reminder-registered" : "actionbar-reminder-unregistered"; BukkitTask task = Bukkit.getScheduler().runTaskTimer(plugin, () -> { if (!isAuthenticated(player.getUniqueId())) { plugin.getMessageUtil().sendActionBar(player, messageKey); } }, interval * 20L, interval * 20L); reminderTasks.put(player.getUniqueId(), task); } } /** * Cancels the reminder task for a player. * * @param uuid the player's UUID */ private void cancelReminderTask(UUID uuid) { BukkitTask task = reminderTasks.remove(uuid); if (task != null) { task.cancel(); } } /** * Saves a player's location. * * @param uuid the player's UUID * @param location the location to save */ public void saveLocation(UUID uuid, Location location) { savedLocations.put(uuid, location); } /** * Gets a player's saved location. * * @param uuid the player's UUID * @return the saved location, or null if not found */ public Location getSavedLocation(UUID uuid) { return savedLocations.get(uuid); } /** * Handles post-login actions for a player. * * @param player the player */ public void handlePostLogin(Player player) { UUID uuid = player.getUniqueId(); // Mark as authenticated setAuthenticated(uuid, true); // Create session String ipAddress = player.getAddress().getAddress().getHostAddress(); createSession(uuid, ipAddress); // Update player data PlayerData playerData = new PlayerData(); playerData.setUuid(uuid); playerData.setUsername(player.getName()); playerData.setLastIp(ipAddress); playerData.setLastLogin(System.currentTimeMillis()); plugin.getDatabaseManager().loadPlayerData(uuid).thenAccept(data -> { if (data != null) { playerData.setHashedPassword(data.getHashedPassword()); playerData.setRegistrationDate(data.getRegistrationDate()); } plugin.getDatabaseManager().savePlayerData(playerData); }); // Teleport if configured if (plugin.getConfig().getBoolean("teleportation.spawn-after-login", false)) { Location spawn = player.getWorld().getSpawnLocation(); player.teleport(spawn); } else if (plugin.getConfig().getBoolean("teleportation.save-last-location", true)) { Location saved = getSavedLocation(uuid); if (saved != null) { player.teleport(saved); } } } /** * Shuts down the authentication manager. */ public void shutdown() { timeoutTasks.values().forEach(BukkitTask::cancel); reminderTasks.values().forEach(BukkitTask::cancel); timeoutTasks.clear(); reminderTasks.clear(); savedLocations.clear(); } }