Remove session in ProtocolSupport directly without expiring

bukkit/src/main/java/com/github/games647/fastlogin/bukkit/BukkitLoginSession.java

@@ -51,12 +51,16 @@ public class BukkitLoginSession extends LoginSession {
         return ArrayUtils.clone(verifyToken);
     }
 
+    /**
+     * @return premium skin if available
+     */
     public synchronized Optional<SkinProperty> getSkin() {
         return Optional.ofNullable(skinProperty);
     }
 
     /**
      * Sets the premium skin property which was retrieved by the session server
+     * @param skinProperty premium skin
      */
     public synchronized void setSkinProperty(SkinProperty skinProperty) {
         this.skinProperty = skinProperty;

bukkit/src/main/java/com/github/games647/fastlogin/bukkit/FastLoginBukkit.java

@@ -152,7 +152,7 @@ public class FastLoginBukkit extends JavaPlugin implements PlatformPlugin<Comman
      * Wait before the server is fully started. This is workaround, because connections right on startup are not
      * injected by ProtocolLib
      *
-     * @return
+     * @return true if ProtocolLib can now intercept packets
      */
     public boolean isServerFullyStarted() {
         return serverStarted;

bukkit/src/main/java/com/github/games647/fastlogin/bukkit/listener/protocolsupport/ProtocolSupportListener.java

@@ -12,6 +12,7 @@ import org.bukkit.command.CommandSender;
 import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.Listener;
+import protocolsupport.api.events.ConnectionCloseEvent;
 import protocolsupport.api.events.PlayerLoginStartEvent;
 import protocolsupport.api.events.PlayerPropertiesResolveEvent;
 
@@ -41,6 +42,12 @@ public class ProtocolSupportListener extends JoinManagement<Player, CommandSende
         super.onLogin(username, new ProtocolLoginSource(loginStartEvent));
     }
 
+    @EventHandler
+    public void onConnectionClosed(ConnectionCloseEvent closeEvent) {
+        InetSocketAddress address = closeEvent.getConnection().getAddress();
+        plugin.getLoginSessions().remove(address.toString());
+    }
+
     @EventHandler
     public void onPropertiesResolve(PlayerPropertiesResolveEvent propertiesResolveEvent) {
         InetSocketAddress address = propertiesResolveEvent.getAddress();

bungee/src/main/java/com/github/games647/fastlogin/bungee/listener/MessageListener.java

@@ -64,7 +64,7 @@ public class MessageListener implements Listener {
 
             String playerName = changeMessage.getPlayerName();
             boolean isSourceInvoker = changeMessage.isSourceInvoker();
-            if (changeMessage.isWillEnable()) {
+            if (changeMessage.shouldEnable()) {
                 if (playerName.equals(forPlayer.getName()) && plugin.getCore().getConfig().get("premium-warning", true)
                         && !core.getPendingConfirms().contains(forPlayer.getUniqueId())) {
                     String message = core.getMessage("premium-warning");

core/src/main/java/com/github/games647/fastlogin/core/messages/ChangePremiumMessage.java

@@ -23,7 +23,7 @@ public class ChangePremiumMessage implements ChannelMessage {
         return playerName;
     }
 
-    public boolean isWillEnable() {
+    public boolean shouldEnable() {
         return willEnable;
     }
 
@@ -54,7 +54,7 @@ public class ChangePremiumMessage implements ChannelMessage {
     public String toString() {
         return this.getClass().getSimpleName() + '{' +
                 "playerName='" + playerName + '\'' +
-                ", willEnable=" + willEnable +
+                ", shouldEnable=" + willEnable +
                 ", isSourceInvoker=" + isSourceInvoker +
                 '}';
     }